20 from setup_modules
import setup_RTCtoSPTCConverters
26 parser = argparse.ArgumentParser(description=
'Training sample preparation:\
27 Prepare a data sample to train the sector map.\n\
28 Usage: basf2 traininPreparation.py -i dataSample.root -- --enable_selection boolean')
35 help=
'enable the selection of training sample based on track parameters')
43 help=
"By default only RecoTracks with valid fit are taken for training. Using this option will disable that. ")
45 arguments = parser.parse_args()
46 use_noKick = arguments.use_NoKick
54 set_log_level(LogLevel.ERROR)
55 log_to_file(
'logVXDTF2Preparation.log', append=
False)
66 rootInputM = register_module(
'RootInput')
67 path.add_module(rootInputM)
70 eventinfoprinter = register_module(
'EventInfoPrinter')
71 path.add_module(eventinfoprinter)
73 path.add_module(
"PrintCollections", printForEvent=1)
77 gearbox = register_module(
'Gearbox')
78 path.add_module(gearbox)
81 geometry = register_module(
'Geometry')
82 path.add_module(geometry)
91 spCreatorPXD = register_module(
'PXDSpacePointCreator')
92 spCreatorPXD.param(
'NameOfInstance',
'PXDSpacePointCreator')
93 spCreatorPXD.param(
'SpacePoints',
'PXDSpacePoints')
94 path.add_module(spCreatorPXD)
96 spCreatorSVD = register_module(
'SVDSpacePointCreator')
97 spCreatorSVD.param(
'OnlySingleClusterSpacePoints',
False)
98 spCreatorSVD.param(
'NameOfInstance',
'SVDSpacePointCreator')
99 spCreatorSVD.param(
'SpacePoints',
'SVDSpacePoints')
100 path.add_module(spCreatorSVD)
105 setup_RTCtoSPTCConverters(path=path,
106 SVDSPscollection=
'SVDSpacePoints',
107 PXDSPscollection=
'PXDSpacePoints',
108 RTCinput=
'MCRecoTracks',
109 sptcOutput=
'checkedSPTCs',
111 logLevel=LogLevel.ERROR,
112 useNoKick=use_noKick,
113 useOnlyFittedTracks=
True)
121 secMapBootStrap = register_module(
'SectorMapBootstrap')
122 secMapBootStrap.param(
'ReadSectorMap',
False)
123 secMapBootStrap.param(
'WriteSectorMap',
False)
124 path.add_module(secMapBootStrap)
129 if os.environ.get(
'USE_BEAST2_GEOMETRY'):
135 nameTag +=
'_SVDOnly'
138 SecMapTrainerBase = register_module(
'VXDTFTrainingDataCollector')
139 SecMapTrainerBase.param(
'NameTag', nameTag)
140 SecMapTrainerBase.param(
'SpacePointTrackCandsName',
'checkedSPTCs')
142 path.add_module(SecMapTrainerBase)
145 path.add_module(
'Progress')
147 path.add_module(
"PrintCollections", printForEvent=1)