MdtTwinDriftCircleVariables.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef PRDTESTERR4_MDTTWINDRITFCIRCLEVARIABLES_H
#define PRDTESTERR4_MDTTWINDRITFCIRCLEVARIABLES_H
#include "MuonPRDTestR4/TesterModuleBase.h"
#include "xAODMuonPrepData/MdtTwinDriftCircleContainer.h"
#include "MuonTesterTree/IdentifierBranch.h"
 
#include <unordered_map>
#include <unordered_set>
namespace MuonValR4{
 
    class MdtTwinDriftCircleVariables: public TesterModuleBase {
        public:
            MdtTwinDriftCircleVariables(MuonTesterTree& tree,
                                        const std::string& inContainer,
                                        MSG::Level msgLvl = MSG::Level::INFO,
                                        const std::string& collName="MdtTwinPrd");
 
            bool declare_keys() override final;
 
            
            bool fill(const EventContext& ctx) override final;

            unsigned int push_back(const xAOD::MdtTwinDriftCircle& dc);
            void dumpAllHitsInChamber(const Identifier& chamberId);
            void enableSeededDump(); 
        private:
           void dump(const ActsTrk::GeometryContext& gctx,
                     const xAOD::MdtTwinDriftCircle& dc);
 
           SG::ReadHandleKey<xAOD::MdtTwinDriftCircleContainer> 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<float>& m_twinLocZ{parent().newVector<float>(m_collName+"_posAlongWire")};
           VectorBranch<float>& m_twinUncertLocZ{parent().newVector<float>(m_collName+"_uncertAlongWire")};
           VectorBranch<uint16_t>& m_tdcCounts{parent().newVector<uint16_t>(m_collName+"_tdc")};
           VectorBranch<uint16_t>& m_adcCounts{parent().newVector<uint16_t>(m_collName+"_adc")};
           VectorBranch<uint16_t>& m_twinTdcCounts{parent().newVector<uint16_t>(m_collName+"_twinTdc")};
           VectorBranch<uint16_t>& m_twinAdcCounts{parent().newVector<uint16_t>(m_collName+"_twinAdc")};
           VectorBranch<uint16_t>& m_twinTube{parent().newVector<uint16_t>(m_collName+"_twinTube")};
           VectorBranch<uint8_t>& m_twinLayer{parent().newVector<uint8_t>(m_collName+"_twinLayer")};

           std::unordered_set<Identifier> m_filteredChamb{};
           std::unordered_map<Identifier, unsigned int> m_idOutIdxMap{};
           std::vector<const xAOD::MdtTwinDriftCircle*> m_dumpedPRDS{};
           bool m_applyFilter{false};
    };
}
#endif