GsfEleMVACut Class Reference

Inheritance diagram for GsfEleMVACut:

Public Member Functions
CandidateType	candidateType () const final
void	getEventContent (const edm::EventBase &) final
	GsfEleMVACut (const edm::ParameterSet &c)
result_type	operator() (const reco::GsfElectronPtr &) const final
void	setConsumes (edm::ConsumesCollector &) final
Public Member Functions inherited from CutApplicatorWithEventContentBase
	CutApplicatorWithEventContentBase ()
	CutApplicatorWithEventContentBase (const edm::ParameterSet &c)
	CutApplicatorWithEventContentBase (const CutApplicatorWithEventContentBase &)=delete
CutApplicatorWithEventContentBase &	operator= (const CutApplicatorWithEventContentBase &)=delete
	~CutApplicatorWithEventContentBase () override
	Destructor. More...
Public Member Functions inherited from CutApplicatorBase
virtual result_type	asCandidate (const argument_type &) const
	CutApplicatorBase ()
	CutApplicatorBase (const edm::ParameterSet &c)
	CutApplicatorBase (const CutApplicatorBase &)=delete
const std::string &	name () const override
result_type	operator() (const argument_type &) const override final
virtual result_type	operator() (const pat::ElectronPtr &) const
virtual result_type	operator() (const reco::PhotonPtr &) const
virtual result_type	operator() (const pat::PhotonPtr &) const
virtual result_type	operator() (const reco::MuonPtr &) const
virtual result_type	operator() (const pat::MuonPtr &) const
virtual result_type	operator() (const reco::PFTauPtr &) const
virtual result_type	operator() (const pat::TauPtr &) const
CutApplicatorBase &	operator= (const CutApplicatorBase &)=delete
	~CutApplicatorBase () override
	Destructor. More...
Public Member Functions inherited from candidate_functions::CandidateCut
	CandidateCut ()
virtual	~CandidateCut ()

Private Member Functions
double	value (const reco::CandidatePtr &cand) const final

Private Attributes
std::vector< StringObjectFunction< reco::GsfElectron > >	cutFormula_
edm::Handle< edm::ValueMap< int > >	mvaCategoriesMap_
const std::vector< std::string >	mvaCutStrings_
edm::Handle< edm::ValueMap< float > >	mvaValueMap_
const int	nCuts_

Additional Inherited Members
Public Types inherited from CutApplicatorBase
enum	CandidateType {   NONE, ELECTRON, MUON, PHOTON,   TAU, PATELECTRON, PATMUON, PATPHOTON,   PATTAU }
Protected Attributes inherited from CutApplicatorWithEventContentBase
std::unordered_map< std::string, edm::InputTag >	contentTags_
std::unordered_map< std::string, edm::EDGetToken >	contentTokens_

Detailed Description

Definition at line 6 of file GsfEleMVACut.cc.

Constructor & Destructor Documentation

GsfEleMVACut::GsfEleMVACut

(

const edm::ParameterSet &

c

)

Definition at line 39 of file GsfEleMVACut.cc.

References CutApplicatorWithEventContentBase::contentTags_, cutFormula_, edm::ParameterSet::getParameter(), and mvaCutStrings_.

                                                   :
  CutApplicatorWithEventContentBase(c),
  mvaCutStrings_(c.getParameter<std::vector<std::string> >("mvaCuts")),
  nCuts_(mvaCutStrings_.size())
{
  edm::InputTag mvaValTag = c.getParameter<edm::InputTag>("mvaValueMapName");
  contentTags_.emplace("mvaVal",mvaValTag);

  edm::InputTag mvaCatTag = c.getParameter<edm::InputTag>("mvaCategoriesMapName");
  contentTags_.emplace("mvaCat",mvaCatTag);

  for (auto &cutString : mvaCutStrings_) {
      cutFormula_.push_back(StringObjectFunction<reco::GsfElectron>(cutString));
  }
}

Member Function Documentation

CandidateType GsfEleMVACut::candidateType ( ) const

inlinefinalvirtual

Reimplemented from CutApplicatorBase.

Definition at line 15 of file GsfEleMVACut.cc.

References CutApplicatorBase::ELECTRON, and value().

                                            {
    return ELECTRON;
  }

void GsfEleMVACut::getEventContent ( const edm::EventBase &  ev )

finalvirtual

Implements CutApplicatorWithEventContentBase.

Definition at line 66 of file GsfEleMVACut.cc.

References CutApplicatorWithEventContentBase::contentTags_, edm::EventBase::getByLabel(), mvaCategoriesMap_, mvaValueMap_, and operator()().

                                                         {

  ev.getByLabel(contentTags_["mvaVal"], mvaValueMap_);
  ev.getByLabel(contentTags_["mvaCat"], mvaCategoriesMap_);
}

CutApplicatorBase::result_type GsfEleMVACut::operator() ( const reco::GsfElectronPtr &  cand ) const

finalvirtual

Reimplemented from CutApplicatorBase.

Definition at line 74 of file GsfEleMVACut.cc.

References edm::ValueMap< T >::begin(), eostools::cat(), edm::ValueMap< T >::contains(), CutApplicatorWithEventContentBase::contentTags_, cutFormula_, Exception, edm::Ptr< T >::id(), edm::ValueMap< T >::idSize(), edm::HandleBase::isValid(), edm::Ptr< T >::key(), mvaCategoriesMap_, mvaValueMap_, nCuts_, AlCaHLTBitMon_QueryRunRegistry::string, pat::PATObject< ObjectType >::userFloat(), pat::PATObject< ObjectType >::userInt(), and heppy_batch::val.

Referenced by getEventContent().

                                                {

  // in case we are by-value
  const std::string& val_name = contentTags_.find("mvaVal")->second.instance();
  const std::string& cat_name = contentTags_.find("mvaCat")->second.instance();
  edm::Ptr<pat::Electron> pat(cand);
  float val = -1.0;
  int   cat = -1;
  if( mvaCategoriesMap_.isValid() && mvaCategoriesMap_->contains( cand.id() ) &&
      mvaValueMap_.isValid() && mvaValueMap_->contains( cand.id() ) ) {
    cat = (*mvaCategoriesMap_)[cand];
    val = (*mvaValueMap_)[cand];
  } else if ( mvaCategoriesMap_.isValid() && mvaValueMap_.isValid() &&
              mvaCategoriesMap_->idSize() == 1 && mvaValueMap_->idSize() == 1 &&
              cand.id() == edm::ProductID() ) {
    // in case we have spoofed a ptr
    //note this must be a 1:1 valuemap (only one product input)
    cat = mvaCategoriesMap_->begin()[cand.key()];
    val = mvaValueMap_->begin()[cand.key()];
  } else if ( mvaCategoriesMap_.isValid() && mvaValueMap_.isValid() ){ // throw an exception
    cat = (*mvaCategoriesMap_)[cand];
    val = (*mvaValueMap_)[cand];
  }

  // Find the cut formula
  const int iCategory = mvaCategoriesMap_.isValid() ? cat : pat->userInt( cat_name );
  if( iCategory >= nCuts_ )
    throw cms::Exception(" Error in MVA categories: ")
      << " found a particle with a category larger than max configured " << std::endl;

  // Look up the MVA value for this particle
  const float mvaValue = mvaValueMap_.isValid() ? val : pat->userFloat( val_name );

  // Apply the cut and return the result
  return mvaValue > cutFormula_[iCategory](*cand);
}

void GsfEleMVACut::setConsumes ( edm::ConsumesCollector &  cc )

finalvirtual

Implements CutApplicatorWithEventContentBase.

Definition at line 55 of file GsfEleMVACut.cc.

References edm::ConsumesCollector::consumes(), CutApplicatorWithEventContentBase::contentTags_, and CutApplicatorWithEventContentBase::contentTokens_.

                                                       {

  auto mvaVal =
    cc.consumes<edm::ValueMap<float> >(contentTags_["mvaVal"]);
  contentTokens_.emplace("mvaVal",mvaVal);

  auto mvaCat =
    cc.consumes<edm::ValueMap<int> >(contentTags_["mvaCat"]);
  contentTokens_.emplace("mvaCat",mvaCat);
}

double GsfEleMVACut::value ( const reco::CandidatePtr &  cand ) const

finalprivatevirtual

Implements candidate_functions::CandidateCut.

Definition at line 111 of file GsfEleMVACut.cc.

References edm::ValueMap< T >::begin(), edm::ValueMap< T >::contains(), CutApplicatorWithEventContentBase::contentTags_, edm::Ptr< T >::id(), edm::ValueMap< T >::idSize(), edm::HandleBase::isValid(), edm::Ptr< T >::key(), mvaCategoriesMap_, mvaValueMap_, AlCaHLTBitMon_QueryRunRegistry::string, pat::PATObject< ObjectType >::userFloat(), and heppy_batch::val.

Referenced by candidateType().

                                                             {

  // in case we are by-value
  const std::string& val_name =contentTags_.find("mvaVal")->second.instance();
  edm::Ptr<pat::Electron> pat(cand);
  float val = 0.0;
  if( mvaCategoriesMap_.isValid() && mvaCategoriesMap_->contains( cand.id() ) &&
      mvaValueMap_.isValid() && mvaValueMap_->contains( cand.id() ) ) {
    val = (*mvaValueMap_)[cand];
  } else if ( mvaCategoriesMap_.isValid() && mvaValueMap_.isValid() &&
              mvaCategoriesMap_->idSize() == 1 && mvaValueMap_->idSize() == 1 &&
              cand.id() == edm::ProductID() ) {
    // in case we have spoofed a ptr
    //note this must be a 1:1 valuemap (only one product input)
    val = mvaValueMap_->begin()[cand.key()];
  } else if ( mvaCategoriesMap_.isValid() && mvaValueMap_.isValid() ){ // throw an exception
    val = (*mvaValueMap_)[cand];
  }

  const float mvaValue = mvaValueMap_.isValid() ? val : pat->userFloat( val_name );
  return mvaValue;
}

Member Data Documentation

std::vector<StringObjectFunction<reco::GsfElectron> > GsfEleMVACut::cutFormula_

private

Definition at line 25 of file GsfEleMVACut.cc.

Referenced by GsfEleMVACut(), and operator()().

edm::Handle<edm::ValueMap<int> > GsfEleMVACut::mvaCategoriesMap_

private

Definition at line 31 of file GsfEleMVACut.cc.

Referenced by getEventContent(), operator()(), and value().

const std::vector<std::string> GsfEleMVACut::mvaCutStrings_

private

Definition at line 24 of file GsfEleMVACut.cc.

Referenced by GsfEleMVACut().

edm::Handle<edm::ValueMap<float> > GsfEleMVACut::mvaValueMap_

private

Definition at line 30 of file GsfEleMVACut.cc.

Referenced by getEventContent(), operator()(), and value().

const int GsfEleMVACut::nCuts_

private

Definition at line 27 of file GsfEleMVACut.cc.

Referenced by operator()().