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<>

Public Member Functions

void beginEvent (const edm::Event &, const edm::EventSetup &) override
 
double discriminate (const TauRef &) const override
 
void endEvent (edm::Event &) override
 
 PATTauDiscriminationAgainstElectronMVA6 (const edm::ParameterSet &cfg)
 
 ~PATTauDiscriminationAgainstElectronMVA6 () override
 
- Public Member Functions inherited from TauDiscriminationProducerBase< TauType, TauDiscriminator >
void produce (edm::Event &, const edm::EventSetup &) override
 
 TauDiscriminationProducerBase (const edm::ParameterSet &iConfig)
 
 TauDiscriminationProducerBase ()
 
 ~TauDiscriminationProducerBase () override
 
- Public Member Functions inherited from edm::stream::EDProducer<>
 EDProducer ()=default
 
bool hasAbilityToProduceInLumis () const final
 
bool hasAbilityToProduceInRuns () const final
 

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_
 
int verbosity_
 
bool vetoEcalCracks_
 

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
 
- 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  srcElectrons = cfg.getParameter<edm::InputTag>("srcElectrons");
35  electronToken = consumes<pat::ElectronCollection>(srcElectrons);
36  vetoEcalCracks_ = cfg.getParameter<bool>("vetoEcalCracks");
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 ( )
inlineoverride

Definition at line 50 of file PATTauDiscriminationAgainstElectronMVA6.cc.

50 {}

Member Function Documentation

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

Reimplemented from TauDiscriminationProducerBase< TauType, TauDiscriminator >.

Definition at line 70 of file PATTauDiscriminationAgainstElectronMVA6.cc.

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

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

Implements TauDiscriminationProducerBase< TauType, TauDiscriminator >.

Definition at line 80 of file PATTauDiscriminationAgainstElectronMVA6.cc.

References funct::abs(), taus_updatedMVAIds_cff::category, deltaR(), PFRecoTauDiscriminationAgainstElectronDeadECAL_cfi::dR, JetPlusTrackCorrections_cfi::Electrons, CustomPhysics_cfi::gamma, edm::Ref< C, T, F >::get(), SiStripPI::max, min(), and pat::PackedCandidate::pdgId().

81 {
82  double mvaValue = 1.;
83  double category = -1.;
84  bool isGsfElectronMatched = false;
85  float deltaRDummy = 9.9;
86  const float ECALBarrelEndcapEtaBorder = 1.479;
87  float tauEtaAtEcalEntrance = theTauRef->etaAtEcalEntrance();
88  float leadChargedPFCandEtaAtEcalEntrance = theTauRef->etaAtEcalEntranceLeadChargedCand();
89 
90  if( (*theTauRef).leadChargedHadrCand().isNonnull()) {
91  int numSignalPFGammaCandsInSigCone = 0;
92  const reco::CandidatePtrVector signalGammaCands = theTauRef->signalGammaCands();
93  for ( const auto & gamma : signalGammaCands ) {
94  double dR = deltaR(gamma->p4(), theTauRef->leadChargedHadrCand()->p4());
95  double signalrad = std::max(0.05, std::min(0.10, 3.0/std::max(1.0, theTauRef->pt())));
96  // gammas inside the tau signal cone
97  if (dR < signalrad) {
98  numSignalPFGammaCandsInSigCone += 1;
99  }
100  }
101  // loop over the electrons
102  for ( const auto & theElectron : *Electrons ) {
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);
108  bool hasGsfTrack = false;
109  pat::PackedCandidate const* packedLeadTauCand = dynamic_cast<pat::PackedCandidate const*>(theTauRef->leadChargedHadrCand().get());
110  if( abs(packedLeadTauCand->pdgId()) == 11 )
111  hasGsfTrack = true;
112  if ( !hasGsfTrack )
113  hasGsfTrack = theElectron.gsfTrack().isNonnull();
114 
115  // veto taus that go to Ecal crack
116  if ( vetoEcalCracks_ && (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);
146  bool hasGsfTrack = false;
147  pat::PackedCandidate const* packedLeadTauCand = dynamic_cast<pat::PackedCandidate const*>(theTauRef->leadChargedHadrCand().get());
148  if( abs(packedLeadTauCand->pdgId()) == 11 ) hasGsfTrack = true;
149 
150  // veto taus that go to Ecal crack
151  if ( vetoEcalCracks_ && (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 }
int pdgId() const override
PDG identifier.
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
T const * get() const
Returns C++ pointer to the item.
Definition: Ref.h:245
T min(T a, T b)
Definition: MathUtil.h:58
double deltaR(double eta1, double eta2, double phi1, double phi2)
Definition: TreeUtility.cc:17
void PATTauDiscriminationAgainstElectronMVA6::endEvent ( edm::Event evt)
overridevirtual

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:137
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.

int PATTauDiscriminationAgainstElectronMVA6::verbosity_
private

Definition at line 67 of file PATTauDiscriminationAgainstElectronMVA6.cc.

bool PATTauDiscriminationAgainstElectronMVA6::vetoEcalCracks_
private

Definition at line 65 of file PATTauDiscriminationAgainstElectronMVA6.cc.