TrackVisualizationTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONTRACKFINDINGTEST_TRACKVISUALIZATIONTOOL_H
#define MUONTRACKFINDINGTEST_TRACKVISUALIZATIONTOOL_H
 
#include <AthenaBaseComps/AthAlgTool.h>
#include <MuonRecToolInterfacesR4/ITrackVisualizationTool.h>
#include <MuonRecToolInterfacesR4/IRootVisualizationService.h>
#include <MuonRecToolInterfacesR4/IRootVisualizationService.h>
#include <MuonRecToolInterfacesR4/ITrackSeedingDiagnosticsTool.h>
#include <ActsGeometryInterfaces/IExtrapolationTool.h>
 
#include "TColor.h"
 
#include <TLegend.h>
 
 
namespace MuonValR4{
    class TrackVisualizationTool : public extends<AthAlgTool, ITrackVisualizationTool> {
        public:
            using base_class::base_class;
            virtual ~TrackVisualizationTool() = default;
            
            virtual StatusCode initialize() override final;
            virtual void displaySeeds(const EventContext& ctx,
                                      const MuonR4::MsTrackSeedContainer& seeds) const override final;
            virtual void displaySeeds(const EventContext& ctx,
                                      const MuonR4::MsTrackSeedContainer& seeds,
                                      PrimitivesVec_t && extPrimitives) const override final;

            virtual void displayTrackSeedObj(const EventContext& ctx,
                                             const MuonR4::MsTrackSeed& seed,
                                             const OptBoundPars_t& parsToExt,
                                             const std::string& objName = "") const override final;
        private:
            using Canvas_t = IRootVisualizationService::ICanvasObject;
            using Edges = Canvas_t::AxisRanges;
            struct PlotLegend {
                PlotLegend(const double xLow, const double yLow,
                           const double xHigh, const double yHigh):
                    legend{std::make_unique<TLegend>(xLow, yLow, xHigh, yHigh)}{}

                void addColor(const int color, const std::string& label);
                void addMarker(const int marker, const std::string& label, const int color = kBlack);
                void fillPrimitives(Canvas_t& canvas);
                std::unique_ptr<TLegend> legend{};
                using LegendEntry_t = std::unordered_map<int, PrimitivePtr_t>;
                LegendEntry_t markers{};
                LegendEntry_t colors{};
            };

            using Location = MuonR4::MsTrackSeed::Location;
            enum class DisplayView{RZ, XY};
            void displaySeeds(const EventContext& ctx,
                              const DisplayView view,
                              const xAOD::MuonSegmentContainer& segments,
                              const MuonR4::MsTrackSeedContainer& seeds,
                              PrimitivesVec_t&& extPrimitives) const;
            void fillTruthSeedPoints(const EventContext& ctx,
                                     const DisplayView view,
                                     PlotLegend& legend, 
                                     Canvas_t& canvas) const;
 
            void displaySeedSegmentsGlobalWithTruth(const EventContext& ctx,
                                                    const DisplayView view,
                                                    const MuonR4::MsTrackSeedContainer& seeds) const;

            static Amg::Vector2D viewVector(const double phi,
                                            const Amg::Vector2D& posOnCylinder,
                                            const DisplayView view);
            ToolHandle<ActsTrk::IExtrapolationTool> m_extrapolationTool{this, "ExtrapolationTool" ,"" };
            ServiceHandle<IRootVisualizationService> m_visualSvc{this, "VisualSvc", "MuonValR4::RootVisualizationService"};
            IRootVisualizationService::ClientToken m_clientToken{};
            SG::ReadHandleKey<xAOD::MuonSegmentContainer> m_truthSegKey{this, "TruthSegkey", "MuonTruthSegments"};
            ToolHandle<MuonR4::ITrackSeedingDiagnosticsTool> m_seedingTool{this, "SeedingTool", ""};
            SG::ReadHandleKey<xAOD::MuonSegmentContainer> m_segmentKey{this, "SegmentContainer", "MuonSegmentsFromR4" };

            ActsTrk::GeoContextReadKey_t m_geoCtxKey{this, "AlignmentKey", "ActsAlignment", "cond handle key"};
            Gaudi::Property<unsigned> m_canvasLimit{this, "CanvasLimit", 5000}; 
            Gaudi::Property<bool> m_saveSinglePDFs{this, "saveSinglePDFs", true};
            Gaudi::Property<bool> m_saveSummaryPDF{this, "saveSummaryPDF",  true};
            Gaudi::Property<std::string> m_subDir{this, "outSubDir", ""};
            Gaudi::Property<std::string> m_canvasPrefix{this, "CanvasPreFix", "MsTrackValid"};
            mutable std::atomic<bool> m_plotsDone{false};
            mutable std::atomic<unsigned> m_objCounter{0};
    };
 
    //** @brief Visualizing seeds and corresponding segments with different colors */
    int seedColorIdx(std::size_t iSeed);
}
 
#endif