df/d45/ALICEKF_8h_source.html

#ifndef ALICEKF_H

#define ALICEKF_H


#include "GPUTPCTrackParam.h"

#include <trackbase/ClusterErrorPara.h>

#include <trackbase_historic/TrackSeed_v1.h>

#include <trackbase/TrkrDefs.h>

#include <trackbase/TrkrClusterContainer.h>

#include <trackbase/TrkrCluster.h>

#include <phfield/PHField.h>

#include <phfield/PHFieldUtility.h>

#include <phfield/PHFieldConfigv1.h>


#include <Acts/Definitions/Algebra.hpp>


#include <Eigen/Core>

#include <Eigen/Dense>


#include <vector>

#include <string>

#include <utility>


using PositionMap = std::map<TrkrDefs::cluskey, Acts::Vector3>;

using TrackSeedAliceSeedMap = std::pair<std::vector<TrackSeed_v1>, std::vector<Eigen::Matrix<double,6,6>>>;


class ALICEKF

{

  public:

  ALICEKF(PHCompositeNode *topNode,

          TrkrClusterContainer* cmap,

          PHField *B,

          double fieldDir,

          unsigned int min_clusters,

          double max_sin_phi,

          int verbosity)

  {

    if(!topNode) std::cout << "no topnode, too bad..." << std::endl;

    _B = B;

    //    _B = PHFieldUtility::GetFieldMapNode(nullptr,topNode);

    _cluster_map = cmap;

    _fieldDir = fieldDir;

    _max_sin_phi = max_sin_phi;

    _v = verbosity;

    _min_clusters_per_track = min_clusters;

    _ClusErrPara = new ClusterErrorPara();

  }


  ~ALICEKF() {delete _ClusErrPara;}


  explicit ALICEKF(const ALICEKF&) = delete;

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


  TrackSeedAliceSeedMap ALICEKalmanFilter(const std::vector<std::vector<TrkrDefs::cluskey>>& chains, bool use_nhits_limit, const PositionMap& globalPositions, std::vector<float>& trackChi2) const;

  bool covIsPosDef(Eigen::Matrix<double,6,6>& cov) const;

  void repairCovariance(Eigen::Matrix<double,6,6>& cov) const;

  bool checknan(double val, const std::string &msg, int num) const;

  double get_Bz(double x, double y, double z) const;

  void useConstBField(bool opt) {_use_const_field = opt;}

  void setConstBField(float b) {_const_field = b; }

  void useFixedClusterError(bool opt) {_use_fixed_clus_error = opt;}

  void setFixedClusterError(int i,double val) {_fixed_clus_error.at(i)=val;}

  double getClusterError(TrkrCluster* c, TrkrDefs::cluskey key, Acts::Vector3 global, int i, int j) const;

  std::vector<double> GetCircleClusterResiduals(const std::vector<std::pair<double,double>>& pts, double R, double X0, double Y0) const;

  std::vector<double> GetLineClusterResiduals(const std::vector<std::pair<double,double>>& pts, double A, double B) const;

  double get_Bzconst() const { return _Bzconst; }


  ClusterErrorPara *_ClusErrPara;

  private:

  PHField* _B = nullptr;

  size_t _min_clusters_per_track = 20;

  TrkrClusterContainer* _cluster_map = nullptr;

  int Verbosity() const

  { return _v; }


  int _v = 0;

  double _Bzconst = 10*0.000299792458f;

  double _fieldDir = -1;

  double _max_sin_phi = 1.;

  bool _use_const_field = false;

  float _const_field = 1.4;

  bool _use_fixed_clus_error = true;

  std::array<double,3> _fixed_clus_error = {.1,.1,.1};


};


#endif