AntiElectronIDCut2.cc

Go to the documentation of this file.

#include "RecoTauTag/RecoTau/interface/AntiElectronIDCut2.h"

#include "FWCore/Utilities/interface/Exception.h"

#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrack.h"
#include <TMath.h>

AntiElectronIDCut2::AntiElectronIDCut2()
{
  //default, keep all taus in cracks for HLT 
  keepAllInEcalCrack_ = true;
  rejectAllInEcalCrack_ = false;

  //Default Cuts to be applied
  Tau_applyCut_hcal3x3OverPLead_ = true; 
  Tau_applyCut_leadPFChargedHadrEoP_ = true; 
  Tau_applyCut_GammaEtaMom_ = true; 
  Tau_applyCut_GammaPhiMom_ = false; 
  Tau_applyCut_GammaEnFrac_ = false; 
  Tau_applyCut_HLTSpecific_ = true; 

  verbosity_ = 0;

}

AntiElectronIDCut2::~AntiElectronIDCut2()
{}

double
AntiElectronIDCut2::Discriminator(float TauPt,
                                  float TauEta,
                                  float TauLeadChargedPFCandPt,
                                  float TauLeadChargedPFCandEtaAtEcalEntrance,
                                  float TauLeadPFChargedHadrEoP,
                                  float TauHcal3x3OverPLead,
                  float TauGammaEtaMom,
                  float TauGammaPhiMom,
                  float TauGammaEnFrac)
{

  if(verbosity_){  
    std::cout<<"Tau GammaEnFrac "<<TauGammaEnFrac<<std::endl;  
    std::cout<<"Tau GammaEtaMom "<<TauGammaEtaMom<<std::endl;  
    std::cout<<"Tau GammaPhiMom "<<TauGammaPhiMom<<std::endl;  
    std::cout<<"Tau Hcal3x3OverPLead "<<TauHcal3x3OverPLead<<std::endl;
    std::cout<<"Tau LeadPFChargedHadrEoP "<<TauLeadPFChargedHadrEoP<<std::endl;
    std::cout<<"Tau LeadChargedPFCandEtaAtEcalEntrance "<<TauLeadChargedPFCandEtaAtEcalEntrance<<std::endl;
    std::cout<<"Tau LeadChargedPFCandPt "<<TauLeadChargedPFCandPt<<std::endl;
    std::cout<<"Tau Eta "<<TauEta<<std::endl;
    std::cout<<"Tau Pt "<<TauPt<<std::endl;
  }
  
  // ensure tau has at least one charged object
  if( TauLeadChargedPFCandPt <= 0 )
    {
      return 0.;
    }
  else {
    // Check if track goes to Ecal Crack
    if(isInEcalCrack(TauLeadChargedPFCandEtaAtEcalEntrance))
      {
    if(keepAllInEcalCrack_)
      return 1.0;
    else if(rejectAllInEcalCrack_)
      return 0.;
      }
  }
  
  bool decision = false;
  //Apply separate cuts for barrel and endcap
  if(TMath::Abs(TauEta) < 1.479)
    {
      
      //Apply cut for barrel
      if(Tau_applyCut_GammaEnFrac_ && TauGammaEnFrac > TauGammaEnFrac_barrel_max_)  
        decision = true;

      if(Tau_applyCut_GammaPhiMom_ && TauGammaPhiMom > TauGammaPhiMom_barrel_max_) 
        decision = true;

      if(Tau_applyCut_GammaEtaMom_ && TauGammaEtaMom > TauGammaEtaMom_barrel_max_)
    decision = true;

      if(Tau_applyCut_hcal3x3OverPLead_ && TauHcal3x3OverPLead > TauHcal3x3OverPLead_barrel_max_)
    decision = true;

      if(Tau_applyCut_leadPFChargedHadrEoP_ &&
     (TauLeadPFChargedHadrEoP < TauLeadPFChargedHadrEoP_barrel_min_ ||
      TauLeadPFChargedHadrEoP > TauLeadPFChargedHadrEoP_barrel_max_)) 
    decision = true;
    }
  else {
    
    //Apply cut for endcap
    if(Tau_applyCut_GammaEnFrac_ && TauGammaEnFrac > TauGammaEnFrac_endcap_max_)   
      decision = true; 
 
    if(Tau_applyCut_GammaPhiMom_ && TauGammaPhiMom > TauGammaPhiMom_endcap_max_)  
      decision = true;
    
    if(Tau_applyCut_GammaEtaMom_ && TauGammaEtaMom > TauGammaEtaMom_endcap_max_)
      decision = true;

    //This cut is for both offline and HLT. For offline, use cut 0.99-1.01, 
    //For HLT use cut 0.7-1.3
    if(Tau_applyCut_leadPFChargedHadrEoP_ &&
       (TauLeadPFChargedHadrEoP < TauLeadPFChargedHadrEoP_endcap_min1_ ||
    TauLeadPFChargedHadrEoP > TauLeadPFChargedHadrEoP_endcap_max1_)) 
      decision = true;

    //This cut is only for HLT. For HLT, use cut like 0.99-1.01 & H3x3/P>0.1
    //For offline, keep the values same as above : 0.99-1.01 & H3x3/P>0, otherwise it may select events in a wrong way. 
    if(Tau_applyCut_HLTSpecific_ && 
       (TauLeadPFChargedHadrEoP < TauLeadPFChargedHadrEoP_endcap_min2_ || 
    TauLeadPFChargedHadrEoP > TauLeadPFChargedHadrEoP_endcap_max2_) &&
       TauHcal3x3OverPLead > TauHcal3x3OverPLead_endcap_max_)
      decision = true;
  }

  return (decision ? 1. : 0.);
}  

double
AntiElectronIDCut2::Discriminator(float TauPt,
                  float TauEta,
                  float TauLeadChargedPFCandPt,
                  float TauLeadChargedPFCandEtaAtEcalEntrance,
                  float TauLeadPFChargedHadrEoP,
                  float TauHcal3x3OverPLead,
                  const std::vector<float>& GammasdEta,
                  const std::vector<float>& GammasdPhi,
                  const std::vector<float>& GammasPt
                  )
{

  double sumPt  = 0.;
  double dEta2  = 0.;
  double dPhi2  = 0.;
  for ( unsigned int i = 0 ; i < GammasPt.size() ; ++i ) {
    double pt_i  = GammasPt[i];
    double phi_i = GammasdPhi[i];
    if ( GammasdPhi[i] > TMath::Pi() ) phi_i = GammasdPhi[i] - 2*TMath::Pi();
    else if ( GammasdPhi[i] < -TMath::Pi() ) phi_i = GammasdPhi[i] + 2*TMath::Pi();
    double eta_i = GammasdEta[i];
    sumPt  +=  pt_i;
    dEta2  += (pt_i*eta_i*eta_i);
    dPhi2  += (pt_i*phi_i*phi_i);
  }

  float TauGammaEnFrac = sumPt/TauPt;

  if ( sumPt > 0. ) {
    dEta2 /= sumPt;
    dPhi2 /= sumPt;
  }

  float TauGammaEtaMom = TMath::Sqrt(dEta2)*TMath::Sqrt(TauGammaEnFrac)*TauPt;
  float TauGammaPhiMom = TMath::Sqrt(dPhi2)*TMath::Sqrt(TauGammaEnFrac)*TauPt;

  return Discriminator(TauPt,
                       TauEta,
                       TauLeadChargedPFCandPt,
                       TauLeadChargedPFCandEtaAtEcalEntrance,
                       TauLeadPFChargedHadrEoP,
                       TauHcal3x3OverPLead,
                       TauGammaEtaMom,
                       TauGammaPhiMom,
               TauGammaEnFrac
               );
}

void 
AntiElectronIDCut2::SetBarrelCutValues(float TauLeadPFChargedHadrEoP_min,
                       float TauLeadPFChargedHadrEoP_max,
                       float TauHcal3x3OverPLead_max,
                       float TauGammaEtaMom_max,
                       float TauGammaPhiMom_max,
                       float TauGammaEnFrac_max
                       )
{
  TauLeadPFChargedHadrEoP_barrel_min_ = TauLeadPFChargedHadrEoP_min;
  TauLeadPFChargedHadrEoP_barrel_max_ = TauLeadPFChargedHadrEoP_max;
  TauHcal3x3OverPLead_barrel_max_ = TauHcal3x3OverPLead_max;
  TauGammaEtaMom_barrel_max_ = TauGammaEtaMom_max;
  TauGammaPhiMom_barrel_max_ = TauGammaPhiMom_max;
  TauGammaEnFrac_barrel_max_ = TauGammaEnFrac_max;
}

void 
AntiElectronIDCut2::SetEndcapCutValues(float TauLeadPFChargedHadrEoP_min_1,
                       float TauLeadPFChargedHadrEoP_max_1,
                       float TauLeadPFChargedHadrEoP_min_2,
                       float TauLeadPFChargedHadrEoP_max_2,
                       float TauHcal3x3OverPLead_max,
                       float TauGammaEtaMom_max,
                       float TauGammaPhiMom_max,
                       float TauGammaEnFrac_max
                       )
{
  TauLeadPFChargedHadrEoP_endcap_min1_ = TauLeadPFChargedHadrEoP_min_1;
  TauLeadPFChargedHadrEoP_endcap_max1_ = TauLeadPFChargedHadrEoP_max_1;
  TauLeadPFChargedHadrEoP_endcap_min2_ = TauLeadPFChargedHadrEoP_min_2;
  TauLeadPFChargedHadrEoP_endcap_max2_ = TauLeadPFChargedHadrEoP_max_2;
  TauHcal3x3OverPLead_endcap_max_ = TauHcal3x3OverPLead_max;
  TauGammaEtaMom_endcap_max_ = TauGammaEtaMom_max;
  TauGammaPhiMom_endcap_max_ = TauGammaPhiMom_max;
  TauGammaEnFrac_endcap_max_ = TauGammaEnFrac_max;
}

void
AntiElectronIDCut2::ApplyCuts(bool applyCut_hcal3x3OverPLead,  
                  bool applyCut_leadPFChargedHadrEoP,  
                  bool applyCut_GammaEtaMom,  
                  bool applyCut_GammaPhiMom,  
                  bool applyCut_GammaEnFrac,  
                  bool applyCut_HLTSpecific 
                  )
{
  Tau_applyCut_hcal3x3OverPLead_ = applyCut_hcal3x3OverPLead; 
  Tau_applyCut_leadPFChargedHadrEoP_ = applyCut_leadPFChargedHadrEoP; 
  Tau_applyCut_GammaEtaMom_ = applyCut_GammaEtaMom; 
  Tau_applyCut_GammaPhiMom_ = applyCut_GammaPhiMom; 
  Tau_applyCut_GammaEnFrac_ = applyCut_GammaEnFrac; 
  Tau_applyCut_HLTSpecific_ = applyCut_HLTSpecific; 
}

bool
AntiElectronIDCut2::isInEcalCrack(double eta) const
{
  bool in_ecal_crack = false;

  eta = fabs(eta);
  for(std::vector<pdouble>::const_iterator etaCrack = ecalCracks_.begin(); 
      etaCrack != ecalCracks_.end(); ++etaCrack)
    if(eta >= etaCrack->first && eta < etaCrack->second)
      in_ecal_crack = true;
  
  return in_ecal_crack;
}