MaterialTrackReader.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ACTSMATERIAL_MATERIALTRACKREADER_H
#define ACTSMATERIAL_MATERIALTRACKREADER_H
 
#include "AthenaBaseComps/AthAlgorithm.h"
#include "GaudiKernel/ServiceHandle.h"
#include "StoreGate/ReadHandleKey.h"
 
#include "ActsGeometryInterfaces/ITrackingGeometrySvc.h"
#include "ActsGeometry/RecordedMaterialTrackCollection.h"
#include <ActsPlugins/Root/RootMaterialTrackIo.hpp>
 
#include "TChain.h"
 
#include <mutex>
 
 
namespace ActsTrk {

 
    class MaterialTrackReader : public AthAlgorithm  {
        public:
 
            using AthAlgorithm::AthAlgorithm;
            virtual StatusCode initialize() override;
            virtual StatusCode execute (const EventContext& ctx) override;
            virtual StatusCode finalize() override;
            virtual unsigned cardinality() const override { return 1; }
            virtual ~MaterialTrackReader();
 
        private:
            Gaudi::Property<std::size_t> m_maxEvents{this, "maxEvents", std::numeric_limits<std::size_t>::max()};
            Gaudi::Property<std::size_t> m_skipEvents{this, "skipEvents", 0ul};
            Gaudi::Property<std::size_t>  m_batchSize{this, "batchSize", 1000ul};
            std::size_t m_nTreeEntries{0ul};
            std::size_t m_currEntry{0ul};
            std::size_t m_procEvents{0ul};
            Gaudi::Property<std::vector<std::string>> m_fileNames{this, "FileNames", {}, "The list of input file names"};
            Gaudi::Property<std::string> m_treeName{this, "TreeName", "material-tracks", "The input tree name"};
            Gaudi::Property<bool> m_readCachedSurfaceInformation{this, "ReadCachedSurface", false, "Read surface information for the root file"};
            ActsPlugins::RootMaterialTrackIo m_accessor{{false, m_readCachedSurfaceInformation}};
            std::unique_ptr<TChain> m_inputChain{};
            SG::WriteHandleKey<RecordedMaterialTrackCollection> m_materialTrackCollectionKey {this, "MaterialTrackCollectionKey", "InputMaterialTracks", "Name of the RecordedMaterialTrackCollection"};
        };
}
 
#endif