RecoMuonPlotOrganizer.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#include "MuonHistUtils/RecoMuonPlotOrganizer.h"
 
namespace Muon {
 
RecoMuonPlotOrganizer::RecoMuonPlotOrganizer(PlotBase* pParent,
                                             const std::string& sDir,
                                             std::vector<int> selPlots)
    : PlotBase(pParent, sDir) {
 
    if (selPlots.empty()) {
        for (unsigned int i = 0; i < MAX_RECOPLOTCLASS; i++) {
            m_selPlots.push_back(i);
        }
    } else {
        m_selPlots = std::move(selPlots);
    }
 
    for (int p : m_selPlots) {
        switch (p) {
            case MUON_TRKPARAM:
                m_oTrkParamPlots = std::make_unique<Trk::ParamPlots>(
                    this, "/kinematics/", "RecoMuon");
                break;
            case MUON_PARAM:
                m_oMuonParamPlots = std::make_unique<Muon::MuonParamPlots>(
                    this, "/parameters/");
                break;
            case MUON_RECOINFO:
                m_oMuRecoInfoPlots =
                    std::make_unique<Muon::RecoInfoPlots>(this, "/parameters/");
                break;
            case MUON_TRKIMPACT:
                m_oImpactPlots =
                    std::make_unique<Trk::ImpactPlots>(this, "/parameters/");
                break;
            case MUON_MOMPULLS:
                m_oMomentumPullPlots =
                    std::make_unique<Muon::MomentumPullPlots>(
                        this, "/momentumPulls/");
                break;
            case MUON_HITS:
                m_oMuonHitSummaryPlots =
                    std::make_unique<Muon::MuonHitSummaryPlots>(this, "/hits/");
                break;
            case MUON_IDHITS:
                m_oIDHitPlots =
                    std::make_unique<Muon::IDHitSummaryPlots>(this, "/hits/");
                break;
            case MUON_ISO:
                m_oMuonIsolationPlots =
                    std::make_unique<Muon::MuonIsolationPlots>(this,
                                                               "/isolation/");
                break;
            case MUON_CHARGEPARAM:
                m_oChargeParamPlotsLowPt =
                    std::make_unique<Muon::ChargeDepParamPlots>(
                        this, "/kinematics/", "lowPt");
                m_oChargeParamPlotsHighPt =
                    std::make_unique<Muon::ChargeDepParamPlots>(
                        this, "/kinematics/", "highPt");
                break;
        }
    }
}
 
RecoMuonPlotOrganizer::~RecoMuonPlotOrganizer() = default;
 
void RecoMuonPlotOrganizer::fill(const xAOD::Muon& mu, float weight) {
    if (m_oIDHitPlots && mu.trackParticle(xAOD::Muon::TrackParticleType::InnerDetectorTrackParticle)) {
        m_oIDHitPlots->fill(
            *mu.trackParticle(xAOD::Muon::TrackParticleType::InnerDetectorTrackParticle), weight);
    }
    if (m_oTrkParamPlots) {
        m_oTrkParamPlots->fill(mu, weight);
    }
    if (m_oMuonParamPlots) {
        m_oMuonParamPlots->fill(mu, weight);
    }
    if (m_oMuRecoInfoPlots) {
        m_oMuRecoInfoPlots->fill(mu, weight);
    }
    if (m_oMomentumPullPlots) {
        m_oMomentumPullPlots->fill(mu, weight);
    }
    if (m_oMuonHitSummaryPlots) {
        m_oMuonHitSummaryPlots->fill(mu, weight);
    }
    if (m_oMuonIsolationPlots) {
        m_oMuonIsolationPlots->fill(mu, weight);
    }
    if (m_oChargeParamPlotsLowPt && m_oChargeParamPlotsHighPt) {
        if (mu.pt() < 8000) {
            m_oChargeParamPlotsLowPt->fill(mu, weight);
        } else {
            m_oChargeParamPlotsHighPt->fill(mu, weight);
        }
    }
 
    // tracking related plots
    const xAOD::TrackParticle* primaryTrk =
        mu.trackParticle(xAOD::Muon::Primary);
    if (!primaryTrk) {
        return;
    }
    if (m_oImpactPlots) {
        m_oImpactPlots->fill(*primaryTrk, weight);
    }
}
 
}  // namespace Muon