10 set_log_level(LogLevel.INFO)
14 import multiprocessing
17 from ROOT
import Belle2
18 from ROOT.Belle2
import SVDHotStripsCalibrationsAlgorithm
20 from caf.framework
import Calibration, CAF, Collection, LocalDatabase, CentralDatabase
21 from caf
import backends
22 from caf
import strategies
25 import reconstruction
as reco
26 import modularAnalysis
as ana
36 def SVDHotStripsCalibrations(files, tags):
43 path.add_module(
'Progress')
45 path.add_module(
'RootInput', branchNames=input_branches)
47 path.add_module(
"Gearbox")
48 path.add_module(
"Geometry", useDB=
True)
56 raw.add_unpackers(path, components=[
'SVD'])
58 collector = register_module(
'SVDOccupancyCalibrationsCollector')
59 collector.param(
"SVDShaperDigitsName",
"SVDShaperDigits")
60 algorithm = SVDHotStripsCalibrationsAlgorithm(
"SVDHotStripsCAF")
66 pre_collector_path=path,
67 database_chain=[CentralDatabase(tag)
for tag
in tags],
69 max_files_per_collector_job=1,
78 if __name__ ==
"__main__":
80 input_files = [os.path.abspath(file)
for file
in [
81 "/group/belle2/dataprod/Data/Raw/e0008/r01309/sub00/physics.0008.01309.HLT5.f00098.root"]]
86 if not len(input_files):
87 print(
"You have to specify some input file(s)\n"
88 "using the standard basf2 command line option - i")
89 print(
"See: basf2 -h")
92 svdOccupCAF = SVDHotStripsCalibrations(input_files,
93 [
'giulia_CDCEDepToADCConversions_rel4_patch',
94 'data_reprocessing_prompt_rel4_patch'])
98 cal_fw.add_calibration(svdOccupCAF)
102 if multiprocessing.cpu_count() < 10: