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 <spdlog/logger.h>
0011 #include <memory>
0012 #include <vector>
0013
0014 #include "ActsExamples/EventData/Trajectories.hpp"
0015 #include "ActsGeometryProvider.h"
0016 #include "DD4hepBField.h"
0017 #include "IterativeVertexFinderConfig.h"
0018 #include "algorithms/interfaces/WithPodConfig.h"
0019
0020 namespace eicrecon {
0021 class IterativeVertexFinder
0022     : public eicrecon::WithPodConfig<eicrecon::IterativeVertexFinderConfig> {
0023 public:
0024   void init(std::shared_ptr<const ActsGeometryProvider> geo_svc,
0025             std::shared_ptr<spdlog::logger> log);
0026   std::unique_ptr<edm4eic::VertexCollection>
0027   produce(std::vector<const ActsExamples::Trajectories*> trajectories);
0028
0029 private:
0030   std::shared_ptr<spdlog::logger> m_log;
0031   std::shared_ptr<const ActsGeometryProvider> m_geoSvc;
0032
0033   std::shared_ptr<const eicrecon::BField::DD4hepBField> m_BField = nullptr;
0034   Acts::GeometryContext m_geoctx;
0035   Acts::MagneticFieldContext m_fieldctx;
0036   IterativeVertexFinderConfig m_cfg;
0037 };
0038 } // namespace eicrecon