getModuleT0Histo.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
 
 
 
import basf2 as b2
import sys
from ROOT import Belle2
from ROOT import TFile, TH1F
 
# ------------------------------------------------------------------------
# Save module T0 constants as a histogram into a root file
#
# usage: basf2 getModuleT0Histo.py expNo runNo globalTag/localDB
# -------------------------------------------------------------------------
 
argvs = sys.argv
if len(argvs) < 4:
    print("usage: basf2", argvs[0], "expNo runNo globalTag/localDB")
    sys.exit()
 
expNo = int(argvs[1])
runNo = int(argvs[2])
tag = argvs[3]
 
 
class SaveModuleT0(b2.Module):
    ''' Save module T0 from database into a root file as histogram '''
 
    def initialize(self):
        ''' initialize '''
 
        
        self.dbdb = Belle2.PyDBObj('TOPCalModuleT0')
 
        file_name = 'moduleT0_DB-' + 'e' + '{:0=4d}'.format(expNo) + '-r' + '{:0=5d}'.format(runNo) + '.root'
        
        self.filefile = TFile.Open(file_name, 'recreate')
        
        self.hh = TH1F('moduleT0', tag + '; slot number; module T0 [ns]', 16, 0.5, 16.5)
 
    def event(self):
        ''' event processing '''
 
        for slot in range(1, 17):
            self.hh.SetBinContent(slot, self.dbdb.getT0(slot))
            self.hh.SetBinError(slot, self.dbdb.getT0Error(slot))
            if self.dbdb.isCalibrated(slot):
                print('slot', slot, 'status = calibrated')
            elif self.dbdb.isUnusable(slot):
                print('slot', slot, 'status = unusable')
            else:
                print('slot', slot, 'status = default')
 
    def terminate(self):
        ''' terminate '''
 
        self.filefile.Write()
        self.filefile.Close()
 
 
# 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]
runList = [runNo]
expList = [expNo]
 
# Event info setter
eventinfosetter = b2.register_module('EventInfoSetter')
eventinfosetter.param({'evtNumList': evtList, 'runList': runList, 'expList': expList})
main.add_module(eventinfosetter)
 
# Save module T0 from database into a root file as histogram
main.add_module(SaveModuleT0())
 
# Process events
b2.process(main)