Back to home page

EIC code displayed by LXR

 
 

    


File indexing completed on 2025-07-11 07:53:34

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/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   /** One time initialization **/
0044   void init() final;
0045 
0046   /** Event by event processing **/
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   /** Check if the last two hits are within a certain angle of the optimum direction **/
0063   bool checkHitPair(const Eigen::Vector3d& hit1, const Eigen::Vector3d& hit2) const;
0064 
0065   /** Convert 2D clusters to 3D coordinates and match associated particle **/
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 } // namespace eicrecon