algorithms/fardetectors/FarDetectorLinearTracking.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2024 - 2025, Simon Gardner
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/MCRecoTrackParticleLinkCollection.h>
0011 #include <edm4eic/MCRecoTrackerHitAssociationCollection.h>
0012 #include <edm4eic/MCRecoTrackerHitLinkCollection.h>
0013 #include <edm4eic/Measurement2DCollection.h>
0014 #include <edm4eic/TrackCollection.h>
0015 #include <edm4hep/MCParticle.h>
0016 #include <Eigen/Core>
0017 #include <cstddef>
0018 #include <gsl/pointers>
0019 #include <optional>
0020 #include <string>
0021 #include <string_view>
0022 #include <vector>
0023
0024 #include "FarDetectorLinearTrackingConfig.h"
0025 #include "algorithms/interfaces/LinkTruthUtils.h"
0026
0027 namespace eicrecon {
0028
0029 using FarDetectorLinearTrackingAlgorithm = algorithms::Algorithm<
0030     algorithms::Input<std::vector<edm4eic::Measurement2DCollection>,
0031                       std::optional<edm4eic::MCRecoTrackerHitLinkCollection>,
0032                       std::optional<edm4eic::MCRecoTrackerHitAssociationCollection>>,
0033     algorithms::Output<edm4eic::TrackCollection,
0034                        std::optional<edm4eic::MCRecoTrackParticleLinkCollection>,
0035                        std::optional<edm4eic::MCRecoTrackParticleAssociationCollection>>>;
0036
0037 class FarDetectorLinearTracking : public FarDetectorLinearTrackingAlgorithm,
0038                                   public WithPodConfig<FarDetectorLinearTrackingConfig> {
0039
0040 public:
0041   FarDetectorLinearTracking(std::string_view name)
0042       : FarDetectorLinearTrackingAlgorithm{
0043             name,
0044             {"inputHitCollections", "inputMCRecoTrackerHitLinks",
0045              "inputMCRecoTrackerHitAssociations"},
0046             {"outputTrackCollection", "outputMCRecoTrackLinks", "outputMCRecoTrackAssociations"},
0047             "Fit track segments from hits in the tracker layers"} {}
0048
0049   /** One time initialization **/
0050   void init() final;
0051
0052   /** Event by event processing **/
0053   void process(const Input&, const Output&) const final;
0054
0055 private:
0056   const dd4hep::rec::CellIDPositionConverter* m_cellid_converter{nullptr};
0057   Eigen::VectorXd m_layerWeights;
0058
0059   Eigen::Vector3d m_optimumDirection;
0060
0061   void checkHitCombination(
0062       Eigen::MatrixXd* hitMatrix, edm4eic::TrackCollection* outputTracks,
0063       edm4eic::MCRecoTrackParticleLinkCollection* trackLinks,
0064       edm4eic::MCRecoTrackParticleAssociationCollection* assocTracks,
0065       const std::vector<gsl::not_null<const edm4eic::Measurement2DCollection*>>& inputHits,
0066       const std::vector<std::vector<edm4hep::MCParticle>>& assocParts,
0067       const std::vector<std::size_t>& layerHitIndex, bool do_assoc) const;
0068
0069   /** Check if the last two hits are within a certain angle of the optimum direction **/
0070   bool checkHitPair(const Eigen::Vector3d& hit1, const Eigen::Vector3d& hit2) const;
0071
0072   /** Convert 2D clusters to 3D coordinates and match associated particle **/
0073   void ConvertClusters(
0074       const edm4eic::Measurement2DCollection& clusters,
0075       const truth::EventLinkNavigator<edm4eic::MCRecoTrackerHitLinkCollection>& link_nav,
0076       std::vector<std::vector<Eigen::Vector3d>>& pointPositions,
0077       std::vector<std::vector<edm4hep::MCParticle>>& assoc_parts) const;
0078 };
0079
0080 } // namespace eicrecon