CaTripletConstructor.h

Go to the documentation of this file.

/* Copyright (C) 2023 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergey Gorbunov [committer] */

 
#pragma once  // include this header only once per compilation unit
 
#include "CaFramework.h"
#include "CaGridEntry.h"
#include "CaMcMatch.h"
#include "CaTriplet.h"
#include "CaVector.h"
#include "CaWindowData.h"
#include "KfActiveLayer.h"
#include "KfFieldRegion.h"
#include "KfSetup.h"
#include "KfSimd.h"
#include "KfTrackKalmanFilter.h"
#include "KfTrackParam.h"
 
namespace cbm::algo::ca
{

  class alignas(kf::VcMemAlign) TripletConstructor {
 
   public:

    TripletConstructor(const ca::Framework& framework, const ca::Parameters<fvec>& pars, WindowData& wData,
                       const fscal mass, const ca::TrackingMode& mode);

    TripletConstructor(const TripletConstructor&) = delete;

    TripletConstructor(TripletConstructor&&) = delete;

    TripletConstructor& operator=(const TripletConstructor&) = delete;

    TripletConstructor& operator=(TripletConstructor&&) = delete;

    ~TripletConstructor() = default;

    void CreateTripletsForHit(Vector<ca::Triplet>& tripletsOut, int istal, int istam, int istar, ca::HitIndex_t ihl);
 
   private:
    typedef std::pair<Vector<TrackParamV>, Vector<ca::HitIndex_t>> Doublet_t;
    typedef std::tuple<Vector<TrackParamV>, Vector<ca::HitIndex_t>, Vector<ca::HitIndex_t>> Triplet_t;
 
    bool InitStations(int istal, int istam, int istar);

    void FindDoublets(kf::TrackKalmanFilter<fvec>& fit);

    void FindTripletHits();

    void FindTriplets();

    void SelectTriplets(Vector<ca::Triplet>& tripletsOut);
 
    void CollectHits(Vector<ca::HitIndex_t>& collectedHits, kf::TrackKalmanFilter<fvec>& fit, const int iSta,
                     const double chi2Cut, const McMatch& mc, const int maxNhits);
 
   private:
    const ca::Framework& fFramework;           
    const Parameters<fvec>& fParameters;       
    const cbm::algo::kf::Setup<fvec>& fSetup;  
    const cbm::algo::kf::Field<fvec>& fField;  
    WindowData& frWData;
    fscal fDefaultMass{constants::phys::MuonMass};  
    ca::TrackingMode fTrackingMode;
 
    bool fIsTargetField{false};  
 
    // Number of stations situated in field region (MVD + STS in CBM)
    int fNfieldStations{0};
 
    int fIstaL{-1};  
    int fIstaM{-1};  
    int fIstaR{-1};  
 
    const kf::ActiveLayer<fvec>* fStaL{nullptr};  
    const kf::ActiveLayer<fvec>* fStaM{nullptr};  
    const kf::ActiveLayer<fvec>* fStaR{nullptr};  
 
    std::array<int, 2> fFldStaTL{};  
    std::array<int, 3> fFldStaLR{};  
 
    ca::HitIndex_t fIhitL;  
    kf::FieldRegion<fvec> fFldL;
 
    // Persistent storage for triplet tracks and hits
    Triplet_t fTripletData;
 
    // Persistent storage for doublet tracks and hits
    Doublet_t fDoubletData;
 
   private:
    static constexpr bool fDebugDublets     = false;  // print debug info for dublets
    static constexpr bool fDebugTriplets    = false;  // print debug info for triplets
    static constexpr bool fDebugCollectHits = false;  // print debug info for CollectHits
  };
 
}  // namespace cbm::algo::ca