16 from ROOT
import Belle2
24 data_dir =
'/group/belle2/dataprod/Data/release-03-02-02/DB00000635/proc00000009_nofilter'
25 skim_dir =
'skim/hlt_bhabha/cdst/sub00'
26 globalTag =
'data_reprocessing_prompt'
27 stagingTags = [
'staging_data_reprocessing']
29 output_dir =
'moduleT0'
36 print(
"usage: basf2", argvs[0],
"experiment runFirst runLast")
38 experiment = int(argvs[1])
39 run_first = int(argvs[2])
40 run_last = int(argvs[3])
42 expNo =
'e' +
'{:0=4d}'.format(experiment)
46 for run
in range(run_first, run_last + 1):
47 runNo =
'r' +
'{:0=5d}'.format(run)
48 for typ
in [
'4S',
'Continuum',
'Scan']:
49 folder = data_dir +
'/' + expNo +
'/' + typ +
'/' + runNo +
'/' + skim_dir
50 files += glob.glob(folder +
'/cdst.*.root')
52 B2ERROR(
'No cdst files found')
57 output_folder = output_dir +
'/' + expNo +
'/' + sampleType +
'/' + method
58 if not os.path.isdir(output_folder):
59 os.makedirs(output_folder)
60 print(
'New folder created: ' + output_folder)
63 fileName = output_folder +
'/moduleT0-' + expNo +
'-'
64 run1 =
'r' +
'{:0=5d}'.format(run_first)
65 run2 =
'r' +
'{:0=5d}'.format(run_last)
66 fileName += run1 +
'_to_' + run2 +
'.root'
67 print(
'Output file:', fileName)
71 ''' exclude (mask-out) BS 13d '''
74 ''' event processing '''
77 if digit.getModuleID() == 13
and digit.getBoardstackNumber() == 3:
78 digit.setHitQuality(Belle2.TOPDigit.c_Junk)
82 use_central_database(globalTag)
83 for tag
in stagingTags:
84 use_central_database(tag)
86 if os.path.isfile(db):
87 use_local_database(db, invertLogging=
True)
89 B2ERROR(db +
": local database not found")
96 main.add_module(
'RootInput', inputFileNames=files)
99 main.add_module(
'TOPGeometryParInitializer')
102 main.add_module(
'TOPTimeRecalibrator', subtractBunchTime=
False)
105 main.add_module(
'TOPChannelMasker')
111 main.add_module(
'TOPBunchFinder', usePIDLikelihoods=
True, subtractRunningOffset=
False)
114 main.add_module(
'TOPModuleT0Calibrator', sample=sampleType, outputFileName=fileName)
117 progress = register_module(
'Progress')
118 main.add_module(progress)