6 from ROOT.Belle2
import TOPDatabaseImporter
7 from ROOT
import gSystem
16 main.add_module(
'EventInfoSetter', evtNumList=[1])
19 main.add_module(
'TOPGeometryParInitializer')
25 use_local_database(
'localDB/localDB.txt', readonly=
False)
28 dbImporter = TOPDatabaseImporter()
31 fileNames = sorted(glob.glob(
'masks/*.root'))
32 numFiles = len(fileNames)
34 print(
'No files found')
37 if not os.path.exists(
'masks/imported'):
38 os.makedirs(
'masks/imported')
40 for i, fileName
in enumerate(fileNames):
41 runFirst = int((fileName.split(
'_r')[1]).split(
'.')[0])
42 expNo = int((fileName.split(
'_e')[1]).split(
'_')[0])
45 if k < len(fileNames):
46 nextName = fileNames[k]
47 runLast = int((nextName.split(
'_r')[1]).split(
'.')[0]) - 1
48 expNext = int((nextName.split(
'_e')[1]).split(
'_')[0])
49 if expNext > expNo
or k == len(fileNames):
51 elif runLast < runFirst:
52 B2ERROR(
"first run:", runFirst,
"last run:", runLast)
53 B2ERROR(
"Last run is less than the first one: exiting!")
55 dbImporter.importChannelMask(fileName, expNo, runFirst, runLast)
56 os.rename(fileName, fileName.replace(
'masks/',
'masks/imported/'))