TrackContainerModule.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef PRDTESTERR4_TRACKCONTAINERMODULE_H
#define PRDTESTERR4_TRACKCONTAINERMODULE_H
#include "MuonPRDTestR4/TesterModuleBase.h"
#include "ActsEvent/TrackContainer.h"
#include "ActsGeometryInterfaces/GeometryContext.h"
 
namespace MuonValR4{
    class TrackFitIterBranch : public MuonVal::VectorBranch<std::uint16_t>,
                               virtual public MuonVal::IParticleDecorationBranch {
        public:
            TrackFitIterBranch(IParticleFourMomBranch& parent);
 
            using VectorBranch<std::uint16_t>::push_back;
            virtual void push_back(const xAOD::IParticle* p) override;
            virtual void push_back(const xAOD::IParticle& p) override;
            virtual void operator+=(const xAOD::IParticle* p) override;
            virtual void operator+=(const xAOD::IParticle& p) override;
 
    };

    class MaterialRecorderBranch : public MuonVal::VectorBranch<float>,
                                   virtual public MuonVal::IParticleDecorationBranch {
        public:
            
            MaterialRecorderBranch(MuonVal::IParticleFourMomBranch& parent);
 
            using MuonVal::VectorBranch<float>::push_back;
            virtual void push_back(const xAOD::IParticle* p) override;
            virtual void push_back(const xAOD::IParticle& p) override;
            virtual void operator+=(const xAOD::IParticle* p) override;
            virtual void operator+=(const xAOD::IParticle& p) override;
            virtual bool init() override final;
 
        private:
            ActsTrk::GeoContextReadKey_t m_geoCtxKey{"ActsAlignment"};
            std::shared_ptr<MuonVal::VectorBranch<float>> m_thickX0{};
            std::shared_ptr<MuonVal::VectorBranch<std::uint8_t>> m_nStates{};
    };  

    class EnergyLossBranch: public MuonVal::VectorBranch<float>,
                            virtual public MuonVal::IParticleDecorationBranch {
        public:
            EnergyLossBranch(MuonVal::IParticleFourMomBranch& parent);
 
            using MuonVal::VectorBranch<float>::push_back;
            virtual void push_back(const xAOD::IParticle* p) override;
            virtual void push_back(const xAOD::IParticle& p) override;
            virtual void operator+=(const xAOD::IParticle* p) override;
            virtual void operator+=(const xAOD::IParticle& p) override;  
    };                
}
 
#endif