RecoParticleFlow/PFTracking/src/ConvBremHeavyObjectCache.cc - master - githubmirror/cmssw

RecoParticleFlow/PFTracking/src/ConvBremHeavyObjectCache.cc

22 строки1 KB

Cms Build

Clang-Format

03 июн 2019, 06:44

03 июн 2019, 06:447622cf3

#include "CommonTools/MVAUtils/interface/GBRForestTools.h"
#include "FWCore/ParameterSet/interface/FileInPath.h"
#include "RecoParticleFlow/PFTracking/interface/ConvBremHeavyObjectCache.h"

namespace convbremhelpers {
  HeavyObjectCache::HeavyObjectCache(const edm::ParameterSet& conf) {
    pfcalib_ = std::make_unique<PFEnergyCalibration>();

    const bool useConvBremFinder_ = conf.getParameter<bool>("useConvBremFinder");

    if (useConvBremFinder_) {
      gbrBarrelLowPt_ =
          createGBRForest(conf.getParameter<edm::FileInPath>("pf_convBremFinderID_mvaWeightFileBarrelLowPt"));
      gbrBarrelHighPt_ =
          createGBRForest(conf.getParameter<edm::FileInPath>("pf_convBremFinderID_mvaWeightFileBarrelHighPt"));
      gbrEndcapsLowPt_ =
          createGBRForest(conf.getParameter<edm::FileInPath>("pf_convBremFinderID_mvaWeightFileEndcapsLowPt"));
      gbrEndcapsHighPt_ =
          createGBRForest(conf.getParameter<edm::FileInPath>("pf_convBremFinderID_mvaWeightFileEndcapsHighPt"));
    }
  }
}  // namespace convbremhelpers