forked from sPHENIX-Collaboration/coresoftware
/
QAG4SimulationUpsilon.h
61 lines (47 loc) · 1.43 KB
/
QAG4SimulationUpsilon.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
57
58
59
60
61
#ifndef QA_QAG4SimulationUpsilon_H
#define QA_QAG4SimulationUpsilon_H
#include <fun4all/SubsysReco.h>
#include <memory>
#include <set>
#include <string>
#include <utility>
class PHCompositeNode;
class PHG4TruthInfoContainer;
class SvtxEvalStack;
/// \class QAG4SimulationUpsilon
class QAG4SimulationUpsilon : public SubsysReco
{
public:
QAG4SimulationUpsilon(const std::string &name = "QAG4SimulationUpsilon");
virtual ~QAG4SimulationUpsilon() {}
int Init(PHCompositeNode *topNode);
int InitRun(PHCompositeNode *topNode);
int process_event(PHCompositeNode *topNode);
// common prefix for QA histograms
std::string get_histo_prefix();
//! If added, only process truth particle associated with the selected list of EmbeddingIDs
//! Call multiple times to add multiple EmbeddingIDs
//! For EmbeddingID<0, all negative embedding IDs are accepted for pile up events.
void addEmbeddingID(int embeddingID);
void setEtaRange(double low, double high)
{
m_etaRange.first = low;
m_etaRange.second = high;
}
void setQuarkoniaPID(const int pid)
{
m_quarkoniaPID = pid;
}
void setDaughterAbsPID(const int pid)
{
m_daughterAbsPID = pid;
}
private:
std::shared_ptr<SvtxEvalStack> _svtxEvalStack;
std::set<int> m_embeddingIDs;
std::pair<double, double> m_etaRange;
PHG4TruthInfoContainer *_truthContainer;
int m_quarkoniaPID = 553;
int m_daughterAbsPID = 11;
};
#endif // QA_QAG4SimulationUpsilon_H