python/runInfo.py

Go to the documentation of this file.

#!/bin/env python
 
# Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
 
import os,sys
from PyCool import cool
 
from CoolRunQuery.utils.AtlRunQueryLookup import InitDetectorMaskDecoder
 
def main( runNum=None, projectName='' ):
    if runNum is None:
        print('ERROR no runNumber given')
        sys.exit(-1)
 
    if projectName == '':
        print('ERROR no projectName given')
        sys.exit(-1)
 
    year=int(projectName[4:6])
 
    print('runInfo from python folder')
    since = ( runNum << 32 )
    until = ( (runNum+1) << 32 )-1
    dbSvc = cool.DatabaseSvcFactory.databaseService()
    #load COMP200 or CONDBR2 depending on date A.Gascon 2014-12-08
 
    if os.path.exists("/afs/cern.ch/work/s/sctcalib/CondReadPwd") :
        pwdinfo = open( '/afs/cern.ch/work/s/sctcalib/CondReadPwd', 'r' ).read()
        pwdinfolist = pwdinfo.split( ' ' )
        pwd = pwdinfolist[0]
    else :
        print('ERROR : unable to open pwd file')
        sys.exit(-1)
 
    if (year > 13):
        RunCtrlDB = 'oracle://ATLAS_COOLPROD;schema=ATLAS_COOLONL_TDAQ;dbname=CONDBR2;user=ATLAS_COOL_READER;password='+pwd
    else:
        RunCtrlDB = 'oracle://ATLAS_COOLPROD;schema=ATLAS_COOLONL_TDAQ;dbname=COMP200;user=ATLAS_COOL_READER;password='+pwd
    
    try:
        RunCtrlDb = dbSvc.openDatabase( RunCtrlDB )
    except Exception as e:
        print('Problem opening database', e)
        sys.exit(-1)
    
    folder_runCtrl = RunCtrlDb.getFolder( '/TDAQ/RunCtrl/EOR' )
    chList = folder_runCtrl.listChannels()
    if not chList:
        print('ERROR : no data in /TDAQ/RunCtrl/EOR')
 
    for Id in chList:
 
        objs = folder_runCtrl.browseObjects( since, until, cool.ChannelSelection(Id) )
        for obj in objs:
            payl = obj.payload()
            #print(payl)
            RunNumber    = payl[ 'RunNumber' ]
            SORTime      = payl[ 'SORTime' ]
            EORTime      = payl[ 'EORTime' ]
            iovstart     = obj.since()
            iovend       = obj.until()
            nLB          = iovend - ( iovstart + 1 )
            ProjectTag   = payl[ 'T0ProjectTag' ]
            DetectorMask = payl[ 'DetectorMask' ]
        objs.close()
        break
    RunCtrlDb.closeDatabase()
 
    ts1 = SORTime/1000000000
    ts2 = EORTime/1000000000
    
    #--- Detector mask
    mask= int (DetectorMask,16)
 
    if (year < 12):
        dName, NotInAll, vetoedbits = InitDetectorMaskDecoder( 1 )
    elif (year < 21):
        dName, NotInAll, vetoedbits = InitDetectorMaskDecoder( 2 )
    else:
        dName, NotInAll, vetoedbits = InitDetectorMaskDecoder( 3 )
 
    res=''
    found = False
    for i in range(64):
        if ( mask & (1 << i) ) :
            if "SCT BA" in dName[i] :
                found = True
                res += "SCTBA,"
            if "SCT BC" in dName[i] :
                found = True
                res += "SCTBC,"
            if "SCT EA" in dName[i] :
                found = True
                res += "SCTEA,"
            if "SCT EC" in dName[i] :
                found = True
                res += "SCTEC,"
 
    if found:
        if len(res)>1:
            res = res[:-1]
        else:
            res += 'NONE'
        
    #print(RunNumber, ProjectTag, ts1, ts2, nLB, res)
    fw = open( 'runInfo.txt', 'w' )
    fw.write( str(RunNumber) + ' ' + str(ProjectTag) + ' ' + str(ts1) + ' ' + str(ts2) + ' ' + str(nLB) + ' ' + res )
    fw.close()
     
if __name__ == "__main__":
    try:
        runNumber = int( sys.argv[1] )
    except Exception:
        print("run number %s could not be converted to int"%(sys.argv[1]))
        sys.exit(-1)
        
    main(runNum = runNumber)