InDetRttPlots.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "InDetRttPlots.h"
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODTruth/TruthParticleContainer.h"
#include "xAODTruth/TruthParticle.h"
#include "xAODTruth/TruthVertex.h"
#include "TruthUtils/MagicNumbers.h"
#include "AthContainers/ConstAccessor.h"
#include <cmath> // std::isnan()
#include <limits>
 
InDetRttPlots::InDetRttPlots(InDetPlotBase* pParent, const std::string& sDir, const InDetRttPlotConfig & cfg) : InDetPlotBase(pParent, sDir),
  m_config(cfg){ 
  
  this->m_iDetailLevel = m_config.detailLevel;
  if (m_config.doTrackParameters)                     m_trackParameters = std::make_unique<InDetPerfPlot_TrackParameters>(this, "Tracks/Selected/Parameters"); 
  if (m_config.doNTracks)                             m_nTracks = std::make_unique<InDetPerfPlot_nTracks>(this, "Tracks/Tracks"); 
  if (m_config.doHitResidualPlot)                     m_hitResidualPlot= std::make_unique<InDetPerfPlot_HitResidual>(this, "Tracks/Hits/Residuals");
  if (m_config.doHitEffPlot)                          m_hitEffPlot= std::make_unique<InDetPerfPlot_HitEfficiency>(this, "Tracks/Hits/Efficiency");
  if (m_config.doFakePlots)                           m_fakePlots= std::make_unique<InDetPerfPlot_FakeRate>(this, "Tracks/FakeRate");
  if (m_config.doMissingTruthFakePlots)               m_missingTruthFakePlots= std::make_unique<InDetPerfPlot_FakeRate>(this, "Tracks/Unlinked/FakeRate");
  if (m_config.doResolutionPlotPrim)                  m_resolutionPlotPrim= std::make_unique<InDetPerfPlot_Resolution>(this, "Tracks/Matched/Resolutions/Primary");
  if (m_config.doResolutionPlotPrim_truthFromB)       m_resolutionPlotPrim_truthFromB= std::make_unique<InDetPerfPlot_Resolution>(this, "Tracks/Matched/Resolutions/TruthFromB");
  if (m_config.doHitsRecoTracksPlots)                 m_hitsRecoTracksPlots= std::make_unique<InDetPerfPlot_Hits>(this, "Tracks/Selected/HitsOnTracks");
  if (m_config.doEffPlots)                            m_effPlots= std::make_unique<InDetPerfPlot_Efficiency>(this, "Tracks/Efficiency");
  if (m_config.doDuplicatePlots)                      m_duplicatePlots = std::make_unique<InDetPerfPlot_Duplicate>(this, "Tracks/Duplicate");
  if (m_config.doVerticesVsMuPlots)                   m_verticesVsMuPlots= std::make_unique<InDetPerfPlot_VerticesVsMu>(this, "Vertices/AllPrimaryVertices");
  if (m_config.doVertexPlots)                         m_vertexPlots= std::make_unique<InDetPerfPlot_Vertex>(this, "Vertices/AllPrimaryVertices");
  if (m_config.doHardScatterVertexPlots)              m_hardScatterVertexPlots= std::make_unique<InDetPerfPlot_Vertex>(this, "Vertices/HardScatteringVertex");
  if (m_config.doHardScatterVertexTruthMatchingPlots) m_hardScatterVertexTruthMatchingPlots= std::make_unique<InDetPerfPlot_VertexTruthMatching>(this, "Vertices/HardScatteringVertex");
  if (m_config.doTrtExtensionPlots)                   m_trtExtensionPlots= std::make_unique<InDetPerfPlot_TRTExtension>(this, "Tracks/TRTExtension");
  if (m_config.doNtupleTruthToReco)                   m_ntupleTruthToReco= std::make_unique<InDetPerfNtuple_TruthToReco>(this, "Ntuples", "TruthToReco");
  if (m_config.doResolutionPlotSecd)                  m_resolutionPlotSecd = std::make_unique<InDetPerfPlot_Resolution>(this, "Tracks/Matched/Resolutions/Secondary");
  if (m_config.doHitsMatchedTracksPlots)              m_hitsMatchedTracksPlots = std::make_unique<InDetPerfPlot_Hits>(this, "Tracks/Matched/HitsOnTracks");
  if (m_config.doHitsFakeTracksPlots)                 m_hitsFakeTracksPlots = std::make_unique<InDetPerfPlot_Hits>(this, "Tracks/Fakes/HitsOnTracks");
  if (m_config.doHitsUnlinkedTracksPlots)             m_hitsUnlinkedTracksPlots = std::make_unique<InDetPerfPlot_Hits>(this, "Tracks/Unlinked/HitsOnTracks");
  if (m_config.doVertexTruthMatchingPlots)            m_vertexTruthMatchingPlots = std::make_unique<InDetPerfPlot_VertexTruthMatching>(this, "Vertices/AllPrimaryVertices", m_iDetailLevel);
 
  if (m_config.doEfficienciesPerAuthor){
      m_effSiSPSeededFinderPlots                = std::make_unique<InDetPerfPlot_Efficiency>(this, "TracksByAuthor/SiSPSeededFinder/Tracks/Efficiency");
      m_effInDetExtensionProcessorPlots         = std::make_unique<InDetPerfPlot_Efficiency>(this, "TracksByAuthor/InDetExtensionProcessor/Tracks/Efficiency");
      m_effTRTSeededTrackFinderPlots            = std::make_unique<InDetPerfPlot_Efficiency>(this, "TracksByAuthor/TRTSeededTrackFinder/Tracks/Efficiency");
      m_effTRTStandalonePlots                   = std::make_unique<InDetPerfPlot_Efficiency>(this, "TracksByAuthor/TRTStandalone/Tracks/Efficiency");
      m_effSiSpacePointsSeedMaker_LargeD0Plots  = std::make_unique<InDetPerfPlot_Efficiency>(this, "TracksByAuthor/SiSpacePointsSeedMaker_LargeD0/Tracks/Efficiency");
  }
 
  if (m_config.doFakesPerAuthor){
      m_fakeSiSPSeededFinderPlots               = std::make_unique<InDetPerfPlot_FakeRate>(this, "TracksByAuthor/SiSPSeededFinder/Tracks/FakeRate");
      m_fakeInDetExtensionProcessorPlots        = std::make_unique<InDetPerfPlot_FakeRate>(this, "TracksByAuthor/InDetExtensionProcessor/Tracks/FakeRate");
      m_fakeTRTSeededTrackFinderPlots           = std::make_unique<InDetPerfPlot_FakeRate>(this, "TracksByAuthor/TRTSeededTrackFinder/Tracks/FakeRate");
      m_fakeTRTStandalonePlots                  = std::make_unique<InDetPerfPlot_FakeRate>(this, "TracksByAuthor/TRTStandalone/Tracks/FakeRate");
      m_fakeSiSpacePointsSeedMaker_LargeD0Plots = std::make_unique<InDetPerfPlot_FakeRate>(this, "TracksByAuthor/SiSpacePointsSeedMaker_LargeD0/Tracks/FakeRate");
  }
 
  if (m_config.doTrackParametersPerAuthor){
    m_trkParaSiSPSeededFinderPlots              = std::make_unique<InDetPerfPlot_TrackParameters>(this, "TracksByAuthor/SiSPSeededFinder/Tracks/Parameters");
    m_trkParaInDetExtensionProcessorPlots       = std::make_unique<InDetPerfPlot_TrackParameters>(this, "TracksByAuthor/InDetExtensionProcessor/Tracks/Parameters");
    m_trkParaTRTSeededTrackFinderPlots          = std::make_unique<InDetPerfPlot_TrackParameters>(this, "TracksByAuthor/TRTSeededTrackFinder/Tracks/Parameters");
    m_trkParaTRTStandalonePlots                 = std::make_unique<InDetPerfPlot_TrackParameters>(this, "TracksByAuthor/TRTStandalone/Tracks/Parameters");
    m_trkParaSiSpacePointsSeedMaker_LargeD0Plots= std::make_unique<InDetPerfPlot_TrackParameters>(this, "TracksByAuthor/SiSpacePointsSeedMaker_LargeD0/Tracks/Parameters");
  }
 
  if (m_config.doResolutionsPerAuthor){
    m_resSiSPSeededFinderPlots                  = std::make_unique<InDetPerfPlot_Resolution>(this, "TracksByAuthor/SiSPSeededFinder/Tracks/Resolution");
    m_resInDetExtensionProcessorPlots           = std::make_unique<InDetPerfPlot_Resolution>(this, "TracksByAuthor/InDetExtensionProcessor/Tracks/Resolution");
    m_resTRTSeededTrackFinderPlots              = std::make_unique<InDetPerfPlot_Resolution>(this, "TracksByAuthor/TRTSeededTrackFinder/Tracks/Resolution");
    m_resTRTStandalonePlots                     = std::make_unique<InDetPerfPlot_Resolution>(this, "TracksByAuthor/TRTStandalone/Tracks/Resolution");
    m_resSiSpacePointsSeedMaker_LargeD0Plots    = std::make_unique<InDetPerfPlot_Resolution>(this, "TracksByAuthor/SiSpacePointsSeedMaker_LargeD0/Tracks/Resolution");
  }
    
  if (m_config.doTrkInJetPlots)                m_trkInJetPlots = std::make_unique<InDetPerfPlot_TrkInJet>(this, "TracksInJets/Tracks");
  if (m_config.doTrkInJetPlots_matched)        m_trkInJetPlots_matched = std::make_unique<InDetPerfPlot_TrkInJet>(this, "TracksInJets/Matched",false);
  if (m_config.doTrkInJetPlots_fake)           m_trkInJetPlots_fake = std::make_unique<InDetPerfPlot_TrkInJet>(this, "TracksInJets/Fakes",false);
  if (m_config.doTrkInJetPlots_unlinked)       m_trkInJetPlots_unlinked = std::make_unique<InDetPerfPlot_TrkInJet>(this, "TracksInJets/Unlinked",false);
  if (m_config.doTrkInJetPlots_bjets)          m_trkInJetPlots_bjets = std::make_unique<InDetPerfPlot_TrkInJet>(this, "TracksInBJets/Tracks");
  if (m_config.doTrkInJetPlots_matched_bjets)  m_trkInJetPlots_matched_bjets = std::make_unique<InDetPerfPlot_TrkInJet>(this, "TracksInBJets/Matched",false);
  if (m_config.doTrkInJetPlots_fake_bjets)     m_trkInJetPlots_fake_bjets = std::make_unique<InDetPerfPlot_TrkInJet>(this, "TracksInBJets/Fakes",false);
  if (m_config.doTrkInJetPlots_unlinked_bjets) m_trkInJetPlots_unlinked_bjets = std::make_unique<InDetPerfPlot_TrkInJet>(this, "TracksInBJets/Unlinked",false);
  if (m_config.doTrkInJetPlots_truthFromB)     m_trkInJetPlots_truthFromB = std::make_unique<InDetPerfPlot_TrkInJet>(this, "TracksInBJets/TruthFromB");
 
  setDetailLevel(m_iDetailLevel);
}
//
//Fill plots for matched particles
//
 
void
InDetRttPlots::fill(const xAOD::TrackParticle& particle, const xAOD::TruthParticle& truthParticle, bool isFromB, float mu, float weight) {
  // fill measurement bias, resolution, and pull plots
 
  // fill ITK resolutions (bias / resolutions)
  SG::ConstAccessor<float> tpTruthProbKeyAcc(m_trackParticleTruthProbKey);
  if (tpTruthProbKeyAcc.isAvailable(particle)) {
    const float prob = tpTruthProbKeyAcc(particle);
    int uniqueID = HepMC::uniqueID(truthParticle);
    if (!HepMC::is_simulation_particle(&truthParticle) && uniqueID != HepMC::UNDEFINED_ID && prob > 0.5) {
        if (m_resolutionPlotPrim) m_resolutionPlotPrim->fill(particle, truthParticle, weight);
    } else if (HepMC::is_simulation_particle(&truthParticle) && prob > 0.7 && m_iDetailLevel >= 200) {
        if (m_resolutionPlotSecd) m_resolutionPlotSecd->fill(particle, truthParticle, weight);
    }
    if ( isFromB ) {
      if (m_resolutionPlotPrim_truthFromB) m_resolutionPlotPrim_truthFromB->fill(particle, truthParticle, weight);
    }
 
    if(m_config.doResolutionsPerAuthor &&  m_iDetailLevel >= 200 and ( !HepMC::is_simulation_particle(&truthParticle) and uniqueID != HepMC::UNDEFINED_ID and prob > 0.5)){
      std::bitset<xAOD::TrackPatternRecoInfo::NumberOfTrackRecoInfo>  patternInfo = particle.patternRecoInfo();
      
      bool isSiSpSeededFinder = patternInfo.test(xAOD::TrackPatternRecoInfo::SiSPSeededFinder);
      bool isInDetExtensionProcessor = patternInfo.test(xAOD::TrackPatternRecoInfo::InDetExtensionProcessor);
      bool isTRTSeededTrackFinder = patternInfo.test(xAOD::TrackPatternRecoInfo::TRTSeededTrackFinder);
      bool isTRTStandalone = patternInfo.test(xAOD::TrackPatternRecoInfo::TRTStandalone);
      bool isSiSpacePointsSeedMaker_LargeD0 = patternInfo.test(xAOD::TrackPatternRecoInfo::SiSpacePointsSeedMaker_LargeD0);
 
      if(isSiSpSeededFinder and not isInDetExtensionProcessor) m_resSiSPSeededFinderPlots->fill(particle, truthParticle, weight);
      if(isInDetExtensionProcessor and not (isTRTSeededTrackFinder or isSiSpacePointsSeedMaker_LargeD0)) m_resInDetExtensionProcessorPlots->fill(particle, truthParticle, weight);
      if(isTRTSeededTrackFinder and not isTRTStandalone) m_resTRTSeededTrackFinderPlots->fill(particle, truthParticle, weight);
      if(isTRTStandalone) m_resTRTStandalonePlots->fill(particle, truthParticle, weight);
      if(isSiSpacePointsSeedMaker_LargeD0) m_resSiSpacePointsSeedMaker_LargeD0Plots->fill(particle, truthParticle, weight);
 
    }
 
    if (m_trtExtensionPlots && !HepMC::is_simulation_particle(&truthParticle) && uniqueID != HepMC::UNDEFINED_ID && prob > 0.5 ) m_trtExtensionPlots->fill(particle, truthParticle, weight);
  }
 
  if(m_hitsMatchedTracksPlots){
    int uniqueID = HepMC::uniqueID(truthParticle);
    if (!HepMC::is_simulation_particle(&truthParticle) && uniqueID != HepMC::UNDEFINED_ID) { 
      m_hitsMatchedTracksPlots->fill(particle, mu, weight);
    }
  }
}
 
//
//Fill basic track properties for reconstructed tracks 
//
 
void
InDetRttPlots::fill(const xAOD::TrackParticle& particle, float weight) {
  if (m_hitResidualPlot) m_hitResidualPlot->fill(particle, weight);
  if (m_hitEffPlot) m_hitEffPlot->fill(particle, weight);
  // fill pt plots
  if (m_trackParameters) m_trackParameters->fill(particle, weight);
 
  if(m_config.doTrackParametersPerAuthor){
    std::bitset<xAOD::TrackPatternRecoInfo::NumberOfTrackRecoInfo>  patternInfo = particle.patternRecoInfo();
    
    bool isSiSpSeededFinder = patternInfo.test(0);
    bool isInDetExtensionProcessor = patternInfo.test(3);
    bool isTRTSeededTrackFinder = patternInfo.test(4);
    bool isTRTStandalone = patternInfo.test(20);
    bool isSiSpacePointsSeedMaker_LargeD0 = patternInfo.test(49);
 
    if(isSiSpSeededFinder and not isInDetExtensionProcessor) m_trkParaSiSPSeededFinderPlots->fill(particle, weight);
    else if(isInDetExtensionProcessor and not (isTRTSeededTrackFinder or isSiSpacePointsSeedMaker_LargeD0)) m_trkParaInDetExtensionProcessorPlots->fill(particle, weight);
    else if(isTRTSeededTrackFinder and not isTRTStandalone) m_trkParaTRTSeededTrackFinderPlots->fill(particle, weight);
    else if(isTRTStandalone) m_trkParaTRTStandalonePlots->fill(particle, weight);
    else if(isSiSpacePointsSeedMaker_LargeD0) m_trkParaSiSpacePointsSeedMaker_LargeD0Plots->fill(particle, weight);
 
  }
 
  if (m_trtExtensionPlots) m_trtExtensionPlots->fill(particle, weight);
}
 
void
InDetRttPlots::fill(const xAOD::TrackParticle& particle, const float mu, const unsigned int nVtx, float weight) {
 
  if (m_trtExtensionPlots) m_trtExtensionPlots->fill(particle, mu, nVtx, weight);
  if (m_hitsRecoTracksPlots) m_hitsRecoTracksPlots->fill(particle, mu, weight);
 
}
 
void
InDetRttPlots::fill(const unsigned int ntracks, const unsigned int truthMu,
            const float actualMu, const unsigned int nvertices,
            const float weight) {
 
  if (m_nTracks) m_nTracks->fill(ntracks, truthMu, actualMu, nvertices, weight);
  
}
//
//Fill plots for selected truth particle
//
 
void
InDetRttPlots::fill(const xAOD::TruthParticle& truthParticle, float weight) {
  // fill truth plots 
  if (m_trackParameters) m_trackParameters->fill(truthParticle, weight);
}
 
//
//Fill Efficiencies
//
 
void
InDetRttPlots::fillEfficiency
(const xAOD::TruthParticle& truth, const xAOD::TrackParticle* track,
 const bool isGood, const unsigned int truthMu, const float actualMu, float weight) {
  if (m_effPlots) m_effPlots->fill(truth, isGood, truthMu, actualMu, weight);
 
  if(m_config.doEfficienciesPerAuthor){
 
    bool isGoodSiSPSeededFinder = false;
    bool isGoodInDetExtensionProcessor = false;
    bool isGoodTRTSeededTrackFinder = false;
    bool isGoodTRTStandalone = false;
    bool isGoodSiSpacePointsSeedMaker_LargeD0 = false;
 
    if(isGood && track){
      std::bitset<xAOD::TrackPatternRecoInfo::NumberOfTrackRecoInfo>  patternInfo = track->patternRecoInfo();
 
      bool isSiSpSeededFinder = patternInfo.test(0);
      bool isInDetExtensionProcessor = patternInfo.test(3);
      bool isTRTSeededTrackFinder = patternInfo.test(4);
      bool isTRTStandalone = patternInfo.test(20);
      bool isSiSpacePointsSeedMaker_LargeD0 = patternInfo.test(49);
 
      isGoodSiSPSeededFinder = isSiSpSeededFinder and not isInDetExtensionProcessor;
      isGoodInDetExtensionProcessor = isInDetExtensionProcessor and
    not (isTRTSeededTrackFinder or isSiSpacePointsSeedMaker_LargeD0);
      isGoodTRTSeededTrackFinder = isTRTSeededTrackFinder and not isTRTStandalone;
      isGoodTRTStandalone = isTRTStandalone;
      isGoodSiSpacePointsSeedMaker_LargeD0 = isSiSpacePointsSeedMaker_LargeD0;
 
    }
 
    m_effSiSPSeededFinderPlots->fill(truth, isGoodSiSPSeededFinder,
                     truthMu, actualMu, weight);
    m_effInDetExtensionProcessorPlots->fill(truth, isGoodInDetExtensionProcessor,
                        truthMu, actualMu, weight);
    m_effTRTSeededTrackFinderPlots->fill(truth, isGoodTRTSeededTrackFinder,
                     truthMu, actualMu, weight);
    m_effTRTStandalonePlots->fill(truth, isGoodTRTStandalone, truthMu, actualMu, weight);
    m_effSiSpacePointsSeedMaker_LargeD0Plots->fill(truth, isGoodSiSpacePointsSeedMaker_LargeD0,
                           truthMu, actualMu, weight);
  }
 
}
 
void InDetRttPlots::fillTechnicalEfficiency
(const xAOD::TruthParticle& truth, const bool isGood,
 const unsigned int truthMu, const float actualMu, float weight) {
  if(m_effPlots) m_effPlots->fillTechnicalEfficiency(truth, isGood, truthMu, actualMu, weight);
}
 
//
//Fill Fake Rates
//
void InDetRttPlots::fillDuplicate
(const xAOD::TruthParticle& truth,
 const std::vector<const xAOD::TrackParticle*>& tracks, float weight) {
  m_duplicatePlots->fill(truth, tracks.size(), weight);
}
 
 
//
//Fill Fake Rates
//
 
void
InDetRttPlots::fillFakeRate(const xAOD::TrackParticle& track, const bool isFake, const bool isAssociatedTruth, const float mu, float weight){
 
  if (m_missingTruthFakePlots) m_missingTruthFakePlots->fill(track, !isAssociatedTruth, weight, mu);
  if(isAssociatedTruth) {
    if (m_fakePlots) m_fakePlots->fill(track, isFake, weight, mu);
    if (m_hitsFakeTracksPlots) m_hitsFakeTracksPlots->fill(track, mu, weight);
    if(m_config.doFakesPerAuthor){
        std::bitset<xAOD::TrackPatternRecoInfo::NumberOfTrackRecoInfo>  patternInfo = track.patternRecoInfo();
        
        bool isSiSpSeededFinder = patternInfo.test(0);
        bool isInDetExtensionProcessor = patternInfo.test(3);
        bool isTRTSeededTrackFinder = patternInfo.test(4);
        bool isTRTStandalone = patternInfo.test(20);
        bool isSiSpacePointsSeedMaker_LargeD0 = patternInfo.test(49);
 
        if(isSiSpSeededFinder and not isInDetExtensionProcessor) m_fakeSiSPSeededFinderPlots->fill(track, isFake, weight, mu); //No extensions 
        if(isInDetExtensionProcessor and not (isTRTSeededTrackFinder or isSiSpacePointsSeedMaker_LargeD0)) m_fakeInDetExtensionProcessorPlots->fill(track, isFake, weight, mu); //Extensions but not Back-tracking
        if(isTRTSeededTrackFinder and not isTRTStandalone) m_fakeTRTSeededTrackFinderPlots->fill(track, isFake, weight, mu); //BackTracking
        if(isTRTStandalone) m_fakeTRTStandalonePlots->fill(track, isFake, weight, mu); //TRT standalone
        if(isSiSpacePointsSeedMaker_LargeD0) m_fakeSiSpacePointsSeedMaker_LargeD0Plots->fill(track, isFake, weight, mu);
    }
  }
  else { 
    if (m_hitsUnlinkedTracksPlots) m_hitsUnlinkedTracksPlots->fill(track, mu, weight);
  }
 
}
 
 
 
//
//Fill Vertexing Plots
//
void
InDetRttPlots::fill(const xAOD::VertexContainer& vertexContainer, const xAOD::Vertex* recoHardScatter, const std::vector<const xAOD::TruthVertex*>& truthHSVertices, const std::vector<const xAOD::TruthVertex*>& truthPUVertices, float weight) {
  // fill vertex-specific properties, for all vertices and for hard-scattering vertex
 
  for (const auto& vtx : vertexContainer.stdcont()) {
    if (vtx->vertexType() == xAOD::VxType::NoVtx) {
      ATH_MSG_DEBUG("IN InDetRttPlots::fill, found xAOD::VxType::NoVtx");
      continue; // skip dummy vertex
    }
    if (m_vertexPlots) m_vertexPlots->fill(*vtx, weight);
    ATH_MSG_DEBUG("IN InDetRttPlots::fill, filling for all vertices");
  }
 
  if (recoHardScatter) {
    if (m_hardScatterVertexPlots) m_hardScatterVertexPlots->fill(*recoHardScatter, weight);
    if (m_hardScatterVertexTruthMatchingPlots){ 
      if(!truthHSVertices.empty() ) m_hardScatterVertexTruthMatchingPlots->fill(*recoHardScatter,truthHSVertices[0],weight);
      else m_hardScatterVertexTruthMatchingPlots->fill(*recoHardScatter,nullptr,weight); 
    }
    ATH_MSG_DEBUG("IN InDetRttPlots::fill, filling for all HS vertex");
  }
 
  if(m_vertexTruthMatchingPlots) m_vertexTruthMatchingPlots->fill(recoHardScatter, vertexContainer, truthHSVertices, truthPUVertices, weight);
}
 
 
void
InDetRttPlots::fill(const xAOD::VertexContainer& vertexContainer, const unsigned int truthMu, const float actualMu, float weight) {
  if (m_verticesVsMuPlots) m_verticesVsMuPlots->fill(vertexContainer, truthMu, actualMu, weight);
}
 
//
//Fill Counters
//
void
InDetRttPlots::fillCounter(const unsigned int freq, const InDetPerfPlot_nTracks::CounterCategory counter, float weight) {
  if (m_nTracks) m_nTracks->fill(freq, counter, weight);
}
 
//Track in Jet Plots
void
InDetRttPlots::fill(const xAOD::TrackParticle& track, const xAOD::Jet& jet, bool isBjet, bool isFake, bool isUnlinked, bool truthIsFromB, float weight){
  if (m_trkInJetPlots) m_trkInJetPlots->fill(track, jet,weight);
  if (isFake){
    if (m_trkInJetPlots_fake) m_trkInJetPlots_fake->fill(track,jet,weight); 
    if (isBjet && m_trkInJetPlots_fake_bjets) m_trkInJetPlots_fake_bjets->fill(track,jet,weight); 
  }
  else if (isUnlinked){
    if (m_trkInJetPlots_unlinked) m_trkInJetPlots_unlinked->fill(track,jet,weight); 
    if (isBjet && m_trkInJetPlots_unlinked_bjets) m_trkInJetPlots_unlinked_bjets->fill(track,jet,weight); 
  }
  else {
    if (m_trkInJetPlots_matched) m_trkInJetPlots_matched->fill(track,jet,weight); 
    if (isBjet && m_trkInJetPlots_matched_bjets) m_trkInJetPlots_matched_bjets->fill(track,jet,weight); 
  }
  if(isBjet){
    if (m_trkInJetPlots_bjets) m_trkInJetPlots_bjets->fill(track, jet,weight);
    if ( truthIsFromB && m_trkInJetPlots_truthFromB) m_trkInJetPlots_truthFromB->fill(track, jet,weight);
  }
}
 
void
InDetRttPlots::fillEfficiency(const xAOD::TruthParticle& truth, const xAOD::Jet& jet, bool isEfficient, bool isBjet, bool truthIsFromB, float weight) {
  if (m_trkInJetPlots) m_trkInJetPlots->fillEfficiency(truth, jet, isEfficient, weight); 
  if(isBjet && m_trkInJetPlots_bjets) m_trkInJetPlots_bjets->fillEfficiency(truth, jet, isEfficient, weight);
  
  if ( isBjet and m_trkInJetPlots_truthFromB and truthIsFromB ) { // truth is from B
    m_trkInJetPlots_truthFromB->fillEfficiency(truth, jet, isEfficient, weight);
  }
}
 
void
InDetRttPlots::fillFakeRate(const xAOD::TrackParticle& track, const xAOD::Jet& jet, bool isFake, bool isBjet, bool truthIsFromB, float weight) {
  if (m_trkInJetPlots) m_trkInJetPlots->fillFakeRate(track, jet, isFake, weight); 
  if(isBjet && m_trkInJetPlots_bjets) m_trkInJetPlots_bjets->fillFakeRate(track, jet, isFake, weight); 
 
  if ( isBjet and m_trkInJetPlots_truthFromB and truthIsFromB ) { // truth is from B
    m_trkInJetPlots_truthFromB->fillFakeRate(track, jet, isFake, weight);
  }
}
 
//IDPVM Ntuple
void
InDetRttPlots::fillNtuple(const xAOD::TrackParticle& track, const xAOD::Vertex* vtx) {
  // Fill track only entries with dummy truth values
  if (m_ntupleTruthToReco){ 
    m_ntupleTruthToReco->fillTrack(track, vtx);
    m_ntupleTruthToReco->fillTree();
  }
}
 
void
InDetRttPlots::fillNtuple(const xAOD::TruthParticle& truth) {
  // Fill truth only entries with dummy track values
  if (m_ntupleTruthToReco){ 
    m_ntupleTruthToReco->fillTruth(truth);
    m_ntupleTruthToReco->fillTree();
  }
}
 
void 
InDetRttPlots::fillNtuple(const xAOD::TrackParticle& track, const xAOD::TruthParticle& truth, const xAOD::Vertex* vtx, const int truthMatchRanking) {
  // Fill track and truth entries
  if (m_ntupleTruthToReco){ 
    m_ntupleTruthToReco->fillTrack(track, vtx, truthMatchRanking);
    m_ntupleTruthToReco->fillTruth(truth);
    m_ntupleTruthToReco->fillTree();
  }
}