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 "CaStation.h"
#include "CaTriplet.h"
#include "CaVector.h"
#include "CaWindowData.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::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 int iMC, const int maxNhits);
 
   private:
    const Parameters<fvec>& fParameters;       
    const cbm::algo::kf::Setup<fvec>& fSetup;  
    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 ca::Station<fvec>* fStaL{nullptr};  
    const ca::Station<fvec>* fStaM{nullptr};  
    const ca::Station<fvec>* fStaR{nullptr};  
 
    const ca::Station<fvec>*
      fFld0Sta[2];  // two stations for approximating the field between the target and the left hit
    const ca::Station<fvec>*
      fFld1Sta[3];  // three stations for approximating the field between the left and the right hit
 
    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