TrackFitUtils.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file TrackFitUtils.h

#ifndef TRACKBASE_TRACKFITUTILS_H
#define TRACKBASE_TRACKFITUTILS_H

#include "TrkrDefs.h"

#include <Acts/Definitions/Algebra.hpp>

#include <tuple>
#include <utility>
#include <vector>

class ActsGeometry;
class TrkrClusterContainer;

class TrackFitUtils
{
 public:
  using position_t = std::pair<double, double>;
  using position_vector_t = std::vector<position_t>;

  using circle_fit_output_t = std::tuple<double, double, double>;

  static circle_fit_output_t circle_fit_by_taubin(const position_vector_t&);

  static circle_fit_output_t circle_fit_by_taubin(const std::vector<Acts::Vector3>&);

  using line_fit_output_t = std::tuple<double, double>;


  static line_fit_output_t line_fit(const position_vector_t&);

  static line_fit_output_t line_fit(const std::vector<Acts::Vector3>&);

  using circle_circle_intersection_output_t = std::tuple<double, double, double, double>;

  static circle_circle_intersection_output_t circle_circle_intersection(double r1, double r2, double x2, double y2);

  static unsigned int addClusters(std::vector<float>& fitpars,
                                  double dca_cut,
                                  ActsGeometry* _tGeometry,
                                  TrkrClusterContainer* _cluster_map,
                                  std::vector<Acts::Vector3>& global_vec,
                                  std::vector<TrkrDefs::cluskey>& cluskey_vec,
                                  unsigned int startLayer,
                                  unsigned int endLayer);

  static unsigned int addClustersOnLine(TrackFitUtils::line_fit_output_t& fitpars,
                                        const bool& isXY,
                                        const double& dca_cut,
                                        ActsGeometry* tGeometry,
                                        TrkrClusterContainer* clusterContainer,
                                        std::vector<Acts::Vector3>& global_vec,
                                        std::vector<TrkrDefs::cluskey>& cluskey_vec,
                                        const unsigned int& startLayer,
                                        const unsigned int& endLayer);

  static std::pair<Acts::Vector3, Acts::Vector3> get_helix_tangent(const std::vector<float>& fitpars, Acts::Vector3& global);

  static Acts::Vector3 get_helix_pca(std::vector<float>& fitpars, Acts::Vector3 global);

  static Acts::Vector2 get_circle_point_pca(float radius, float x0, float y0, Acts::Vector3 global);

  static std::vector<float> fitClusters(std::vector<Acts::Vector3>& global_vec,
                                        std::vector<TrkrDefs::cluskey> cluskey_vec);
  static void getTrackletClusters(ActsGeometry* _tGeometry,
                                  TrkrClusterContainer* _cluster_map,
                                  std::vector<Acts::Vector3>& global_vec,
                                  std::vector<TrkrDefs::cluskey>& cluskey_vec);

  static Acts::Vector3 getPCALinePoint(Acts::Vector3 global, Acts::Vector3 tangent, Acts::Vector3 posref);

  static std::vector<double> getLineClusterResiduals(position_vector_t& rz_pts, float slope, float intercept);
  static std::vector<double> getCircleClusterResiduals(position_vector_t& xy_pts, float R, float X0, float Y0);
};

#endif