algorithms/tracking/TrackProjector.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2022 - 2025 Dmitry Romanov, Dmitry Kalinkin
0003
0004 #pragma once
0005
0006 #include <ActsExamples/EventData/Trajectories.hpp>
0007 #include <algorithms/algorithm.h>
0008 #include <edm4eic/TrackCollection.h>
0009 #include <edm4eic/TrackSegmentCollection.h>
0010 #include <memory>
0011 #include <string>
0012 #include <string_view>
0013 #include <vector>
0014
0015 #include "ActsGeometryProvider.h"
0016 #include "algorithms/interfaces/WithPodConfig.h"
0017
0018 namespace eicrecon {
0019
0020 using TrackProjectorAlgorithm = algorithms::Algorithm<
0021     algorithms::Input<std::vector<ActsExamples::Trajectories>, edm4eic::TrackCollection>,
0022     algorithms::Output<edm4eic::TrackSegmentCollection>>;
0023
0024 class TrackProjector : public TrackProjectorAlgorithm, public WithPodConfig<NoConfig> {
0025 public:
0026   TrackProjector(std::string_view name)
0027       : TrackProjectorAlgorithm{name,
0028                                 {"inputActsTrajectories"},
0029                                 {"outputTrackSegments"},
0030                                 "Exports track states as segments"} {}
0031
0032   void init() final;
0033   void process(const Input&, const Output&) const final;
0034
0035 private:
0036   std::shared_ptr<const ActsGeometryProvider> m_geo_provider;
0037 };
0038
0039 } // namespace eicrecon