KfVertexFitter.h

Go to the documentation of this file.

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

 
#pragma once
 
#include "KfDefs.h"
#include "KfSetup.h"
#include "KfTrackParam.h"
#include "KfVertex.h"
 
#include <tuple>
#include <vector>
 
namespace cbm::algo::kf
{
  template<DoFitTime FlagFitTime = DoFitTime::Y>
  class VertexFitter {
   public:
    VertexFitter() = default;

    VertexFitter(std::shared_ptr<const Setup<double>> kfSetup) : fKfSetup(kfSetup) {}

    ~VertexFitter() = default;

    void SetKfSetup(std::shared_ptr<const Setup<double>> kfSetup) { fKfSetup = kfSetup; }

    void SetConstrainToTargetZ() { fConstrainToTargetZ = true; }

    void SetNoConstrainToTargetZ() { fConstrainToTargetZ = false; }

    void SetConstrainToBeamXY(double x, double y, double sigmaX, double sigmaY)
    {
      fConstrainToBeamXY = true;
      fBeamX             = x;
      fBeamY             = y;
      fBeamSigmaX        = sigmaX;
      fBeamSigmaY        = sigmaY;
    }

    void SetNoConstrainToBeamXY() { fConstrainToBeamXY = false; }

    void SetRmsCut(double c) { fRmsCut = c; }

    void SetNoRmsCut() { fRmsCut = -1.; }

    void SetNofIterations(int n) { fNofIterations = n; }

 
    bool IsConstrainToTargetZ() const { return fConstrainToTargetZ; }
    bool IsConstrainToBeamXY() const { return fConstrainToBeamXY; }
    double GetRmsCut() const { return fRmsCut; }
    int GetNofIterations() const { return fNofIterations; }
    std::shared_ptr<const Setup<double>> GetKfSetup() const { return fKfSetup; }

    template<typename T>
    std::tuple<VertexD, std::vector<char>> FitVertex(const std::vector<const TrackParam<T>*>& tracks,
                                                     const std::vector<const TrackParam<T>*>& linearizations,
                                                     const VertexD& vtxGuess) const;
    VertexD GetGuessFromTheTarget() const;
 
   private:
    void dummy();  
 
   private:
    std::shared_ptr<const Setup<double>> fKfSetup{};  
 
    bool fConstrainToTargetZ{false};  
    bool fConstrainToBeamXY{false};   
 
    double fBeamX{0.};        
    double fBeamY{0.};        
    double fBeamSigmaX{0.1};  
    double fBeamSigmaY{0.1};  
    double fRmsCut{5.};       
    int fNofIterations{3};    
  };
 
}  // namespace cbm::algo::kf