CaloTruthEval.h

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

#ifndef G4EVAL_CALOTRUTHEVAL_H
#define G4EVAL_CALOTRUTHEVAL_H

#include "BaseTruthEval.h"

#include <map>
#include <set>
#include <string>

class PHCompositeNode;

class PHG4Hit;
class PHG4HitContainer;
class PHG4Particle;
class PHG4TruthInfoContainer;
class PHG4Shower;
class PHG4VtxPoint;

class CaloTruthEval
{
 public:
  CaloTruthEval(PHCompositeNode* topNode, const std::string& caloname);
  virtual ~CaloTruthEval();

  int get_caloid() { return _caloid; }

  void next_event(PHCompositeNode* topNode);

  void do_caching(bool do_cache) { _do_cache = do_cache; }

  void set_strict(bool strict)
  {
    _strict = strict;
    _basetrutheval.set_strict(strict);
  }

  unsigned int get_errors() { return _errors; }

  void set_verbosity(int verbosity)
  {
    _verbosity = verbosity;
    _basetrutheval.set_verbosity(verbosity);
  }

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

  bool has_reduced_node_pointers();

  PHG4Shower* get_primary_shower(PHG4Shower* shower);

  PHG4Shower* get_primary_shower(PHG4Particle* particle);

  PHG4Particle* get_primary_particle(PHG4Shower* shower);

  PHG4Particle* get_primary_particle(PHG4Particle* particle);

  int get_embed(PHG4Particle* particle);

  PHG4VtxPoint* get_vertex(PHG4Particle* particle);

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

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

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

  bool is_primary(PHG4Shower* shower);

  bool is_primary(PHG4Particle* particle);

  float get_shower_energy_deposit(PHG4Particle* primary);

  float get_shower_eh_ratio(PHG4Particle* primary);

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

  bool has_full_node_pointers();

  std::set<PHG4Hit*> all_truth_hits(PHG4Shower* shower);

  std::set<PHG4Hit*> all_truth_hits(PHG4Particle* particle);

  PHG4Particle* get_parent_particle(PHG4Hit* g4hit);

  PHG4Particle* get_primary_particle(PHG4Hit* g4hit);

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

  std::set<PHG4Hit*> get_shower_hits_from_primary(PHG4Particle* primary);

 private:
  void get_node_pointers(PHCompositeNode* topNode);

  BaseTruthEval _basetrutheval;

  std::string _caloname;
  int _caloid;
  PHG4TruthInfoContainer* _truthinfo = nullptr;
  PHG4HitContainer* _g4hits = nullptr;
  int _g4hit_container_id = -1;

  bool _strict = false;
  int _verbosity = 1;
  unsigned int _errors = 0;

  bool _do_cache = true;
  std::map<PHG4Particle*, float> _cache_get_shower_energy_deposit;
  std::map<PHG4Shower*, std::set<PHG4Hit*> > _cache_all_truth_hits_g4shower;
  std::map<PHG4Particle*, std::set<PHG4Hit*> > _cache_all_truth_hits_g4particle;
  std::map<PHG4Hit*, PHG4Particle*> _cache_get_primary_particle_g4hit;
  std::map<PHG4Particle*, std::set<PHG4Hit*> > _cache_get_shower_hits_from_primary;
};

#endif  // G4EVAL_CALOTRUTHEVAL_H