TruthRelatedMuonPlotOrganizer.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "MuonHistUtils/TruthRelatedMuonPlotOrganizer.h"
#include "xAODTruth/TruthParticleContainer.h"
#include "xAODTruth/TruthParticleAuxContainer.h"
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODTracking/TrackParticleAuxContainer.h"
#include "AthContainers/ConstAccessor.h"
 
typedef ElementLink< xAOD::TruthParticleContainer > TruthLink;
 
namespace Muon{
  
TruthRelatedMuonPlotOrganizer::TruthRelatedMuonPlotOrganizer(PlotBase* pParent, const std::string& sDir, bool doBinnedResolutionPlots, std::vector<int> selPlots):
  PlotBase(pParent, sDir)
  // Truth related plots
{
  
  if (selPlots.empty()) {
    for (unsigned int i=0; i<MAX_TRUTHRELATEDPLOTCLASS; i++) m_selPlots.push_back(i);
  }
  else m_selPlots = std::move(selPlots);
  
  for (auto p: m_selPlots) {
    switch (p) {    
    case TRK_MATCHEDTRUE:
      m_oMatchedPlots = std::make_unique< Trk::ParamPlots>(this, "/kinematics/", "Matched Muons");
      break;
    case TRK_MATCHEDRECO:
      m_oMatchedRecoPlots = std::make_unique< Trk::ParamPlots>(this, "/kinematicsReco/", "Matched Muons");
      break;
    case TRK_MSHITDIFF:
      m_oMSHitDiffPlots = std::make_unique< Trk::MSHitDiffPlots>(this,"/hits/");
      break;
    case MUON_HITDIFF:
      m_oMuonHitDiffSummaryPlots = std::make_unique< Muon::MuonHitDiffSummaryPlots>(this,"/hits/");
      break;
    case MUON_TRUTHHIT:
      m_oMuonTruthHitPlots = std::make_unique< Muon::MuonTruthHitPlots>(this,"/truthHits/");
      break;
    case MUON_RESOL:
      m_oMuonResolutionPlots = std::make_unique< Muon::MuonResolutionPlots>(this, "/resolution/","",doBinnedResolutionPlots);
      break;
    case TRK_DEFPARAMPULLS:
      m_oDefParamPullPlots = std::make_unique< Trk::DefParamPullPlots>(this, "/pulls/","");
      break;
    case MUON_PULLSTAIL:          
      m_oMomentumTruthPullPlots_Tail = std::make_unique< Muon::MomentumTruthPullPlots>(this,"/momentumPulls/","Tail");
      break;
    case MUON_PULLSNOTAIL:
      m_oMomentumTruthPullPlots_NoTail = std::make_unique< Muon::MomentumTruthPullPlots>(this,"/momentumPulls/","NoTail");
      break;
    case MUON_PARAMELOSS:
      m_oMatchedRecoElossPlots = std::make_unique< Muon::MuonParamElossPlots>(this,"/Eloss/");     
      break;
    }
  }
  
}
TruthRelatedMuonPlotOrganizer::~TruthRelatedMuonPlotOrganizer() = default;
 
  void TruthRelatedMuonPlotOrganizer::fill(const xAOD::TruthParticle& truthMu, const xAOD::Muon& mu, const xAOD::TrackParticleContainer* MSTracks, float weight){
    if (m_oMatchedPlots) m_oMatchedPlots->fill( truthMu , weight);
    if (m_oMuonHitDiffSummaryPlots) m_oMuonHitDiffSummaryPlots->fill(mu, truthMu, weight);
    if (m_oMuonTruthHitPlots) m_oMuonTruthHitPlots->fill(mu,weight);
 
  // for eloss
  if (m_oMatchedRecoElossPlots) m_oMatchedRecoElossPlots->fill(truthMu, mu , weight );
 
  // Tracking related plots
  const xAOD::TrackParticle* primaryTrk = mu.trackParticle(xAOD::Muon::Primary);
  //const xAOD::TrackParticle* meTrk = mu.trackParticle(xAOD::Muon::ExtrapolatedMuonSpectrometerTrackParticle);
  
  if (!primaryTrk) return;
  if (m_oMatchedRecoPlots) m_oMatchedRecoPlots->fill( *primaryTrk , weight);
  if (m_oMSHitDiffPlots) m_oMSHitDiffPlots->fill(*primaryTrk, truthMu, weight);
  if (m_oMuonResolutionPlots) m_oMuonResolutionPlots->fill(*primaryTrk, truthMu, weight);
  if (m_oDefParamPullPlots) m_oDefParamPullPlots->fill(*primaryTrk, truthMu, weight);
  
  if (m_oMomentumTruthPullPlots_NoTail || m_oMomentumTruthPullPlots_Tail) {
    //muon spectrometer track at MS entry (not extrapolated)
    const xAOD::TrackParticle *msTrk(nullptr);  
    //muon extrapolated to IP
    //const xAOD::TrackParticle* msExtrapTrk(0);
  
    //for 20.1.0...
    //for 20.1.3...
    //const xAOD::TrackParticle* msExtrapTrk = mu.trackParticle(xAOD::Muon::ExtrapolatedMuonSpectrometerTrackParticle);
  
    //trying to accomodate both in a way that the code compiles in both releases
    static const SG::ConstAccessor< ElementLink<xAOD::TrackParticleContainer> >
      extrapLinkAcc("extrapolatedMuonSpectrometerTrackParticleLink");
    if (extrapLinkAcc.isAvailable(mu) && extrapLinkAcc(mu).isValid() ) {
      //cool, we got both links:
      //int correctEnumForExtrap = ((int)xAOD::Muon::MuonSpectrometerTrackParticle)+2;
      //msExtrapTrk = mu.trackParticle((xAOD::Muon::TrackParticleType) correctEnumForExtrap);
      msTrk = mu.trackParticle( xAOD::Muon::MuonSpectrometerTrackParticle );
    }
    else {
      // gymnastics to get msTrk...
      const ElementLink<xAOD::TrackParticleContainer>& msExtrapTrkLink = mu.trackParticleLink(xAOD::Muon::MuonSpectrometerTrackParticle);
      static const SG::ConstAccessor<TruthLink>
        truthParticleLinkAcc("truthParticleLink");
      if (msExtrapTrkLink.isValid()) {
    //msExtrapTrk = mu.trackParticle( xAOD::Muon::MuonSpectrometerTrackParticle );
    TruthLink truthLink_muTrk;
    if( truthParticleLinkAcc.isAvailable(**msExtrapTrkLink) ) {
      truthLink_muTrk = truthParticleLinkAcc(**msExtrapTrkLink);
    }
    if (truthLink_muTrk.isValid()) {
      for (const auto trk: *MSTracks) {
        TruthLink truthLink_msTrk;
        if( truthParticleLinkAcc.isAvailable(*trk) ) {
          truthLink_msTrk = truthParticleLinkAcc(*trk);
          if (truthLink_msTrk.isValid()) {
        if (truthLink_msTrk == truthLink_muTrk) {
          msTrk = trk; //got it!
          break;
        }
          }
        }
      }
    }
      }
    }
#ifndef XAOD_ANALYSIS
    float eloss=0; 
    if (mu.parameter(eloss,xAOD::Muon::EnergyLoss)) { 
      if ( mu.energyLossType()!=xAOD::Muon::Tail ) { //to test MEASURED energy loss 
    if (m_oMomentumTruthPullPlots_NoTail) m_oMomentumTruthPullPlots_NoTail->fill(mu, msTrk, truthMu, weight); 
      } 
      else { 
    if (m_oMomentumTruthPullPlots_Tail) m_oMomentumTruthPullPlots_Tail->fill(mu, msTrk, truthMu, weight); //to test PARAMETRIZED energy loss 
      } 
    }
#endif // not XAOD_ANALYSIS
  }
}
 
  void TruthRelatedMuonPlotOrganizer::fill(const xAOD::TruthParticle& truthMu, const xAOD::TrackParticle& muTP, float weight){
  // Tracking related plots
    if (m_oMatchedPlots) m_oMatchedPlots->fill( truthMu, weight );
    if (m_oMatchedRecoPlots) m_oMatchedRecoPlots->fill( muTP , weight);
    if (m_oDefParamPullPlots) m_oDefParamPullPlots->fill(muTP, truthMu, weight);
    if (m_oMuonResolutionPlots) m_oMuonResolutionPlots->fill(muTP, truthMu, weight);
    if (m_oMSHitDiffPlots) m_oMSHitDiffPlots->fill(muTP, truthMu, weight);
}
  
}//close namespace