MuonTree.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONHISTUTILS_MUONTREE_H
#define MUONHISTUTILS_MUONTREE_H
 
#include <memory>
#include <vector>
 
#include "TTree.h"
#include "TrkValHistUtils/PlotBase.h"
#include "xAODEventInfo/EventInfo.h"
#include "xAODMuon/Muon.h"
#include "xAODTruth/TruthParticle.h"
 
namespace Muon {
 
class MuonTree : public PlotBase {
   public:
    MuonTree(PlotBase* pParent, const std::string& sDir, bool mcFlag);
    MuonTree(PlotBase* pParent, const std::string& sDir);
    ~MuonTree();
 
    void initializePlots();
    void fillEventBranches(const xAOD::EventInfo* eventInfo, bool isData);
    void fillRecoMuonBranches(const xAOD::Muon& mu);
    void fillTruthMuonBranches(const xAOD::TruthParticle& truthMu,
                               bool isGoodTruthTrack);
    void fillTruthMuonBranches(const xAOD::TruthParticle& truthMu,
                               const xAOD::Muon& mu,
                               const xAOD::TrackParticleContainer* MSTracks,
                               bool isGoodTruthTrack);
    void postFillTreeActions();
    TTree* getTree();
 
   private:
    std::unique_ptr<TTree> m_tree;
 
    // steering flag
    bool m_isMC;
 
    // Event info variables
    uint32_t m_runNumber;
    uint32_t m_lumiBlock;
    unsigned long long m_eventNumber;
    uint32_t m_mcChannelNumber;
    Float_t m_mcBeamSpotWeight;
    //
    // Reco variables
    //
 
    // general properties
    std::vector<float> m_pt;
    std::vector<float> m_eta;
    std::vector<float> m_phi;
    std::vector<float> m_e;
    std::vector<float> m_rapidity;
 
    // authors and types
    std::vector<uint16_t> m_allAuthors;
    std::vector<int> m_muonType;
 
    // summary values
    std::vector<int8_t> m_innerSmallHits;
    std::vector<int8_t> m_innerLargeHits;
    std::vector<int8_t> m_innerSmallHoles;
    std::vector<int8_t> m_innerLargeHoles;
 
    // parameters
    std::vector<float> m_msInnerMatchChi2;
    std::vector<int> m_msInnerMatchDOF;
    std::vector<float> m_msOuterMatchChi2;
    std::vector<int> m_msOuterMatchDOF;
    std::vector<float> m_CaloLRLikelihood;
    std::vector<int> m_CaloMuonIDTag;
    std::vector<float> m_EnergyLoss;
 
    // quality
    std::vector<int> m_Quality;
 
    // isolation
    std::vector<float> m_iso_etcone20;
    std::vector<float> m_iso_ptcone20;
    std::vector<float> m_iso_topoetcon20;
    std::vector<float> m_iso_ptvarcon20;
    std::vector<float> m_iso_neflowisol20;
 
    // associated TrackParticle
    std::vector<float> m_trkp_d0;
    std::vector<float> m_trkp_z0;
    std::vector<float> m_trkp_qOverP;
    std::vector<float> m_trkp_chiSquared;
    std::vector<float> m_trkp_numberDoF;
 
    // used MuonSegments
    std::vector<size_t> m_nMuonSegments;
    std::vector<std::vector<float> > m_museg_x;
    std::vector<std::vector<float> > m_museg_y;
    std::vector<std::vector<float> > m_museg_z;
    std::vector<std::vector<int> > m_museg_sector;
    std::vector<std::vector<unsigned int> > m_museg_chamberIndex;
    std::vector<std::vector<unsigned int> > m_museg_technology;
    std::vector<std::vector<int> > m_museg_nPrecisionHits;
    std::vector<std::vector<int> > m_museg_nPhiLayers;
    std::vector<std::vector<int> > m_museg_nTrigEtaLayers;
 
    //
    // Truth variables
    //
 
    // Trk::Param
    //  all TruthTrack
    std::vector<bool> m_th_isGoodTruthTrack;
 
    std::vector<double> m_th_pt;
    std::vector<double> m_th_eta;
    std::vector<double> m_th_phi;
 
    // Trk::TruthInfo
    //  all TruthTrack
    std::vector<int> m_th_truthType;
    std::vector<int> m_th_truthOrigin;
 
    // Trk::TruthTrkExtrapolation
    //  all TruthTrack
    std::vector<float> m_th_CaloEntry_p;
    std::vector<float> m_th_MuonEntry_p;
    std::vector<float> m_th_MuonExit_p;
 
    // Trk::MSHit
    //  all TruthTrack
    std::vector<int8_t> m_th_nprecLayers;
    std::vector<int8_t> m_th_nphiLayers;
    std::vector<int8_t> m_th_ntrigEtaLayers;
};
 
}  // namespace Muon
 
#endif