Belle II Software
release-05-01-25
cosmicsHoughTrackingValidation.py
1
#!/usr/bin/env python3
2
# -*- coding: utf-8 -*-
3
4
"""
5
<header>
6
<contact>software-tracking@belle2.org</contact>
7
<input>CosmicsSimNoBkg.root</input>
8
<description>Validates the hough finder working on segments in cosmics events.</description>
9
</header>
10
"""
11
12
VALIDATION_OUTPUT_FILE =
'CosmicsHoughTrackingValidation.root'
13
N_EVENTS = 1000
14
ACTIVE =
False
15
16
import
basf2
17
basf2.set_random_seed(1337)
18
19
import
os
20
import
trackfindingcdc.cdcdisplay
as
cdcdisplay
21
22
import
logging
23
24
from
tracking.validation.run
import
TrackingValidationRun
25
26
27
class
CosmicsHough
(
TrackingValidationRun
):
28
"""Validate the CDC Hough track-segment finder with cosmic rays"""
29
30
n_events = N_EVENTS
31
32
generator_module =
'Cosmics'
33
34
root_input_file =
'../CosmicsSimNoBkg.root'
35
36
components =
None
37
38
def
finder_module
(self, path):
39
"""Add the CDC Hough track finder and related modules to the basf2 path"""
40
path.add_module(
'TFCDC_WireHitPreparer'
)
41
path.add_module(
'TFCDC_ClusterPreparer'
)
42
path.add_module(
'TFCDC_SegmentFinderFacetAutomaton'
,
43
SegmentOrientation=
"downwards"
)
44
path.add_module(
'TFCDC_AxialTrackCreatorSegmentHough'
,
45
tracks=
"CDCAxialTrackVector"
)
46
path.add_module(
'TFCDC_StereoHitFinder'
,
47
inputTracks=
"CDCAxialTrackVector"
)
48
path.add_module(
'TFCDC_TrackExporter'
,
49
inputTracks=
"CDCAxialTrackVector"
)
50
51
interactive_display =
False
52
if
interactive_display:
53
cdc_display_module = cdcdisplay.CDCSVGDisplayModule(os.getcwd(), interactive=
True
)
54
cdc_display_module.draw_recotracks =
True
55
cdc_display_module.draw_recotrack_seed_trajectories =
True
56
path.add_module(cdc_display_module)
57
58
59
tracking_coverage = {
60
'WhichParticles'
: [
'CDC'
],
# Include all particles seen in CDC, also secondaries
61
'UsePXDHits'
:
False
,
62
'UseSVDHits'
:
False
,
63
'UseCDCHits'
:
True
,
64
'UseOnlyAxialCDCHits'
:
False
,
65
"UseReassignedHits"
:
True
,
66
}
67
68
pulls =
True
69
70
output_file_name = VALIDATION_OUTPUT_FILE
71
72
73
def
main
():
74
validation_run =
CosmicsHough
()
75
validation_run.configure_and_execute_from_commandline()
76
77
if
__name__ ==
'__main__'
:
78
logging.basicConfig(level=logging.INFO)
79
if
ACTIVE:
80
main
()
cosmicsHoughTrackingValidation.CosmicsHough
Definition:
cosmicsHoughTrackingValidation.py:27
tracking.validation.run
Definition:
run.py:1
trackfindingcdc.cdcdisplay
Definition:
__init__.py:1
main
int main(int argc, char **argv)
Run all tests.
Definition:
test_main.cc:77
tracking.run.tracked_event_generation.ReadOrGenerateTrackedEventsRun.finder_module
finder_module
Name of the finder module to be used - can be everything that is accepted by tracking....
Definition:
tracked_event_generation.py:25
tracking.validation.run.TrackingValidationRun
Definition:
run.py:18
tracking
validation
cosmicsHoughTrackingValidation.py
Generated on Fri Nov 5 2021 04:04:59 for Belle II Software by
1.8.17