Inheritance diagram for PFRecoTauDiscriminationByHPSSelection:

Classes
struct	DecayModeCuts

Public Member Functions
double	discriminate (const reco::PFTauRef &) override

	PFRecoTauDiscriminationByHPSSelection (const edm::ParameterSet &)

	~PFRecoTauDiscriminationByHPSSelection ()

Public Member Functions inherited from TauDiscriminationProducerBase< TauType, TauDiscriminator >
virtual void	beginEvent (const edm::Event &evt, const edm::EventSetup &evtSetup)

virtual double	discriminate (const TauRef &tau)=0

virtual void	endEvent (edm::Event &evt)

void	produce (edm::Event &, const edm::EventSetup &)

	TauDiscriminationProducerBase (const edm::ParameterSet &iConfig)

	TauDiscriminationProducerBase ()

virtual	~TauDiscriminationProducerBase ()

Public Member Functions inherited from edm::EDProducer
	EDProducer ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducer ()

Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

virtual	~ProducerBase ()

Public Member Functions inherited from edm::EDConsumerBase
	EDConsumerBase ()

ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Types
typedef std::map< IntPair, DecayModeCuts >	DecayModeCutMap

typedef std::pair< double, double >	DoublePair

typedef std::pair< unsigned int, unsigned int >	IntPair

typedef StringObjectFunction < reco::PFTau >	TauFunc

Private Attributes
DecayModeCutMap	decayModeCuts_

double	matchingCone_

double	minPt_

bool	requireTauChargedHadronsToBeChargedPFCands_

TauFunc	signalConeFun_

int	verbosity_

Additional Inherited Members
Public Types inherited from TauDiscriminationProducerBase< TauType, TauDiscriminator >
typedef std::vector< TauType >	TauCollection

typedef edm::Ref< TauCollection >	TauRef

typedef edm::RefProd < TauCollection >	TauRefProd

Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

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

edm::InputTag	TauProducer_

Detailed Description

Definition at line 21 of file PFRecoTauDiscriminationByHPSSelection.cc.

Member Typedef Documentation

typedef std::map<IntPair, DecayModeCuts> PFRecoTauDiscriminationByHPSSelection::DecayModeCutMap

private

Definition at line 48 of file PFRecoTauDiscriminationByHPSSelection.cc.

typedef std::pair<double, double> PFRecoTauDiscriminationByHPSSelection::DoublePair

private

Definition at line 47 of file PFRecoTauDiscriminationByHPSSelection.cc.

typedef std::pair<unsigned int, unsigned int> PFRecoTauDiscriminationByHPSSelection::IntPair

private

Definition at line 46 of file PFRecoTauDiscriminationByHPSSelection.cc.

typedef StringObjectFunction<reco::PFTau> PFRecoTauDiscriminationByHPSSelection::TauFunc

private

Definition at line 29 of file PFRecoTauDiscriminationByHPSSelection.cc.

Constructor & Destructor Documentation

PFRecoTauDiscriminationByHPSSelection::PFRecoTauDiscriminationByHPSSelection ( const edm::ParameterSet & pset )

explicit

Definition at line 60 of file PFRecoTauDiscriminationByHPSSelection.cc.

References PFRecoTauDiscriminationByHPSSelection::DecayModeCuts::assumeStripMass_, hpstanc_transforms::cuts, decayModeCuts_, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), matchingCone_, PFRecoTauDiscriminationByHPSSelection::DecayModeCuts::maxMass_, PFRecoTauDiscriminationByHPSSelection::DecayModeCuts::maxPi0Mass_, PFRecoTauDiscriminationByHPSSelection::DecayModeCuts::minMass_, PFRecoTauDiscriminationByHPSSelection::DecayModeCuts::minPi0Mass_, minPt_, PFRecoTauDiscriminationByHPSSelection::DecayModeCuts::nChargedPFCandsMin_, PFRecoTauDiscriminationByHPSSelection::DecayModeCuts::nTracksMin_, requireTauChargedHadronsToBeChargedPFCands_, AlCaHLTBitMon_QueryRunRegistry::string, and verbosity_.

   : PFTauDiscriminationProducerBase(pset),
     signalConeFun_(pset.getParameter<std::string>("coneSizeFormula")) 
 {
   // Get the matchign cut
   matchingCone_ = pset.getParameter<double>("matchingCone");
   minPt_ = pset.getParameter<double>("minTauPt");
   // Get the mass cuts for each decay mode
   typedef std::vector<edm::ParameterSet> VPSet;
   const VPSet& decayModes = pset.getParameter<VPSet>("decayModes");
   BOOST_FOREACH( const edm::ParameterSet &decayMode, decayModes ) {
     // The mass window(s)
     DecayModeCuts cuts;
     if ( decayMode.exists("nTracksMin") ) {
       cuts.nTracksMin_ = decayMode.getParameter<unsigned>("nTracksMin");
     } else {
       cuts.nTracksMin_ = 0;
     }
     if ( decayMode.exists("nChargedPFCandsMin") ) {
       cuts.nChargedPFCandsMin_ = decayMode.getParameter<unsigned>("nChargedPFCandsMin");
     } else {
       cuts.nChargedPFCandsMin_ = 0;
     }
     cuts.minMass_ = decayMode.getParameter<double>("minMass");
     cuts.maxMass_ = new TauFunc(decayMode.getParameter<std::string>("maxMass"));
     if ( decayMode.exists("minPi0Mass") ) {
       cuts.minPi0Mass_ = decayMode.getParameter<double>("minPi0Mass");
       cuts.maxPi0Mass_ = decayMode.getParameter<double>("maxPi0Mass");
     } else {
       cuts.minPi0Mass_ = -1.e3;
       cuts.maxPi0Mass_ = 1.e9;
     }
     if ( decayMode.exists("assumeStripMass") ) {
       cuts.assumeStripMass_ = decayMode.getParameter<double>("assumeStripMass");
     } else {
       cuts.assumeStripMass_ = -1.0;
     }
     decayModeCuts_.insert(std::make_pair(
             // The decay mode as a key
             std::make_pair(
                 decayMode.getParameter<uint32_t>("nCharged"),
                 decayMode.getParameter<uint32_t>("nPiZeros")),
             cuts
           ));
   }
   requireTauChargedHadronsToBeChargedPFCands_ = pset.getParameter<bool>("requireTauChargedHadronsToBeChargedPFCands");
   verbosity_ = pset.exists("verbosity") ?
     pset.getParameter<int>("verbosity") : 0;
 }

PFRecoTauDiscriminationByHPSSelection::~PFRecoTauDiscriminationByHPSSelection ( )

Definition at line 110 of file PFRecoTauDiscriminationByHPSSelection.cc.

References decayModeCuts_.

 {
   for ( DecayModeCutMap::iterator it = decayModeCuts_.begin();
     it != decayModeCuts_.end(); ++it ) {
     delete it->second.maxMass_;
   }
 }

Member Function Documentation

double PFRecoTauDiscriminationByHPSSelection::discriminate ( const reco::PFTauRef & tau )

override

Definition at line 119 of file PFRecoTauDiscriminationByHPSSelection.cc.

References reco::PFRecoTauChargedHadron::algo(), PFRecoTauDiscriminationByHPSSelection::DecayModeCuts::assumeStripMass_, gather_cfg::cout, decayModeCuts_, deltaR(), reco::PFRecoTauChargedHadron::kChargedPFCandidate, reco::PFRecoTauChargedHadron::kPFNeutralHadron, reco::PFRecoTauChargedHadron::kTrack, matchingCone_, PFRecoTauDiscriminationByHPSSelection::DecayModeCuts::maxMass_, PFRecoTauDiscriminationByHPSSelection::DecayModeCuts::maxPi0Mass_, PFRecoTauDiscriminationByHPSSelection::DecayModeCuts::minMass_, PFRecoTauDiscriminationByHPSSelection::DecayModeCuts::minPi0Mass_, minPt_, PFRecoTauDiscriminationByHPSSelection::DecayModeCuts::nChargedPFCandsMin_, PFRecoTauDiscriminationByHPSSelection::DecayModeCuts::nTracksMin_, reco::LeafCandidate::p4(), requireTauChargedHadronsToBeChargedPFCands_, signalConeFun_, AlCaHLTBitMon_QueryRunRegistry::string, and verbosity_.

 {
   if ( verbosity_ ) {
     std::cout << "<PFRecoTauDiscriminationByHPSSelection::discriminate>:" << std::endl;
     std::cout << " nCharged = " << tau->signalTauChargedHadronCandidates().size() << std::endl;
     std::cout << " nPiZeros = " << tau->signalPiZeroCandidates().size() << std::endl;
   }
 
   // Check if we pass the min pt
   if ( tau->pt() < minPt_ ) {
     if ( verbosity_ ) {
       std::cout << " fails minPt cut." << std::endl;
     }
     return 0.0;
   }
 
   // See if we select this decay mode
   DecayModeCutMap::const_iterator massWindowIter =
       decayModeCuts_.find(std::make_pair(tau->signalTauChargedHadronCandidates().size(),
                                          tau->signalPiZeroCandidates().size()));
   // Check if decay mode is supported
   if ( massWindowIter == decayModeCuts_.end() ) {
     if ( verbosity_ ) {
       std::cout << " fails mass-window definition requirement." << std::endl;
     }
     return 0.0;
   }
   const DecayModeCuts& massWindow = massWindowIter->second;
 
   if ( massWindow.nTracksMin_ > 0 ) {
     unsigned int nTracks = 0;
     const std::vector<reco::PFRecoTauChargedHadron>& chargedHadrons = tau->signalTauChargedHadronCandidates();
     for ( std::vector<reco::PFRecoTauChargedHadron>::const_iterator chargedHadron = chargedHadrons.begin();
       chargedHadron != chargedHadrons.end(); ++chargedHadron ) {
       if ( chargedHadron->algoIs(reco::PFRecoTauChargedHadron::kChargedPFCandidate) || chargedHadron->algoIs(reco::PFRecoTauChargedHadron::kTrack) ) {
     ++nTracks;
       }
     }
     if ( verbosity_ ) {
       std::cout << "nTracks = " << nTracks << " (min = " << massWindow.nTracksMin_ << ")" << std::endl;
     }
     if ( nTracks < massWindow.nTracksMin_ ) {
       if ( verbosity_ ) {
     std::cout << " fails nTracks requirement for mass-window." << std::endl;
       }
       return 0.0;
     }
   }
   if ( massWindow.nChargedPFCandsMin_ > 0 ) {
     unsigned int nChargedPFCands = 0;
     const std::vector<reco::PFRecoTauChargedHadron>& chargedHadrons = tau->signalTauChargedHadronCandidates();
     for ( std::vector<reco::PFRecoTauChargedHadron>::const_iterator chargedHadron = chargedHadrons.begin();
       chargedHadron != chargedHadrons.end(); ++chargedHadron ) {
       if ( chargedHadron->algoIs(reco::PFRecoTauChargedHadron::kChargedPFCandidate) ) {
     ++nChargedPFCands;
       }
     }
     if ( verbosity_ ) {
       std::cout << "nChargedPFCands = " << nChargedPFCands << " (min = " << massWindow.nChargedPFCandsMin_ << ")" << std::endl;
     }
     if ( nChargedPFCands < massWindow.nChargedPFCandsMin_ ) {
       if ( verbosity_ ) {
     std::cout << " fails nChargedPFCands requirement for mass-window." << std::endl;
       }
       return 0.0;
     }
   }
 
   math::XYZTLorentzVector tauP4 = tau->p4();
   if ( verbosity_ ) {
     std::cout << "tau: Pt = " << tauP4.pt() << ", eta = " << tauP4.eta() << ", phi = " << tauP4.phi() << ", mass = " << tauP4.mass() << std::endl;
   }
   // Find the total pizero p4
   reco::Candidate::LorentzVector stripsP4;
   BOOST_FOREACH(const reco::RecoTauPiZero& cand, 
       tau->signalPiZeroCandidates()){
     math::XYZTLorentzVector candP4 = cand.p4();
     stripsP4 += candP4;
   }
 
   // Apply strip mass assumption corrections
   if (massWindow.assumeStripMass_ >= 0) {
     BOOST_FOREACH(const reco::RecoTauPiZero& cand, 
         tau->signalPiZeroCandidates()){
       math::XYZTLorentzVector uncorrected = cand.p4();
       math::XYZTLorentzVector corrected = 
         applyMassConstraint(uncorrected, massWindow.assumeStripMass_);
       math::XYZTLorentzVector correction = corrected - uncorrected;
       tauP4 += correction;
       stripsP4 += correction;
     }
   }
   if ( verbosity_ ) {
     std::cout << "strips: Pt = " << stripsP4.pt() << ", eta = " << stripsP4.eta() << ", phi = " << stripsP4.phi() << ", mass = " << stripsP4.mass() << std::endl;
   }
 
   // Check if tau fails mass cut
   double maxMass_value = (*massWindow.maxMass_)(*tau);
   if ( tauP4.M() > maxMass_value || tauP4.M() < massWindow.minMass_ ) {
     if ( verbosity_ ) {
       std::cout << " fails tau mass-window cut." << std::endl;
     }
     return 0.0;
   }
 
   // Check if it fails the pi0 IM cut
   if ( stripsP4.M() > massWindow.maxPi0Mass_ ||
        stripsP4.M() < massWindow.minPi0Mass_ ) {
     if ( verbosity_ ) {
       std::cout << " fails strip mass-window cut." << std::endl;
     }
     return 0.0;
   }
 
   // Check if tau passes matching cone cut
   //std::cout << "dR(tau, jet) = " << deltaR(tauP4, tau->jetRef()->p4()) << std::endl;
   if ( deltaR(tauP4, tau->jetRef()->p4()) > matchingCone_ ) {
     if ( verbosity_ ) {
       std::cout << " fails matching-cone cut." << std::endl;
     }
     return 0.0;
   }
   
   // Check if tau passes cone cut
   double cone_size = signalConeFun_(*tau);
   // Check if any charged objects fail the signal cone cut
   BOOST_FOREACH(const reco::PFRecoTauChargedHadron& cand, tau->signalTauChargedHadronCandidates()) {
     if ( verbosity_ ) {
       std::cout << "dR(tau, signalPFChargedHadr) = " << deltaR(cand.p4(), tauP4) << std::endl;
     }
     if ( deltaR(cand.p4(), tauP4) > cone_size ) {
       if ( verbosity_ ) {
     std::cout << " fails signal-cone cut for charged hadron(s)." << std::endl;
       }
       return 0.0;
     }
   }
   // Now check the pizeros
   BOOST_FOREACH(const reco::RecoTauPiZero& cand, tau->signalPiZeroCandidates()) {
     if ( verbosity_ ) {
       std::cout << "dR(tau, signalPiZero) = " << deltaR(cand.p4(), tauP4) << std::endl;
     }
     if ( deltaR(cand.p4(), tauP4) > cone_size ) {
       if ( verbosity_ ) {
     std::cout << " fails signal-cone cut for strip(s)." << std::endl;
       }
       return 0.0;
     }
   }
 
   if ( requireTauChargedHadronsToBeChargedPFCands_ ) {
     BOOST_FOREACH(const reco::PFRecoTauChargedHadron& cand, tau->signalTauChargedHadronCandidates()) {
       if ( verbosity_ ) {
     std::string algo_string;
     if      ( cand.algo() == reco::PFRecoTauChargedHadron::kChargedPFCandidate ) algo_string = "ChargedPFCandidate";
     else if ( cand.algo() == reco::PFRecoTauChargedHadron::kTrack              ) algo_string = "Track";
     else if ( cand.algo() == reco::PFRecoTauChargedHadron::kPFNeutralHadron    ) algo_string = "PFNeutralHadron";
     else                                                                         algo_string = "Undefined";
     std::cout << "algo(signalPFChargedHadr) = " << algo_string << std::endl;
       }
       if ( !(cand.algo() == reco::PFRecoTauChargedHadron::kChargedPFCandidate) ) {
     if ( verbosity_ ) {
       std::cout << " fails cut on PFRecoTauChargedHadron algo." << std::endl;
     }
     return 0.0;
       }
     }
   }
 
   // Otherwise, we pass!
   if ( verbosity_ ) {
     std::cout << " passes all cuts." << std::endl;
   }
   return 1.0;
 }