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