BaseTruthEval.h

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

#ifndef G4EVAL_BASETRUTHEVAL_H
#define G4EVAL_BASETRUTHEVAL_H

#include <set>

class PHCompositeNode;
class PHG4Hit;
class PHG4Particle;
class PHG4Shower;
class PHG4TruthInfoContainer;
class PHG4VtxPoint;

class BaseTruthEval
{
 public:
  explicit BaseTruthEval(PHCompositeNode* topNode);
  virtual ~BaseTruthEval();

  void next_event(PHCompositeNode* topNode);

  void set_strict(bool strict) { m_Strict = strict; }

  unsigned int get_errors() { return m_Errors; }

  void set_verbosity(int verbosity) { m_Verbosity = verbosity; }

  // ---reduced sim node or better---------------------------------------------

  bool has_reduced_node_pointers();

  int get_embed(PHG4Particle* particle);

  PHG4VtxPoint* get_vertex(PHG4Particle* particle);

  bool is_primary(PHG4Shower* shower);

  bool is_primary(PHG4Particle* particle);

  PHG4Shower* get_primary_shower(PHG4Shower* shower);

  PHG4Shower* get_primary_shower(PHG4Particle* particle);

  PHG4Particle* get_primary_particle(PHG4Particle* particle);

  PHG4Particle* get_primary_particle(PHG4Shower* shower);

  std::set<PHG4Shower*> all_secondary_showers(PHG4Shower* shower);

  bool are_same_shower(PHG4Shower* s1, PHG4Shower* s2);

  bool are_same_particle(PHG4Particle* p1, PHG4Particle* p2);

  bool are_same_vertex(PHG4VtxPoint* vtx1, PHG4VtxPoint* vtx2);

  // ---full sim node required--------------------------------------------------

  bool has_full_node_pointers() { return has_reduced_node_pointers(); }

  PHG4Particle* get_particle(PHG4Hit* g4hit);

  PHG4Shower* get_primary_shower(PHG4Hit* g4hit);

  PHG4Particle* get_primary_particle(PHG4Hit* g4hit);

  PHG4Particle* get_particle(const int trackid);

  bool is_g4hit_from_primary_shower(PHG4Hit* g4hit, PHG4Shower* shower);

  bool is_g4hit_from_particle(PHG4Hit* g4hit, PHG4Particle* particle);

 private:
  void get_node_pointers(PHCompositeNode* topNode);

  PHG4TruthInfoContainer* m_TruthInfo = nullptr;

  bool m_Strict = false;
  int m_Verbosity = 0;
  unsigned int m_Errors = 0;
};

#endif  // G4EVAL_BASETRUTHEVAL_H