AmCData.py

Go to the documentation of this file.

#!/usr/bin/env python
'''
AmC Data 
Usage:
    nuwa.py --no-history -A "0.1*s" -n -1 -m"Quickstart.EH1Tables" -m"Quickstart.Calibrate" -m"Quickstart.CalculateCalibStats"  -m"Quickstart.Reconstruct" -m"MuonTagging.MuonTagLoose" -m"CalibrationTagging.AmCTag" -m"CalibrationTagging.AmCData" -o output.root input.root
Help:
    nuwa.py -m"CalibrationTagging.AmCData -h"
    
Created By: tsang@caltech.edu 2011/08/22
Based on: ADCoincData.py by zhang@caltech.edu
'''

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

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

ServiceMode = gbl.ServiceMode

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

    def __init__(self, name):
        UserTaggingAlg.__init__(self, name)
        
        self.runStartTime = None
        self.isSimulation = False

    # -------------------------------------------
    def initTagList(self):        
        myTag = self.addTag('/Event/Tag/Calib/AmC', '').setBaseHeader('/Events/Rec/AdSimple')
        myData = myTag.addData('/Event/Tag/Calib/AmC' , '/Event/Data/Calib/AmC')
        myData.addFloat('ePrompt' , 'xPrompt' , 'yPrompt' , 'zPrompt' ,
                        'eDelayed', 'xDelayed', 'yDelayed', 'zDelayed', 'deltaT_us')
        
    # -------------------------------------------
    def check(self, evt):                

        tagHeader = evt['/Event/Tag/Calib/AmC']  # '/Event/Tag/Calib/AmC' == tagPath
        if not tagHeader:
          return
        
        myData = self.getTag('/Event/Tag/Calib/AmC').getData('/Event/Tag/Calib/AmC')
        self.SaveAmCData(myData, tagHeader)
        self.tagIt('/Event/Tag/Calib/AmC')
        return SUCCESS
    
    # -------------------------------------------
    def SaveAmCData(self, myData, tagHeader):
        '''
        Keyword arguments:
            myData -- the Data object defined in UserTagging.Models
            tagHeader -- the AD coincidence UserTag object
        '''

        savedRecHdrs = tagHeader.inputHeaders()
        promptRecHdr = savedRecHdrs[0]
        delayedRecHdr = savedRecHdrs[1]

        # Don't process SAB data
        #if(promptRecHdr.context().GetSite() == Site.kSAB):
        #    return

        if promptRecHdr == None:
          print 'promptRecHdr == None'
          return FAILURE
        if delayedRecHdr == None:
          print 'delayedRecHdr == None'
          return FAILURE

        promptRecResult = promptRecHdr.recTrigger()
        delayedRecResult = delayedRecHdr.recTrigger()

        # Position-related info
        pRecPosStat = promptRecResult.positionStatus()
        pRecX = promptRecResult.position().x()
        pRecY = promptRecResult.position().y()
        pRecZ = promptRecResult.position().z()
        myData.set('xPrompt',pRecX)
        myData.set('yPrompt',pRecY)
        myData.set('zPrompt',pRecZ)
        dRecPosStat = delayedRecResult.positionStatus()
        dRecX = delayedRecResult.position().x()
        dRecY = delayedRecResult.position().y()
        dRecZ = delayedRecResult.position().z()
        myData.set('xDelayed',dRecX)
        myData.set('yDelayed',dRecY)
        myData.set('zDelayed',dRecZ)

        # Energy-related info
        pRecEStat = promptRecResult.energyStatus()
        pRecE = promptRecResult.energy()
        pRecE *= 184./160.
        myData.set('ePrompt',pRecE)
        dRecEStat = delayedRecResult.energyStatus()
        dRecE = delayedRecResult.energy()
        dRecE *= 184./160.
        myData.set('eDelayed',dRecE)
    
        # Time-related info
        deltaT = gbl.TimeStamp( delayedRecResult.triggerTime() )
        deltaT.Subtract( promptRecResult.triggerTime() )
        myData.set('deltaT_us',deltaT.GetSeconds() * 1e6)

        # Calibstats
#        nPERMS = calibStats.get('nPERMS').value()
#        Co60Data.set('nPERMS_co60',nPERMS)
#        nPESum = calibStats.get('nPESum').value()
#        Co60Data.set('nPESum_co60',nPESum)
        
#####  Job Configuration for nuwa.py ########################################
options = None

def configure(argv=[]):
    """Configuration with command line arguments"""   
    global options 
    from optparse import OptionParser
    
    parser = OptionParser()
    parser.add_option('--simulation',
        dest='isSimulation',
        action='store_true', default=False,
        help='flag for simulation files (save extra gen/truth info)')
    (options, args) = parser.parse_args(args=argv)

def run(app):
    from DybPython.Tools import mapify
    #app.ExtSvc += ["CoordSysSvc"]
    
    myAlg = AmCData("AmCData")
    myAlg.isSimulation = options.isSimulation
    app.addAlgorithm(myAlg)