Track.h

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

#ifndef PHGENFIT_TRACK_H
#define PHGENFIT_TRACK_H

#include <trackbase/TrkrDefs.h>

#include <TMatrixDSymfwd.h>
#include <TVector3.h>

//STL
#include <map>
#include <memory>
#include <vector>
#include <iostream>

//GenFit

namespace genfit
{
class AbsTrackRep;
class MeasuredStateOnPlane;
class Track;
}  // namespace genfit

namespace PHGenFit
{
class Measurement;

class Track
{
 public:
  Track(genfit::AbsTrackRep* rep, TVector3 seed_pos, TVector3 seed_mom, TMatrixDSym seed_cov, const int v = 0);

  Track(const PHGenFit::Track& t);

  ~Track();

  int addMeasurement(PHGenFit::Measurement* measurement);
  int addMeasurements(std::vector<PHGenFit::Measurement*>& measurements);

  int deleteLastMeasurement();

  int updateOneMeasurementKalman(
      const std::vector<PHGenFit::Measurement*>& measurements,
      std::map<double, std::shared_ptr<PHGenFit::Track> >& incr_chi2s_new_tracks,
      const int base_tp_idx = -1,
      const int direction = 1,
      const float blowup_factor = 1.,
      const bool use_fitted_state = false) const;

  double extrapolateToPlane(genfit::MeasuredStateOnPlane& state, TVector3 O, TVector3 n, const int tr_point_id = 0) const;
  double extrapolateToLine(genfit::MeasuredStateOnPlane& state, TVector3 line_point, TVector3 line_direction, const int tr_point_id = 0) const;
  double extrapolateToCylinder(genfit::MeasuredStateOnPlane& state, double radius, TVector3 line_point, TVector3 line_direction, const int tr_point_id = 0, const int direction = 1) const;
  double extrapolateToPoint(genfit::MeasuredStateOnPlane& state, TVector3 P, const int tr_point_id = 0) const;

  genfit::MeasuredStateOnPlane* extrapolateToPlane(TVector3 O, TVector3 n, const int tr_point_id = 0) const;
  genfit::MeasuredStateOnPlane* extrapolateToLine(TVector3 line_point, TVector3 line_direction, const int tr_point_id = 0) const;
  genfit::MeasuredStateOnPlane* extrapolateToCylinder(double radius, TVector3 line_point, TVector3 line_direction, const int tr_point_id = 0, const int direction = 1) const;
  genfit::MeasuredStateOnPlane* extrapolateToPoint(TVector3 P, const int tr_point_id = 0) const;
  genfit::Track* getGenFitTrack() { return _track; }

  genfit::Track* getGenFitTrack() const { return _track; }

  double get_chi2() const;

  double get_ndf() const;

  double get_charge() const;

  TVector3 get_mom() const;

  // old tracking
  const std::vector<unsigned int>& get_cluster_IDs() const
  {
    return _clusterIDs;
  }
  void set_cluster_IDs(const std::vector<unsigned int>& clusterIDs)
  {
    _clusterIDs = clusterIDs;
  }

  // new tracking
  const std::vector<TrkrDefs::cluskey>& get_cluster_keys() const
  {
    return _clusterkeys;
  }
  void set_cluster_keys(const std::vector<TrkrDefs::cluskey>& clusterkeys)
  {
    _clusterkeys = clusterkeys;
  }

  void set_vertex_id(const unsigned int vert_id)
  {
    _vertex_id = vert_id;
  }

  unsigned int get_vertex_id()
  {
    //std::cout << " Track: returning vertex_id = " << _vertex_id << std::endl;
    return _vertex_id;
  }


  int get_verbosity() const
  {
    return verbosity;
  }

  void set_verbosity(int verbosity)
  {
    this->verbosity = verbosity;
  }

  //SMART(genfit::Track) getGenFitTrack() {return _track;}

 private:
  Track operator=(Track &trk) = delete; 

  int verbosity;

  genfit::Track* _track;
  //std::vector<PHGenFit::Measurement*> _measurements;
  std::vector<unsigned int> _clusterIDs;
  std::vector<TrkrDefs::cluskey> _clusterkeys;
  unsigned int _vertex_id;

  //SMART(genfit::Track) _track;
};
}  // namespace PHGenFit

#endif