24 from array
import array
27 from ROOT.Belle2
import SVDCoGCalibrationFunction
28 from ROOT.Belle2
import SVDCoGTimeCalibrations
32 import matplotlib.pyplot
as plt
36 filename = sys.argv[2]
37 branches = [
'SVDShaperDigits',
'SVDRecoDigits',
'SVDClusters',
'EventT0',
'Tracks',
'RecoTracks']
39 trk_outputFile =
"TrackFilterControlNtuples_" + localdb +
".root"
45 if filename.rfind(
".root") != -1:
46 inputFileList.append(filename)
48 with open(filename,
'r')
as f:
49 inputFileList = [line.strip()
for line
in f]
53 use_central_database(
'data_reprocessing_proc7')
54 if '_CHECK' not in localdb:
55 use_central_database(
'svd_NOCoGCorrections')
56 if os.path.isdir(localdb):
57 print(
'WARNING, you are using CoG corrections in ' + localdb +
' exiting')
60 use_local_database(localdb +
"/database.txt", localdb, invertLogging=
True)
62 localdb = localdb.split(
'_')
63 if not os.path.isdir(localdb[0]):
64 print(
'WARNING, the localDB ' + localdb[0] +
' containing the corrections is missing, exiting.')
67 use_local_database(localdb[0] +
"/database.txt", localdb[0], invertLogging=
True)
69 if os.path.isdir(localdb):
70 print(
'WARNING, you are using CoG corrections in ' + localdb +
' exiting')
73 use_local_database(localdb +
"/database.txt", localdb, invertLogging=
True)
77 rootinput = register_module(
'RootInput')
78 rootinput.param(
'inputFileNames', inputFileList)
79 rootinput.param(
'branchNames', branches)
80 main.add_module(rootinput)
82 main.add_module(
"Gearbox")
83 main.add_module(
"Geometry", useDB=
True)
86 trkFlt = register_module(
'TrackFilter')
87 trkFlt.param(
'outputFileName', trk_outputFile)
88 trkFlt.param(
'outputINArrayName',
'SelectedTracks')
89 trkFlt.param(
'outputOUTArrayName',
'ExcludedTracks')
90 trkFlt.param(
'min_NumHitSVD', nSVD)
91 trkFlt.param(
'min_NumHitCDC', nCDC)
92 trkFlt.param(
'min_Pvalue', pVal)
96 fil = register_module(
'SVDShaperDigitsFromTracks')
97 fil.param(
'outputINArrayName',
'SVDShaperDigitsFromTracks')
100 add_svd_reconstruction(main)
102 for moda
in main.modules():
103 if moda.name() ==
'SVDCoGTimeEstimator':
104 moda.param(
"ShaperDigits",
'SVDShaperDigitsFromTracks')
105 moda.param(
"RecoDigits",
'SVDRecoDigitsFromTracks')
106 if moda.name() ==
'SVDSimpleClusterizer':
107 moda.param(
"Clusters",
'SVDClustersFromTracks')
108 moda.param(
"RecoDigits",
'SVDRecoDigitsFromTracks')
109 if moda.name() ==
'SVDSpacePointCreator':
110 moda.param(
"SVDClusters",
'SVDClustersFromTracks')
112 calib = SVDCoGTimeCalibrationImporterModule()
113 calib.set_localdb(localdb)
114 calib.notApplyCorrectForCDCLatency(
False)
115 main.add_module(calib)
118 progress = register_module(
'ProgressBar')
119 main.add_module(progress)