python.MagFieldUtils Namespace Reference

Classes
class	MagFieldDCSInfo

Functions
	getFieldForRun (run, quiet=False, lumiblock=None)
	getTimeForLB (run, LB)

Variables
	_timeForLB = dict()
	_fieldForLB = dict()
	run = int(sys.argv[1])
	lumiblock = None
	magfield = getFieldForRun(run,lumiblock=lumiblock)

Function Documentation

getFieldForRun()

python.MagFieldUtils.getFieldForRun	(		run,
			quiet = False,
			lumiblock = None )

Definition at line 47 of file MagFieldUtils.py.

def getFieldForRun(run,quiet=False,lumiblock=None):
    "Get the magnetic field currents (MagFieldDCSInfo) for a given run"
    # access the TDAQ schema to translate run number into timestamp
    # and get the filename tag
 
    runiov=(run << 32)+(lumiblock or 0)
    if runiov in _fieldForLB:
        return _fieldForLB[runiov]
 
    newdb=(run>=236107)
    if not quiet:
        print ("Reading magnetic field for run %i, CONDBR2 %s" % (run,newdb))
    # setup appropriate connection and folder parameters
    if newdb:
        dbname='CONDBR2'
        sorfolder='/TDAQ/RunCtrl/SOR'
        fntname='T0ProjectTag'
    else:
        dbname='COMP200'
        sorfolder='/TDAQ/RunCtrl/SOR_Params'
        fntname='FilenameTag'
 
    tdaqDB=indirectOpen('COOLONL_TDAQ/%s' % dbname)
    if (tdaqDB is None):
        print ("MagFieldUtils.getFieldForRun ERROR: Cannot connect to COOLONL_TDAQ/%s" % dbname)
        return None
    sortime=0
    try:
        tdaqfolder=tdaqDB.getFolder(sorfolder)
        runiov=run << 32
        obj=tdaqfolder.findObject(runiov,0)
        payload=obj.payload()
        sortime=payload['SORTime']
        fnt=payload[fntname]
    except Exception as e:
        print ("MagFieldUtils.getFieldForRun ERROR accessing folder %s" % sorfolder)
        print (e)
    tdaqDB.closeDatabase()
    
    # if lumiblock is specifed, actually want the start time of the LB
    if lumiblock is not None:
        if not quiet:
            print ("Reading specific timestamp for lumiblock %i" % lumiblock)
        
        lbtime=getTimeForLB(run,lumiblock)
        if (lbtime==0 and lumiblock>1):
            # sometimes fails as last LB is missing in LBLB - try previous
            print ("MagFieldUtils.getFieldForRun WARNING: Cannot find LB %i, trying %i" % (lumiblock,lumiblock-1))
            lbtime=getTimeForLB(run,lumiblock-1)
        if (lbtime==0):
            print ("MagFieldUtils.getFieldForRun WARNING: Cannot find LB %i, fall back on SOR time" % lumiblock)
        if (lbtime>0):
            # use this time instead of SORtime
            if not quiet:
                print ("Lumiblock starts %i seconds from start of run" % int((lbtime-sortime)/1.E9))
            sortime=lbtime
        else:
            print ("MagFieldUtils.getFieldForRun ERROR accessing /TRIGGER/LUMI/LBLB")
            print ("Fall back on SOR time from %s" % sorfolder)
            lbtime=sortime
    
    # if we do not have a valid time, exit
    if (sortime==0): return None
    
    # now having got the start of run timestamp, lookup the field info in DCS
    dcsDB=indirectOpen('COOLOFL_DCS/%s' % dbname)
    if (dcsDB is None):
        print ("MagFieldUtils.getFieldForRun ERROR: Cannot connect to COOLOFL_DCS/%s" % dbname)
        return None
    data=None
    try:
        # map of expected channel names to slots in data[] variable
        # follows original order from run1/COMP200
        # has changed in CONDBR2, but use of named channels recovers this
        currentmap={'CentralSol_Current':0,'CentralSol_SCurrent':1,
                    'Toroids_Current':2,'Toroids_SCurrent':3}
        dcsfolder=dcsDB.getFolder('/EXT/DCS/MAGNETS/SENSORDATA')
        objs=dcsfolder.findObjects(sortime,cool.ChannelSelection.all())
        data=[-1.,-1.,-1.,-1.]
        for obj in objs:
            chan=obj.channelId()
            channame=dcsfolder.channelName(chan)
            if channame in currentmap.keys():
                data[currentmap[channame]]=obj.payload()['value']
    except Exception as e:
        print ("MagFieldUtils.getFieldForRun ERROR accessing /EXT/DCS/MAGNETS/SENSORDATA")
        print (e)
    dcsDB.closeDatabase()
    # if problem accessing folder, exit
    if data is None:
        return None
    # return a MagFIeldDCSInfo object containing the result
    retval=MagFieldDCSInfo(data[0],data[1],data[2],data[3],fnt)
    _fieldForLB[runiov]=retval
    return retval
 

getTimeForLB()

python.MagFieldUtils.getTimeForLB	(		run,
			LB )

Definition at line 143 of file MagFieldUtils.py.

def getTimeForLB(run,LB):
    "Return the time a specific run/LB, given the folder, or 0 for bad/no data"
    if LB is None: 
        LB=0
 
    runiov=(run << 32)+LB
 
    if runiov in _timeForLB:
        print ("getTimeForLB: Returning cached time for run %i, LumiBlock %i " % (run,LB))
        return _timeForLB[runiov]
 
    if (run>=236107):
        dbname="CONDBR2"
    else:
        dbname="COMP200"
 
    #print ("Querying DB for time of run %i LB %i" % (run,LB))
 
    try:
        trigDB=indirectOpen('COOLONL_TRIGGER/%s' % dbname)
        if (trigDB is None):
            print ("MagFieldUtils.getTimeForLB ERROR: Cannot connect to COOLONL_TDAQ/%s" % dbname)
            return 0
 
        lblbfolder=trigDB.getFolder('/TRIGGER/LUMI/LBLB')
        obj=lblbfolder.findObject(runiov,0)
        payload=obj.payload()
        lbtime=payload['StartTime']
        _timeForLB[runiov]=lbtime
        trigDB.closeDatabase()
        return lbtime
    except Exception as e:
        print ("MagFieldUtils.getTimeForLB WARNING: accessing /TRIGGER/LUMI/LBLB for run %i, LB %i" % (run,LB))
        print (e)
        return 0
 
 
# command line driver for convenience

Variable Documentation

_fieldForLB

python.MagFieldUtils._fieldForLB = dict()

protected

Definition at line 14 of file MagFieldUtils.py.

_timeForLB

python.MagFieldUtils._timeForLB = dict()

protected

Definition at line 13 of file MagFieldUtils.py.

lumiblock

python.MagFieldUtils.lumiblock = None

Definition at line 187 of file MagFieldUtils.py.

magfield

python.MagFieldUtils.magfield = getFieldForRun(run,lumiblock=lumiblock)

Definition at line 190 of file MagFieldUtils.py.

run

python.MagFieldUtils.run = int(sys.argv[1])

Definition at line 186 of file MagFieldUtils.py.