checkCalibDB.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 TOP calibration in central DB
#
# usage: basf2 checkCalibDB.py expNo runNo [globalTag]
# --------------------------------------------------------------------
 
argvs = sys.argv
if len(argvs) < 3:
    print("usage: basf2", argvs[0], "expNo runNo [globalTag]")
    sys.exit()
 
expNo = int(argvs[1])
runNo = int(argvs[2])
tag = '(default)'
if len(argvs) == 4:
    tag = argvs[3]
 
 
class CheckCalibDB(b2.Module):
    ''' print calibration status of TOPCal payloads '''
 
    def printChannel(self, payload):
        ''' print status of a payload given by the argument '''
 
        db = Belle2.PyDBObj(payload)
        if not db:
            b2.B2ERROR(payload + ' not found')
            return
        calibrated = 0
        all_entries = 0
        for moduleID in range(1, 17):
            for channel in range(512):
                all_entries += 1
                if db.isCalibrated(moduleID, channel):
                    calibrated += 1
        print(payload + ': ' + str(calibrated) + '/' + str(all_entries) + ' calibrated')
 
    def printChannelMask(self):
        ''' print status of channel masks '''
 
        payload = 'TOPCalChannelMask'
        db = Belle2.PyDBObj(payload)
        if not db:
            b2.B2ERROR(payload + ' not found')
            return
        active = db.getNumOfActiveChannels()
        all_entries = db.getNumOfChannels()
        print(payload + ': ' + str(active) + '/' + str(all_entries) + ' active')
 
    def printModule(self, payload):
        ''' print status of a payload given by the argument '''
 
        db = Belle2.PyDBObj(payload)
        if not db:
            b2.B2ERROR(payload + ' not found')
            return
        calibrated = 0
        all_entries = 0
        for moduleID in range(1, 17):
            all_entries += 1
            if db.isCalibrated(moduleID):
                calibrated += 1
        print(payload + ': ' + str(calibrated) + '/' + str(all_entries) + ' calibrated')
 
    def printCommon(self, payload):
        ''' print status of a payload given by the argument '''
 
        db = Belle2.PyDBObj(payload)
        if not db:
            b2.B2ERROR(payload + ' not found')
            return
        calibrated = 0
        all_entries = 1
        if db.isCalibrated():
            calibrated += 1
        print(payload + ': ' + str(calibrated) + '/' + str(all_entries) + ' calibrated')
 
    def printTimeBase(self):
        ''' print status of time base calibration '''
 
        payload = 'TOPCalTimebase'
        db = Belle2.PyDBObj(payload)
        if not db:
            b2.B2ERROR(payload + ' not found')
            return
        calibrated = 0
        all_entries = 8192
        for sampleTimes in db.getSampleTimes():
            if sampleTimes.isCalibrated():
                calibrated += 1
        print(payload + ': ' + str(calibrated) + '/' + str(all_entries) + ' calibrated')
 
    def event(self):
        ''' event processing '''
 
        print()
        print('Calibration status of GT =', tag)
        print('Experiment =', expNo, 'Run =', runNo)
        print()
        self.printTimeBaseprintTimeBase()
        self.printChannelprintChannel('TOPCalChannelT0')
        self.printModuleprintModule('TOPCalModuleT0')
        self.printCommonprintCommon('TOPCalCommonT0')
        self.printChannelprintChannel('TOPCalChannelNoise')
        self.printChannelprintChannel('TOPCalChannelPulseHeight')
        self.printChannelprintChannel('TOPCalChannelRQE')
        self.printChannelprintChannel('TOPCalChannelThresholdEff')
        self.printChannelprintChannel('TOPCalChannelThreshold')
        self.printChannelprintChannel('TOPCalIntegratedCharge')
        self.printModuleprintModule('TOPCalModuleAlignment')
        self.printChannelMaskprintChannelMask()
        print()
 
 
# Central database
if len(argvs) == 4:
    b2.conditions.append_globaltag(tag)
 
# Create path
main = b2.create_path()
 
# Set number of events to generate
eventinfosetter = b2.register_module('EventInfoSetter')
eventinfosetter.param({'evtNumList': [1], 'runList': [runNo], 'expList': [expNo]})
main.add_module(eventinfosetter)
 
# Run checker
main.add_module(CheckCalibDB())
 
# Process events
b2.process(main)