ReadTripsProbsFromCool.py

Go to the documentation of this file.

#!/bin/env python
 
# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
#
# ./ReadTripsProbsFromCool.py  --schema='COOLOFL_TILE/OFLP200'  --folder='OFL02' --tag='UPD4'
 
import getopt,sys,os
os.environ['TERM'] = 'linux'
 
def usage():
    print ("Usage: ",sys.argv[0]," [OPTION] ... ")
    print ("Dumps the TileCal drawer trips probabilities from various schemas / folders")
    print ("")
    print ("-h, --help     shows this help")
    print ("-f, --folder=  specify status folder, by default OFL02 ")
    print ("-t, --tag=     specify tag to use, f.i. UPD1 or UPD4")
    print ("-r, --run=     specify run  number, by default uses latest iov")
    print ("-l, --lumi=    specify lumi block number, default is 0")
    print ("-s, --schema=  specify schema to use, like 'COOLOFL_TILE/OFLP200' or 'sqlite://;schema=tileSqlite.db;dbname=OFLP200'")
    print ("-S, --server=  specify server - ORACLE or FRONTIER, default is FRONTIER")
 
letters = "hr:l:S:s:t:f:"
keywords = ["help","run=","lumi=","server=","schema=","tag=","folder="]
 
try:
    opts, extraparams = getopt.getopt(sys.argv[1:],letters,keywords)
except getopt.GetoptError as err:
    print (str(err))
    usage()
    sys.exit(2)
 
# defaults
run = 2147483647
lumi = 0
server = ''
schema = 'COOLOFL_TILE/OFLP200'
folderPath =  "/TILE/OFL02/STATUS/ADC"
tag = "SDR-BS8T-10"
 
for o, a in opts:
    a = a.strip()
    if o in ("-f","--folder"):
        folderPath = "/TILE/%s/STATUS/ADC" % a
    elif  o in ("-t","--tag"):
        tag = a
    elif o in ("-S","--server"):
        server = a
    elif o in ("-s","--schema"):
        schema = a
    elif o in ("-r","--run"):
        run = int(a)
    elif o in ("-l","--lumi"):
        lumi = int(a)
    elif o in ("-h","--help"):
        usage()
        sys.exit(2)
    else:
        raise RuntimeError("unhandled option")
 
 
 
from TileCalibBlobPython import TileCalibTools
from TileCalibBlobObjs.Classes import TileCalibUtils
 
from TileCalibBlobPython.TileCalibLogger import getLogger
log = getLogger("ReadTripsProbs")
import logging
logLevel=logging.DEBUG
log.setLevel(logLevel)
log1 = getLogger("TileCalibTools")
log1.setLevel(logLevel)
 
 
#=== set database
db = TileCalibTools.openDbConn(schema, server)
folderTag = TileCalibTools.getFolderTag(db, folderPath, tag)
log.info("Initializing folder %s with tag %s", folderPath, folderTag)
 
blobReader = TileCalibTools.TileBlobReader(db, folderPath, folderTag)
 
import cppyy
util = cppyy.gbl.TileCalibUtils()
tripsCalibDrawer = blobReader.getDrawer(util.trips_ros(), util.trips_drawer(), (run,lumi))
 
if tripsCalibDrawer.getNChans() != util.max_ros() \
   or tripsCalibDrawer.getObjSizeUint32() != (util.max_drawer() + 1):
    log.info("There no drawer trips probabilities in tag %s", folderTag)
    sys.exit(2)
 
 
print ("")
print ("Module\tTrip (probability)")
print ("")
 
for ros in range(1, util.max_ros()):
    denominator = tripsCalibDrawer.getData(ros, 0, util.max_drawer())
    for mod in range(0, min(64, TileCalibUtils.getMaxDrawer(ros))):
        modName = TileCalibUtils.getDrawerString(ros, mod)
        trip = tripsCalibDrawer.getData(ros, 0, mod)
        #log.info("%s\t%s" % (modName, str(float(trip)/ denominator)))
        print ("%s\t%s" % (modName, str(float(trip)/ denominator)))
 
 
#=== close DB
db.closeDatabase()