In This Package:

SimData.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
import PyCintex


class SimData(UserTaggingAlg):
    '''Add MC Truth Info to UserData'''
        
    # ----------------------------------------    
    def initTagList(self):
        
        # nSim should be one except for pile-up events        
        # Truth information, size of nSim
        self.addTag('Dummy', ''
            ).addData('SimData', '/Event/UserData/General/SimData'
            ).addInt('nSim'
            ).addIntArray('capTargetZ'
            ).addFloatArray('tCap', 'xCap', 'yCap', 'zCap'
            ).addFloatArray('QEInGdLS', 'QEInLS', 'QEInIAV', 'QEInOAV', 'QEInOIL', 'QESum')
    
        self.target_de_name = '/dd/Structure/AD/db-ade1/db-sst1/db-oil1'
        
    # ----------------------------------------
    def check(self, evt):

        readoutHdr = evt['/Event/Readout/ReadoutHeader']
        if not readoutHdr:
            self.warning('cannot find readoutHdr')
            return
        
        simHdrs = readoutHdr.findHeaders(gbl.DayaBay.SimHeader.classID())
        self.SaveSimData(simHdrs)
        
        self.tagIt('Dummy')
        
    # ----------------------------------------
    def SaveSimData(self, simHdrs):

        myData = self.getTag('Dummy').getData('SimData')
        
        de = self.getDet(self.target_de_name)
        if not de:
            self.info('Failed to get DE' + self.target_de_name)
            return FAILURE
        Gaudi = PyCintex.makeNamespace('Gaudi')
                        
        nSim = len(simHdrs) 
                
        for simHdr in simHdrs:
            uo_hdr = simHdr.unobservableStatistics()
            if not uo_hdr:
                self.info("no unobservableStatistics header in sim header")
                continue
            uo_stats = uo_hdr.stats()
            if not uo_stats:
                self.info("no unobservableStatistics in sim header")
                continue
            QEInGdLS = uo_stats["QEDepInGdLS"].sum()/units.MeV
            QEInLS   = uo_stats["QEDepInLS"].sum()/units.MeV
            QEInIAV  = uo_stats["QEDepInIAV"].sum()/units.MeV
            QEInOAV  = uo_stats["QEDepInOAV"].sum()/units.MeV
            QEInOIL  = uo_stats["QEDepInOIL"].sum()/units.MeV
            QESum = QEInGdLS+QEInLS+QEInIAV+QEInOAV+QEInOIL
            myData.append('QEInGdLS', QEInGdLS
                ).append('QEInLS'  , QEInLS
                ).append('QEInIAV' , QEInIAV
                ).append('QEInOAV' , QEInOAV
                ).append('QEInOIL' , QEInOIL
                ).append('QESum'   , QESum
                
                ).append('capTargetZ', int(uo_stats["capTargetZ"].sum())
                ).append('tCap', uo_stats["tCap"].sum()/units.microsecond
                ).append('xCap', uo_stats["xCap"].sum()/units.mm
                ).append('yCap', uo_stats["yCap"].sum()/units.mm
                ).append('zCap', uo_stats["zCap"].sum()/units.mm)
            
        myData.set("nSim", nSim)        


#####  Job Configuration for nuwa.py ########################################

def configure(argv=[]):
    pass

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

---