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