CaloRawTowerEval.h

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

#ifndef G4EVAL_CALORAWTOWEREVAL_H
#define G4EVAL_CALORAWTOWEREVAL_H

#include "CaloTruthEval.h"

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

class PHCompositeNode;

class PHG4CellContainer;
class PHG4Hit;
class PHG4HitContainer;
class PHG4Particle;
class PHG4Shower;
class PHG4TruthInfoContainer;

class RawTower;
class RawTowerContainer;

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

  int get_caloid() { return get_truth_eval()->get_caloid(); }

  void next_event(PHCompositeNode* topNode);

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

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

  unsigned int get_errors() { return _errors + _trutheval.get_errors(); }

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

  CaloTruthEval* get_truth_eval() { return &_trutheval; }

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

  bool has_reduced_node_pointers();

  // shower interface

  std::set<PHG4Shower*> all_truth_primary_showers(RawTower* tower);

  PHG4Shower* max_truth_primary_shower_by_energy(RawTower* tower);

  std::set<RawTower*> all_towers_from(PHG4Shower* primary);

  RawTower* best_tower_from(PHG4Shower* primary);

  float get_energy_contribution(RawTower* tower, PHG4Shower* primary);

  // particle interface

  std::set<PHG4Particle*> all_truth_primary_particles(RawTower* tower);

  PHG4Particle* max_truth_primary_particle_by_energy(RawTower* tower);

  std::set<RawTower*> all_towers_from(PHG4Particle* primary);

  RawTower* best_tower_from(PHG4Particle* primary);

  float get_energy_contribution(RawTower* tower, PHG4Particle* primary);

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

  bool has_full_node_pointers();

  std::set<PHG4Hit*> all_truth_hits(RawTower* tower);

 private:
  void get_node_pointers(PHCompositeNode* topNode);

  std::string _caloname;
  CaloTruthEval _trutheval;
  RawTowerContainer* _towers = nullptr;
  PHG4CellContainer* _g4cells = nullptr;
  PHG4HitContainer* _g4hits = nullptr;
  PHG4TruthInfoContainer* _truthinfo = nullptr;

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

  bool _do_cache = true;

  std::map<RawTower*, std::set<PHG4Shower*> > _cache_all_truth_primary_showers;
  std::map<RawTower*, PHG4Shower*> _cache_max_truth_primary_shower_by_energy;
  std::map<PHG4Shower*, std::set<RawTower*> > _cache_all_towers_from_primary_shower;
  std::map<PHG4Shower*, RawTower*> _cache_best_tower_from_primary_shower;
  std::map<std::pair<RawTower*, PHG4Shower*>, float> _cache_get_energy_contribution_primary_shower;

  std::map<RawTower*, std::set<PHG4Particle*> > _cache_all_truth_primary_particles;
  std::map<RawTower*, PHG4Particle*> _cache_max_truth_primary_particle_by_energy;
  std::map<PHG4Particle*, std::set<RawTower*> > _cache_all_towers_from_primary_particle;
  std::map<PHG4Particle*, RawTower*> _cache_best_tower_from_primary_particle;
  std::map<std::pair<RawTower*, PHG4Particle*>, float> _cache_get_energy_contribution_primary_particle;

  std::map<RawTower*, std::set<PHG4Hit*> > _cache_all_truth_hits;
};

#endif  // G4EVAL_CALORAWTOWEREVAL_H