#include <HPSPFRecoTauAlgorithm.h>

Inheritance diagram for HPSPFRecoTauAlgorithm:

Classes
class	HPSTauIsolationSorter

class	HPSTauPtSorter

Public Member Functions
reco::PFTau	buildPFTau (const reco::PFTauTagInfoRef &, const reco::Vertex &)

	HPSPFRecoTauAlgorithm ()

	HPSPFRecoTauAlgorithm (const edm::ParameterSet &)

	~HPSPFRecoTauAlgorithm ()

Public Member Functions inherited from PFRecoTauAlgorithmBase
	PFRecoTauAlgorithmBase ()

	PFRecoTauAlgorithmBase (const edm::ParameterSet &)

void	setTransientTrackBuilder (const TransientTrackBuilder *)

virtual	~PFRecoTauAlgorithmBase ()

Private Member Functions
void	applyElectronRejection (reco::PFTau &, double)

void	applyMassConstraint (math::XYZTLorentzVector &, double)

void	applyMuonRejection (reco::PFTau &)

void	associateIsolationCandidates (reco::PFTau &, double)

void	buildOneProng (const reco::PFTauTagInfoRef &, const std::vector< reco::PFCandidatePtr > &)

void	buildOneProngStrip (const reco::PFTauTagInfoRef &, const std::vector< std::vector< reco::PFCandidatePtr >> &, const std::vector< reco::PFCandidatePtr > &)

void	buildOneProngTwoStrips (const reco::PFTauTagInfoRef &, const std::vector< std::vector< reco::PFCandidatePtr >> &, const std::vector< reco::PFCandidatePtr > &)

void	buildThreeProngs (const reco::PFTauTagInfoRef &, const std::vector< reco::PFCandidatePtr > &)

void	configure (const edm::ParameterSet &)

math::XYZTLorentzVector	createMergedLorentzVector (const std::vector< reco::PFCandidatePtr > &)

reco::PFTau	getBestTauCandidate (reco::PFTauCollection &)

bool	isNarrowTau (const reco::PFTau &, double)

bool	refitThreeProng (reco::PFTau &)

void	removeCandidateFromRefVector (const reco::PFCandidatePtr &, std::vector< reco::PFCandidatePtr > &)

Private Attributes
PFCandidateMergerBase *	candidateMerger_

double	chargeIsolationCone_

std::string	coneMetric_

TFormula	coneSizeFormula

std::string	coneSizeFormula_

bool	doOneProngs_

bool	doOneProngStrips_

bool	doOneProngTwoStrips_

bool	doThreeProngs_

std::string	emMerger_

double	gammaIsolationCone_

double	leadPionThreshold_

double	matchingCone_

double	maxSignalCone_

double	minSignalCone_

double	neutrHadrIsolationCone_

std::vector< double >	oneProngStripMassWindow_

std::vector< double >	oneProngTwoStripsMassWindow_

std::vector< double >	oneProngTwoStripsPi0MassWindow_

std::string	overlapCriterion_

reco::PFTauCollection	pfTaus_

double	stripPtThreshold_

double	tauThreshold_

std::vector< double >	threeProngMassWindow_

bool	useIsolationAnnulus_

Additional Inherited Members
Protected Attributes inherited from PFRecoTauAlgorithmBase
const TransientTrackBuilder *	TransientTrackBuilder_

Detailed Description

Definition at line 25 of file HPSPFRecoTauAlgorithm.h.

Constructor & Destructor Documentation

HPSPFRecoTauAlgorithm::HPSPFRecoTauAlgorithm ( )

Definition at line 6 of file HPSPFRecoTauAlgorithm.cc.

                                             :
     PFRecoTauAlgorithmBase()
 {
 }

HPSPFRecoTauAlgorithm::HPSPFRecoTauAlgorithm ( const edm::ParameterSet & config )

Definition at line 11 of file HPSPFRecoTauAlgorithm.cc.

References configure().

                                                                          :
     PFRecoTauAlgorithmBase(config)
 {
   configure(config);
 }

HPSPFRecoTauAlgorithm::~HPSPFRecoTauAlgorithm ( )

Definition at line 17 of file HPSPFRecoTauAlgorithm.cc.

References candidateMerger_.

 {
   if(candidateMerger_ !=0 ) delete candidateMerger_;
 }

Member Function Documentation

void HPSPFRecoTauAlgorithm::applyElectronRejection	(	reco::PFTau &	tau,
		double	stripEnergy
	)

private

Definition at line 588 of file HPSPFRecoTauAlgorithm.cc.

References funct::abs(), edm::Ptr< T >::isNonnull(), reco::PFTau::leadPFChargedHadrCand(), reco::PFTau::setbremsRecoveryEOverPLead(), reco::PFTau::setecalStripSumEOverPLead(), reco::PFTau::setelectronPreIDDecision(), reco::PFTau::setelectronPreIDOutput(), reco::PFTau::setelectronPreIDTrack(), reco::PFTau::setemFraction(), reco::PFTau::sethcal3x3OverPLead(), reco::PFTau::sethcalMaxOverPLead(), reco::PFTau::sethcalTotOverPLead(), reco::PFTau::setmaximumHCALPFClusterEt(), and funct::sin().

Referenced by buildOneProng(), buildOneProngStrip(), buildOneProngTwoStrips(), and buildThreeProngs().

 {
   //Here we apply the common electron rejection variables.
   //The only not common is the E/P that is applied in the decay mode
   //construction
 
 
   if(tau.leadPFChargedHadrCand().isNonnull()) {
     PFCandidatePtr leadCharged = tau.leadPFChargedHadrCand();
     math::XYZVector caloDir(leadCharged->positionAtECALEntrance().x(),
                             leadCharged->positionAtECALEntrance().y(),
                             leadCharged->positionAtECALEntrance().z());
 
     tau.setmaximumHCALPFClusterEt(leadCharged->hcalEnergy()*sin(caloDir.theta()));
 
 
 
     if(leadCharged->trackRef().isNonnull()) {
       TrackRef track = leadCharged->trackRef();
       tau.setemFraction(leadCharged->ecalEnergy()/(leadCharged->ecalEnergy()+leadCharged->hcalEnergy()));
       //For H/P trust particle Flow ! :Just take HCAL energy of the candidate
       //end of story
       tau.sethcalTotOverPLead(leadCharged->hcalEnergy()/track->p());
       tau.sethcalMaxOverPLead(leadCharged->hcalEnergy()/track->p());
       tau.sethcal3x3OverPLead(leadCharged->hcalEnergy()/track->p());
       tau.setelectronPreIDTrack(track);
       tau.setelectronPreIDOutput(leadCharged->mva_e_pi());
       //Since PF uses brem recovery we will store the default ecal energy here
       tau.setbremsRecoveryEOverPLead(leadCharged->ecalEnergy()/track->p());
       tau.setecalStripSumEOverPLead((leadCharged->ecalEnergy()-stripEnergy)/track->p());
       bool electronDecision;
       if(std::abs(leadCharged->pdgId())==11)
         electronDecision=true;
       else
         electronDecision=false;
       tau.setelectronPreIDDecision(electronDecision);
     }
   }
 }

void HPSPFRecoTauAlgorithm::applyMassConstraint	(	math::XYZTLorentzVector &	vec,
		double	mass
	)

private

Definition at line 703 of file HPSPFRecoTauAlgorithm.cc.

References mathSSE::sqrt().

Referenced by buildOneProngStrip(), and buildOneProngTwoStrips().

 {
   double factor = sqrt(vec.energy()*vec.energy()-mass*mass)/vec.P();
   vec.SetCoordinates(vec.px()*factor,vec.py()*factor,vec.pz()*factor,vec.energy());
 }

void HPSPFRecoTauAlgorithm::applyMuonRejection ( reco::PFTau & tau )

private

Definition at line 556 of file HPSPFRecoTauAlgorithm.cc.

References edm::Ptr< T >::isNonnull(), edm::Ref< C, T, F >::isNonnull(), reco::PFTau::leadPFChargedHadrCand(), RPCpg::mu, reco::PFTau::setCaloComp(), reco::PFTau::setMuonDecision(), and reco::PFTau::setSegComp().

Referenced by buildOneProng(), buildOneProngStrip(), buildOneProngTwoStrips(), and buildThreeProngs().

 {
 
   // Require that no signal track has segment matches
 
   //Also:
   //The segment compatibility is the number of matched Muon Segments
   //the old available does not exist in the muons anymore so i will fill the data format with that
   bool decision=true;
   float caloComp=0.0;
   float segComp=0.0;
 
   if(tau.leadPFChargedHadrCand().isNonnull()) {
     MuonRef mu =tau.leadPFChargedHadrCand()->muonRef();
     if(mu.isNonnull()){
       segComp=(float)(mu->matches().size());
       if(mu->caloCompatibility()>caloComp)
         caloComp = mu->caloCompatibility();
 
       if(segComp<1.0)
         decision=false;
 
       tau.setCaloComp(caloComp);
       tau.setSegComp(segComp);
       tau.setMuonDecision(decision);
     }
 
   }
 }

void HPSPFRecoTauAlgorithm::associateIsolationCandidates	(	reco::PFTau &	tau,
		double	tauCone
	)

private

Definition at line 447 of file HPSPFRecoTauAlgorithm.cc.

References chargeIsolationCone_, reco::deltaR(), PFRecoTauDiscriminationAgainstElectronDeadECAL_cfi::dR, reco::LeafCandidate::eta(), gammaIsolationCone_, i, edm::Ref< C, T, F >::isNull(), neutrHadrIsolationCone_, reco::PFTau::pfTauTagInfoRef(), reco::LeafCandidate::phi(), dt_dqm_sourceclient_common_cff::reco, removeCandidateFromRefVector(), reco::PFTau::setisolationPFCands(), reco::PFTau::setisolationPFChargedHadrCands(), reco::PFTau::setisolationPFChargedHadrCandsPtSum(), reco::PFTau::setisolationPFGammaCands(), reco::PFTau::setisolationPFGammaCandsEtSum(), reco::PFTau::setisolationPFNeutrHadrCands(), reco::PFTau::signalPFChargedHadrCands(), reco::PFTau::signalPFGammaCands(), and useIsolationAnnulus_.

Referenced by buildOneProng(), buildOneProngStrip(), buildOneProngTwoStrips(), and buildThreeProngs().

 {
 
 
   using namespace reco;
 
   //Information to get filled
   double sumPT=0;
   double sumET=0;
 
   if(tau.pfTauTagInfoRef().isNull()) return;
 
   std::vector<PFCandidatePtr> hadrons;
   std::vector<PFCandidatePtr> gammas;
   std::vector<PFCandidatePtr> neutral;
 
   //Remove candidates outside the cones
   if(useIsolationAnnulus_)
   {
     const std::vector<reco::PFCandidatePtr>& pfChargedHadrCands = tau.pfTauTagInfoRef()->PFChargedHadrCands();
     double tauEta=tau.eta();
     double tauPhi=tau.phi();
     for ( std::vector<reco::PFCandidatePtr>::const_iterator pfChargedHadrCand = pfChargedHadrCands.begin();
       pfChargedHadrCand != pfChargedHadrCands.end(); ++pfChargedHadrCand ) {
       double dR = reco::deltaR(tauEta, tauPhi, (*pfChargedHadrCand)->eta(), (*pfChargedHadrCand)->phi());
       if ( dR > tauCone && dR < chargeIsolationCone_ ) hadrons.push_back(*pfChargedHadrCand);
     }
 
     const std::vector<reco::PFCandidatePtr>& pfGammaCands = tau.pfTauTagInfoRef()->PFGammaCands();
     for ( std::vector<reco::PFCandidatePtr>::const_iterator pfGammaCand = pfGammaCands.begin();
       pfGammaCand != pfGammaCands.end(); ++pfGammaCand ) {
       double dR = reco::deltaR(tauEta, tauPhi, (*pfGammaCand)->eta(), (*pfGammaCand)->phi());
       if ( dR > tauCone && dR < gammaIsolationCone_ ) gammas.push_back(*pfGammaCand);
     }
 
     const std::vector<reco::PFCandidatePtr>& pfNeutralHadrCands = tau.pfTauTagInfoRef()->PFNeutrHadrCands();
     for ( std::vector<reco::PFCandidatePtr>::const_iterator pfNeutralHadrCand = pfNeutralHadrCands.begin();
       pfNeutralHadrCand != pfNeutralHadrCands.end(); ++pfNeutralHadrCand ) {
       double dR = reco::deltaR(tauEta, tauPhi, (*pfNeutralHadrCand)->eta(), (*pfNeutralHadrCand)->phi());
       if ( dR > tauCone && dR < neutrHadrIsolationCone_ ) hadrons.push_back(*pfNeutralHadrCand);
     }
   } else {
     double tauEta=tau.eta();
     double tauPhi=tau.phi();
     const std::vector<reco::PFCandidatePtr>& pfChargedHadrCands = tau.pfTauTagInfoRef()->PFChargedHadrCands();
     for ( std::vector<reco::PFCandidatePtr>::const_iterator pfChargedHadrCand = pfChargedHadrCands.begin();
       pfChargedHadrCand != pfChargedHadrCands.end(); ++pfChargedHadrCand ) {
       double dR = reco::deltaR(tauEta, tauPhi, (*pfChargedHadrCand)->eta(), (*pfChargedHadrCand)->phi());
       if ( dR < chargeIsolationCone_ ) hadrons.push_back(*pfChargedHadrCand);
     }
 
     const std::vector<reco::PFCandidatePtr>& pfGammaCands = tau.pfTauTagInfoRef()->PFGammaCands();
     for ( std::vector<reco::PFCandidatePtr>::const_iterator pfGammaCand = pfGammaCands.begin();
       pfGammaCand != pfGammaCands.end(); ++pfGammaCand ) {
       double dR = reco::deltaR(tauEta, tauPhi, (*pfGammaCand)->eta(), (*pfGammaCand)->phi());
       if ( dR < gammaIsolationCone_ ) gammas.push_back(*pfGammaCand);
     }
 
     const std::vector<reco::PFCandidatePtr>& pfNeutralHadrCands = tau.pfTauTagInfoRef()->PFNeutrHadrCands();
     for ( std::vector<reco::PFCandidatePtr>::const_iterator pfNeutralHadrCand = pfNeutralHadrCands.begin();
       pfNeutralHadrCand != pfNeutralHadrCands.end(); ++pfNeutralHadrCand ) {
       double dR = reco::deltaR(tauEta, tauPhi, (*pfNeutralHadrCand)->eta(), (*pfNeutralHadrCand)->phi());
       if ( dR < neutrHadrIsolationCone_ ) hadrons.push_back(*pfNeutralHadrCand);
     }
   }
 
   //remove the signal Constituents from the collections
   for(std::vector<PFCandidatePtr>::const_iterator i=tau.signalPFChargedHadrCands().begin();i!=tau.signalPFChargedHadrCands().end();++i)
   {
     removeCandidateFromRefVector(*i,hadrons);
   }
 
   for(std::vector<PFCandidatePtr>::const_iterator i=tau.signalPFGammaCands().begin();i!=tau.signalPFGammaCands().end();++i)
   {
     removeCandidateFromRefVector(*i,gammas);
     removeCandidateFromRefVector(*i,hadrons);//special case where we included a hadron if the strip!
   }
 
 
   //calculate isolation deposits
   for(unsigned int i=0;i<hadrons.size();++i)
   {
     sumPT+=hadrons[i]->pt();
   }
 
   for(unsigned int i=0;i<gammas.size();++i)
   {
     sumET+=gammas[i]->pt();
   }
 
 
   tau.setisolationPFChargedHadrCandsPtSum(sumPT);
   tau.setisolationPFGammaCandsEtSum(sumET);
   tau.setisolationPFChargedHadrCands(hadrons);
   tau.setisolationPFNeutrHadrCands(neutral);
   tau.setisolationPFGammaCands(gammas);
 
   std::vector<PFCandidatePtr> isoAll = hadrons;
   for(unsigned int i=0;i<gammas.size();++i)
     isoAll.push_back(gammas[i]);
 
   for(unsigned int i=0;i<neutral.size();++i)
     isoAll.push_back(neutral[i]);
 
   tau.setisolationPFCands(isoAll);
 }

void HPSPFRecoTauAlgorithm::buildOneProng	(	const reco::PFTauTagInfoRef &	tagInfo,
		const std::vector< reco::PFCandidatePtr > &	hadrons
	)

private

Definition at line 88 of file HPSPFRecoTauAlgorithm.cc.

References applyElectronRejection(), applyMuonRejection(), associateIsolationCandidates(), getBestTauCandidate(), h, leadPionThreshold_, matchingCone_, RecoPFTauTag_cff::PFTau, pfTaus_, reco::PFTau::setleadPFCand(), reco::PFTau::setleadPFChargedHadrCand(), reco::PFTau::setpfTauTagInfoRef(), reco::PFTau::setsignalPFCands(), reco::PFTau::setsignalPFChargedHadrCands(), metsig::tau, and tauThreshold_.

Referenced by buildPFTau().

 {
   PFTauCollection  taus;
 
   if(hadrons.size()>0)
     for(unsigned int h=0;h<hadrons.size();++h) {
       PFCandidatePtr hadron = hadrons[h];
 
       //In the one prong case the lead Track pt should be above the tau Threshold!
       //since all the tau is just one track!
       if(hadron->pt()>tauThreshold_)
         if(hadron->pt()>leadPionThreshold_)
           //The track should be within the matching cone
           if(ROOT::Math::VectorUtil::DeltaR(hadron->p4(),tagInfo->pfjetRef()->p4())<matchingCone_) {
             //OK Lets create a Particle Flow Tau!
             PFTau tau = PFTau(hadron->charge(),hadron->p4(),hadron->vertex());
 
             //Associate the Tag Info to the tau
             tau.setpfTauTagInfoRef(tagInfo);
 
             //Put the Hadron in the signal Constituents
             std::vector<PFCandidatePtr> signal;
             signal.push_back(hadron);
 
             //Set The signal candidates of the PF Tau
             tau.setsignalPFChargedHadrCands(signal);
             tau.setsignalPFCands(signal);
             tau.setleadPFChargedHadrCand(hadron);
             tau.setleadPFCand(hadron);
 
             //Fill isolation variables
             associateIsolationCandidates(tau,0.0);
 
             //Apply Muon rejection algorithms
             applyMuonRejection(tau);
             applyElectronRejection(tau,0.0);
 
             //Save this candidate
             taus.push_back(tau);
           }
     }
   if(taus.size()>0) {
     pfTaus_.push_back(getBestTauCandidate(taus));
   }
 
 }

void HPSPFRecoTauAlgorithm::buildOneProngStrip	(	const reco::PFTauTagInfoRef &	,
		const std::vector< std::vector< reco::PFCandidatePtr >> &	,
		const std::vector< reco::PFCandidatePtr > &
	)

private

Definition at line 138 of file HPSPFRecoTauAlgorithm.cc.

References applyElectronRejection(), applyMassConstraint(), applyMuonRejection(), associateIsolationCandidates(), coneMetric_, createMergedLorentzVector(), getBestTauCandidate(), isNarrowTau(), j, matchingCone_, max(), oneProngStripMassWindow_, pfTaus_, removeCandidateFromRefVector(), stripPtThreshold_, RecoTauPiZeroBuilderPlugins_cfi::strips, metsig::tau, and tauThreshold_.

Referenced by buildPFTau().

 {
   //Create output Collection
   PFTauCollection taus;
 
 
 
 
   //make taus like this only if there is at least one hadron+ 1 strip
   if(hadrons.size()>0&&strips.size()>0){
     //Combinatorics between strips and clusters
     for(std::vector<std::vector<PFCandidatePtr>>::const_iterator candVector=strips.begin();candVector!=strips.end();++candVector)
       for(std::vector<PFCandidatePtr>::const_iterator hadron=hadrons.begin();hadron!=hadrons.end();++hadron) {
 
         //First Cross cleaning ! If you asked to clusterize the candidates
         //with tracks too then you should not double count the track
         std::vector<PFCandidatePtr> emConstituents = *candVector;
         removeCandidateFromRefVector(*hadron,emConstituents);
 
         //Create a LorentzVector for the strip
         math::XYZTLorentzVector strip = createMergedLorentzVector(emConstituents);
 
         //TEST: Apply Strip Constraint
         applyMassConstraint(strip,0.1349);
 
         //create the Particle Flow Tau: Hadron plus Strip
         PFTau tau((*hadron)->charge(),
                   (*hadron)->p4()+strip,
                   (*hadron)->vertex());
 
         //Check tau threshold,  mass, Matching Cone window
         if(tau.pt()>tauThreshold_&&strip.pt()>stripPtThreshold_)
           if(tau.mass()>oneProngStripMassWindow_[0]&&tau.mass()<oneProngStripMassWindow_[1])//Apply mass window
             if(ROOT::Math::VectorUtil::DeltaR(tau.p4(),tagInfo->pfjetRef()->p4())<matchingCone_) { //Apply matching cone
               //Set The Tag Infor ref
               tau.setpfTauTagInfoRef(tagInfo);
 
               //Create the signal vectors
               std::vector<PFCandidatePtr> signal;
               std::vector<PFCandidatePtr> signalH;
               std::vector<PFCandidatePtr> signalG;
 
               //Store the hadron in the PFTau
               signalH.push_back(*hadron);
               signal.push_back(*hadron);
 
               //calculate the cone size : For the strip use it as one candidate !
               double tauCone=0.0;
               if(coneMetric_ =="angle")
                 tauCone=std::max(fabs(ROOT::Math::VectorUtil::Angle(tau.p4(),(*hadron)->p4())),
                                  fabs(ROOT::Math::VectorUtil::Angle(tau.p4(),strip)));
               else if(coneMetric_ == "DR")
                 tauCone=std::max(ROOT::Math::VectorUtil::DeltaR(tau.p4(),(*hadron)->p4()),
                                  ROOT::Math::VectorUtil::DeltaR(tau.p4(),strip));
 
               if(emConstituents.size()>0)
                 for(std::vector<PFCandidatePtr>::const_iterator j=emConstituents.begin();j!=emConstituents.end();++j)  {
                   signal.push_back(*j);
                   signalG.push_back(*j);
                 }
 
               //Set the PFTau
               tau.setsignalPFChargedHadrCands(signalH);
               tau.setsignalPFGammaCands(signalG);
               tau.setsignalPFCands(signal);
               tau.setleadPFChargedHadrCand(*hadron);
               tau.setleadPFNeutralCand(emConstituents[0]);
 
               //Set the lead Candidate->Can be the hadron or the leading PFGamma(When we clear the Dataformat we will put the strip)
               if((*hadron)->pt()>emConstituents[0]->pt())
                 tau.setleadPFCand(*hadron);
               else
                 tau.setleadPFCand(emConstituents[0]);
 
               //Apply the signal cone size formula
               if(isNarrowTau(tau,tauCone)) {
                 //calculate the isolation Deposits
                 associateIsolationCandidates(tau,tauCone);
                 //Set Muon Rejection
                 applyMuonRejection(tau);
                 applyElectronRejection(tau,strip.energy());
 
                 taus.push_back(tau);
               }
             }
       }
   }
 
   if(taus.size()>0) {
     pfTaus_.push_back(getBestTauCandidate(taus));
   }
 
 }

void HPSPFRecoTauAlgorithm::buildOneProngTwoStrips	(	const reco::PFTauTagInfoRef &	,
		const std::vector< std::vector< reco::PFCandidatePtr >> &	,
		const std::vector< reco::PFCandidatePtr > &
	)

private

Definition at line 234 of file HPSPFRecoTauAlgorithm.cc.

References applyElectronRejection(), applyMassConstraint(), applyMuonRejection(), associateIsolationCandidates(), coneMetric_, createMergedLorentzVector(), getBestTauCandidate(), isNarrowTau(), j, matchingCone_, max(), oneProngTwoStripsMassWindow_, oneProngTwoStripsPi0MassWindow_, pfTaus_, removeCandidateFromRefVector(), stripPtThreshold_, RecoTauPiZeroBuilderPlugins_cfi::strips, metsig::tau, and tauThreshold_.

Referenced by buildPFTau().

 {
 
 
   PFTauCollection taus;
 
   //make taus like this only if there is at least one hadron+ 2 strips
   if(hadrons.size()>0&&strips.size()>1){
     //Combinatorics between strips and clusters
     for(unsigned int Nstrip1=0;Nstrip1<strips.size()-1;++Nstrip1)
       for(unsigned int Nstrip2=Nstrip1+1;Nstrip2<strips.size();++Nstrip2)
         for(std::vector<PFCandidatePtr>::const_iterator hadron=hadrons.begin();hadron!=hadrons.end();++hadron) {
 
 
 
           //Create the strips and the vectors .Again cross clean the track if associated
           std::vector<PFCandidatePtr> emConstituents1 = strips[Nstrip1];
           std::vector<PFCandidatePtr> emConstituents2 = strips[Nstrip2];
           removeCandidateFromRefVector(*hadron,emConstituents1);
           removeCandidateFromRefVector(*hadron,emConstituents2);
 
 
           //Create a LorentzVector for the strip
           math::XYZTLorentzVector strip1 = createMergedLorentzVector(emConstituents1);
           math::XYZTLorentzVector strip2 = createMergedLorentzVector(emConstituents2);
 
 
 
           //Apply Mass Constraints
           applyMassConstraint(strip1,0.0);
           applyMassConstraint(strip2,0.0);
 
 
           PFTau tau((*hadron)->charge(),
                     (*hadron)->p4()+strip1+strip2,
                     (*hadron)->vertex());
 
 
           if(tau.pt()>tauThreshold_&&strip1.pt()>stripPtThreshold_&&strip2.pt()>stripPtThreshold_)
             if((strip1+strip2).M() >oneProngTwoStripsPi0MassWindow_[0] &&(strip1+strip2).M() <oneProngTwoStripsPi0MassWindow_[1] )//pi0 conmstraint for two strips
               if(tau.mass()>oneProngTwoStripsMassWindow_[0]&&tau.mass()<oneProngTwoStripsMassWindow_[1] )//Apply mass window
                 if(ROOT::Math::VectorUtil::DeltaR(tau.p4(),tagInfo->pfjetRef()->p4())<matchingCone_) { //Apply matching cone
                   //create the PFTau
                   tau.setpfTauTagInfoRef(tagInfo);
 
 
                   //Create the signal vectors
                   std::vector<PFCandidatePtr> signal;
                   std::vector<PFCandidatePtr> signalH;
                   std::vector<PFCandidatePtr> signalG;
 
                   //Store the hadron in the PFTau
                   signalH.push_back(*hadron);
                   signal.push_back(*hadron);
 
                   //calculate the cone size from the reconstructed Objects
                   double tauCone=1000.0;
                   if(coneMetric_ =="angle") {
                     tauCone=std::max(std::max(fabs(ROOT::Math::VectorUtil::Angle(tau.p4(),(*hadron)->p4())),
                                               fabs(ROOT::Math::VectorUtil::Angle(tau.p4(),strip1))),
                                               fabs(ROOT::Math::VectorUtil::Angle(tau.p4(),strip2)));
                   }
                   else if(coneMetric_ =="DR") {
                     tauCone=std::max(std::max((ROOT::Math::VectorUtil::DeltaR(tau.p4(),(*hadron)->p4())),
                                               (ROOT::Math::VectorUtil::DeltaR(tau.p4(),strip1))),
                                               (ROOT::Math::VectorUtil::DeltaR(tau.p4(),strip2)));
 
                   }
 
                   for(std::vector<PFCandidatePtr>::const_iterator j=emConstituents1.begin();j!=emConstituents1.end();++j)  {
                     signal.push_back(*j);
                     signalG.push_back(*j);
                   }
 
                   for(std::vector<PFCandidatePtr>::const_iterator j=emConstituents2.begin();j!=emConstituents2.end();++j)  {
                     signal.push_back(*j);
                     signalG.push_back(*j);
                   }
 
                   //Set the PFTau
                   tau.setsignalPFChargedHadrCands(signalH);
                   tau.setsignalPFGammaCands(signalG);
                   tau.setsignalPFCands(signal);
                   tau.setleadPFChargedHadrCand(*hadron);
 
                   //Set the lead Candidate->Can be the hadron or the leading PFGamma(When we clear the Dataformat we will put the strip)
                   if((*hadron)->pt()>emConstituents1[0]->pt())
                     tau.setleadPFCand(*hadron);
                   else
                     tau.setleadPFCand(emConstituents1[0]);
 
                   //Apply the cone size formula
                   if(isNarrowTau(tau,tauCone)) {
 
                     //calculate the isolation Deposits
                     associateIsolationCandidates(tau,tauCone);
 
                     applyMuonRejection(tau);
 
                     //For two strips take the nearest strip to the track
                     if(ROOT::Math::VectorUtil::DeltaR(strip1,(*hadron)->p4())<
                        ROOT::Math::VectorUtil::DeltaR(strip2,(*hadron)->p4()))
                       applyElectronRejection(tau,strip1.energy());
                     else
                       applyElectronRejection(tau,strip2.energy());
 
                     taus.push_back(tau);
                   }
                 }
         }
   }
 
   if(taus.size()>0) {
     pfTaus_.push_back(getBestTauCandidate(taus));
   }
 }

PFTau HPSPFRecoTauAlgorithm::buildPFTau	(	const reco::PFTauTagInfoRef &	tagInfo,
		const reco::Vertex &	vertex
	)

virtual

Implements PFRecoTauAlgorithmBase.

Definition at line 23 of file HPSPFRecoTauAlgorithm.cc.

References TransientTrackBuilder::build(), buildOneProng(), buildOneProngStrip(), buildOneProngTwoStrips(), buildThreeProngs(), candidateMerger_, doOneProngs_, doOneProngStrips_, doOneProngTwoStrips_, doThreeProngs_, getBestTauCandidate(), edm::Ptr< T >::isNonnull(), edm::Ref< C, T, F >::isNonnull(), metsig::jet, reco::PFTau::leadPFChargedHadrCand(), singlePfTauSkim_cff::leadTrack, PFCandidateMergerBase::mergeCandidates(), pfTaus_, reco::PFTau::pfTauTagInfoRef(), reco::PFTau::setleadPFChargedHadrCandsignedSipt(), reco::LeafCandidate::setP4(), reco::PFTau::setpfTauTagInfoRef(), IPTools::signedTransverseImpactParameter(), TauTagTools::sortRefVectorByPt(), RecoTauPiZeroBuilderPlugins_cfi::strips, and PFRecoTauAlgorithmBase::TransientTrackBuilder_.

 {
   PFTau pfTau;
 
   pfTaus_.clear();
   //make the strips globally.
   std::vector<std::vector<PFCandidatePtr>> strips = candidateMerger_->mergeCandidates(tagInfo->PFCands());
 
 
   //Sort the hadrons globally and once!
   std::vector<PFCandidatePtr> hadrons = tagInfo->PFChargedHadrCands();
   if(hadrons.size()>1)
     TauTagTools::sortRefVectorByPt(hadrons);
 
 
   //OK For this Tau Tag Info we should create all the possible taus
 
   //One Prongs
   if(doOneProngs_)
     buildOneProng(tagInfo,hadrons);
 
   //One Prong Strips
   if(doOneProngStrips_)
     buildOneProngStrip(tagInfo,strips,hadrons);
 
   //One Prong TwoStrips
   if(doOneProngTwoStrips_)
     buildOneProngTwoStrips(tagInfo,strips,hadrons);
 
   //Three Prong
   if(doThreeProngs_)
     buildThreeProngs(tagInfo,hadrons);
 
 
   //Lets see if we created any taus
   if(pfTaus_.size()>0) {
 
     pfTau = getBestTauCandidate(pfTaus_);
 
     //Set the IP for the leading track
     if(TransientTrackBuilder_!=0 &&pfTau.leadPFChargedHadrCand()->trackRef().isNonnull()) {
       const TransientTrack leadTrack=TransientTrackBuilder_->build(pfTau.leadPFChargedHadrCand()->trackRef());
       if(pfTau.pfTauTagInfoRef().isNonnull())
         if(pfTau.pfTauTagInfoRef()->pfjetRef().isNonnull()) {
           PFJetRef jet = pfTau.pfTauTagInfoRef()->pfjetRef();
           GlobalVector jetDir(jet->px(),jet->py(),jet->pz());
           if(IPTools::signedTransverseImpactParameter(leadTrack,jetDir,vertex).first)
             pfTau.setleadPFChargedHadrCandsignedSipt(IPTools::signedTransverseImpactParameter(leadTrack,jetDir,vertex).second.significance());
         }
     }
   }
   else {      //null PFTau
     //Simone asked that in case there is no tau returned make a tau
     //without refs and the LV of the jet
     pfTau.setpfTauTagInfoRef(tagInfo);
     pfTau.setP4(tagInfo->pfjetRef()->p4());
   }
 
   return pfTau;
 }

void HPSPFRecoTauAlgorithm::buildThreeProngs	(	const reco::PFTauTagInfoRef &	tagInfo,
		const std::vector< reco::PFCandidatePtr > &	hadrons
	)

private

Definition at line 354 of file HPSPFRecoTauAlgorithm.cc.

References a, funct::abs(), applyElectronRejection(), applyMuonRejection(), associateIsolationCandidates(), b, trackerHits::c, DeDxDiscriminatorTools::charge(), coneMetric_, getBestTauCandidate(), isNarrowTau(), leadPionThreshold_, reco::LeafCandidate::mass(), matchingCone_, max(), reco::LeafCandidate::p4(), RecoPFTauTag_cff::PFTau, pfTaus_, reco::LeafCandidate::pt(), refitThreeProng(), reco::PFTau::setleadPFCand(), reco::PFTau::setleadPFChargedHadrCand(), reco::PFTau::setpfTauTagInfoRef(), reco::PFTau::setsignalPFCands(), reco::PFTau::setsignalPFChargedHadrCands(), metsig::tau, tauThreshold_, and threeProngMassWindow_.

Referenced by buildPFTau().

 {
   PFTauCollection taus;
   //get Hadrons
 
 
   //Require at least three hadrons
   if(hadrons.size()>2)
     for(unsigned int a=0;a<hadrons.size()-2;++a)
       for(unsigned int b=a+1;b<hadrons.size()-1;++b)
         for(unsigned int c=b+1;c<hadrons.size();++c) {
 
           PFCandidatePtr h1 = hadrons[a];
           PFCandidatePtr h2 = hadrons[b];
           PFCandidatePtr h3 = hadrons[c];
 
           //check charge Compatibility and lead track
           int charge=h1->charge()+h2->charge()+h3->charge();
           if(std::abs(charge)==1 && h1->pt()>leadPionThreshold_)
             //check the track refs
             if(h1->trackRef()!=h2->trackRef()&&h1->trackRef()!=h3->trackRef()&&h2->trackRef()!=h3->trackRef())
             {
 
               //create the tau
               PFTau tau = PFTau(charge,h1->p4()+h2->p4()+h3->p4(),h1->vertex());
               tau.setpfTauTagInfoRef(tagInfo);
 
               if(tau.pt()>tauThreshold_)//Threshold
                 if(ROOT::Math::VectorUtil::DeltaR(tau.p4(),tagInfo->pfjetRef()->p4())<matchingCone_) {//Matching Cone
 
                   std::vector<PFCandidatePtr> signal;
                   signal.push_back(h1);
                   signal.push_back(h2);
                   signal.push_back(h3);
                   //calculate the tau cone by getting the maximum distance
                   double tauCone = 10000.0;
                   if(coneMetric_=="DR")
                   {
                     tauCone = std::max(ROOT::Math::VectorUtil::DeltaR(tau.p4(),h1->p4()),
                               std::max(ROOT::Math::VectorUtil::DeltaR(tau.p4(),h2->p4()),
                                        ROOT::Math::VectorUtil::DeltaR(tau.p4(),h3->p4())));
                   }
                   else if(coneMetric_=="angle")
                   {
                     tauCone =std::max(fabs(ROOT::Math::VectorUtil::Angle(tau.p4(),h1->p4())),
                              std::max(fabs(ROOT::Math::VectorUtil::Angle(tau.p4(),h2->p4())),
                                       fabs(ROOT::Math::VectorUtil::Angle(tau.p4(),h3->p4()))));
                   }
 
                   //Set The PFTau
                   tau.setsignalPFChargedHadrCands(signal);
                   tau.setsignalPFCands(signal);
                   tau.setleadPFChargedHadrCand(h1);
                   tau.setleadPFCand(h1);
 
                   if(isNarrowTau(tau,tauCone)) {
                     //calculate the isolation Deposits
                     associateIsolationCandidates(tau,tauCone);
                     applyMuonRejection(tau);
                     applyElectronRejection(tau,0.0);
                     taus.push_back(tau);
 
                   }
                 }
             }
         }
 
   if(taus.size()>0) {
     PFTau bestTau  = getBestTauCandidate(taus);
     if(refitThreeProng(bestTau))
       //Apply mass constraint
       if ( bestTau.mass() > threeProngMassWindow_[0] && bestTau.mass() < threeProngMassWindow_[1] )//MassWindow
         pfTaus_.push_back(bestTau);
   }
 
 }

void HPSPFRecoTauAlgorithm::configure ( const edm::ParameterSet & p )

private

Definition at line 632 of file HPSPFRecoTauAlgorithm.cc.

References candidateMerger_, chargeIsolationCone_, TauTagTools::computeConeSizeTFormula(), coneMetric_, coneSizeFormula, coneSizeFormula_, doOneProngs_, doOneProngStrips_, doOneProngTwoStrips_, doThreeProngs_, emMerger_, edm::hlt::Exception, gammaIsolationCone_, edm::ParameterSet::getParameter(), leadPionThreshold_, matchingCone_, maxSignalCone_, minSignalCone_, neutrHadrIsolationCone_, oneProngStripMassWindow_, oneProngTwoStripsMassWindow_, oneProngTwoStripsPi0MassWindow_, overlapCriterion_, AlCaHLTBitMon_QueryRunRegistry::string, stripPtThreshold_, tauThreshold_, threeProngMassWindow_, and useIsolationAnnulus_.

Referenced by HPSPFRecoTauAlgorithm().

 {
   emMerger_                      = p.getParameter<std::string>("emMergingAlgorithm");
   overlapCriterion_              = p.getParameter<std::string>("candOverlapCriterion");
   doOneProngs_                   = p.getParameter<bool>("doOneProng");
   doOneProngStrips_              = p.getParameter<bool>("doOneProngStrip");
   doOneProngTwoStrips_           = p.getParameter<bool>("doOneProngTwoStrips");
   doThreeProngs_                 = p.getParameter<bool>("doThreeProng");
   tauThreshold_                  = p.getParameter<double>("tauPtThreshold");
   leadPionThreshold_             = p.getParameter<double>("leadPionThreshold");
   stripPtThreshold_               = p.getParameter<double>("stripPtThreshold");
   chargeIsolationCone_           = p.getParameter<double>("chargeHadrIsolationConeSize");
   gammaIsolationCone_            = p.getParameter<double>("gammaIsolationConeSize");
   neutrHadrIsolationCone_        = p.getParameter<double>("neutrHadrIsolationConeSize");
   useIsolationAnnulus_           = p.getParameter<bool>("useIsolationAnnulus");
   oneProngStripMassWindow_       = p.getParameter<std::vector<double> >("oneProngStripMassWindow");
   oneProngTwoStripsMassWindow_   = p.getParameter<std::vector<double> >("oneProngTwoStripsMassWindow");
   oneProngTwoStripsPi0MassWindow_= p.getParameter<std::vector<double> >("oneProngTwoStripsPi0MassWindow");
   threeProngMassWindow_          = p.getParameter<std::vector<double> >("threeProngMassWindow");
   matchingCone_                  = p.getParameter<double>("matchingCone");
   coneMetric_                    = p.getParameter<std::string>("coneMetric");
   coneSizeFormula_               = p.getParameter<std::string>("coneSizeFormula");
   minSignalCone_                 = p.getParameter<double>("minimumSignalCone");
   maxSignalCone_                 = p.getParameter<double>("maximumSignalCone");
 
 
 
   //Initialize The Merging Algorithm!
   if(emMerger_ =="StripBased")
     candidateMerger_ =  new PFCandidateStripMerger(p);
   //Add the Pi0 Merger from Evan here
 
 
   if(oneProngStripMassWindow_.size()!=2)
     throw cms::Exception("") << "OneProngStripMassWindow must be a vector of size 2 [min,max] " << std::endl;
   if(oneProngTwoStripsMassWindow_.size()!=2)
     throw cms::Exception("") << "OneProngTwoStripsMassWindow must be a vector of size 2 [min,max] " << std::endl;
   if(threeProngMassWindow_.size()!=2)
     throw cms::Exception("") << "ThreeProngMassWindow must be a vector of size 2 [min,max] " << std::endl;
   if(coneMetric_!= "angle" && coneMetric_ != "DR")
     throw cms::Exception("") << "Cone Metric should be angle or DR " << std::endl;
 
   coneSizeFormula = TauTagTools::computeConeSizeTFormula(coneSizeFormula_,"Signal cone size Formula");
 
 
 }

math::XYZTLorentzVector HPSPFRecoTauAlgorithm::createMergedLorentzVector ( const std::vector< reco::PFCandidatePtr > & cands )

private

Definition at line 682 of file HPSPFRecoTauAlgorithm.cc.

References i.

Referenced by buildOneProngStrip(), and buildOneProngTwoStrips().

 {
   math::XYZTLorentzVector sum;
   for(unsigned int i=0;i<cands.size();++i) {
     sum+=cands[i]->p4();
   }
   return sum;
 }

reco::PFTau HPSPFRecoTauAlgorithm::getBestTauCandidate ( reco::PFTauCollection & taus )

private

Definition at line 759 of file HPSPFRecoTauAlgorithm.cc.

References overlapCriterion_, and MCScenario_CRAFT1_22X::sorter().

Referenced by buildOneProng(), buildOneProngStrip(), buildOneProngTwoStrips(), buildPFTau(), and buildThreeProngs().

 {
   reco::PFTauCollection::iterator it;
   if(overlapCriterion_ =="Isolation"){
     HPSTauIsolationSorter sorter;
     it = std::min_element(taus.begin(),taus.end(),sorter);
 
   }
   else if(overlapCriterion_ =="Pt"){
     HPSTauPtSorter sorter;
     it = std::min_element(taus.begin(),taus.end(),sorter);
   }
 
   return *it;
 }

bool HPSPFRecoTauAlgorithm::isNarrowTau	(	const reco::PFTau &	tau,
		double	cone
	)

private

Definition at line 433 of file HPSPFRecoTauAlgorithm.cc.

References coneSizeFormula, reco::LeafCandidate::energy(), reco::LeafCandidate::et(), maxSignalCone_, and minSignalCone_.

Referenced by buildOneProngStrip(), buildOneProngTwoStrips(), and buildThreeProngs().

 {
   double allowedConeSize=coneSizeFormula.Eval(tau.energy(),tau.et());
   if (allowedConeSize<minSignalCone_) allowedConeSize=minSignalCone_;
   if (allowedConeSize>maxSignalCone_) allowedConeSize=maxSignalCone_;
 
   if(cone<allowedConeSize)
     return true;
   else
     return false;
 }

bool HPSPFRecoTauAlgorithm::refitThreeProng ( reco::PFTau & tau )

private

Definition at line 713 of file HPSPFRecoTauAlgorithm.cc.

References TransientTrackBuilder::build(), TransientVertex::hasRefittedTracks(), TransientVertex::isValid(), reco::TrackBase::momentum(), p1, p2, p3, TransientVertex::position(), reco::TrackBase::px(), reco::TrackBase::py(), reco::TrackBase::pz(), TransientVertex::refittedTracks(), reco::LeafCandidate::setP4(), reco::LeafCandidate::setVertex(), reco::PFTau::signalPFChargedHadrCands(), mathSSE::sqrt(), PFRecoTauAlgorithmBase::TransientTrackBuilder_, KalmanVertexFitter::vertex(), PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

Referenced by buildThreeProngs().

 {
   bool response=false;
   //Get Hadrons
   std::vector<reco::PFCandidatePtr> hadrons = tau.signalPFChargedHadrCands();
   PFCandidatePtr h1 = hadrons[0];
   PFCandidatePtr h2 = hadrons[1];
   PFCandidatePtr h3 = hadrons[2];
 
   //Make transient tracks
   std::vector<TransientTrack> transientTracks;
   transientTracks.push_back(TransientTrackBuilder_->build(h1->trackRef()));
   transientTracks.push_back(TransientTrackBuilder_->build(h2->trackRef()));
   transientTracks.push_back(TransientTrackBuilder_->build(h3->trackRef()));
 
   //Apply the Vertex Fit
   KalmanVertexFitter fitter(true);
   TransientVertex myVertex = fitter.vertex(transientTracks);
 
   //Just require a valid vertex+ 3 refitted tracks
   if(myVertex.isValid()&&
      myVertex.hasRefittedTracks()&&
      myVertex.refittedTracks().size()==3) {
 
     response=true;
     math::XYZPoint vtx(myVertex.position().x(),myVertex.position().y(),myVertex.position().z());
 
     //Create a LV for each refitted track
     const std::vector<reco::TransientTrack>& refittedTracks = myVertex.refittedTracks();
     const reco::Track& track1 = refittedTracks.at(0).track();
     math::XYZTLorentzVector p1(track1.px(), track1.py(), track1.pz(), sqrt(track1.momentum().mag2() + 0.139*0.139));
     const reco::Track& track2 = refittedTracks.at(1).track();
     math::XYZTLorentzVector p2(track2.px(), track2.py(), track2.pz(), sqrt(track2.momentum().mag2() + 0.139*0.139));
     const reco::Track& track3 = refittedTracks.at(2).track();
     math::XYZTLorentzVector p3(track3.px(), track3.py(), track3.pz(), sqrt(track3.momentum().mag2() + 0.139*0.139));
 
     //Update the tau p4
     tau.setP4(p1+p2+p3);
     //Update the vertex
     tau.setVertex(vtx);
   }
   return response;
 
 }

void HPSPFRecoTauAlgorithm::removeCandidateFromRefVector	(	const reco::PFCandidatePtr &	cand,
		std::vector< reco::PFCandidatePtr > &	vec
	)

private

Definition at line 692 of file HPSPFRecoTauAlgorithm.cc.

References spr::find().

Referenced by associateIsolationCandidates(), buildOneProngStrip(), and buildOneProngTwoStrips().

 {
   std::vector<PFCandidatePtr> newVec;
 
   std::vector<PFCandidatePtr>::iterator it;
   it = std::find(vec.begin(),vec.end(),cand);
   if(it!=vec.end())
     vec.erase(it);
 }