algorithms/tracking/ActsToTracks.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2024 Dmitry Kalinkin
0003
0004 #pragma once
0005
0006 #include <ActsExamples/EventData/Trajectories.hpp>
0007 #include <algorithms/algorithm.h>
0008 #include <edm4eic/MCRecoTrackParticleAssociationCollection.h>
0009 #include <edm4eic/MCRecoTrackerHitAssociationCollection.h>
0010 #include <edm4eic/Measurement2DCollection.h>
0011 #include <edm4eic/TrackCollection.h>
0012 #include <edm4eic/TrackParametersCollection.h>
0013 #include <edm4eic/TrajectoryCollection.h>
0014 #include <optional>
0015 #include <string>
0016 #include <string_view>
0017 #include <vector>
0018
0019 #include "algorithms/interfaces/WithPodConfig.h"
0020
0021 namespace eicrecon {
0022
0023 using ActsToTracksAlgorithm = algorithms::Algorithm<
0024     algorithms::Input<edm4eic::Measurement2DCollection, std::vector<ActsExamples::Trajectories>,
0025                       std::optional<edm4eic::MCRecoTrackerHitAssociationCollection>>,
0026     algorithms::Output<edm4eic::TrajectoryCollection, edm4eic::TrackParametersCollection,
0027                        edm4eic::TrackCollection,
0028                        std::optional<edm4eic::MCRecoTrackParticleAssociationCollection>>>;
0029
0030 class ActsToTracks : public ActsToTracksAlgorithm, public WithPodConfig<NoConfig> {
0031 public:
0032   ActsToTracks(std::string_view name)
0033       : ActsToTracksAlgorithm{name,
0034                               {
0035                                   "inputMeasurements",
0036                                   "inputActsTrajectories",
0037                                   "inputRawTrackerHitAssociations",
0038                               },
0039                               {
0040                                   "outputTrajectories",
0041                                   "outputTrackParameters",
0042                                   "outputTracks",
0043                                   "outputTrackAssociations",
0044                               },
0045                               "Converts ACTS trajectories to EDM4eic"} {};
0046
0047   void init() final;
0048   void process(const Input&, const Output&) const final;
0049 };
0050
0051 } // namespace eicrecon