MuonSpectrometer/MuonCalib/MuonCalibDbOperations/python/ReadCool.py

Go to the documentation of this file.

# Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
 
from __future__ import print_function
 
import CoolConvUtilities.AtlCoolLib as AtlCoolLib
from PyCool import cool,coral
import zlib,StringIO
import sys
import re
from MuonCalibDbOperations.MuonCalibResolveTag import ResolveTag
from MuonCalibDbOperations.MuonCalibConvertTimeSlewing import TimeSlewingApplied, NoTs2Ts, GasmonDriftTimeOffsetT0
    
sys.argv=[sys.argv[0], '-b']
 
from ROOT import TGraphErrors, TSpline3, gDirectory
from MuonCalibIdentifier.MuonFixedIdUnpack import MuonFixedIdUnpack
 
def UnpackData(data):
    if type(data)==str:
        return data
    if type(data)==coral.Blob:
        blob=StringIO.StringIO()
        for i in range(data.size()):
            blob.write(chr(data[i]))
        unpacked=zlib.decompress(blob.getvalue())
        return unpacked
 
def iov_keygen(iov):
    return iov[0]*10000000 + iov[1]
 
def DumpFolderSummary(db_string, folder, tag, run=None):
    
    try:
        db=AtlCoolLib.indirectOpen(db_string, readOnly=True, debug=True)
    except Exception as e:
        print ('Problem opening database',e)
        sys.exit(-1)
 
#get folder and tag 
    cool_folder=db.getFolder(folder)
    cool_tag=ResolveTag(cool_folder, tag)
    
    counters={}
    if run:
        objs=cool_folder.browseObjects((run<<32), ((run+1)<<32), cool.ChannelSelection.all(), cool_tag)
    else:
        objs=cool_folder.browseObjects(0, (999999<<32), cool.ChannelSelection.all(), cool_tag)
    for obj in objs:
        file_col=obj.payload()['file']
        file_items=re.split('[A-Z,a-z]*', file_col)
        site="default"
        head_id=-1
        if len(file_items)==2:
            try:
                head_id = int(file_items[1])
                site = re.sub("[0-9]", "", file_col)
            except ValueError:
                site=file_col
        ts="NOTS"
        if TimeSlewingApplied(obj):
            ts="TS"     
        ident=(obj.since()>>32, obj.until()>>32, site, head_id, ts)
        if ident not in counters:
            counters[ident] = 0
        counters[ident] +=1
    for ident in sorted(counters.keys(), key=iov_keygen):
        print ("[", ident[0], ",", ident[1], "[", ident[2], ident[3], ident[4], ":" , counters[ident])
            
 
def ReadRtCool(db_string, folder, tag, run_number):
 
    try:
        db=AtlCoolLib.indirectOpen(db_string, readOnly=True, debug=True)
    except Exception as e:
        print ('Problem opening database',e)
        sys.exit(-1)
 
#get folder and tag 
    cool_folder=db.getFolder(folder)
    cool_tag=ResolveTag(cool_folder, tag)
 
#data to be filled
    graphs={}
    splines={}  
    iovs=set([])
    
    myiov=(run_number<<32)
    
    objs=cool_folder.browseObjects(myiov,myiov, cool.ChannelSelection.all(), cool_tag)
    for obj in objs:
        iov=(obj.since(), obj.until())
        iovs.add(iov)
        sp=UnpackData(obj.payload()['data']).split('\n')
        sp1=sp[0].split(',')
        chamber=int(sp1[0])
        n_points=int(sp1[1])    
        sp2=sp[1].split(',')
        graphs[chamber]=TGraphErrors(n_points)
        up=MuonFixedIdUnpack(chamber)
        if up.stationNameString()=='XXX':
            print ("Invalid station name in ", obj.payload()['file'])
            sys.exit(1)
        nm=up.stationNameString() + "_" + str(up.stationPhi()) + "_" + str(up.stationEta())
#       print (nm)
        ts_applied=TimeSlewingApplied(obj)
        for i in range(0, n_points):
            r=float(sp2[3*i])
            t=float(sp2[3*i+1])
            s=float(sp2[3*i+2])
            if not ts_applied:
                t_new=NoTs2Ts(r, t)
                t=t_new
            graphs[chamber].SetPoint(i, r, t)
            graphs[chamber].SetPointError(i, 0, s)
        splines[chamber]=TSpline3("sp_" + nm, graphs[chamber])
        if gDirectory.IsWritable():
            splines[chamber].Write("sp_" + nm)
            graphs[chamber].Write("gr_" + nm)
    
    return graphs, splines, iovs
 
 
def ReadT0Cool(db_string, folder, tag, run_number):
    
    try:
        db=AtlCoolLib.indirectOpen(db_string, readOnly=True, debug=True)
    except Exception as e:
        print ('Problem opening database',e)
        sys.exit(-1)
 
#get folder and tag 
    cool_folder=db.getFolder(folder)
    cool_tag=ResolveTag(cool_folder, tag)
    myiov=(run_number<<32)
#data
    t0_values={}
    
    objs=cool_folder.browseObjects(myiov,myiov, cool.ChannelSelection.all(), cool_tag)
    
    for obj in objs:
        sp=UnpackData(obj.payload()['data']).split('\n')
        sp1=sp[0].split(',')
        ch_sp=sp1[0][2:].split("_")
        ch_id=(ch_sp[0], int(ch_sp[1]), int(ch_sp[2]))
        ntubes=int(sp1[-1])
        
        t0_items=sp[1].split(",")
        t0_values[ch_id] = []
        ts_applied=TimeSlewingApplied(obj)
        for i in range(0, ntubes):
            t0=float(t0_items[3*i+0])
            if not ts_applied:
                t0+= GasmonDriftTimeOffsetT0
            t0_values[ch_id].append((t0, int(t0_items[3*i+1]), float(t0_items[3*i+2])))
    
    return t0_values
        
    
def GetFolderTag(db_string, folder):
    dbSvc=cool.DatabaseSvcFactory.databaseService()
    try:
        db=dbSvc.openDatabase(db_string)
    except Exception as e:
        print ('Problem opening database',e)
        sys.exit(-1)
 
#get folder and tag 
    cool_folder=db.getFolder(folder)
    for x in cool_folder.listTags():
        return x