ElectronMVAEstimatorRun2Spring16GeneralPurpose Class Reference

#include <ElectronMVAEstimatorRun2Spring16GeneralPurpose.h>

Inheritance diagram for ElectronMVAEstimatorRun2Spring16GeneralPurpose:

Classes
struct	AllVariables

Public Types
enum	mvaCategories { UNDEFINED = -1, CAT_EB1_PT10plus = 0, CAT_EB2_PT10plus = 1, CAT_EE_PT10plus = 2 }

Public Member Functions
void	constrainMVAVariables (AllVariables &) const
std::unique_ptr< const GBRForest >	createSingleReader (const int iCategory, const edm::FileInPath &weightFile)
	ElectronMVAEstimatorRun2Spring16GeneralPurpose (const edm::ParameterSet &conf)
	ElectronMVAEstimatorRun2Spring16GeneralPurpose ()
	ElectronMVAEstimatorRun2Spring16GeneralPurpose (const std::string &mvaTag, const std::string &conversionsTag="reducedEgamma:reducedConversions", const std::string &beamspotTag="offlineBeamSpot")
std::vector< float >	fillMVAVariables (const edm::Ptr< reco::Candidate > &particle, const edm::Event &) const override
std::vector< float >	fillMVAVariables (const reco::GsfElectron *particle, const edm::Handle< reco::ConversionCollection > conversions, const reco::BeamSpot *beamSpot) const
int	findCategory (const edm::Ptr< reco::Candidate > &particle) const override
int	findCategory (const reco::GsfElectron *particle) const
virtual const std::string &	getName () const override final
virtual int	getNCategories () const override
virtual const std::string &	getTag () const override final
void	init (const std::vector< std::string > weightFileNames)
bool	isEndcapCategory (int category) const
float	mvaValue (const edm::Ptr< reco::Candidate > &particle, const edm::Event &) const override
float	mvaValue (const reco::GsfElectron *particle, const edm::EventBase &) const
float	mvaValue (const int iCategory, const std::vector< float > &vars) const
void	setConsumes (edm::ConsumesCollector &&) const override final
	~ElectronMVAEstimatorRun2Spring16GeneralPurpose ()
Public Member Functions inherited from AnyMVAEstimatorRun2Base
	AnyMVAEstimatorRun2Base (const edm::ParameterSet &conf)
virtual void	getEventContent (const edm::Event &iEvent) const final
template<typename... Args>
std::vector< float >	packMVAVariables (const Args...args) const
virtual	~AnyMVAEstimatorRun2Base ()

Public Attributes
const int	nCategories = 3
Public Attributes inherited from AnyMVAEstimatorRun2Base
const edm::ParameterSet	_conf

Private Attributes
AllVariables	allMVAVars_
const edm::InputTag	beamSpotLabel_
const edm::InputTag	conversionsLabelAOD_
const edm::InputTag	conversionsLabelMiniAOD_
std::vector< std::unique_ptr< const GBRForest > >	gbrForest_s
const std::string	MethodName_
const std::string	name_ = "ElectronMVAEstimatorRun2Spring16GeneralPurpose"
const std::string	tag_

Detailed Description

Definition at line 22 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.h.

Member Enumeration Documentation

enum ElectronMVAEstimatorRun2Spring16GeneralPurpose::mvaCategories

Enumerator
UNDEFINED
CAT_EB1_PT10plus
CAT_EB2_PT10plus
CAT_EE_PT10plus

Definition at line 29 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.h.

                     {
    UNDEFINED = -1,
    CAT_EB1_PT10plus = 0,
    CAT_EB2_PT10plus = 1,
    CAT_EE_PT10plus  = 2
  };

Constructor & Destructor Documentation

ElectronMVAEstimatorRun2Spring16GeneralPurpose::ElectronMVAEstimatorRun2Spring16GeneralPurpose

(

const edm::ParameterSet &

conf

)

Definition at line 13 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.cc.

References edm::ParameterSet::getParameter(), init(), and mvaElectronID_PHYS14_PU20bx25_nonTrig_V1_cff::weightFileNames.

                                                                                                                         :
  AnyMVAEstimatorRun2Base(conf),
  tag_(conf.getParameter<std::string>("mvaTag")),
  MethodName_("BDTG method"),
  beamSpotLabel_(conf.getParameter<edm::InputTag>("beamSpot")),
  conversionsLabelAOD_(conf.getParameter<edm::InputTag>("conversionsAOD")),
  conversionsLabelMiniAOD_(conf.getParameter<edm::InputTag>("conversionsMiniAOD")) {
  
  const std::vector <std::string> weightFileNames
    = conf.getParameter<std::vector<std::string> >("weightFileNames");
  init(weightFileNames);
}

ElectronMVAEstimatorRun2Spring16GeneralPurpose::ElectronMVAEstimatorRun2Spring16GeneralPurpose

(

)

Definition at line 45 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.cc.

                                                                                              :
  AnyMVAEstimatorRun2Base(edm::ParameterSet()) {}

ElectronMVAEstimatorRun2Spring16GeneralPurpose::ElectronMVAEstimatorRun2Spring16GeneralPurpose	(	const std::string &	mvaTag,
		const std::string &	conversionsTag = "reducedEgamma:reducedConversions",
		const std::string &	beamspotTag = "offlineBeamSpot"
	)

Definition at line 48 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.cc.

References ~ElectronMVAEstimatorRun2Spring16GeneralPurpose().

                                                                                                                                                                                    :
  AnyMVAEstimatorRun2Base( edm::ParameterSet() ),
  tag_(mvaTag),
  MethodName_("BDTG method"),
  beamSpotLabel_(edm::InputTag(beamspotTag)),
  conversionsLabelAOD_(edm::InputTag(conversionsTag)),
  conversionsLabelMiniAOD_(conversionsLabelAOD_) {
 }

ElectronMVAEstimatorRun2Spring16GeneralPurpose::~ElectronMVAEstimatorRun2Spring16GeneralPurpose

(

)

Definition at line 60 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.cc.

Referenced by ElectronMVAEstimatorRun2Spring16GeneralPurpose().

                                                 {
}

Member Function Documentation

void ElectronMVAEstimatorRun2Spring16GeneralPurpose::constrainMVAVariables

(

AllVariables &

allMVAVars

)

const

Definition at line 416 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.cc.

References ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::deta, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::detacalo, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::dphi, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::eleEoPout, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::EoP, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::fbrem, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::gsfchi2, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::kfchi2, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::OneMinusE1x5E5x5, and ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::R9.

Referenced by fillMVAVariables(), and getTag().

                                                                                                         {

  // Check that variables do not have crazy values

  if(allMVAVars.fbrem < -1.)
    allMVAVars.fbrem = -1.;
  
  allMVAVars.deta = fabs(allMVAVars.deta);
  if(allMVAVars.deta > 0.06)
    allMVAVars.deta = 0.06;
  
  
  allMVAVars.dphi = fabs(allMVAVars.dphi);
  if(allMVAVars.dphi > 0.6)
    allMVAVars.dphi = 0.6;
  

  if(allMVAVars.EoP > 20.)
    allMVAVars.EoP = 20.;
  
  if(allMVAVars.eleEoPout > 20.)
    allMVAVars.eleEoPout = 20.;
  
  
  allMVAVars.detacalo = fabs(allMVAVars.detacalo);
  if(allMVAVars.detacalo > 0.2)
    allMVAVars.detacalo = 0.2;
  
  if(allMVAVars.OneMinusE1x5E5x5 < -1.)
    allMVAVars.OneMinusE1x5E5x5 = -1;
  
  if(allMVAVars.OneMinusE1x5E5x5 > 2.)
    allMVAVars.OneMinusE1x5E5x5 = 2.; 
  
  
  if(allMVAVars.R9 > 5)
    allMVAVars.R9 = 5;
  
  if(allMVAVars.gsfchi2 > 200.)
    allMVAVars.gsfchi2 = 200;
  
  
  if(allMVAVars.kfchi2 > 10.)
    allMVAVars.kfchi2 = 10.;
}

std::unique_ptr< const GBRForest > ElectronMVAEstimatorRun2Spring16GeneralPurpose::createSingleReader	(	const int	iCategory,
		const edm::FileInPath &	weightFile
	)

Definition at line 177 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.cc.

References allMVAVars_, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::convVtxFitProbability, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::deta, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::detacalo, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::dphi, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::eleEoPout, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::EoP, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::etawidth, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::expectedMissingInnerHits, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::fbrem, fillMVAVariables(), edm::FileInPath::fullPath(), ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::gsfchi2, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::gsfhits, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::HoE, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::IoEmIoP, isEndcapCategory(), ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::kfchi2, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::kfhits, MethodName_, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::OneMinusE1x5E5x5, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::phiwidth, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::PreShowerOverRaw, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::pt, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::R9, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::SCeta, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::see, and ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::spp.

Referenced by init(), and isEndcapCategory().

                                                                        {

  //
  // Create the reader  
  //
  TMVA::Reader tmpTMVAReader( "!Color:Silent:!Error" );

  //
  // Configure all variables and spectators. Note: the order and names
  // must match what is found in the xml weights file!
  //

  // Pure ECAL -> shower shapes
  tmpTMVAReader.AddVariable("ele_oldsigmaietaieta", &allMVAVars_.see);
  tmpTMVAReader.AddVariable("ele_oldsigmaiphiiphi", &allMVAVars_.spp);
  tmpTMVAReader.AddVariable("ele_oldcircularity",   &allMVAVars_.OneMinusE1x5E5x5);
  tmpTMVAReader.AddVariable("ele_oldr9",            &allMVAVars_.R9);
  tmpTMVAReader.AddVariable("ele_scletawidth",      &allMVAVars_.etawidth);
  tmpTMVAReader.AddVariable("ele_sclphiwidth",      &allMVAVars_.phiwidth);
  tmpTMVAReader.AddVariable("ele_oldhe",               &allMVAVars_.HoE);
 
  //Pure tracking variables
  tmpTMVAReader.AddVariable("ele_kfhits",           &allMVAVars_.kfhits);
  tmpTMVAReader.AddVariable("ele_kfchi2",           &allMVAVars_.kfchi2);
  tmpTMVAReader.AddVariable("ele_gsfchi2",        &allMVAVars_.gsfchi2);

  // Energy matching
  tmpTMVAReader.AddVariable("ele_fbrem",           &allMVAVars_.fbrem);

  tmpTMVAReader.AddVariable("ele_gsfhits",         &allMVAVars_.gsfhits);
  tmpTMVAReader.AddVariable("ele_expected_inner_hits",             &allMVAVars_.expectedMissingInnerHits);
  tmpTMVAReader.AddVariable("ele_conversionVertexFitProbability",  &allMVAVars_.convVtxFitProbability);

  tmpTMVAReader.AddVariable("ele_ep",              &allMVAVars_.EoP);
  tmpTMVAReader.AddVariable("ele_eelepout",        &allMVAVars_.eleEoPout);
  tmpTMVAReader.AddVariable("ele_IoEmIop",         &allMVAVars_.IoEmIoP);
  
  // Geometrical matchings
  tmpTMVAReader.AddVariable("ele_deltaetain",      &allMVAVars_.deta);
  tmpTMVAReader.AddVariable("ele_deltaphiin",      &allMVAVars_.dphi);
  tmpTMVAReader.AddVariable("ele_deltaetaseed",    &allMVAVars_.detacalo);

  tmpTMVAReader.AddVariable("ele_pt",             &allMVAVars_.pt);
  tmpTMVAReader.AddVariable("scl_eta",            &allMVAVars_.SCeta);

   // Endcap only variables
  if( isEndcapCategory(iCategory) )
    tmpTMVAReader.AddVariable("ele_psEoverEraw",    &allMVAVars_.PreShowerOverRaw);
  
  //
  // Book the method and set up the weights file
  //
  tmpTMVAReader.BookMVA(MethodName_ , weightFile.fullPath());

  return std::unique_ptr<const GBRForest> ( new GBRForest( dynamic_cast<TMVA::MethodBDT*>( tmpTMVAReader.FindMVA(MethodName_) ) ) );
}

std::vector< float > ElectronMVAEstimatorRun2Spring16GeneralPurpose::fillMVAVariables ( const edm::Ptr< reco::Candidate > &  particle, const edm::Event &  iEvent  ) const

overridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 236 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.cc.

References beamSpotLabel_, conversions_cfi::conversions, conversionsLabelAOD_, conversionsLabelMiniAOD_, Exception, edm::Ptr< T >::get(), edm::Event::getByLabel(), edm::HandleBase::isValid(), NULL, and edm::Handle< T >::product().

Referenced by createSingleReader(), getTag(), and mvaValue().

                                                {

  // 
  // Declare all value maps corresponding to the products we defined earlier
  //
  edm::Handle<reco::BeamSpot> theBeamSpot;
  edm::Handle<reco::ConversionCollection> conversions;

  // Get data needed for conversion rejection
  iEvent.getByLabel(beamSpotLabel_, theBeamSpot);

  // Conversions in miniAOD and AOD have different names,
  // but the same type, so we use the same handle with different tokens.
  iEvent.getByLabel(conversionsLabelAOD_, conversions);
  if( !conversions.isValid() )
    iEvent.getByLabel(conversionsLabelMiniAOD_, conversions);

  // Make sure everything is retrieved successfully
  if(! (theBeamSpot.isValid() 
    && conversions.isValid() ) 
     )
    throw cms::Exception("MVA failure: ")
      << "Failed to retrieve event content needed for this MVA" 
      << std::endl
      << "Check python MVA configuration file."
      << std::endl;

  // Try to cast the particle into a reco particle.
  // This should work for both reco and pat.
  const edm::Ptr<reco::GsfElectron> eleRecoPtr = ( edm::Ptr<reco::GsfElectron> )particle;
  if( eleRecoPtr.get() == NULL )
    throw cms::Exception("MVA failure: ")
      << " given particle is expected to be reco::GsfElectron or pat::Electron," << std::endl
      << " but appears to be neither" << std::endl;
  return fillMVAVariables(eleRecoPtr.get(), conversions, theBeamSpot.product());
}

std::vector< float > ElectronMVAEstimatorRun2Spring16GeneralPurpose::fillMVAVariables	(	const reco::GsfElectron *	particle,
		const edm::Handle< reco::ConversionCollection >	conversions,
		const reco::BeamSpot *	beamSpot
	)		const

Definition at line 276 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.cc.

References funct::abs(), reco::Vertex::chi2(), pat::Electron::closestCtfTrackRef(), reco::GsfElectron::closestCtfTrackRef(), constexpr, constrainMVAVariables(), ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::convVtxFitProbability, reco::GsfElectron::deltaEtaSeedClusterTrackAtCalo(), reco::GsfElectron::deltaEtaSuperClusterTrackAtVtx(), reco::GsfElectron::deltaPhiSuperClusterTrackAtVtx(), ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::deta, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::detacalo, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::dphi, reco::GsfElectron::ecalEnergy(), reco::GsfElectron::eEleClusterOverPout(), ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::eleEoPout, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::EoP, reco::GsfElectron::eSuperClusterOverP(), ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::etawidth, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::expectedMissingInnerHits, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::fbrem, reco::GsfElectron::fbrem(), findCategory(), reco::GsfElectron::full5x5_e1x5(), reco::GsfElectron::full5x5_e5x5(), reco::GsfElectron::full5x5_hcalOverEcal(), reco::GsfElectron::full5x5_r9(), reco::GsfElectron::full5x5_sigmaIetaIeta(), reco::GsfElectron::full5x5_sigmaIphiIphi(), edm::Ref< C, T, F >::get(), ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::gsfchi2, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::gsfhits, reco::GsfElectron::gsfTrack(), ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::HoE, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::IoEmIoP, edm::Ref< C, T, F >::isAvailable(), ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::isBarrel, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::isEndcap, isEndcapCategory(), edm::Ref< C, T, F >::isNonnull(), edm::Ref< C, T, F >::isNull(), reco::Vertex::isValid(), ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::kfchi2, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::kfhits, ConversionTools::matchedConversion(), reco::HitPattern::MISSING_INNER_HITS, eostools::move(), reco::Vertex::ndof(), NULL, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::OneMinusE1x5E5x5, AnyMVAEstimatorRun2Base::packMVAVariables(), ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::phiwidth, reco::BeamSpot::position(), ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::PreShowerOverRaw, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::pt, reco::LeafCandidate::pt(), ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::R9, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::SCeta, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::see, ElectronMVAEstimatorRun2Spring16GeneralPurpose::AllVariables::spp, reco::GsfElectron::superCluster(), reco::GsfElectron::trackMomentumAtVtx(), and badGlobalMuonTaggersAOD_cff::vtx.

                                                                                                                {


  // Both pat and reco particles have exactly the same accessors, so we use a reco ptr 
  // throughout the code, with a single exception as of this writing, handled separately below.
  auto superCluster = eleRecoPtr->superCluster();
  
  AllVariables allMVAVars;

  // Pure ECAL -> shower shapes
  allMVAVars.see            = eleRecoPtr->full5x5_sigmaIetaIeta();
  allMVAVars.spp            = eleRecoPtr->full5x5_sigmaIphiIphi();
  allMVAVars.OneMinusE1x5E5x5 = 1. - eleRecoPtr->full5x5_e1x5() / eleRecoPtr->full5x5_e5x5();
  allMVAVars.R9             = eleRecoPtr->full5x5_r9();
  allMVAVars.etawidth       = superCluster->etaWidth();
  allMVAVars.phiwidth       = superCluster->phiWidth();
  allMVAVars.HoE            = eleRecoPtr->full5x5_hcalOverEcal(); //hadronicOverEm();
  // Endcap only variables
  allMVAVars.PreShowerOverRaw  = superCluster->preshowerEnergy() / superCluster->rawEnergy();

  // To get to CTF track information in pat::Electron, we have to have the pointer
  // to pat::Electron, it is not accessible from the pointer to reco::GsfElectron.
  // This behavior is reported and is expected to change in the future (post-7.4.5 some time).
  bool validKF= false; 
  reco::TrackRef myTrackRef = eleRecoPtr->closestCtfTrackRef();
  const pat::Electron * elePatPtr = dynamic_cast<const pat::Electron *>(eleRecoPtr);
  // Check if this is really a pat::Electron, and if yes, get the track ref from this new
  // pointer instead
  if( elePatPtr != NULL )
    myTrackRef = elePatPtr->closestCtfTrackRef();
  validKF = (myTrackRef.isAvailable() && (myTrackRef.isNonnull()) );  

  //Pure tracking variables
  allMVAVars.kfhits          = (validKF) ? myTrackRef->hitPattern().trackerLayersWithMeasurement() : -1. ;
  allMVAVars.kfchi2          = (validKF) ? myTrackRef->normalizedChi2() : 0;
  allMVAVars.gsfchi2         = eleRecoPtr->gsfTrack()->normalizedChi2();

  // Energy matching
  allMVAVars.fbrem           = eleRecoPtr->fbrem();

  allMVAVars.gsfhits         = eleRecoPtr->gsfTrack()->hitPattern().trackerLayersWithMeasurement();
  allMVAVars.expectedMissingInnerHits = eleRecoPtr->gsfTrack()
    ->hitPattern().numberOfHits(reco::HitPattern::MISSING_INNER_HITS);

  reco::ConversionRef conv_ref = ConversionTools::matchedConversion(*eleRecoPtr,
                                    conversions, 
                                    theBeamSpot->position());
  double vertexFitProbability = -1.; 
  if(!conv_ref.isNull()) {
    const reco::Vertex &vtx = conv_ref.get()->conversionVertex(); if (vtx.isValid()) {
      vertexFitProbability = TMath::Prob( vtx.chi2(), vtx.ndof());
    } 
  }
  allMVAVars.convVtxFitProbability    = vertexFitProbability;

  allMVAVars.EoP             = eleRecoPtr->eSuperClusterOverP();
  allMVAVars.eleEoPout       = eleRecoPtr->eEleClusterOverPout();
  float pAtVertex            = eleRecoPtr->trackMomentumAtVtx().R();
  allMVAVars.IoEmIoP         = (1.0/eleRecoPtr->ecalEnergy()) - (1.0 / pAtVertex );

  // Geometrical matchings
  allMVAVars.deta            = eleRecoPtr->deltaEtaSuperClusterTrackAtVtx();
  allMVAVars.dphi            = eleRecoPtr->deltaPhiSuperClusterTrackAtVtx();
  allMVAVars.detacalo        = eleRecoPtr->deltaEtaSeedClusterTrackAtCalo();

  // Spectator variables  
  allMVAVars.pt              = eleRecoPtr->pt();
  float scEta = superCluster->eta();
  constexpr float ebeeSplit = 1.479;
  allMVAVars.isBarrel        = ( std::abs(scEta) < ebeeSplit );
  allMVAVars.isEndcap        = ( std::abs(scEta) >= ebeeSplit );
  allMVAVars.SCeta           = scEta;
  allMVAVars.pt              = eleRecoPtr->pt();

  constrainMVAVariables(allMVAVars);

  std::vector<float> vars;

  if( isEndcapCategory( findCategory( eleRecoPtr ) ) ) {
    vars = std::move( packMVAVariables(allMVAVars.see,
                                       allMVAVars.spp,
                                       allMVAVars.OneMinusE1x5E5x5,
                                       allMVAVars.R9,
                                       allMVAVars.etawidth,
                                       allMVAVars.phiwidth,
                                       allMVAVars.HoE,
                                      //Pure tracking variables
                                       allMVAVars.kfhits,
                                       allMVAVars.kfchi2,
                                       allMVAVars.gsfchi2,                                       
                                       // Energy matching
                                       allMVAVars.fbrem,                                       
                                       allMVAVars.gsfhits,
                                       allMVAVars.expectedMissingInnerHits,
                                       allMVAVars.convVtxFitProbability,
                                       allMVAVars.EoP,
                                       allMVAVars.eleEoPout,
                                       allMVAVars.IoEmIoP,                                       
                                       // Geometrical matchings
                                       allMVAVars.deta,
                                       allMVAVars.dphi,
                                       allMVAVars.detacalo,                                       

                                       allMVAVars.pt,
                                       allMVAVars.SCeta,                                       
                                        // Endcap only variables
                                       allMVAVars.PreShowerOverRaw)
                      );
  } else {
    vars = std::move( packMVAVariables(allMVAVars.see,
                                       allMVAVars.spp,
                                       allMVAVars.OneMinusE1x5E5x5,
                                       allMVAVars.R9,
                                       allMVAVars.etawidth,
                                       allMVAVars.phiwidth,
                                       allMVAVars.HoE,
                                       //Pure tracking variables
                                       allMVAVars.kfhits,
                                       allMVAVars.kfchi2,
                                       allMVAVars.gsfchi2,                                       
                                       // Energy matching
                                       allMVAVars.fbrem,                                       
                                       allMVAVars.gsfhits,
                                       allMVAVars.expectedMissingInnerHits,
                                       allMVAVars.convVtxFitProbability,
                                       allMVAVars.EoP,
                                       allMVAVars.eleEoPout,
                                       allMVAVars.IoEmIoP,                                       
                                       // Geometrical matchings
                                       allMVAVars.deta,
                                       allMVAVars.dphi,
                                       allMVAVars.detacalo,                                       
                                       allMVAVars.pt,
                                       allMVAVars.SCeta)
                     );
  }
  return vars;
}

int ElectronMVAEstimatorRun2Spring16GeneralPurpose::findCategory ( const edm::Ptr< reco::Candidate > &  particle ) const

overridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 130 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.cc.

References edm::Ptr< T >::get(), and NULL.

Referenced by fillMVAVariables(), getTag(), and mvaValue().

                                                                                                             {
  
  // Try to cast the particle into a reco particle.
  // This should work for both reco and pat.
  const edm::Ptr<reco::GsfElectron> eleRecoPtr = ( edm::Ptr<reco::GsfElectron> )particle;
  if( eleRecoPtr.get() == NULL )
    throw cms::Exception("MVA failure: ")
      << " given particle is expected to be reco::GsfElectron or pat::Electron," << std::endl
      << " but appears to be neither" << std::endl;
   return findCategory(eleRecoPtr.get());
}

int ElectronMVAEstimatorRun2Spring16GeneralPurpose::findCategory

(

const reco::GsfElectron *

particle

)

const

Definition at line 142 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.cc.

References funct::abs(), CAT_EB1_PT10plus, CAT_EB2_PT10plus, CAT_EE_PT10plus, stringResolutionProvider_cfi::eta, isEndcapCategory(), reco::GsfElectron::superCluster(), and UNDEFINED.

                                                                                                           {
  float eta = eleRecoPtr->superCluster()->eta();

  //
  // Determine the category
  //
  int  iCategory = UNDEFINED;
  const float ebSplit = 0.800;// barrel is split into two regions
  const float ebeeSplit = 1.479; // division between barrel and endcap


  if (std::abs(eta) < ebSplit) 
    iCategory = CAT_EB1_PT10plus;

  if (std::abs(eta) >= ebSplit && std::abs(eta) < ebeeSplit)
    iCategory = CAT_EB2_PT10plus;

  if (std::abs(eta) >= ebeeSplit) 
    iCategory = CAT_EE_PT10plus;
  
  return iCategory;
}

virtual const std::string& ElectronMVAEstimatorRun2Spring16GeneralPurpose::getName ( void  ) const

inlinefinaloverridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 105 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.h.

References name_.

Referenced by plotting.Plot::draw().

{ return name_; }

virtual int ElectronMVAEstimatorRun2Spring16GeneralPurpose::getNCategories ( ) const

inlineoverridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 103 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.h.

References python.rootplot.argparse::category, isEndcapCategory(), and nCategories.

{ return nCategories; }

virtual const std::string& ElectronMVAEstimatorRun2Spring16GeneralPurpose::getTag ( void  ) const

inlinefinaloverridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 106 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.h.

References ecalDrivenElectronSeedsParameters_cff::beamSpot, constrainMVAVariables(), conversions_cfi::conversions, fillMVAVariables(), findCategory(), setConsumes(), and tag_.

{ return tag_; }

void ElectronMVAEstimatorRun2Spring16GeneralPurpose::init

(

const std::vector< std::string >

weightFileNames

)

Definition at line 26 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.cc.

References createSingleReader(), Exception, gbrForest_s, mps_fire::i, nCategories, and candidateChargeBTagComputer_cfi::weightFile.

Referenced by ElectronMVAEstimatorRun2Spring16GeneralPurpose().

                                                                                                     {
  if( (int)(weightFileNames.size()) != nCategories )
    throw cms::Exception("MVA config failure: ")
      << "wrong number of weightfiles" << std::endl;

  gbrForest_s.clear();
  // Create a TMVA reader object for each category
  for(int i=0; i<nCategories; i++){

    // Use unique_ptr so that all readers are properly cleaned up
    // when the vector clear() is called in the destructor

    edm::FileInPath weightFile( weightFileNames[i] );
    gbrForest_s.push_back( createSingleReader(i, weightFile ) );

  }

}

bool ElectronMVAEstimatorRun2Spring16GeneralPurpose::isEndcapCategory

(

int

category

)

const

Definition at line 166 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.cc.

References CAT_EE_PT10plus, createSingleReader(), and GeomDetEnumerators::isEndcap().

Referenced by createSingleReader(), fillMVAVariables(), findCategory(), and getNCategories().

                                      {

  bool isEndcap = false;
  if( category == CAT_EE_PT10plus )
    isEndcap = true;

  return isEndcap;
}

float ElectronMVAEstimatorRun2Spring16GeneralPurpose::mvaValue ( const edm::Ptr< reco::Candidate > &  particle, const edm::Event &  iEvent  ) const

overridevirtual

Implements AnyMVAEstimatorRun2Base.

Definition at line 79 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.cc.

References fillMVAVariables(), findCategory(), and eostools::move().

Referenced by mvaValue(), and setConsumes().

                                                                               {
  
  const int iCategory = findCategory( particle );
  const std::vector<float> vars = std::move( fillMVAVariables( particle, iEvent ) );  
  return mvaValue(iCategory, vars);
}

float ElectronMVAEstimatorRun2Spring16GeneralPurpose::mvaValue	(	const reco::GsfElectron *	particle,
		const edm::EventBase &	iEvent
	)		const

Definition at line 87 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.cc.

References ecalDrivenElectronSeedsParameters_cff::beamSpot, beamSpotLabel_, conversions_cfi::conversions, conversionsLabelAOD_, fillMVAVariables(), findCategory(), edm::EventBase::getByLabel(), eostools::move(), mvaValue(), and edm::Handle< T >::product().

                                                                               {
  edm::Handle<reco::ConversionCollection> conversions;
  edm::Handle<reco::BeamSpot> beamSpot;
  iEvent.getByLabel(conversionsLabelAOD_, conversions);
  iEvent.getByLabel(beamSpotLabel_, beamSpot);
  const int iCategory = findCategory( particle );
  const std::vector<float> vars = std::move( fillMVAVariables( particle, conversions, beamSpot.product() ) );  
  return mvaValue(iCategory, vars);
}

float ElectronMVAEstimatorRun2Spring16GeneralPurpose::mvaValue	(	const int	iCategory,
		const std::vector< float > &	vars
	)		const

Definition at line 98 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.cc.

References gather_cfg::cout, debug, gbrForest_s, MethodName_, and mps_fire::result.

                                                                     {
  const float result = gbrForest_s.at(iCategory)->GetClassifier(vars.data());

  const bool debug = false;
  if(debug) {
    std::cout << " *** Inside the class MethodName_ " << MethodName_ << std::endl;
    std::cout << " bin " << iCategory
          << " fbrem " <<  vars[11]
          << " kfchi2 " << vars[9]
          << " mykfhits " << vars[8]
          << " gsfchi2 " << vars[10]
          << " deta " <<  vars[18]
          << " dphi " << vars[19]
          << " detacalo " << vars[20]
          << " see " << vars[0]
          << " spp " << vars[1]
          << " etawidth " << vars[4]
          << " phiwidth " << vars[5]
          << " OneMinusE1x5E5x5 " << vars[2]
          << " R9 " << vars[3]
          << " HoE " << vars[6]
          << " EoP " << vars[15]
          << " IoEmIoP " << vars[17]
          << " eleEoPout " << vars[16]
          << " eta " << vars[24]
          << " pt " << vars[21] << std::endl;
    std::cout << " ### MVA " << result << std::endl;
  }

  return result;
}

void ElectronMVAEstimatorRun2Spring16GeneralPurpose::setConsumes ( edm::ConsumesCollector &&  cc ) const

finaloverridevirtual

Reimplemented from AnyMVAEstimatorRun2Base.

Definition at line 64 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.cc.

References beamSpotLabel_, conversionsLabelAOD_, conversionsLabelMiniAOD_, and mvaValue().

Referenced by getTag().

                                                                                                {

  // All tokens for event content needed by this MVA

  // Beam spot (same for AOD and miniAOD)
   cc.consumes<reco::BeamSpot>(beamSpotLabel_);

  // Conversions collection (different names in AOD and miniAOD)
  cc.mayConsume<reco::ConversionCollection>(conversionsLabelAOD_);
  cc.mayConsume<reco::ConversionCollection>(conversionsLabelMiniAOD_);
  

}

Member Data Documentation

AllVariables ElectronMVAEstimatorRun2Spring16GeneralPurpose::allMVAVars_

private

Definition at line 140 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.h.

Referenced by createSingleReader().

const edm::InputTag ElectronMVAEstimatorRun2Spring16GeneralPurpose::beamSpotLabel_

private

Definition at line 146 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.h.

Referenced by fillMVAVariables(), mvaValue(), and setConsumes().

const edm::InputTag ElectronMVAEstimatorRun2Spring16GeneralPurpose::conversionsLabelAOD_

private

Definition at line 148 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.h.

Referenced by fillMVAVariables(), mvaValue(), and setConsumes().

const edm::InputTag ElectronMVAEstimatorRun2Spring16GeneralPurpose::conversionsLabelMiniAOD_

private

Definition at line 149 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.h.

Referenced by fillMVAVariables(), and setConsumes().

std::vector< std::unique_ptr<const GBRForest> > ElectronMVAEstimatorRun2Spring16GeneralPurpose::gbrForest_s

private

Definition at line 136 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.h.

Referenced by init(), and mvaValue().

const std::string ElectronMVAEstimatorRun2Spring16GeneralPurpose::MethodName_

private

Definition at line 139 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.h.

Referenced by createSingleReader(), and mvaValue().

const std::string ElectronMVAEstimatorRun2Spring16GeneralPurpose::name_ = "ElectronMVAEstimatorRun2Spring16GeneralPurpose"

private

Definition at line 129 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.h.

Referenced by getName().

const int ElectronMVAEstimatorRun2Spring16GeneralPurpose::nCategories = 3

Definition at line 28 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.h.

Referenced by getNCategories(), and init().

const std::string ElectronMVAEstimatorRun2Spring16GeneralPurpose::tag_

private

Definition at line 133 of file ElectronMVAEstimatorRun2Spring16GeneralPurpose.h.

Referenced by getTag().