Skip to content

Commit

Permalink
Merge pull request #25316 from cms-tau-pog/CMSSW_9_4_X_tau-pog_update…
Browse files Browse the repository at this point in the history 
…AntiEDisc

Update anti-e tau discriminant (94X)
  • Loading branch information
@cmsbuild
cmsbuild committed Dec 5, 2018
2 parents 8abc46d + f2e1266 commit 182d723
Show file tree
Hide file tree
Showing 5 changed files with 132 additions and 150 deletions.
12 changes: 7 additions & 5 deletions RecoTauTag/RecoTau/interface/AntiElectronIDMVA6.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -123,15 +123,15 @@ class AntiElectronIDMVA6

// this function can be called for all categories
double MVAValue(const reco::PFTau& thePFTau,
const reco::GsfElectron& theGsfEle, bool usePhiAtEcalEntranceExtrapolation);
const reco::GsfElectron& theGsfEle);
// this function can be called for category 1 only !!
double MVAValue(const reco::PFTau& thePFTau, bool usePhiAtEcalEntranceExtrapolation);
double MVAValue(const reco::PFTau& thePFTau);

// this function can be called for all categories
double MVAValue(const pat::Tau& theTau,
const pat::Electron& theEle, bool usePhiAtEcalEntranceExtrapolation);
const pat::Electron& theEle);
// this function can be called for category 1 only !!
double MVAValue(const pat::Tau& theTau, bool usePhiAtEcalEntranceExtrapolation);
double MVAValue(const pat::Tau& theTau);
// track extrapolation to ECAL entrance (used to re-calculate varibales that might not be available on miniAOD)
bool atECalEntrance(const reco::Candidate* part, math::XYZPoint &pos);

Expand All @@ -153,7 +153,9 @@ class AntiElectronIDMVA6
std::string mvaName_NoEleMatch_wGwoGSF_EC_;
std::string mvaName_woGwGSF_EC_;
std::string mvaName_wGwGSF_EC_;


bool usePhiAtEcalEntranceExtrapolation_;

Float_t* Var_NoEleMatch_woGwoGSF_Barrel_;
Float_t* Var_NoEleMatch_wGwoGSF_Barrel_;
Float_t* Var_woGwGSF_Barrel_;
Expand Down
13 changes: 5 additions & 8 deletions RecoTauTag/RecoTau/plugins/PATTauDiscriminationAgainstElectronMVA6.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,6 @@ class PATTauDiscriminationAgainstElectronMVA6 : public PATTauDiscriminationProdu
{
mva_ = std::make_unique<AntiElectronIDMVA6>(cfg);

usePhiAtEcalEntranceExtrapolation_ = cfg.getParameter<bool>("usePhiAtEcalEntranceExtrapolation");
srcElectrons = cfg.getParameter<edm::InputTag>("srcElectrons");
electronToken = consumes<pat::ElectronCollection>(srcElectrons);
verbosity_ = ( cfg.exists("verbosity") ) ?
Expand Down Expand Up @@ -61,7 +60,6 @@ class PATTauDiscriminationAgainstElectronMVA6 : public PATTauDiscriminationProdu
edm::Handle<TauCollection> taus_;

std::unique_ptr<PATTauDiscriminator> category_output_;
bool usePhiAtEcalEntranceExtrapolation_;

int verbosity_;
};
Expand Down Expand Up @@ -89,22 +87,21 @@ double PATTauDiscriminationAgainstElectronMVA6::discriminate(const TauRef& theTa
if( (*theTauRef).leadChargedHadrCand().isNonnull()) {
int numSignalPFGammaCandsInSigCone = 0;
const reco::CandidatePtrVector signalGammaCands = theTauRef->signalGammaCands();
for( reco::CandidatePtrVector::const_iterator gamma = signalGammaCands.begin(); gamma != signalGammaCands.end(); ++gamma ){
double dR = deltaR((*gamma)->p4(), theTauRef->leadChargedHadrCand()->p4());
for ( const auto & gamma : signalGammaCands ) {
double dR = deltaR(gamma->p4(), theTauRef->leadChargedHadrCand()->p4());
double signalrad = std::max(0.05, std::min(0.10, 3.0/std::max(1.0, theTauRef->pt())));
// gammas inside the tau signal cone
if (dR < signalrad) {
numSignalPFGammaCandsInSigCone += 1;
}
}
// loop over the electrons
for( unsigned int ie = 0; ie < Electrons->size(); ++ie ){
const pat::Electron& theElectron = Electrons->at(ie);
for ( const auto & theElectron : *Electrons ) {
if ( theElectron.pt() > 10. ) { // CV: only take electrons above some minimal energy/Pt into account...
double deltaREleTau = deltaR(theElectron.p4(), theTauRef->p4());
deltaRDummy = deltaREleTau;
if( deltaREleTau < 0.3 ){
double mva_match = mva_->MVAValue(*theTauRef, theElectron, usePhiAtEcalEntranceExtrapolation_);
double mva_match = mva_->MVAValue(*theTauRef, theElectron);
bool hasGsfTrack = false;
pat::PackedCandidate const* packedLeadTauCand = dynamic_cast<pat::PackedCandidate const*>(theTauRef->leadChargedHadrCand().get());
if( abs(packedLeadTauCand->pdgId()) == 11 )
Expand Down Expand Up @@ -142,7 +139,7 @@ double PATTauDiscriminationAgainstElectronMVA6::discriminate(const TauRef& theTa
} // end of loop over electrons

if ( !isGsfElectronMatched ) {
mvaValue = mva_->MVAValue(*theTauRef, usePhiAtEcalEntranceExtrapolation_);
mvaValue = mva_->MVAValue(*theTauRef);
bool hasGsfTrack = false;
pat::PackedCandidate const* packedLeadTauCand = dynamic_cast<pat::PackedCandidate const*>(theTauRef->leadChargedHadrCand().get());
if( abs(packedLeadTauCand->pdgId()) == 11 ) hasGsfTrack = true;
Expand Down
55 changes: 22 additions & 33 deletions RecoTauTag/RecoTau/plugins/PFRecoTauDiscriminationAgainstElectronMVA6.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -29,12 +29,11 @@ class PFRecoTauDiscriminationAgainstElectronMVA6 : public PFTauDiscriminationPro
public:
explicit PFRecoTauDiscriminationAgainstElectronMVA6(const edm::ParameterSet& cfg)
: PFTauDiscriminationProducerBase(cfg),
mva_(nullptr),
mva_(),
category_output_()
{
mva_ = new AntiElectronIDMVA6(cfg);
mva_ = std::make_unique<AntiElectronIDMVA6>(cfg);

usePhiAtEcalEntranceExtrapolation_ = cfg.getParameter<bool>("usePhiAtEcalEntranceExtrapolation");
srcGsfElectrons_ = cfg.getParameter<edm::InputTag>("srcGsfElectrons");
GsfElectrons_token = consumes<reco::GsfElectronCollection>(srcGsfElectrons_);

Expand All @@ -51,26 +50,20 @@ class PFRecoTauDiscriminationAgainstElectronMVA6 : public PFTauDiscriminationPro

void endEvent(edm::Event&) override;

~PFRecoTauDiscriminationAgainstElectronMVA6() override
{
delete mva_;
}
~PFRecoTauDiscriminationAgainstElectronMVA6() override {}

private:
bool isInEcalCrack(double) const;

std::string moduleLabel_;

AntiElectronIDMVA6* mva_;
float* mvaInput_;
std::unique_ptr<AntiElectronIDMVA6> mva_;

edm::InputTag srcGsfElectrons_;
edm::EDGetTokenT<reco::GsfElectronCollection> GsfElectrons_token;
edm::Handle<reco::GsfElectronCollection> gsfElectrons_;
edm::Handle<TauCollection> taus_;

std::unique_ptr<PFTauDiscriminator> category_output_;
bool usePhiAtEcalEntranceExtrapolation_;

int verbosity_;
};
Expand Down Expand Up @@ -98,28 +91,26 @@ double PFRecoTauDiscriminationAgainstElectronMVA6::discriminate(const PFTauRef&
float sumEtaTimesEnergy = 0.;
float sumEnergy = 0.;
const std::vector<reco::PFCandidatePtr>& signalPFCands = thePFTauRef->signalPFCands();
for ( std::vector<reco::PFCandidatePtr>::const_iterator pfCandidate = signalPFCands.begin();
pfCandidate != signalPFCands.end(); ++pfCandidate ) {
sumEtaTimesEnergy += ((*pfCandidate)->positionAtECALEntrance().eta()*(*pfCandidate)->energy());
sumEnergy += (*pfCandidate)->energy();
for ( const auto & pfCandidate : signalPFCands ) {
sumEtaTimesEnergy += (pfCandidate->positionAtECALEntrance().eta()*pfCandidate->energy());
sumEnergy += pfCandidate->energy();
}
if ( sumEnergy > 0. ) {
tauEtaAtEcalEntrance = sumEtaTimesEnergy/sumEnergy;
}

float leadChargedPFCandEtaAtEcalEntrance = -99.;
float leadChargedPFCandPt = -99.;
for ( std::vector<reco::PFCandidatePtr>::const_iterator pfCandidate = signalPFCands.begin();
pfCandidate != signalPFCands.end(); ++pfCandidate ) {
for ( const auto & pfCandidate : signalPFCands ) {
const reco::Track* track = nullptr;
if ( (*pfCandidate)->trackRef().isNonnull() ) track = (*pfCandidate)->trackRef().get();
else if ( (*pfCandidate)->muonRef().isNonnull() && (*pfCandidate)->muonRef()->innerTrack().isNonnull() ) track = (*pfCandidate)->muonRef()->innerTrack().get();
else if ( (*pfCandidate)->muonRef().isNonnull() && (*pfCandidate)->muonRef()->globalTrack().isNonnull() ) track = (*pfCandidate)->muonRef()->globalTrack().get();
else if ( (*pfCandidate)->muonRef().isNonnull() && (*pfCandidate)->muonRef()->outerTrack().isNonnull() ) track = (*pfCandidate)->muonRef()->outerTrack().get();
else if ( (*pfCandidate)->gsfTrackRef().isNonnull() ) track = (*pfCandidate)->gsfTrackRef().get();
if ( pfCandidate->trackRef().isNonnull() ) track = pfCandidate->trackRef().get();
else if ( pfCandidate->muonRef().isNonnull() && pfCandidate->muonRef()->innerTrack().isNonnull() ) track = pfCandidate->muonRef()->innerTrack().get();
else if ( pfCandidate->muonRef().isNonnull() && pfCandidate->muonRef()->globalTrack().isNonnull() ) track = pfCandidate->muonRef()->globalTrack().get();
else if ( pfCandidate->muonRef().isNonnull() && pfCandidate->muonRef()->outerTrack().isNonnull() ) track = pfCandidate->muonRef()->outerTrack().get();
else if ( pfCandidate->gsfTrackRef().isNonnull() ) track = pfCandidate->gsfTrackRef().get();
if ( track ) {
if ( track->pt() > leadChargedPFCandPt ) {
leadChargedPFCandEtaAtEcalEntrance = (*pfCandidate)->positionAtECALEntrance().eta();
leadChargedPFCandEtaAtEcalEntrance = pfCandidate->positionAtECALEntrance().eta();
leadChargedPFCandPt = track->pt();
}
}
Expand All @@ -130,10 +121,9 @@ double PFRecoTauDiscriminationAgainstElectronMVA6::discriminate(const PFTauRef&
int numSignalPFGammaCandsInSigCone = 0;
const std::vector<reco::PFCandidatePtr>& signalPFGammaCands = thePFTauRef->signalPFGammaCands();

for ( std::vector<reco::PFCandidatePtr>::const_iterator pfGamma = signalPFGammaCands.begin();
pfGamma != signalPFGammaCands.end(); ++pfGamma ) {
for ( const auto & pfGamma : signalPFGammaCands ) {

double dR = deltaR((*pfGamma)->p4(), thePFTauRef->leadPFChargedHadrCand()->p4());
double dR = deltaR(pfGamma->p4(), thePFTauRef->leadPFChargedHadrCand()->p4());
double signalrad = std::max(0.05, std::min(0.10, 3.0/std::max(1.0, thePFTauRef->pt())));

// pfGammas inside the tau signal cone
Expand All @@ -143,16 +133,15 @@ double PFRecoTauDiscriminationAgainstElectronMVA6::discriminate(const PFTauRef&
}

// loop over the electrons
for ( reco::GsfElectronCollection::const_iterator theGsfElectron = gsfElectrons_->begin();
theGsfElectron != gsfElectrons_->end(); ++theGsfElectron ) {
if ( theGsfElectron->pt() > 10. ) { // CV: only take electrons above some minimal energy/Pt into account...
double deltaREleTau = deltaR(theGsfElectron->p4(), thePFTauRef->p4());
for ( const auto & theGsfElectron : *gsfElectrons_ ) {
if ( theGsfElectron.pt() > 10. ) { // CV: only take electrons above some minimal energy/Pt into account...
double deltaREleTau = deltaR(theGsfElectron.p4(), thePFTauRef->p4());
deltaRDummy = deltaREleTau;
if ( deltaREleTau < 0.3 ) {
double mva_match = mva_->MVAValue(*thePFTauRef, *theGsfElectron, usePhiAtEcalEntranceExtrapolation_);
double mva_match = mva_->MVAValue(*thePFTauRef, theGsfElectron);
bool hasGsfTrack = thePFTauRef->leadPFChargedHadrCand()->gsfTrackRef().isNonnull();
if ( !hasGsfTrack )
hasGsfTrack = theGsfElectron->gsfTrack().isNonnull();
hasGsfTrack = theGsfElectron.gsfTrack().isNonnull();

//// Veto taus that go to Ecal crack
if ( isInEcalCrack(tauEtaAtEcalEntrance) || isInEcalCrack(leadChargedPFCandEtaAtEcalEntrance) ) {
Expand Down Expand Up @@ -186,7 +175,7 @@ double PFRecoTauDiscriminationAgainstElectronMVA6::discriminate(const PFTauRef&
} // end of loop over electrons

if ( !isGsfElectronMatched ) {
mvaValue = mva_->MVAValue(*thePFTauRef, usePhiAtEcalEntranceExtrapolation_);
mvaValue = mva_->MVAValue(*thePFTauRef);
bool hasGsfTrack = thePFTauRef->leadPFChargedHadrCand()->gsfTrackRef().isNonnull();

//// Veto taus that go to Ecal crack
Expand Down
2 changes: 1 addition & 1 deletion RecoTauTag/RecoTau/python/PATTauDiscriminationAgainstElectronMVA6_cfi.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@

patTauDiscriminationAgainstElectronMVA6 = cms.EDProducer("PATTauDiscriminationAgainstElectronMVA6",
# tau collection to discriminate
PFTauProducer = cms.InputTag('slimmedTaus'),
PATTauProducer = cms.InputTag('slimmedTaus'),

# Require leading pion ensures that:
# 1) these is at least one track above threshold (0.5 GeV) in the signal cone
Expand Down

0 comments on commit 182d723

Please sign in to comment.