CaTrackExtender.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 "CaBranch.h"
#include "CaHit.h"
#include "CaParameters.h"
#include "CaSimd.h"
#include "CaVector.h"
#include "CaWindowData.h"
#include "KfTrackKalmanFilter.h"
#include "KfTrackParam.h"
 
namespace cbm::algo::ca
{
 
  class Track;
  class Framework;
  class InputData;
 
  class TrackExtender {
 
   public:
    TrackExtender(const ca::Parameters<fvec>& pars, const fscal mass);
 
    ~TrackExtender();
 
    TrackExtender(const TrackExtender&) = default;
 
    TrackExtender(TrackExtender&&) = default;
 
    TrackExtender& operator=(const TrackExtender&) = delete;
 
    TrackExtender& operator=(TrackExtender&&) = delete;
 
 
    fscal ExtendBranch(ca::Branch& t, WindowData& wData);
 
   private:
 
    void FindMoreHits(ca::Branch& t, TrackParamV& T, const kf::FitDirection direction, const fvec qp0);
 
    void FitBranch(const ca::Branch& t, TrackParamV& T, const kf::FitDirection direction, const fvec qp0,
                   const bool initParams = true);
 
 
    void FitBranchFast(const ca::Branch& t, TrackParamV& T, const kf::FitDirection direction, const fvec qp0,
                       const bool initParams = true);
 
   private:
 
    const Parameters<fvec>& fParameters;  
    const cbm::algo::kf::Setup<fvec>& fSetup;
    WindowData* frWData;
    fscal fDefaultMass{constants::phys::MuonMass};  
  };
 
}  // namespace cbm::algo::ca