MaterialTrackWriter.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ACTSMATERIAL_MATERIALTRACKWRITER_H
#define ACTSMATERIAL_MATERIALTRACKWRITER_H
 
#include "AthenaBaseComps/AthHistogramAlgorithm.h"
 
#include "StoreGate/ReadHandleKey.h"
 
#include "ActsEvent/ContextUtility.h"
#include "ActsGeometry/RecordedMaterialTrackCollection.h"
 
#include <ActsPlugins/Root/RootMaterialTrackIo.hpp>
 
#include "TTree.h"
#include "TFile.h"
 
#include <mutex>
 
 
namespace ActsTrk {

 
    class MaterialTrackWriter : public AthHistogramAlgorithm  {
        public:
            using AthHistogramAlgorithm::AthHistogramAlgorithm;
            virtual StatusCode initialize() override;
            virtual StatusCode execute (const EventContext& ctx) override;           
            virtual ~MaterialTrackWriter();
 
        private:
            mutable std::mutex m_writeMutex;
            Gaudi::Property<std::string> m_outStream{this, "OutStream", "ACTSMATERIALWRITER", "The output file name"};
            Gaudi::Property<std::string> m_treeName{this, "TreeName", "materialTracks", "The output tree name"};
            Gaudi::Property<bool> m_storeSurface{this, "StoreSurface", false, "Write the surface to which the material step correpond"};
            Gaudi::Property<bool> m_storeVolume{this, "StoreVolume", false, "Write the volume to which the material step correpond"};
            Gaudi::Property<bool> m_prePostStep{this, "PrePostStep", false, "Write out pre and post step (for G4), otherwise central step position"};
            Gaudi::Property<bool> m_recalculateTotals{this, "RecalculateTotals", false, "Re-calculate total values from individual steps (for cross-checks)"};

            SG::ReadHandleKey<RecordedMaterialTrackCollection> m_materialTrackCollectionKey {this, "MaterialTrackCollectionKey", "OutputMaterialTracks", "Name of the RecordedMaterialTrackCollection"};
 
            Gaudi::Property<bool> m_useTrackingGeo{this, "useTrackingGeometry", false, 
                                                    "Use the tracking geometry to retrieve the geometry context"};
            
            
            ContextUtility m_ctxProvider{this};
            using Config_t = ActsPlugins::RootMaterialTrackIo::Config;
            ActsPlugins::RootMaterialTrackIo m_accessor{Config_t{}};
           
            TTree* m_outputTree{};
 
        };
}
 
#endif