CMS 3D CMS Logo

List of all members | Public Member Functions | Private Member Functions | Private Attributes
PATTauDiscriminationAgainstElectronMVA6 Class Reference
Inheritance diagram for PATTauDiscriminationAgainstElectronMVA6:
TauDiscriminationProducerBase< TauType, TauDiscriminator > edm::stream::EDProducer<> edm::stream::EDProducerBase edm::ProducerBase edm::EDConsumerBase edm::ProductRegistryHelper

Public Member Functions

void beginEvent (const edm::Event &, const edm::EventSetup &)
 
double discriminate (const TauRef &) const
 
void endEvent (edm::Event &)
 
 PATTauDiscriminationAgainstElectronMVA6 (const edm::ParameterSet &cfg)
 
 ~PATTauDiscriminationAgainstElectronMVA6 ()
 
- Public Member Functions inherited from TauDiscriminationProducerBase< TauType, TauDiscriminator >
void produce (edm::Event &, const edm::EventSetup &)
 
 TauDiscriminationProducerBase (const edm::ParameterSet &iConfig)
 
 TauDiscriminationProducerBase ()
 
virtual ~TauDiscriminationProducerBase ()
 
- Public Member Functions inherited from edm::stream::EDProducer<>
 EDProducer ()=default
 
- Public Member Functions inherited from edm::stream::EDProducerBase
 EDProducerBase ()
 
ModuleDescription const & moduleDescription () const
 
virtual ~EDProducerBase ()
 
- Public Member Functions inherited from edm::ProducerBase
void callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
 
std::vector< edm::ProductResolverIndex > const & indiciesForPutProducts (BranchType iBranchType) const
 
 ProducerBase ()
 
void registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
 
std::function< void(BranchDescription const &)> registrationCallback () const
 used by the fwk to register list of products More...
 
void resolvePutIndicies (BranchType iBranchType, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)
 
virtual ~ProducerBase () noexcept(false)
 
- Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfoconsumesInfo () const
 
 EDConsumerBase ()
 
 EDConsumerBase (EDConsumerBase const &)=delete
 
 EDConsumerBase (EDConsumerBase &&)=default
 
ProductResolverIndexAndSkipBit indexFrom (EDGetToken, BranchType, TypeID const &) const
 
void itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
void itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
std::vector< ProductResolverIndexAndSkipBit > const & itemsToGetFromEvent () const
 
void labelsForToken (EDGetToken iToken, Labels &oLabels) const
 
void modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
 
EDConsumerBase const & operator= (EDConsumerBase const &)=delete
 
EDConsumerBaseoperator= (EDConsumerBase &&)=default
 
bool registeredToConsume (ProductResolverIndex, bool, BranchType) const
 
bool registeredToConsumeMany (TypeID const &, BranchType) const
 
void updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
 
virtual ~EDConsumerBase () noexcept(false)
 

Private Member Functions

bool isInEcalCrack (double) const
 

Private Attributes

std::unique_ptr< PATTauDiscriminatorcategory_output_
 
edm::Handle< pat::ElectronCollectionElectrons
 
edm::EDGetTokenT< pat::ElectronCollectionelectronToken
 
std::string moduleLabel_
 
std::unique_ptr< AntiElectronIDMVA6mva_
 
edm::InputTag srcElectrons
 
edm::Handle< TauCollectiontaus_
 
bool usePhiAtEcalEntranceExtrapolation_
 
int verbosity_
 

Additional Inherited Members

- Public Types inherited from TauDiscriminationProducerBase< TauType, TauDiscriminator >
typedef std::vector< TauType > TauCollection
 
typedef edm::Ref< TauCollectionTauRef
 
typedef edm::RefProd< TauCollectionTauRefProd
 
- Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... > CacheTypes
 
typedef CacheTypes::GlobalCache GlobalCache
 
typedef AbilityChecker< T... > HasAbility
 
typedef CacheTypes::LuminosityBlockCache LuminosityBlockCache
 
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCacheLuminosityBlockContext
 
typedef CacheTypes::LuminosityBlockSummaryCache LuminosityBlockSummaryCache
 
typedef CacheTypes::RunCache RunCache
 
typedef RunContextT< RunCache, GlobalCacheRunContext
 
typedef CacheTypes::RunSummaryCache RunSummaryCache
 
- Public Types inherited from edm::stream::EDProducerBase
typedef EDProducerAdaptorBase ModuleType
 
- Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList TypeLabelList
 
- Public Types inherited from edm::EDConsumerBase
typedef ProductLabels Labels
 
- Static Public Member Functions inherited from edm::stream::EDProducerBase
static const std::string & baseType ()
 
static void fillDescriptions (ConfigurationDescriptions &descriptions)
 
static void prevalidate (ConfigurationDescriptions &descriptions)
 
- 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)
 
- Protected Attributes inherited from TauDiscriminationProducerBase< TauType, TauDiscriminator >
std::string moduleLabel_
 
double prediscriminantFailValue_
 
edm::EDGetTokenT< TauCollectionTau_token
 
size_t tauIndex_
 
edm::InputTag TauProducer_
 

Detailed Description

Definition at line 24 of file PATTauDiscriminationAgainstElectronMVA6.cc.

Constructor & Destructor Documentation

PATTauDiscriminationAgainstElectronMVA6::PATTauDiscriminationAgainstElectronMVA6 ( const edm::ParameterSet cfg)
inlineexplicit

Definition at line 27 of file PATTauDiscriminationAgainstElectronMVA6.cc.

References looper::cfg, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), and PATTauDiscriminationAgainstElectronMVA6_cfi::srcElectrons.

29  mva_(),
31  {
32  mva_ = std::make_unique<AntiElectronIDMVA6>(cfg);
33 
34  usePhiAtEcalEntranceExtrapolation_ = cfg.getParameter<bool>("usePhiAtEcalEntranceExtrapolation");
35  srcElectrons = cfg.getParameter<edm::InputTag>("srcElectrons");
36  electronToken = consumes<pat::ElectronCollection>(srcElectrons);
37  verbosity_ = ( cfg.exists("verbosity") ) ?
38  cfg.getParameter<int>("verbosity") : 0;
39 
40  // add category index
41  produces<PATTauDiscriminator>("category");
42  }
T getParameter(std::string const &) const
bool exists(std::string const &parameterName) const
checks if a parameter exists
TauDiscriminationProducerBase< pat::Tau, pat::PATTauDiscriminator > PATTauDiscriminationProducerBase
edm::EDGetTokenT< pat::ElectronCollection > electronToken
PATTauDiscriminationAgainstElectronMVA6::~PATTauDiscriminationAgainstElectronMVA6 ( )
inline

Definition at line 50 of file PATTauDiscriminationAgainstElectronMVA6.cc.

50 {}

Member Function Documentation

void PATTauDiscriminationAgainstElectronMVA6::beginEvent ( const edm::Event evt,
const edm::EventSetup es 
)
virtual

Reimplemented from TauDiscriminationProducerBase< TauType, TauDiscriminator >.

Definition at line 69 of file PATTauDiscriminationAgainstElectronMVA6.cc.

References JetPlusTrackCorrections_cfi::Electrons, and edm::Event::getByToken().

70 {
71  mva_->beginEvent(evt, es);
72 
75 
77 }
edm::RefProd< TauCollection > TauRefProd
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:457
edm::EDGetTokenT< pat::ElectronCollection > electronToken
edm::EDGetTokenT< TauCollection > Tau_token
double PATTauDiscriminationAgainstElectronMVA6::discriminate ( const TauRef theTauRef) const
virtual

Implements TauDiscriminationProducerBase< TauType, TauDiscriminator >.

Definition at line 79 of file PATTauDiscriminationAgainstElectronMVA6.cc.

References funct::abs(), edm::PtrVector< T >::begin(), python.rootplot.argparse::category, deltaR(), PFRecoTauDiscriminationAgainstElectronDeadECAL_cfi::dR, JetPlusTrackCorrections_cfi::Electrons, edm::PtrVector< T >::end(), CustomPhysics_cfi::gamma, edm::Ref< C, T, F >::get(), pat::Electron::gsfTrack(), edm::Ref< C, T, F >::isNonnull(), hpstanc_transforms::max, min(), reco::GsfElectron::p4(), pat::PackedCandidate::pdgId(), and reco::LeafCandidate::pt().

80 {
81  double mvaValue = 1.;
82  double category = -1.;
83  bool isGsfElectronMatched = false;
84  float deltaRDummy = 9.9;
85  const float ECALBarrelEndcapEtaBorder = 1.479;
86  float tauEtaAtEcalEntrance = theTauRef->etaAtEcalEntrance();
87  float leadChargedPFCandEtaAtEcalEntrance = theTauRef->etaAtEcalEntranceLeadChargedCand();
88 
89  if( (*theTauRef).leadChargedHadrCand().isNonnull()) {
90  int numSignalPFGammaCandsInSigCone = 0;
91  const reco::CandidatePtrVector signalGammaCands = theTauRef->signalGammaCands();
92  for( reco::CandidatePtrVector::const_iterator gamma = signalGammaCands.begin(); gamma != signalGammaCands.end(); ++gamma ){
93  double dR = deltaR((*gamma)->p4(), theTauRef->leadChargedHadrCand()->p4());
94  double signalrad = std::max(0.05, std::min(0.10, 3.0/std::max(1.0, theTauRef->pt())));
95  // gammas inside the tau signal cone
96  if (dR < signalrad) {
97  numSignalPFGammaCandsInSigCone += 1;
98  }
99  }
100  // loop over the electrons
101  for( unsigned int ie = 0; ie < Electrons->size(); ++ie ){
102  const pat::Electron& theElectron = Electrons->at(ie);
103  if ( theElectron.pt() > 10. ) { // CV: only take electrons above some minimal energy/Pt into account...
104  double deltaREleTau = deltaR(theElectron.p4(), theTauRef->p4());
105  deltaRDummy = deltaREleTau;
106  if( deltaREleTau < 0.3 ){
107  double mva_match = mva_->MVAValue(*theTauRef, theElectron, usePhiAtEcalEntranceExtrapolation_);
108  bool hasGsfTrack = 0;
109  pat::PackedCandidate const* packedLeadTauCand = dynamic_cast<pat::PackedCandidate const*>(theTauRef->leadChargedHadrCand().get());
110  if( abs(packedLeadTauCand->pdgId()) == 11 )
111  hasGsfTrack = 1;
112  if ( !hasGsfTrack )
113  hasGsfTrack = theElectron.gsfTrack().isNonnull();
114 
115  // veto taus that go to Ecal crack
116  if ( isInEcalCrack(tauEtaAtEcalEntrance) || isInEcalCrack(leadChargedPFCandEtaAtEcalEntrance) ) {
117  // add category index
118  category_output_->setValue(tauIndex_, category);
119  // return MVA output value
120  return -99;
121  }
122  // Veto taus that go to Ecal crack
123  if( std::abs(tauEtaAtEcalEntrance) < ECALBarrelEndcapEtaBorder ){ // Barrel
124  if( numSignalPFGammaCandsInSigCone == 0 && hasGsfTrack ){
125  category = 5.;
126  }
127  else if ( numSignalPFGammaCandsInSigCone >= 1 && hasGsfTrack ) {
128  category = 7.;
129  }
130  } else { // Endcap
131  if ( numSignalPFGammaCandsInSigCone == 0 && hasGsfTrack ) {
132  category = 13.;
133  }
134  else if ( numSignalPFGammaCandsInSigCone >= 1 && hasGsfTrack ) {
135  category = 15.;
136  }
137  }
138  mvaValue = std::min(mvaValue, mva_match);
139  isGsfElectronMatched = true;
140  } // deltaR < 0.3
141  } // electron pt > 10
142  } // end of loop over electrons
143 
144  if ( !isGsfElectronMatched ) {
145  mvaValue = mva_->MVAValue(*theTauRef, usePhiAtEcalEntranceExtrapolation_);
146  bool hasGsfTrack = 0;
147  pat::PackedCandidate const* packedLeadTauCand = dynamic_cast<pat::PackedCandidate const*>(theTauRef->leadChargedHadrCand().get());
148  if( abs(packedLeadTauCand->pdgId()) == 11 ) hasGsfTrack = 1;
149 
150  // veto taus that go to Ecal crack
151  if ( isInEcalCrack(tauEtaAtEcalEntrance) || isInEcalCrack(leadChargedPFCandEtaAtEcalEntrance) ) {
152  // add category index
153  category_output_->setValue(tauIndex_, category);
154  // return MVA output value
155  return -99;
156  }
157  // veto taus that go to Ecal crack
158  if( std::abs(tauEtaAtEcalEntrance) < ECALBarrelEndcapEtaBorder ){ // Barrel
159  if( numSignalPFGammaCandsInSigCone == 0 && !hasGsfTrack ){
160  category = 0.;
161  }
162  else if ( numSignalPFGammaCandsInSigCone >= 1 && !hasGsfTrack ) {
163  category = 2.;
164  }
165  } else { // Endcap
166  if ( numSignalPFGammaCandsInSigCone == 0 && !hasGsfTrack ) {
167  category = 8.;
168  }
169  else if ( numSignalPFGammaCandsInSigCone >= 1 && !hasGsfTrack ) {
170  category = 10.;
171  }
172  }
173  }
174  }
175  if ( verbosity_ ) {
176  edm::LogPrint("PATTauAgainstEleMVA6") << "<PATTauDiscriminationAgainstElectronMVA6::discriminate>:" ;
177  edm::LogPrint("PATTauAgainstEleMVA6") << " tau: Pt = " << theTauRef->pt() << ", eta = " << theTauRef->eta() << ", phi = " << theTauRef->phi();
178  edm::LogPrint("PATTauAgainstEleMVA6") << " deltaREleTau = " << deltaRDummy << ", isGsfElectronMatched = " << isGsfElectronMatched;
179  edm::LogPrint("PATTauAgainstEleMVA6") << " #Prongs = " << theTauRef->signalChargedHadrCands().size();
180  edm::LogPrint("PATTauAgainstEleMVA6") << " MVA = " << mvaValue << ", category = " << category;
181  }
182  // add category index
183  category_output_->setValue(tauIndex_, category);
184  // return MVA output value
185  return mvaValue;
186 }
virtual double pt() const final
transverse momentum
bool isNonnull() const
Checks for non-null.
Definition: Ref.h:252
const LorentzVector & p4(P4Kind kind) const
Definition: GsfElectron.cc:225
const_iterator begin() const
Definition: PtrVector.h:130
reco::GsfTrackRef gsfTrack() const
override the reco::GsfElectron::gsfTrack method, to access the internal storage of the supercluster ...
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
const_iterator end() const
Definition: PtrVector.h:135
T const * get() const
Returns C++ pointer to the item.
Definition: Ref.h:244
T min(T a, T b)
Definition: MathUtil.h:58
double deltaR(double eta1, double eta2, double phi1, double phi2)
Definition: TreeUtility.cc:17
Analysis-level electron class.
Definition: Electron.h:52
virtual int pdgId() const
PDG identifier.
void PATTauDiscriminationAgainstElectronMVA6::endEvent ( edm::Event evt)
virtual

Reimplemented from TauDiscriminationProducerBase< TauType, TauDiscriminator >.

Definition at line 188 of file PATTauDiscriminationAgainstElectronMVA6.cc.

References eostools::move(), and edm::Event::put().

189 {
190  // add all category indices to event
191  evt.put(std::move(category_output_), "category");
192 }
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:122
def move(src, dest)
Definition: eostools.py:510
bool PATTauDiscriminationAgainstElectronMVA6::isInEcalCrack ( double  eta) const
private

Definition at line 195 of file PATTauDiscriminationAgainstElectronMVA6.cc.

References DEFINE_FWK_MODULE.

196 {
197  double absEta = fabs(eta);
198  return (absEta > 1.460 && absEta < 1.558);
199 }

Member Data Documentation

std::unique_ptr<PATTauDiscriminator> PATTauDiscriminationAgainstElectronMVA6::category_output_
private

Definition at line 63 of file PATTauDiscriminationAgainstElectronMVA6.cc.

edm::Handle<pat::ElectronCollection> PATTauDiscriminationAgainstElectronMVA6::Electrons
private

Definition at line 60 of file PATTauDiscriminationAgainstElectronMVA6.cc.

edm::EDGetTokenT<pat::ElectronCollection> PATTauDiscriminationAgainstElectronMVA6::electronToken
private

Definition at line 59 of file PATTauDiscriminationAgainstElectronMVA6.cc.

std::string PATTauDiscriminationAgainstElectronMVA6::moduleLabel_
private
std::unique_ptr<AntiElectronIDMVA6> PATTauDiscriminationAgainstElectronMVA6::mva_
private

Definition at line 56 of file PATTauDiscriminationAgainstElectronMVA6.cc.

edm::InputTag PATTauDiscriminationAgainstElectronMVA6::srcElectrons
private

Definition at line 58 of file PATTauDiscriminationAgainstElectronMVA6.cc.

edm::Handle<TauCollection> PATTauDiscriminationAgainstElectronMVA6::taus_
private

Definition at line 61 of file PATTauDiscriminationAgainstElectronMVA6.cc.

bool PATTauDiscriminationAgainstElectronMVA6::usePhiAtEcalEntranceExtrapolation_
private

Definition at line 64 of file PATTauDiscriminationAgainstElectronMVA6.cc.

int PATTauDiscriminationAgainstElectronMVA6::verbosity_
private

Definition at line 66 of file PATTauDiscriminationAgainstElectronMVA6.cc.