ElectronIdMVAProducer Class Reference

Inheritance diagram for ElectronIdMVAProducer:

Public Member Functions
	ElectronIdMVAProducer (const edm::ParameterSet &)
	~ElectronIdMVAProducer () override
Public Member Functions inherited from edm::EDFilter
	EDFilter ()
SerialTaskQueue *	globalLuminosityBlocksQueue ()
SerialTaskQueue *	globalRunsQueue ()
ModuleDescription const &	moduleDescription () const
	~EDFilter () override
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 ()
std::vector< edm::ProductResolverIndex > const &	putTokenIndexToProductResolverIndex () const
std::vector< bool > const &	recordProvenanceList () const
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, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)
TypeLabelList const &	typeLabelList () const
	used by the fwk to register the list of products of this module More...
	~ProducerBase () noexcept(false) override
Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const
void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...
	EDConsumerBase ()
	EDConsumerBase (EDConsumerBase &&)=default
	EDConsumerBase (EDConsumerBase const &)=delete
ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const
std::vector< ESProxyIndex > const &	esGetTokenIndicesVector (edm::Transition iTrans) const
std::vector< ESRecordIndex > const &	esGetTokenRecordIndicesVector (edm::Transition iTrans) const
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::array< std::vector< ModuleDescription const * > *, NumBranchTypes > &modulesAll, std::vector< ModuleProcessName > &modulesInPreviousProcesses, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
EDConsumerBase &	operator= (EDConsumerBase &&)=default
EDConsumerBase const &	operator= (EDConsumerBase const &)=delete
bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
void	selectInputProcessBlocks (ProductRegistry const &productRegistry, ProcessBlockHelperBase const &processBlockHelperBase)
ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const
void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)
virtual	~EDConsumerBase () noexcept(false)

Private Member Functions
bool	filter (edm::Event &, const edm::EventSetup &) override

Private Attributes
double	_Rho
const EcalClusterLazyTools::ESGetTokens	ecalClusterToolsESGetTokens_
edm::EDGetTokenT< reco::GsfElectronCollection >	electronToken_
edm::EDGetTokenT< double >	eventrhoToken_
std::string	method_
EGammaMvaEleEstimator *	mvaID_
std::vector< std::string >	mvaWeightFiles_
bool	NoIP_
edm::EDGetTokenT< EcalRecHitCollection >	reducedEBRecHitCollectionToken_
edm::EDGetTokenT< EcalRecHitCollection >	reducedEERecHitCollectionToken_
bool	Trig_
bool	verbose_
edm::EDGetTokenT< reco::VertexCollection >	vertexToken_

Additional Inherited Members
Public Types inherited from edm::EDFilter
typedef EDFilter	ModuleType
Public Types inherited from edm::ProducerBase
using	ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const *, const char *, edm::ProductResolverIndex > >
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList
Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels
Static Public Member Functions inherited from edm::EDFilter
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &)
static bool	wantsGlobalLuminosityBlocks ()
static bool	wantsGlobalRuns ()
static bool	wantsInputProcessBlocks ()
static bool	wantsProcessBlocks ()
static bool	wantsStreamLuminosityBlocks ()
static bool	wantsStreamRuns ()
Protected Member Functions inherited from edm::ProducerBase
template<class ProductType >
BranchAliasSetterT< ProductType >	produces ()
	declare what type of product will make and with which optional label More...
template<typename ProductType , BranchType B>
BranchAliasSetterT< ProductType >	produces ()
template<typename ProductType , Transition B>
BranchAliasSetterT< ProductType >	produces ()
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)
template<BranchType B>
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)
template<Transition B>
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)
template<typename ProductType , Transition B>
BranchAliasSetterT< ProductType >	produces (std::string instanceName)
template<class ProductType >
BranchAliasSetterT< ProductType >	produces (std::string instanceName)
template<typename ProductType , BranchType B>
BranchAliasSetterT< ProductType >	produces (std::string instanceName)
ProducesCollector	producesCollector ()
Protected Member Functions inherited from edm::EDConsumerBase
EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)
template<BranchType B = InEvent>
EDConsumerBaseAdaptor< B >	consumes (edm::InputTag tag) noexcept
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 ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()
template<Transition Tr = Transition::Event>
constexpr auto	esConsumes () noexcept
template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)
template<Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag tag) noexcept
template<Transition Tr = Transition::Event>
ESGetTokenGeneric	esConsumes (eventsetup::EventSetupRecordKey const &iRecord, eventsetup::DataKey const &iKey)
	Used with EventSetupRecord::doGet. More...
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)
template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

Detailed Description

Definition at line 24 of file ElectronIdMVAProducer.cc.

Constructor & Destructor Documentation

ElectronIdMVAProducer()

ElectronIdMVAProducer::ElectronIdMVAProducer ( const edm::ParameterSet &  iConfig )

explicit

Definition at line 61 of file ElectronIdMVAProducer.cc.

    : ecalClusterToolsESGetTokens_{consumesCollector()} {
  verbose_ = iConfig.getUntrackedParameter<bool>("verbose", false);
  vertexToken_ = consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("vertexTag"));
  electronToken_ = consumes<reco::GsfElectronCollection>(iConfig.getParameter<edm::InputTag>("electronTag"));
  eventrhoToken_ = consumes<double>(edm::InputTag("kt6PFJets", "rho"));
  reducedEBRecHitCollectionToken_ =
      consumes<EcalRecHitCollection>(iConfig.getParameter<edm::InputTag>("reducedEBRecHitCollection"));
  reducedEERecHitCollectionToken_ =
      consumes<EcalRecHitCollection>(iConfig.getParameter<edm::InputTag>("reducedEERecHitCollection"));
  method_ = iConfig.getParameter<std::string>("method");
  std::vector<std::string> fpMvaWeightFiles = iConfig.getParameter<std::vector<std::string> >("mvaWeightFile");
  Trig_ = iConfig.getParameter<bool>("Trig");
  NoIP_ = iConfig.getParameter<bool>("NoIP");
 
  produces<edm::ValueMap<float> >("");
 
  mvaID_ = new EGammaMvaEleEstimator();
 
  EGammaMvaEleEstimator::MVAType type_;
  if (Trig_ && !NoIP_) {
    type_ = EGammaMvaEleEstimator::kTrig;
  }
 
  if (Trig_ && NoIP_) {
    type_ = EGammaMvaEleEstimator::kTrigNoIP;
  }
 
  if (!Trig_) {
    type_ = EGammaMvaEleEstimator::kNonTrig;
  }
 
  bool manualCat_ = true;
 
  std::string path_mvaWeightFileEleID;
  for (unsigned ifile = 0; ifile < fpMvaWeightFiles.size(); ++ifile) {
    path_mvaWeightFileEleID = edm::FileInPath(fpMvaWeightFiles[ifile].c_str()).fullPath();
    mvaWeightFiles_.push_back(path_mvaWeightFileEleID);
  }
 
  mvaID_->initialize(method_, type_, manualCat_, mvaWeightFiles_);
}

References edm::EDConsumerBase::consumesCollector().

~ElectronIdMVAProducer()

ElectronIdMVAProducer::~ElectronIdMVAProducer ( )

override

Definition at line 104 of file ElectronIdMVAProducer.cc.

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

Member Function Documentation

filter()

bool ElectronIdMVAProducer::filter ( edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

overrideprivatevirtual

Implements edm::EDFilter.

Definition at line 114 of file ElectronIdMVAProducer.cc.

                                                                                {
  std::unique_ptr<edm::ValueMap<float> > out(new edm::ValueMap<float>());
 
  edm::Handle<reco::VertexCollection> vertexCollection;
  iEvent.getByToken(vertexToken_, vertexCollection);
 
  reco::Vertex dummy;
  const reco::Vertex* pv = &dummy;
  if (!vertexCollection->empty()) {
    pv = &*vertexCollection->begin();
  } else {  // create a dummy PV
    reco::Vertex::Error e;
    e(0, 0) = 0.0015 * 0.0015;
    e(1, 1) = 0.0015 * 0.0015;
    e(2, 2) = 15. * 15.;
    reco::Vertex::Point p(0, 0, 0);
    dummy = reco::Vertex(p, e, 0, 0, 0);
  }
 
  EcalClusterLazyTools lazyTools(iEvent,
                                 ecalClusterToolsESGetTokens_.get(iSetup),
                                 reducedEBRecHitCollectionToken_,
                                 reducedEERecHitCollectionToken_);
 
  edm::ESHandle<TransientTrackBuilder> builder;
  iSetup.get<TransientTrackRecord>().get("TransientTrackBuilder", builder);
  TransientTrackBuilder thebuilder = *(builder.product());
 
  edm::Handle<reco::GsfElectronCollection> egCollection;
  iEvent.getByToken(electronToken_, egCollection);
  const reco::GsfElectronCollection egCandidates = (*egCollection.product());
 
  _Rho = 0;
  edm::Handle<double> rhoPtr;
  iEvent.getByToken(eventrhoToken_, rhoPtr);
  _Rho = *rhoPtr;
 
  std::vector<float> values;
  values.reserve(egCollection->size());
 
  for (reco::GsfElectronCollection::const_iterator egIter = egCandidates.begin(); egIter != egCandidates.end();
       ++egIter) {
    double mvaVal = -999999;
    if (!NoIP_) {
      mvaVal = mvaID_->mvaValue(*egIter, *pv, thebuilder, lazyTools, verbose_);
    }
    if (NoIP_) {
      mvaVal = mvaID_->mvaValue(*egIter, *pv, _Rho, /*thebuilder,*/ lazyTools, verbose_);
    }
 
    values.push_back(mvaVal);
  }
 
  edm::ValueMap<float>::Filler filler(*out);
  filler.insert(egCollection, values.begin(), values.end());
  filler.fill();
 
  iEvent.put(std::move(out));
 
  return true;
}

References _Rho, MillePedeFileConverter_cfg::e, ecalClusterToolsESGetTokens_, electronToken_, eventrhoToken_, trigObjTnPSource_cfi::filler, EcalClusterLazyToolsBase::ESGetTokens::get(), edm::EventSetup::get(), get, iEvent, eostools::move(), mvaID_, EGammaMvaEleEstimator::mvaValue(), NoIP_, MillePedeFileConverter_cfg::out, AlCaHLTBitMon_ParallelJobs::p, edm::Handle< T >::product(), MetAnalyzer::pv(), reducedEBRecHitCollectionToken_, reducedEERecHitCollectionToken_, contentValuesCheck::values, verbose_, HltBtagValidation_cff::Vertex, spclusmultinvestigator_cfi::vertexCollection, and vertexToken_.

Member Data Documentation

_Rho

double ElectronIdMVAProducer::_Rho

private

Definition at line 41 of file ElectronIdMVAProducer.cc.

Referenced by filter().

ecalClusterToolsESGetTokens_

const EcalClusterLazyTools::ESGetTokens ElectronIdMVAProducer::ecalClusterToolsESGetTokens_

private

Definition at line 39 of file ElectronIdMVAProducer.cc.

Referenced by filter().

electronToken_

edm::EDGetTokenT<reco::GsfElectronCollection> ElectronIdMVAProducer::electronToken_

private

Definition at line 35 of file ElectronIdMVAProducer.cc.

Referenced by filter().

eventrhoToken_

edm::EDGetTokenT<double> ElectronIdMVAProducer::eventrhoToken_

private

Definition at line 36 of file ElectronIdMVAProducer.cc.

Referenced by filter().

method_

std::string ElectronIdMVAProducer::method_

private

Definition at line 42 of file ElectronIdMVAProducer.cc.

mvaID_

EGammaMvaEleEstimator* ElectronIdMVAProducer::mvaID_

private

Definition at line 47 of file ElectronIdMVAProducer.cc.

Referenced by filter().

mvaWeightFiles_

std::vector<std::string> ElectronIdMVAProducer::mvaWeightFiles_

private

Definition at line 43 of file ElectronIdMVAProducer.cc.

NoIP_

bool ElectronIdMVAProducer::NoIP_

private

Definition at line 45 of file ElectronIdMVAProducer.cc.

Referenced by filter().

reducedEBRecHitCollectionToken_

edm::EDGetTokenT<EcalRecHitCollection> ElectronIdMVAProducer::reducedEBRecHitCollectionToken_

private

Definition at line 37 of file ElectronIdMVAProducer.cc.

Referenced by filter().

reducedEERecHitCollectionToken_

edm::EDGetTokenT<EcalRecHitCollection> ElectronIdMVAProducer::reducedEERecHitCollectionToken_

private

Definition at line 38 of file ElectronIdMVAProducer.cc.

Referenced by filter().

Trig_

bool ElectronIdMVAProducer::Trig_

private

Definition at line 44 of file ElectronIdMVAProducer.cc.

verbose_

bool ElectronIdMVAProducer::verbose_

private

Definition at line 33 of file ElectronIdMVAProducer.cc.

Referenced by filter().

vertexToken_

edm::EDGetTokenT<reco::VertexCollection> ElectronIdMVAProducer::vertexToken_

private

Definition at line 34 of file ElectronIdMVAProducer.cc.

Referenced by filter().