ElectronIdMVABased Class Reference

#include <MyAnalyzer/ElectronIdMVABased/src/ElectronIdMVABased.cc>

Inheritance diagram for ElectronIdMVABased:

Public Member Functions
	ElectronIdMVABased (const edm::ParameterSet &)
	~ElectronIdMVABased ()
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
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 &) override

Private Attributes
edm::EDGetTokenT < reco::GsfElectronCollection >	electronToken
ElectronMVAEstimator *	mvaID_
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::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

Description: [one line class summary]

Implementation: [Notes on implementation]

Definition at line 42 of file ElectronIdMVABased.cc.

Constructor & Destructor Documentation

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

explicit

Definition at line 75 of file ElectronIdMVABased.cc.

References edm::ParameterSet::getParameter().

                                                                     {
    vertexToken = consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("vertexTag"));
    electronToken = consumes<reco::GsfElectronCollection>(iConfig.getParameter<edm::InputTag>("electronTag"));
    mvaWeightFileEleID = iConfig.getParameter<string>("HZZmvaWeightFile");
    thresholdBarrel = iConfig.getParameter<double>("thresholdBarrel");
    thresholdEndcap = iConfig.getParameter<double>("thresholdEndcap");
    thresholdIsoBarrel = iConfig.getParameter<double>("thresholdIsoDR03Barrel");
    thresholdIsoEndcap = iConfig.getParameter<double>("thresholdIsoDR03Endcap");

    produces<reco::GsfElectronCollection>();
    path_mvaWeightFileEleID = edm::FileInPath ( mvaWeightFileEleID.c_str() ).fullPath();
    FILE * fileEleID = fopen(path_mvaWeightFileEleID.c_str(), "r");
    if (fileEleID) {
      fclose(fileEleID);
    }
    else {
      string err = "ElectronIdMVABased: cannot open weight file '";
      err += path_mvaWeightFileEleID;
      err += "'";
      throw invalid_argument( err );
    }

    mvaID_ = new ElectronMVAEstimator(path_mvaWeightFileEleID);
}

ElectronIdMVABased::~ElectronIdMVABased

(

)

Definition at line 101 of file ElectronIdMVABased.cc.

{

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

}

Member Function Documentation

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

overrideprivatevirtual

Implements edm::EDFilter.

Definition at line 116 of file ElectronIdMVABased.cc.

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

                                                                             {
    using namespace edm;

    std::auto_ptr<reco::GsfElectronCollection> mvaElectrons(new reco::GsfElectronCollection);

    Handle<reco::VertexCollection>  vertexCollection;
    iEvent.getByToken(vertexToken, vertexCollection);
    int nVtx = vertexCollection->size();

    Handle<reco::GsfElectronCollection> egCollection;
    iEvent.getByToken(electronToken,egCollection);
    const reco::GsfElectronCollection egCandidates = (*egCollection.product());
    for ( reco::GsfElectronCollection::const_iterator egIter = egCandidates.begin(); egIter != egCandidates.end(); ++egIter) {
      double mvaVal = mvaID_->mva( *egIter, nVtx );
      double isoDr03 = egIter->dr03TkSumPt() + egIter->dr03EcalRecHitSumEt() + egIter->dr03HcalTowerSumEt();
      double eleEta = fabs(egIter->eta());
      if (eleEta <= 1.485 && mvaVal > thresholdBarrel && isoDr03 < thresholdIsoBarrel) {
        mvaElectrons->push_back( *egIter );
        reco::GsfElectron::MvaOutput myMvaOutput;
        myMvaOutput.mva = mvaVal;
        mvaElectrons->back().setMvaOutput(myMvaOutput);
      }
      else if (eleEta > 1.485 && mvaVal > thresholdEndcap  && isoDr03 < thresholdIsoEndcap) {
        mvaElectrons->push_back( *egIter );
        reco::GsfElectron::MvaOutput myMvaOutput;
        myMvaOutput.mva = mvaVal;
        mvaElectrons->back().setMvaOutput(myMvaOutput);
      }


    }


    iEvent.put(mvaElectrons);

    return true;
}

Member Data Documentation

edm::EDGetTokenT<reco::GsfElectronCollection> ElectronIdMVABased::electronToken

private

Definition at line 53 of file ElectronIdMVABased.cc.

ElectronMVAEstimator* ElectronIdMVABased::mvaID_

private

Definition at line 61 of file ElectronIdMVABased.cc.

string ElectronIdMVABased::mvaWeightFileEleID

private

Definition at line 54 of file ElectronIdMVABased.cc.

string ElectronIdMVABased::path_mvaWeightFileEleID

private

Definition at line 55 of file ElectronIdMVABased.cc.

double ElectronIdMVABased::thresholdBarrel

private

Definition at line 56 of file ElectronIdMVABased.cc.

double ElectronIdMVABased::thresholdEndcap

private

Definition at line 57 of file ElectronIdMVABased.cc.

double ElectronIdMVABased::thresholdIsoBarrel

private

Definition at line 58 of file ElectronIdMVABased.cc.

double ElectronIdMVABased::thresholdIsoEndcap

private

Definition at line 59 of file ElectronIdMVABased.cc.

edm::EDGetTokenT<reco::VertexCollection> ElectronIdMVABased::vertexToken

private

Definition at line 52 of file ElectronIdMVABased.cc.