Track.h

#ifndef TRACKING_EVENT_TRACK_H_
#define TRACKING_EVENT_TRACK_H_
 
//----------------------//
//   C++ Standard Lib   //
//----------------------//
#include <iostream>
#include <optional>
#include <vector>
 
//----------//
//   ROOT   //
//----------//
#include "TObject.h"
 
// --- ACTS --- //
// #include "Acts/Definitions/TrackParametrization.hpp"
// #include "Acts/EventData/TrackParameters.hpp"
 
namespace ldmx {
 
 
 
enum TrackStateType {
  RefPoint = 0,
  AtTarget = 1,
  AtFirstMeasurement = 2,
  AtLastMeasurement = 3,
  AtECAL = 4,
  AtBeamOrigin = 5,
  Invalid = 6,
  AtHCAL = 7
};
 
class Track {
 public:
  // Track states won't be visualized in the root tree from the TBrowser, but it
  // will be accessible when reading back the rootfile using for example the
  // monitoring code.
  struct TrackState {
    double ref_x_, ref_y_, ref_z_;
    std::vector<double> params_;
    std::vector<double> cov_;
    TrackStateType ts_type_;
  };
 
  Track(){};
 
  virtual ~Track(){};
 
  friend std::ostream& operator<<(std::ostream& o, const Track& d);
 
  // To match the Framework Bus clear. It's doing nothing
  void clear() {};
 
  void setNhits(int nhits) { n_hits_ = nhits; }
  int getNhits() const { return n_hits_; }
 
  std::optional<TrackState> getTrackState(TrackStateType tstype) const {
    for (auto ts : track_states_)
      if (ts.ts_type_ == tstype) return std::optional<TrackState>(ts);
 
    return std::nullopt;
  }
 
  // void setNholes(int nholes) {n_holes_ = nholes;}
  // int  getNholes() const {return n_holes_;}
 
  void setNoutliers(int nout) { n_outliers_ = nout; }
  int getNoutliers() const { return n_outliers_; }
 
  void setNdf(int ndf) { ndf_ = ndf; }
  int getNdf() const { return ndf_; };
 
  void setNsharedHits(int nsh) { n_shared_hits_ = nsh; }
  int getNsharedHits() const { return n_shared_hits_; }
 
  void setChi2(double chi2) { chi2_ = chi2; }
  double getChi2() const { return chi2_; }
 
  void setTrackID(int trackid) { track_id_ = trackid; };
  int getTrackID() const { return track_id_; };
 
  void setTruthProb(double truthProb) { truth_prob_ = truthProb; };
  double getTruthProb() const { return truth_prob_; };
 
  void setPdgID(int pdgID) { pdg_id_ = pdgID; };
  int getPdgID() const { return pdg_id_; };
 
  // in units of e
  int q() const { return perigee_pars_[4] > 0 ? 1 : -1; }
 
  // Add measurement indices to tracks
  // For reco  tracks they corresponds to the indices in the measurement
  // container For truth tracks they corresponds to the indices of the
  // SimHitCointainer
 
  void addMeasurementIndex(unsigned int measIdx) {
    meas_idxs_.push_back(measIdx);
  }
  std::vector<unsigned int> getMeasurementsIdxs() const { return meas_idxs_; }
 
  void addOutlierIndex(unsigned int measIdx) {
    outlier_idxs_.push_back(measIdx);
  }
  std::vector<unsigned int> getOutlierIdxs() const { return outlier_idxs_; }
 
  void addHoleIndex(unsigned int measIdx) { hole_idxs_.push_back(measIdx); }
  std::vector<unsigned int> getHoleIdxs() const { return hole_idxs_; }
 
  void addSharedIndex(unsigned int measIdx) { shared_idxs_.push_back(measIdx); }
  std::vector<unsigned int> getSharedIdxs() const { return shared_idxs_; }
 
  // void setPerigeeParameters(const Acts::BoundVector& par)  {perigee_pars_ =
  // par; } Acts::BoundVector getPerigeeParameters() {return perigee_pars_;}
 
  // void setPerigeeCov(const Acts::BoundMatrix& cov) {perigee_cov_ = cov;}
  // Acts::BoundMatrix getPerigeeCov() {return perigee_cov_;}
 
  // void setPerigeeState(const Acts::BoundVector& par, const Acts::BoundMatrix&
  // cov) {
  //   perigee_pars_ = par;
  //   perigee_cov_  = cov;
  // }
 
  // Vector representation
  void setPerigeeParameters(const std::vector<double>& par) {
    perigee_pars_ = par;
  }
  std::vector<double> getPerigeeParameters() const { return perigee_pars_; }
 
  void setPerigeeCov(const std::vector<double>& cov) { perigee_cov_ = cov; }
  std::vector<double> getPerigeeCov() const { return perigee_cov_; }
 
  void setPerigeeLocation(const std::vector<double>& perigee) {
    perigee_ = perigee;
  }
 
  void setPerigeeLocation(const double& x_, const double& y_,
                          const double& z_) {
    perigee_[0] = x_;
    perigee_[1] = y_;
    perigee_[2] = z_;
  }
 
  void setMomentum(const double& px, const double& py, const double& pz) {
    momentum_[0] = px;
    momentum_[1] = py;
    momentum_[2] = pz;
  }
 
  void setPosition(const double& x_, const double& y_, const double& z_) {
    position_[0] = x_;
    position_[1] = y_;
    position_[2] = z_;
  }
 
  std::vector<double> getPerigeeLocation() const { return perigee_; };
  double getPerigeeX() const { return perigee_[0]; };
  double getPerigeeY() const { return perigee_[1]; };
  double getPerigeeZ() const { return perigee_[2]; };
 
  std::vector<double> getMomentum() const { return momentum_; };
  std::vector<double> getPosition() const { return position_; };
 
  // getters -- TODO use an enum instead
 
  double getD0() const { return perigee_pars_[0]; };
  double getZ0() const { return perigee_pars_[1]; };
  double getPhi() const { return perigee_pars_[2]; };
  double getTheta() const { return perigee_pars_[3]; };
  double getQoP() const { return perigee_pars_[4]; };
  double getT() const { return perigee_pars_[5]; };
 
  void addTrackState(const ldmx::Track::TrackState& ts) {
    track_states_.push_back(ts);
  };
 
  std::vector<TrackState> getTrackStates() const { return track_states_; }
 
 protected:
  int n_hits_{0};
  int n_outliers_{0};
  int ndf_{0};
  int n_shared_hits_{0};
  int n_holes_{0};
 
  // particle hypothesis if truth track
  // int pdgID_{0};
 
  double chi2_{0};
 
  // The parameters and covariance matrix wrt the perigee surface
  // Acts::BoundVector perigee_pars_;
  // Acts::BoundSymMatrix perigee_cov_;
 
  // 6 elements
  // d0 / z0 / phi / theta / qop / t
  std::vector<double> perigee_pars_{0., 0., 0., 0., 0., 0.};
 
  // 21 elements
  // d0d0 d0z0 d0phi d0th  d0qop  d0t
  //      z0z0 z0phi z0th  z0qop  z0t
  //           phph  phith phqop  pht
  //                  thth thqop  tht
  //                       qopqop qopt
  //                              t
  std::vector<double> perigee_cov_;
 
  // The perigee location
  std::vector<double> perigee_{0., 0., 0.};
 
  // The 3-momentum at the perigee
  std::vector<double> momentum_{0., 0., 0.};
 
  // The 3-position at the perigee
  std::vector<double> position_{0., 0., 0.};
 
  // The vector of measurement IDs
  std::vector<unsigned int> meas_idxs_{};
 
  // The vector of outlier IDs
  std::vector<unsigned int> outlier_idxs_{};
 
  // The vector of hole IDs
  std::vector<unsigned int> hole_idxs_{};
 
  // The vector of shared hit IDs
  std::vector<unsigned int> shared_idxs_{};
 
  // ID of the matched particle in the SimParticles map
  int track_id_{-1};
 
  // Truth probability
  double truth_prob_{0.};
 
  // pdgID
  int pdg_id_{0};
 
  // Track States
  std::vector<TrackState> track_states_;
 
  ClassDef(Track, 4);
 
};  // Track
 
typedef std::vector<ldmx::Track> Tracks;
// typedef std::vector<std::reference_wrapper<const ldmx::Track>> Tracks;
 
}  // namespace ldmx
 
#endif  // TRACKING_EVENT_TRACK_H_