InDetPerfPlot_FakeRate.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#include "InDetPerfPlot_FakeRate.h"
// #include "TrkValHistUtils/EfficiencyPurityCalculator.h"
#include "xAODTruth/TruthParticle.h"
#include "xAODTruth/TruthVertex.h"
#include "InDetPhysValMonitoringUtilities.h"
#include <cmath>
#include <regex>
using namespace IDPVM;
 
InDetPerfPlot_FakeRate::InDetPerfPlot_FakeRate(InDetPlotBase* pParent, const std::string& sDir, FakeRateCategory fakeRateCategory) :
  InDetPlotBase(pParent, sDir),
  m_fakerate_vs_eta{},
  m_fakerate_vs_pt{},
  m_fakerate_vs_lowpt{},
  m_fakerate_vs_phi{},
  m_fakerate_vs_d0{},
  m_fakerate_vs_z0{},
  m_fakerate_vs_mu{},
  m_fakeRateCategory{fakeRateCategory} {
  // nop
}
 
void unlinked_ylabel(TEfficiency *teff, const std::string& replacement)
{
    if (teff) {
    std::string title = std::regex_replace(teff->GetTitle(), std::regex("linked"), replacement);
    std::string xlabel = teff->GetTotalHistogram()->GetXaxis()->GetTitle();
    std::string ylabel = std::regex_replace(teff->GetTotalHistogram()->GetYaxis()->GetTitle(), std::regex("linked"), replacement);
    std::string s = title + ";" + xlabel + ";" + ylabel;
    teff->SetTitle(s.c_str());
    }
}
 
void
InDetPerfPlot_FakeRate::initializePlots() {
 
  if(m_fakeRateCategory == FakeRateCategory::Total){
    std::string nameOverride = "_total";
    book(m_fakerate_vs_eta, "fakerate_vs_eta", "fakerate_vs_eta"+nameOverride);
    book(m_fakerate_vs_pt, "fakerate_vs_pt", "fakerate_vs_pt"+nameOverride);
    book(m_fakerate_vs_lowpt, "fakerate_vs_lowpt", "fakerate_vs_lowpt"+nameOverride);
    book(m_fakerate_vs_phi, "fakerate_vs_phi", "fakerate_vs_phi"+nameOverride);
    book(m_fakerate_vs_d0, "fakerate_vs_d0", "fakerate_vs_d0"+nameOverride);
    book(m_fakerate_vs_z0, "fakerate_vs_z0", "fakerate_vs_z0"+nameOverride);
    book(m_fakerate_vs_mu, "fakerate_vs_mu", "fakerate_vs_mu"+nameOverride);
  }
  else{
    book(m_fakerate_vs_eta, "fakerate_vs_eta");
    book(m_fakerate_vs_pt, "fakerate_vs_pt");
    book(m_fakerate_vs_lowpt, "fakerate_vs_lowpt");
    book(m_fakerate_vs_phi, "fakerate_vs_phi");
    book(m_fakerate_vs_d0, "fakerate_vs_d0");
    book(m_fakerate_vs_z0, "fakerate_vs_z0");
    book(m_fakerate_vs_mu, "fakerate_vs_mu");
  }
 
  
  if (m_fakeRateCategory != FakeRateCategory::Linked) {
    std::string replacement = (m_fakeRateCategory==FakeRateCategory::Unlinked)? "unlinked" : "";
      unlinked_ylabel(m_fakerate_vs_eta, replacement);
      unlinked_ylabel(m_fakerate_vs_pt, replacement);
      unlinked_ylabel(m_fakerate_vs_lowpt, replacement);
      unlinked_ylabel(m_fakerate_vs_phi, replacement);
      unlinked_ylabel(m_fakerate_vs_d0, replacement);
      unlinked_ylabel(m_fakerate_vs_z0, replacement);
      unlinked_ylabel(m_fakerate_vs_mu, replacement);
  }
}
 
void
InDetPerfPlot_FakeRate::fill(const xAOD::TrackParticle& track, const bool isFake, float weight, float mu) {
 
  double eta = track.eta();
  double pt = track.pt() / Gaudi::Units::GeV; // convert MeV to GeV
  double phi = track.phi();
  double d0 = track.d0();
  double z0 = track.z0();
 
  fillHisto(m_fakerate_vs_eta, eta, isFake, weight);
  fillHisto(m_fakerate_vs_pt, pt, isFake, weight);
  fillHisto(m_fakerate_vs_lowpt, pt, isFake, weight);
  fillHisto(m_fakerate_vs_phi, phi, isFake, weight); 
  fillHisto(m_fakerate_vs_d0, d0, isFake, weight);
  fillHisto(m_fakerate_vs_z0, z0, isFake, weight);
  fillHisto(m_fakerate_vs_mu, mu, isFake, weight);
}
 
void
InDetPerfPlot_FakeRate::finalizePlots() {
 
}