MdtDriftCircleVariables.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef PRDTESTERR4_MDTDRITFCIRCLEVARIABLES_H
#define PRDTESTERR4_MDTDRITFCIRCLEVARIABLES_H
#include "MuonPRDTestR4/TesterModuleBase.h"
#include "xAODMuonPrepData/MdtDriftCircleContainer.h"
#include "MuonTesterTree/IdentifierBranch.h"
 
#include <unordered_map>
#include <unordered_set>
namespace MuonValR4{
 
    class MdtDriftCircleVariables: public TesterModuleBase {
        public:
            MdtDriftCircleVariables(MuonTesterTree& tree,
                                    const std::string& inContainer,
                                    MSG::Level msgLvl = MSG::Level::INFO,
                                    const std::string& collName="MdtPrd");
 
            bool declare_keys() override final;
 
            
            bool fill(const EventContext& ctx) override final;
 
            unsigned int push_back(const xAOD::MdtDriftCircle& dc);
            void dumpAllHitsInChamber(const Identifier& chamberId);
            void enableSeededDump(); 
        private:
           void dump(const ActsGeometryContext& gctx,
                     const xAOD::MdtDriftCircle& dc);
 
           SG::ReadHandleKey<xAOD::MdtDriftCircleContainer> m_key{};
 
           std::string m_collName{};
           MdtIdentifierBranch m_id{parent(), m_collName};
           ThreeVectorBranch m_globPos{parent(), m_collName+"_globalPos"};
           VectorBranch<float>& m_driftRadius{parent().newVector<float>(m_collName+"_driftRadius")};
           VectorBranch<float>& m_driftRadiusUncert{parent().newVector<float>(m_collName+"_uncertDriftRadius")};
           VectorBranch<uint16_t>& m_tdcCounts{parent().newVector<uint16_t>(m_collName+"_tdc")};
           VectorBranch<uint16_t>& m_adcCounts{parent().newVector<uint16_t>(m_collName+"_adc")};
    
           std::unordered_set<Identifier> m_filteredChamb{};
           std::unordered_map<Identifier, unsigned int> m_idOutIdxMap{};
           std::vector<const xAOD::MdtDriftCircle*> m_dumpedPRDS{};
           bool m_applyFilter{false};
    };
}
#endif