checkDB-commonT0.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
 
 
 
import basf2 as b2
import sys
from ROOT import Belle2
 
# ------------------------------------------------------------------------
# useful tool for checking the status of common T0 calibration in DB
#
# usage: basf2 checkDB-commonT0.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 TOPCalCommonT0 '''
 
    def initialize(self):
        ''' initialize '''
 
        
        self.dbdb = Belle2.PyDBObj('TOPCalCommonT0')
        
        self.lastRunlastRun = None
        
        self.bunchTimeSepbunchTimeSep = 47.163878 / 24
 
        print()
        print('Common 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' + '{:0=5d}'.format(evtMetaData.getRun())
 
        if not self.dbdb:
            b2.B2ERROR(runNo + ': payload not found')
            return
        if not self.dbdb.hasChanged():
            self.lastRunlastRun = runNo
            return
        if self.lastRunlastRun:
            print('... to ' + self.lastRunlastRun)
            self.lastRunlastRun = None
 
        if self.dbdb.isCalibrated():
            status = 'calibrated'
        elif self.dbdb.isUnusable():
            status = 'unusable'
        elif self.dbdb.isRoughlyCalibrated():
            status = 'roughly calibrated'
        else:
            status = 'default'
        n = round(self.dbdb.getT0() / self.bunchTimeSepbunchTimeSep, 0)
        if n == 0:
            print(runNo + ': T0 =',
                  round(self.dbdb.getT0(), 4), '+/-', round(self.dbdb.getT0Error(), 4),
                  status)
        else:
            print(runNo + ': T0 =',
                  round(self.dbdb.getT0(), 4), '+/-', round(self.dbdb.getT0Error(), 4),
                  status, '-- out of range')
 
    def terminate(self):
        ''' terminate '''
 
        if self.lastRunlastRun:
            print('... to ' + self.lastRunlastRun)
            self.lastRunlastRun = 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)