CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
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)
 
 ProducerBase ()
 
void registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
 
std::function< void(BranchDescription
const &)> 
registrationCallback () const
 used by the fwk to register list of products More...
 
virtual ~ProducerBase ()
 
- Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfoconsumesInfo () const
 
 EDConsumerBase ()
 
ProductHolderIndexAndSkipBit indexFrom (EDGetToken, BranchType, TypeID const &) const
 
void itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
 
void itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
 
std::vector
< ProductHolderIndexAndSkipBit >
const & 
itemsToGetFromEvent () const
 
void labelsForToken (EDGetToken iToken, Labels &oLabels) const
 
void modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const
 
void modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
 
bool registeredToConsume (ProductHolderIndex, bool, BranchType) const
 
bool registeredToConsumeMany (TypeID const &, BranchType) const
 
void updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)
 
virtual ~EDConsumerBase ()
 

Static Public Member Functions

static void globalEndJob (gsfidhelper::HeavyObjectCache const *)
 
static std::unique_ptr
< gsfidhelper::HeavyObjectCache
initializeGlobalCache (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::GsfElectronCollection
electronToken
 
std::vector< string > mvaWeightFileEleID
 
string path_mvaWeightFileEleID
 
double thresholdBarrel
 
double thresholdEndcap
 
double thresholdIsoBarrel
 
double thresholdIsoEndcap
 
edm::EDGetTokenT
< reco::VertexCollection
vertexToken
 

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, GlobalCache
LuminosityBlockContext
 
typedef
CacheTypes::LuminosityBlockSummaryCache 
LuminosityBlockSummaryCache
 
typedef CacheTypes::RunCache RunCache
 
typedef RunContextT< RunCache,
GlobalCache
RunContext
 
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().

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

Implements edm::stream::EDFilterBase.

Definition at line 122 of file ElectronIdMVABased.cc.

References constexpr, edm::Event::getByToken(), reco::GsfElectron::MvaOutput::mva_Isolated, electronIdMVABased_cfi::mvaElectrons, edm::Event::put(), and GoodVertex_cfg::vertexCollection.

122  {
123  using namespace edm;
124 
125  constexpr double etaEBEE = 1.485;
126 
127  std::auto_ptr<reco::GsfElectronCollection> mvaElectrons(new reco::GsfElectronCollection);
128 
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(mvaElectrons);
155 
156  return true;
157 }
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:464
tuple vertexCollection
std::vector< GsfElectron > GsfElectronCollection
collection of GsfElectron objects
#define constexpr
OrphanHandle< PROD > put(std::auto_ptr< PROD > product)
Put a new product.
Definition: Event.h:120
edm::EDGetTokenT< reco::VertexCollection > vertexToken
edm::EDGetTokenT< reco::GsfElectronCollection > electronToken
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::unique_ptr<gsfidhelper::HeavyObjectCache>(new 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.