algorithms/tracking/TrackParamTruthInit.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 <edm4eic/TrackParametersCollection.h>
0008 #include <edm4hep/MCParticleCollection.h>
0009 #include <spdlog/logger.h>
0010 #include <memory>
0011 #include <random>
0012
0013 #include "ActsGeometryProvider.h"
0014 #include "TrackParamTruthInitConfig.h"
0015 #include "algorithms/interfaces/ParticleSvc.h"
0016 #include "algorithms/interfaces/WithPodConfig.h"
0017
0018 namespace eicrecon {
0019 class TrackParamTruthInit : public WithPodConfig<TrackParamTruthInitConfig> {
0020
0021 public:
0022   void init(std::shared_ptr<const ActsGeometryProvider> geo_svc,
0023             const std::shared_ptr<spdlog::logger> logger);
0024
0025   std::unique_ptr<edm4eic::TrackParametersCollection>
0026   produce(const edm4hep::MCParticleCollection* mcparticles);
0027
0028 private:
0029   std::shared_ptr<spdlog::logger> m_log;
0030   std::shared_ptr<const ActsGeometryProvider> m_geoSvc;
0031
0032   const algorithms::ParticleSvc& m_particleSvc = algorithms::ParticleSvc::instance();
0033
0034   std::default_random_engine generator; // TODO: need something more appropriate here
0035   std::uniform_int_distribution<int> m_uniformIntDist{-1, 1}; // defaults to min=-1, max=1
0036   std::normal_distribution<double> m_normDist;
0037 };
0038 } // namespace eicrecon