TruthMuonPlotOrganizer.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#include "MuonHistUtils/TruthMuonPlotOrganizer.h"
 
namespace Muon {
 
TruthMuonPlotOrganizer::TruthMuonPlotOrganizer(PlotBase* pParent,
                                               const std::string& sDir,
                                               std::vector<int> selPlots)
    : PlotBase(pParent, sDir) {
    if (selPlots.empty()) {
        for (unsigned int i = 0; i < MAX_TRUTHPLOTCLASS; i++) {
            m_selPlots.push_back(i);
        }
    } else {
        m_selPlots = std::move(selPlots);
    }
 
    for (auto p : m_selPlots) {
        switch (p) {
            case TRUTH_PARAM:
                m_oTruthPlots =
                    std::make_unique<Trk::ParamPlots>(this, "/", "Truth Muons");
                break;
            case TRUTH_INFO:
                m_oTruthInfoPlots =
                    std::make_unique<Trk::TruthInfoPlots>(this, "/");
                break;
            case TRUTH_TRKEXTRAP:
                m_oTruthTrkExtrapolationPlots =
                    std::make_unique<Trk::TruthTrkExtrapolationPlots>(this,
                                                                      "/");
                break;
            case TRUTH_MSHITS:
                m_oTruthMSHitPlots =
                    std::make_unique<Trk::MSHitPlots>(this, "/");
                break;
        }
    }
}
TruthMuonPlotOrganizer::~TruthMuonPlotOrganizer() = default;
 
void TruthMuonPlotOrganizer::fill(const xAOD::TruthParticle& truthMu,
                                  float weight) {
    // General Truth Plots
    if (m_oTruthPlots) {
        m_oTruthPlots->fill(truthMu, weight);
    }
    if (m_oTruthInfoPlots) {
        m_oTruthInfoPlots->fill(truthMu, weight);
    }
    if (m_oTruthTrkExtrapolationPlots) {
        m_oTruthTrkExtrapolationPlots->fill(truthMu, weight);
    }
    if (m_oTruthMSHitPlots) {
        m_oTruthMSHitPlots->fill(truthMu, weight);
    }
}
 
}  // namespace Muon