TrackSummaryModule.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONPRDTESTR4_TRACKSUMMARYMODULE_H
#define MUONPRDTESTR4_TRACKSUMMARYMODULE_H
 
#include "MuonPRDTestR4/TesterModuleBase.h"
#include "MuonRecToolInterfacesR4/ITrackSummaryTool.h"
#include "ActsEvent/TrackContainer.h"
#include "MuonTrackEvent/HitSummary.h"
 
 
namespace MuonValR4{
    class TrackSummaryValueBranch : public MuonVal::VectorBranch<std::uint8_t> {
        public:
            using Category = MuonR4::HitSummary::HitCategory;
            using Status = MuonR4::HitSummary::Status;
            using LayerIndex = MuonR4::HitSummary::LayerIndex;
            TrackSummaryValueBranch(TTree* tree,
                                    const std::string& collName,
                                    Category cat,
                                    Status status,
                                    LayerIndex layer,
                                    bool isSmall);
            using VectorBranch<std::uint8_t>::push_back;
            void push_back(const MuonR4::HitSummary& summary);
        private:
            Category m_cat{Category::nCategories};
            Status m_status{Status::MaxValue};
            LayerIndex m_layer{LayerIndex::LayerIndexMax};
            bool m_isSmall{false};
    };
 
    class TrackSummaryModule: public MuonVal::MuonTesterBranch {
        public:
            TrackSummaryModule(MuonVal::MuonTesterTree& parent,
                                const std::string& collName,
                                const MuonR4::ITrackSummaryTool* summaryTool);
            bool fill(const EventContext& ctx) override;
            bool init() override;
 
            using ConstTrack_t = MuonR4::ITrackSummaryTool::ConstTrack_t;
 
            void push_back(const EventContext& ctx, const ConstTrack_t track);
            void push_back(const EventContext& ctx, const MuonR4::MsTrackSeed& seed);
            void push_back(const MuonR4::HitSummary& summary);
 
 
        private:
            using ValuePtr_t = std::shared_ptr<TrackSummaryValueBranch>;
            std::vector<ValuePtr_t> m_values{};
            const MuonR4::ITrackSummaryTool* m_summaryTool{nullptr};
            
    };
}
 
#endif