18 from caf
import backends
19 from caf.framework
import CAF
20 from basf2
import B2ERROR
21 from top_calibration
import commonT0_calibration_BF, commonT0_calibration_LL
25 globalTags = [
'Reco_master_patch',
'data_reprocessing_proc10']
27 data_dir =
'/group/belle2/dataprod/Data/OfficialReco/proc10/'
28 bhabha_skim_dir =
'skim/hlt_bhabha/cdst/sub00'
29 dimuon_skim_dir =
'offskim/offskim_mumutop/cdst/sub00'
30 main_output_dir =
'top_calibration'
31 default_sample =
'bhabha'
33 new_cdst_format =
False
40 print(
"usage: basf2", argvs[0],
"experiment runFirst runLast [sample method]")
41 print(
" sample = bhabha/dimuon (D = bhabha)")
42 print(
" method = BF/LL (D = BF)")
44 experiment = int(argvs[1])
45 run_first = int(argvs[2])
46 run_last = int(argvs[3])
47 sample = default_sample
48 method = default_method
56 if sample ==
'bhabha':
57 skim_dir = bhabha_skim_dir
58 elif sample ==
'dimuon':
59 skim_dir = dimuon_skim_dir
61 B2ERROR(
"Invalid sample name: " + sample)
66 expNo =
'e' +
'{:0=4d}'.format(experiment)
67 for run
in range(run_first, run_last + 1):
68 runNo =
'r' +
'{:0=5d}'.format(run)
69 filename = f
"{data_dir}/{expNo}/*/{runNo}/{skim_dir}/cdst*.root"
70 inputFiles += glob.glob(filename)
72 if len(inputFiles) == 0:
73 B2ERROR(
'No cdst files found in ' + data_dir +
' for exp=' + str(experiment) +
74 ' runFirst=' + str(run_first) +
' runLast=' + str(run_last) +
75 ' (skim_dir=' + skim_dir +
')')
79 run_range =
'r' +
'{:0=5d}'.format(run_first) +
'-' +
'{:0=5d}'.format(run_last)
80 output_dir = f
"{main_output_dir}/commonT0-{sample}-{method}-{expNo}-{run_range}"
84 cal = commonT0_calibration_BF(inputFiles, globalTags, localDBs, new_cdst_format)
86 cal = commonT0_calibration_LL(inputFiles, sample, globalTags, localDBs, new_cdst_format)
88 B2ERROR(
'Invalid method name: ' + method)
90 cal.backend_args = {
"queue":
"s"}
94 cal_fw.add_calibration(cal)
95 cal_fw.output_dir = output_dir