sTGCPRDVariables.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MuonPRDTEST_sTGCPRDVARIABLES_H
#define MuonPRDTEST_sTGCPRDVARIABLES_H
 
#include "MuonPRDTest/PrdTesterModule.h"
#include "MuonPrepRawData/sTgcPrepDataContainer.h"
#include "MuonTesterTree/TwoVectorBranch.h"
 
namespace MuonPRDTest{
    class sTGCPRDVariables : public PrdTesterModule {
    public:

 
        sTGCPRDVariables(MuonTesterTree& tree,
                         const std::string& container_name,
                         MSG::Level msglvl);
    
        ~sTGCPRDVariables() = default;
    
        bool fill(const EventContext& ctx) override final;
    
        bool declare_keys() override final;

        unsigned int push_back(const Muon::sTgcPrepData& prd);

        void dumpAllHitsInChamber(const MuonGM::sTgcReadoutElement& detEle);
        void enableSeededDump();
 
    
    private:
        unsigned int dump(const Muon::sTgcPrepData& prd);
        SG::ReadHandleKey<Muon::sTgcPrepDataContainer> m_key{};
        ScalarBranch<unsigned int>& m_NSWsTGC_nPRD{parent().newScalar<unsigned int>("N_PRD_sTGC")};
        VectorBranch<double>& m_NSWsTGC_PRD_localPosX{parent().newVector<double>("PRD_sTGC_localPosX")};
        VectorBranch<double>& m_NSWsTGC_PRD_localPosY{parent().newVector<double>("PRD_sTGC_localPosY")};
        ThreeVectorBranch m_NSWsTGC_PRD_globalPos{parent(), "PRD_sTGC_globalPos"};
        VectorBranch<double>& m_NSWsTGC_PRD_covMatrix_1_1{parent().newVector<double>("PRD_sTGC_covMatrix_1_1")};
        VectorBranch<double>& m_NSWsTGC_PRD_covMatrix_2_2{parent().newVector<double>("PRD_sTGC_covMatrix_2_2")};
        sTgcIdentifierBranch m_NSWsTGC_PRD_id{parent(), "PRD_sTGC"};
 
        VectorBranch<int>& m_NSWsTGC_PRD_charge{parent().newVector<int>("PRD_sTGC_charge")};
        VectorBranch<short>& m_NSWsTGC_PRD_time{parent().newVector<short>("PRD_sTGC_time")};
        VectorBranch<int>& m_NSWsTGC_PRD_nStrips{parent().newVector<int>("PRD_sTGC_nStrips")};
        
        MatrixBranch<int>& m_NSWsTGC_PRD_stripCharge{parent().newMatrix<int>("PRD_sTGC_stripCharge")};
        MatrixBranch<short>& m_NSWsTGC_PRD_stripTime{parent().newMatrix<short>("PRD_sTGC_stripTime")};
        MatrixBranch<uint16_t>& m_NSWsTGC_PRD_stripChannel{parent().newMatrix<uint16_t>("PRD_sTGC_stripChannel")};
 
        VectorBranch<uint8_t>& m_stgcAuthor{parent().newVector<uint8_t>("stgcAuthor")};
        VectorBranch<uint8_t>& m_stgcQuality{parent().newVector<uint8_t>("stgcQuality")};
        
        
        std::unordered_set<Identifier> m_filteredChamb{};
        std::unordered_map<Identifier, unsigned int> m_filteredPRDs{};
        bool m_applyFilter{false};
        bool m_externalPush{false};
    };
};
 
#endif  // MuonPRDTEST_sTGCPRDVARIABLES_H