15 add_cdc_cr_track_finding,
16 add_cdc_track_finding,
17 add_cr_track_fit_and_track_creator,
18 add_eclcdc_track_finding,
20 add_hit_preparation_modules,
23 add_flipping_of_recoTracks,
24 add_pxd_cr_track_finding,
25 add_pxd_track_finding,
26 add_svd_track_finding,
27 add_track_fit_and_track_creator,
28 add_prefilter_track_fit_and_track_creator,
29 add_vxd_track_finding_vxdtf2,
30 add_svd_standalone_tracking,
38from pxd
import add_pxd_reconstruction
41def add_tracking_reconstruction(path, components=None, pruneTracks=False, skipGeometryAdding=False,
42 mcTrackFinding=False, trackFitHypotheses=None,
43 reco_tracks="RecoTracks", prune_temporary_tracks=True, fit_tracks=True,
44 use_second_cdc_hits=False, skipHitPreparerAdding=False,
45 svd_standalone_mode="VXDTF2",
46 use_svd_to_cdc_ckf=True, use_ecl_to_cdc_ckf=False,
47 add_cdcTrack_QI=True, add_vxdTrack_QI=False, add_recoTrack_QI=False,
48 pxd_filtering_offline=False,
49 create_intercepts_for_pxd_ckf=False,
50 append_full_grid_cdc_eventt0=True,
51 v0_finding=True, flip_recoTrack=True,
52 skip_full_grid_cdc_eventt0_if_svd_time_present=True):
54 This function adds the **standard tracking reconstruction** modules
64 :ref:`CDC to SVD CKF<tracking_cdc2svd_ckf>` are combined
with SVD standalone tracks found
in the previous step using the \
65 :ref:`CDC to SVD Seed CKF<tracking_svdcdc_merger_ckf>`
75 PXD hits are
not available on HLT. At the end of the tracking chain on HLT we have the\
76 :ref:`PXD Region Of Interest Finding<tracking_pxdDataReduction>`, that consists of extrapolating\
77 the tracks on the PXD sensors
and defining regions
in which we expect to find the hit.\
78 Only fired pixels inside these regions reach Event Builder 2.
81 in particular we search
for displaced vertices to reconstruct K-short, Lambda
and\
82 photon-conversions, see :ref:`V0 Finding<tracking_v0Finding>`.
85 flip
and refit them to fix the charge, see :ref:`Flip&Refit<trk_flipNrefit>`.
89 :param path: the path to add the tracking reconstruction modules to
90 :param components: the list of geometry components
in use
or None for all components.
91 :param pruneTracks:
if true, delete all hits
except the first
and the last
in the found tracks.
92 :param skipGeometryAdding: (advanced flag) the tracking modules need the geometry module
and will add it,
93 if it
is not already present
in the path. In a setup
with multiple (conditional) paths however, it can
not
94 determine,
if the geometry
is already loaded. This flag can be used o just turn off the geometry adding at
95 all (but you will have to add it on your own then).
96 :param skipHitPreparerAdding: (advanced flag)
if true, do
not add the hit preparation (esp. VXD cluster creation
97 modules. This
is useful
if they have been added before already.
98 :param mcTrackFinding:
if true, use the MC track finders instead of the realistic ones.
99 :param reco_tracks: name of the StoreArray where the reco tracks should be stored
100 :param prune_temporary_tracks:
if false, store all information of the single CDC
and VXD tracks before merging.
102 :param fit_tracks:
if false, the final track find
and the TrackCreator module will no be executed
103 :param use_second_cdc_hits:
if true, the second hit information will be used
in the CDC track finding.
104 :param trackFitHypotheses: which pdg hypothesis to fit. Defaults to [211, 321, 2212].
105 :param svd_standalone_mode: Which SVD standalone tracking
is used.
106 Options are
"VXDTF2",
"SVDHough",
"VXDTF2_and_SVDHough",
and "SVDHough_and_VXDTF2".
108 :param use_svd_to_cdc_ckf:
if true, add SVD to CDC CKF module.
109 :param use_ecl_to_cdc_ckf:
if true, add ECL to CDC CKF module.
110 :param add_cdcTrack_QI:
if true, add the MVA track quality estimation
111 to the path that sets the quality indicator property of the found CDC standalone tracks
112 :param add_vxdTrack_QI:
if true, add the MVA track quality estimation
113 to the path that sets the quality indicator property of the found VXDTF2 tracks
114 (ATTENTION: Standard triplet QI of VXDTF2
is replaced
in this case
115 -> setting this option to
'True' will have some influence on the final track collection)
116 :param add_recoTrack_QI:
if true, add the MVA track quality estimation
117 to the path that sets the quality indicator property of all found reco tracks
118 (Both other QIs needed
as input.)
119 :param pxd_filtering_offline: If
True, PXD data reduction (ROI filtering)
is applied during the track reconstruction.
120 The reconstructed SVD/CDC tracks are used to define the ROIs
and reject all PXD clusters outside of these.
121 :param create_intercepts_for_pxd_ckf: If
True, the PXDROIFinder
is added to the path to create PXDIntercepts to be used
122 for hit filtering when creating the CKF relations. This independent of the offline PXD digit filtering which
is
123 steered by
'pxd_filtering_offline'. This can be applied
for both data
and MC.
124 :param append_full_grid_cdc_eventt0: If
True, the module FullGridChi2TrackTimeExtractor
is added to the path
125 and provides the CDC temporary EventT0.
126 :param v0_finding:
if false, the V0Finder module
is not executed
127 :param flip_recoTrack:
if true, add the recoTracks flipping function
in the postfilter (only
if PXD
is present)
128 :param skip_full_grid_cdc_eventt0_if_svd_time_present:
if true,
and if also append_full_grid_cdc_eventt0
is true, the
129 FullGridChi2TrackTimeExtractor
is only executed
in the events where no SVD-based EventT0
is found. If false, but
130 append_full_grid_cdc_eventt0
is true, FullGridChi2TrackTimeExtractor will be executed
in each event regardless of
131 SVD EventT0 being present. Has no effect
if append_full_grid_cdc_eventt0
is false. Default: true
134 add_prefilter_tracking_reconstruction(
136 components=components,
137 skipGeometryAdding=skipGeometryAdding,
138 mcTrackFinding=mcTrackFinding,
139 trackFitHypotheses=trackFitHypotheses,
140 reco_tracks=reco_tracks,
141 prune_temporary_tracks=prune_temporary_tracks,
142 fit_tracks=fit_tracks,
143 use_second_cdc_hits=use_second_cdc_hits,
144 skipHitPreparerAdding=skipHitPreparerAdding,
145 svd_standalone_mode=svd_standalone_mode,
146 use_svd_to_cdc_ckf=use_svd_to_cdc_ckf,
147 use_ecl_to_cdc_ckf=use_ecl_to_cdc_ckf,
148 add_cdcTrack_QI=add_cdcTrack_QI,
149 add_vxdTrack_QI=add_vxdTrack_QI,
150 add_recoTrack_QI=add_recoTrack_QI,
151 pxd_filtering_offline=pxd_filtering_offline,
152 create_intercepts_for_pxd_ckf=create_intercepts_for_pxd_ckf,
153 append_full_grid_cdc_eventt0=append_full_grid_cdc_eventt0,
154 skip_full_grid_cdc_eventt0_if_svd_time_present=skip_full_grid_cdc_eventt0_if_svd_time_present)
156 add_postfilter_tracking_reconstruction(path,
157 components=components,
158 pruneTracks=pruneTracks,
159 reco_tracks=reco_tracks,
160 use_second_cdc_hits=use_second_cdc_hits,
161 prune_temporary_tracks=prune_temporary_tracks,
162 v0_finding=v0_finding,
163 flip_recoTrack=flip_recoTrack,
164 mcTrackFinding=mcTrackFinding)
167def add_prefilter_tracking_reconstruction(path, components=None, skipGeometryAdding=False,
168 mcTrackFinding=False, trackFitHypotheses=None, reco_tracks="RecoTracks",
169 prune_temporary_tracks=True, fit_tracks=True,
170 use_second_cdc_hits=False, skipHitPreparerAdding=False,
171 svd_standalone_mode="VXDTF2",
172 use_svd_to_cdc_ckf=True, use_ecl_to_cdc_ckf=False,
173 add_cdcTrack_QI=True, add_vxdTrack_QI=False, add_recoTrack_QI=False,
174 pxd_filtering_offline=False,
175 create_intercepts_for_pxd_ckf=False,
176 append_full_grid_cdc_eventt0=True,
177 skip_full_grid_cdc_eventt0_if_svd_time_present=True):
179 This function adds the tracking reconstruction modules required to calculate HLT filter decision
182 :param path: The path to add the tracking reconstruction modules to
183 :param components: the list of geometry components in use
or None for all components.
184 :param skipGeometryAdding: Advances flag: The tracking modules need the geometry module
and will add it,
185 if it
is not already present
in the path. In a setup
with multiple (conditional) paths however, it can
not
186 determine,
if the geometry
is already loaded. This flag can be used o just turn off the geometry adding at
187 all (but you will have to add it on your own then).
188 :param skipHitPreparerAdding: Advanced flag: do
not add the hit preparation (esp. VXD cluster creation
189 modules. This
is useful
if they have been added before already.
190 :param mcTrackFinding: Use the MC track finders instead of the realistic ones.
191 :param reco_tracks: Name of the StoreArray where the reco tracks should be stored
192 :param prune_temporary_tracks: If false, store all information of the single CDC
and VXD tracks before merging.
194 :param fit_tracks: If false, the final track find
and the TrackCreator module will no be executed
195 :param use_second_cdc_hits: If true, the second hit information will be used
in the CDC track finding.
196 :param trackFitHypotheses: Which pdg hypothesis to fit. Defaults to [211, 321, 2212].
197 :param svd_standalone_mode: Which SVD standalone tracking
is used.
198 Options are
"VXDTF2",
"SVDHough",
"VXDTF2_and_SVDHough",
and "SVDHough_and_VXDTF2".
200 :param use_svd_to_cdc_ckf:
if true, add SVD to CDC CKF module.
201 :param use_ecl_to_cdc_ckf:
if true, add ECL to CDC CKF module.
202 :param add_cdcTrack_QI: If true, add the MVA track quality estimation
203 to the path that sets the quality indicator property of the found CDC standalone tracks
204 :param add_vxdTrack_QI: If true, add the MVA track quality estimation
205 to the path that sets the quality indicator property of the found VXDTF2 tracks
206 (ATTENTION: Standard triplet QI of VXDTF2
is replaced
in this case
207 -> setting this option to
'True' will have some influence on the final track collection)
208 :param add_recoTrack_QI: If true, add the MVA track quality estimation
209 to the path that sets the quality indicator property of all found reco tracks
210 (Both other QIs needed
as input.)
211 :param pxd_filtering_offline: If
True, PXD data reduction (ROI filtering)
is applied during the track reconstruction.
212 The reconstructed SVD/CDC tracks are used to define the ROIs
and reject all PXD clusters outside of these.
213 :param create_intercepts_for_pxd_ckf: If
True, the PXDROIFinder
is added to the path to create PXDIntercepts to be used
214 for hit filtering when creating the CKF relations. This independent of the offline PXD digit filtering which
is
215 steered by
'pxd_filtering_offline'. This can be applied
for both data
and MC.
216 :param append_full_grid_cdc_eventt0: If
True, the module FullGridChi2TrackTimeExtractor
is added to the path
217 and provides the CDC temporary EventT0.
218 :param skip_full_grid_cdc_eventt0_if_svd_time_present:
if true,
and if also append_full_grid_cdc_eventt0
is true, the
219 FullGridChi2TrackTimeExtractor
is only executed
in the events where no SVD-based EventT0
is found. If false, but
220 append_full_grid_cdc_eventt0
is true, FullGridChi2TrackTimeExtractor will be executed
in each event regardless of
221 SVD EventT0 being present. Has no effect
if append_full_grid_cdc_eventt0
is false. Default: true
224 if not is_svd_used(components)
and not is_cdc_used(components):
227 if (add_cdcTrack_QI
or add_vxdTrack_QI
or add_recoTrack_QI)
and not fit_tracks:
228 b2.B2ERROR(
"MVA track qualiy indicator requires `fit_tracks` to be enabled. Turning all off.")
229 add_cdcTrack_QI =
False
230 add_vxdTrack_QI =
False
231 add_recoTrack_QI =
False
233 if add_recoTrack_QI
and (
not add_cdcTrack_QI
or not add_vxdTrack_QI):
234 b2.B2ERROR(
"RecoTrack qualiy indicator requires CDC and VXD QI as input. Turning it all of.")
235 add_cdcTrack_QI =
False
236 add_vxdTrack_QI =
False
237 add_recoTrack_QI =
False
239 if not skipGeometryAdding:
240 add_geometry_modules(path, components=components)
242 if not skipHitPreparerAdding:
243 add_hit_preparation_modules(path,
244 components=components,
245 pxd_filtering_offline=pxd_filtering_offline,
246 create_intercepts_for_pxd_ckf=create_intercepts_for_pxd_ckf)
249 if 'SetupGenfitExtrapolation' not in path:
250 path.add_module(
'SetupGenfitExtrapolation',
251 energyLossBrems=
False, noiseBrems=
False)
254 add_mc_track_finding(path, components=components, reco_tracks=reco_tracks,
255 use_second_cdc_hits=use_second_cdc_hits)
257 add_track_finding(path, components=components, reco_tracks=reco_tracks,
258 prune_temporary_tracks=prune_temporary_tracks,
259 use_second_cdc_hits=use_second_cdc_hits,
260 svd_standalone_mode=svd_standalone_mode,
261 use_svd_to_cdc_ckf=use_svd_to_cdc_ckf,
262 use_ecl_to_cdc_ckf=use_ecl_to_cdc_ckf,
263 add_cdcTrack_QI=add_cdcTrack_QI,
264 add_vxdTrack_QI=add_vxdTrack_QI,
265 pxd_filtering_offline=pxd_filtering_offline,
266 create_intercepts_for_pxd_ckf=create_intercepts_for_pxd_ckf)
270 add_time_extraction(path, append_full_grid_cdc_eventt0, components=components,
271 skip_full_grid_cdc_eventt0_if_svd_time_present=skip_full_grid_cdc_eventt0_if_svd_time_present)
274 add_prefilter_track_fit_and_track_creator(path,
275 trackFitHypotheses=trackFitHypotheses,
276 reco_tracks=reco_tracks,
277 add_mva_quality_indicator=add_recoTrack_QI)
280def add_postfilter_tracking_reconstruction(path, components=None, pruneTracks=False, reco_tracks="RecoTracks",
281 use_second_cdc_hits=False, prune_temporary_tracks=True, v0_finding=True,
282 flip_recoTrack=True, mcTrackFinding=False, kink_finding=True):
284 This function adds the tracking reconstruction modules not required to calculate HLT filter
287 :param path: The path to add the tracking reconstruction modules to
288 :param components: the list of geometry components
in use
or None for all components.
289 :param pruneTracks: Delete all hits
except the first
and the last
in the found tracks.
290 :param reco_tracks: Name of the StoreArray where the reco tracks should be stored
291 :param use_second_cdc_hits: If true, the second hit information will be used
in the CDC track finding.
292 :param prune_temporary_tracks: If false, store all information of the single CDC
and VXD tracks before merging.
294 :param v0_finding: If false, the V0 module will
not be executed
295 :param flip_recoTrack:
if true, add the recoTracks flipping function
in the postfilter (only
if PXD
is present)
296 :param mcTrackFinding: Use the MC track finders instead of the realistic ones.
300 if components
and not (
'SVD' in components
or 'CDC' in components):
304 if flip_recoTrack
and not mcTrackFinding
and is_pxd_used(components):
305 add_flipping_of_recoTracks(path, reco_tracks=
"RecoTracks")
309 path.add_module(
'V0Finder', RecoTracks=reco_tracks, v0FitterMode=1)
313 path.add_module(
'KinkFinder', RecoTracks=reco_tracks)
316 path.add_module(
'TrackTimeEstimator')
318 add_mc_matcher(path, components=components, reco_tracks=reco_tracks,
319 use_second_cdc_hits=use_second_cdc_hits)
323 add_prune_tracks(path, components=components, reco_tracks=reco_tracks)
325 if prune_temporary_tracks
or pruneTracks:
326 path.add_module(
"PruneRecoHits")
329def add_time_extraction(path, append_full_grid_cdc_eventt0=False, components=None,
330 skip_full_grid_cdc_eventt0_if_svd_time_present=True):
332 Add time extraction components via tracking
334 :param path: The path to add the tracking reconstruction modules to
335 :param append_full_grid_cdc_eventt0: If True, the module FullGridChi2TrackTimeExtractor
is added to the path
336 and provides the CDC temporary EventT0.
337 :param components: the list of geometry components
in use
or None for all components.
338 :param skip_full_grid_cdc_eventt0_if_svd_time_present:
if true,
and if also append_full_grid_cdc_eventt0
is true, the
339 FullGridChi2TrackTimeExtractor
is only executed
in the events where no SVD-based EventT0
is found. If false, but
340 append_full_grid_cdc_eventt0
is true, FullGridChi2TrackTimeExtractor will be executed
in each event regardless of
341 SVD EventT0 being present. Has no effect
if append_full_grid_cdc_eventt0
is false. Default: true
345 if is_svd_used(components):
346 path.add_module(
"SVDEventT0Estimator")
348 if is_cdc_used(components)
and append_full_grid_cdc_eventt0:
349 path.add_module(
"FullGridChi2TrackTimeExtractor",
350 skipIfSVDEventT0Present=skip_full_grid_cdc_eventt0_if_svd_time_present)
353def add_cr_tracking_reconstruction(path, components=None, prune_tracks=False,
354 skip_geometry_adding=False, event_time_extraction=True,
355 merge_tracks=True, use_second_cdc_hits=False):
357 This function adds the reconstruction modules for cr tracking to a path.
359 :param path: The path to which to add the tracking reconstruction modules
361 :param components: the list of geometry components
in use
or None for all components.
362 :param prune_tracks: Delete all hits
except the first
and the last
in the found tracks.
364 :param skip_geometry_adding: Advanced flag: The tracking modules need the geometry module
and will add it,
365 if it
is not already present
in the path. In a setup
with multiple (conditional) paths however, it cannot
366 determine
if the geometry
is already loaded. This flag can be used to just turn off the geometry adding
367 (but you will have to add it on your own).
368 :param event_time_extraction: extract the event time
369 :param merge_tracks: The upper
and lower half of the tracks should be merged together
in one track
370 :param use_second_cdc_hits: If true, the second hit information will be used
in the CDC track finding.
378 if not is_cdc_used(components):
381 if not skip_geometry_adding:
382 add_geometry_modules(path, components)
384 add_hit_preparation_modules(path, components=components, create_intercepts_for_pxd_ckf=
False)
387 if 'SetupGenfitExtrapolation' not in path:
388 path.add_module(
'SetupGenfitExtrapolation',
389 energyLossBrems=
False, noiseBrems=
False)
392 add_cr_track_finding(path, reco_tracks=
"RecoTracks", components=components,
393 merge_tracks=merge_tracks, use_second_cdc_hits=use_second_cdc_hits)
397 add_cr_track_fit_and_track_creator(path, components=components, prune_tracks=prune_tracks,
398 event_timing_extraction=event_time_extraction)
402 add_cr_track_fit_and_track_creator(path, components=components, prune_tracks=prune_tracks,
403 event_timing_extraction=
False,
404 reco_tracks=
"NonMergedRecoTracks", tracks=
"NonMergedTracks")
407def add_mc_tracking_reconstruction(path, components=None, pruneTracks=False, use_second_cdc_hits=False):
409 This function adds the standard reconstruction modules for MC tracking
412 :param path: The path to add the tracking reconstruction modules to
413 :param components: the list of geometry components
in use
or None for all components.
414 :param pruneTracks: Delete all hits expect the first
and the last
from the found tracks.
415 :param use_second_cdc_hits: If true, the second hit information will be used
in the CDC track finding.
417 add_tracking_reconstruction(path,
418 components=components,
419 pruneTracks=pruneTracks,
421 use_second_cdc_hits=use_second_cdc_hits)
424def add_track_finding(path, components=None, reco_tracks="RecoTracks",
425 prune_temporary_tracks=True, use_second_cdc_hits=False,
426 use_mc_truth=False, svd_ckf_mode="SVD_after", add_both_directions=True,
427 svd_standalone_mode="VXDTF2",
428 use_svd_to_cdc_ckf=True, use_ecl_to_cdc_ckf=False,
429 add_cdcTrack_QI=True, add_vxdTrack_QI=False,
430 pxd_filtering_offline=False, use_HLT_ROIs=False,
431 create_intercepts_for_pxd_ckf=False):
433 Add the CKF to the path with all the track finding related to
and needed
for it.
434 :param path: The path to add the tracking reconstruction modules to
435 :param reco_tracks: The store array name where to output all tracks
436 :param use_mc_truth: Use the truth information
in the CKF modules
437 :param svd_ckf_mode: how to apply the CKF (
with or without SVD standalone tracking). Defaults to
"SVD_after".
438 :param add_both_directions: Curlers may be found
in the wrong orientation by the CDC track finder, so
try to
439 extrapolate also
in the other direction.
440 :param svd_standalone_mode: Which SVD standalone tracking
is used.
441 Options are
"VXDTF2",
"SVDHough",
"VXDTF2_and_SVDHough",
and "SVDHough_and_VXDTF2".
443 :param use_second_cdc_hits: whether to use the secondary CDC hit during CDC track finding
or not
444 :param components: the list of geometry components
in use
or None for all components.
445 :param prune_temporary_tracks: If false, store all information of the single CDC
and VXD tracks before merging.
447 :param use_svd_to_cdc_ckf:
if true, add SVD to CDC CKF module.
448 :param use_ecl_to_cdc_ckf:
if true, add ECL to CDC CKF module.
449 :param add_cdcTrack_QI: If true, add the MVA track quality estimation
450 to the path that sets the quality indicator property of the found CDC standalone tracks
451 :param add_vxdTrack_QI: If true, add the MVA track quality estimation
452 to the path that sets the quality indicator property of the found VXDTF2 tracks
453 (ATTENTION: Standard triplet QI of VXDTF2
is replaced
in this case
454 -> setting this option to
'True' will have some influence on the final track collection)
455 :param pxd_filtering_offline: If
True, PXD data reduction (ROI filtering)
is applied during the track reconstruction.
456 The reconstructed SVD/CDC tracks are used to define the ROIs
and reject all PXD clusters outside of these.
457 :param use_HLT_ROIs: Don
't calculate the ROIs here but use the ones from the HLT (does obviously not work for simulation) :param create_intercepts_for_pxd_ckf: If True, the PXDROIFinder
is added to the path to create PXDIntercepts to be used
458 for hit filtering when creating the CKF relations. This independent of the offline PXD digit filtering which
is
459 steered by
'pxd_filtering_offline'. This can be applied
for both data
and MC.
461 if not is_svd_used(components)
and not is_cdc_used(components):
464 if use_ecl_to_cdc_ckf
and not is_cdc_used(components):
465 b2.B2WARNING(
"ECL CKF cannot be used without CDC. Turning it off.")
466 use_ecl_to_cdc_ckf =
False
468 if use_ecl_to_cdc_ckf
and not is_ecl_used(components):
469 b2.B2ERROR(
"ECL CKF cannot be used without ECL. Turning it off.")
470 use_ecl_to_cdc_ckf =
False
473 if 'RegisterEventLevelTrackingInfo' not in path:
474 path.add_module(
'RegisterEventLevelTrackingInfo')
477 cdc_reco_tracks =
"CDCRecoTracks"
478 svd_cdc_reco_tracks =
"SVDCDCRecoTracks"
479 ecl_reco_tracks =
"ECLRecoTracks"
480 combined_ecl_reco_tracks =
"combinedECLRecoTracks"
483 svd_reco_tracks =
"SVDRecoTracks"
484 pxd_reco_tracks =
"PXDRecoTracks"
487 temporary_reco_track_list = []
490 latest_reco_tracks =
None
492 if not is_pxd_used(components):
493 if use_ecl_to_cdc_ckf
and is_cdc_used(components):
494 combined_ecl_reco_tracks = reco_tracks
495 elif (
not use_ecl_to_cdc_ckf)
and is_svd_used(components):
496 svd_cdc_reco_tracks = reco_tracks
497 elif (
not use_ecl_to_cdc_ckf)
and (
not is_svd_used(components))
and is_cdc_used(components):
498 cdc_reco_tracks = reco_tracks
500 if is_cdc_used(components):
501 add_cdc_track_finding(path, use_second_hits=use_second_cdc_hits, output_reco_tracks=cdc_reco_tracks,
502 add_mva_quality_indicator=add_cdcTrack_QI)
503 temporary_reco_track_list.append(cdc_reco_tracks)
504 latest_reco_tracks = cdc_reco_tracks
506 if is_svd_used(components):
507 add_svd_track_finding(path, components=components, input_reco_tracks=latest_reco_tracks,
508 output_reco_tracks=svd_cdc_reco_tracks, use_mc_truth=use_mc_truth,
509 temporary_reco_tracks=svd_reco_tracks,
510 svd_ckf_mode=svd_ckf_mode, add_both_directions=add_both_directions,
511 use_svd_to_cdc_ckf=use_svd_to_cdc_ckf, prune_temporary_tracks=prune_temporary_tracks,
512 add_mva_quality_indicator=add_vxdTrack_QI,
513 svd_standalone_mode=svd_standalone_mode)
514 temporary_reco_track_list.append(svd_reco_tracks)
515 temporary_reco_track_list.append(svd_cdc_reco_tracks)
516 latest_reco_tracks = svd_cdc_reco_tracks
518 if use_ecl_to_cdc_ckf
and is_cdc_used(components):
519 add_eclcdc_track_finding(path, components=components, output_reco_tracks=ecl_reco_tracks,
520 prune_temporary_tracks=prune_temporary_tracks)
524 path.add_module(
"RecoTrackStoreArrayCombiner",
525 Temp1RecoTracksStoreArrayName=latest_reco_tracks,
526 Temp2RecoTracksStoreArrayName=ecl_reco_tracks,
527 recoTracksStoreArrayName=combined_ecl_reco_tracks)
528 temporary_reco_track_list.append(ecl_reco_tracks)
529 temporary_reco_track_list.append(combined_ecl_reco_tracks)
530 latest_reco_tracks = combined_ecl_reco_tracks
532 if is_pxd_used(components):
534 In case we want to use offline PXD hit filtering ('pxd_filtering_offline == True'), we can decide to either use ROIs
535 created by HLT (
'use_HLT_ROIs == True')
or to create ROIs on the fly (
'use_HLT_ROIs == False').
536 In addition, we can also just run the ROI finder to create PXDIntercepts (
'create_intercepts_for_pxd_ckf == True')
537 that can be used
in the ToPXDCKF relation filters to reduce the number of relations that are created. In this case
538 there
is no need to apply the ROI filtering.
540 if pxd_filtering_offline
or create_intercepts_for_pxd_ckf:
542 intercepts_name =
"PXDIntercepts"
543 if not use_HLT_ROIs
or create_intercepts_for_pxd_ckf:
544 path.add_module(
"DAFRecoFitter", recoTracksStoreArrayName=latest_reco_tracks)
546 roiName =
"ROIs_offline"
547 intercepts_name =
"CKF" + intercepts_name
549 reco_tracks=latest_reco_tracks,
550 intercepts_name=intercepts_name,
553 if pxd_filtering_offline:
554 pxd_digifilter = b2.register_module(
'PXDdigiFilter')
555 pxd_digifilter.param(
'ROIidsName', roiName)
556 pxd_digifilter.param(
'PXDDigitsName',
'PXDDigits')
557 pxd_digifilter.param(
'PXDDigitsInsideROIName',
'PXDDigits')
558 pxd_digifilter.param(
'overrideDB',
True)
559 pxd_digifilter.param(
'usePXDDataReduction',
True)
560 path.add_module(pxd_digifilter)
562 if pxd_filtering_offline
or create_intercepts_for_pxd_ckf:
563 add_pxd_reconstruction(path)
565 add_pxd_track_finding(path, components=components, input_reco_tracks=latest_reco_tracks,
566 use_mc_truth=use_mc_truth, output_reco_tracks=reco_tracks,
567 temporary_reco_tracks=pxd_reco_tracks,
568 add_both_directions=add_both_directions)
569 temporary_reco_track_list.append(pxd_reco_tracks)
571 if prune_temporary_tracks:
572 for temporary_reco_track_name
in temporary_reco_track_list:
573 if temporary_reco_track_name != reco_tracks:
574 path.add_module(
'PruneRecoTracks', storeArrayName=temporary_reco_track_name)
577def add_cr_track_finding(path, reco_tracks="RecoTracks", components=None,
578 merge_tracks=True, use_second_cdc_hits=False):
581 if 'RegisterEventLevelTrackingInfo' not in path:
582 path.add_module(
'RegisterEventLevelTrackingInfo')
584 if not is_cdc_used(components):
585 b2.B2FATAL(
"CDC must be in components")
587 reco_tracks_from_track_finding = reco_tracks
589 reco_tracks_from_track_finding =
"NonMergedRecoTracks"
591 cdc_reco_tracks =
"CDCRecoTracks"
592 if not is_pxd_used(components)
and not is_svd_used(components):
593 cdc_reco_tracks = reco_tracks_from_track_finding
595 svd_cdc_reco_tracks =
"SVDCDCRecoTracks"
596 if not is_pxd_used(components):
597 svd_cdc_reco_tracks = reco_tracks_from_track_finding
599 full_reco_tracks = reco_tracks_from_track_finding
602 add_cdc_cr_track_finding(path, merge_tracks=
False, use_second_cdc_hits=use_second_cdc_hits,
603 output_reco_tracks=cdc_reco_tracks)
605 latest_reco_tracks = cdc_reco_tracks
607 if is_svd_used(components):
608 add_svd_track_finding(path, components=components, input_reco_tracks=latest_reco_tracks,
609 output_reco_tracks=svd_cdc_reco_tracks,
610 svd_ckf_mode=
"cosmics", add_both_directions=
True, svd_standalone_mode=
"VXDTF2")
611 latest_reco_tracks = svd_cdc_reco_tracks
613 if is_pxd_used(components):
614 add_pxd_cr_track_finding(path, components=components, input_reco_tracks=latest_reco_tracks,
615 output_reco_tracks=full_reco_tracks, add_both_directions=
True)
619 path.add_module(
"CosmicsTrackMerger", inputRecoTracks=reco_tracks_from_track_finding,
620 outputRecoTracks=reco_tracks)
623def add_mc_track_finding(path, components=None, reco_tracks="RecoTracks", use_second_cdc_hits=False):
625 Add the MC based TrackFinder to the path.
627 :param path: The path to add the tracking reconstruction modules to
628 :param components: the list of geometry components in use
or None for all components.
629 :param reco_tracks: Name of the StoreArray where the reco tracks should be stored
630 :param use_second_cdc_hits: If true, the second hit information will be used
in the CDC track finding.
632 if is_cdc_used(components)
or is_pxd_used(components)
or is_svd_used(components):
634 path.add_module(
'TrackFinderMCTruthRecoTracks',
635 RecoTracksStoreArrayName=reco_tracks,
636 UseSecondCDCHits=use_second_cdc_hits,
637 UsePXDHits=is_pxd_used(components),
638 UseSVDHits=is_svd_used(components),
639 UseCDCHits=is_cdc_used(components))
642def add_tracking_for_PXDDataReduction_simulation(path, components, svd_cluster='__ROIsvdClusters'):
644 This function adds the standard reconstruction modules for tracking to be used
for the simulation of PXD data
647 :param path: The path to add the tracking reconstruction modules to
648 :param components: the list of geometry components
in use
or None for all components, always exclude the PXD.
651 if not is_svd_used(components):
655 if 'SetupGenfitExtrapolation' not in path:
656 material_effects = b2.register_module(
'SetupGenfitExtrapolation')
657 material_effects.set_name(
658 'SetupGenfitExtrapolationForPXDDataReduction')
659 path.add_module(material_effects)
662 svd_reco_tracks =
"__ROIsvdRecoTracks"
665 add_svd_standalone_tracking(path, components=[
'SVD'], reco_tracks=svd_reco_tracks, suffix=
"__ROI",
666 svd_clusters=svd_cluster)
669 dafRecoFitter = b2.register_module(
"DAFRecoFitter")
670 dafRecoFitter.set_name(
"SVD-only DAFRecoFitter")
671 dafRecoFitter.param(
'recoTracksStoreArrayName', svd_reco_tracks)
672 dafRecoFitter.param(
'svdHitsStoreArrayName', svd_cluster)
673 path.add_module(dafRecoFitter)
676def add_roiFinder(path, reco_tracks="RecoTracks", intercepts_name="PXDIntercepts", roiName="ROIs"):
678 Add the ROI finding to the path creating ROIs out of reco tracks by extrapolating them to the PXD volume.
679 :param path: Where to add the module to.
680 :param reco_tracks: Which tracks to use in the extrapolation step.
681 :param roiName: Name of the produced/stored ROIs.
684 pxdDataRed = b2.register_module('PXDROIFinder')
686 'recoTrackListName': reco_tracks,
687 'PXDInterceptListName': intercepts_name,
688 'ROIListName': roiName,
690 pxdDataRed.param(param_pxdDataRed)
691 path.add_module(pxdDataRed)
694def add_vxd_standalone_cosmics_finder(
696 reco_tracks="RecoTracks",
697 pxd_spacepoints_name="PXDSpacePoints",
698 svd_spacepoints_name="SVDSpacePoints",
703 Convenience function for adding VXD standalone cosmics track finding
for B = 0 Tesla
706 The result
is a StoreArray
with name
@param reco_tracks containing one
or zero reco tracks per event.
707 This track candidates have an arbitrary but large momentum seed
in the direction of the fitted line.
708 The position
and momentum seed
is obtained using a principal component analysis method.
710 :param path: basf2 path
711 :param reco_tracks: Name of the output RecoTracks; defaults to RecoTracks.
712 :param spacepoints_name: name of store array containing the spacepoints; defaults to SpacePoints
713 :param quality_cut: Cut on the chi squared value of the line fit; candidates
with values above the cut will be
714 rejected; defaults to 0.0001
715 :param min_sps: Minimal number of SpacePoints required to build a track candidate; defaults to 3;
716 :param max_rejected_sps: Maximal number of retries to refit a track after the worst spacepoint was removed;
721 if 'RegisterEventLevelTrackingInfo' not in path:
722 path.add_module(
'RegisterEventLevelTrackingInfo')
724 if "PXDSpacePointCreator" not in [m.name()
for m
in path.modules()]:
725 path.add_module(
'PXDSpacePointCreator', SpacePoints=pxd_spacepoints_name)
729 track_finder = b2.register_module(
'TrackFinderVXDCosmicsStandalone')
730 track_finder.param(
'SpacePointTrackCandArrayName',
"")
731 track_finder.param(
'SpacePoints', [pxd_spacepoints_name, svd_spacepoints_name])
732 track_finder.param(
'QualityCut', quality_cut)
733 track_finder.param(
'MinSPs', min_sps)
734 track_finder.param(
'MaxRejectedSPs', max_rejected_sps)
735 path.add_module(track_finder)
737 converter = b2.register_module(
'SPTC2RTConverter')
738 converter.param(
'recoTracksStoreArrayName', reco_tracks)
739 path.add_module(converter)