PHTpcTrackFollower.h

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

#ifndef PHTPCTRACKFOLLOWER_H_
#define PHTPCTRACKFOLLOWER_H_

#include <TVector3.h>  // for TVector3

#include <utility>  // for pair
#include <vector>   // for vector

class PHField;
class PHTpcLookup;
class TrkrClusterContainer;

namespace PHGenFit
{
  class SpacepointMeasurement2;
}
namespace PHGenFit2
{
  class Fitter;
}
namespace PHGenFit2
{
  class Track;
}
namespace genfit
{
  class MeasuredStateOnPlane;
}
namespace genfit
{
  class Track;
}
namespace kdfinder
{
  template <class T>
  class TrackCandidate;
}

class PHTpcTrackFollower
{
 public:
  PHTpcTrackFollower();
  virtual ~PHTpcTrackFollower() {}

  std::vector<PHGenFit2::Track*> followTracks(std::vector<kdfinder::TrackCandidate<double>*>& candidates,
                                              PHField* B, PHTpcLookup* lookup, PHGenFit2::Fitter* fitter);

  PHGenFit2::Track* propagateTrack(kdfinder::TrackCandidate<double>* candidate,
                                   PHTpcLookup* lookup, PHGenFit2::Fitter* fitter);

  void set_optimization_helix(bool opt = true) { mOptHelix = opt; }
  void set_optimization_precise_fit(bool opt = true) { mOptPreciseFit = opt; }

 protected:
  PHGenFit2::Track* candidate_to_genfit(kdfinder::TrackCandidate<double>* candidate);
  PHGenFit::SpacepointMeasurement2* hit_to_measurement(std::vector<double>& hit);
  int get_track_layer(PHGenFit2::Track* track, int dir = 1);
  int get_track_layer(genfit::Track* gftrack, int dir = 1);
  std::pair<genfit::MeasuredStateOnPlane*, double> get_projected_coordinate(genfit::Track* gftrack, int dir, double radius);
  std::pair<genfit::MeasuredStateOnPlane*, double> get_projected_coordinate(genfit::Track* gftrack, int dir, const TVector3& point);
  int followTrack(PHGenFit2::Track* track, PHTpcLookup* lookup, PHGenFit2::Fitter* fitter, int dir = 1);

 private:
  bool mOptHelix;
  bool mOptPreciseFit;
};

#endif /* PHTPCTRACKFOLLOWER_H_ */