SpallB12.py

Go to the documentation of this file.

#!/usr/bin/env python
# chao@bnl.gov, 2011-08-07
'''
Spallation B-12 Tagger
Usage:
    nuwa.py --no-history -A "1*s" -n -1 -m"Tagger.MuonTagger.SpallB12" input.root
'''

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

from GaudiPython import SUCCESS, FAILURE
from GaudiPython import gbl
import GaudiKernel.SystemOfUnits as units

ServiceMode = gbl.ServiceMode

# ===========================================
class SpallB12Tagging(UserTaggingAlg):
    'Coincidence Data for AD coincidence events'

    def __init__(self, name):
        UserTaggingAlg.__init__(self, name)
        
        self.location = '/Events/Rec/AdSimple'
    # -------------------------------------------
    def initTagList(self):        
        self.daqDetailsSvc = self.svc('IDaqDetailsSvc','DetailsFromRecordSvc')
        
        if self.daqDetailsSvc == None: 
            self.error("Failed to get IDaqDetailsSvc") 
            return FAILURE 
        
        self.myTag = self.addTag('SpallB12', '/Event/Tag/SpallB12'
            ).setBaseHeader(self.location)
            
        self.myData = self.myTag.addData('SpallB12', '/Event/Data/SpallB12')
        self.myData.addInt('runNo', 'fileNo', 'nMuon'
            ).addFloatArray('dtLastMuon_ms', 'muon_nPESum')
        for paraName in RecDataHelper.DATA_SUMMARY_INT:
            self.myData.addInt(paraName)
        for paraName in RecDataHelper.DATA_SUMMARY_FLOAT:
            self.myData.addFloat(paraName)

        # add CalibStats info
        for paraName in CalibDataHelper.DATA_STATS_INT:
            self.myData.addInt('calib_'+paraName)
        for paraName in CalibDataHelper.DATA_STATS_FLOAT:
            self.myData.addFloat('calib_'+paraName)
                   
    # -------------------------------------------
    def check(self, evt):     
        recHdr = evt[self.location]
        if not recHdr:
            self.warning('cannot find header at ' + self.location)
            return FAILURE

        detId = recHdr.context().GetDetId()
        if not (detId == 1 or detId == 2):
            return False
        
        data = RecDataHelper.summary(recHdr)                  
        if not (4 < data['e'] < 20): 
            return False

        # self.info('energy ' + str(data['e']) + ', det ' + str(detId))

        timeStamp = recHdr.timeStamp()
        dtLastMuon_ms = []
        muon_nPESum = []
        aes = self.getAES('/Event/Data/CalibStats')
        for i in range(1, len(aes)):
            aes_header = aes[i]
            dt = gbl.TimeStamp(timeStamp)
            dt.Subtract(aes_header.timeStamp())
            
            # only save last 500 ms 
            if dt.GetSeconds()>0.5:
                break  
            
            # veto 1 ms after an AD muon
            aes_detId = aes_header.context().GetDetId()    
            aes_nPESum = aes_header.getFloat('nPESum')
            if dt.GetSeconds()<1e-3 and aes_nPESum>3.5e4 and aes_detId == detId:
                break
            
            # muon need to be on same AD
            if not (aes_detId == detId):
                # self.info('skip aes: aesdet:%d det:%d' % (aes_detId, detId))
                continue
                        
            # deposit muon energy in AD need to be large
            if (aes_nPESum < 3e5):
                # self.info('skip: npe' + str(aes_nPESum))
                continue
                
            # found a related muon!
            dtLastMuon_ms.append(dt.GetSeconds()*1e3)
            muon_nPESum.append(aes_nPESum)

        
        nMuon = len(dtLastMuon_ms)
        if nMuon == 0: 
            # no related muon, skip tagging
            return False
        
        self.myTag.tagIt()
        
        # save User Data
        self.myData.set("nMuon", nMuon)
        for i in range(nMuon):
            self.myData.append("dtLastMuon_ms", dtLastMuon_ms[i])
            self.myData.append("muon_nPESum", muon_nPESum[i])

        for paraName in RecDataHelper.DATA_SUMMARY_INT + RecDataHelper.DATA_SUMMARY_FLOAT:
            self.myData.set(paraName, data[paraName])
        
        calibStats = evt['/Event/Data/CalibStats']
        if calibStats:
            for paraName in CalibDataHelper.DATA_STATS_INT:
                self.myData.set('calib_'+paraName, calibStats.getInt(paraName))
            for paraName in CalibDataHelper.DATA_STATS_FLOAT:
                self.myData.set('calib_'+paraName, calibStats.getFloat(paraName))
        else:
            self.warning("no calibStats at /Event/Data/CalibStats")

        svcMode = ServiceMode(recHdr.context(), 0)
        runDetails = self.daqDetailsSvc.runDetails(svcMode)
        if runDetails:
            runNo = runDetails.runNumber()
        else:
            runNo = -1
        fileDetails = self.daqDetailsSvc.fileDetails(svcMode)
        if fileDetails:
            fileNo = fileDetails.fileNumber()
        else:
            fileNo = -1
        self.myData.set("runNo", runNo)
        self.myData.set("fileNo", fileNo)
        
        
        
#####  Job Configuration for nuwa.py ########################################
options = None

def configure(argv=[]):
    """Configuration with command line arguments"""   
    global options 


def run(app):
    app.ExtSvc += ["DetailsFromRecordSvc"]
    
    myAlg = SpallB12Tagging("SpallB12Tagging")
    app.addAlgorithm(myAlg)