Belle II Software  release-05-01-25
evaluateVXDTF1.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3 
4 
14 
15 
16 from basf2 import *
17 import tracking
18 from tracking.harvesting_validation.combined_module import CombinedTrackingValidationModule
19 
20 
21 # ---------------------------------------------------------------------------------------
22 # Settings
23 usePXD = False
24 
25 performFit = False
26 
27 # Logging and Debug Levels
28 set_log_level(LogLevel.ERROR)
29 log_to_file('logVXDTF1Evaluation.log', append=False)
30 
31 
32 # ---------------------------------------------------------------------------------------
33 path = create_path()
34 
35 # Input
36 rootInput = register_module('RootInput')
37 path.add_module(rootInput)
38 
39 # Event Info Module
40 eventinfoprinter = register_module('EventInfoPrinter')
41 path.add_module(eventinfoprinter)
42 
43 # Gearbox
44 gearbox = register_module('Gearbox')
45 path.add_module(gearbox)
46 
47 # Geometry
48 geometry = register_module('Geometry')
49 geometry.param('components', ['BeamPipe',
50  'MagneticFieldConstant4LimitedRSVD',
51  'PXD',
52  'SVD',
53  'CDC'])
54 path.add_module(geometry)
55 
56 # Event counter
57 eventCounter = register_module('EventCounter')
58 path.add_module(eventCounter)
59 
60 genFitExtrapolation = register_module('SetupGenfitExtrapolation')
61 path.add_module(genFitExtrapolation)
62 
63 tracking.add_vxd_track_finding(path, "RecoTracks", components=["SVD"])
64 
65 if performFit:
66  fitter = register_module('DAFRecoFitter')
67  path.add_module(fitter)
68 
69 # Matching
70 mcTrackMatcherModule = register_module('MCRecoTracksMatcher')
71 mcTrackMatcherModule.param({
72  'UseCDCHits': False,
73  'UseSVDHits': True,
74  'UsePXDHits': False,
75  'mcRecoTracksStoreArrayName': 'MCRecoTracks',
76  'MinimalPurity': .66,
77 })
78 path.add_module(mcTrackMatcherModule)
79 
80 # Evaluation of matching
81 trackingValidationModule = CombinedTrackingValidationModule(
82  "",
83  contact="",
84  output_file_name="VXDTF1Validation.root",
85  expert_level=2)
86 path.add_module(trackingValidationModule)
87 
88 process(path)
89 print(statistics)
tracking.harvesting_validation.combined_module
Definition: combined_module.py:1
tracking.validation.run.TrackingValidationRun.create_path
def create_path(self)
Definition: run.py:114