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::EDFilter< edm::GlobalCache< gsfidhelper::HeavyObjectCache > > edm::stream::EDFilterBase edm::ProducerBase edm::EDConsumerBase edm::ProductRegistryHelper

Public Member Functions

 ElectronIdMVABased (const edm::ParameterSet &, const gsfidhelper::HeavyObjectCache *)
 
 ~ElectronIdMVABased ()
 
- Public Member Functions inherited from edm::stream::EDFilter< edm::GlobalCache< gsfidhelper::HeavyObjectCache > >
 EDFilter ()=default
 
- Public Member Functions inherited from edm::stream::EDFilterBase
 EDFilterBase ()
 
ModuleDescription const & moduleDescription () const
 
virtual ~EDFilterBase ()
 
- Public Member Functions inherited from edm::ProducerBase
void callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
 
std::vector< edm::ProductResolverIndex > const & indiciesForPutProducts (BranchType iBranchType) const
 
 ProducerBase ()
 
void registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
 
std::function< void(BranchDescription const &)> registrationCallback () const
 used by the fwk to register list of products More...
 
void resolvePutIndicies (BranchType iBranchType, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)
 
virtual ~ProducerBase () noexcept(false)
 
- Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfoconsumesInfo () const
 
void convertCurrentProcessAlias (std::string const &processName)
 Convert "@currentProcess" in InputTag process names to the actual current process name. More...
 
 EDConsumerBase ()
 
 EDConsumerBase (EDConsumerBase const &)=delete
 
 EDConsumerBase (EDConsumerBase &&)=default
 
ProductResolverIndexAndSkipBit indexFrom (EDGetToken, BranchType, TypeID const &) const
 
void itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
void itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
std::vector< ProductResolverIndexAndSkipBit > const & itemsToGetFrom (BranchType iType) const
 
void labelsForToken (EDGetToken iToken, Labels &oLabels) const
 
void modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
 
EDConsumerBase const & operator= (EDConsumerBase const &)=delete
 
EDConsumerBaseoperator= (EDConsumerBase &&)=default
 
bool registeredToConsume (ProductResolverIndex, bool, BranchType) const
 
bool registeredToConsumeMany (TypeID const &, BranchType) const
 
void updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
 
virtual ~EDConsumerBase () noexcept(false)
 

Static Public Member Functions

static void globalEndJob (gsfidhelper::HeavyObjectCache const *)
 
static std::unique_ptr< gsfidhelper::HeavyObjectCacheinitializeGlobalCache (const edm::ParameterSet &conf)
 
- Static Public Member Functions inherited from edm::stream::EDFilterBase
static const std::string & baseType ()
 
static void fillDescriptions (ConfigurationDescriptions &descriptions)
 
static void prevalidate (ConfigurationDescriptions &descriptions)
 

Private Member Functions

virtual bool filter (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::EDFilter< 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
 
- Public Types inherited from edm::stream::EDFilterBase
typedef EDFilterAdaptorBase ModuleType
 
- Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList TypeLabelList
 
- Public Types inherited from edm::EDConsumerBase
typedef ProductLabels Labels
 
- Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > consumes (edm::InputTag const &tag)
 
EDGetToken consumes (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken consumes (TypeToGet const &id, edm::InputTag const &tag)
 
ConsumesCollector consumesCollector ()
 Use a ConsumesCollector to gather consumes information from helper functions. More...
 
template<typename ProductType , BranchType B = InEvent>
void consumesMany ()
 
void consumesMany (const TypeToGet &id)
 
template<BranchType B>
void consumesMany (const TypeToGet &id)
 
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > mayConsume (edm::InputTag const &tag)
 
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 

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 ( )

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

bool ElectronIdMVABased::filter ( edm::Event iEvent,
const edm::EventSetup iSetup 
)
overrideprivatevirtual

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  return true;
157 }
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:122
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:460
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:510
static void ElectronIdMVABased::globalEndJob ( gsfidhelper::HeavyObjectCache const *  )
inlinestatic

Definition at line 68 of file ElectronIdMVABased.cc.

References ALCARECOTkAlBeamHalo_cff::filter.

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  }

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.