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 <Acts/EventData/VectorMultiTrajectory.hpp>
0007 #include <Acts/EventData/VectorTrackContainer.hpp>
0008 #include <algorithms/algorithm.h>
0009 #include <edm4eic/MCRecoTrackParticleAssociationCollection.h>
0010 #include <edm4eic/EDM4eicVersion.h>
0011 #include <edm4eic/MCRecoTrackerHitAssociationCollection.h>
0012 #include <edm4eic/Measurement2DCollection.h>
0013 #include <edm4eic/TrackCollection.h>
0014 #include <edm4eic/TrackParametersCollection.h>
0015 #include <edm4eic/TrackSeedCollection.h>
0016 #include <edm4eic/TrajectoryCollection.h>
0017 #include <optional>
0018 #include <string>
0019 #include <string_view>
0020
0021 #include "algorithms/interfaces/WithPodConfig.h"
0022
0023 #if EDM4EIC_BUILD_VERSION >= EDM4EIC_VERSION(8, 7, 0)
0024 #include <edm4eic/MCRecoTrackParticleLinkCollection.h>
0025 #endif
0026
0027 namespace eicrecon {
0028
0029 using ActsToTracksAlgorithm = algorithms::Algorithm<
0030     algorithms::Input<edm4eic::Measurement2DCollection, edm4eic::TrackSeedCollection,
0031                       Acts::ConstVectorMultiTrajectory, Acts::ConstVectorTrackContainer,
0032                       std::optional<edm4eic::MCRecoTrackerHitAssociationCollection>>,
0033     algorithms::Output<edm4eic::TrajectoryCollection, edm4eic::TrackParametersCollection,
0034                        edm4eic::TrackCollection,
0035 #if EDM4EIC_BUILD_VERSION >= EDM4EIC_VERSION(8, 7, 0)
0036                        std::optional<edm4eic::MCRecoTrackParticleLinkCollection>,
0037 #endif
0038                        std::optional<edm4eic::MCRecoTrackParticleAssociationCollection>>>;
0039
0040 class ActsToTracks : public ActsToTracksAlgorithm, public WithPodConfig<NoConfig> {
0041 public:
0042   ActsToTracks(std::string_view name)
0043       : ActsToTracksAlgorithm{name,
0044                               {
0045                                   "inputMeasurements",
0046                                   "inputTrackSeeds",
0047                                   "inputActsTrackStates",
0048                                   "inputActsTracks",
0049                                   "inputRawTrackerHitAssociations",
0050                               },
0051                               {
0052                                   "outputTrajectories",
0053                                   "outputTrackParameters",
0054                                   "outputTracks",
0055 #if EDM4EIC_BUILD_VERSION >= EDM4EIC_VERSION(8, 7, 0)
0056                                   "outputTrackLinks",
0057 #endif
0058                                   "outputTrackAssociations",
0059                               },
0060                               "Converts ACTS tracks to EDM4eic"} {};
0061
0062   void init() final;
0063   void process(const Input&, const Output&) const final;
0064 };
0065
0066 } // namespace eicrecon