JetTruthEval.h

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

#ifndef G4EVAL_JETTRUTHEVAL_H
#define G4EVAL_JETTRUTHEVAL_H

#include "CaloEvalStack.h"
#include "SvtxEvalStack.h"

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

class Jet;
class JetContainer;

class PHCompositeNode;

class PHG4Hit;
class PHG4Particle;
class PHG4Shower;
class PHG4TruthInfoContainer;

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

  void next_event(PHCompositeNode* topNode);

  void do_caching(bool do_cache)
  {
    _do_cache = do_cache;
    _svtxevalstack.do_caching(do_cache);
    _cemcevalstack.do_caching(do_cache);
    _hcalinevalstack.do_caching(do_cache);
    _hcaloutevalstack.do_caching(do_cache);
    _femcevalstack.do_caching(do_cache);
    _fhcalevalstack.do_caching(do_cache);
    _eemcevalstack.do_caching(do_cache);
  }

  void set_strict(bool strict)
  {
    _strict = strict;
    _svtxevalstack.set_strict(strict);
    _cemcevalstack.set_strict(strict);
    _hcalinevalstack.set_strict(strict);
    _hcaloutevalstack.set_strict(strict);
    _femcevalstack.set_strict(strict);
    _fhcalevalstack.set_strict(strict);
    _eemcevalstack.set_strict(strict);
  }

  unsigned int get_errors()
  {
    return _errors + _svtxevalstack.get_errors() + _cemcevalstack.get_errors() + _hcalinevalstack.get_errors() + _hcaloutevalstack.get_errors() + _femcevalstack.get_errors() + _fhcalevalstack.get_errors() + _eemcevalstack.get_errors();
  }

  void set_verbosity(int verbosity)
  {
    _verbosity = verbosity;
    _svtxevalstack.set_verbosity(verbosity);
    _cemcevalstack.set_verbosity(verbosity);
    _hcalinevalstack.set_verbosity(verbosity);
    _hcaloutevalstack.set_verbosity(verbosity);
    _femcevalstack.set_verbosity(verbosity);
    _fhcalevalstack.set_verbosity(verbosity);
    _eemcevalstack.set_verbosity(verbosity);
  }

  // set track node name
  void set_track_nodename(const std::string& name)
  {
    _svtxevalstack.set_track_nodename(name);
  }

  SvtxEvalStack* get_svtx_eval_stack() { return &_svtxevalstack; }

  CaloEvalStack* get_cemc_eval_stack() { return &_cemcevalstack; }

  CaloEvalStack* get_hcalin_eval_stack() { return &_hcalinevalstack; }

  CaloEvalStack* get_hcalout_eval_stack() { return &_hcaloutevalstack; }

  CaloEvalStack* get_femc_eval_stack() { return &_femcevalstack; }

  CaloEvalStack* get_fhcal_eval_stack() { return &_fhcalevalstack; }

  CaloEvalStack* get_eemc_eval_stack() { return &_eemcevalstack; }

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

  Jet* get_truth_jet(PHG4Particle* truthparticle);

  std::set<PHG4Particle*> all_truth_particles(Jet* truthjet);

  std::set<PHG4Shower*> all_truth_showers(Jet* truthjet);

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

  std::set<PHG4Hit*> all_truth_hits(Jet* truthjet);

 private:
  void get_node_pointers(PHCompositeNode* topNode);

  std::string _truthjetname;
  SvtxEvalStack _svtxevalstack;
  CaloEvalStack _cemcevalstack;
  CaloEvalStack _hcalinevalstack;
  CaloEvalStack _hcaloutevalstack;
  CaloEvalStack _femcevalstack;
  CaloEvalStack _fhcalevalstack;
  CaloEvalStack _eemcevalstack;

  PHG4TruthInfoContainer* _truthinfo = nullptr;
  JetContainer* _truthjets = nullptr;

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

  bool _do_cache = true;
  std::map<Jet*, std::set<PHG4Particle*> > _cache_all_truth_particles;
  std::map<Jet*, std::set<PHG4Shower*> > _cache_all_truth_showers;
  std::map<Jet*, std::set<PHG4Hit*> > _cache_all_truth_hits;
  std::map<PHG4Particle*, Jet*> _cache_get_truth_jet;
};

#endif  // JETTRUTHEVAL_H__