Belle II Software  release-05-01-25
VXDQE_TrainingDataCollector.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3 
4 
12 
13 from basf2 import *
14 from tracking import add_vxd_track_finding_vxdtf2, add_mc_matcher, add_hit_preparation_modules
15 
16 estimationMethod = 'tripletFit'
17 clusterInfo = 'Average'
18 
19 eval_file_train = "Upsilon4S_ForMVA_10000Events_w16th_overlay_Bkg.root" # Change to correct training file
20 sector_map = None # Default SectorMap
21 
22 # ---------------------------------------------------------------------------------------
23 name = 'VXDEQE_CollectedTrainingData_Default'
24 
25 # Logging and Debug Level
26 set_log_level(LogLevel.INFO)
27 
28 path = create_path()
29 
30 rootInput = register_module('RootInput')
31 rootInput.param('inputFileName', eval_file_train)
32 path.add_module(rootInput)
33 
34 # Event Info Module
35 eventinfoprinter = register_module('EventInfoPrinter')
36 path.add_module(eventinfoprinter)
37 
38 path.add_module("Gearbox")
39 path.add_module("Geometry")
40 
41 add_hit_preparation_modules(path, components=['SVD'])
42 
43 add_vxd_track_finding_vxdtf2(path, reco_tracks="RecoTracks", components=['SVD'], suffix="", sectormap_file=sector_map)
44 
45 add_mc_matcher(path, components=['SVD'])
46 
47 data = register_module('VXDQETrainingDataCollector')
48 data.param('EstimationMethod', estimationMethod)
49 data.param('ClusterInformation', clusterInfo)
50 data.param('TrainingDataOutputName', name + '.root')
51 data.param('SpacePointTrackCandsStoreArrayName', 'SPTrackCands')
52 data.param('MCRecoTracksStoreArrayName', 'MCRecoTracks')
53 path.add_module(data)
54 
55 print_path(path)
56 process(path)
57 print(statistics)
tracking.validation.run.TrackingValidationRun.create_path
def create_path(self)
Definition: run.py:114