TruthMuonPlotOrganizer.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONHISTUTILS_TRUTHMUONPLOTORGANIZER_H
#define MUONHISTUTILS_TRUTHMUONPLOTORGANIZER_H
 
#include <vector>
 
#include "TrkValHistUtils/MSHitPlots.h"
#include "TrkValHistUtils/ParamPlots.h"
#include "TrkValHistUtils/PlotBase.h"
#include "TrkValHistUtils/TruthInfoPlots.h"
#include "TrkValHistUtils/TruthTrkExtrapolationPlots.h"
#include "xAODTruth/TruthParticle.h"
 
namespace Muon {
enum TRUTHPLOTCLASS {
    TRUTH_PARAM,
    TRUTH_INFO,
    TRUTH_TRKEXTRAP,
    TRUTH_MSHITS,
    MAX_TRUTHPLOTCLASS
};
 
class TruthMuonPlotOrganizer : public PlotBase {
   public:
    TruthMuonPlotOrganizer(PlotBase* pParent, const std::string& sDir,
                           std::vector<int> selPlots = {});
    ~TruthMuonPlotOrganizer();
 
    std::vector<int> m_selPlots;
 
    void fill(const xAOD::TruthParticle& truthMu, float weight = 1.0);
 
    // Truth plots
    std::unique_ptr<Trk::ParamPlots> m_oTruthPlots{};
    std::unique_ptr<Trk::TruthInfoPlots> m_oTruthInfoPlots{};
    std::unique_ptr<Trk::TruthTrkExtrapolationPlots>
        m_oTruthTrkExtrapolationPlots{};
    std::unique_ptr<Trk::MSHitPlots> m_oTruthMSHitPlots{};
};
 
}  // namespace Muon
 
#endif