Belle II Software
release-05-01-25
cosmicsAxialHoughTrackingValidation.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 axial hough finder working on segments in cosmics events.</description>
9
</header>
10
"""
11
12
VALIDATION_OUTPUT_FILE =
'CosmicsAxialHoughTrackingValidation.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
CosmicsAxialHough
(
TrackingValidationRun
):
28
"""Validate the CDC axial Hough track 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 track-finding module 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
path.add_module(
'TFCDC_TrackExporter'
)
46
47
interactive_display =
False
48
if
interactive_display:
49
cdc_display_module = cdcdisplay.CDCSVGDisplayModule(os.getcwd(), interactive=
True
)
50
cdc_display_module.draw_recotracks =
True
51
cdc_display_module.draw_recotrack_seed_trajectories =
True
52
path.add_module(cdc_display_module)
53
54
55
tracking_coverage = {
56
'WhichParticles'
: [
'CDC'
],
# Include all particles seen in CDC, also secondaries
57
'UsePXDHits'
:
False
,
58
'UseSVDHits'
:
False
,
59
'UseCDCHits'
:
True
,
60
'UseOnlyAxialCDCHits'
:
True
,
61
"UseReassignedHits"
:
True
,
62
}
63
64
65
pulls =
True
66
67
output_file_name = VALIDATION_OUTPUT_FILE
68
69
70
def
main
():
71
validation_run =
CosmicsAxialHough
()
72
validation_run.configure_and_execute_from_commandline()
73
74
if
__name__ ==
'__main__'
:
75
logging.basicConfig(level=logging.INFO)
76
if
ACTIVE:
77
main
()
tracking.validation.run
Definition:
run.py:1
trackfindingcdc.cdcdisplay
Definition:
__init__.py:1
cosmicsAxialHoughTrackingValidation.CosmicsAxialHough
Definition:
cosmicsAxialHoughTrackingValidation.py:27
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
cosmicsAxialHoughTrackingValidation.py
Generated on Fri Nov 5 2021 04:04:58 for Belle II Software by
1.8.17