algorithms/tracking/TrackProjector.h

0001 // Created by Dmitry Romanov
0002 // Subject to the terms in the LICENSE file found in the top-level directory.
0003 //
0004
0005 #pragma once
0006
0007 #include <ActsExamples/EventData/Trajectories.hpp>
0008 #include <edm4eic/TrackSegmentCollection.h>
0009 #include <spdlog/logger.h>
0010 #include <memory>
0011 #include <vector>
0012
0013 #include "ActsGeometryProvider.h"
0014
0015
0016 namespace eicrecon {
0017
0018         /** Extract the particles form fit trajectories.
0019          *
0020          * \ingroup tracking
0021          */
0022         class TrackProjector {
0023
0024         public:
0025
0026             void init(std::shared_ptr<const ActsGeometryProvider> geo_svc, std::shared_ptr<spdlog::logger> logger);
0027
0028             std::unique_ptr<edm4eic::TrackSegmentCollection> execute(std::vector<const ActsExamples::Trajectories*> trajectories);
0029
0030         private:
0031             std::shared_ptr<const ActsGeometryProvider> m_geo_provider;
0032             std::shared_ptr<spdlog::logger> m_log;
0033
0034         };
0035
0036
0037
0038 } // eicrecon