CaTrackFinder.h

Go to the documentation of this file.

/* Copyright (C) 2022 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergei Zharko [committer], Maksym Zyzak */
 
 
#pragma once  // include this header only once per compilation unit
 
#include "CaSimd.h"
#include "CaTimesliceHeader.h"
#include "CaTrackExtender.h"
#include "CaTrackFinderWindow.h"
#include "CaTrackFitter.h"
#include "CaVector.h"
#include "KfTrackParam.h"
 
#include <vector>
 
namespace cbm::algo::ca
{
 
  class Track;
  class Triplet;
  class Framework;
 
  class TrackFinder {
   public:
    typedef std::pair<Vector<Track>, Vector<ca::HitIndex_t>> Output_t;
 
    TrackFinder(const ca::Parameters<fvec>& pars, const fscal mass, const ca::TrackingMode& mode,
                TrackingMonitorData& monitorData, int nThreads, double& recoTime);
    ~TrackFinder() = default;
 
    TrackFinder(const TrackFinder&) = delete;
 
    TrackFinder(TrackFinder&&) = delete;
 
    TrackFinder& operator=(const TrackFinder&) = delete;
 
    TrackFinder& operator=(TrackFinder&&) = delete;
 
    Output_t FindTracks(const InputData& input, TimesliceHeader& tsHeader);
 
    const auto& GetRecoTracksContainer(int iThread) const { return fvRecoTracks[iThread]; }
    const auto& GetRecoHitIndicesContainer(int iThread) const { return fvRecoHitIndices[iThread]; }
    TrackingMode GetTrackingMode() const { return fTrackingMode; }
    const std::vector<ca::WindowData>& GetWData() const { return fvWData; }
 
   private:
    // -------------------------------
    // Private methods
    void FindTracksThread(const InputData& input, int iThread, std::pair<fscal, fscal>& windowRange, int& statNwindows,
                          int& statNhitsProcessed);
    //   bool checkTripletMatch(const ca::Triplet& l, const ca::Triplet& r, fscal& dchi2) const;
 
    // -------------------------------
    // Data members
 
    Vector<CaHitTimeInfo> fHitTimeInfo;
 
    const Parameters<fvec>& fParameters;            
    fscal fDefaultMass{constants::phys::MuonMass};  
    ca::TrackingMode fTrackingMode;
 
    TrackingMonitorData& fMonitorData;                     
    std::vector<TrackingMonitorData> fvMonitorDataThread;  
 
    std::vector<ca::WindowData> fvWData;  
 
    int fNofThreads;      
    double& fCaRecoTime;  // time of the track finder + fitter
 
    std::vector<Vector<Track>> fvRecoTracks;           
    std::vector<Vector<HitIndex_t>> fvRecoHitIndices;  
 
    fscal fWindowLength  = 0.;   
    fscal fStatTsStart  = 0.;
    fscal fStatTsEnd    = 0.;
    int fStatNhitsTotal = 0;
  };
 
}  // namespace cbm::algo::ca