CMS 3D CMS Logo

List of all members | Public Member Functions | Static Public Member Functions | Private Member Functions | Private Attributes
ElectronIdMVABased Class Reference

#include <MyAnalyzer/ElectronIdMVABased/src/ElectronIdMVABased.cc>

Inheritance diagram for ElectronIdMVABased:
edm::stream::EDProducer< edm::GlobalCache< gsfidhelper::HeavyObjectCache > >

Public Member Functions

 ElectronIdMVABased (const edm::ParameterSet &, const gsfidhelper::HeavyObjectCache *)
 
 ~ElectronIdMVABased () override
 
- Public Member Functions inherited from edm::stream::EDProducer< edm::GlobalCache< gsfidhelper::HeavyObjectCache > >
 EDProducer ()=default
 
bool hasAbilityToProduceInLumis () const final
 
bool hasAbilityToProduceInRuns () const final
 

Static Public Member Functions

static void globalEndJob (gsfidhelper::HeavyObjectCache const *)
 
static std::unique_ptr< gsfidhelper::HeavyObjectCacheinitializeGlobalCache (const edm::ParameterSet &conf)
 

Private Member Functions

void produce (edm::Event &, const edm::EventSetup &) override
 

Private Attributes

edm::EDGetTokenT< reco::GsfElectronCollectionelectronToken
 
std::vector< string > mvaWeightFileEleID
 
string path_mvaWeightFileEleID
 
double thresholdBarrel
 
double thresholdEndcap
 
double thresholdIsoBarrel
 
double thresholdIsoEndcap
 
edm::EDGetTokenT< reco::VertexCollectionvertexToken
 

Additional Inherited Members

- Public Types inherited from edm::stream::EDProducer< edm::GlobalCache< gsfidhelper::HeavyObjectCache > >
typedef CacheContexts< T... > CacheTypes
 
typedef CacheTypes::GlobalCache GlobalCache
 
typedef AbilityChecker< T... > HasAbility
 
typedef CacheTypes::LuminosityBlockCache LuminosityBlockCache
 
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCacheLuminosityBlockContext
 
typedef CacheTypes::LuminosityBlockSummaryCache LuminosityBlockSummaryCache
 
typedef CacheTypes::RunCache RunCache
 
typedef RunContextT< RunCache, GlobalCacheRunContext
 
typedef CacheTypes::RunSummaryCache RunSummaryCache
 

Detailed Description

Description: [one line class summary]

Implementation: [Notes on implementation]

Definition at line 57 of file ElectronIdMVABased.cc.

Constructor & Destructor Documentation

ElectronIdMVABased::ElectronIdMVABased ( const edm::ParameterSet iConfig,
const gsfidhelper::HeavyObjectCache  
)
explicit

Definition at line 97 of file ElectronIdMVABased.cc.

References edm::ParameterSet::getParameter(), electronIdMVABased_cfi::thresholdBarrel, and electronIdMVABased_cfi::thresholdEndcap.

97  {
98  vertexToken = consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("vertexTag"));
99  electronToken = consumes<reco::GsfElectronCollection>(iConfig.getParameter<edm::InputTag>("electronTag"));
100  thresholdBarrel = iConfig.getParameter<double>("thresholdBarrel");
101  thresholdEndcap = iConfig.getParameter<double>("thresholdEndcap");
102  thresholdIsoBarrel = iConfig.getParameter<double>("thresholdIsoDR03Barrel");
103  thresholdIsoEndcap = iConfig.getParameter<double>("thresholdIsoDR03Endcap");
104 
105  produces<reco::GsfElectronCollection>();
106 }
T getParameter(std::string const &) const
edm::EDGetTokenT< reco::VertexCollection > vertexToken
edm::EDGetTokenT< reco::GsfElectronCollection > electronToken
ElectronIdMVABased::~ElectronIdMVABased ( )
override

Definition at line 109 of file ElectronIdMVABased.cc.

110 {
111  // do anything here that needs to be done at desctruction time
112  // (e.g. close files, deallocate resources etc.)
113 
114 }

Member Function Documentation

static void ElectronIdMVABased::globalEndJob ( gsfidhelper::HeavyObjectCache const *  )
inlinestatic

Definition at line 68 of file ElectronIdMVABased.cc.

68  {
69  }
static std::unique_ptr<gsfidhelper::HeavyObjectCache> ElectronIdMVABased::initializeGlobalCache ( const edm::ParameterSet conf)
inlinestatic

Definition at line 64 of file ElectronIdMVABased.cc.

64  {
65  return std::make_unique<gsfidhelper::HeavyObjectCache>(conf);
66  }
void ElectronIdMVABased::produce ( edm::Event iEvent,
const edm::EventSetup iSetup 
)
overrideprivate

Definition at line 122 of file ElectronIdMVABased.cc.

References constexpr, DEFINE_FWK_MODULE, edm::Event::getByToken(), eostools::move(), reco::GsfElectron::MvaOutput::mva_Isolated, electronIdMVABased_cfi::mvaElectrons, edm::Handle< T >::product(), edm::Event::put(), electronIdMVABased_cfi::thresholdBarrel, electronIdMVABased_cfi::thresholdEndcap, and particleFlowSuperClusterECAL_cfi::vertexCollection.

122  {
123  using namespace edm;
124 
125  constexpr double etaEBEE = 1.485;
126 
127  auto mvaElectrons = std::make_unique<reco::GsfElectronCollection>();
128 
130  iEvent.getByToken(vertexToken, vertexCollection);
131  int nVtx = vertexCollection->size();
132 
134  iEvent.getByToken(electronToken,egCollection);
135  const reco::GsfElectronCollection egCandidates = (*egCollection.product());
136  for ( reco::GsfElectronCollection::const_iterator egIter = egCandidates.begin(); egIter != egCandidates.end(); ++egIter) {
137  double mvaVal = globalCache()->mvaID_->mva( *egIter, nVtx );
138  double isoDr03 = egIter->dr03TkSumPt() + egIter->dr03EcalRecHitSumEt() + egIter->dr03HcalTowerSumEt();
139  double eleEta = fabs(egIter->eta());
140  if (eleEta <= etaEBEE && mvaVal > thresholdBarrel && isoDr03 < thresholdIsoBarrel) {
141  mvaElectrons->push_back( *egIter );
142  reco::GsfElectron::MvaOutput myMvaOutput;
143  myMvaOutput.mva_Isolated = mvaVal;
144  mvaElectrons->back().setMvaOutput(myMvaOutput);
145  }
146  else if (eleEta > etaEBEE && mvaVal > thresholdEndcap && isoDr03 < thresholdIsoEndcap) {
147  mvaElectrons->push_back( *egIter );
148  reco::GsfElectron::MvaOutput myMvaOutput;
149  myMvaOutput.mva_Isolated = mvaVal;
150  mvaElectrons->back().setMvaOutput(myMvaOutput);
151  }
152  }
153 
154  iEvent.put(std::move(mvaElectrons));
155 
156 }
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:137
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:579
std::vector< GsfElectron > GsfElectronCollection
collection of GsfElectron objects
#define constexpr
edm::EDGetTokenT< reco::VertexCollection > vertexToken
edm::EDGetTokenT< reco::GsfElectronCollection > electronToken
T const * product() const
Definition: Handle.h:81
HLT enums.
def move(src, dest)
Definition: eostools.py:511

Member Data Documentation

edm::EDGetTokenT<reco::GsfElectronCollection> ElectronIdMVABased::electronToken
private

Definition at line 77 of file ElectronIdMVABased.cc.

std::vector<string> ElectronIdMVABased::mvaWeightFileEleID
private

Definition at line 78 of file ElectronIdMVABased.cc.

string ElectronIdMVABased::path_mvaWeightFileEleID
private

Definition at line 79 of file ElectronIdMVABased.cc.

double ElectronIdMVABased::thresholdBarrel
private

Definition at line 80 of file ElectronIdMVABased.cc.

double ElectronIdMVABased::thresholdEndcap
private

Definition at line 81 of file ElectronIdMVABased.cc.

double ElectronIdMVABased::thresholdIsoBarrel
private

Definition at line 82 of file ElectronIdMVABased.cc.

double ElectronIdMVABased::thresholdIsoEndcap
private

Definition at line 83 of file ElectronIdMVABased.cc.

edm::EDGetTokenT<reco::VertexCollection> ElectronIdMVABased::vertexToken
private

Definition at line 76 of file ElectronIdMVABased.cc.