SegmentVariables.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONPRDTEST_SEGMENTVARIABLES_H
#define MUONPRDTEST_SEGMENTVARIABLES_H
 
#include "MuonPRDTest/PrdTesterModule.h"
 
#include "StoreGate/ReadDecorHandleKeyArray.h"
#include "xAODMuon/MuonSegmentContainer.h"
 
namespace MuonPRDTest{
    class SegmentVariables : public PrdTesterModule {
        public:
            SegmentVariables(MuonTesterTree& tree, 
                             const std::string& containerKey,
                             const std::string& outName,
                             MSG::Level msglvl);
 
            bool fill(const EventContext& ctx) override final;
            bool declare_keys() override final;
        
            unsigned int push_back(const xAOD::MuonSegment& segment);
            bool addVariable(std::shared_ptr<IAuxElementDecorationBranch> br);
        private:
            unsigned int fill(const xAOD::MuonSegment& segment);
 
            SG::ReadHandleKey<xAOD::MuonSegmentContainer> m_key{};
            std::string m_name{};
            ThreeVectorBranch m_pos{parent(), m_name+"_pos"};
            ThreeVectorBranch m_dir{parent(), m_name+"_dir"};
            VectorBranch<char>& m_etaIdx{parent().newVector<char>(m_name+"_etaIndex")};
            VectorBranch<uint8_t>& m_sector{parent().newVector<uint8_t>(m_name+"_sector")};
            VectorBranch<uint8_t>& m_chamberIdx{parent().newVector<uint8_t>(m_name+"_chamberIdx")};
            VectorBranch<float>& m_chi2{parent().newVector<float>(m_name+"_chi2")};
            VectorBranch<unsigned int>& m_nDoF{parent().newVector<unsigned int>(m_name+"_nDoF")};
            VectorBranch<uint8_t>& m_nPrecHits{parent().newVector<uint8_t>(m_name+"_nPrecHits")};
            VectorBranch<uint8_t>& m_nTrigEtaLayers{parent().newVector<uint8_t>(m_name+"_nTrigEtaLayers")};
            VectorBranch<uint8_t>& m_nTrigPhiLayers{parent().newVector<uint8_t>(m_name+"_nTrigPhiLayers")};
            std::vector<std::shared_ptr<IAuxElementDecorationBranch>> m_addBranches{};
            bool m_filterMode{false};
            std::unordered_map<const xAOD::MuonSegment*, unsigned int> m_idxLookUp{};
 
    };
}
#endif