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/AthReentrantAlgorithm.h"
#include "GaudiKernel/ServiceHandle.h"
#include "StoreGate/ReadHandleKey.h"
 
#include "ActsGeometryInterfaces/ITrackingGeometrySvc.h"
#include "ActsMaterial/RecordedMaterialTrackCollection.h"
 
#include <ActsPlugins/Root/RootMaterialTrackIo.hpp>
 
#include "TTree.h"
#include "TFile.h"
 
#include <mutex>
 
 
namespace ActsTrk {

 
    class MaterialTrackWriter : public AthReentrantAlgorithm  {
        public:
            MaterialTrackWriter(const std::string &name, ISvcLocator *pSvcLocator);
            virtual StatusCode initialize() override;
            virtual StatusCode execute (const EventContext& ctx) const override;
            virtual StatusCode finalize() override;
            virtual ~MaterialTrackWriter();
 
        private:
            mutable std::mutex m_writeMutex;
            Gaudi::Property<std::string> m_fileName{this, "FileName", "MaterialTracks.root", "The output file name"};
            Gaudi::Property<std::string> m_treeName{this, "TreeName", "material-tracks", "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"};

            ServiceHandle<ActsTrk::ITrackingGeometrySvc> m_trackingGeometrySvc{this, "TrackingGeometrySvc","ActsTrackingGeometrySvc", "The ACTS geometry service for the geometry context"};
            mutable ActsPlugins::RootMaterialTrackIo m_accessor ATLAS_THREAD_SAFE;
            TFile* m_outputFile{};
            mutable TTree* m_outputTree ATLAS_THREAD_SAFE = {};
 
        };
}
 
#endif