#include <GEDGsfElectronProducer.h>

Inheritance diagram for GEDGsfElectronProducer:

Public Member Functions
	GEDGsfElectronProducer (const edm::ParameterSet &, const gsfAlgoHelpers::HeavyObjectCache *)

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

Public Member Functions inherited from GsfElectronBaseProducer
	GsfElectronBaseProducer (const edm::ParameterSet &, const gsfAlgoHelpers::HeavyObjectCache *)

	~GsfElectronBaseProducer () override

Public Member Functions inherited from edm::stream::EDProducer< edm::GlobalCache< gsfAlgoHelpers::HeavyObjectCache > >
	EDProducer ()=default

bool	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () const final

Private Member Functions
void	fillGsfElectronValueMap (edm::Event &event, edm::ValueMap< reco::GsfElectronRef >::Filler &filler)

void	matchWithPFCandidates (edm::Event &event)

void	setMVAOutputs (reco::GsfElectronCollection &electrons, const gsfAlgoHelpers::HeavyObjectCache *, const std::map< reco::GsfTrackRef, reco::GsfElectron::MvaOutput > &mvaOutputs, reco::VertexCollection const &vertices) const

Private Attributes
edm::EDGetTokenT< reco::PFCandidateCollection >	egmPFCandidateCollection_

std::map< reco::GsfTrackRef, reco::GsfElectron::MvaInput >	gsfMVAInputMap_

std::map< reco::GsfTrackRef, reco::GsfElectron::MvaOutput >	gsfMVAOutputMap_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer< edm::GlobalCache< gsfAlgoHelpers::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

Static Public Member Functions inherited from GsfElectronBaseProducer
static void	fillDescriptions (edm::ConfigurationDescriptions &)

static void	globalEndJob (gsfAlgoHelpers::HeavyObjectCache const *)

static std::unique_ptr< gsfAlgoHelpers::HeavyObjectCache >	initializeGlobalCache (const edm::ParameterSet &conf)

Protected Member Functions inherited from GsfElectronBaseProducer
void	beginEvent (edm::Event &, const edm::EventSetup &)

void	fillEvent (reco::GsfElectronCollection &electrons, edm::Event &event)

const edm::OrphanHandle< reco::GsfElectronCollection > &	orphanHandle () const

Protected Attributes inherited from GsfElectronBaseProducer
GsfElectronAlgo *	algo_

const GsfElectronAlgo::CutsConfiguration	cutsCfg_

const GsfElectronAlgo::CutsConfiguration	cutsCfgPflow_

ElectronHcalHelper::Configuration	hcalCfg_

ElectronHcalHelper::Configuration	hcalCfgPflow_

GsfElectronAlgo::InputTagsConfiguration	inputCfg_

edm::EDGetTokenT< edm::ValueMap< float > >	pfMVA_

GsfElectronAlgo::StrategyConfiguration	strategyCfg_

Detailed Description

Definition at line 9 of file GEDGsfElectronProducer.h.

Constructor & Destructor Documentation

GEDGsfElectronProducer::GEDGsfElectronProducer	(	const edm::ParameterSet &	cfg,
		const gsfAlgoHelpers::HeavyObjectCache *	hoc
	)

explicit

Definition at line 30 of file GEDGsfElectronProducer.cc.

   : GsfElectronBaseProducer(cfg,hoc)
   , egmPFCandidateCollection_(consumes<reco::PFCandidateCollection>(cfg.getParameter<edm::InputTag>("egmPFCandidatesTag")))
 {
    produces<edm::ValueMap<reco::GsfElectronRef> >();
 }

Member Function Documentation

void GEDGsfElectronProducer::fillGsfElectronValueMap	(	edm::Event &	event,
		edm::ValueMap< reco::GsfElectronRef >::Filler &	filler
	)

private

Definition at line 57 of file GEDGsfElectronProducer.cc.

References begin, egmPFCandidateCollection_, end, runEdmFileComparison::found, edm::helper::Filler< Map >::insert(), GsfElectronBaseProducer::orphanHandle(), slimmedMuons_cfi::pfCandidates, and MuonErrorMatrixValues_cff::values.

Referenced by produce().

 {
   // Read the collection of PFCandidates
   edm::Handle<reco::PFCandidateCollection> pfCandidates;
   
   bool found = event.getByToken(egmPFCandidateCollection_, pfCandidates);
   if(!found) {
     edm::LogError("GEDGsfElectronProducer")
        <<" cannot get PFCandidates! ";
   }
 
   //Loop over the collection of PFFCandidates
   reco::PFCandidateCollection::const_iterator it = pfCandidates->begin();
   reco::PFCandidateCollection::const_iterator itend = pfCandidates->end() ;
   std::vector<reco::GsfElectronRef> values;
 
   for ( ; it != itend ; ++it) {
     reco::GsfElectronRef myRef;
     // First check that the GsfTrack is non null
     if( it->gsfTrackRef().isNonnull()) {
       // now look for the corresponding GsfElectron
       const auto itcheck = std::find_if( orphanHandle()->begin(),orphanHandle()->end(),
                                          [it](const auto& ele) {return (ele.gsfTrack() == it->gsfTrackRef());} );
       if (itcheck != orphanHandle()->end()) {
     // Build the Ref from the handle and the index
     myRef = reco::GsfElectronRef(orphanHandle(),itcheck-orphanHandle()->begin());
       }
     }
     values.push_back(myRef);
   }
   filler.insert(pfCandidates,values.begin(),values.end());
 }

void GEDGsfElectronProducer::matchWithPFCandidates ( edm::Event & event )

private

Definition at line 94 of file GEDGsfElectronProducer.cc.

References reco::GsfElectron::MvaInput::deltaEta, reco::GsfElectron::MvaInput::earlyBrem, egmPFCandidateCollection_, runEdmFileComparison::found, gsfMVAInputMap_, gsfMVAOutputMap_, reco::GsfElectron::MvaInput::hadEnergy, reco::GsfElectron::MvaInput::lateBrem, reco::PFCandidateEGammaExtra::MVA_DeltaEtaTrackCluster, reco::PFCandidateEGammaExtra::MVA_FirstBrem, reco::PFCandidateEGammaExtra::MVA_LateBrem, slimmedMuons_cfi::pfCandidates, reco::GsfElectron::MvaInput::sigmaEtaEta, and reco::GsfElectron::MvaOutput::status.

Referenced by produce().

 {
   gsfMVAInputMap_.clear();
   gsfMVAOutputMap_.clear();
 
   // Read the collection of PFCandidates
   edm::Handle<reco::PFCandidateCollection> pfCandidates;
   
   bool found = event.getByToken(egmPFCandidateCollection_, pfCandidates);
   if(!found) {
     edm::LogError("GEDGsfElectronProducer")
        <<" cannot get PFCandidates! ";
   }
 
   //Loop over the collection of PFFCandidates
   reco::PFCandidateCollection::const_iterator it = pfCandidates->begin();
   reco::PFCandidateCollection::const_iterator itend = pfCandidates->end() ;
   
   for ( ; it != itend ; ++it) {
     reco::GsfElectronRef myRef;
     // First check that the GsfTrack is non null
     if( it->gsfTrackRef().isNonnull()) {
 
       reco::GsfElectron::MvaOutput myMvaOutput;
       // at the moment, undefined
       myMvaOutput.status = it->egammaExtraRef()->electronStatus() ;
       gsfMVAOutputMap_[it->gsfTrackRef()] = myMvaOutput;
 
       reco::GsfElectron::MvaInput myMvaInput;
       myMvaInput.earlyBrem = it->egammaExtraRef()->mvaVariable(reco::PFCandidateEGammaExtra::MVA_FirstBrem);
       myMvaInput.lateBrem = it->egammaExtraRef()->mvaVariable(reco::PFCandidateEGammaExtra::MVA_LateBrem);
       myMvaInput.deltaEta = it->egammaExtraRef()->mvaVariable(reco::PFCandidateEGammaExtra::MVA_DeltaEtaTrackCluster);
       myMvaInput.sigmaEtaEta = it->egammaExtraRef()->sigmaEtaEta();
       myMvaInput.hadEnergy = it->egammaExtraRef()->hadEnergy();
       gsfMVAInputMap_[it->gsfTrackRef()] = myMvaInput;
     }
   }
 }

void GEDGsfElectronProducer::produce	(	edm::Event &	event,
		const edm::EventSetup &	setup
	)

override

Definition at line 39 of file GEDGsfElectronProducer.cc.

References GsfElectronBaseProducer::algo_, GsfElectronAlgo::completeElectrons(), nano_cff::electrons, GsfElectronBaseProducer::fillEvent(), fillGsfElectronValueMap(), edm::Event::get(), gsfMVAInputMap_, gsfMVAOutputMap_, GsfElectronBaseProducer::inputCfg_, matchWithPFCandidates(), eostools::move(), edm::second(), setMVAOutputs(), and GsfElectronAlgo::InputTagsConfiguration::vtxCollectionTag.

  {
   matchWithPFCandidates(event);
   reco::GsfElectronCollection electrons;
   algo_->completeElectrons(electrons, event, setup, globalCache()) ;
   setMVAOutputs(electrons, globalCache(),gsfMVAOutputMap_, event.get(inputCfg_.vtxCollectionTag));
   for(auto& el : electrons) el.setMvaInput(gsfMVAInputMap_.find(el.gsfTrack())->second); // set MVA inputs
   fillEvent(electrons, event) ;
 
   // ValueMap
   auto valMap_p = std::make_unique<edm::ValueMap<reco::GsfElectronRef>>();
   edm::ValueMap<reco::GsfElectronRef>::Filler valMapFiller(*valMap_p);
   fillGsfElectronValueMap(event,valMapFiller);
   valMapFiller.fill();
   event.put(std::move(valMap_p));  
   // Done with the ValueMap
  }

void GEDGsfElectronProducer::setMVAOutputs	(	reco::GsfElectronCollection &	electrons,
		const gsfAlgoHelpers::HeavyObjectCache *	hoc,
		const std::map< reco::GsfTrackRef, reco::GsfElectron::MvaOutput > &	mvaOutputs,
		reco::VertexCollection const &	vertices
	)		const

private

Definition at line 134 of file GEDGsfElectronProducer.cc.

References gsfAlgoHelpers::HeavyObjectCache::iElectronMVAEstimator, reco::GsfElectron::MvaOutput::mva_e_pi, reco::GsfElectron::MvaOutput::mva_Isolated, and gsfAlgoHelpers::HeavyObjectCache::sElectronMVAEstimator.

Referenced by produce().

 {
   for (auto el = electrons.begin(); el != electrons.end(); el++) {
     float mva_NIso_Value = hoc->sElectronMVAEstimator->mva(*el, vertices);
     float mva_Iso_Value = hoc->iElectronMVAEstimator->mva(*el, vertices.size());
     GsfElectron::MvaOutput mvaOutput;
     mvaOutput.mva_e_pi = mva_NIso_Value;
     mvaOutput.mva_Isolated = mva_Iso_Value;
     el->setMvaOutput(mvaOutput);
   }
 }