NtracksPlots.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDETTRACKPERFMON_PLOTS_NTRACKSPLOTS_H
#define INDETTRACKPERFMON_PLOTS_NTRACKSPLOTS_H


#include "../PlotMgr.h"
#include "../TrackAnalysisCollections.h"
 
 
namespace IDTPM {
 
  class NtracksPlots : public PlotMgr {
 
  public:

    NtracksPlots(
        PlotMgr* pParent,
        const std::string& dirName,
        const std::string& anaTag,
        const std::string& trackType,
        bool doTrigger = false,
        bool doTruthMuPlots = false );

    virtual ~NtracksPlots() = default;

    StatusCode fillPlots(
        const std::vector< size_t >& trackCounts,
        const std::vector< size_t >& vertexCounts,
        float truthMu,
        float actualMu,
        float weight );

    void initializePlots(); // needed to override PlotBase
    StatusCode bookPlots();

    void finalizePlots();
 
  private:
 
    std::string m_trackType;
    bool m_doTrigger{};
    bool m_doTruthMuPlots{};

    enum Counter : size_t {
        ALL       = TrackAnalysisCollections::ALL,
        SELECTED  = TrackAnalysisCollections::SELECTED,
        INROI     = TrackAnalysisCollections::INROI,
        MATCHED   = TrackAnalysisCollections::MATCHED,
        NCOUNTERS = TrackAnalysisCollections::NCOUNTERS
    };
 
    std::string m_counterName[ NCOUNTERS ] = {
        "all", "selected", "selectedInRoI", "matched"
    };
 
    TH1*  m_nTracks[ NCOUNTERS ]{};
    TH2*  m_nTracks_vs_nVertices[ NCOUNTERS ]{};
    TH2*  m_nTracks_vs_truthMu[ NCOUNTERS ]{};
    TH2*  m_nTracks_vs_actualMu[ NCOUNTERS ]{};
    TProfile*  m_avg_nTracks_vs_truthMu[ NCOUNTERS ]{};
    TProfile*  m_avg_nTracks_vs_actualMu[ NCOUNTERS ]{};
 
  }; // class NtracksPlots
 
} // namespace IDTPM
 
#endif // > ! INDETTRACKPERFMON_PLOTS_NTRACKSPLOTS_H