File indexing completed on 2025-07-11 07:53:34
0001
0002
0003
0004 #pragma once
0005
0006 #include <DDRec/CellIDPositionConverter.h>
0007 #include <algorithms/algorithm.h>
0008 #include <algorithms/interfaces/WithPodConfig.h>
0009 #include <edm4eic/MCRecoTrackParticleAssociationCollection.h>
0010 #include <edm4eic/MCRecoTrackerHitAssociationCollection.h>
0011 #include <edm4eic/Measurement2DCollection.h>
0012 #include <edm4eic/TrackCollection.h>
0013 #include <edm4hep/MCParticle.h>
0014 #include <Eigen/Core>
0015 #include <cstddef>
0016 #include <gsl/pointers>
0017 #include <optional>
0018 #include <string>
0019 #include <string_view>
0020 #include <vector>
0021
0022 #include "FarDetectorLinearTrackingConfig.h"
0023
0024 namespace eicrecon {
0025
0026 using FarDetectorLinearTrackingAlgorithm = algorithms::Algorithm<
0027 algorithms::Input<std::vector<edm4eic::Measurement2DCollection>,
0028 std::optional<edm4eic::MCRecoTrackerHitAssociationCollection>>,
0029 algorithms::Output<edm4eic::TrackCollection,
0030 std::optional<edm4eic::MCRecoTrackParticleAssociationCollection>>>;
0031
0032 class FarDetectorLinearTracking : public FarDetectorLinearTrackingAlgorithm,
0033 public WithPodConfig<FarDetectorLinearTrackingConfig> {
0034
0035 public:
0036 FarDetectorLinearTracking(std::string_view name)
0037 : FarDetectorLinearTrackingAlgorithm{
0038 name,
0039 {"inputHitCollections", "inputMCRecoTrackerHitAssociations"},
0040 {"outputTrackCollection", "outputMCRecoTrackAssociations"},
0041 "Fit track segments from hits in the tracker layers"} {}
0042
0043
0044 void init() final;
0045
0046
0047 void process(const Input&, const Output&) const final;
0048
0049 private:
0050 const dd4hep::rec::CellIDPositionConverter* m_cellid_converter{nullptr};
0051 Eigen::VectorXd m_layerWeights;
0052
0053 Eigen::Vector3d m_optimumDirection;
0054
0055 void checkHitCombination(
0056 Eigen::MatrixXd* hitMatrix, edm4eic::TrackCollection* outputTracks,
0057 edm4eic::MCRecoTrackParticleAssociationCollection* assocTracks,
0058 const std::vector<gsl::not_null<const edm4eic::Measurement2DCollection*>>& inputHits,
0059 const std::vector<std::vector<edm4hep::MCParticle>>& assocParts,
0060 const std::vector<std::size_t>& layerHitIndex) const;
0061
0062
0063 bool checkHitPair(const Eigen::Vector3d& hit1, const Eigen::Vector3d& hit2) const;
0064
0065
0066 void ConvertClusters(const edm4eic::Measurement2DCollection& clusters,
0067 const edm4eic::MCRecoTrackerHitAssociationCollection& assoc_hits,
0068 std::vector<std::vector<Eigen::Vector3d>>& pointPositions,
0069 std::vector<std::vector<edm4hep::MCParticle>>& assoc_parts) const;
0070 };
0071
0072 }