ElectronIdMVAProducer Class Reference

Inheritance diagram for ElectronIdMVAProducer:

Public Member Functions
	ElectronIdMVAProducer (const edm::ParameterSet &)
	~ElectronIdMVAProducer ()
Public Member Functions inherited from edm::EDFilter
	EDFilter ()
ModuleDescription const &	moduleDescription () const
virtual	~EDFilter ()
Public Member Functions inherited from edm::ProducerBase
	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
	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
bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)
virtual	~EDConsumerBase ()

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

Private Attributes
double	_Rho
edm::EDGetTokenT < reco::GsfElectronCollection >	electronToken_
edm::EDGetTokenT< double >	eventrhoToken_
string	method_
EGammaMvaEleEstimator *	mvaID_
vector< 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
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList
Static Public Member Functions inherited from edm::EDFilter
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &)
Protected Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
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

Definition at line 26 of file ElectronIdMVAProducer.cc.

Constructor & Destructor Documentation

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

explicit

Definition at line 62 of file ElectronIdMVAProducer.cc.

References edm::FileInPath::fullPath(), edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), compare_using_db::ifile, EGammaMvaEleEstimator::kNonTrig, EGammaMvaEleEstimator::kTrig, and EGammaMvaEleEstimator::kTrigNoIP.

                                                                           {
  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<string>("method");
  std::vector<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;
  
  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_);
  
}

ElectronIdMVAProducer::~ElectronIdMVAProducer

(

)

Definition at line 98 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

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

privatevirtual

Implements edm::EDFilter.

Definition at line 112 of file ElectronIdMVAProducer.cc.

References alignCSCRings::e, edm::helper::Filler< Map >::fill(), edm::EventSetup::get(), edm::Event::getByToken(), edm::helper::Filler< Map >::insert(), dbtoconf::out, AlCaHLTBitMon_ParallelJobs::p, edm::Event::put(), makeHLTPrescaleTable::values, and GoodVertex_cfg::vertexCollection.

Referenced by Vispa.Plugins.Browser.BrowserTabController.BrowserTabController::filter(), Vispa.Plugins.Browser.BrowserTabController.BrowserTabController::find(), Vispa.Plugins.Browser.BrowserTabController.BrowserTabController::setDataAccessor(), and Vispa.Plugins.Browser.BrowserTabController.BrowserTabController::switchCenterView().

                                                                                {
  using namespace edm;
  
  std::auto_ptr<edm::ValueMap<float> > out(new edm::ValueMap<float>() );
  
  Handle<reco::VertexCollection>  vertexCollection;
  iEvent.getByToken(vertexToken_, vertexCollection);
  
  Vertex dummy;
  const Vertex *pv = &dummy;
  if ( vertexCollection->size() != 0) {
    pv = &*vertexCollection->begin();
  } else { // create a dummy PV
    Vertex::Error e;
    e(0, 0) = 0.0015 * 0.0015;
    e(1, 1) = 0.0015 * 0.0015;
    e(2, 2) = 15. * 15.;
    Vertex::Point p(0, 0, 0);
    dummy = Vertex(p, e, 0, 0, 0);
  }
  
  EcalClusterLazyTools lazyTools(iEvent, iSetup, reducedEBRecHitCollectionToken_, reducedEERecHitCollectionToken_);
  
  edm::ESHandle<TransientTrackBuilder> builder;
  iSetup.get<TransientTrackRecord>().get("TransientTrackBuilder", builder);
  TransientTrackBuilder thebuilder = *(builder.product());
  
  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(out);
  
  return true;
}

Member Data Documentation

double ElectronIdMVAProducer::_Rho

private

Definition at line 42 of file ElectronIdMVAProducer.cc.

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

private

Definition at line 37 of file ElectronIdMVAProducer.cc.

edm::EDGetTokenT<double> ElectronIdMVAProducer::eventrhoToken_

private

Definition at line 38 of file ElectronIdMVAProducer.cc.

string ElectronIdMVAProducer::method_

private

Definition at line 43 of file ElectronIdMVAProducer.cc.

EGammaMvaEleEstimator* ElectronIdMVAProducer::mvaID_

private

Definition at line 48 of file ElectronIdMVAProducer.cc.

vector<string> ElectronIdMVAProducer::mvaWeightFiles_

private

Definition at line 44 of file ElectronIdMVAProducer.cc.

bool ElectronIdMVAProducer::NoIP_

private

Definition at line 46 of file ElectronIdMVAProducer.cc.

edm::EDGetTokenT<EcalRecHitCollection> ElectronIdMVAProducer::reducedEBRecHitCollectionToken_

private

Definition at line 39 of file ElectronIdMVAProducer.cc.

edm::EDGetTokenT<EcalRecHitCollection> ElectronIdMVAProducer::reducedEERecHitCollectionToken_

private

Definition at line 40 of file ElectronIdMVAProducer.cc.

bool ElectronIdMVAProducer::Trig_

private

Definition at line 45 of file ElectronIdMVAProducer.cc.

bool ElectronIdMVAProducer::verbose_

private

Definition at line 35 of file ElectronIdMVAProducer.cc.

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

private

Definition at line 36 of file ElectronIdMVAProducer.cc.