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