DQDumpPmt.py

Go to the documentation of this file.

#!/usr/bin/env python
# useage: nuwa.py --no-history -n -1 -m"RollingGain" -m"DQDump.DQDumpPmt" rawdata 

# Load DybPython
from DybPython.DybPythonAlg import DybPythonAlg
from GaudiPython import SUCCESS, FAILURE
from GaudiPython import gbl
import GaudiKernel.SystemOfUnits as units
DetectorSensor = gbl.DayaBay.DetectorSensor
ServiceMode = gbl.ServiceMode

from DybDbi import gDbi
from DybDbi import GDataQualityPmt as DQPmt
from DybDbi import ContextRange,Site,SimFlag,TimeStamp

import os,sys
import MySQLdb
os.environ['DBCONF'] = 'tmp_ligs_offline_db'

# Make your algorithm
class DumpPmtToDbAlg(DybPythonAlg):
    "Template Python Algorithm"
    def __init__(self,name):
        DybPythonAlg.__init__(self,name)
        self.runNo = 0
        self.fileNo = 0
        return

    def getSeqNo(self, runNo, fileNo):
        """Get file seqno by runno and fileno"""
        try:
            con = MySQLdb.connect( host = "dayabaydb.lbl.gov",
                                   user = "dayabay",
                                   passwd = "3quarks",
                                   db = "offline_db" )
            cursor = con.cursor()
            cursor.execute("SELECT seqno FROM DaqRawDataFileInfo WHERE runNo = "
                            + str(runNo) + " AND " + "fileNo = " + str(fileNo))
            row = cursor.fetchone()
            if row:
                print "get seqno successful"
        except MySQLdb.Error,e:
            print "Error %d: %s" % (e.args[0], e.args[1])
            sys.exit(1)
        cursor.close()
        con.close()
        return row[0]

    def getFileTimeInfo(self, seqNo):
        """Talk to offline DB and get the time range of the run"""
        try:
            con = MySQLdb.connect( host = "dayabaydb.lbl.gov",
                                   user = "dayabay",
                                   passwd = "3quarks",
                                   db = "offline_db" )
            cursor = con.cursor()
            cursor.execute("SELECT timestart,timeend FROM DaqRawDataFileInfoVld WHERE seqno = " + str(seqNo))
            row = cursor.fetchone()
            if row:
                print "get file time info successfully"
        except MySQLdb.Error,e:
            print "Error %d: %s" % (e.args[0], e.args[1])
            sys.exit(1)

        cursor.close()
        con.close()
        return row

    def dumptodb(self):
        seqNo = self.getSeqNo(self.runNo, self.fileNo)
        fileTime = self.getFileTimeInfo(seqNo)
        import calendar
        starttime = calendar.timegm( fileTime[0].timetuple() )
        endtime = calendar.timegm( fileTime[1].timetuple() )
        
        wrt = DQPmt.Wrt()
        cr = ContextRange( Site.kAll,
                           SimFlag.kData|SimFlag.kMC,
                           TimeStamp(starttime),
                           TimeStamp(endtime) )
        wrt.ctx(contextrange=cr, versiondate=TimeStamp(0,0))

        FILENAME='pmtDataTable_01.txt'
        f = open(FILENAME,'r')
        for line in f:
            if line[0] != '#':
                line = line.strip()
                para = line.split()
                status = int( para[2] )
                chi2ndf = float( para[5] )
                pmtID = para[0]
                pmtInfo = para[1]
                gain = float( para[6] )
                gainErr = float( para[7] )
                darkRate = float( para[15] )
                darkRateErr = float( para[16] )
                elecNoiseRate = float( para[17] )
                elecNoiseRateErr = float( para[18] )
                preAdc = float( para[19] )
                preAdcErr = float( para[20] )
                record = DQPmt.Create( RunNo = self.runNo,
                                       FileNo = self.fileNo,
                                       PmtId = DetectorSensor( int(pmtID) ),
                                       Describ = pmtInfo,
                                       Status = status,
                                       Chi2ndf = chi2ndf,
                                       Gain = gain,
                                       GainErr = gainErr,
                                       DarkRate = darkRate,
                                       DarkRateErr = darkRateErr,
                                       ElecNoiseRate = elecNoiseRate,
                                       ElecNoiseRateErr = elecNoiseRateErr,
                                       PreAdc = preAdc,
                                       PreAdcErr = preAdcErr )
                wrt.Write(record)
        wrt.Close()
        return

    def initialize(self):
        status = DybPythonAlg.initialize(self)
        if status.isFailure(): return status
        
        self.daqDetailsSvc = self.svc('IDaqDetailsSvc','DetailsFromRecordSvc')
        if self.daqDetailsSvc == None:
            self.error("Failed to get IDaqDetailsSvc")
            return FAILURE
        
        self.info("initializing")
        return SUCCESS

    def execute(self):
        if((self.runNo == 0) and (self.fileNo == 0) ):
            evt = self.evtSvc()
            readoutHdr = evt["/Event/Readout/ReadoutHeader"]
            if readoutHdr == None:
                self.error("Failed to get current readout header")
                return FAILURE
            svcMode = ServiceMode(readoutHdr.context(), 0)
            self.runNo = self.daqDetailsSvc.runDetails(svcMode).runNumber()
            self.fileNo = self.daqDetailsSvc.fileDetails(svcMode).fileNumber() 
        return SUCCESS
        
    def finalize(self):
        self.info("finalizing")
        self.dumptodb()
        os.system("mv pmtDataTable_01.txt pmtDataTable_01_finished.txt")
        status = DybPythonAlg.finalize(self)
        return status

#####  Job Configuration for nuwa.py ########################################
def configure():
    """ Template job module """
    return

def run(app):
    '''
    Configure and add the algorithm to job
    '''
    app.ExtSvc += ["DetailsFromRecordSvc"]
    # Add Python algorithm to job here
    myAlg = DumpPmtToDbAlg("MyDumpPmtAlg")
    # Add algorithm to job
    app.addAlgorithm(myAlg)
    pass