clang -cc1 -cc1 -triple x86_64-unknown-linux-gnu -analyze -disable-free -main-file-name DAnalysisUtilities.cc -analyzer-store=region -analyzer-opt-analyze-nested-blocks -analyzer-checker=core -analyzer-checker=apiModeling -analyzer-checker=unix -analyzer-checker=deadcode -analyzer-checker=cplusplus -analyzer-checker=security.insecureAPI.UncheckedReturn -analyzer-checker=security.insecureAPI.getpw -analyzer-checker=security.insecureAPI.gets -analyzer-checker=security.insecureAPI.mktemp -analyzer-checker=security.insecureAPI.mkstemp -analyzer-checker=security.insecureAPI.vfork -analyzer-checker=nullability.NullPassedToNonnull -analyzer-checker=nullability.NullReturnedFromNonnull -analyzer-output plist -w -setup-static-analyzer -mrelocation-model pic -pic-level 2 -fhalf-no-semantic-interposition -mframe-pointer=none -fmath-errno -fno-rounding-math -mconstructor-aliases -munwind-tables -target-cpu x86-64 -tune-cpu generic -fno-split-dwarf-inlining -debugger-tuning=gdb -resource-dir /w/halld-scifs17exp/home/sdobbs/clang/llvm-project/install/lib/clang/12.0.0 -D HAVE_CCDB -D HAVE_RCDB -D HAVE_EVIO -D HAVE_TMVA=1 -D RCDB_MYSQL=1 -D RCDB_SQLITE=1 -D SQLITE_USE_LEGACY_STRUCT=ON -I .Linux_CentOS7.7-x86_64-gcc4.8.5/libraries/ANALYSIS -I libraries/ANALYSIS -I . -I libraries -I libraries/include -I /w/halld-scifs17exp/home/sdobbs/clang/halld_recon/Linux_CentOS7.7-x86_64-gcc4.8.5/include -I external/xstream/include -I /usr/include/tirpc -I /group/halld/Software/builds/Linux_CentOS7.7-x86_64-gcc4.8.5/root/root-6.08.06/include -I /w/halld-scifs17exp/halld2/home/sdobbs/Software/jana/jana_0.8.2/Linux_CentOS7.7-x86_64-gcc4.8.5/include -I /group/halld/Software/builds/Linux_CentOS7.7-x86_64-gcc4.8.5/ccdb/ccdb_1.06.06/include -I /group/halld/Software/builds/Linux_CentOS7.7-x86_64-gcc4.8.5/rcdb/rcdb_0.06.00/cpp/include -I /usr/include/mysql -I /group/halld/Software/builds/Linux_CentOS7.7-x86_64-gcc4.8.5/sqlitecpp/SQLiteCpp-2.2.0^bs130/include -I /group/halld/Software/builds/Linux_CentOS7.7-x86_64-gcc4.8.5/sqlite/sqlite-3.13.0^bs130/include -I /group/halld/Software/builds/Linux_CentOS7.7-x86_64-gcc4.8.5/hdds/hdds-4.9.0/Linux_CentOS7.7-x86_64-gcc4.8.5/src -I /group/halld/Software/builds/Linux_CentOS7.7-x86_64-gcc4.8.5/xerces-c/xerces-c-3.1.4/include -I /group/halld/Software/builds/Linux_CentOS7.7-x86_64-gcc4.8.5/evio/evio-4.4.6/Linux-x86_64/include -internal-isystem /usr/lib/gcc/x86_64-redhat-linux/4.8.5/../../../../include/c++/4.8.5 -internal-isystem /usr/lib/gcc/x86_64-redhat-linux/4.8.5/../../../../include/c++/4.8.5/x86_64-redhat-linux -internal-isystem /usr/lib/gcc/x86_64-redhat-linux/4.8.5/../../../../include/c++/4.8.5/backward -internal-isystem /usr/local/include -internal-isystem /w/halld-scifs17exp/home/sdobbs/clang/llvm-project/install/lib/clang/12.0.0/include -internal-externc-isystem /include -internal-externc-isystem /usr/include -O2 -std=c++11 -fdeprecated-macro -fdebug-compilation-dir /home/sdobbs/work/clang/halld_recon/src -ferror-limit 19 -fgnuc-version=4.2.1 -fcxx-exceptions -fexceptions -vectorize-loops -vectorize-slp -analyzer-output=html -faddrsig -o /tmp/scan-build-2021-01-21-110224-160369-1 -x c++ libraries/ANALYSIS/DAnalysisUtilities.cc
1 | #ifdef VTRACE |
2 | #include "vt_user.h" |
3 | #endif |
4 | |
5 | #include "DAnalysisUtilities.h" |
6 | #include "ANALYSIS/DParticleComboCreator.h" |
7 | |
8 | DAnalysisUtilities::DAnalysisUtilities(JEventLoop* locEventLoop) |
9 | { |
10 | DEBUG_LEVEL=0; |
11 | gPARMS->SetDefaultParameter("DAnalysisUtilities:DEBUG_LEVEL",DEBUG_LEVEL); |
12 | |
13 | locEventLoop->GetSingle(dPIDAlgorithm); |
14 | |
15 | dTargetZCenter = 65.0; |
16 | |
17 | DApplication *locApplication = dynamic_cast<DApplication*> (locEventLoop->GetJApplication()); |
18 | DGeometry *locGeometry = locApplication ? locApplication->GetDGeometry(locEventLoop->GetJEvent().GetRunNumber()) : NULL; |
19 | locGeometry->GetTargetZ(dTargetZCenter); |
20 | |
21 | |
22 | dMagneticFieldMap = locApplication->GetBfield(locEventLoop->GetJEvent().GetRunNumber()); |
23 | dIsNoFieldFlag = (dynamic_cast<const DMagneticFieldMapNoField*>(dMagneticFieldMap) != NULL); |
24 | |
25 | |
26 | |
27 | dTrackSelectionTag = "PreSelect"; |
28 | dShowerSelectionTag = "PreSelect"; |
29 | gPARMS->SetDefaultParameter("COMBO:TRACK_SELECT_TAG", dTrackSelectionTag); |
30 | gPARMS->SetDefaultParameter("COMBO:SHOWER_SELECT_TAG", dShowerSelectionTag); |
31 | } |
32 | |
33 | bool DAnalysisUtilities::Check_IsBDTSignalEvent(JEventLoop* locEventLoop, const DReaction* locReaction, bool locExclusiveMatchFlag, bool locIncludeDecayingToReactionFlag) const |
34 | { |
35 | #ifdef VTRACE |
36 | VT_TRACER("DAnalysisUtilities::Check_IsBDTSignalEvent()"); |
37 | #endif |
38 | |
39 | |
40 | |
41 | |
42 | |
43 | |
44 | |
45 | |
46 | |
47 | |
48 | |
49 | if(dParticleComboCreator == nullptr) |
50 | dParticleComboCreator = new DParticleComboCreator(locEventLoop, nullptr, nullptr, nullptr); |
51 | DReaction_factory_Thrown* dThrownReactionFactory = static_cast<DReaction_factory_Thrown*>(locEventLoop->GetFactory("DReaction", "Thrown")); |
52 | if(!dThrownReactionFactory->brun_was_called()) |
53 | { |
54 | dThrownReactionFactory->brun(locEventLoop, locEventLoop->GetJEvent().GetRunNumber()); |
55 | dThrownReactionFactory->Set_brun_called(); |
56 | } |
57 | |
58 | vector<const DReaction*> locThrownReactions; |
59 | locEventLoop->Get(locThrownReactions, "Thrown"); |
60 | if(locThrownReactions.empty()) |
61 | return false; |
62 | auto locActualThrownReaction = locThrownReactions[0]; |
63 | |
64 | |
65 | size_t locStepIndex = 0; |
66 | vector<DReactionStep> locNewReactionSteps; |
67 | const DReaction* locCurrentReaction = locReaction; |
68 | DReaction locNewReaction("Interim"); |
69 | |
70 | do |
71 | { |
72 | const DReactionStep* locReactionStep = locCurrentReaction->Get_ReactionStep(locStepIndex); |
73 | Particle_t locInitialPID = locReactionStep->Get_InitialPID(); |
74 | if((locInitialPID != omega) && (locInitialPID != phiMeson)) |
75 | { |
76 | ++locStepIndex; |
77 | continue; |
78 | } |
79 | |
80 | |
81 | auto locDecayProducts = locReactionStep->Get_FinalPIDs(); |
82 | |
83 | |
84 | for(size_t loc_i = 0; loc_i < locStepIndex; ++loc_i) |
85 | { |
86 | const DReactionStep* locProductionStep = locCurrentReaction->Get_ReactionStep(loc_i); |
87 | auto locPIDs = locProductionStep->Get_FinalPIDs(); |
88 | |
89 | |
90 | bool locFoundFlag = false; |
91 | for(size_t loc_j = 0; loc_j < locPIDs.size(); ++loc_j) |
92 | { |
93 | Particle_t locFinalStatePID = locPIDs[loc_j]; |
94 | if(locFinalStatePID != locInitialPID) |
95 | continue; |
96 | locPIDs.erase(locPIDs.begin() + loc_j); |
97 | locPIDs.insert(locPIDs.begin(), locDecayProducts.begin(), locDecayProducts.end()); |
98 | locFoundFlag = true; |
99 | break; |
100 | } |
101 | if(!locFoundFlag) |
102 | continue; |
103 | |
104 | |
105 | DReactionStep locNewReactionStep; |
106 | locNewReactionStep.Set_InitialParticleID(locProductionStep->Get_InitialPID()); |
107 | locNewReactionStep.Set_TargetParticleID(locProductionStep->Get_TargetPID()); |
108 | int locMissingParticleIndex = locProductionStep->Get_MissingParticleIndex(); |
109 | for(int loc_j = 0; loc_j < int(locPIDs.size()); ++loc_j) |
110 | locNewReactionStep.Add_FinalParticleID(locPIDs[loc_j], (loc_j == locMissingParticleIndex)); |
111 | locNewReactionSteps.push_back(locNewReactionStep); |
112 | |
113 | |
114 | DReaction locReactionBuild("Build"); |
115 | locReactionBuild.Clear_ReactionSteps(); |
116 | for(size_t loc_j = 0; loc_j < locCurrentReaction->Get_NumReactionSteps(); ++loc_j) |
117 | { |
118 | if(loc_j == loc_i) |
119 | locReactionBuild.Add_ReactionStep(&locNewReactionSteps.back()); |
120 | else if(loc_j == locStepIndex) |
121 | continue; |
122 | else |
123 | locReactionBuild.Add_ReactionStep(locCurrentReaction->Get_ReactionStep(loc_j)); |
124 | } |
125 | locNewReaction = locReactionBuild; |
126 | locCurrentReaction = &locNewReaction; |
127 | break; |
128 | } |
129 | } |
130 | while(locStepIndex < locCurrentReaction->Get_NumReactionSteps()); |
131 | |
132 | |
133 | deque<pair<const DMCThrown*, deque<const DMCThrown*> > > locThrownSteps; |
134 | Get_ThrownParticleSteps(locEventLoop, locThrownSteps); |
135 | |
136 | if(locThrownSteps.size() == 1) |
137 | return Check_ThrownsMatchReaction(locEventLoop, locCurrentReaction, locExclusiveMatchFlag); |
138 | |
139 | |
140 | map<Particle_t, size_t> locNumDecayingParticles_Thrown; |
141 | for(size_t loc_i = 0; loc_i < locThrownSteps.size(); ++loc_i) |
142 | { |
143 | if(locThrownSteps[loc_i].first == NULL) |
144 | continue; |
145 | Particle_t locPID = locThrownSteps[loc_i].first->PID(); |
146 | if(locNumDecayingParticles_Thrown.find(locPID) == locNumDecayingParticles_Thrown.end()) |
147 | locNumDecayingParticles_Thrown[locPID] = 1; |
148 | else |
149 | ++locNumDecayingParticles_Thrown[locPID]; |
150 | } |
151 | |
152 | map<Particle_t, size_t> locNumDecayingParticles_Reaction; |
153 | for(size_t loc_i = 0; loc_i < locCurrentReaction->Get_NumReactionSteps(); ++loc_i) |
154 | { |
155 | Particle_t locPID = locCurrentReaction->Get_ReactionStep(loc_i)->Get_InitialPID(); |
156 | if(locPID == Gamma) |
157 | continue; |
158 | if(locNumDecayingParticles_Reaction.find(locPID) == locNumDecayingParticles_Reaction.end()) |
159 | locNumDecayingParticles_Reaction[locPID] = 1; |
160 | else |
161 | ++locNumDecayingParticles_Reaction[locPID]; |
162 | } |
163 | |
164 | |
165 | map<Particle_t, size_t>::iterator locIterator = locNumDecayingParticles_Reaction.begin(); |
166 | for(; locIterator != locNumDecayingParticles_Reaction.end(); ++locIterator) |
167 | { |
168 | Particle_t locPID = locIterator->first; |
169 | if(locNumDecayingParticles_Thrown.find(locPID) == locNumDecayingParticles_Thrown.end()) |
170 | return false; |
171 | } |
172 | |
173 | |
174 | |
175 | locStepIndex = 1; |
176 | do |
177 | { |
178 | pair<const DMCThrown*, deque<const DMCThrown*> > locStepPair = locThrownSteps[locStepIndex]; |
179 | |
180 | |
181 | const DMCThrown* locThrownParent = locStepPair.first; |
182 | Particle_t locInitialPID = locThrownParent->PID(); |
183 | bool locInitialPIDFoundFlag = (locNumDecayingParticles_Reaction.find(locInitialPID) != locNumDecayingParticles_Reaction.end()); |
184 | |
185 | |
186 | |
187 | if(((!locInitialPIDFoundFlag) && locIncludeDecayingToReactionFlag) || (locInitialPID == omega) || (locInitialPID == phiMeson)) |
188 | { |
189 | Replace_DecayingParticleWithProducts(locThrownSteps, locStepIndex); |
190 | if(locNumDecayingParticles_Thrown[locInitialPID] == 1) |
191 | locNumDecayingParticles_Thrown.erase(locInitialPID); |
192 | else |
193 | --locNumDecayingParticles_Thrown[locInitialPID]; |
194 | } |
195 | else |
196 | ++locStepIndex; |
197 | } |
198 | while(locStepIndex < locThrownSteps.size()); |
199 | |
200 | if(!locIncludeDecayingToReactionFlag) |
201 | { |
202 | |
203 | DReaction* locThrownReaction = dThrownReactionFactory->Build_ThrownReaction(locEventLoop, locThrownSteps); |
204 | auto locThrownCombo = dParticleComboCreator->Build_ThrownCombo(locEventLoop, locThrownReaction, locThrownSteps); |
205 | bool locCheckResult = Check_ThrownsMatchReaction(locActualThrownReaction, locThrownCombo, locCurrentReaction, locExclusiveMatchFlag); |
206 | |
207 | dParticleComboCreator->Reset(); |
208 | dThrownReactionFactory->Recycle_Reaction(locThrownReaction); |
209 | return locCheckResult; |
210 | } |
211 | |
212 | |
213 | if(locExclusiveMatchFlag) |
214 | { |
215 | if(locReaction->Get_NumReactionSteps() != locThrownSteps.size()) |
216 | return false; |
217 | } |
218 | |
219 | |
220 | |
221 | |
222 | |
223 | |
224 | vector<Particle_t> locPIDVector; |
225 | vector<int> locResumeAtIndex; |
226 | for(locIterator = locNumDecayingParticles_Reaction.begin(); locIterator != locNumDecayingParticles_Reaction.end(); ++locIterator) |
227 | { |
228 | Particle_t locPID = locIterator->first; |
229 | size_t locNumThrown = locNumDecayingParticles_Thrown[locPID]; |
230 | if(locNumThrown < locIterator->second) |
231 | return false; |
232 | if(locNumThrown == locIterator->second) |
233 | continue; |
234 | for(size_t loc_i = 0; loc_i < locNumThrown - locIterator->second; ++loc_i) |
235 | { |
236 | locPIDVector.push_back(locPID); |
237 | locResumeAtIndex.push_back(0); |
238 | } |
239 | } |
240 | |
241 | |
242 | if(locPIDVector.empty()) |
243 | { |
244 | DReaction* locThrownReaction = dThrownReactionFactory->Build_ThrownReaction(locEventLoop, locThrownSteps); |
245 | auto locThrownCombo = dParticleComboCreator->Build_ThrownCombo(locEventLoop, locThrownReaction, locThrownSteps); |
246 | bool locCheckResult = Check_ThrownsMatchReaction(locActualThrownReaction, locThrownCombo, locCurrentReaction, locExclusiveMatchFlag); |
247 | |
248 | dParticleComboCreator->Reset(); |
249 | dThrownReactionFactory->Recycle_Reaction(locThrownReaction); |
250 | return locCheckResult; |
251 | } |
252 | |
253 | |
254 | |
255 | |
256 | deque<int> locDecayReplacementIndices; |
257 | int locParticleIndex = 0; |
258 | |
259 | |
260 | deque<deque<pair<const DMCThrown*, deque<const DMCThrown*> > > > locReplacementThrownSteps(1, locThrownSteps); |
261 | do |
262 | { |
263 | if(locParticleIndex == -1) |
264 | break; |
265 | |
266 | deque<pair<const DMCThrown*, deque<const DMCThrown*> > > locCurrentThrownSteps = locReplacementThrownSteps.back(); |
267 | if(locParticleIndex == int(locPIDVector.size())) |
268 | { |
269 | |
270 | DReaction* locThrownReaction = dThrownReactionFactory->Build_ThrownReaction(locEventLoop, locCurrentThrownSteps); |
271 | auto locThrownCombo = dParticleComboCreator->Build_ThrownCombo(locEventLoop, locThrownReaction, locCurrentThrownSteps); |
272 | bool locCheckResult = Check_ThrownsMatchReaction(locActualThrownReaction, locThrownCombo, locCurrentReaction, locExclusiveMatchFlag); |
273 | |
274 | dParticleComboCreator->Reset(); |
275 | dThrownReactionFactory->Recycle_Reaction(locThrownReaction); |
276 | |
277 | if(locCheckResult) |
278 | return true; |
279 | locReplacementThrownSteps.pop_back(); |
280 | --locParticleIndex; |
281 | continue; |
282 | } |
283 | |
284 | Particle_t locPID = locPIDVector[locParticleIndex]; |
285 | |
286 | bool locFoundFlag = false; |
287 | for(size_t loc_i = locResumeAtIndex[locParticleIndex]; loc_i < locCurrentThrownSteps.size(); ++loc_i) |
288 | { |
289 | if(locCurrentThrownSteps[loc_i].first == NULL) |
290 | continue; |
291 | Particle_t locInitialPID = locCurrentThrownSteps[loc_i].first->PID(); |
292 | if(locInitialPID != locPID) |
293 | continue; |
294 | locFoundFlag = true; |
295 | |
296 | Replace_DecayingParticleWithProducts(locCurrentThrownSteps, loc_i); |
297 | locReplacementThrownSteps.push_back(locCurrentThrownSteps); |
298 | locResumeAtIndex[locParticleIndex] = loc_i + 1; |
299 | |
300 | break; |
301 | } |
302 | |
303 | if(locFoundFlag) |
304 | ++locParticleIndex; |
305 | else |
306 | { |
307 | locResumeAtIndex[locParticleIndex] = 0; |
308 | locReplacementThrownSteps.pop_back(); |
309 | --locParticleIndex; |
310 | } |
311 | } |
312 | while(true); |
313 | |
314 | return false; |
315 | } |
316 | |
317 | void DAnalysisUtilities::Replace_DecayingParticleWithProducts(deque<pair<const DMCThrown*, deque<const DMCThrown*> > >& locThrownSteps, size_t locStepIndex) const |
318 | { |
319 | if(locStepIndex >= locThrownSteps.size()) |
320 | return; |
321 | |
322 | |
323 | int locInitialParticleDecayFromStepIndex = -1; |
324 | const DMCThrown* locThrownParent = locThrownSteps[locStepIndex].first; |
325 | if(locThrownParent == NULL) |
326 | return; |
327 | |
328 | for(size_t loc_j = 0; loc_j < locStepIndex; ++loc_j) |
329 | { |
330 | for(size_t loc_k = 0; loc_k < locThrownSteps[loc_j].second.size(); ++loc_k) |
331 | { |
332 | if(locThrownParent != locThrownSteps[loc_j].second[loc_k]) |
333 | continue; |
334 | locInitialParticleDecayFromStepIndex = loc_j; |
335 | break; |
336 | } |
337 | if(locInitialParticleDecayFromStepIndex != -1) |
338 | break; |
339 | } |
340 | |
341 | if(locInitialParticleDecayFromStepIndex == -1) |
342 | { |
343 | cout << "ERROR: SOMETHING IS WRONG WITH DAnalysisUtilities::Replace_DecayingParticleWithProducts(). ABORTING." << endl; |
344 | abort(); |
345 | } |
346 | |
347 | |
348 | deque<const DMCThrown*>& locProductionStepFinalParticles = locThrownSteps[locInitialParticleDecayFromStepIndex].second; |
349 | deque<const DMCThrown*>& locDecayStepFinalParticles = locThrownSteps[locStepIndex].second; |
350 | for(size_t loc_j = 0; loc_j < locProductionStepFinalParticles.size(); ++loc_j) |
351 | { |
352 | if(locProductionStepFinalParticles[loc_j] != locThrownParent) |
353 | continue; |
354 | locProductionStepFinalParticles.erase(locProductionStepFinalParticles.begin() + loc_j); |
355 | locProductionStepFinalParticles.insert(locProductionStepFinalParticles.end(), locDecayStepFinalParticles.begin(), locDecayStepFinalParticles.end()); |
356 | break; |
357 | } |
358 | |
359 | locThrownSteps.erase(locThrownSteps.begin() + locStepIndex); |
360 | } |
361 | |
362 | bool DAnalysisUtilities::Check_ThrownsMatchReaction(JEventLoop* locEventLoop, const DReaction* locReaction, bool locExclusiveMatchFlag) const |
363 | { |
364 | |
365 | |
366 | |
367 | |
368 | if(dParticleComboCreator == nullptr) |
369 | dParticleComboCreator = new DParticleComboCreator(locEventLoop, nullptr, nullptr, nullptr); |
370 | auto locThrownCombo = dParticleComboCreator->Build_ThrownCombo(locEventLoop); |
371 | |
372 | vector<const DReaction*> locReactions; |
373 | locEventLoop->Get(locReactions, "Thrown"); |
374 | if(locReactions.empty()) |
375 | return false; |
376 | |
377 | auto locResult = Check_ThrownsMatchReaction(locReactions[0], locThrownCombo, locReaction, locExclusiveMatchFlag); |
378 | dParticleComboCreator->Reset(); |
379 | return locResult; |
380 | } |
381 | |
382 | bool DAnalysisUtilities::Check_ThrownsMatchReaction(const DReaction* locThrownReaction, const DParticleCombo* locThrownCombo, const DReaction* locReaction, bool locExclusiveMatchFlag) const |
383 | { |
384 | #ifdef VTRACE |
385 | VT_TRACER("DAnalysisUtilities::Check_ThrownsMatchReaction()"); |
386 | #endif |
387 | |
388 | |
389 | |
390 | |
391 | |
392 | |
393 | if(locThrownCombo == NULL) |
394 | return false; |
395 | if(locExclusiveMatchFlag) |
396 | { |
397 | if(locReaction->Get_NumReactionSteps() != locThrownCombo->Get_NumParticleComboSteps()) |
398 | return false; |
399 | } |
400 | |
401 | |
402 | map<size_t, int> locReactionInitialParticleDecayFromStepIndexMap; |
403 | for(size_t loc_i = 0; loc_i < locReaction->Get_NumReactionSteps(); ++loc_i) |
404 | { |
405 | const DReactionStep* locReactionStep = locReaction->Get_ReactionStep(loc_i); |
406 | if(loc_i == 0) |
407 | { |
408 | if(locReactionStep->Get_InitialPID() == Gamma) |
409 | locReactionInitialParticleDecayFromStepIndexMap[0] = -1; |
410 | } |
411 | |
412 | for(size_t loc_j = 0; loc_j < locReactionStep->Get_NumFinalPIDs(); ++loc_j) |
413 | { |
414 | Particle_t locFinalStatePID = locReactionStep->Get_FinalPID(loc_j); |
415 | |
416 | for(size_t loc_k = loc_i; loc_k < locReaction->Get_NumReactionSteps(); ++loc_k) |
417 | { |
418 | if(locReaction->Get_ReactionStep(loc_k)->Get_InitialPID() != locFinalStatePID) |
419 | continue; |
420 | if(locReactionInitialParticleDecayFromStepIndexMap.find(loc_k) != locReactionInitialParticleDecayFromStepIndexMap.end()) |
421 | continue; |
422 | locReactionInitialParticleDecayFromStepIndexMap[loc_k] = loc_i; |
423 | break; |
424 | } |
425 | } |
426 | } |
427 | |
428 | |
429 | set<size_t> locMatchedInputStepIndices; |
430 | map<int, int> locStepMatching; |
431 | map<int, int> locReverseStepMatching; |
432 | for(size_t loc_i = 0; loc_i < locThrownCombo->Get_NumParticleComboSteps(); ++loc_i) |
433 | { |
434 | int locInitialParticleDecayFromStepIndex = DAnalysis::Get_InitialParticleDecayFromIndices(locThrownReaction, loc_i).first; |
435 | |
436 | |
437 | size_t locStartSearchIndex = 0; |
438 | if(locReverseStepMatching.find(locInitialParticleDecayFromStepIndex) != locReverseStepMatching.end()) |
439 | locStartSearchIndex = locReverseStepMatching[locInitialParticleDecayFromStepIndex] + 1; |
440 | |
441 | |
442 | bool locMatchFoundFlag = false; |
443 | int locPossibleMatchIndex = -1; |
444 | for(size_t loc_j = locStartSearchIndex; loc_j < locReaction->Get_NumReactionSteps(); ++loc_j) |
445 | { |
446 | const DReactionStep* locReactionStep = locReaction->Get_ReactionStep(loc_j); |
447 | if(locMatchedInputStepIndices.find(loc_j) != locMatchedInputStepIndices.end()) |
448 | continue; |
449 | |
450 | |
451 | if(!DAnalysis::Check_ChannelEquality(locThrownReaction->Get_ReactionStep(loc_i), locReactionStep, false, !locExclusiveMatchFlag)) |
452 | continue; |
453 | |
454 | |
455 | if(locReactionInitialParticleDecayFromStepIndexMap.find(loc_j) == locReactionInitialParticleDecayFromStepIndexMap.end()) |
456 | { |
457 | |
458 | |
459 | locPossibleMatchIndex = loc_j; |
460 | continue; |
461 | } |
462 | |
463 | int locReactionInitialParticleDecayFromStepIndex = locReactionInitialParticleDecayFromStepIndexMap[loc_j]; |
464 | if(locReactionInitialParticleDecayFromStepIndex != -1) |
465 | { |
466 | if(locStepMatching.find(locReactionInitialParticleDecayFromStepIndex) == locStepMatching.end()) |
467 | continue; |
468 | int locReactionInitialParticleDecayFromStepIndexMappedBackToThrown = locStepMatching[locReactionInitialParticleDecayFromStepIndex]; |
469 | if(locInitialParticleDecayFromStepIndex != locReactionInitialParticleDecayFromStepIndexMappedBackToThrown) |
470 | continue; |
471 | } |
472 | else if(locInitialParticleDecayFromStepIndex != -1) |
473 | continue; |
474 | |
475 | |
476 | locMatchedInputStepIndices.insert(loc_j); |
477 | locStepMatching[loc_j] = loc_i; |
478 | locReverseStepMatching[loc_i] = loc_j; |
479 | locMatchFoundFlag = true; |
480 | break; |
481 | } |
482 | |
483 | if((!locMatchFoundFlag) && (locPossibleMatchIndex != -1)) |
484 | { |
485 | |
486 | locMatchedInputStepIndices.insert(locPossibleMatchIndex); |
487 | locStepMatching[locPossibleMatchIndex] = loc_i; |
488 | locReverseStepMatching[loc_i] = locPossibleMatchIndex; |
489 | locMatchFoundFlag = true; |
490 | } |
491 | if(locExclusiveMatchFlag && (!locMatchFoundFlag)) |
492 | return false; |
493 | } |
494 | if(locExclusiveMatchFlag) |
495 | return true; |
496 | |
497 | |
498 | for(size_t loc_i = 0; loc_i < locReaction->Get_NumReactionSteps(); ++loc_i) |
499 | { |
500 | if(locMatchedInputStepIndices.find(loc_i) == locMatchedInputStepIndices.end()) |
501 | return false; |
502 | } |
503 | |
504 | return true; |
505 | } |
506 | |
507 | void DAnalysisUtilities::Get_UnusedChargedTracks(JEventLoop* locEventLoop, const DParticleCombo* locParticleCombo, vector<const DChargedTrack*>& locUnusedChargedTracks) const |
508 | { |
509 | locUnusedChargedTracks.clear(); |
510 | locEventLoop->Get(locUnusedChargedTracks, "Combo"); |
511 | std::sort(locUnusedChargedTracks.begin(), locUnusedChargedTracks.end()); |
512 | |
513 | auto locChargedSourceObjects = locParticleCombo->Get_FinalParticle_SourceObjects(d_Charged); |
514 | for(size_t loc_i = 0; loc_i < locChargedSourceObjects.size(); ++loc_i) |
515 | { |
516 | for(auto locIterator = locUnusedChargedTracks.begin(); locIterator != locUnusedChargedTracks.end();) |
517 | { |
518 | if(locChargedSourceObjects[loc_i] != *locIterator) |
519 | ++locIterator; |
520 | else |
521 | locIterator = locUnusedChargedTracks.erase(locIterator); |
522 | } |
523 | } |
524 | } |
525 | |
526 | void DAnalysisUtilities::Get_UnusedTimeBasedTracks(JEventLoop* locEventLoop, const DParticleCombo* locParticleCombo, vector<const DTrackTimeBased*>& locUnusedTimeBasedTracks) const |
527 | { |
528 | locUnusedTimeBasedTracks.clear(); |
529 | locEventLoop->Get(locUnusedTimeBasedTracks); |
530 | |
531 | vector<const DTrackTimeBased*> locComboTimeBasedTracks; |
532 | locEventLoop->Get(locComboTimeBasedTracks, "Combo"); |
533 | locUnusedTimeBasedTracks.insert(locUnusedTimeBasedTracks.end(), locComboTimeBasedTracks.begin(), locComboTimeBasedTracks.end()); |
534 | |
535 | auto locChargedSourceObjects = locParticleCombo->Get_FinalParticle_SourceObjects(d_Charged); |
536 | for(size_t loc_i = 0; loc_i < locChargedSourceObjects.size(); ++loc_i) |
537 | { |
538 | |
539 | auto locChargedTrack = static_cast<const DChargedTrack*>(locChargedSourceObjects[loc_i]); |
540 | for(auto locIterator = locUnusedTimeBasedTracks.begin(); locIterator != locUnusedTimeBasedTracks.end();) |
541 | { |
542 | if(locChargedTrack->candidateid != (*locIterator)->candidateid) |
543 | ++locIterator; |
544 | else |
545 | locIterator = locUnusedTimeBasedTracks.erase(locIterator); |
546 | } |
547 | } |
548 | } |
549 | |
550 | void DAnalysisUtilities::Get_UnusedWireBasedTracks(JEventLoop* locEventLoop, const DParticleCombo* locParticleCombo, vector<const DTrackWireBased*>& locUnusedWireBasedTracks) const |
551 | { |
552 | locUnusedWireBasedTracks.clear(); |
553 | locEventLoop->Get(locUnusedWireBasedTracks); |
554 | |
555 | auto locChargedSourceObjects = locParticleCombo->Get_FinalParticle_SourceObjects(d_Charged); |
556 | for(size_t loc_i = 0; loc_i < locChargedSourceObjects.size(); ++loc_i) |
557 | { |
558 | |
559 | auto locChargedTrack = static_cast<const DChargedTrack*>(locChargedSourceObjects[loc_i]); |
560 | for(auto locIterator = locUnusedWireBasedTracks.begin(); locIterator != locUnusedWireBasedTracks.end();) |
561 | { |
562 | if(locChargedTrack->candidateid != (*locIterator)->candidateid) |
563 | ++locIterator; |
564 | else |
565 | locIterator = locUnusedWireBasedTracks.erase(locIterator); |
566 | } |
567 | } |
568 | } |
569 | |
570 | void DAnalysisUtilities::Get_UnusedTrackCandidates(JEventLoop* locEventLoop, const DParticleCombo* locParticleCombo, vector<const DTrackCandidate*>& locUnusedTrackCandidates) const |
571 | { |
572 | locUnusedTrackCandidates.clear(); |
573 | locEventLoop->Get(locUnusedTrackCandidates); |
| 1 | Calling 'JEventLoop::Get' | |
|
574 | |
575 | auto locChargedSourceObjects = locParticleCombo->Get_FinalParticle_SourceObjects(d_Charged); |
576 | set<unsigned int> locUsedCandidateIndices; |
577 | for(size_t loc_i = 0; loc_i < locChargedSourceObjects.size(); ++loc_i) |
578 | { |
579 | |
580 | auto locChargedTrack = static_cast<const DChargedTrack*>(locChargedSourceObjects[loc_i]); |
581 | locUsedCandidateIndices.insert(locChargedTrack->candidateid - 1); |
582 | } |
583 | |
584 | for(int loc_i = locUnusedTrackCandidates.size() - 1; loc_i >= 0; --loc_i) |
585 | { |
586 | if(locUsedCandidateIndices.find(loc_i) != locUsedCandidateIndices.end()) |
587 | locUnusedTrackCandidates.erase(locUnusedTrackCandidates.begin() + loc_i); |
588 | } |
589 | } |
590 | |
591 | void DAnalysisUtilities::Get_UnusedNeutralShowers(JEventLoop* locEventLoop, const DParticleCombo* locParticleCombo, vector<const DNeutralShower*>& locUnusedNeutralShowers) const |
592 | { |
593 | locUnusedNeutralShowers.clear(); |
594 | locEventLoop->Get(locUnusedNeutralShowers, dShowerSelectionTag.c_str()); |
595 | |
596 | auto locNeutralSourceObjects = locParticleCombo->Get_FinalParticle_SourceObjects(d_Neutral); |
597 | for(size_t loc_i = 0; loc_i < locNeutralSourceObjects.size(); ++loc_i) |
598 | { |
599 | auto locNeutralShower = static_cast<const DNeutralShower*>(locNeutralSourceObjects[loc_i]); |
600 | for(auto locIterator = locUnusedNeutralShowers.begin(); locIterator != locUnusedNeutralShowers.end();) |
601 | { |
602 | if(locNeutralShower != *locIterator) |
603 | ++locIterator; |
604 | else |
605 | locIterator = locUnusedNeutralShowers.erase(locIterator); |
606 | } |
607 | } |
608 | } |
609 | |
610 | void DAnalysisUtilities::Get_UnusedNeutralParticles(JEventLoop* locEventLoop, const DParticleCombo* locParticleCombo, vector<const DNeutralParticle*>& locUnusedNeutralParticles) const |
611 | { |
612 | locUnusedNeutralParticles.clear(); |
613 | locEventLoop->Get(locUnusedNeutralParticles, dShowerSelectionTag.c_str()); |
614 | |
615 | auto locNeutralSourceObjects = locParticleCombo->Get_FinalParticle_SourceObjects(d_Neutral); |
616 | for(size_t loc_i = 0; loc_i < locNeutralSourceObjects.size(); ++loc_i) |
617 | { |
618 | auto locNeutralShower = static_cast<const DNeutralShower*>(locNeutralSourceObjects[loc_i]); |
619 | for(auto locIterator = locUnusedNeutralParticles.begin(); locIterator != locUnusedNeutralParticles.end();) |
620 | { |
621 | if(locNeutralShower != (*locIterator)->dNeutralShower) |
622 | ++locIterator; |
623 | else |
624 | locIterator = locUnusedNeutralParticles.erase(locIterator); |
625 | } |
626 | } |
627 | } |
628 | |
629 | void DAnalysisUtilities::Get_ThrownParticleSteps(JEventLoop* locEventLoop, deque<pair<const DMCThrown*, deque<const DMCThrown*> > >& locThrownSteps) const |
630 | { |
631 | vector<const DMCThrown*> locMCThrowns; |
632 | locEventLoop->Get(locMCThrowns); |
633 | |
634 | map<size_t, const DMCThrown*> locIDMap; |
635 | for(size_t loc_i = 0; loc_i < locMCThrowns.size(); ++loc_i) |
636 | locIDMap[locMCThrowns[loc_i]->myid] = locMCThrowns[loc_i]; |
637 | |
638 | locThrownSteps.clear(); |
639 | if(locMCThrowns.empty()) |
640 | return; |
641 | locThrownSteps.push_back(pair<const DMCThrown*, deque<const DMCThrown*> >(NULL, deque<const DMCThrown*>() ) ); |
642 | |
643 | for(size_t loc_i = 0; loc_i < locMCThrowns.size(); ++loc_i) |
644 | { |
645 | if(IsResonance(locMCThrowns[loc_i]->PID())) |
646 | continue; |
647 | |
648 | if(locMCThrowns[loc_i]->PID() == Unknown) |
649 | continue; |
650 | |
651 | |
652 | int locParentID = locMCThrowns[loc_i]->parentid; |
653 | if(locParentID == 0) |
654 | { |
655 | locThrownSteps[0].second.push_back(locMCThrowns[loc_i]); |
656 | continue; |
657 | } |
658 | if(locIDMap.find(locParentID) == locIDMap.end()) |
659 | continue; |
660 | |
661 | |
662 | Particle_t locParentPID = locIDMap[locParentID]->PID(); |
663 | bool locDoneFlag = false; |
664 | while(((locParentPID == Unknown) || IsResonance(locParentPID)) && (!locDoneFlag)) |
665 | { |
666 | |
667 | locParentID = locIDMap[locParentID]->parentid; |
668 | if(locParentID == 0) |
669 | { |
670 | locThrownSteps[0].second.push_back(locMCThrowns[loc_i]); |
671 | locDoneFlag = true; |
672 | } |
673 | else if(locIDMap.find(locParentID) == locIDMap.end()) |
674 | locDoneFlag = true; |
675 | else |
676 | locParentPID = locIDMap[locParentID]->PID(); |
677 | } |
678 | |
679 | if(Is_FinalStateParticle(locParentPID) == 1) |
680 | continue; |
681 | |
682 | |
683 | bool locListedAsDecayingFlag = false; |
684 | for(size_t loc_j = 1; loc_j < locThrownSteps.size(); ++loc_j) |
685 | { |
686 | if(locThrownSteps[loc_j].first->myid != locParentID) |
687 | continue; |
688 | locThrownSteps[loc_j].second.push_back(locMCThrowns[loc_i]); |
689 | locListedAsDecayingFlag = true; |
690 | break; |
691 | } |
692 | if(locListedAsDecayingFlag) |
693 | continue; |
694 | |
695 | |
696 | |
697 | const DMCThrown* locThrownParent = locIDMap[locParentID]; |
698 | bool locFoundFlag = false; |
699 | for(size_t loc_j = 0; loc_j < locThrownSteps.size(); ++loc_j) |
700 | { |
701 | for(size_t loc_k = 0; loc_k < locThrownSteps[loc_j].second.size(); ++loc_k) |
702 | { |
703 | if(locThrownSteps[loc_j].second[loc_k] != locThrownParent) |
704 | continue; |
705 | locFoundFlag = true; |
706 | break; |
707 | } |
708 | if(locFoundFlag) |
709 | break; |
710 | } |
711 | if(!locFoundFlag) |
712 | continue; |
713 | |
714 | |
715 | locThrownSteps.push_back(pair<const DMCThrown*, deque<const DMCThrown*> >(locThrownParent, deque<const DMCThrown*>(1, locMCThrowns[loc_i]) )); |
716 | } |
717 | |
718 | |
719 | |
720 | |
721 | |
722 | |
723 | |
724 | |
725 | |
726 | |
727 | |
728 | } |
729 | |
730 | bool DAnalysisUtilities::Are_ThrownPIDsSameAsDesired(JEventLoop* locEventLoop, const deque<Particle_t>& locDesiredPIDs, Particle_t locMissingPID) const |
731 | { |
732 | vector<const DMCThrown*> locMCThrowns; |
733 | locEventLoop->Get(locMCThrowns, "FinalState"); |
734 | deque<Particle_t> locDesiredPIDs_Copy = locDesiredPIDs; |
735 | |
736 | bool locMissingPIDMatchedFlag = false; |
737 | for(size_t loc_i = 0; loc_i < locMCThrowns.size(); ++loc_i) |
738 | { |
739 | Particle_t locPID = (Particle_t)(locMCThrowns[loc_i]->type); |
740 | |
741 | if((!locMissingPIDMatchedFlag) && (locMissingPID == locPID)) |
742 | { |
743 | |
744 | locMissingPIDMatchedFlag = true; |
745 | continue; |
746 | } |
747 | |
748 | bool locPIDFoundFlag = false; |
749 | for(deque<Particle_t>::iterator locIterator = locDesiredPIDs_Copy.begin(); locIterator != locDesiredPIDs_Copy.end(); ++locIterator) |
750 | { |
751 | if(*locIterator != locPID) |
752 | continue; |
753 | locDesiredPIDs_Copy.erase(locIterator); |
754 | locPIDFoundFlag = true; |
755 | break; |
756 | } |
757 | if(!locPIDFoundFlag) |
758 | return false; |
759 | } |
760 | |
761 | return (locDesiredPIDs_Copy.empty()); |
762 | } |
763 | |
764 | DLorentzVector DAnalysisUtilities::Calc_MissingP4(const DReaction* locReaction, const DParticleCombo* locParticleCombo, bool locUseKinFitDataFlag) const |
765 | { |
766 | set<pair<const JObject*, unsigned int> > locSourceObjects; |
767 | return Calc_MissingP4(locReaction, locParticleCombo, 0, -1, set<size_t>(), locSourceObjects, locUseKinFitDataFlag); |
768 | } |
769 | |
770 | DLorentzVector DAnalysisUtilities::Calc_MissingP4(const DReaction* locReaction, const DParticleCombo* locParticleCombo, set<pair<const JObject*, unsigned int> >& locSourceObjects, bool locUseKinFitDataFlag) const |
771 | { |
772 | return Calc_MissingP4(locReaction, locParticleCombo, 0, -1, set<size_t>(), locSourceObjects, locUseKinFitDataFlag); |
773 | } |
774 | |
775 | DLorentzVector DAnalysisUtilities::Calc_MissingP4(const DReaction* locReaction, const DParticleCombo* locParticleCombo, size_t locStepIndex, int locUpToStepIndex, set<size_t> locUpThroughIndices, bool locUseKinFitDataFlag) const |
776 | { |
777 | set<pair<const JObject*, unsigned int> > locSourceObjects; |
778 | return Calc_MissingP4(locReaction, locParticleCombo, locStepIndex, locUpToStepIndex, locUpThroughIndices, locSourceObjects, locUseKinFitDataFlag); |
779 | } |
780 | |
781 | DLorentzVector DAnalysisUtilities::Calc_MissingP4(const DReaction* locReaction, const DParticleCombo* locParticleCombo, size_t locStepIndex, int locUpToStepIndex, set<size_t> locUpThroughIndices, set<pair<const JObject*, unsigned int> >& locSourceObjects, bool locUseKinFitDataFlag) const |
782 | { |
783 | |
784 | |
785 | if(locUseKinFitDataFlag && (locParticleCombo->Get_KinFitResults() == NULL)) |
786 | return Calc_MissingP4(locReaction, locParticleCombo, locStepIndex, locUpToStepIndex, locUpThroughIndices, locSourceObjects, false); |
787 | |
788 | DLorentzVector locMissingP4; |
789 | const DParticleComboStep* locParticleComboStep = locParticleCombo->Get_ParticleComboStep(locStepIndex); |
790 | auto locReactionStep = locReaction->Get_ReactionStep(locStepIndex); |
791 | |
792 | const DKinematicData* locKinematicData = NULL; |
793 | if(locStepIndex == 0) |
794 | { |
795 | |
796 | locKinematicData = locParticleComboStep->Get_InitialParticle_Measured(); |
797 | locSourceObjects.insert(pair<const JObject*, unsigned int>(locKinematicData, abs(PDGtype(locKinematicData->PID())))); |
798 | if(locUseKinFitDataFlag) |
799 | locKinematicData = locParticleComboStep->Get_InitialParticle(); |
800 | locMissingP4 += locKinematicData->lorentzMomentum(); |
801 | } |
802 | |
803 | |
804 | Particle_t locPID = locReactionStep->Get_TargetPID(); |
805 | if(locPID != Unknown) |
806 | { |
807 | double locMass = ParticleMass(locPID); |
808 | locMissingP4 += DLorentzVector(DVector3(0.0, 0.0, 0.0), locMass); |
809 | } |
810 | |
811 | auto locParticles = locUseKinFitDataFlag ? locParticleComboStep->Get_FinalParticles() : locParticleComboStep->Get_FinalParticles_Measured(); |
812 | for(size_t loc_j = 0; loc_j < locParticles.size(); ++loc_j) |
813 | { |
814 | if(int(loc_j) == locReactionStep->Get_MissingParticleIndex()) |
815 | continue; |
816 | if((int(locStepIndex) == locUpToStepIndex) && (locUpThroughIndices.find(loc_j) == locUpThroughIndices.end())) |
817 | continue; |
818 | |
819 | int locDecayStepIndex = DAnalysis::Get_DecayStepIndex(locReaction, locStepIndex, loc_j); |
820 | if(locDecayStepIndex > 0) |
821 | { |
822 | |
823 | locMissingP4 += Calc_MissingP4(locReaction, locParticleCombo, locDecayStepIndex, locUpToStepIndex, locUpThroughIndices, locSourceObjects, locUseKinFitDataFlag); |
824 | } |
825 | else |
826 | { |
827 | Particle_t locPID = locReactionStep->Get_FinalPID(loc_j); |
828 | auto locDetectedP4 = locParticles[loc_j]->lorentzMomentum(); |
829 | locMissingP4 -= locDetectedP4; |
830 | locSourceObjects.insert(pair<const JObject*, unsigned int>(locParticleComboStep->Get_FinalParticle_SourceObject(loc_j), abs(PDGtype(locPID)))); |
831 | } |
832 | } |
833 | |
834 | return locMissingP4; |
835 | } |
836 | |
837 | TMatrixFSym DAnalysisUtilities::Calc_MissingP3Covariance(const DReaction* locReaction, const DParticleCombo* locParticleCombo) const |
838 | { |
839 | |
840 | return Calc_MissingP3Covariance(locReaction, locParticleCombo, 0, -1, set<size_t>()); |
841 | } |
842 | |
843 | TMatrixFSym DAnalysisUtilities::Calc_MissingP3Covariance(const DReaction* locReaction, const DParticleCombo* locParticleCombo, size_t locStepIndex, int locUpToStepIndex, set<size_t> locUpThroughIndices) const |
844 | { |
845 | |
846 | |
847 | |
848 | |
849 | |
850 | |
851 | |
852 | const DParticleComboStep* locParticleComboStep = locParticleCombo->Get_ParticleComboStep(locStepIndex); |
853 | auto locReactionStep = locReaction->Get_ReactionStep(locStepIndex); |
854 | TMatrixFSym locMissingCovarianceMatrix(3); |
855 | locMissingCovarianceMatrix.Zero(); |
856 | |
857 | const DKinematicData* locKinematicData = NULL; |
858 | if(locStepIndex == 0) |
859 | { |
860 | |
861 | locKinematicData = locParticleComboStep->Get_InitialParticle_Measured(); |
862 | TMatrixFSym locParticleCovarianceMatrix = *(locKinematicData->errorMatrix().get()); |
863 | locParticleCovarianceMatrix.ResizeTo(3, 3); |
864 | locMissingCovarianceMatrix += locParticleCovarianceMatrix; |
865 | } |
866 | |
867 | auto locParticles = locParticleComboStep->Get_FinalParticles_Measured(); |
868 | for(size_t loc_j = 0; loc_j < locParticles.size(); ++loc_j) |
869 | { |
870 | if(int(loc_j) == locReactionStep->Get_MissingParticleIndex()) |
871 | continue; |
872 | if((int(locStepIndex) == locUpToStepIndex) && (locUpThroughIndices.find(loc_j) == locUpThroughIndices.end())) |
873 | continue; |
874 | |
875 | int locDecayStepIndex = DAnalysis::Get_DecayStepIndex(locReaction, locStepIndex, loc_j); |
876 | if(locDecayStepIndex > 0) |
877 | locMissingCovarianceMatrix += Calc_MissingP3Covariance(locReaction, locParticleCombo, locDecayStepIndex, locUpToStepIndex, locUpThroughIndices); |
878 | else |
879 | { |
880 | TMatrixFSym locParticleCovarianceMatrix = *(locParticles[loc_j]->errorMatrix().get()); |
881 | locParticleCovarianceMatrix.ResizeTo(3, 3); |
882 | locMissingCovarianceMatrix += locParticleCovarianceMatrix; |
883 | } |
884 | } |
885 | |
886 | return locMissingCovarianceMatrix; |
887 | } |
888 | |
889 | DLorentzVector DAnalysisUtilities::Calc_FinalStateP4(const DReaction* locReaction, const DParticleCombo* locParticleCombo, size_t locStepIndex, bool locUseKinFitDataFlag) const |
890 | { |
891 | set<pair<const JObject*, unsigned int> > locSourceObjects; |
892 | return Calc_FinalStateP4(locReaction, locParticleCombo, locStepIndex, set<size_t>(), locSourceObjects, locUseKinFitDataFlag); |
893 | } |
894 | |
895 | DLorentzVector DAnalysisUtilities::Calc_FinalStateP4(const DReaction* locReaction, const DParticleCombo* locParticleCombo, size_t locStepIndex, set<pair<const JObject*, unsigned int> >& locSourceObjects, bool locUseKinFitDataFlag) const |
896 | { |
897 | return Calc_FinalStateP4(locReaction, locParticleCombo, locStepIndex, set<size_t>(), locSourceObjects, locUseKinFitDataFlag); |
898 | } |
899 | |
900 | DLorentzVector DAnalysisUtilities::Calc_FinalStateP4(const DReaction* locReaction, const DParticleCombo* locParticleCombo, size_t locStepIndex, set<size_t> locToIncludeIndices, bool locUseKinFitDataFlag) const |
901 | { |
902 | set<pair<const JObject*, unsigned int> > locSourceObjects; |
903 | return Calc_FinalStateP4(locReaction, locParticleCombo, locStepIndex, locToIncludeIndices, locSourceObjects, locUseKinFitDataFlag); |
904 | } |
905 | |
906 | DLorentzVector DAnalysisUtilities::Calc_FinalStateP4(const DReaction* locReaction, const DParticleCombo* locParticleCombo, size_t locStepIndex, set<size_t> locToIncludeIndices, set<pair<const JObject*, unsigned int> >& locSourceObjects, bool locUseKinFitDataFlag) const |
907 | { |
908 | if(locUseKinFitDataFlag && (locParticleCombo->Get_KinFitResults() == NULL)) |
909 | return Calc_FinalStateP4(locReaction, locParticleCombo, locStepIndex, locToIncludeIndices, locSourceObjects, false); |
910 | |
911 | DLorentzVector locFinalStateP4; |
912 | const DParticleComboStep* locParticleComboStep = locParticleCombo->Get_ParticleComboStep(locStepIndex); |
913 | if(locParticleComboStep == NULL) |
914 | return (DLorentzVector()); |
915 | auto locReactionStep = locReaction->Get_ReactionStep(locStepIndex); |
916 | |
917 | auto locParticles = locUseKinFitDataFlag ? locParticleComboStep->Get_FinalParticles() : locParticleComboStep->Get_FinalParticles_Measured(); |
918 | |
919 | |
920 | if(locStepIndex != 0) |
921 | { |
922 | Particle_t locPID = locReactionStep->Get_TargetPID(); |
923 | if(locPID != Unknown) |
924 | locFinalStateP4 -= DLorentzVector(DVector3(0.0, 0.0, 0.0), ParticleMass(locPID)); |
925 | } |
926 | |
927 | bool locDoSubsetFlag = !locToIncludeIndices.empty(); |
928 | for(size_t loc_i = 0; loc_i < locParticles.size(); ++loc_i) |
929 | { |
930 | if(locDoSubsetFlag && (locToIncludeIndices.find(loc_i) == locToIncludeIndices.end())) |
931 | continue; |
932 | |
933 | if(locReactionStep->Get_MissingParticleIndex() == int(loc_i)) |
934 | return (DLorentzVector()); |
935 | |
936 | int locDecayStepIndex = DAnalysis::Get_DecayStepIndex(locReaction, locStepIndex, loc_i); |
937 | if(locDecayStepIndex >= 0) |
938 | { |
939 | |
940 | if((!locUseKinFitDataFlag) || (!IsFixedMass(locReactionStep->Get_FinalPID(loc_i)))) |
941 | locFinalStateP4 += Calc_FinalStateP4(locReaction, locParticleCombo, locDecayStepIndex, set<size_t>(), locSourceObjects, locUseKinFitDataFlag); |
942 | else |
943 | { |
944 | locFinalStateP4 += locParticles[loc_i]->lorentzMomentum(); |
945 | |
946 | Calc_FinalStateP4(locReaction, locParticleCombo, locDecayStepIndex, set<size_t>(), locSourceObjects, locUseKinFitDataFlag); |
947 | } |
948 | } |
949 | else |
950 | { |
951 | locFinalStateP4 += locParticles[loc_i]->lorentzMomentum(); |
952 | locSourceObjects.insert(pair<const JObject*, unsigned int>(locParticleComboStep->Get_FinalParticle_SourceObject(loc_i), abs(PDGtype(locParticles[loc_i]->PID())))); |
953 | } |
954 | } |
955 | return locFinalStateP4; |
956 | } |
957 | |
958 | int DAnalysisUtilities::Calc_Energy_UnusedShowers(JEventLoop* locEventLoop, const DParticleCombo* locParticleCombo, double &locEnergy_UnusedShowers, int &locNumber_UnusedShowers_Quality, double &locEnergy_UnusedShowers_Quality) const |
959 | { |
960 | DVector3 locVertex(0.0, 0.0, dTargetZCenter); |
961 | const DEventRFBunch* locEventRFBunch = locParticleCombo->Get_EventRFBunch(); |
962 | double locRFTime = (locEventRFBunch != NULL) ? locEventRFBunch->dTime : numeric_limits<double>::quiet_NaN(); |
963 | |
964 | vector<const DNeutralShower*> locUnusedNeutralShowers; |
965 | Get_UnusedNeutralShowers(locEventLoop, locParticleCombo, locUnusedNeutralShowers); |
966 | |
967 | int locNumber_UnusedShowers = 0; |
968 | locEnergy_UnusedShowers = 0.; |
969 | locEnergy_UnusedShowers_Quality = 0.; |
970 | locNumber_UnusedShowers_Quality = 0; |
971 | for(size_t loc_i = 0; loc_i < locUnusedNeutralShowers.size(); ++loc_i) { |
972 | const DNeutralShower* locUnusedNeutralShower = locUnusedNeutralShowers[loc_i]; |
973 | |
974 | |
975 | double locFlightTime = (locUnusedNeutralShower->dSpacetimeVertex.Vect() - locVertex).Mag()/SPEED_OF_LIGHT; |
976 | double locDeltaT = locUnusedNeutralShower->dSpacetimeVertex.T() - locFlightTime - locRFTime; |
977 | |
978 | |
979 | if(fabs(locDeltaT) > 4.) |
980 | continue; |
981 | |
982 | locEnergy_UnusedShowers += locUnusedNeutralShower->dEnergy; |
983 | locNumber_UnusedShowers++; |
984 | |
985 | if(locUnusedNeutralShower->dQuality > 0.5) { |
986 | locEnergy_UnusedShowers_Quality += locUnusedNeutralShower->dEnergy; |
987 | locNumber_UnusedShowers_Quality++; |
988 | } |
989 | } |
990 | |
991 | return locNumber_UnusedShowers; |
992 | } |
993 | |
994 | int DAnalysisUtilities::Calc_Momentum_UnusedTracks(JEventLoop* locEventLoop, const DParticleCombo* locParticleCombo, double &locSumPMag_UnusedTracks, TVector3 &locSumP3_UnusedTracks) const |
995 | { |
996 | vector<const DChargedTrack*> locUnusedChargedTracks; |
997 | Get_UnusedChargedTracks(locEventLoop, locParticleCombo, locUnusedChargedTracks); |
998 | |
999 | for(size_t loc_i = 0; loc_i < locUnusedChargedTracks.size(); ++loc_i) { |
1000 | const DChargedTrack* locUnusedChargedTrack = locUnusedChargedTracks[loc_i]; |
1001 | const DChargedTrackHypothesis *locUnusedChargedTrackHypothesis = locUnusedChargedTrack->Get_BestTrackingFOM(); |
1002 | |
1003 | locSumPMag_UnusedTracks += locUnusedChargedTrackHypothesis->pmag(); |
1004 | locSumP3_UnusedTracks += locUnusedChargedTrackHypothesis->momentum(); |
1005 | } |
1006 | |
1007 | return (int)locUnusedChargedTracks.size(); |
1008 | } |
1009 | |
1010 | double DAnalysisUtilities::Calc_DOCAToVertex(const DKinematicData* locKinematicData, const DVector3& locVertex) const |
1011 | { |
1012 | DVector3 locPOCA; |
1013 | return Calc_DOCAToVertex(locKinematicData, locVertex, locPOCA); |
1014 | } |
1015 | |
1016 | double DAnalysisUtilities::Calc_DOCAToVertex(const DKinematicData* locKinematicData, const DVector3& locVertex, DVector3& locPOCA) const |
1017 | { |
1018 | DVector3 locUnitDir = (1.0/locKinematicData->momentum().Mag())*locKinematicData->momentum(); |
1019 | DVector3 locPosition = locKinematicData->position(); |
1020 | return Calc_DOCAToVertex(locUnitDir, locPosition, locVertex, locPOCA); |
1021 | } |
1022 | |
1023 | double DAnalysisUtilities::Calc_DOCAToVertex(const DKinFitParticle* locKinFitParticle, const DVector3& locVertex) const |
1024 | { |
1025 | DVector3 locPOCA; |
1026 | return Calc_DOCAToVertex(locKinFitParticle, locVertex, locPOCA); |
1027 | } |
1028 | |
1029 | double DAnalysisUtilities::Calc_DOCAToVertex(const DKinFitParticle* locKinFitParticle, const DVector3& locVertex, DVector3& locPOCA) const |
1030 | { |
1031 | DVector3 locUnitDir(locKinFitParticle->Get_Momentum().Unit().X(),locKinFitParticle->Get_Momentum().Unit().Y(),locKinFitParticle->Get_Momentum().Unit().Z()); |
1032 | DVector3 locPosition(locKinFitParticle->Get_Position().X(),locKinFitParticle->Get_Position().Y(),locKinFitParticle->Get_Position().Z()); |
1033 | return Calc_DOCAToVertex(locUnitDir, locPosition, locVertex, locPOCA); |
1034 | } |
1035 | |
1036 | double DAnalysisUtilities::Calc_DOCAToVertex(const DVector3& locUnitDir, const DVector3& locPosition, const DVector3& locVertex) const |
1037 | { |
1038 | DVector3 locPOCA1, locPOCA2; |
1039 | return Calc_DOCA(locUnitDir, locUnitDir, locPosition, locVertex, locPOCA1, locPOCA2); |
1040 | } |
1041 | |
1042 | double DAnalysisUtilities::Calc_DOCAToVertex(const DVector3& locUnitDir, const DVector3& locPosition, const DVector3& locVertex, DVector3& locPOCA) const |
1043 | { |
1044 | DVector3 locPOCA2; |
1045 | return Calc_DOCA(locUnitDir, locUnitDir, locPosition, locVertex, locPOCA, locPOCA2); |
1046 | } |
1047 | |
1048 | double DAnalysisUtilities::Calc_DOCAVertex(const DKinFitParticle* locKinFitParticle1, const DKinFitParticle* locKinFitParticle2, DVector3& locDOCAVertex) const |
1049 | { |
1050 | DVector3 locPOCA1, locPOCA2; |
1051 | double locDOCA; |
1052 | |
1053 | if (dIsNoFieldFlag==false && Calc_DOCA(locKinFitParticle1,locKinFitParticle2, |
1054 | locPOCA1,locPOCA2,locDOCA)==NOERROR){ |
1055 | locDOCAVertex=0.5*(locPOCA1+locPOCA2); |
1056 | return locDOCA; |
1057 | } |
1058 | |
1059 | |
1060 | DVector3 locUnitDir1(locKinFitParticle1->Get_Momentum().Unit().X(),locKinFitParticle1->Get_Momentum().Unit().Y(),locKinFitParticle1->Get_Momentum().Unit().Z()); |
1061 | DVector3 locUnitDir2(locKinFitParticle2->Get_Momentum().Unit().X(),locKinFitParticle2->Get_Momentum().Unit().Y(),locKinFitParticle2->Get_Momentum().Unit().Z()); |
1062 | DVector3 locVertex1(locKinFitParticle1->Get_Position().X(),locKinFitParticle1->Get_Position().Y(),locKinFitParticle1->Get_Position().Z()); |
1063 | DVector3 locVertex2(locKinFitParticle2->Get_Position().X(),locKinFitParticle2->Get_Position().Y(),locKinFitParticle2->Get_Position().Z()); |
1064 | return Calc_DOCAVertex(locUnitDir1, locUnitDir2, locVertex1, locVertex2, locDOCAVertex); |
1065 | } |
1066 | |
1067 | double DAnalysisUtilities::Calc_DOCAVertex(const DKinematicData* locKinematicData1, const DKinematicData* locKinematicData2, DVector3& locDOCAVertex) const |
1068 | { |
1069 | DVector3 locPOCA1, locPOCA2; |
1070 | double locDOCA; |
1071 | |
1072 | if (dIsNoFieldFlag==false && Calc_DOCA(locKinematicData1,locKinematicData2, |
1073 | locPOCA1,locPOCA2,locDOCA)==NOERROR){ |
1074 | locDOCAVertex=0.5*(locPOCA1+locPOCA2); |
1075 | return locDOCA; |
1076 | } |
1077 | |
1078 | |
1079 | DVector3 locUnitDir1 = (1.0/locKinematicData1->momentum().Mag())*locKinematicData1->momentum(); |
1080 | DVector3 locUnitDir2 = (1.0/locKinematicData2->momentum().Mag())*locKinematicData2->momentum(); |
1081 | DVector3 locVertex1 = locKinematicData1->position(); |
1082 | DVector3 locVertex2 = locKinematicData2->position(); |
1083 | return Calc_DOCAVertex(locUnitDir1, locUnitDir2, locVertex1, locVertex2, locDOCAVertex); |
1084 | } |
1085 | |
1086 | double DAnalysisUtilities::Calc_DOCAVertex(const DVector3 &locUnitDir1, const DVector3 &locUnitDir2, const DVector3 &locVertex1, const DVector3 &locVertex2, DVector3& locDOCAVertex) const |
1087 | { |
1088 | DVector3 locPOCA1, locPOCA2; |
1089 | double locDOCA = Calc_DOCA(locUnitDir1, locUnitDir2, locVertex1, locVertex2, locPOCA1, locPOCA2); |
1090 | locDOCAVertex = 0.5*(locPOCA1 + locPOCA2); |
1091 | return locDOCA; |
1092 | } |
1093 | |
1094 | double DAnalysisUtilities::Calc_DOCA(const DKinFitParticle* locKinFitParticle1, const DKinFitParticle* locKinFitParticle2) const |
1095 | { |
1096 | DVector3 locPOCA1, locPOCA2; |
1097 | return Calc_DOCA(locKinFitParticle1, locKinFitParticle2, locPOCA1, locPOCA2); |
1098 | } |
1099 | |
1100 | double DAnalysisUtilities::Calc_DOCA(const DKinematicData* locKinematicData1, const DKinematicData* locKinematicData2) const |
1101 | { |
1102 | DVector3 locPOCA1, locPOCA2; |
1103 | return Calc_DOCA(locKinematicData1, locKinematicData2, locPOCA1, locPOCA2); |
1104 | } |
1105 | |
1106 | double DAnalysisUtilities::Calc_DOCA(const DKinFitParticle* locKinFitParticle1, const DKinFitParticle* locKinFitParticle2, DVector3 &locPOCA1, DVector3 &locPOCA2) const |
1107 | { |
1108 | DVector3 locUnitDir1(locKinFitParticle1->Get_Momentum().Unit().X(),locKinFitParticle1->Get_Momentum().Unit().Y(),locKinFitParticle1->Get_Momentum().Unit().Z()); |
1109 | DVector3 locUnitDir2(locKinFitParticle2->Get_Momentum().Unit().X(),locKinFitParticle2->Get_Momentum().Unit().Y(),locKinFitParticle2->Get_Momentum().Unit().Z()); |
1110 | DVector3 locVertex1(locKinFitParticle1->Get_Position().X(),locKinFitParticle1->Get_Position().Y(),locKinFitParticle1->Get_Position().Z()); |
1111 | DVector3 locVertex2(locKinFitParticle2->Get_Position().X(),locKinFitParticle2->Get_Position().Y(),locKinFitParticle2->Get_Position().Z()); |
1112 | return Calc_DOCA(locUnitDir1, locUnitDir2, locVertex1, locVertex2, locPOCA1, locPOCA2); |
1113 | } |
1114 | |
1115 | double DAnalysisUtilities::Calc_DOCA(const DKinematicData* locKinematicData1, const DKinematicData* locKinematicData2, DVector3 &locPOCA1, DVector3 &locPOCA2) const |
1116 | { |
1117 | DVector3 locUnitDir1 = (1.0/locKinematicData1->momentum().Mag())*locKinematicData1->momentum(); |
1118 | DVector3 locUnitDir2 = (1.0/locKinematicData2->momentum().Mag())*locKinematicData2->momentum(); |
1119 | DVector3 locVertex1 = locKinematicData1->position(); |
1120 | DVector3 locVertex2 = locKinematicData2->position(); |
1121 | return Calc_DOCA(locUnitDir1, locUnitDir2, locVertex1, locVertex2, locPOCA1, locPOCA2); |
1122 | } |
1123 | |
1124 | double DAnalysisUtilities::Calc_DOCA(const DVector3 &locUnitDir1, const DVector3 &locUnitDir2, const DVector3 &locVertex1, const DVector3 &locVertex2) const |
1125 | { |
1126 | DVector3 locPOCA1, locPOCA2; |
1127 | return Calc_DOCA(locUnitDir1, locUnitDir2, locVertex1, locVertex2, locPOCA1, locPOCA2); |
1128 | } |
1129 | |
1130 | double DAnalysisUtilities::Calc_DOCA(const DVector3 &locUnitDir1, const DVector3 &locUnitDir2, const DVector3 &locVertex1, const DVector3 &locVertex2, DVector3 &locPOCA1, DVector3 &locPOCA2) const |
1131 | { |
1132 | |
1133 | |
1134 | double locUnitDot = locUnitDir1.Dot(locUnitDir2); |
1135 | double locDenominator = locUnitDot*locUnitDot - 1.0; |
1136 | double locDistVertToInterDOCA1 = 0.0, locDistVertToInterDOCA2 = 0.0; |
1137 | |
1138 | if(fabs(locDenominator) < 1.0e-15) |
1139 | { |
1140 | locDistVertToInterDOCA1 = (locVertex2 - locVertex1).Dot(locUnitDir2)/locUnitDot; |
1141 | locDistVertToInterDOCA2 = (locVertex1 - locVertex2).Dot(locUnitDir1)/locUnitDot; |
1142 | } |
1143 | else |
1144 | { |
1145 | double locA = (locVertex1 - locVertex2).Dot(locUnitDir1); |
1146 | double locB = (locVertex1 - locVertex2).Dot(locUnitDir2); |
1147 | locDistVertToInterDOCA1 = (locA - locUnitDot*locB)/locDenominator; |
1148 | locDistVertToInterDOCA2 = (locUnitDot*locA - locB)/locDenominator; |
1149 | } |
1150 | |
1151 | locPOCA1 = locVertex1 + locDistVertToInterDOCA1*locUnitDir1; |
1152 | locPOCA2 = locVertex2 + locDistVertToInterDOCA2*locUnitDir2; |
1153 | return (locPOCA1 - locPOCA2).Mag(); |
1154 | } |
1155 | |
1156 | |
1157 | |
1158 | |
1159 | jerror_t |
1160 | DAnalysisUtilities::Calc_DOCA(const DKinFitParticle* locKinFitParticle1, |
1161 | const DKinFitParticle* locKinFitParticle2, |
1162 | DVector3 &pos1_out,DVector3 &pos2_out, |
1163 | double &doca) const{ |
1164 | |
1165 | double q1=locKinFitParticle1->Get_Charge(); |
1166 | double q2=locKinFitParticle2->Get_Charge(); |
1167 | |
1168 | |
1169 | DVector3 pos1_in=locKinFitParticle1->Get_Position(); |
1170 | DVector3 pos2_in=locKinFitParticle2->Get_Position(); |
1171 | |
1172 | |
1173 | DVector3 mom1_in=locKinFitParticle1->Get_Momentum(); |
1174 | DVector3 mom2_in=locKinFitParticle2->Get_Momentum(); |
1175 | |
1176 | return Calc_DOCA(q1,q2,pos1_in,pos2_in,mom1_in,mom2_in,pos1_out,pos2_out, |
1177 | doca); |
1178 | } |
1179 | |
1180 | |
1181 | |
1182 | |
1183 | jerror_t DAnalysisUtilities::Calc_DOCA(const DKinematicData* kinematicData1, |
1184 | const DKinematicData* kinematicData2, |
1185 | DVector3 &pos1_out,DVector3 &pos2_out, |
1186 | double &doca |
1187 | ) const { |
1188 | |
1189 | double q1=kinematicData1->charge(); |
1190 | double q2=kinematicData2->charge(); |
1191 | |
1192 | |
1193 | DVector3 pos1_in=kinematicData1->position(); |
1194 | DVector3 pos2_in=kinematicData2->position(); |
1195 | |
1196 | |
1197 | DVector3 mom1_in=kinematicData1->momentum(); |
1198 | DVector3 mom2_in=kinematicData2->momentum(); |
1199 | |
1200 | return Calc_DOCA(q1,q2,pos1_in,pos2_in,mom1_in,mom2_in,pos1_out,pos2_out, |
1201 | doca); |
1202 | } |
1203 | |
1204 | |
1205 | |
1206 | jerror_t DAnalysisUtilities::Calc_DOCA(double q1,double q2, |
1207 | const DVector3 &pos1_in, |
1208 | const DVector3 &pos2_in, |
1209 | const DVector3 &mom1_in, |
1210 | const DVector3 &mom2_in, |
1211 | DVector3 &pos1_out,DVector3 &pos2_out, |
1212 | double &doca |
1213 | ) const { |
1214 | DVector3 avg_pos=0.5*(pos1_in+pos2_in); |
1215 | DVector3 diff_pos=pos1_in-pos2_in; |
1216 | double B=fabs(dMagneticFieldMap->GetBz(avg_pos.x(),avg_pos.y(),avg_pos.z())); |
1217 | double kap1=-0.003*B*q1/(2.*mom1_in.Perp()); |
1218 | double kap2=-0.003*B*q2/(2.*mom2_in.Perp()); |
1219 | double dx=diff_pos.x(),dy=diff_pos.y(),dz=diff_pos.z(); |
1220 | double tan1=tan(M_PI_2-mom1_in.Theta()); |
1221 | double tan1sq=tan1*tan1; |
1222 | double phi1=mom1_in.Phi(); |
1223 | double cos1=cos(phi1),sin1=sin(phi1); |
1224 | double cos1sq=cos1*cos1,sin1sq=sin1*sin1; |
1225 | double tan2=tan(M_PI_2-mom2_in.Theta()); |
1226 | double tan2sq=tan2*tan2; |
1227 | double phi2=mom2_in.Phi(); |
1228 | double cos2=cos(phi2),sin2=sin(phi2); |
1229 | double cos2sq=cos2*cos2,sin2sq=sin2*sin2; |
1230 | double dx_sq=dx*dx,dy_sq=dy*dy; |
1231 | double cos2sin1_minus_cos1sin2=cos2*sin1 - cos1*sin2; |
1232 | |
1233 | double B1=pow(2*dx*kap2*sin1sq*sin2 - 2*dx*kap1*sin1*sin2sq - sin1sq*sin2sq |
1234 | + tan1sq + 2*dx*kap2*sin2*tan1sq - 2*dx*kap1*sin1*tan2sq |
1235 | - 2*sin1*sin2*(2*dx_sq*kap1*kap2 + tan1*tan2) |
1236 | + sin1sq*(1 + tan2sq) + cos1sq*(-2*cos2*dy*kap2 |
1237 | + 4*dx*kap2*sin2 + sin2sq |
1238 | + tan2sq) |
1239 | + 2*cos2*(-2*dy*kap2*sin1sq + dy*kap1*sin1*sin2 |
1240 | - dy*kap2*tan1sq + sin1*(2*dx*dy*kap1*kap2 |
1241 | - dz*kap2*tan1 |
1242 | + dz*kap1*tan2)) |
1243 | + 2*cos1*(cos2sq*dy*kap1 + dy*kap2*sin1*sin2 + dy*kap1*tan2sq |
1244 | + sin2*(2*dx*dy*kap1*kap2 + dz*kap2*tan1 |
1245 | - dz*kap1*tan2) |
1246 | - cos2*(2*dy_sq*kap1*kap2 + dx*kap2*sin1 |
1247 | + dx*kap1*sin2 + sin1*sin2 + tan1*tan2)),2) |
1248 | + 8*(cos2sin1_minus_cos1sin2)*(cos1*kap1*(cos2 - 2*dy*kap2) |
1249 | + kap2*(-1 + 2*dx*kap1*sin1 - tan1sq) |
1250 | + kap1*(sin1*sin2 + tan1*tan2))* |
1251 | (2*cos2*dy_sq*kap2*sin1 + cos2*dx*sin1*sin2 - dz*tan1 |
1252 | - 2*dx*dz*kap2*sin2*tan1 + dz*sin1*sin2*tan2 + cos2*dx*tan1*tan2 |
1253 | + dy*(-2*dx*kap2*sin1*sin2 + sin1*sin2sq + 2*cos2*dz*kap2*tan1 |
1254 | + sin2*tan1*tan2 - sin1*(1 + tan2sq)) |
1255 | + cos1*(cos2*(2*dx*dy*kap2 + dy*sin2 + dz*tan2) |
1256 | - dx*(2*dx*kap2*sin2 + sin2sq + tan2sq))); |
1257 | if (B1<0.){ |
1258 | doca=9.9e9; |
1259 | return VALUE_OUT_OF_RANGE; |
1260 | } |
1261 | |
1262 | double B2=pow(2*dx*kap2*sin1sq*sin2 - 2*dx*kap1*sin1*sin2sq - sin1sq*sin2sq |
1263 | + tan1sq + 2*dx*kap2*sin2*tan1sq - 2*dx*kap1*sin1*tan2sq |
1264 | - 2*sin1*sin2*(2*dx_sq*kap1*kap2 + tan1*tan2) |
1265 | + sin1sq*(1 + tan2sq) + cos1sq*(-2*cos2*dy*kap2 |
1266 | + 4*dx*kap2*sin2 + sin2sq |
1267 | + tan2sq) |
1268 | + 2*cos2*(-2*dy*kap2*sin1sq + dy*kap1*sin1*sin2 -dy*kap2*tan1sq |
1269 | + sin1*(2*dx*dy*kap1*kap2 - dz*kap2*tan1 |
1270 | + dz*kap1*tan2)) |
1271 | + 2*cos1*(cos2sq*dy*kap1 + dy*kap2*sin1*sin2 + dy*kap1*tan2sq |
1272 | + sin2*(2*dx*dy*kap1*kap2 + dz*kap2*tan1 |
1273 | - dz*kap1*tan2) |
1274 | - cos2*(2*dy_sq*kap1*kap2 + dx*kap2*sin1 |
1275 | + dx*kap1*sin2 + sin1*sin2 + tan1*tan2)),2) |
1276 | + 8*(cos2sin1_minus_cos1sin2)*(cos1*kap1*(cos2 - 2*dy*kap2) |
1277 | + kap2*(-1 + 2*dx*kap1*sin1 - tan1sq) |
1278 | + kap1*(sin1*sin2 + tan1*tan2)) |
1279 | * (2*cos2*dy_sq*kap2*sin1 + cos2*dx*sin1*sin2 - dz*tan1 |
1280 | - 2*dx*dz*kap2*sin2*tan1 + dz*sin1*sin2*tan2 + cos2*dx*tan1*tan2 |
1281 | + dy*(-2*dx*kap2*sin1*sin2 + sin1*sin2sq + 2*cos2*dz*kap2*tan1 |
1282 | + sin2*tan1*tan2 - sin1*(1 + tan2sq)) |
1283 | + cos1*(cos2*(2*dx*dy*kap2 + dy*sin2 + dz*tan2) |
1284 | - dx*(2*dx*kap2*sin2 + sin2sq + tan2sq))); |
1285 | if (B2<0){ |
1286 | doca=9.9e9; |
1287 | return VALUE_OUT_OF_RANGE; |
1288 | } |
1289 | |
1290 | double A1=2*cos1*cos2sq*dy*kap1 - 2*cos1sq*cos2*dy*kap2 |
1291 | - 4*cos1*cos2*dy_sq*kap1*kap2 - 2*cos1*cos2*dx*kap2*sin1 |
1292 | + 4*cos2*dx*dy*kap1*kap2*sin1 + cos2sq*sin1sq - 4*cos2*dy*kap2*sin1sq |
1293 | - 2*cos1*cos2*dx*kap1*sin2 + 4*cos1sq*dx*kap2*sin2 |
1294 | + 4*cos1*dx*dy*kap1*kap2*sin2 - 2*cos1*cos2*sin1*sin2 |
1295 | + 2*cos2*dy*kap1*sin1*sin2 + 2*cos1*dy*kap2*sin1*sin2 |
1296 | - 4*dx_sq*kap1*kap2*sin1*sin2 + 2*dx*kap2*sin1sq*sin2 + cos1sq*sin2sq |
1297 | - 2*dx*kap1*sin1*sin2sq - 2*cos2*dz*kap2*sin1*tan1 |
1298 | + 2*cos1*dz*kap2*sin2*tan1 + cos2sq*tan1sq - 2*cos2*dy*kap2*tan1sq |
1299 | + 2*dx*kap2*sin2*tan1sq + sin2sq*tan1sq + 2*cos2*dz*kap1*sin1*tan2 |
1300 | - 2*cos1*dz*kap1*sin2*tan2 - 2*cos1*cos2*tan1*tan2 - 2*sin1*sin2*tan1*tan2 |
1301 | + cos1sq*tan2sq + 2*cos1*dy*kap1*tan2sq - 2*dx*kap1*sin1*tan2sq |
1302 | + sin1sq*tan2sq; |
1303 | double C1=-4.*(cos2sin1_minus_cos1sin2) |
1304 | *(cos1*kap1*(cos2 - 2*dy*kap2) + kap2*(-1 + 2*dx*kap1*sin1 - tan1sq) |
1305 | + kap1*(sin1*sin2 + tan1*tan2)); |
1306 | |
1307 | double A2=2*cos1*cos2sq*dy*kap1 - 2*cos1sq*cos2*dy*kap2 |
1308 | - 4*cos1*cos2*dy_sq*kap1*kap2 - 4*cos2sq*dx*kap1*sin1 |
1309 | + 2*cos1*cos2*dx*kap2*sin1 + 4*cos2*dx*dy*kap1*kap2*sin1 + cos2sq*sin1sq |
1310 | + 2*cos1*cos2*dx*kap1*sin2 + 4*cos1*dx*dy*kap1*kap2*sin2 |
1311 | - 2*cos1*cos2*sin1*sin2 - 2*cos2*dy*kap1*sin1*sin2 |
1312 | - 2*cos1*dy*kap2*sin1*sin2 - 4*dx_sq*kap1*kap2*sin1*sin2 |
1313 | + 2*dx*kap2*sin1sq*sin2 + cos1sq*sin2sq + 4*cos1*dy*kap1*sin2sq |
1314 | - 2*dx*kap1*sin1*sin2sq + 2*cos2*dz*kap2*sin1*tan1 |
1315 | - 2*cos1*dz*kap2*sin2*tan1 + cos2sq*tan1sq - 2*cos2*dy*kap2*tan1sq |
1316 | + 2*dx*kap2*sin2*tan1sq + sin2sq*tan1sq - 2*cos2*dz*kap1*sin1*tan2 |
1317 | + 2*cos1*dz*kap1*sin2*tan2 - 2*cos1*cos2*tan1*tan2 - 2*sin1*sin2*tan1*tan2 |
1318 | + cos1sq*tan2sq + 2*cos1*dy*kap1*tan2sq - 2*dx*kap1*sin1*tan2sq |
1319 | + sin1sq*tan2sq; |
1320 | double C2=4.*(cos2sin1_minus_cos1sin2) |
1321 | *(kap2*(cos1*cos2 + sin1*sin2 + tan1*tan2) |
1322 | + kap1*(-1 + 2*cos2*dy*kap2 - 2*dx*kap2*sin2 - tan2sq)); |
1323 | |
1324 | |
1325 | double s1_minus=(A1-sqrt(B1))/C1; |
1326 | double s2_minus=(A2-sqrt(B2))/C2; |
1327 | double s1_plus=(A1+sqrt(B1))/C1; |
1328 | double s2_plus=(A2+sqrt(B2))/C2; |
1329 | |
1330 | |
1331 | double x1=pos1_in.x(); |
1332 | double y1=pos1_in.y(); |
1333 | double z1=pos1_in.z(); |
1334 | double x2=pos2_in.x(); |
1335 | double y2=pos2_in.y(); |
1336 | double z2=pos2_in.z(); |
1337 | |
1338 | |
1339 | if (fabs(s1_minus+s2_minus)<fabs(s1_plus+s2_plus)){ |
1340 | pos1_out.SetXYZ(x1+cos1*s1_minus,y1+sin1*s1_minus,z1+tan1*s1_minus); |
1341 | pos2_out.SetXYZ(x2+cos2*s2_minus,y2+sin2*s2_minus,z2+tan2*s2_minus); |
1342 | } |
1343 | else{ |
1344 | pos1_out.SetXYZ(x1+cos1*s1_plus,y1+sin1*s1_plus,z1+tan1*s1_plus); |
1345 | pos2_out.SetXYZ(x2+cos2*s2_plus,y2+sin2*s2_plus,z2+tan2*s2_plus); |
1346 | } |
1347 | |
1348 | doca=(pos1_out-pos2_out).Mag(); |
1349 | |
1350 | if (DEBUG_LEVEL>0){ |
1351 | DVector3 myvertex=0.5*(pos1_out+pos2_out); |
1352 | printf("Vertex position (x,y,z)=(%3.2f, %3.2f, %3.2f)\n",myvertex.x(), |
1353 | myvertex.y(),myvertex.z()); |
1354 | } |
1355 | |
1356 | return NOERROR; |
1357 | } |
1358 | |
1359 | |
1360 | |
1361 | double DAnalysisUtilities::Calc_CrudeTime(const vector<const DKinematicData*>& locParticles, const DVector3& locCommonVertex) const |
1362 | { |
1363 | |
1364 | DVector3 locPOCA; |
1365 | DVector3 locDeltaVertex; |
1366 | DVector3 locMomentum; |
1367 | double locAverageTime = 0.0; |
1368 | for(size_t loc_i = 0; loc_i < locParticles.size(); ++loc_i) |
1369 | { |
1370 | Calc_DOCAToVertex(locParticles[loc_i], locCommonVertex, locPOCA); |
1371 | locDeltaVertex = locPOCA - locParticles[loc_i]->position(); |
1372 | locMomentum = locParticles[loc_i]->momentum(); |
1373 | double locTime = locParticles[loc_i]->time() + locDeltaVertex.Dot(locMomentum)*locParticles[loc_i]->energy()/(29.9792458*locMomentum.Mag2()); |
1374 | locAverageTime += locTime; |
1375 | } |
1376 | return locAverageTime/(double(locParticles.size())); |
1377 | } |
1378 | |
1379 | double DAnalysisUtilities::Calc_CrudeTime(const vector<DKinFitParticle*>& locParticles, const DVector3& locCommonVertex) const |
1380 | { |
1381 | |
1382 | DVector3 locPOCA; |
1383 | DVector3 locDeltaVertex; |
1384 | double locAverageTime = 0.0; |
1385 | for(size_t loc_i = 0; loc_i < locParticles.size(); ++loc_i) |
1386 | { |
1387 | double locTime = 0.0; |
1388 | double locE = locParticles[loc_i]->Get_ShowerEnergy(); |
1389 | if((locParticles[loc_i]->Get_Charge() == 0) && (locE > 0.0)) |
1390 | { |
1391 | double locMass = locParticles[loc_i]->Get_Mass(); |
1392 | double locPMag = sqrt(locE*locE - locMass*locMass); |
1393 | DVector3 locPosition = locParticles[loc_i]->Get_Position(); |
1394 | DVector3 locDPosition(locPosition.X(), locPosition.Y(), locPosition.Z()); |
1395 | DVector3 locDeltaVertex = locDPosition - locCommonVertex; |
1396 | locTime = locParticles[loc_i]->Get_Time() + locDeltaVertex.Mag()*locE/(29.9792458*locPMag); |
1397 | } |
1398 | else |
1399 | { |
1400 | Calc_DOCAToVertex(locParticles[loc_i], locCommonVertex, locPOCA); |
1401 | locDeltaVertex = locPOCA - DVector3(locParticles[loc_i]->Get_Position().X(),locParticles[loc_i]->Get_Position().Y(),locParticles[loc_i]->Get_Position().Z()); |
1402 | DVector3 locMomentum(locParticles[loc_i]->Get_Momentum().X(),locParticles[loc_i]->Get_Momentum().Y(),locParticles[loc_i]->Get_Momentum().Z()); |
1403 | locTime = locParticles[loc_i]->Get_Time() + locDeltaVertex.Dot(locMomentum)*locParticles[loc_i]->Get_Energy()/(29.9792458*locMomentum.Mag2()); |
1404 | } |
1405 | locAverageTime += locTime; |
1406 | } |
1407 | return locAverageTime/(double(locParticles.size())); |
1408 | } |
1409 | |
1410 | DVector3 DAnalysisUtilities::Calc_CrudeVertex(const vector<const DTrackTimeBased*>& locParticles) const |
1411 | { |
1412 | |
1413 | |
1414 | DVector3 locVertex(0.0, 0.0, dTargetZCenter); |
1415 | |
1416 | if(locParticles.size() == 0) |
1417 | return locVertex; |
1418 | if(locParticles.size() == 1) |
1419 | return locParticles[0]->position(); |
1420 | |
1421 | double locDOCA, locSmallestDOCA; |
1422 | DVector3 locTempVertex; |
1423 | |
1424 | locSmallestDOCA = 9.9E9; |
1425 | for(int loc_j = 0; loc_j < (int(locParticles.size()) - 1); ++loc_j) |
1426 | { |
1427 | for(size_t loc_k = loc_j + 1; loc_k < locParticles.size(); ++loc_k) |
1428 | { |
1429 | locDOCA = Calc_DOCAVertex(locParticles[loc_j], locParticles[loc_k], locTempVertex); |
1430 | |
1431 | if(locDOCA < locSmallestDOCA) |
1432 | { |
1433 | locSmallestDOCA = locDOCA; |
1434 | locVertex = locTempVertex; |
1435 | } |
1436 | } |
1437 | } |
1438 | return locVertex; |
1439 | } |
1440 | |
1441 | DVector3 DAnalysisUtilities::Calc_CrudeVertex(const vector<const DChargedTrackHypothesis*>& locParticles) const |
1442 | { |
1443 | |
1444 | |
1445 | DVector3 locVertex(0.0, 0.0, dTargetZCenter); |
1446 | |
1447 | if(locParticles.size() == 0) |
1448 | return locVertex; |
1449 | if(locParticles.size() == 1) |
1450 | return locParticles[0]->position(); |
1451 | |
1452 | double locDOCA, locSmallestDOCA; |
1453 | DVector3 locTempVertex; |
1454 | |
1455 | locSmallestDOCA = 9.9E9; |
1456 | for(int loc_j = 0; loc_j < (int(locParticles.size()) - 1); ++loc_j) |
1457 | { |
1458 | for(size_t loc_k = loc_j + 1; loc_k < locParticles.size(); ++loc_k) |
1459 | { |
1460 | locDOCA = Calc_DOCAVertex(locParticles[loc_j], locParticles[loc_k], locTempVertex); |
1461 | if(locDOCA < locSmallestDOCA) |
1462 | { |
1463 | locSmallestDOCA = locDOCA; |
1464 | locVertex = locTempVertex; |
1465 | } |
1466 | } |
1467 | } |
1468 | return locVertex; |
1469 | } |
1470 | |
1471 | DVector3 DAnalysisUtilities::Calc_CrudeVertex(const vector<const DKinematicData*>& locParticles) const |
1472 | { |
1473 | |
1474 | |
1475 | DVector3 locVertex(0.0, 0.0, dTargetZCenter); |
1476 | |
1477 | if(locParticles.size() == 0) |
1478 | return locVertex; |
1479 | if(locParticles.size() == 1) |
1480 | return locParticles[0]->position(); |
1481 | |
1482 | double locDOCA, locSmallestDOCA; |
1483 | DVector3 locTempVertex; |
1484 | |
1485 | locSmallestDOCA = 9.9E9; |
1486 | for(int loc_j = 0; loc_j < (int(locParticles.size()) - 1); ++loc_j) |
1487 | { |
1488 | for(size_t loc_k = loc_j + 1; loc_k < locParticles.size(); ++loc_k) |
1489 | { |
1490 | locDOCA = Calc_DOCAVertex(locParticles[loc_j], locParticles[loc_k], locTempVertex); |
1491 | if(locDOCA < locSmallestDOCA) |
1492 | { |
1493 | locSmallestDOCA = locDOCA; |
1494 | locVertex = locTempVertex; |
1495 | } |
1496 | } |
1497 | } |
1498 | return locVertex; |
1499 | } |
1500 | |
1501 | DVector3 DAnalysisUtilities::Calc_CrudeVertex(const vector<shared_ptr<DKinFitParticle>>& locParticles) const |
1502 | { |
1503 | |
1504 | |
1505 | DVector3 locVertex(0.0, 0.0, dTargetZCenter); |
1506 | |
1507 | if(locParticles.size() == 0) |
1508 | return locVertex; |
1509 | |
1510 | if(locParticles.size() == 1) |
1511 | return DVector3(locParticles[0]->Get_Position().X(), locParticles[0]->Get_Position().Y(), locParticles[0]->Get_Position().Z()); |
1512 | |
1513 | double locDOCA, locSmallestDOCA; |
1514 | DVector3 locTempVertex; |
1515 | |
1516 | locSmallestDOCA = 9.9E9; |
1517 | for(int loc_j = 0; loc_j < (int(locParticles.size()) - 1); ++loc_j) |
1518 | { |
1519 | for(size_t loc_k = loc_j + 1; loc_k < locParticles.size(); ++loc_k) |
1520 | { |
1521 | locDOCA = Calc_DOCAVertex(locParticles[loc_j].get(), locParticles[loc_k].get(), locTempVertex); |
1522 | if(locDOCA < locSmallestDOCA) |
1523 | { |
1524 | locSmallestDOCA = locDOCA; |
1525 | locVertex = locTempVertex; |
1526 | } |
1527 | } |
1528 | } |
1529 | return locVertex; |
1530 | } |
1531 | |
1532 | set<set<size_t> > DAnalysisUtilities::Build_IndexCombos(const DReactionStep* locReactionStep, deque<Particle_t> locToIncludePIDs) const |
1533 | { |
1534 | |
1535 | set<set<size_t> > locCombos; |
1536 | |
1537 | auto locReactionStepPIDs = locReactionStep->Get_FinalPIDs(); |
1538 | int locMissingParticleIndex = locReactionStep->Get_MissingParticleIndex(); |
1539 | |
1540 | if(locToIncludePIDs.empty()) |
1541 | { |
1542 | set<size_t> locCombo; |
1543 | for(size_t loc_i = 0; loc_i < locReactionStepPIDs.size(); ++loc_i) |
1544 | { |
1545 | if(int(loc_i) == locMissingParticleIndex) |
1546 | continue; |
1547 | locCombo.insert(loc_i); |
1548 | } |
1549 | locCombos.insert(locCombo); |
1550 | return locCombos; |
1551 | } |
1552 | |
1553 | |
1554 | deque<deque<size_t> > locPossibilities(locToIncludePIDs.size(), deque<size_t>()); |
1555 | deque<int> locResumeAtIndices(locToIncludePIDs.size(), 0); |
1556 | |
1557 | |
1558 | for(size_t loc_i = 0; loc_i < locReactionStepPIDs.size(); ++loc_i) |
1559 | { |
1560 | if(int(loc_i) == locMissingParticleIndex) |
1561 | continue; |
1562 | Particle_t locPID = locReactionStepPIDs[loc_i]; |
1563 | |
1564 | |
1565 | for(size_t loc_j = 0; loc_j < locToIncludePIDs.size(); ++loc_j) |
1566 | { |
1567 | if(locToIncludePIDs[loc_j] == locPID) |
1568 | locPossibilities[loc_j].push_back(loc_i); |
1569 | } |
1570 | } |
1571 | |
1572 | |
1573 | int locParticleIndex = 0; |
1574 | deque<size_t> locComboDeque; |
1575 | while(true) |
1576 | { |
1577 | if(locParticleIndex == int(locPossibilities.size())) |
1578 | { |
1579 | set<size_t> locComboSet; |
1580 | for(size_t loc_i = 0; loc_i < locComboDeque.size(); ++loc_i) |
1581 | locComboSet.insert(locComboDeque[loc_i]); |
1582 | locCombos.insert(locComboSet); |
1583 | |
1584 | if(!Handle_Decursion(locParticleIndex, locComboDeque, locResumeAtIndices, locPossibilities)) |
1585 | break; |
1586 | continue; |
1587 | } |
1588 | |
1589 | int& locResumeAtIndex = locResumeAtIndices[locParticleIndex]; |
1590 | |
1591 | |
1592 | |
1593 | Particle_t locToIncludePID = locToIncludePIDs[locParticleIndex]; |
1594 | for(int loc_i = locParticleIndex - 1; loc_i >= 0; --loc_i) |
1595 | { |
1596 | if(locToIncludePIDs[loc_i] == locToIncludePID) |
1597 | { |
1598 | if(locResumeAtIndex < locResumeAtIndices[loc_i]) |
1599 | locResumeAtIndex = locResumeAtIndices[loc_i]; |
1600 | break; |
1601 | } |
1602 | } |
1603 | |
1604 | if(locResumeAtIndex >= int(locPossibilities[locParticleIndex].size())) |
1605 | { |
1606 | if(!Handle_Decursion(locParticleIndex, locComboDeque, locResumeAtIndices, locPossibilities)) |
1607 | break; |
1608 | continue; |
1609 | } |
1610 | |
1611 | |
1612 | locComboDeque.push_back(locPossibilities[locParticleIndex][locResumeAtIndex]); |
1613 | ++locResumeAtIndex; |
1614 | ++locParticleIndex; |
1615 | } |
1616 | |
1617 | return locCombos; |
1618 | } |
1619 | |
1620 | bool DAnalysisUtilities::Handle_Decursion(int& locParticleIndex, deque<size_t>& locComboDeque, deque<int>& locResumeAtIndices, deque<deque<size_t> >& locPossibilities) const |
1621 | { |
1622 | do |
1623 | { |
1624 | if(locParticleIndex < int(locResumeAtIndices.size())) |
1625 | locResumeAtIndices[locParticleIndex] = 0; |
1626 | |
1627 | --locParticleIndex; |
1628 | if(locParticleIndex < 0) |
1629 | return false; |
1630 | |
1631 | locComboDeque.pop_back(); |
1632 | } |
1633 | while(locResumeAtIndices[locParticleIndex] == int(locPossibilities[locParticleIndex].size())); |
1634 | |
1635 | return true; |
1636 | } |
1637 | |
1638 | |
1639 | |
1640 | |
1641 | double DAnalysisUtilities::Propagate_Track(int locCharge, const DVector3& locPropagateToPoint, DLorentzVector& locMeasuredX4, DLorentzVector& locMeasuredP4, TMatrixFSym* locCovarianceMatrix) const |
1642 | { |
1643 | |
1644 | double locDistance = (locMeasuredX4.Vect() - locPropagateToPoint).Mag(); |
1645 | if(!Get_IsBFieldNearBeamline() || (locCharge == 0) || (locDistance < dMinDistanceForStraightTrack)) |
1646 | { |
1647 | |
1648 | DVector3 locPOCA; |
1649 | auto locP3 = locMeasuredP4.Vect(); |
1650 | Calc_DOCAToVertex(locP3.Unit(), locMeasuredX4.Vect(), locPropagateToPoint, locPOCA); |
1651 | locMeasuredX4.SetVect(locPOCA); |
1652 | auto locDistanceVector = locPOCA - locMeasuredX4.Vect(); |
1653 | |
1654 | auto locDeltaPathLength = (locP3.Dot(locDistanceVector) > 0.0) ? -1.0*locDistanceVector.Mag() : locDistanceVector.Mag(); |
1655 | locMeasuredX4.SetT(locMeasuredX4.T() - locDeltaPathLength/(locMeasuredP4.Beta()*SPEED_OF_LIGHT)); |
1656 | return locDeltaPathLength; |
1657 | } |
1658 | |
1659 | |
1660 | double locTotalDeltaPathLength = 0.0; |
1661 | DLorentzVector locTempPropagatedPosition = locMeasuredX4; |
1662 | DLorentzVector locTempPropagatedMomentum = locMeasuredP4; |
1663 | if(fabs(locMeasuredP4.Pz()) > 0.0) |
1664 | { |
1665 | locTotalDeltaPathLength += (locPropagateToPoint.Z() - locMeasuredX4.Z())*locMeasuredP4.P()/locMeasuredP4.Pz(); |
1666 | Propagate_Track(locTotalDeltaPathLength, locCharge, locTempPropagatedPosition, locTempPropagatedMomentum, NULL); |
1667 | } |
1668 | |
1669 | |
1670 | locTotalDeltaPathLength += Calc_PathLength_Step(locCharge, locPropagateToPoint, locTempPropagatedPosition, locTempPropagatedMomentum); |
1671 | |
1672 | |
1673 | locTotalDeltaPathLength += Calc_PathLength_FineGrained(locCharge, locPropagateToPoint, locTempPropagatedPosition.Vect(), locTempPropagatedMomentum.Vect()); |
1674 | |
1675 | |
1676 | |
1677 | Propagate_Track(locTotalDeltaPathLength, locCharge, locMeasuredX4, locMeasuredP4, locCovarianceMatrix); |
1678 | |
1679 | |
1680 | return -1.0*locTotalDeltaPathLength; |
1681 | } |
1682 | |
1683 | double DAnalysisUtilities::Calc_PathLength_Step(int locCharge, const DVector3& locPropagateToPoint, DLorentzVector& locMeasuredX4, DLorentzVector& locMeasuredP4) const |
1684 | { |
1685 | |
1686 | |
1687 | DVector3 locBField = Get_BField(locPropagateToPoint); |
1688 | double locA = -0.00299792458*(double(locCharge))*locBField.Mag(); |
1689 | |
1690 | double locPMag = locMeasuredP4.P(); |
1691 | double locPathLengthOneRotation = 2.0*TMath::Pi()*locPMag/locA; |
1692 | double locStepDistance = locPathLengthOneRotation/12.0; |
1693 | |
1694 | |
1695 | double locStepDeltaZ = locMeasuredP4.Pz()*locStepDistance/locPMag; |
1696 | double locTargetDeltaZ = 1.0; |
1697 | if(fabs(locStepDeltaZ) > 5.0) |
1698 | locStepDistance = locTargetDeltaZ*locPMag/locMeasuredP4.Pz(); |
1699 | |
1700 | |
1701 | double locStepDirection = 1.0; |
1702 | double locTotalDeltaPathLength = 0.0; |
1703 | double locDeltaX3 = (locMeasuredX4.Vect() - locPropagateToPoint).Mag(); |
1704 | double locRhoS = locStepDistance*locA/locPMag; |
1705 | double locRhoSCutOff = 0.1; |
1706 | while(locRhoS > locRhoSCutOff) |
1707 | { |
1708 | double locDeltaPathLength = locStepDirection*locStepDistance; |
1709 | locTotalDeltaPathLength += locDeltaPathLength; |
1710 | |
1711 | Propagate_Track(locDeltaPathLength, locCharge, locMeasuredX4, locMeasuredP4, NULL); |
1712 | |
1713 | double locNewDeltaX3 = (locMeasuredX4.Vect() - locPropagateToPoint).Mag(); |
1714 | bool locGettingCloserFlag = (locNewDeltaX3 < locDeltaX3); |
1715 | locDeltaX3 = locNewDeltaX3; |
1716 | |
1717 | if(locGettingCloserFlag) |
1718 | continue; |
1719 | |
1720 | |
1721 | locStepDirection *= -1.0; |
1722 | locStepDistance /= 2.0; |
1723 | locRhoS = locStepDistance*locA/locPMag; |
1724 | |
1725 | |
1726 | if(locRhoS < locRhoSCutOff) |
1727 | { |
1728 | locTotalDeltaPathLength -= locDeltaPathLength; |
1729 | Propagate_Track(-1.0*locDeltaPathLength, locCharge, locMeasuredX4, locMeasuredP4, NULL); |
1730 | } |
1731 | } |
1732 | |
1733 | return locTotalDeltaPathLength; |
1734 | } |
1735 | |
1736 | double DAnalysisUtilities::Calc_PathLength_FineGrained(int locCharge, const DVector3& locPropagateToPoint, DVector3 locMeasuredPosition, DVector3 locMeasuredMomentum) const |
1737 | { |
1738 | |
1739 | |
1740 | |
1741 | |
1742 | |
1743 | |
1744 | |
1745 | |
1746 | |
1747 | |
1748 | |
1749 | |
1750 | |
1751 | |
1752 | |
1753 | DVector3 locBField = Get_BField(locMeasuredPosition); |
1754 | double locA = -0.00299792458*(double(locCharge))*locBField.Mag(); |
1755 | double locPMag = locMeasuredMomentum.Mag(); |
1756 | double locRho = locA/locPMag; |
1757 | |
1758 | DVector3 locDeltaX3 = locMeasuredPosition - locPropagateToPoint; |
1759 | double locF = locMeasuredMomentum.Pz()/locPMag; |
1760 | double locC = locDeltaX3.Z(); |
1761 | double locD = locRho*(locMeasuredMomentum.Px()*locDeltaX3.X() + locMeasuredMomentum.Py()*locDeltaX3.Y())/locA; |
1762 | double locPerpPSq = locMeasuredMomentum.Px()*locMeasuredMomentum.Px() + locMeasuredMomentum.Py()*locMeasuredMomentum.Py(); |
1763 | double locE = locRho*(locPerpPSq/locA - locMeasuredMomentum.Py()*locDeltaX3.X() + locMeasuredMomentum.Px()*locDeltaX3.Y())/locA; |
1764 | return -1.0*(locF*locC + locD)/(locF*locF + locE*locRho); |
1765 | } |
1766 | |
1767 | void DAnalysisUtilities::Propagate_Track(double locDeltaPathLength, int locCharge, DLorentzVector& locX4, DLorentzVector& locP4, TMatrixFSym* locCovarianceMatrix) const |
1768 | { |
1769 | |
1770 | |
1771 | DVector3 locBField = Get_BField(locX4.Vect()); |
1772 | if(!(locBField.Mag() > 0.0)) |
1773 | return; |
1774 | DVector3 locH = locBField.Unit(); |
1775 | double locA = -0.00299792458*(double(locCharge))*locBField.Mag(); |
1776 | |
1777 | double locPMag = locP4.P(); |
1778 | double locRhoS = locDeltaPathLength*locA/locPMag; |
1779 | |
1780 | DLorentzVector locDeltaX4; |
1781 | locDeltaX4.SetX(locP4.Px()*sin(locRhoS)/locA - locP4.Py()*(1.0 - cos(locRhoS))/locA); |
1782 | locDeltaX4.SetY(locP4.Py()*sin(locRhoS)/locA + locP4.Px()*(1.0 - cos(locRhoS))/locA); |
1783 | locDeltaX4.SetZ(locP4.Pz()*locDeltaPathLength/locPMag); |
1784 | locDeltaX4.SetT(locDeltaPathLength/(locP4.Beta()*SPEED_OF_LIGHT)); |
1785 | locX4 += locDeltaX4; |
1786 | |
1787 | if(locCovarianceMatrix == NULL) |
1788 | { |
1789 | locP4.SetVect(locP4.Vect() - locDeltaX4.Vect().Cross(locA*locH)); |
1790 | return; |
1791 | } |
1792 | |
1793 | |
1794 | TMatrixF locTransformMatrix(7, 7); |
1795 | locTransformMatrix.Zero(); |
1796 | |
1797 | double locPCubed = locPMag*locPMag*locPMag; |
1798 | double locSOverP3 = locDeltaPathLength/locPCubed; |
1799 | |
1800 | double locSOverP3_PxPx = locSOverP3*locP4.Px()*locP4.Px(); |
1801 | double locSOverP3_PxPy = locSOverP3*locP4.Px()*locP4.Py(); |
1802 | double locSOverP3_PxPz = locSOverP3*locP4.Px()*locP4.Pz(); |
1803 | |
1804 | double locSOverP3_PyPy = locSOverP3*locP4.Py()*locP4.Py(); |
1805 | double locSOverP3_PyPz = locSOverP3*locP4.Py()*locP4.Pz(); |
1806 | double locSOverP3_PzPz = locSOverP3*locP4.Pz()*locP4.Pz(); |
1807 | |
1808 | |
1809 | locTransformMatrix(0, 0) = (1.0 + locA*locSOverP3_PxPy)*cos(locRhoS) + locA*locSOverP3_PxPx*sin(locRhoS); |
1810 | locTransformMatrix(0, 1) = (-1.0 + locA*locSOverP3_PxPy)*sin(locRhoS) + locA*locSOverP3_PyPy*cos(locRhoS); |
1811 | locTransformMatrix(0, 2) = locA*locSOverP3_PyPz*cos(locRhoS) + locA*locSOverP3_PxPz*sin(locRhoS); |
1812 | |
1813 | |
1814 | locTransformMatrix(1, 0) = -1.0*locA*locSOverP3_PxPx*cos(locRhoS) + (1.0 + locA*locSOverP3_PxPy)*sin(locRhoS); |
1815 | locTransformMatrix(1, 1) = (1.0 - locA*locSOverP3_PxPy)*cos(locRhoS) + locA*locSOverP3_PyPy*sin(locRhoS); |
1816 | locTransformMatrix(1, 2) = locA*locSOverP3_PyPz*sin(locRhoS) - locA*locSOverP3_PxPz*cos(locRhoS); |
1817 | |
1818 | |
1819 | locTransformMatrix(2, 2) = 1.0; |
1820 | |
1821 | |
1822 | locTransformMatrix(3, 0) = (1.0/locA + locSOverP3_PxPy)*sin(locRhoS) - locSOverP3_PxPx*cos(locRhoS); |
1823 | locTransformMatrix(3, 1) = -1.0/locA + (1.0/locA - locSOverP3_PxPy)*cos(locRhoS) + locSOverP3_PyPy*sin(locRhoS); |
1824 | locTransformMatrix(3, 2) = locSOverP3_PyPz*sin(locRhoS) - locSOverP3_PxPz*cos(locRhoS); |
1825 | locTransformMatrix(3, 3) = 1.0; |
1826 | |
1827 | |
1828 | locTransformMatrix(4, 0) = 1.0/locA - (1.0/locA + locSOverP3_PxPy)*cos(locRhoS) + locSOverP3_PxPx*sin(locRhoS); |
1829 | locTransformMatrix(4, 1) = (1.0/locA - locSOverP3_PxPy)*sin(locRhoS) - locSOverP3_PyPy*cos(locRhoS); |
1830 | locTransformMatrix(4, 2) = -1.0*locSOverP3_PyPz*cos(locRhoS) - locSOverP3_PxPz*sin(locRhoS); |
1831 | locTransformMatrix(4, 4) = 1.0; |
1832 | |
1833 | |
1834 | locTransformMatrix(5, 0) = -1.0*locSOverP3_PxPz; |
1835 | locTransformMatrix(5, 1) = -1.0*locSOverP3_PyPz; |
1836 | locTransformMatrix(5, 2) = locDeltaPathLength/locPMag - locSOverP3_PzPz; |
1837 | locTransformMatrix(5, 5) = 1.0; |
1838 | |
1839 | |
1840 | locTransformMatrix(6, 0) = -1.0*locP4.M2()*locSOverP3*locP4.Px()/(SPEED_OF_LIGHT*locP4.E()); |
1841 | locTransformMatrix(6, 1) = -1.0*locP4.M2()*locSOverP3*locP4.Py()/(SPEED_OF_LIGHT*locP4.E()); |
1842 | locTransformMatrix(6, 2) = -1.0*locP4.M2()*locSOverP3*locP4.Pz()/(SPEED_OF_LIGHT*locP4.E()); |
1843 | locTransformMatrix(6, 6) = 1.0; |
1844 | |
1845 | |
1846 | locCovarianceMatrix->Similarity(locTransformMatrix); |
1847 | |
1848 | |
1849 | locP4.SetVect(locP4.Vect() - locDeltaX4.Vect().Cross(locA*locH)); |
1850 | } |
1 | |
2 | |
3 | |
4 | |
5 | |
6 | |
7 | |
8 | #ifndef _JEventLoop_ |
9 | #define _JEventLoop_ |
10 | |
11 | #include <sys/time.h> |
12 | |
13 | #include <vector> |
14 | #include <list> |
15 | #include <string> |
16 | #include <utility> |
17 | #include <typeinfo> |
18 | #include <string.h> |
19 | #include <map> |
20 | #include <utility> |
21 | using std::vector; |
22 | using std::list; |
23 | using std::string; |
24 | using std::type_info; |
25 | |
26 | #include <JANA/jerror.h> |
27 | #include <JANA/JObject.h> |
28 | #include <JANA/JException.h> |
29 | #include <JANA/JEvent.h> |
30 | #include <JANA/JThread.h> |
31 | #include <JANA/JFactory_base.h> |
32 | #include <JANA/JCalibration.h> |
33 | #include <JANA/JGeometry.h> |
34 | #include <JANA/JResourceManager.h> |
35 | #include <JANA/JStreamLog.h> |
36 | |
37 | |
38 | #include "cint.h" |
39 | |
40 | |
41 | |
42 | namespace jana{ |
43 | |
44 | |
45 | template<class T> class JFactory; |
46 | class JApplication; |
47 | class JEventProcessor; |
48 | |
49 | |
50 | class JEventLoop{ |
51 | public: |
52 | |
53 | friend class JApplication; |
54 | |
55 | enum data_source_t{ |
56 | DATA_NOT_AVAILABLE = 1, |
57 | DATA_FROM_CACHE, |
58 | DATA_FROM_SOURCE, |
59 | DATA_FROM_FACTORY |
60 | }; |
61 | |
62 | typedef struct{ |
63 | string caller_name; |
64 | string caller_tag; |
65 | string callee_name; |
66 | string callee_tag; |
67 | double start_time; |
68 | double end_time; |
69 | data_source_t data_source; |
70 | }call_stack_t; |
71 | |
72 | typedef struct{ |
73 | const char* factory_name; |
74 | string tag; |
75 | const char* filename; |
76 | int line; |
77 | }error_call_stack_t; |
78 | |
79 | JEventLoop(JApplication *app); |
80 | virtual ~JEventLoop(); |
81 | virtual const char* className(void){return static_className();} |
82 | static const char* static_className(void){return "JEventLoop";} |
83 | |
84 | JApplication* GetJApplication(void) const {return app;} |
85 | void RefreshProcessorListFromJApplication(void); |
86 | virtual jerror_t AddFactory(JFactory_base* factory); |
87 | jerror_t RemoveFactory(JFactory_base* factory); |
88 | JFactory_base* GetFactory(const string data_name, const char *tag="", bool allow_deftag=true); |
89 | vector<JFactory_base*> GetFactories(void) const {return factories;} |
90 | void GetFactoryNames(vector<string> &factorynames); |
91 | void GetFactoryNames(map<string,string> &factorynames); |
92 | map<string,string> GetDefaultTags(void) const {return default_tags;} |
93 | jerror_t ClearFactories(void); |
94 | jerror_t PrintFactories(int sparsify=0); |
95 | jerror_t Print(const string data_name, const char *tag=""); |
96 | |
97 | JCalibration* GetJCalibration(); |
98 | template<class T> bool GetCalib(string namepath, map<string,T> &vals); |
99 | template<class T> bool GetCalib(string namepath, vector<T> &vals); |
100 | template<class T> bool GetCalib(string namepath, T &val); |
101 | |
102 | JGeometry* GetJGeometry(); |
103 | template<class T> bool GetGeom(string namepath, map<string,T> &vals); |
104 | template<class T> bool GetGeom(string namepath, T &val); |
105 | |
106 | JResourceManager* GetJResourceManager(void); |
107 | string GetResource(string namepath); |
108 | template<class T> bool GetResource(string namepath, T vals, int event_number=0); |
109 | |
110 | void Initialize(void); |
111 | jerror_t Loop(void); |
112 | jerror_t OneEvent(uint64_t event_number); |
113 | jerror_t OneEvent(void); |
114 | inline void Pause(void){pause = 1;} |
115 | inline void Resume(void){pause = 0;} |
116 | inline void Quit(void){quit = 1;} |
117 | inline bool GetQuit(void) const {return quit;} |
118 | void QuitProgram(void); |
119 | |
120 | |
121 | bool HasRandomAccess(void); |
122 | void AddToEventQueue(uint64_t event_number){ next_events_to_process.push_back(event_number); } |
123 | void AddToEventQueue(list<uint64_t> &event_numbers) { next_events_to_process.insert(next_events_to_process.end(), event_numbers.begin(), event_numbers.end()); } |
124 | list<uint64_t> GetEventQueue(void){ return next_events_to_process; } |
125 | void ClearEventQueue(void){ next_events_to_process.clear(); } |
126 | |
127 | template<class T> JFactory<T>* GetSingle(const T* &t, const char *tag="", bool exception_if_not_one=true); |
128 | template<class T> JFactory<T>* Get(vector<const T*> &t, const char *tag="", bool allow_deftag=true); |
129 | template<class T> JFactory<T>* GetFromFactory(vector<const T*> &t, const char *tag="", data_source_t &data_source=null_data_source, bool allow_deftag=true); |
130 | template<class T> jerror_t GetFromSource(vector<const T*> &t, JFactory_base *factory=NULL); |
131 | inline JEvent& GetJEvent(void){return event;} |
132 | inline void SetJEvent(JEvent *event){this->event = *event;} |
133 | inline void SetAutoFree(int auto_free){this->auto_free = auto_free;} |
134 | inline pthread_t GetPThreadID(void) const {return pthread_id;} |
135 | double GetInstantaneousRate(void) const {return rate_instantaneous;} |
136 | double GetIntegratedRate(void) const {return rate_integrated;} |
137 | double GetLastEventProcessingTime(void) const {return delta_time_single;} |
138 | unsigned int GetNevents(void) const {return Nevents;} |
139 | |
140 | inline bool CheckEventBoundary(uint64_t event_numberA, uint64_t event_numberB); |
141 | |
142 | inline bool GetCallStackRecordingStatus(void){ return record_call_stack; } |
143 | inline void DisableCallStackRecording(void){ record_call_stack = false; } |
144 | inline void EnableCallStackRecording(void){ record_call_stack = true; } |
145 | inline void CallStackStart(JEventLoop::call_stack_t &cs, const string &caller_name, const string &caller_tag, const string callee_name, const string callee_tag); |
146 | inline void CallStackEnd(JEventLoop::call_stack_t &cs); |
147 | inline vector<call_stack_t> GetCallStack(void){return call_stack;} |
148 | inline void AddToCallStack(call_stack_t &cs){if(record_call_stack) call_stack.push_back(cs);} |
149 | inline void AddToErrorCallStack(error_call_stack_t &cs){error_call_stack.push_back(cs);} |
150 | inline vector<error_call_stack_t> GetErrorCallStack(void){return error_call_stack;} |
151 | void PrintErrorCallStack(void); |
152 | |
153 | const JObject* FindByID(JObject::oid_t id); |
154 | template<class T> const T* FindByID(JObject::oid_t id); |
155 | JFactory_base* FindOwner(const JObject *t); |
156 | JFactory_base* FindOwner(JObject::oid_t id); |
157 | |
158 | |
159 | template<class T> void SetRef(T *t); |
160 | template<class T> T* GetRef(void); |
161 | template<class T> vector<T*> GetRefsT(void); |
162 | vector<pair<const char*, void*> > GetRefs(void){ return user_refs; } |
163 | template<class T> void RemoveRef(T *t); |
164 | |
165 | |
166 | uint64_t GetStatus(void){return event.GetStatus();} |
167 | bool GetStatusBit(uint32_t bit){return event.GetStatusBit(bit);} |
168 | bool SetStatusBit(uint32_t bit, bool val=true){return event.SetStatusBit(bit, val);} |
169 | bool ClearStatusBit(uint32_t bit){return event.ClearStatusBit(bit);} |
170 | void ClearStatus(void){event.ClearStatus();} |
171 | void SetStatusBitDescription(uint32_t bit, string description){event.SetStatusBitDescription(bit, description);} |
172 | string GetStatusBitDescription(uint32_t bit){return event.GetStatusBitDescription(bit);} |
173 | void GetStatusBitDescriptions(map<uint32_t, string> &status_bit_descriptions){return event.GetStatusBitDescriptions(status_bit_descriptions);} |
174 | string StatusWordToString(void); |
175 | |
176 | private: |
177 | JEvent event; |
178 | vector<JFactory_base*> factories; |
179 | vector<JEventProcessor*> processors; |
180 | vector<error_call_stack_t> error_call_stack; |
181 | vector<call_stack_t> call_stack; |
182 | JApplication *app; |
183 | JThread *jthread; |
184 | bool initialized; |
185 | bool print_parameters_called; |
186 | int pause; |
187 | int quit; |
188 | int auto_free; |
189 | pthread_t pthread_id; |
190 | map<string, string> default_tags; |
191 | vector<pair<string,string> > auto_activated_factories; |
192 | bool record_call_stack; |
193 | string caller_name; |
194 | string caller_tag; |
195 | vector<uint64_t> event_boundaries; |
196 | int32_t event_boundaries_run; |
197 | list<uint64_t> next_events_to_process; |
198 | |
199 | uint64_t Nevents; |
200 | uint64_t Nevents_rate; |
201 | double delta_time_single; |
202 | double delta_time_rate; |
203 | double delta_time; |
204 | double rate_instantaneous; |
205 | double rate_integrated; |
206 | |
207 | static data_source_t null_data_source; |
208 | |
209 | vector<pair<const char*, void*> > user_refs; |
210 | }; |
211 | |
212 | |
213 | |
214 | #ifdef G__DICTIONARY |
215 | typedef JEventLoop::call_stack_t call_stack_t; |
216 | typedef JEventLoop::error_call_stack_t error_call_stack_t; |
217 | #endif |
218 | |
219 | #if !defined(__CINT__) && !defined(__CLING__) |
220 | |
221 | |
222 | |
223 | |
224 | template<class T> |
225 | JFactory<T>* JEventLoop::GetSingle(const T* &t, const char *tag, bool exception_if_not_one) |
226 | { |
227 | |
228 | |
229 | |
230 | |
231 | |
232 | |
233 | |
234 | |
235 | |
236 | |
237 | |
238 | vector<const T*> v; |
239 | JFactory<T> *fac = Get(v, tag); |
240 | |
241 | if(v.size()!=1){ |
242 | t = NULL; |
243 | if(exception_if_not_one) throw v.size(); |
244 | } |
245 | |
246 | t = v[0]; |
247 | |
248 | return fac; |
249 | } |
250 | |
251 | |
252 | |
253 | |
254 | template<class T> |
255 | JFactory<T>* JEventLoop::Get(vector<const T*> &t, const char *tag, bool allow_deftag) |
256 | { |
257 | |
258 | |
259 | |
260 | |
261 | |
262 | |
263 | |
264 | |
265 | |
266 | |
267 | |
268 | |
269 | |
270 | |
271 | |
272 | |
273 | |
274 | |
275 | |
276 | |
277 | |
278 | |
279 | |
280 | |
281 | |
282 | |
283 | |
284 | |
285 | |
286 | |
287 | |
288 | |
289 | |
290 | const char *mytag = tag==NULL ? "":tag; |
| |
291 | if(strlen(mytag)==0 && allow_deftag){ |
| |
292 | map<string, string>::const_iterator iter = default_tags.find(T::static_className()); |
293 | if(iter!=default_tags.end())tag = iter->second.c_str(); |
| 4 | | Assuming the condition is true | |
|
| |
| |
294 | } |
295 | |
296 | |
297 | |
298 | |
299 | |
300 | |
301 | call_stack_t cs; |
302 | |
303 | |
304 | if(record_call_stack) CallStackStart(cs, caller_name, caller_tag, T::static_className(), tag); |
| 7 | | Assuming field 'record_call_stack' is false | |
|
| |
305 | |
306 | |
307 | JFactory<T>* factory=NULL; |
308 | try{ |
309 | factory = GetFromFactory(t, tag, cs.data_source, allow_deftag); |
| 9 | | Passing value via 2nd parameter 'tag' | |
|
| 10 | | Calling 'JEventLoop::GetFromFactory' | |
|
310 | if(!factory){ |
311 | |
312 | |
313 | |
314 | |
315 | |
316 | |
317 | |
318 | string p; |
319 | try{ |
320 | gPARMS->GetParameter("JANA:AUTOFACTORYCREATE", p); |
321 | }catch(...){} |
322 | if(p.size()==0){ |
323 | jout<<std::endl; |
324 | _DBG__; |
325 | jout<<"No factory of type \""<<T::static_className()<<"\" with tag \""<<tag<<"\" exists."<<std::endl; |
326 | jout<<"If you are reading objects from a file, I can auto-create a factory"<<std::endl; |
327 | jout<<"of the appropriate type to hold the objects, but this feature is turned"<<std::endl; |
328 | jout<<"off by default. To turn it on, set the \"JANA:AUTOFACTORYCREATE\""<<std::endl; |
329 | jout<<"configuration parameter. This can usually be done by passing the"<<std::endl; |
330 | jout<<"following argument to the program from the command line:"<<std::endl; |
331 | jout<<std::endl; |
332 | jout<<" -PJANA:AUTOFACTORYCREATE=1"<<std::endl; |
333 | jout<<std::endl; |
334 | jout<<"Note that since the most commonly expected occurance of this situation."<<std::endl; |
335 | jout<<"is an error, the program will now throw an exception so that the factory."<<std::endl; |
336 | jout<<"call stack can be printed."<<std::endl; |
337 | jout<<std::endl; |
338 | throw exception(); |
339 | }else{ |
340 | AddFactory(new JFactory<T>(tag)); |
341 | jout<<__FILE__<<":"<<__LINE__<<" Auto-created "<<T::static_className()<<":"<<tag<<" factory"<<std::endl; |
342 | |
343 | |
344 | |
345 | factory = GetFromFactory(t, tag, cs.data_source, allow_deftag); |
346 | } |
347 | } |
348 | }catch(exception &e){ |
349 | |
350 | |
351 | error_call_stack_t ecs; |
352 | ecs.factory_name = T::static_className(); |
353 | ecs.tag = tag; |
354 | ecs.filename = NULL; |
355 | error_call_stack.push_back(ecs); |
356 | throw e; |
357 | } |
358 | |
359 | |
360 | if(record_call_stack) CallStackEnd(cs); |
361 | |
362 | return factory; |
363 | } |
364 | |
365 | |
366 | |
367 | |
368 | template<class T> |
369 | JFactory<T>* JEventLoop::GetFromFactory(vector<const T*> &t, const char *tag, data_source_t &data_source, bool allow_deftag) |
370 | { |
371 | |
372 | |
373 | vector<JFactory_base*>::iterator iter=factories.begin(); |
374 | JFactory<T> *factory = NULL; |
375 | string className(T::static_className()); |
376 | |
377 | |
378 | |
379 | const char *mytag = tag==NULL ? "":tag; |
| 11 | | Assuming 'tag' is equal to NULL | |
|
| 12 | | Assuming pointer value is null | |
|
| |
380 | if(strlen(mytag)==0 && allow_deftag){ |
| |
381 | map<string, string>::const_iterator iter = default_tags.find(className); |
382 | if(iter!=default_tags.end())tag = iter->second.c_str(); |
| 15 | | Assuming the condition is false | |
|
| |
383 | } |
384 | |
385 | for(; iter!=factories.end(); iter++){ |
| 17 | | Calling 'operator!=<jana::JFactory_base **, std::vector<jana::JFactory_base *>>' | |
|
| 20 | | Returning from 'operator!=<jana::JFactory_base **, std::vector<jana::JFactory_base *>>' | |
|
| 21 | | Loop condition is true. Entering loop body | |
|
386 | |
387 | |
388 | |
389 | |
390 | |
391 | |
392 | |
393 | |
394 | |
395 | if(className == (*iter)->GetDataClassName())factory = (JFactory<T>*)(*iter); |
| |
396 | if(factory == NULL)continue; |
| 23 | | Assuming 'factory' is not equal to NULL | |
|
| |
397 | const char *factag = factory->Tag()==NULL ? "":factory->Tag(); |
| 25 | | Assuming the condition is true | |
|
| |
398 | if(!strcmp(factag, tag)){ |
| 27 | | Null pointer passed to 2nd parameter expecting 'nonnull' |
|
399 | break; |
400 | }else{ |
401 | factory=NULL; |
402 | } |
403 | } |
404 | |
405 | |
406 | if(!factory){ |
407 | data_source = DATA_NOT_AVAILABLE; |
408 | return NULL; |
409 | } |
410 | |
411 | |
412 | |
413 | |
414 | if(factory->evnt_was_called()){ |
415 | factory->CopyFrom(t); |
416 | data_source = DATA_FROM_CACHE; |
417 | return factory; |
418 | } |
419 | |
420 | |
421 | if(factory->GetCheckSourceFirst()){ |
422 | |
423 | |
424 | |
425 | |
426 | |
427 | jerror_t err = GetFromSource(t, factory); |
428 | if(err == NOERROR){ |
429 | |
430 | |
431 | |
432 | |
433 | |
434 | |
435 | |
436 | |
437 | |
438 | |
439 | |
440 | |
441 | |
442 | |
443 | factory->SetFactoryPointers(); |
444 | data_source = DATA_FROM_SOURCE; |
445 | |
446 | return factory; |
447 | } |
448 | } |
449 | |
450 | |
451 | |
452 | factory->Get(t); |
453 | factory->SetFactoryPointers(); |
454 | data_source = DATA_FROM_FACTORY; |
455 | |
456 | return factory; |
457 | } |
458 | |
459 | |
460 | |
461 | |
462 | template<class T> |
463 | jerror_t JEventLoop::GetFromSource(vector<const T*> &t, JFactory_base *factory) |
464 | { |
465 | |
466 | |
467 | |
468 | |
469 | |
470 | |
471 | |
472 | |
473 | |
474 | |
475 | if(!factory)throw OBJECT_NOT_AVAILABLE; |
476 | |
477 | return event.GetObjects(t, factory); |
478 | } |
479 | |
480 | |
481 | |
482 | |
483 | inline void JEventLoop::CallStackStart(JEventLoop::call_stack_t &cs, const string &caller_name, const string &caller_tag, const string callee_name, const string callee_tag) |
484 | { |
485 | |
486 | |
487 | |
488 | |
489 | |
490 | |
491 | struct itimerval tmr; |
492 | getitimer(ITIMER_PROF, &tmr); |
493 | |
494 | cs.caller_name = this->caller_name; |
495 | cs.caller_tag = this->caller_tag; |
496 | this->caller_name = cs.callee_name = callee_name; |
497 | this->caller_tag = cs.callee_tag = callee_tag; |
498 | cs.start_time = tmr.it_value.tv_sec + tmr.it_value.tv_usec/1.0E6; |
499 | } |
500 | |
501 | |
502 | |
503 | |
504 | inline void JEventLoop::CallStackEnd(JEventLoop::call_stack_t &cs) |
505 | { |
506 | |
507 | |
508 | |
509 | |
510 | struct itimerval tmr; |
511 | getitimer(ITIMER_PROF, &tmr); |
512 | cs.end_time = tmr.it_value.tv_sec + tmr.it_value.tv_usec/1.0E6; |
513 | caller_name = cs.caller_name; |
514 | caller_tag = cs.caller_tag; |
515 | call_stack.push_back(cs); |
516 | } |
517 | |
518 | |
519 | |
520 | |
521 | inline bool JEventLoop::CheckEventBoundary(uint64_t event_numberA, uint64_t event_numberB) |
522 | { |
523 | |
524 | |
525 | |
526 | |
527 | |
528 | |
529 | |
530 | |
531 | |
532 | |
533 | |
534 | if(event.GetRunNumber()!=event_boundaries_run){ |
535 | event_boundaries.clear(); |
536 | JCalibration *jcalib = GetJCalibration(); |
537 | if(jcalib)jcalib->GetEventBoundaries(event_boundaries); |
538 | event_boundaries_run = event.GetRunNumber(); |
539 | } |
540 | |
541 | |
542 | for(unsigned int i=0; i<event_boundaries.size(); i++){ |
543 | uint64_t eb = event_boundaries[i]; |
544 | if((eb - event_numberA)*(eb - event_numberB) < 0.0 || eb==event_numberA){ |
545 | |
546 | return true; |
547 | } |
548 | } |
549 | |
550 | return false; |
551 | } |
552 | |
553 | |
554 | |
555 | |
556 | template<class T> |
557 | const T* JEventLoop::FindByID(JObject::oid_t id) |
558 | { |
559 | |
560 | |
561 | |
562 | |
563 | |
564 | |
565 | |
566 | |
567 | |
568 | |
569 | |
570 | |
571 | |
572 | |
573 | |
574 | for(unsigned int i=0; i<factories.size(); i++){ |
575 | if(factories[i]->GetDataClassName() != T::static_className())continue; |
576 | |
577 | |
578 | |
579 | const JObject *my_obj = factories[i]->GetByID(id); |
580 | if(my_obj)return dynamic_cast<const T*>(my_obj); |
581 | } |
582 | |
583 | return NULL; |
584 | } |
585 | |
586 | |
587 | |
588 | |
589 | template<class T> |
590 | bool JEventLoop::GetCalib(string namepath, map<string,T> &vals) |
591 | { |
592 | |
593 | |
594 | |
595 | |
596 | |
597 | |
598 | |
599 | vals.clear(); |
600 | |
601 | JCalibration *calib = GetJCalibration(); |
602 | if(!calib){ |
603 | _DBG_<<"Unable to get JCalibration object for run "<<event.GetRunNumber()<<std::endl; |
604 | return true; |
605 | } |
606 | |
607 | return calib->Get(namepath, vals, event.GetEventNumber()); |
608 | } |
609 | |
610 | |
611 | |
612 | |
613 | template<class T> bool JEventLoop::GetCalib(string namepath, vector<T> &vals) |
614 | { |
615 | |
616 | |
617 | |
618 | vals.clear(); |
619 | |
620 | JCalibration *calib = GetJCalibration(); |
621 | if(!calib){ |
622 | _DBG_<<"Unable to get JCalibration object for run "<<event.GetRunNumber()<<std::endl; |
623 | return true; |
624 | } |
625 | |
626 | return calib->Get(namepath, vals, event.GetEventNumber()); |
627 | } |
628 | |
629 | |
630 | |
631 | |
632 | template<class T> bool JEventLoop::GetCalib(string namepath, T &val) |
633 | { |
634 | |
635 | |
636 | |
637 | |
638 | |
639 | |
640 | vector<T> vals; |
641 | bool ret = GetCalib(namepath, vals); |
642 | if(vals.empty()) return true; |
643 | val = vals[0]; |
644 | |
645 | return ret; |
646 | } |
647 | |
648 | |
649 | |
650 | |
651 | template<class T> |
652 | bool JEventLoop::GetGeom(string namepath, map<string,T> &vals) |
653 | { |
654 | |
655 | |
656 | |
657 | |
658 | |
659 | |
660 | |
661 | vals.clear(); |
662 | |
663 | JGeometry *geom = GetJGeometry(); |
664 | if(!geom){ |
665 | _DBG_<<"Unable to get JGeometry object for run "<<event.GetRunNumber()<<std::endl; |
666 | return true; |
667 | } |
668 | |
669 | return geom->Get(namepath, vals); |
670 | } |
671 | |
672 | |
673 | |
674 | |
675 | template<class T> bool JEventLoop::GetGeom(string namepath, T &val) |
676 | { |
677 | |
678 | |
679 | |
680 | JGeometry *geom = GetJGeometry(); |
681 | if(!geom){ |
682 | _DBG_<<"Unable to get JGeometry object for run "<<event.GetRunNumber()<<std::endl; |
683 | return true; |
684 | } |
685 | |
686 | return geom->Get(namepath, val); |
687 | } |
688 | |
689 | |
690 | |
691 | |
692 | template<class T> |
693 | void JEventLoop::SetRef(T *t) |
694 | { |
695 | pair<const char*, void*> p(typeid(T).name(), (void*)t); |
696 | user_refs.push_back(p); |
697 | } |
698 | |
699 | |
700 | |
701 | |
702 | template<class T> bool JEventLoop::GetResource(string namepath, T vals, int event_number) |
703 | { |
704 | JResourceManager *resource_manager = GetJResourceManager(); |
705 | if(!resource_manager){ |
706 | string mess = string("Unable to get the JResourceManager object (namepath=\"")+namepath+"\")"; |
707 | throw JException(mess); |
708 | } |
709 | |
710 | return resource_manager->Get(namepath, vals, event_number); |
711 | } |
712 | |
713 | |
714 | |
715 | |
716 | template<class T> |
717 | T* JEventLoop::GetRef(void) |
718 | { |
719 | |
720 | for(unsigned int i=0; i<user_refs.size(); i++){ |
721 | if(user_refs[i].first == typeid(T).name()) return (T*)user_refs[i].second; |
722 | } |
723 | |
724 | return NULL; |
725 | } |
726 | |
727 | |
728 | |
729 | |
730 | template<class T> |
731 | vector<T*> JEventLoop::GetRefsT(void) |
732 | { |
733 | vector<T*> refs; |
734 | for(unsigned int i=0; i<user_refs.size(); i++){ |
735 | if(user_refs[i].first == typeid(T).name()){ |
736 | refs.push_back((T*)user_refs[i].second); |
737 | } |
738 | } |
739 | |
740 | return refs; |
741 | } |
742 | |
743 | |
744 | |
745 | |
746 | template<class T> |
747 | void JEventLoop::RemoveRef(T *t) |
748 | { |
749 | vector<pair<const char*, void*> >::iterator iter; |
750 | for(iter=user_refs.begin(); iter!= user_refs.end(); iter++){ |
751 | if(iter->second == (void*)t){ |
752 | user_refs.erase(iter); |
753 | return; |
754 | } |
755 | } |
756 | _DBG_<<" Attempt to remove user reference not in event loop!" << std::endl; |
757 | } |
758 | |
759 | |
760 | #endif //__CINT__ __CLING__ |
761 | |
762 | } |
763 | |
764 | |
765 | |
766 | #endif // _JEventLoop_ |
767 | |
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 | #ifndef _STL_ITERATOR_H |
61 | #define _STL_ITERATOR_H 1 |
62 | |
63 | #include <bits/cpp_type_traits.h> |
64 | #include <ext/type_traits.h> |
65 | #include <bits/move.h> |
66 | |
67 | namespace std _GLIBCXX_VISIBILITY(default) |
68 | { |
69 | _GLIBCXX_BEGIN_NAMESPACE_VERSION |
70 | |
71 | |
72 | |
73 | |
74 | |
75 | |
76 | |
77 | |
78 | |
79 | |
80 | |
81 | |
82 | |
83 | |
84 | |
85 | |
86 | |
87 | |
88 | |
89 | |
90 | |
91 | |
92 | |
93 | |
94 | |
95 | template<typename _Iterator> |
96 | class reverse_iterator |
97 | : public iterator<typename iterator_traits<_Iterator>::iterator_category, |
98 | typename iterator_traits<_Iterator>::value_type, |
99 | typename iterator_traits<_Iterator>::difference_type, |
100 | typename iterator_traits<_Iterator>::pointer, |
101 | typename iterator_traits<_Iterator>::reference> |
102 | { |
103 | protected: |
104 | _Iterator current; |
105 | |
106 | typedef iterator_traits<_Iterator> __traits_type; |
107 | |
108 | public: |
109 | typedef _Iterator iterator_type; |
110 | typedef typename __traits_type::difference_type difference_type; |
111 | typedef typename __traits_type::pointer pointer; |
112 | typedef typename __traits_type::reference reference; |
113 | |
114 | |
115 | |
116 | |
117 | |
118 | |
119 | |
120 | reverse_iterator() : current() { } |
121 | |
122 | |
123 | |
124 | |
125 | explicit |
126 | reverse_iterator(iterator_type __x) : current(__x) { } |
127 | |
128 | |
129 | |
130 | |
131 | reverse_iterator(const reverse_iterator& __x) |
132 | : current(__x.current) { } |
133 | |
134 | |
135 | |
136 | |
137 | |
138 | template<typename _Iter> |
139 | reverse_iterator(const reverse_iterator<_Iter>& __x) |
140 | : current(__x.base()) { } |
141 | |
142 | |
143 | |
144 | |
145 | iterator_type |
146 | base() const |
147 | { return current; } |
148 | |
149 | |
150 | |
151 | |
152 | |
153 | |
154 | |
155 | |
156 | |
157 | |
158 | |
159 | reference |
160 | operator*() const |
161 | { |
162 | _Iterator __tmp = current; |
163 | return *--__tmp; |
164 | } |
165 | |
166 | |
167 | |
168 | |
169 | |
170 | |
171 | pointer |
172 | operator->() const |
173 | { return &(operator*()); } |
174 | |
175 | |
176 | |
177 | |
178 | |
179 | |
180 | reverse_iterator& |
181 | operator++() |
182 | { |
183 | --current; |
184 | return *this; |
185 | } |
186 | |
187 | |
188 | |
189 | |
190 | |
191 | |
192 | reverse_iterator |
193 | operator++(int) |
194 | { |
195 | reverse_iterator __tmp = *this; |
196 | --current; |
197 | return __tmp; |
198 | } |
199 | |
200 | |
201 | |
202 | |
203 | |
204 | |
205 | reverse_iterator& |
206 | operator--() |
207 | { |
208 | ++current; |
209 | return *this; |
210 | } |
211 | |
212 | |
213 | |
214 | |
215 | |
216 | |
217 | reverse_iterator |
218 | operator--(int) |
219 | { |
220 | reverse_iterator __tmp = *this; |
221 | ++current; |
222 | return __tmp; |
223 | } |
224 | |
225 | |
226 | |
227 | |
228 | |
229 | |
230 | reverse_iterator |
231 | operator+(difference_type __n) const |
232 | { return reverse_iterator(current - __n); } |
233 | |
234 | |
235 | |
236 | |
237 | |
238 | |
239 | |
240 | reverse_iterator& |
241 | operator+=(difference_type __n) |
242 | { |
243 | current -= __n; |
244 | return *this; |
245 | } |
246 | |
247 | |
248 | |
249 | |
250 | |
251 | |
252 | reverse_iterator |
253 | operator-(difference_type __n) const |
254 | { return reverse_iterator(current + __n); } |
255 | |
256 | |
257 | |
258 | |
259 | |
260 | |
261 | |
262 | reverse_iterator& |
263 | operator-=(difference_type __n) |
264 | { |
265 | current += __n; |
266 | return *this; |
267 | } |
268 | |
269 | |
270 | |
271 | |
272 | |
273 | |
274 | reference |
275 | operator[](difference_type __n) const |
276 | { return *(*this + __n); } |
277 | }; |
278 | |
279 | |
280 | |
281 | |
282 | |
283 | |
284 | |
285 | |
286 | |
287 | |
288 | |
289 | template<typename _Iterator> |
290 | inline bool |
291 | operator==(const reverse_iterator<_Iterator>& __x, |
292 | const reverse_iterator<_Iterator>& __y) |
293 | { return __x.base() == __y.base(); } |
294 | |
295 | template<typename _Iterator> |
296 | inline bool |
297 | operator<(const reverse_iterator<_Iterator>& __x, |
298 | const reverse_iterator<_Iterator>& __y) |
299 | { return __y.base() < __x.base(); } |
300 | |
301 | template<typename _Iterator> |
302 | inline bool |
303 | operator!=(const reverse_iterator<_Iterator>& __x, |
304 | const reverse_iterator<_Iterator>& __y) |
305 | { return !(__x == __y); } |
306 | |
307 | template<typename _Iterator> |
308 | inline bool |
309 | operator>(const reverse_iterator<_Iterator>& __x, |
310 | const reverse_iterator<_Iterator>& __y) |
311 | { return __y < __x; } |
312 | |
313 | template<typename _Iterator> |
314 | inline bool |
315 | operator<=(const reverse_iterator<_Iterator>& __x, |
316 | const reverse_iterator<_Iterator>& __y) |
317 | { return !(__y < __x); } |
318 | |
319 | template<typename _Iterator> |
320 | inline bool |
321 | operator>=(const reverse_iterator<_Iterator>& __x, |
322 | const reverse_iterator<_Iterator>& __y) |
323 | { return !(__x < __y); } |
324 | |
325 | template<typename _Iterator> |
326 | inline typename reverse_iterator<_Iterator>::difference_type |
327 | operator-(const reverse_iterator<_Iterator>& __x, |
328 | const reverse_iterator<_Iterator>& __y) |
329 | { return __y.base() - __x.base(); } |
330 | |
331 | template<typename _Iterator> |
332 | inline reverse_iterator<_Iterator> |
333 | operator+(typename reverse_iterator<_Iterator>::difference_type __n, |
334 | const reverse_iterator<_Iterator>& __x) |
335 | { return reverse_iterator<_Iterator>(__x.base() - __n); } |
336 | |
337 | |
338 | |
339 | template<typename _IteratorL, typename _IteratorR> |
340 | inline bool |
341 | operator==(const reverse_iterator<_IteratorL>& __x, |
342 | const reverse_iterator<_IteratorR>& __y) |
343 | { return __x.base() == __y.base(); } |
344 | |
345 | template<typename _IteratorL, typename _IteratorR> |
346 | inline bool |
347 | operator<(const reverse_iterator<_IteratorL>& __x, |
348 | const reverse_iterator<_IteratorR>& __y) |
349 | { return __y.base() < __x.base(); } |
350 | |
351 | template<typename _IteratorL, typename _IteratorR> |
352 | inline bool |
353 | operator!=(const reverse_iterator<_IteratorL>& __x, |
354 | const reverse_iterator<_IteratorR>& __y) |
355 | { return !(__x == __y); } |
356 | |
357 | template<typename _IteratorL, typename _IteratorR> |
358 | inline bool |
359 | operator>(const reverse_iterator<_IteratorL>& __x, |
360 | const reverse_iterator<_IteratorR>& __y) |
361 | { return __y < __x; } |
362 | |
363 | template<typename _IteratorL, typename _IteratorR> |
364 | inline bool |
365 | operator<=(const reverse_iterator<_IteratorL>& __x, |
366 | const reverse_iterator<_IteratorR>& __y) |
367 | { return !(__y < __x); } |
368 | |
369 | template<typename _IteratorL, typename _IteratorR> |
370 | inline bool |
371 | operator>=(const reverse_iterator<_IteratorL>& __x, |
372 | const reverse_iterator<_IteratorR>& __y) |
373 | { return !(__x < __y); } |
374 | |
375 | template<typename _IteratorL, typename _IteratorR> |
376 | #if __cplusplus >= 201103L |
377 | |
378 | inline auto |
379 | operator-(const reverse_iterator<_IteratorL>& __x, |
380 | const reverse_iterator<_IteratorR>& __y) |
381 | -> decltype(__y.base() - __x.base()) |
382 | #else |
383 | inline typename reverse_iterator<_IteratorL>::difference_type |
384 | operator-(const reverse_iterator<_IteratorL>& __x, |
385 | const reverse_iterator<_IteratorR>& __y) |
386 | #endif |
387 | { return __y.base() - __x.base(); } |
388 | |
389 | |
390 | |
391 | |
392 | |
393 | |
394 | |
395 | |
396 | |
397 | |
398 | |
399 | |
400 | |
401 | template<typename _Container> |
402 | class back_insert_iterator |
403 | : public iterator<output_iterator_tag, void, void, void, void> |
404 | { |
405 | protected: |
406 | _Container* container; |
407 | |
408 | public: |
409 | |
410 | typedef _Container container_type; |
411 | |
412 | |
413 | explicit |
414 | back_insert_iterator(_Container& __x) : container(&__x) { } |
415 | |
416 | |
417 | |
418 | |
419 | |
420 | |
421 | |
422 | |
423 | |
424 | |
425 | |
426 | |
427 | #if __cplusplus < 201103L |
428 | back_insert_iterator& |
429 | operator=(typename _Container::const_reference __value) |
430 | { |
431 | container->push_back(__value); |
432 | return *this; |
433 | } |
434 | #else |
435 | back_insert_iterator& |
436 | operator=(const typename _Container::value_type& __value) |
437 | { |
438 | container->push_back(__value); |
439 | return *this; |
440 | } |
441 | |
442 | back_insert_iterator& |
443 | operator=(typename _Container::value_type&& __value) |
444 | { |
445 | container->push_back(std::move(__value)); |
446 | return *this; |
447 | } |
448 | #endif |
449 | |
450 | |
451 | back_insert_iterator& |
452 | operator*() |
453 | { return *this; } |
454 | |
455 | |
456 | back_insert_iterator& |
457 | operator++() |
458 | { return *this; } |
459 | |
460 | |
461 | back_insert_iterator |
462 | operator++(int) |
463 | { return *this; } |
464 | }; |
465 | |
466 | |
467 | |
468 | |
469 | |
470 | |
471 | |
472 | |
473 | |
474 | |
475 | |
476 | |
477 | template<typename _Container> |
478 | inline back_insert_iterator<_Container> |
479 | back_inserter(_Container& __x) |
480 | { return back_insert_iterator<_Container>(__x); } |
481 | |
482 | |
483 | |
484 | |
485 | |
486 | |
487 | |
488 | |
489 | |
490 | |
491 | |
492 | template<typename _Container> |
493 | class front_insert_iterator |
494 | : public iterator<output_iterator_tag, void, void, void, void> |
495 | { |
496 | protected: |
497 | _Container* container; |
498 | |
499 | public: |
500 | |
501 | typedef _Container container_type; |
502 | |
503 | |
504 | explicit front_insert_iterator(_Container& __x) : container(&__x) { } |
505 | |
506 | |
507 | |
508 | |
509 | |
510 | |
511 | |
512 | |
513 | |
514 | |
515 | |
516 | |
517 | #if __cplusplus < 201103L |
518 | front_insert_iterator& |
519 | operator=(typename _Container::const_reference __value) |
520 | { |
521 | container->push_front(__value); |
522 | return *this; |
523 | } |
524 | #else |
525 | front_insert_iterator& |
526 | operator=(const typename _Container::value_type& __value) |
527 | { |
528 | container->push_front(__value); |
529 | return *this; |
530 | } |
531 | |
532 | front_insert_iterator& |
533 | operator=(typename _Container::value_type&& __value) |
534 | { |
535 | container->push_front(std::move(__value)); |
536 | return *this; |
537 | } |
538 | #endif |
539 | |
540 | |
541 | front_insert_iterator& |
542 | operator*() |
543 | { return *this; } |
544 | |
545 | |
546 | front_insert_iterator& |
547 | operator++() |
548 | { return *this; } |
549 | |
550 | |
551 | front_insert_iterator |
552 | operator++(int) |
553 | { return *this; } |
554 | }; |
555 | |
556 | |
557 | |
558 | |
559 | |
560 | |
561 | |
562 | |
563 | |
564 | |
565 | |
566 | |
567 | template<typename _Container> |
568 | inline front_insert_iterator<_Container> |
569 | front_inserter(_Container& __x) |
570 | { return front_insert_iterator<_Container>(__x); } |
571 | |
572 | |
573 | |
574 | |
575 | |
576 | |
577 | |
578 | |
579 | |
580 | |
581 | |
582 | |
583 | |
584 | |
585 | |
586 | template<typename _Container> |
587 | class insert_iterator |
588 | : public iterator<output_iterator_tag, void, void, void, void> |
589 | { |
590 | protected: |
591 | _Container* container; |
592 | typename _Container::iterator iter; |
593 | |
594 | public: |
595 | |
596 | typedef _Container container_type; |
597 | |
598 | |
599 | |
600 | |
601 | |
602 | insert_iterator(_Container& __x, typename _Container::iterator __i) |
603 | : container(&__x), iter(__i) {} |
604 | |
605 | |
606 | |
607 | |
608 | |
609 | |
610 | |
611 | |
612 | |
613 | |
614 | |
615 | |
616 | |
617 | |
618 | |
619 | |
620 | |
621 | |
622 | |
623 | |
624 | |
625 | |
626 | |
627 | |
628 | #if __cplusplus < 201103L |
629 | insert_iterator& |
630 | operator=(typename _Container::const_reference __value) |
631 | { |
632 | iter = container->insert(iter, __value); |
633 | ++iter; |
634 | return *this; |
635 | } |
636 | #else |
637 | insert_iterator& |
638 | operator=(const typename _Container::value_type& __value) |
639 | { |
640 | iter = container->insert(iter, __value); |
641 | ++iter; |
642 | return *this; |
643 | } |
644 | |
645 | insert_iterator& |
646 | operator=(typename _Container::value_type&& __value) |
647 | { |
648 | iter = container->insert(iter, std::move(__value)); |
649 | ++iter; |
650 | return *this; |
651 | } |
652 | #endif |
653 | |
654 | |
655 | insert_iterator& |
656 | operator*() |
657 | { return *this; } |
658 | |
659 | |
660 | insert_iterator& |
661 | operator++() |
662 | { return *this; } |
663 | |
664 | |
665 | insert_iterator& |
666 | operator++(int) |
667 | { return *this; } |
668 | }; |
669 | |
670 | |
671 | |
672 | |
673 | |
674 | |
675 | |
676 | |
677 | |
678 | |
679 | |
680 | |
681 | template<typename _Container, typename _Iterator> |
682 | inline insert_iterator<_Container> |
683 | inserter(_Container& __x, _Iterator __i) |
684 | { |
685 | return insert_iterator<_Container>(__x, |
686 | typename _Container::iterator(__i)); |
687 | } |
688 | |
689 | |
690 | |
691 | _GLIBCXX_END_NAMESPACE_VERSION |
692 | } |
693 | |
694 | namespace __gnu_cxx _GLIBCXX_VISIBILITY(default) |
695 | { |
696 | _GLIBCXX_BEGIN_NAMESPACE_VERSION |
697 | |
698 | |
699 | |
700 | |
701 | |
702 | |
703 | |
704 | |
705 | using std::iterator_traits; |
706 | using std::iterator; |
707 | template<typename _Iterator, typename _Container> |
708 | class __normal_iterator |
709 | { |
710 | protected: |
711 | _Iterator _M_current; |
712 | |
713 | typedef iterator_traits<_Iterator> __traits_type; |
714 | |
715 | public: |
716 | typedef _Iterator iterator_type; |
717 | typedef typename __traits_type::iterator_category iterator_category; |
718 | typedef typename __traits_type::value_type value_type; |
719 | typedef typename __traits_type::difference_type difference_type; |
720 | typedef typename __traits_type::reference reference; |
721 | typedef typename __traits_type::pointer pointer; |
722 | |
723 | _GLIBCXX_CONSTEXPR __normal_iterator() : _M_current(_Iterator()) { } |
724 | |
725 | explicit |
726 | __normal_iterator(const _Iterator& __i) : _M_current(__i) { } |
727 | |
728 | |
729 | template<typename _Iter> |
730 | __normal_iterator(const __normal_iterator<_Iter, |
731 | typename __enable_if< |
732 | (std::__are_same<_Iter, typename _Container::pointer>::__value), |
733 | _Container>::__type>& __i) |
734 | : _M_current(__i.base()) { } |
735 | |
736 | |
737 | reference |
738 | operator*() const |
739 | { return *_M_current; } |
740 | |
741 | pointer |
742 | operator->() const |
743 | { return _M_current; } |
744 | |
745 | __normal_iterator& |
746 | operator++() |
747 | { |
748 | ++_M_current; |
749 | return *this; |
750 | } |
751 | |
752 | __normal_iterator |
753 | operator++(int) |
754 | { return __normal_iterator(_M_current++); } |
755 | |
756 | |
757 | __normal_iterator& |
758 | operator--() |
759 | { |
760 | --_M_current; |
761 | return *this; |
762 | } |
763 | |
764 | __normal_iterator |
765 | operator--(int) |
766 | { return __normal_iterator(_M_current--); } |
767 | |
768 | |
769 | reference |
770 | operator[](const difference_type& __n) const |
771 | { return _M_current[__n]; } |
772 | |
773 | __normal_iterator& |
774 | operator+=(const difference_type& __n) |
775 | { _M_current += __n; return *this; } |
776 | |
777 | __normal_iterator |
778 | operator+(const difference_type& __n) const |
779 | { return __normal_iterator(_M_current + __n); } |
780 | |
781 | __normal_iterator& |
782 | operator-=(const difference_type& __n) |
783 | { _M_current -= __n; return *this; } |
784 | |
785 | __normal_iterator |
786 | operator-(const difference_type& __n) const |
787 | { return __normal_iterator(_M_current - __n); } |
788 | |
789 | const _Iterator& |
790 | base() const |
791 | { return _M_current; } |
792 | }; |
793 | |
794 | |
795 | |
796 | |
797 | |
798 | |
799 | |
800 | |
801 | |
802 | |
803 | template<typename _IteratorL, typename _IteratorR, typename _Container> |
804 | inline bool |
805 | operator==(const __normal_iterator<_IteratorL, _Container>& __lhs, |
806 | const __normal_iterator<_IteratorR, _Container>& __rhs) |
807 | { return __lhs.base() == __rhs.base(); } |
808 | |
809 | template<typename _Iterator, typename _Container> |
810 | inline bool |
811 | operator==(const __normal_iterator<_Iterator, _Container>& __lhs, |
812 | const __normal_iterator<_Iterator, _Container>& __rhs) |
813 | { return __lhs.base() == __rhs.base(); } |
814 | |
815 | template<typename _IteratorL, typename _IteratorR, typename _Container> |
816 | inline bool |
817 | operator!=(const __normal_iterator<_IteratorL, _Container>& __lhs, |
818 | const __normal_iterator<_IteratorR, _Container>& __rhs) |
819 | { return __lhs.base() != __rhs.base(); } |
820 | |
821 | template<typename _Iterator, typename _Container> |
822 | inline bool |
823 | operator!=(const __normal_iterator<_Iterator, _Container>& __lhs, |
824 | const __normal_iterator<_Iterator, _Container>& __rhs) |
825 | { return __lhs.base() != __rhs.base(); } |
| 18 | | Assuming the condition is true | |
|
| 19 | | Returning the value 1, which participates in a condition later | |
|
826 | |
827 | |
828 | template<typename _IteratorL, typename _IteratorR, typename _Container> |
829 | inline bool |
830 | operator<(const __normal_iterator<_IteratorL, _Container>& __lhs, |
831 | const __normal_iterator<_IteratorR, _Container>& __rhs) |
832 | { return __lhs.base() < __rhs.base(); } |
833 | |
834 | template<typename _Iterator, typename _Container> |
835 | inline bool |
836 | operator<(const __normal_iterator<_Iterator, _Container>& __lhs, |
837 | const __normal_iterator<_Iterator, _Container>& __rhs) |
838 | { return __lhs.base() < __rhs.base(); } |
839 | |
840 | template<typename _IteratorL, typename _IteratorR, typename _Container> |
841 | inline bool |
842 | operator>(const __normal_iterator<_IteratorL, _Container>& __lhs, |
843 | const __normal_iterator<_IteratorR, _Container>& __rhs) |
844 | { return __lhs.base() > __rhs.base(); } |
845 | |
846 | template<typename _Iterator, typename _Container> |
847 | inline bool |
848 | operator>(const __normal_iterator<_Iterator, _Container>& __lhs, |
849 | const __normal_iterator<_Iterator, _Container>& __rhs) |
850 | { return __lhs.base() > __rhs.base(); } |
851 | |
852 | template<typename _IteratorL, typename _IteratorR, typename _Container> |
853 | inline bool |
854 | operator<=(const __normal_iterator<_IteratorL, _Container>& __lhs, |
855 | const __normal_iterator<_IteratorR, _Container>& __rhs) |
856 | { return __lhs.base() <= __rhs.base(); } |
857 | |
858 | template<typename _Iterator, typename _Container> |
859 | inline bool |
860 | operator<=(const __normal_iterator<_Iterator, _Container>& __lhs, |
861 | const __normal_iterator<_Iterator, _Container>& __rhs) |
862 | { return __lhs.base() <= __rhs.base(); } |
863 | |
864 | template<typename _IteratorL, typename _IteratorR, typename _Container> |
865 | inline bool |
866 | operator>=(const __normal_iterator<_IteratorL, _Container>& __lhs, |
867 | const __normal_iterator<_IteratorR, _Container>& __rhs) |
868 | { return __lhs.base() >= __rhs.base(); } |
869 | |
870 | template<typename _Iterator, typename _Container> |
871 | inline bool |
872 | operator>=(const __normal_iterator<_Iterator, _Container>& __lhs, |
873 | const __normal_iterator<_Iterator, _Container>& __rhs) |
874 | { return __lhs.base() >= __rhs.base(); } |
875 | |
876 | |
877 | |
878 | |
879 | |
880 | template<typename _IteratorL, typename _IteratorR, typename _Container> |
881 | #if __cplusplus >= 201103L |
882 | |
883 | inline auto |
884 | operator-(const __normal_iterator<_IteratorL, _Container>& __lhs, |
885 | const __normal_iterator<_IteratorR, _Container>& __rhs) |
886 | -> decltype(__lhs.base() - __rhs.base()) |
887 | #else |
888 | inline typename __normal_iterator<_IteratorL, _Container>::difference_type |
889 | operator-(const __normal_iterator<_IteratorL, _Container>& __lhs, |
890 | const __normal_iterator<_IteratorR, _Container>& __rhs) |
891 | #endif |
892 | { return __lhs.base() - __rhs.base(); } |
893 | |
894 | template<typename _Iterator, typename _Container> |
895 | inline typename __normal_iterator<_Iterator, _Container>::difference_type |
896 | operator-(const __normal_iterator<_Iterator, _Container>& __lhs, |
897 | const __normal_iterator<_Iterator, _Container>& __rhs) |
898 | { return __lhs.base() - __rhs.base(); } |
899 | |
900 | template<typename _Iterator, typename _Container> |
901 | inline __normal_iterator<_Iterator, _Container> |
902 | operator+(typename __normal_iterator<_Iterator, _Container>::difference_type |
903 | __n, const __normal_iterator<_Iterator, _Container>& __i) |
904 | { return __normal_iterator<_Iterator, _Container>(__i.base() + __n); } |
905 | |
906 | _GLIBCXX_END_NAMESPACE_VERSION |
907 | } |
908 | |
909 | #if __cplusplus >= 201103L |
910 | |
911 | namespace std _GLIBCXX_VISIBILITY(default) |
912 | { |
913 | _GLIBCXX_BEGIN_NAMESPACE_VERSION |
914 | |
915 | |
916 | |
917 | |
918 | |
919 | |
920 | |
921 | |
922 | |
923 | |
924 | |
925 | |
926 | |
927 | |
928 | |
929 | template<typename _Iterator> |
930 | class move_iterator |
931 | { |
932 | protected: |
933 | _Iterator _M_current; |
934 | |
935 | typedef iterator_traits<_Iterator> __traits_type; |
936 | |
937 | public: |
938 | typedef _Iterator iterator_type; |
939 | typedef typename __traits_type::iterator_category iterator_category; |
940 | typedef typename __traits_type::value_type value_type; |
941 | typedef typename __traits_type::difference_type difference_type; |
942 | |
943 | typedef _Iterator pointer; |
944 | typedef value_type&& reference; |
945 | |
946 | move_iterator() |
947 | : _M_current() { } |
948 | |
949 | explicit |
950 | move_iterator(iterator_type __i) |
951 | : _M_current(__i) { } |
952 | |
953 | template<typename _Iter> |
954 | move_iterator(const move_iterator<_Iter>& __i) |
955 | : _M_current(__i.base()) { } |
956 | |
957 | iterator_type |
958 | base() const |
959 | { return _M_current; } |
960 | |
961 | reference |
962 | operator*() const |
963 | { return std::move(*_M_current); } |
964 | |
965 | pointer |
966 | operator->() const |
967 | { return _M_current; } |
968 | |
969 | move_iterator& |
970 | operator++() |
971 | { |
972 | ++_M_current; |
973 | return *this; |
974 | } |
975 | |
976 | move_iterator |
977 | operator++(int) |
978 | { |
979 | move_iterator __tmp = *this; |
980 | ++_M_current; |
981 | return __tmp; |
982 | } |
983 | |
984 | move_iterator& |
985 | operator--() |
986 | { |
987 | --_M_current; |
988 | return *this; |
989 | } |
990 | |
991 | move_iterator |
992 | operator--(int) |
993 | { |
994 | move_iterator __tmp = *this; |
995 | --_M_current; |
996 | return __tmp; |
997 | } |
998 | |
999 | move_iterator |
1000 | operator+(difference_type __n) const |
1001 | { return move_iterator(_M_current + __n); } |
1002 | |
1003 | move_iterator& |
1004 | operator+=(difference_type __n) |
1005 | { |
1006 | _M_current += __n; |
1007 | return *this; |
1008 | } |
1009 | |
1010 | move_iterator |
1011 | operator-(difference_type __n) const |
1012 | { return move_iterator(_M_current - __n); } |
1013 | |
1014 | move_iterator& |
1015 | operator-=(difference_type __n) |
1016 | { |
1017 | _M_current -= __n; |
1018 | return *this; |
1019 | } |
1020 | |
1021 | reference |
1022 | operator[](difference_type __n) const |
1023 | { return std::move(_M_current[__n]); } |
1024 | }; |
1025 | |
1026 | |
1027 | |
1028 | |
1029 | template<typename _IteratorL, typename _IteratorR> |
1030 | inline bool |
1031 | operator==(const move_iterator<_IteratorL>& __x, |
1032 | const move_iterator<_IteratorR>& __y) |
1033 | { return __x.base() == __y.base(); } |
1034 | |
1035 | template<typename _Iterator> |
1036 | inline bool |
1037 | operator==(const move_iterator<_Iterator>& __x, |
1038 | const move_iterator<_Iterator>& __y) |
1039 | { return __x.base() == __y.base(); } |
1040 | |
1041 | template<typename _IteratorL, typename _IteratorR> |
1042 | inline bool |
1043 | operator!=(const move_iterator<_IteratorL>& __x, |
1044 | const move_iterator<_IteratorR>& __y) |
1045 | { return !(__x == __y); } |
1046 | |
1047 | template<typename _Iterator> |
1048 | inline bool |
1049 | operator!=(const move_iterator<_Iterator>& __x, |
1050 | const move_iterator<_Iterator>& __y) |
1051 | { return !(__x == __y); } |
1052 | |
1053 | template<typename _IteratorL, typename _IteratorR> |
1054 | inline bool |
1055 | operator<(const move_iterator<_IteratorL>& __x, |
1056 | const move_iterator<_IteratorR>& __y) |
1057 | { return __x.base() < __y.base(); } |
1058 | |
1059 | template<typename _Iterator> |
1060 | inline bool |
1061 | operator<(const move_iterator<_Iterator>& __x, |
1062 | const move_iterator<_Iterator>& __y) |
1063 | { return __x.base() < __y.base(); } |
1064 | |
1065 | template<typename _IteratorL, typename _IteratorR> |
1066 | inline bool |
1067 | operator<=(const move_iterator<_IteratorL>& __x, |
1068 | const move_iterator<_IteratorR>& __y) |
1069 | { return !(__y < __x); } |
1070 | |
1071 | template<typename _Iterator> |
1072 | inline bool |
1073 | operator<=(const move_iterator<_Iterator>& __x, |
1074 | const move_iterator<_Iterator>& __y) |
1075 | { return !(__y < __x); } |
1076 | |
1077 | template<typename _IteratorL, typename _IteratorR> |
1078 | inline bool |
1079 | operator>(const move_iterator<_IteratorL>& __x, |
1080 | const move_iterator<_IteratorR>& __y) |
1081 | { return __y < __x; } |
1082 | |
1083 | template<typename _Iterator> |
1084 | inline bool |
1085 | operator>(const move_iterator<_Iterator>& __x, |
1086 | const move_iterator<_Iterator>& __y) |
1087 | { return __y < __x; } |
1088 | |
1089 | template<typename _IteratorL, typename _IteratorR> |
1090 | inline bool |
1091 | operator>=(const move_iterator<_IteratorL>& __x, |
1092 | const move_iterator<_IteratorR>& __y) |
1093 | { return !(__x < __y); } |
1094 | |
1095 | template<typename _Iterator> |
1096 | inline bool |
1097 | operator>=(const move_iterator<_Iterator>& __x, |
1098 | const move_iterator<_Iterator>& __y) |
1099 | { return !(__x < __y); } |
1100 | |
1101 | |
1102 | template<typename _IteratorL, typename _IteratorR> |
1103 | inline auto |
1104 | operator-(const move_iterator<_IteratorL>& __x, |
1105 | const move_iterator<_IteratorR>& __y) |
1106 | -> decltype(__x.base() - __y.base()) |
1107 | { return __x.base() - __y.base(); } |
1108 | |
1109 | template<typename _Iterator> |
1110 | inline auto |
1111 | operator-(const move_iterator<_Iterator>& __x, |
1112 | const move_iterator<_Iterator>& __y) |
1113 | -> decltype(__x.base() - __y.base()) |
1114 | { return __x.base() - __y.base(); } |
1115 | |
1116 | template<typename _Iterator> |
1117 | inline move_iterator<_Iterator> |
1118 | operator+(typename move_iterator<_Iterator>::difference_type __n, |
1119 | const move_iterator<_Iterator>& __x) |
1120 | { return __x + __n; } |
1121 | |
1122 | template<typename _Iterator> |
1123 | inline move_iterator<_Iterator> |
1124 | make_move_iterator(_Iterator __i) |
1125 | { return move_iterator<_Iterator>(__i); } |
1126 | |
1127 | template<typename _Iterator, typename _ReturnType |
1128 | = typename conditional<__move_if_noexcept_cond |
1129 | <typename iterator_traits<_Iterator>::value_type>::value, |
1130 | _Iterator, move_iterator<_Iterator>>::type> |
1131 | inline _ReturnType |
1132 | __make_move_if_noexcept_iterator(_Iterator __i) |
1133 | { return _ReturnType(__i); } |
1134 | |
1135 | |
1136 | |
1137 | _GLIBCXX_END_NAMESPACE_VERSION |
1138 | } |
1139 | |
1140 | #define _GLIBCXX_MAKE_MOVE_ITERATOR(_Iter) std::make_move_iterator(_Iter) |
1141 | #define _GLIBCXX_MAKE_MOVE_IF_NOEXCEPT_ITERATOR(_Iter) \ |
1142 | std::__make_move_if_noexcept_iterator(_Iter) |
1143 | #else |
1144 | #define _GLIBCXX_MAKE_MOVE_ITERATOR(_Iter) (_Iter) |
1145 | #define _GLIBCXX_MAKE_MOVE_IF_NOEXCEPT_ITERATOR(_Iter) (_Iter) |
1146 | #endif // C++11 |
1147 | |
1148 | #endif |