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 
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