forked from sPHENIX-Collaboration/coresoftware
/
QAG4SimulationVertex.h
56 lines (38 loc) · 1.32 KB
/
QAG4SimulationVertex.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
// Tell emacs that this is a C++ source
// -*- C++ -*-.
#ifndef QAG4SIMULATIONVERTEX_H
#define QAG4SIMULATIONVERTEX_H
#include <fun4all/SubsysReco.h>
#include <g4eval/SvtxEvalStack.h>
#include <memory>
#include <set>
#include <string>
class PHCompositeNode;
class PHG4TruthInfoContainer;
class SvtxTrackMap;
class SvtxVertexMap;
class QAG4SimulationVertex : public SubsysReco
{
public:
QAG4SimulationVertex(const std::string &name = "QAG4SimulationVertex");
virtual ~QAG4SimulationVertex() = default;
int Init(PHCompositeNode *topNode);
int InitRun(PHCompositeNode *topNode);
int process_event(PHCompositeNode *topNode);
std::string get_histo_prefix();
void addEmbeddingID(int embeddingID);
void setTrackgMapName(const std::string &name) { m_trackMapName = name; }
void setVertexMapName(const std::string &name) { m_vertexMapName = name; }
private:
int load_nodes(PHCompositeNode *);
PHG4TruthInfoContainer *m_truthContainer = nullptr;
unsigned int _nlayers_maps = 3;
std::unique_ptr<SvtxEvalStack> m_svtxEvalStack;
SvtxTrackMap *m_trackMap = nullptr;
SvtxVertexMap *m_vertexMap = nullptr;
PHG4TruthInfoContainer *m_truthInfo = nullptr;
std::set<int> m_embeddingIDs;
std::string m_trackMapName = "SvtxTrackMap";
std::string m_vertexMapName = "SvtxVertexMap";
};
#endif // QAG4SIMULATIONVERTEX_H