SpacePointTesterModule.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef PRDTESTERR4_MUONSPACEPOINTTESTMODULE_H
#define PRDTESTERR4_MUONSPACEPOINTTESTMODULE_H
#include "MuonPRDTestR4/TesterModuleBase.h"
 
#include "MuonSpacePoint/SpacePointContainer.h"
#include "MuonTesterTree/IdentifierBranch.h"
namespace MuonValR4{
    class SpacePointTesterModule: public TesterModuleBase {
        public:
            SpacePointTesterModule(MuonTesterTree& tree,
                                   const std::string& inContainer,
                                   MSG::Level msgLvl = MSG::Level::INFO,
                                   const std::string& collName="");
 
            bool declare_keys() override final;
 
            bool fill(const EventContext& ctx) override final;
 
            
            unsigned int push_back(const MuonR4::SpacePointBucket& bucket);
            unsigned int push_back(const MuonR4::SpacePoint& spacePoint);
 
 
        private:
           std::string m_collName{};
           SG::ReadHandleKey<MuonR4::SpacePointContainer> m_key{};
           VectorBranch<uint16_t>& m_bucketNumber{parent().newVector<uint16_t>(m_collName+"bucket_index")};
           MuonIdentifierBranch m_bucketId{parent(), m_collName+"_bucket"};
           VectorBranch<float>& m_bucketMin{parent().newVector<float>(m_collName+"_bucketMinX")};
           VectorBranch<float>& m_bucketMax{parent().newVector<float>(m_collName+"_bucketMaxX")};
           MatrixBranch<uint16_t>& m_bucketPoints{parent().newMatrix<uint16_t>(m_collName+"_bucketSpacePoints")};
 
         
           ThreeVectorBranch m_spPos{parent(), m_collName+"spacePoint_Position"};
           VectorBranch<float>& m_driftR{parent().newVector<float>(m_collName+"_spacePointDriftR")};
           VectorBranch<float>& m_covX{parent().newVector<float>(m_collName+"_spacePointCovX")};
           VectorBranch<float>& m_covY{parent().newVector<float>(m_collName+"_spacePointCovY")};
           VectorBranch<float>& m_covT{parent().newVector<float>(m_collName+"_spacePointCovT")};
           VectorBranch<bool>& m_measEta{parent().newVector<bool>(m_collName+"_spacePointMeasEta")};
           VectorBranch<bool>& m_measPhi{parent().newVector<bool>(m_collName+"_spacePointMeasPhi")};
           VectorBranch<unsigned int>& m_nEtaInstances{parent().newVector<unsigned int>(m_collName+"_spacePointNumEtaInUse")};
           VectorBranch<unsigned int>& m_nPhiInstances{parent().newVector<unsigned int>(m_collName+"_spacePointNumPhiInUse")};
 

           MuonIdentifierBranch m_spacePointId{parent(), m_collName+"_spacePoint"};
           VectorBranch<unsigned char>& m_techIdx{parent().newVector<unsigned char>(m_collName+"_spacePointTechnology")};
           VectorBranch<unsigned char>& m_layer{parent().newVector<unsigned char>(m_collName+"_spacePointLayer")};
           VectorBranch<uint16_t>& m_channel{parent().newVector<uint16_t>(m_collName+"_spacePoint1stChannel")};
           VectorBranch<int16_t>& m_phiChannel{parent().newVector<int16_t>(m_collName+"_spacePoint2ndChannel")};

           std::unordered_map<const MuonR4::SpacePoint*, unsigned int> m_spacePointIdx{};
           std::unordered_map<const MuonR4::SpacePointBucket*, unsigned int> m_bucketIdx{};
           bool m_applyFilter{false};
           bool m_internalFill{false};
    };
}
#endif