MmClusterVariables.h

Go to the documentation of this file.

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

            unsigned int push_back(const xAOD::MMCluster& cluster);
            void dumpAllHitsInChamber(const Identifier& chamberId);
            void enableSeededDump(); 
        private:
           void dump(const ActsTrk::GeometryContext& gctx,
                     const xAOD::MMCluster& dc);
 
           SG::ReadHandleKey<xAOD::MMClusterContainer> m_key{};
 
           std::string m_collName{};
           MmIdentifierBranch m_id{parent(), m_collName};
           ThreeVectorBranch m_globPos{parent(), m_collName+"_globalPos"};
           VectorBranch<float>& m_locPos{parent().newVector<float>(m_collName+"_localPos")};
           VectorBranch<float>& m_locCov{parent().newVector<float>(m_collName+"_localCov")};
           VectorBranch<uint16_t>& m_time{parent().newVector<uint16_t>(m_collName+"_time")};
           VectorBranch<uint32_t>& m_charge{parent().newVector<uint32_t>(m_collName+"_charge")};
           VectorBranch<uint8_t>& m_author{parent().newVector<uint8_t>(m_collName+"_author")};
           std::unordered_set<Identifier> m_filteredChamb{};
           std::unordered_map<Identifier, unsigned int> m_idOutIdxMap{};
           std::vector<const xAOD::MMCluster*> m_dumpedPRDS{};
           bool m_applyFilter{false};
    };
}
#endif