CalibStats.py

Go to the documentation of this file.

#!/usr/bin/env python
# zhang@caltech.edu, 2011-05-05
'''
UserData of a collection of calibration statistics
Usage:
    nuwa.py --no-history -A "1*s" -n -1 -m"QuickStart.Calibrate" -m"Tagger.CalibStats" input.root
'''

from UserTagging.UserTaggingAlg import UserTaggingAlg
from Tagger.DataHelpers import CalibDataHelper

from GaudiPython import SUCCESS, FAILURE
from GaudiPython import gbl

# ===========================================
class CalibStats(UserTaggingAlg):
    'UserData of a collection of calibration statistics'

    def __init__(self, name):
        UserTaggingAlg.__init__(self, name)
        self.location = '/Event/CalibReadout/CalibReadoutHeader'
        
    # -------------------------------------------
    def initTagList(self):
        self.pmtGeomSvc = self.svc('IPmtGeomInfoSvc','PmtGeomInfoSvc')
        if self.pmtGeomSvc == None:
            self.error("Failed to get PmtGeomInfoSvc")
            return FAILURE
        
        self.myTag = self.addTag('CalibStats', '').setBaseHeader(self.location)
        self.myData = self.myTag.addData('CalibStats' , '/Event/Data/CalibStats')
        for paraName in CalibDataHelper.DATA_STATS_INT:
            self.myData.addInt(paraName)
        for paraName in CalibDataHelper.DATA_STATS_FLOAT:
            self.myData.addFloat(paraName)
        for paraName in CalibDataHelper.DATA_DTLAST_FLOAT:
            self.myData.addFloat(paraName)  
        for paraName in CalibDataHelper.DATA_DTNEXT_FLOAT:
            self.myData.addFloat(paraName) 
                                  
    # -------------------------------------------
    def check(self, evt):
        calibHdr = evt[self.location]
        if not calibHdr:
            self.warning('cannot find %s' % self.location)
            return FAILURE
        data = CalibDataHelper.stats(calibHdr)
        for paraName in CalibDataHelper.DATA_STATS_INT:
            self.myData.set(paraName, data[paraName])
        for paraName in CalibDataHelper.DATA_STATS_FLOAT:
            self.myData.set(paraName, data[paraName])        
                
        # shall we add CalibReadoutHeader to the inputHeaders of UserData?
        # self.myData.addInputHeader(calibHdr)    
        # trigger the saving process
        self.myTag.tagIt()
        
        # save dt to last detector info
        calibAES = self.getAES(self.location)
        if not calibAES:
            self.warning("AES not found in %s" % self.location)
            return
        data = CalibDataHelper.dtLast(calibAES)
        for paraName in CalibDataHelper.DATA_DTLAST_FLOAT:
            self.myData.set(paraName, data[paraName])
        for paraName in CalibDataHelper.DATA_DTNEXT_FLOAT:
            self.myData.set(paraName, -1)
    
    # -------------------------------------------
    def post_save(self, evt):        
        # save dt to next detector info
        calibStatsAES = self.getAES('/Event/Data/CalibStats')
        if not calibStatsAES:
            self.warning("AES not found at /Event/Data/CalibStats")
            return
        
        header = calibStatsAES[0]
        detId = header.context().GetDetId()
        paraName = 'dtNext' + CalibDataHelper.DET_NAMES.get(detId,'') + '_ms'
        timeStamp = header.timeStamp()
        # print '#', self.execNumber, paraName, timeStamp.GetSec(), timeStamp.GetNanoSec()
        for i in range(1, len(calibStatsAES)):
            aes_header = calibStatsAES[i]
            aes_detId = aes_header.context().GetDetId()
            if aes_detId == detId:
                dt = gbl.TimeStamp(timeStamp)
                dt.Subtract(aes_header.timeStamp())
                aes_header.setFloat(paraName, dt.GetSeconds()*1e3)
                # print 'break at', i
                break  # subsequent aes headers already set this dtNext
            else:
                if aes_header.getFloat(paraName) < 0:
                    dt = gbl.TimeStamp(timeStamp)
                    dt.Subtract(aes_header.timeStamp())
                    aes_header.setFloat(paraName, dt.GetSeconds()*1e3)
        
        # # debug
        # print '========='
        # for i in range(0, len(calibStatsAES)):   
        #     aes_header = calibStatsAES[i]
        #     print 'aes', i,  CalibDataHelper.DET_NAMES.get(aes_header.context().GetDetId(),'unknown')
        #     for paraName in CalibDataHelper.DATA_DTNEXT_FLOAT:  
        #         print paraName, aes_header.getFloat(paraName)
        # print
        
#####  Job Configuration for nuwa.py ########################################

def configure(argv=[]):
    # Setup Service to get PMT position in detectors
    from DetHelpers.DetHelpersConf import PmtGeomInfoSvc
    pgiSvc = PmtGeomInfoSvc()
    pgiSvc.StreamItems = ["/dd/Structure/DayaBay"]   
    pass

def run(app):    
    app.ExtSvc += ["PmtGeomInfoSvc"]
    myAlg = CalibStats("CalibStats")
    app.addAlgorithm(myAlg)