checkDB-moduleT0.py

#!/usr/bin/env python
 
 
 
import basf2 as b2
import sys
from ROOT import Belle2
 
# ------------------------------------------------------------------------
# useful tool for checking the status of module T0 calibration in DB
#
# usage: basf2 checkDB-moduleT0.py expNo runFirst runLast globalTag/localDB
# -------------------------------------------------------------------------
 
argvs = sys.argv
if len(argvs) < 5:
    print("usage: basf2", argvs[0], "expNo runFirst runLast globalTag/localDB")
    sys.exit()
 
expNo = int(argvs[1])
runFirst = int(argvs[2])
runLast = int(argvs[3])
tag = argvs[4]
 
 
class CheckCalibDB(b2.Module):
    ''' print content of TOPCalModuleT0 '''
 
    def initialize(self):
        ''' initialize '''
 
        
        self.db = Belle2.PyDBObj('TOPCalModuleT0')
        
        self.lastRun = None
 
        print()
        print('Module T0 calibration status of GT =', tag)
        print('Experiment =', expNo, 'Runs =', runFirst, 'to', runLast)
        print()
 
    def event(self):
        ''' event processing '''
 
        evtMetaData = Belle2.PyStoreObj('EventMetaData')
        runNo = 'r' + f'{evtMetaData.getRun():05d}'
 
        if not self.db:
            b2.B2ERROR(runNo + ': payload not found')
            return
        if not self.db.hasChanged():
            self.lastRun = runNo
            return
        if self.lastRun:
            print('... to ' + self.lastRun)
            self.lastRun = None
 
        print(runNo + ':')
        for slot in range(1, 17):
            if self.db.isCalibrated(slot):
                status = 'calibrated'
            elif self.db.isUnusable(slot):
                status = 'unusable'
            else:
                status = 'default'
            print('  slot' + f'{slot:02d}' + ': T0 =',
                  round(self.db.getT0(slot), 4), '+/-', round(self.db.getT0Error(slot), 4),
                  status)
 
    def terminate(self):
        ''' terminate '''
 
        if self.lastRun:
            print('... to ' + self.lastRun)
            self.lastRun = None
 
 
# Database
if '.txt' in tag:
    b2.conditions.append_testing_payloads(tag)
else:
    b2.conditions.append_globaltag(tag)
 
# Create path
main = b2.create_path()
 
# Set number of events to generate
evtList = [1 for run in range(runFirst, runLast + 1)]
runList = [run for run in range(runFirst, runLast + 1)]
expList = [expNo for run in range(runFirst, runLast + 1)]
 
# Event info setter
eventinfosetter = b2.register_module('EventInfoSetter')
eventinfosetter.param({'evtNumList': evtList, 'runList': runList, 'expList': expList})
main.add_module(eventinfosetter)
 
# Run checker
main.add_module(CheckCalibDB())
 
# Process events
b2.process(main)