GeneralTauPlots.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
 
#include "GeneralTauPlots.h"
#include "AthenaKernel/Units.h"
 
namespace Tau{
 
GeneralTauPlots::GeneralTauPlots(PlotBase* pParent, const std::string& sDir, const std::string& sTauJetContainerName):
   PlotBase(pParent, sDir),
   m_oTauKinematicPlots(this, "", sTauJetContainerName),
   m_sTauJetContainerName(sTauJetContainerName)
{   
}
 
void GeneralTauPlots::initializePlots(){
 
   m_tauCharge      = Book1D("Charge",m_sTauJetContainerName + " Tau charge; charge; # Taus",7,-3.,4.);
   m_tauNChargedTracks = Book1D("NChargedTracks", m_sTauJetContainerName + " Tau n charged tracks; nCharged; # Taus",10,0.,10.);
   m_tauNIsolatedTracks = Book1D("NIsolatedTracks", m_sTauJetContainerName + " Tau n isolated tracks; nIsolated; # Taus",10,0.,10.);
   m_tauNCoreTracks = Book1D("NCoreTracks",m_sTauJetContainerName + " Tau n core tracks; nCore; # Taus",15,0.,15.);
   m_tauNWideTracks = Book1D("NWideTracks",m_sTauJetContainerName + " Tau n wide tracks; nWide; # Taus",20,0.,20.);
   m_ptHighPt = Book1D("ptHighPt", m_sTauJetContainerName+" HighPt"+"; pt; # Taus",20, 0.0, 1500.0);
   m_RNNEleScore = Book1D("RNNEleScore", m_sTauJetContainerName+" RNNEleScore;RNNEleScore;# Tau", 50,0.,1.);
   m_RNNEleScoreSigTrans = Book1D("RNNEleScoreSigTrans", m_sTauJetContainerName+" RNNEleScoreSigTrans;RNNEleScoreSigTrans;"+"# Tau", 50,0.,1.);
   m_GNTauScore = Book1D("GNTauScore", m_sTauJetContainerName+" GNTauScore;GNTauScore;# Tau", 50,0.,1.);
   m_GNTauScoreSigTrans = Book1D("GNTauScoreSigTrans", m_sTauJetContainerName+" GNTauScoreSigTrans;GNTauScoreSigTrans;"+"# Tau", 50,0.,1.);
   m_ptGNTauLoose = Book1D("ptGNTauSigLoose",m_sTauJetContainerName+" GNTauSigLoose; pt; # Taus", 20, 0.0, 150.0);
   m_ptGNTauLooseHighPt = Book1D("ptGNTauSigLooseHighPt", m_sTauJetContainerName+" GNTauSigLooseHighPt; pt"+"; # Taus",20, 0.0, 1500.0);
   m_ptGNTauMedium = Book1D("ptGNTauSigMedium",m_sTauJetContainerName+" GNTauSigMedium; pt; # Taus", 20, 0.0, 150.0);
   m_ptGNTauMediumHighPt = Book1D("ptGNTauSigMediumHighPt", m_sTauJetContainerName+" GNTauSigMediumHighPt; pt"+"; # Taus",20, 0.0, 1500.0);
   m_ptGNTauTight = Book1D("ptGNTauSigTight",m_sTauJetContainerName+" GNTauSigTight; pt; # Taus", 20, 0.0, 150.0);
   m_ptGNTauTightHighPt = Book1D("ptGNTauSigTightHighPt", m_sTauJetContainerName+" GNTauSigTightHighPt; pt"+"; # Taus",20, 0.0, 1500.0);
 
   m_tauTrackSummary = Book1D("tauTrackSummary", m_sTauJetContainerName+" Track Type Summary", 7,0,7);
   m_tauTrackSummary->GetXaxis()->SetLabelSize(0.05);
   for(int i=1; i<= 7;i++){
      m_tauTrackSummary->GetXaxis()->SetBinLabel(i,m_lable[i-1]);
   }
 
}
  
void GeneralTauPlots::fill(const xAOD::TauJet& tau, float weight) {
  m_oTauKinematicPlots.fill(tau, weight);
  m_tauCharge->Fill(tau.charge(), weight); 
  m_tauNChargedTracks->Fill(tau.nTracks(), weight);
  m_tauNIsolatedTracks->Fill(tau.nTracks(xAOD::TauJetParameters::classifiedIsolation), weight);
  m_tauNCoreTracks->Fill(tau.nTracks(xAOD::TauJetParameters::coreTrack), weight);
  m_tauNWideTracks->Fill(tau.nTracks(xAOD::TauJetParameters::wideTrack), weight); 
  m_ptHighPt->Fill(tau.pt()/Athena::Units::GeV, weight);
 
 
  static const SG::ConstAccessor<float> acc_RNNEleScore("RNNEleScore");
  if ( acc_RNNEleScore.isAvailable(tau) ) {
     float rnnScore = tau.discriminant(xAOD::TauJetParameters::RNNEleScore);
     if ( rnnScore > -2.0 ) m_RNNEleScore->Fill(rnnScore, weight);
  }
  static const SG::ConstAccessor<float> acc_RNNEleScoreSigTrans("RNNEleScoreSigTrans_v1");
  if ( acc_RNNEleScoreSigTrans.isAvailable(tau) ) {
     float rnnScore = acc_RNNEleScoreSigTrans(tau);
     m_RNNEleScoreSigTrans->Fill(rnnScore, weight);
  }
  static const SG::ConstAccessor<float> acc_GNTauScore("GNTauScore_v0prune");
  if ( acc_GNTauScore.isAvailable(tau) ) {
     float gntauScore = acc_GNTauScore(tau);
     if ( gntauScore > -2.0 ) m_GNTauScore->Fill(gntauScore, weight);
  }
  static const SG::ConstAccessor<float> acc_GNTauScoreSigTrans("GNTauScoreSigTrans_v0prune");
  if ( acc_GNTauScoreSigTrans.isAvailable(tau) ) {
     float gntauScoreSigTrans = acc_GNTauScoreSigTrans(tau);
     if ( gntauScoreSigTrans > -2.0 ) m_GNTauScoreSigTrans->Fill(gntauScoreSigTrans, weight);
  }
 
  static const SG::ConstAccessor<char> acc_GNTauL("GNTauL_v0prune");
  if( acc_GNTauL.isAvailable(tau) && acc_GNTauL(tau)) {
     m_ptGNTauLoose      ->Fill(tau.pt()/Athena::Units::GeV, weight);
     m_ptGNTauLooseHighPt->Fill(tau.pt()/Athena::Units::GeV, weight);
  }
  static const SG::ConstAccessor<char> acc_GNTauM("GNTauM_v0prune");
  if( acc_GNTauM.isAvailable(tau) && acc_GNTauM(tau)) {
     m_ptGNTauMedium      ->Fill(tau.pt()/Athena::Units::GeV, weight);
     m_ptGNTauMediumHighPt->Fill(tau.pt()/Athena::Units::GeV, weight);
  }
  static const SG::ConstAccessor<char> acc_GNTauT("GNTauT_v0prune");
  if( acc_GNTauT.isAvailable(tau) && acc_GNTauT(tau)) {
     m_ptGNTauTight      ->Fill(tau.pt()/Athena::Units::GeV, weight);
     m_ptGNTauTightHighPt->Fill(tau.pt()/Athena::Units::GeV, weight);
  }
 
  // fill histogram with track summary type
  for(int j=0; j < 7; j++){
     int tracktype; 
     if( j==0) { tracktype = tau.nTracks(xAOD::TauJetParameters::coreTrack);}
     else if( j==1) { tracktype = tau.nTracks(xAOD::TauJetParameters::wideTrack);}
     else if( j==2) { continue;}
     else if( j==3) { tracktype = tau.nTracks();}
     else if( j==4) { tracktype = tau.nTracks(xAOD::TauJetParameters::classifiedIsolation);}  
     else if( j==5) { tracktype = tau.nTracks(xAOD::TauJetParameters::classifiedConversion);}
     else { tracktype = tau.nTracks(xAOD::TauJetParameters::classifiedFake);} 
 
     for(int i=0; i < tracktype; i++){
       m_tauTrackSummary->Fill(j+0.5, weight);
     }
  }
}
 
 
}