RootVisualizationService.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONVISUALIZATIONHELPERSR4_ROOVISUALIZATIONSERVICE_H
#define MUONVISUALIZATIONHELPERSR4_ROOVISUALIZATIONSERVICE_H
#include <MuonRecToolInterfacesR4/IRootVisualizationService.h>
#include <GaudiKernel/EventContext.h>
#include <AthenaBaseComps/AthService.h>
 
#include "DrawCanvasObject.h"
 
#include <TFile.h>
 
#include <mutex>
 
namespace MuonValR4{
    class RootVisualizationService: public extends<AthService, IRootVisualizationService>{
        public:
            using base_class::base_class;
 
 
            virtual StatusCode registerClient(const ClientToken& token) override final;
 
            virtual std::shared_ptr<ICanvasObject> 
                    prepareCanvas(const EventContext& ctx, const ClientToken& token,
                                  const std::string& canvasName) override final;
            virtual StatusCode finalize() override final;
            virtual StatusCode initialize() override final;
 
            using PlotPtr_t = std::shared_ptr<detail::DrawCanvasObject>;
            using PlotVec_t = std::vector<PlotPtr_t>;
 
        private:
            void paintObjects(const ClientToken& token,
                              PlotVec_t&& toDraw);

            Gaudi::Property<double> m_canvasExtraScale{this, "CanvasExtraScale" , 1.5};
            Gaudi::Property<bool> m_quadCanvas{this, "QuadraticCanas", true};
            Gaudi::Property<unsigned> m_canvasWidth{this, "CanvasWidth", 800};
            Gaudi::Property<unsigned> m_canvasHeight{this, "CanvasHeight", 600};
            Gaudi::Property<std::string> m_outDir{this, "outputDir" , "./Displays/"};
            Gaudi::Property<std::string> m_outRootFileName{this, "outputROOTFile", "AllDisplays.root"};
  
            struct PlotsPerClient{
                PlotVec_t toDraw{};
                bool elementsDrawn{false};
            };
 
            using StorageMap_t = std::map<ClientToken, PlotsPerClient>;
            StorageMap_t m_storage{};
            std::unique_ptr<TFile> m_outFile{};
 
            std::mutex m_storageMutex{};
            std::mutex m_canvasMutex{};
 
 
    };
}
#endif