algorithms/tracking/IterativeVertexFinder.h

0001 // Created by Joe Osborn
0002 // Subject to the terms in the LICENSE file found in the top-level directory.
0003 //
0004
0005 #pragma once
0006
0007 #include <Acts/Geometry/GeometryContext.hpp>
0008 #include <Acts/MagneticField/MagneticFieldContext.hpp>
0009 #include <edm4eic/VertexCollection.h>
0010 #include <edm4eic/ReconstructedParticle.h>
0011 #include <spdlog/logger.h>
0012 #include <memory>
0013 #include <vector>
0014
0015 #include "ActsExamples/EventData/Trajectories.hpp"
0016 #include "ActsGeometryProvider.h"
0017 #include "DD4hepBField.h"
0018 #include "IterativeVertexFinderConfig.h"
0019 #include "algorithms/interfaces/WithPodConfig.h"
0020
0021 namespace eicrecon {
0022 class IterativeVertexFinder
0023     : public eicrecon::WithPodConfig<eicrecon::IterativeVertexFinderConfig> {
0024 public:
0025   void init(std::shared_ptr<const ActsGeometryProvider> geo_svc,
0026             std::shared_ptr<spdlog::logger> log);
0027   std::unique_ptr<edm4eic::VertexCollection>
0028   produce(std::vector<const ActsExamples::Trajectories*> trajectories, const edm4eic::ReconstructedParticleCollection* reconParticles);
0029
0030 private:
0031   std::shared_ptr<spdlog::logger> m_log;
0032   std::shared_ptr<const ActsGeometryProvider> m_geoSvc;
0033
0034   std::shared_ptr<const eicrecon::BField::DD4hepBField> m_BField = nullptr;
0035   Acts::GeometryContext m_geoctx;
0036   Acts::MagneticFieldContext m_fieldctx;
0037   IterativeVertexFinderConfig m_cfg;
0038 };
0039 } // namespace eicrecon