CoincTag.py

Go to the documentation of this file.

#!/usr/bin/env python
'''
'''

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

dTCut = 1 * units.ms

class CoincTag(UserTaggingAlg):
    '''Coincidence Events'''
        
    # ===========================================    
    def initTagList(self):
        
        self.addTag('Coinc' , '/Event/UserTag/Coinc/All')
        self.addTag('Prompt' , '/Event/UserTag/Coinc/Prompt')
        self.addTag('Delayed', '/Event/UserTag/Coinc/Delayed')
        
        self.dTCut = dTCut
        self.info("dT set to %.3f microsecond" % (self.dTCut/units.microsecond,))
        
        self.lastTimeStamp = 0
        # initialize a list of input headers to hold the rec headers of multiplets
        self.inputHeaders = []
        
    # ===========================================
    def check(self, evt):
        recHdr = evt["/Event/Rec/AdSimple"]
        if not recHdr:
            self.warning('cannot find recHdr')
            return FAILURE

        timeStamp = recHdr.timeStamp().GetSeconds() * units.second
        # self.info("executing #%d, TS=%.6f msec" % (self.execNumber, timeStamp))
        
        if (timeStamp - self.lastTimeStamp) > self.dTCut:
            if len(self.inputHeaders) == 0:
                self.inputHeaders += [ recHdr ]
            elif len(self.inputHeaders) == 1:
                self.inputHeaders = [ recHdr ]
            else:
                # write a tag header object
                
                # write all events
                self.info( '#%d: found %d coinc events' % 
                    (self.execNumber, len(self.inputHeaders)) )
                self.getTag('Coinc').setInputHeaders(self.inputHeaders)
                
                # write prompt & delayed events
                promptHeader  = self.getTag('Prompt').header
                delayedHeader = self.getTag('Delayed').header
                
                for i, hdrI in enumerate(self.inputHeaders[:-1]):
                    for j, hdrJ in enumerate(self.inputHeaders[i+1:]):
                        promptHeader.addInputHeader(hdrI)
                        delayedHeader.addInputHeader(hdrJ)
                     
                # clear out the inputHeaders
                self.inputHeaders = [ recHdr ]
                self.lastTimeStamp = timeStamp
                
                self.tagIt('Coinc')
                self.tagIt('Prompt')
                self.tagIt('Delayed')
                return
        else:
            self.inputHeaders += [ recHdr ]
            
        self.lastTimeStamp = timeStamp
        
#####  Job Configuration for nuwa.py ########################################

def configure(argv=[]):
    """Configuration with command line arguments"""   
    global dTCut 
    from DybPython.Tools import unitify
    if len(argv) < 1:
        dTCut = unitify('1*ms') # default
        return
    try:
        dTCut = unitify(argv[0], 'second')
    except:
        import sys
        sys.exit('dTCut: ' + argv[0] + ': unit is not available, or wrong syntax.')

def run(app):
    myAlg = CoincTag("UserTagg::CoincTag")
    app.addAlgorithm(myAlg)
    pass