g4simulation/g4eval/SvtxEvalStack.h

0001 #ifndef G4EVAL_SVTXEVALSTACK_H
0002 #define G4EVAL_SVTXEVALSTACK_H
0003
0004 #include "SvtxVertexEval.h"
0005
0006 class PHCompositeNode;
0007 class SvtxClusterEval;
0008 class SvtxHitEval;
0009 class SvtxTrackEval;
0010 class SvtxTruthEval;
0011
0012 #include <string>  // for string
0013
0014 // This user class provides pointers to the
0015 // full set of tracking evaluators and
0016 // protects the user from future introduction
0017 // of new eval heirachies (new eval objects can
0018 // be introduced without rewrites)
0019
0020 class SvtxEvalStack
0021 {
0022  public:
0023   SvtxEvalStack(PHCompositeNode* topNode);
0024   virtual ~SvtxEvalStack() {}
0025
0026   void next_event(PHCompositeNode* topNode);
0027   void do_caching(bool do_cache) { _vertexeval.do_caching(do_cache); }
0028   void set_strict(bool strict) { _vertexeval.set_strict(strict); }
0029   // void set_over_write_vertexmap(bool over_write) {_vertexeval.set_over_write_vertexmap(over_write);}
0030   void set_use_initial_vertex(bool use_init_vtx) { _vertexeval.set_use_initial_vertex(use_init_vtx); }
0031   void set_use_genfit_vertex(bool use_genfit_vtx) { _vertexeval.set_use_genfit_vertex(use_genfit_vtx); }
0032   void set_verbosity(int verbosity) { _vertexeval.set_verbosity(verbosity); }
0033
0034   SvtxVertexEval* get_vertex_eval() { return &_vertexeval; }
0035   SvtxTrackEval* get_track_eval() { return _vertexeval.get_track_eval(); }
0036   SvtxClusterEval* get_cluster_eval() { return _vertexeval.get_cluster_eval(); }
0037   SvtxHitEval* get_hit_eval() { return _vertexeval.get_hit_eval(); }
0038   SvtxTruthEval* get_truth_eval() { return _vertexeval.get_truth_eval(); }
0039
0040   unsigned int get_errors() { return _vertexeval.get_errors(); }
0041
0042   void set_track_nodename(const std::string& name) { _vertexeval.set_track_nodename(name); }
0043
0044  private:
0045   SvtxVertexEval _vertexeval;  // right now this is the top-level eval
0046 };
0047
0048 #endif  // G4EVAL_SVTXEVALSTACK_H