algorithms/tracking/TracksToParticles.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2022 - 2024, Sylvester Joosten, Wouter Deconinck, Dmitry Romanov, Christopher Dilks, Dmitry Kalinkin
0003
0004 #pragma once
0005
0006 #include <algorithms/algorithm.h>
0007 #include <edm4eic/MCRecoParticleAssociationCollection.h>
0008 #include <edm4eic/EDM4eicVersion.h>
0009 #include <edm4eic/MCRecoTrackParticleAssociationCollection.h>
0010 #include <edm4eic/ReconstructedParticleCollection.h>
0011 #include <edm4eic/TrackCollection.h>
0012 #include <optional>
0013 #include <string>
0014 #include <string_view>
0015
0016 #include "algorithms/interfaces/WithPodConfig.h"
0017
0018 #if EDM4EIC_BUILD_VERSION >= EDM4EIC_VERSION(8, 7, 0)
0019 #include <edm4eic/MCRecoParticleLinkCollection.h>
0020 #endif
0021
0022 namespace eicrecon {
0023
0024 using TracksToParticlesAlgorithm = algorithms::Algorithm<
0025     algorithms::Input<edm4eic::TrackCollection,
0026                       std::optional<edm4eic::MCRecoTrackParticleAssociationCollection>>,
0027     algorithms::Output<edm4eic::ReconstructedParticleCollection,
0028 #if EDM4EIC_BUILD_VERSION >= EDM4EIC_VERSION(8, 7, 0)
0029                        std::optional<edm4eic::MCRecoParticleLinkCollection>,
0030 #endif
0031                        std::optional<edm4eic::MCRecoParticleAssociationCollection>>>;
0032
0033 class TracksToParticles : public TracksToParticlesAlgorithm, public WithPodConfig<NoConfig> {
0034 public:
0035   TracksToParticles(std::string_view name)
0036       : TracksToParticlesAlgorithm{name,
0037                                    {"inputTracksCollection", "inputTrackAssociationsCollection"},
0038                                    {"outputReconstructedParticlesCollection",
0039 #if EDM4EIC_BUILD_VERSION >= EDM4EIC_VERSION(8, 7, 0)
0040                                     "outputLinks",
0041 #endif
0042                                     "outputAssociationsCollection"},
0043                                    "Converts track to particles with associations"} {};
0044
0045   void init() final;
0046   void process(const Input&, const Output&) const final;
0047 };
0048
0049 } // namespace eicrecon