14 from tracking
import add_vxd_track_finding_vxdtf2, add_mc_matcher, add_hit_preparation_modules
16 estimationMethod =
'tripletFit'
17 clusterInfo =
'Average'
19 eval_file_train =
"Upsilon4S_ForMVA_10000Events_w16th_overlay_Bkg.root"
23 name =
'VXDEQE_CollectedTrainingData_Default'
26 set_log_level(LogLevel.INFO)
30 rootInput = register_module(
'RootInput')
31 rootInput.param(
'inputFileName', eval_file_train)
32 path.add_module(rootInput)
35 eventinfoprinter = register_module(
'EventInfoPrinter')
36 path.add_module(eventinfoprinter)
38 path.add_module(
"Gearbox")
39 path.add_module(
"Geometry")
41 add_hit_preparation_modules(path, components=[
'SVD'])
43 add_vxd_track_finding_vxdtf2(path, reco_tracks=
"RecoTracks", components=[
'SVD'], suffix=
"", sectormap_file=sector_map)
45 add_mc_matcher(path, components=[
'SVD'])
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')