PFTauQualityCutWrapper.cc

Go to the documentation of this file.

#include "RecoTauTag/TauTagTools/interface/PFTauQualityCutWrapper.h"

// ****************************************************
// ******   Isolation filters  ************************
// ****************************************************

using namespace reco;

void
PFTauQualityCutWrapper::isolationChargedObjects(const PFTau& pfTau, const Vertex& pv, std::vector<reco::LeafCandidate>& output)
{
   if( isoQCuts.useTracksInsteadOfPF )
   {
      TrackRefVector result = TauTagTools::filteredTracks(pfTau.isolationTracks(), 
                                             isoQCuts.minTrackPt,
                                             isoQCuts.minTrackPixelHits,
                                             isoQCuts.minTrackHits,
                                             isoQCuts.maxTransverseImpactParameter,
                                             isoQCuts.maxTrackChi2,
                                             isoQCuts.maxDeltaZ,
                                             pv,
                                             pv.position().z() ); //????

      size_t nTracks = result.size();
      for(size_t iTrack = 0; iTrack < nTracks; ++iTrack)
      {
         // this sucks
         int charge         = result[iTrack]->charge();
         math::XYZVector p3 = result[iTrack]->momentum();
         reco::Particle::LorentzVector p4(p3.R(), p3.x(), p3.y(), p3.z());
         output.push_back(reco::LeafCandidate(charge, p4));
      }
   } else
   {
      PFCandidateRefVector preresult = TauTagTools::filteredPFChargedHadrCands(pfTau.isolationPFChargedHadrCands(), 
                                             isoQCuts.minTrackPt,
                                             isoQCuts.minTrackPixelHits,
                                             isoQCuts.minTrackHits,
                                             isoQCuts.maxTransverseImpactParameter,
                                             isoQCuts.maxTrackChi2,
                         10000.,                    //                                             isoQCuts.maxDeltaZ,
                                             pv,
                                             pv.position().z() ); //????


      for(unsigned int i=0;i<preresult.size();++i)
    if(preresult.at(i)->trackRef().isNonnull()) {
      //get the vertex weight and require to be >50%
      float w = pv.trackWeight(preresult.at(i)->trackRef());
      if(w>0.0) 
        output.push_back(reco::LeafCandidate(preresult.at(i)->charge(), preresult[i]->p4()));
      
    }


   }
   
}


void
PFTauQualityCutWrapper::isolationPUObjects(const PFTau& pfTau, const Vertex& pv, std::vector<reco::LeafCandidate>& output)
{

      PFCandidateRefVector preresult = TauTagTools::filteredPFChargedHadrCands(pfTau.isolationPFChargedHadrCands(), 
                             isoQCuts.minGammaEt,0,0,1000.,100000,
                         10000.,                    //                                             isoQCuts.maxDeltaZ,
                                             pv,
                                             pv.position().z() ); //????


      for(unsigned int i=0;i<preresult.size();++i)
    if(preresult.at(i)->trackRef().isNonnull()) {
      //get the vertex weight and require to be >50%
      float w = pv.trackWeight(preresult.at(i)->trackRef());
      if(w==0.0) 
        output.push_back(reco::LeafCandidate(preresult.at(i)->charge(), preresult[i]->p4()));
      
    }
   
}



void
PFTauQualityCutWrapper::isolationGammaObjects(const PFTau& pfTau, std::vector<reco::LeafCandidate>& output)
{
   PFCandidateRefVector result = TauTagTools::filteredPFGammaCands(pfTau.isolationPFGammaCands(), isoQCuts.minGammaEt);

   size_t nGammas = result.size();
   for(size_t iGamma = 0; iGamma < nGammas; ++iGamma)
   {
      output.push_back(reco::LeafCandidate(result[iGamma]->charge(), result[iGamma]->p4()));
   }
   
}

// ****************************************************
// ******   Signal region filters *********************
// ****************************************************

void
PFTauQualityCutWrapper::signalChargedObjects(const PFTau& pfTau, const Vertex& pv, std::vector<reco::LeafCandidate>& output)
{
   if( signalQCuts.useTracksInsteadOfPF )
   {
      TrackRefVector result = TauTagTools::filteredTracks(pfTau.signalTracks(), 
                                             signalQCuts.minTrackPt,
                                             signalQCuts.minTrackPixelHits,
                                             signalQCuts.minTrackHits,
                                             signalQCuts.maxTransverseImpactParameter,
                                             signalQCuts.maxTrackChi2,
                                             signalQCuts.maxDeltaZ,
                                             pv,
                                             pv.position().z() ); //????

      size_t nTracks = result.size();
      for(size_t iTrack = 0; iTrack < nTracks; ++iTrack)
      {
         // this sucks
         int charge         = result[iTrack]->charge();
         math::XYZVector p3 = result[iTrack]->momentum();
         reco::Particle::LorentzVector p4(p3.R(), p3.x(), p3.y(), p3.z());
         output.push_back(reco::LeafCandidate(charge, p4));
      }
   } else
   {

     //First perform basic filtering without vertex dz
      PFCandidateRefVector preresult = TauTagTools::filteredPFChargedHadrCands(pfTau.signalPFChargedHadrCands(), 

                                             signalQCuts.minTrackPt,
                                             signalQCuts.minTrackPixelHits,
                                             signalQCuts.minTrackHits,
                                             signalQCuts.maxTransverseImpactParameter,
                                             signalQCuts.maxTrackChi2,
                             10000.,   //                              signalQCuts.maxDeltaZ,
                                             pv,
                                             pv.position().z() ); //????
      
      //Now check the vertex association
      for(unsigned int i=0;i<preresult.size();++i)
    if(preresult.at(i)->trackRef().isNonnull()) {
      //get the vertex weight and require to be >50%
      float w = pv.trackWeight(preresult.at(i)->trackRef());
      if(w>0.0) 
        output.push_back(reco::LeafCandidate(preresult.at(i)->charge(), preresult[i]->p4()));
                       
    }
      

   }
   
}

void
PFTauQualityCutWrapper::signalGammaObjects(const PFTau& pfTau, std::vector<reco::LeafCandidate>& output)
{
   PFCandidateRefVector result = TauTagTools::filteredPFGammaCands(pfTau.signalPFGammaCands(), signalQCuts.minGammaEt);

   size_t nGammas = result.size();
   for(size_t iGamma = 0; iGamma < nGammas; ++iGamma)
   {
      output.push_back(reco::LeafCandidate(result[iGamma]->charge(), result[iGamma]->p4()));
   }
   
}