CbmDcaVertexFinder.h

Go to the documentation of this file.

/* Copyright (C) 2023 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Dario Ramirez [committer] */
 
#ifndef CBMDCAVERTEXFINDER_H
#define CBMDCAVERTEXFINDER_H
 
#include "TH2.h"
#include "TMatrixFSym.h"
#include "TVector3.h"
 
#include <optional>
class CbmVertex;
class CbmGlobalTrack;

class CbmDcaVertexFinder {
 public:
  struct Qa {
    std::shared_ptr<TH2> pca_y_vs_x;
    std::shared_ptr<TH2> pca_x_vs_z;
    std::shared_ptr<TH2> pca_y_vs_z;
  };

  struct PCA {
    TVector3 point;
    double d_trk;
  };
 
  CbmDcaVertexFinder();

  CbmDcaVertexFinder(double max_dca);

  CbmDcaVertexFinder(const std::vector<CbmGlobalTrack*> tracks);

  CbmDcaVertexFinder(const std::vector<CbmGlobalTrack*> tracks, const double max_dca);

  void SetTracks(const std::vector<CbmGlobalTrack*> tracks);
 
  ~CbmDcaVertexFinder() = default;

  std::optional<CbmDcaVertexFinder::PCA> FindPca(CbmGlobalTrack* trk_i, CbmGlobalTrack* trk_j);

  std::optional<CbmVertex> FindVertex();
 
  void EnableQa(Qa qa) { fQA = std::make_optional<Qa>(qa); }
 
 private:
  const double fMaxDca;
  std::vector<CbmGlobalTrack*> fInputTracks;
  double fNbPairs;
  std::optional<Qa> fQA;
 
  TMatrixFSym fCovMatrix;
};
 
#endif