MvtxStandaloneTracking.h

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

// Tell emacs that this is a C++ source
//  -*- C++ -*-.
#ifndef MVTX_StandaloneTracking_H
#define MVTX_StandaloneTracking_H

#include <vector>
#include <set>

#include <TMatrixD.h>
#include <TVectorD.h>

class TrkrCluster;
class TrkrClusterContainer;
class PHCompositeNode;

class MvtxStandaloneTracking
{
public:

  struct MvtxTrack
  {
    std::vector<TrkrCluster*> ClusterList;
    std::vector<double> dx;
    std::vector<double> dz;
    double m_xy;
    double b_xy;
    double chi2_xy;
    double m_zy;
    double b_zy;
    double chi2_zy;
  };
  typedef std::vector<MvtxTrack> MvtxTrackList;

  MvtxStandaloneTracking();

  ~MvtxStandaloneTracking();

  void RunTracking(PHCompositeNode* topNode, MvtxTrackList &trklst, std::vector<int> &lyrs);

  void RunGhostRejection( MvtxTrackList &trklst);

  void SetWindowX(float w) { window_x_ = w; }
  void SetWindowZ(float w) { window_z_ = w; }

  void SetGhostRejection( bool yn ) { ghostrejection_ = yn; }

  void Verbosity(int v ) { verbosity_ = v; }

private:

  void AssociateClusters(MvtxTrackList &trklst, std::vector<int> &lyrs);

  void TrackFitXY(MvtxTrack &trk);

  void TrackFitZY(MvtxTrack &trk);

  TVectorD SolveGLS(TMatrixD &X, TVectorD &y, TMatrixD &L);


  double CalcSlope(double x0, double y0, double x1, double y1);
  double CalcIntecept(double x0, double y0, double m);
  double CalcProjection(double x, double m, double b);

  void PrintTrackCandidates(MvtxTrackList &trklst);

  TrkrClusterContainer* clusters_;

  float window_x_;
  float window_z_;

  bool ghostrejection_;

  // verbosity
  int verbosity_;

  static const int NLYR = 4;

};

#endif //__MvtxStandaloneTracking_h__#