ECCE @ EIC Software
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
G4_KFParticle.C
Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file G4_KFParticle.C
1 #ifndef MACRO_G4KFPARTICLE_C
2 #define MACRO_G4KFPARTICLE_C
3 
4 #include <GlobalVariables.C>
5 
6 #include <kfparticle_sphenix/KFParticle_sPHENIX.h>
7 
8 #include <fun4all/Fun4AllServer.h>
9 
10 R__LOAD_LIBRARY(libkfparticle_sphenix.so)
11 
12 namespace Enable
13 {
14  bool KFPARTICLE = false;
15  bool KFPARTICLE_SAVE_NTUPLE = false;
16  bool KFPARTICLE_APPEND_TO_DST = true;
17  bool KFPARTICLE_TRUTH_MATCH = false;
18  bool KFPARTICLE_DETECTOR_INFO = false;
19  int KFPARTICLE_VERBOSITY = 0;
20  std::string KFPARTICLE_TRACKMAP = "SvtxTrackMap";
21  std::string KFPARTICLE_VERTEXMAP = "SvtxVertexMap";
22 } // namespace Enable
23 
24 namespace KFParticleBaseCut
25 {
26  float minTrackPT = 0.5; // GeV
27  float maxTrackchi2nDoF = 2;
28  float minTrackIPchi2 = 15; // IP = DCA of track with vertex
29  float maxVertexchi2nDoF = 2;
30  float maxTrackTrackDCA = 0.05; // cm
31  float minMotherPT = 0; // GeV
32 } // namespace KFParticleBaseCut
33 
34 void KFParticleInit() {} //I guess this line isnt needed
35 
36 void KFParticle_Upsilon_Reco()
37 {
38  int verbosity = std::max(Enable::VERBOSITY, Enable::KFPARTICLE_VERBOSITY);
39 
40  Fun4AllServer *se = Fun4AllServer::instance();
41 
42  std::string motherName = "Upsilon";
43  KFParticle_sPHENIX *kfparticle = new KFParticle_sPHENIX("KFParticle_" + motherName + "_Reco");
44  kfparticle->Verbosity(verbosity);
45 
46  kfparticle->saveDST(Enable::KFPARTICLE_APPEND_TO_DST);
47  kfparticle->saveOutput(Enable::KFPARTICLE_SAVE_NTUPLE);
48  kfparticle->doTruthMatching(Enable::KFPARTICLE_TRUTH_MATCH);
49  kfparticle->getDetectorInfo(Enable::KFPARTICLE_DETECTOR_INFO);
50 
51  kfparticle->setContainerName(motherName);
52  kfparticle->setOutputName("KFParticleOutput_" + motherName + "_reconstruction.root");
53 
54  std::pair<std::string, int> daughterList[99];
55  kfparticle->setNumberOfTracks(2);
56  daughterList[0] = make_pair("electron", +1);
57  daughterList[1] = make_pair("electron", -1);
58  kfparticle->getChargeConjugate(false);
59  kfparticle->setDaughters(daughterList);
60  kfparticle->setMinimumTrackPT(KFParticleBaseCut::minTrackPT);
61  kfparticle->setMinimumTrackIPchi2(0); // Upsilon decays are prompt, tracks are more likely to point to vertex
62  kfparticle->setMaximumTrackchi2nDOF(KFParticleBaseCut::maxTrackchi2nDoF);
63 
64  kfparticle->setMaximumVertexchi2nDOF(KFParticleBaseCut::maxVertexchi2nDoF);
65  kfparticle->setMaximumDaughterDCA(KFParticleBaseCut::maxTrackTrackDCA);
66 
67  kfparticle->setMotherName(motherName);
68  kfparticle->setMinimumMass(7);
69  kfparticle->setMaximumMass(11);
70  kfparticle->setMotherPT(KFParticleBaseCut::minMotherPT);
71  kfparticle->constrainToPrimaryVertex(false);
72 
73  se->registerSubsystem(kfparticle);
74 
75  return;
76 }
77 
78 
79 void KFParticle_D0_Reco()
80 {
81  int verbosity = std::max(Enable::VERBOSITY, Enable::KFPARTICLE_VERBOSITY);
82 
83  Fun4AllServer *se = Fun4AllServer::instance();
84 
85  std::string motherName = "D0";
86  KFParticle_sPHENIX *kfparticle = new KFParticle_sPHENIX("KFParticle_" + motherName + "_Reco");
87  kfparticle->Verbosity(verbosity);
88 
89  kfparticle->saveDST(Enable::KFPARTICLE_APPEND_TO_DST);
90  kfparticle->saveOutput(Enable::KFPARTICLE_SAVE_NTUPLE);
91  kfparticle->doTruthMatching(Enable::KFPARTICLE_TRUTH_MATCH);
92  kfparticle->getDetectorInfo(Enable::KFPARTICLE_DETECTOR_INFO);
93 
94  kfparticle->setContainerName(motherName);
95  kfparticle->setOutputName("KFParticleOutput_" + motherName + "_reconstruction.root");
96 
97  std::pair<std::string, int> daughterList[99];
98  kfparticle->setNumberOfTracks(2);
99  daughterList[0] = make_pair("kaon", -1);
100  daughterList[1] = make_pair("pion", +1);
101  kfparticle->getChargeConjugate(true);
102  kfparticle->setDaughters(daughterList);
103  kfparticle->setMinimumTrackPT(KFParticleBaseCut::minTrackPT);
104  kfparticle->setMinimumTrackIPchi2(KFParticleBaseCut::minTrackIPchi2);
105  kfparticle->setMaximumTrackchi2nDOF(KFParticleBaseCut::maxTrackchi2nDoF);
106 
107  kfparticle->setMaximumVertexchi2nDOF(KFParticleBaseCut::maxVertexchi2nDoF);
108  kfparticle->setMaximumDaughterDCA(KFParticleBaseCut::maxTrackTrackDCA);
109 
110  kfparticle->setMotherName(motherName);
111  kfparticle->setMinimumMass(1.750);
112  kfparticle->setMaximumMass(1.950);
113  kfparticle->setMotherPT(KFParticleBaseCut::minMotherPT);
114  kfparticle->constrainToPrimaryVertex(false);
115 
116  se->registerSubsystem(kfparticle);
117 
118  return;
119 }
120 
121 
122 void KFParticle_Lambdac_Reco()
123 {
124  int verbosity = std::max(Enable::VERBOSITY, Enable::KFPARTICLE_VERBOSITY);
125 
126  Fun4AllServer *se = Fun4AllServer::instance();
127 
128  std::string motherName = "Lambdac";
129  KFParticle_sPHENIX *kfparticle = new KFParticle_sPHENIX("KFParticle_" + motherName + "_Reco");
130  kfparticle->Verbosity(verbosity);
131 
132  kfparticle->saveDST(Enable::KFPARTICLE_APPEND_TO_DST);
133  kfparticle->saveOutput(Enable::KFPARTICLE_SAVE_NTUPLE);
134  kfparticle->doTruthMatching(Enable::KFPARTICLE_TRUTH_MATCH);
135  kfparticle->getDetectorInfo(Enable::KFPARTICLE_DETECTOR_INFO);
136 
137  kfparticle->setContainerName(motherName);
138  kfparticle->setOutputName("KFParticleOutput_" + motherName + "_reconstruction.root");
139 
140  std::pair<std::string, int> daughterList[99];
141  kfparticle->setNumberOfTracks(3);
142  daughterList[0] = make_pair("proton", +1);
143  daughterList[1] = make_pair("kaon", -1);
144  daughterList[2] = make_pair("pion", +1);
145  kfparticle->getChargeConjugate(true);
146  kfparticle->setDaughters(daughterList);
147  kfparticle->setMinimumTrackPT(KFParticleBaseCut::minTrackPT);
148  kfparticle->setMinimumTrackIPchi2(KFParticleBaseCut::minTrackIPchi2);
149  kfparticle->setMaximumTrackchi2nDOF(KFParticleBaseCut::maxTrackchi2nDoF);
150 
151  kfparticle->setMaximumVertexchi2nDOF(KFParticleBaseCut::maxVertexchi2nDoF);
152  kfparticle->setMaximumDaughterDCA(KFParticleBaseCut::maxTrackTrackDCA);
153 
154  kfparticle->setMotherName(motherName);
155  kfparticle->setMinimumMass(2.150);
156  kfparticle->setMaximumMass(2.400);
157  kfparticle->setMotherPT(KFParticleBaseCut::minMotherPT);
158  kfparticle->constrainToPrimaryVertex(false);
159 
160  se->registerSubsystem(kfparticle);
161 
162  return;
163 }
164 
165 
166 #endif