CaTripletConstructorSW.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 "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) TripletConstructorSW {
 
   public:

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

    TripletConstructorSW(const TripletConstructorSW&) = delete;

    TripletConstructorSW(TripletConstructorSW&&) = delete;

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

    TripletConstructorSW& operator=(TripletConstructorSW&&) = delete;

    ~TripletConstructorSW() = default;

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

    void SuppressDoubletClones();

    void FindTripletHits();

    void FitTriplets(Vector<ca::Triplet>& tripletsOut);
 
    void CollectHits(Vector<ca::HitIndex_t>& collectedHits, const SearchWindowMap::SearchWindow& area2D, fscal time,
                     fscal timeRange, const int iSta, const fscal areaExtension, const int maxNhits, const McMatch& mc);
 
    int FindClosestHitWithMc(const SearchWindowMap::SearchWindow& area2D, const int iSta, const McMatch& mc);
 
 
   private:
    const ca::Framework& fFramework;           
    const Parameters<fvec>& fParameters;       
    const cbm::algo::kf::Setup<fvec>& fSetup;  
    const cbm::algo::kf::Field<fvec>& fField;  
    const SearchWindowMapContainer& fSwMaps;   
    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};  
 
    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
    Vector<ca::HitIndex_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