Analysis Software
Documentation for sPHENIX simulation software
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
HFReco.C
Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file HFReco.C
1 #include <fun4all/Fun4AllServer.h>
2 
3 #pragma GCC diagnostic push
4 
5 #pragma GCC diagnostic ignored "-Wundefined-internal"
6 
7 #include <kfparticle_sphenix/KFParticle_sPHENIX.h>
8 #include <resonancejettagging/ResonanceJetTagging.h>
9 #include <fun4all/Fun4AllReturnCodes.h>
10 
11 #pragma GCC diagnostic pop
12 
13 R__LOAD_LIBRARY(libkfparticle_sphenix.so)
14 
15 namespace HFjets
16 {
17 namespace Enable
18 {
19  bool KFPARTICLE_SAVE_NTUPLE = false;
20  bool KFPARTICLE_APPEND_TO_DST = true;
21  bool KFPARTICLE_TRUTH_MATCH = true;
22  bool KFPARTICLE_DETECTOR_INFO = false;
23  bool KFPARTICLE_CALO_INFO = false;
24  bool KFPARTICLE_ALL_PV_INFO = false;
25  int KFPARTICLE_VERBOSITY = INT_MAX;
26  bool runTracking = false;
27  bool runTruthTrigger = true;
28  string reconstructionName = "myTestReco"; //Used for naming output folder, file and nodes
29 } // namespace Enable
30 
31 
32 
33 
34 namespace KFPARTICLE
35 {
36  bool runQA = false;
37 
38  std::string D0Name = "D0";
39  std::string D0toK3piName = "D0TOK3PI";
40  std::pair<float, float> D0MassRange(1.7, 2.0);
41  bool D0Trigger = false;
42  bool runD0toKpiReco = false;
43  std::string D0toKpiDecayDescriptor = "[" + D0Name + " -> K^- pi^+]cc";
44  bool runD0toK3piReco = false;
45  std::string D0toK3piDecayDescriptor = "[" + D0Name + " -> K^- pi^- pi^+ pi^+]cc";
46 
47  std::string DplusName = "Dplus";
48  std::pair<float, float> DplusMassRange(1.7, 2.00);
49  bool DplusTrigger = false;
50  bool runDplustoK2piReco = false;
51  std::string DplustoK2piDecayDescriptor = "[" + DplusName + " -> K^- pi^+ pi^+]cc";
52 
53  std::string LambdacName = "Lambdac";
54  std::pair<float, float> LambdacMassRange(2.15, 2.45);
55  bool LambdacTrigger = false;
56  bool runLambdactopKpiReco = false;
57  std::string LambdacDecayDescriptor = "[" + LambdacName + " -> proton^+ K^- pi^+]cc";
58 }
59 
60 namespace KFParticleBaseCut
61 {
62  bool KFPARTICLE_ALLOW_ZERO_MASS_TRACKS = true;
63  bool constrainToPV = true;
64  float minTrackPT = 0.7; // GeV
65  float maxTrackchi2nDoF = 5; // An 3
66  float minTrackIPchi2 = 0; // IP = DCA of track with vertex AuAu 9, An 9
67  float minTrackIP = 0.0025; // cm, An 0 AuAu 0.008
68  float maxVertexchi2nDoF = 5;
69  float maxTrackTrackDCA = 0.008; // cm An 0.03
70  float minMotherPT = 1.5; // GeV An 0
71 } // namespace KFParticleBaseCut
72 
73 
74 int KFParticle_Set_Reco(ResonanceJetTagging::TAG tag)
75 {
76 
77  Fun4AllServer *se = Fun4AllServer::instance();
78 
79  std::string particle_name;
80  switch (tag) {
81  case ResonanceJetTagging::TAG::D0:
82  particle_name = KFPARTICLE::D0Name;
83  break;
84  case ResonanceJetTagging::TAG::D0TOK3PI:
85  particle_name = KFPARTICLE::D0toK3piName;
86  break;
87  case ResonanceJetTagging::TAG::DPLUS:
88  particle_name = KFPARTICLE::DplusName;
89  break;
90  case ResonanceJetTagging::TAG::LAMBDAC:
91  particle_name = KFPARTICLE::LambdacName;
92  break;
93  default:
94  std::cout<<"ERROR:decay parameters not set, ABORTING!";
95  return Fun4AllReturnCodes::ABORTRUN;
96  break;
97 
98  }
99 
100  KFParticle_sPHENIX *kfparticle = new KFParticle_sPHENIX("KFParticle_" + particle_name + "_Reco");
101 
102  switch (tag) {
103  case ResonanceJetTagging::TAG::D0:
104  kfparticle->setDecayDescriptor(KFPARTICLE::D0toKpiDecayDescriptor);
105  kfparticle->setContainerName(KFPARTICLE::D0Name);
106  kfparticle->setOutputName("KFParticleOutput_" + KFPARTICLE::D0Name + "_reconstruction.root");
107  break;
108  case ResonanceJetTagging::TAG::D0TOK3PI:
109  kfparticle->setDecayDescriptor(KFPARTICLE::D0toK3piDecayDescriptor);
110  kfparticle->setContainerName(KFPARTICLE::D0toK3piName);
111  kfparticle->setOutputName("KFParticleOutput_" + KFPARTICLE::D0toK3piName + "_reconstruction.root");
112  break;
113  case ResonanceJetTagging::TAG::DPLUS:
114  kfparticle->setDecayDescriptor(KFPARTICLE::DplustoK2piDecayDescriptor);
115  kfparticle->setContainerName(KFPARTICLE::DplusName);
116  kfparticle->setOutputName("KFParticleOutput_" + KFPARTICLE::DplusName + "_reconstruction.root");
117  break;
118  case ResonanceJetTagging::TAG::LAMBDAC:
119  kfparticle->setDecayDescriptor(KFPARTICLE::LambdacDecayDescriptor);
120  kfparticle->setContainerName(KFPARTICLE::LambdacName);
121  kfparticle->setOutputName("KFParticleOutput_" + KFPARTICLE::LambdacName + "_reconstruction.root");
122  break;
123  default:
124  std::cout<<"ERROR:decay parameters not set, ABORTING!";
125  return Fun4AllReturnCodes::ABORTRUN;
126  break;
127  }
128 
129  kfparticle->Verbosity(Enable::KFPARTICLE_VERBOSITY);
130  kfparticle->saveDST(Enable::KFPARTICLE_APPEND_TO_DST);
131  kfparticle->saveOutput(Enable::KFPARTICLE_SAVE_NTUPLE);
132  kfparticle->doTruthMatching(Enable::KFPARTICLE_TRUTH_MATCH);
133  kfparticle->getDetectorInfo(Enable::KFPARTICLE_DETECTOR_INFO);
134  kfparticle->getCaloInfo(Enable::KFPARTICLE_CALO_INFO);
135  kfparticle->getAllPVInfo(Enable::KFPARTICLE_ALL_PV_INFO);
136 
137 
138  //Track parameters
139  kfparticle->setMinimumTrackPT(KFParticleBaseCut::minTrackPT);
140  kfparticle->setMaximumTrackchi2nDOF(KFParticleBaseCut::maxTrackchi2nDoF);
141  kfparticle->setMinimumTrackIPchi2(KFParticleBaseCut::minTrackIPchi2);
142  kfparticle->setMinimumTrackIP(KFParticleBaseCut::minTrackIP);
143  kfparticle->allowZeroMassTracks(KFParticleBaseCut::KFPARTICLE_ALLOW_ZERO_MASS_TRACKS);
144 
145  //Vertex parameters
146  kfparticle->setMaximumVertexchi2nDOF(KFParticleBaseCut::maxVertexchi2nDoF);
147  kfparticle->setMaximumDaughterDCA(KFParticleBaseCut::maxTrackTrackDCA);
148  kfparticle->constrainToPrimaryVertex(KFParticleBaseCut::constrainToPV);
149 
150  //Parent param
151  kfparticle->setMotherPT(KFParticleBaseCut::minMotherPT);
152 
153  switch (tag) {
154  case ResonanceJetTagging::TAG::D0:
155  kfparticle->setMinimumMass(KFPARTICLE::D0MassRange.first);
156  kfparticle->setMaximumMass(KFPARTICLE::D0MassRange.second);
157  kfparticle->setMinDIRA(0.9); //AuAu 0.98 An 0.98
158  kfparticle->setMotherIPchi2(3);
159  kfparticle->setFlightDistancechi2(0);
160  // kfparticle->setNumberOfTracks(2);
161  kfparticle->setMaximumDaughterDCA(KFParticleBaseCut::maxTrackTrackDCA); //AuAu 0.005 An 0.03
162  break;
163  case ResonanceJetTagging::TAG::D0TOK3PI:
164  kfparticle->setMinimumMass(KFPARTICLE::D0MassRange.first);
165  kfparticle->setMaximumMass(KFPARTICLE::D0MassRange.second);
166  kfparticle->setMinDIRA(0.95);
167  kfparticle->setMotherIPchi2(15);
168  kfparticle->setFlightDistancechi2(0);
169  // kfparticle->setNumberOfTracks(4);
170  kfparticle->setMaximumDaughterDCA(KFParticleBaseCut::maxTrackTrackDCA);
171  break;
172  case ResonanceJetTagging::TAG::DPLUS:
173  kfparticle->setMinimumMass(KFPARTICLE::DplusMassRange.first);
174  kfparticle->setMaximumMass(KFPARTICLE::DplusMassRange.second);
175  std::cout<<"ERROR: TAG::DPLUS decay parameters not set, ABORTING!";
176  return Fun4AllReturnCodes::ABORTRUN;
177  break;
178  case ResonanceJetTagging::TAG::LAMBDAC:
179  kfparticle->setMinimumMass(KFPARTICLE::LambdacMassRange.first);
180  kfparticle->setMaximumMass(KFPARTICLE::LambdacMassRange.second);
181  std::cout<<"ERROR: TAG::Lambdac decay parameters not set, ABORTING!";
182  return Fun4AllReturnCodes::ABORTRUN;
183  break;
184  default:
185  std::cout<<"ERROR:decay parameters not set, ABORTING!";
186  return Fun4AllReturnCodes::ABORTRUN;
187  break;
188  }
189 
190  se->registerSubsystem(kfparticle);
191 
192  //KFPARTICLE::runD0Reco = true;
193 
194  return 0;
195 }
196 } // namespace HFjets
197