TrkrTruthTrackv1.h

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

#ifndef TRACKBASE_TRKRTRUTHTRACKV1_H
#define TRACKBASE_TRKRTRUTHTRACKV1_H

#include "TrkrTruthTrack.h"

class PHG4Particle;
class PHG4VtxPoint;

class TrkrTruthTrackv1 : public TrkrTruthTrack
{
 public:
  TrkrTruthTrackv1();
  TrkrTruthTrackv1(unsigned int, PHG4Particle*, PHG4VtxPoint*);

  ~TrkrTruthTrackv1() override = default;

  unsigned int getTrackid() const override { return trackid; };
  float getX0() const override { return X0; };
  float getY0() const override { return Y0; };
  float getZ0() const override { return Z0; };

  float getPseudoRapidity() const override { return pseudoRapidity; };
  float getPt() const override { return pt; };
  float getPhi() const override { return phi; };

  std::vector<TrkrDefs::cluskey>& getClusters() override { return clusters; };
  void addCluster(TrkrDefs::cluskey) override;

  bool has_hitsetkey(TrkrDefs::hitsetkey) const override; // note, only works when the data is sorted
  bool has_hitsetkey(TrkrDefs::cluskey)   const override;
  std::pair<bool, TrkrDefs::cluskey> get_cluskey(TrkrDefs::hitsetkey) const override; // bool is if there is the key, and 

  void setTrackid(unsigned int _) override { trackid = _; };

  void setX0(float _) override { X0 = _; };
  void setY0(float _) override { Y0 = _; };
  void setZ0(float _) override { Z0 = _; };

  void setPseudoRapity(float _) override { pseudoRapidity = _; };
  void setPt(float _) override { pt = _; };
  void setPhi(float _) override { phi = _; };

  void identify(std::ostream& os = std::cout) const override;

  struct CompHitSetKey {
    bool operator()(const TrkrDefs::cluskey& lhs, const TrkrDefs::cluskey& rhs) {
      return TrkrDefs::getHitSetKeyFromClusKey(lhs) < TrkrDefs::getHitSetKeyFromClusKey(rhs);
    };
    bool operator()(const TrkrDefs::cluskey& lhs, const TrkrDefs::hitsetkey& rhs) {
      return TrkrDefs::getHitSetKeyFromClusKey(lhs) < rhs; 
    };
    bool operator()(const TrkrDefs::hitsetkey& lhs, const TrkrDefs::cluskey& rhs) {
      return lhs < TrkrDefs::getHitSetKeyFromClusKey(rhs);
    };
  };

 protected:
  unsigned int trackid;
  float X0;
  float Y0;
  float Z0;

  float pseudoRapidity;
  float pt;
  float phi;

  std::vector<TrkrDefs::cluskey> clusters;
  ClassDefOverride(TrkrTruthTrackv1, 1)
};
#endif