ADRecData.py

Go to the documentation of this file.

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

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

class ADRecData(UserTaggingAlg):
    '''Add AD Reconstruction Info to UserData'''
        
    # ----------------------------------------    
    def initTagList(self):   
        self.addTag('Dummy', ''
            ).addData('AdSimple', '/Event/UserData/AD/ADRecData/AdSimple'
            ).addInt('t_sec', 't_nsec', 'triggerType', 'detID'
            ).addFloat('energy', 'x', 'y', 'z')

    # ----------------------------------------
    def check(self, evt):
        ADTag = evt['/Event/UserTag/Detector/AD']
        if not ADTag:
            return False
        
        recHdr = evt['/Event/Rec/AdSimple']
        if not recHdr:
            self.warning('cannot find recHdr')
            return
        self.SaveRecData(recHdr)            

        self.tagIt('Dummy')

    # ----------------------------------------
    def SaveRecData(self, recHdr):
        '''Save Recon Info to UserData'''

        result = recHdr.recTrigger()
        self.getTag('Dummy').getData('AdSimple'
            ).set("energy", result.energy()/units.MeV
            ).set("x", result.position().x()/units.mm
            ).set("y", result.position().y()/units.mm
            ).set("z", result.position().z()/units.mm
            ).set("t_sec", recHdr.timeStamp().GetSec()
            ).set("t_nsec", recHdr.timeStamp().GetNanoSec()
            ).set("triggerType", result.triggerType()
            ).set("detID", result.detector().detectorId())
               
#####  Job Configuration for nuwa.py ########################################

def configure(argv=[]):
    pass

def run(app):
    myAlg = ADRecData("UserData::ADRecData")
    app.addAlgorithm(myAlg)
    pass