2 from ROOT
import Belle2
19 basf2.set_log_level(basf2.LogLevel.DEBUG)
20 basf2.set_debug_level(20)
26 main = basf2.create_path()
29 dstfiles = [sys.argv[1]]
33 print(
'Files to be processed:')
38 os.makedirs(sys.argv[0].split(
'.py')[0] +
'/data', exist_ok=
True)
39 os.makedirs(sys.argv[0].split(
'.py')[0] +
'/hist', exist_ok=
True)
42 outputfile = sys.argv[0].split(
'.py')[0] +
'/hist/histo.' + sys.argv[1].split(
'/')[-1].split(ending)[0] +
'.root'
43 dstputfile = sys.argv[0].split(
'.py')[0] +
'/data/dst.' + sys.argv[1].split(
'/')[-1].split(ending)[0] +
'.root'
46 main.add_module(
"RootInput", inputFileNames=dstfiles)
49 main.add_module(
'Gearbox')
50 main.add_module(
'Geometry')
53 main.add_module(
'Progress', maxN=3)
69 main.add_module(
'CDCTriggerRecoMatcher', TrgTrackCollectionName=neurotrigger.hwneurotracks,
70 hitCollectionName=neurotrigger.hwneuroinputsegmenthits, axialOnly=
True)
71 main.add_module(
'CDCTriggerRecoMatcher', TrgTrackCollectionName=neurotrigger.hwsimneurotracks,
72 hitCollectionName=neurotrigger.hwneuroinputsegmenthits, axialOnly=
True)
73 main.add_module(
'CDCTriggerRecoMatcher', TrgTrackCollectionName=neurotrigger.hwneuroinput2dfindertracks,
74 hitCollectionName=neurotrigger.hwneuroinputsegmenthits, axialOnly=
True)
78 main.add_module(
'SetupGenfitExtrapolation')
81 main.add_module(
'HistoManager',
82 histoFileName=outputfile)
83 main.add_module(
'CDCTriggerNeuroDQM',
84 simNeuroTracksName=neurotrigger.hwsimneurotracks,
85 unpackedNeuroInput2dTracksName=neurotrigger.hwneuroinput2dfindertracks,
97 main.add_module(
'RootOutput', outputFileName=dstputfile)
101 print(basf2.statistics)