ADSingleData.py

Go to the documentation of this file.

#!/usr/bin/env python
# xiadm@ihep.ac.cn, 2011-07-03
'''
Single Data for AD single events
Usage:
    nuwa.py --no-history -A "1*s" -n -1 -m"Tagger.SingleTagger.ADSingleData" input.root
Help:
    nuwa.py -m"Tagger.SingleTagger.ADSingleData -h"
'''

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

# ===========================================
class ADSingleData(UserTaggingAlg):
    'single Data for AD single events'

    def __init__(self, name):
        UserTaggingAlg.__init__(self, name)
        
        self.tag_data_locations = {
            '/Event/Tag/Single/AD1SingleTag' : '/Event/Data/SingleData/AD1SingleData',
            '/Event/Tag/Single/AD2SingleTag' : '/Event/Data/SingleData/AD2SingleData',
        }
        self.location = '/Event/Rec/AdSimple'
        self.isSimulation = False

    # -------------------------------------------
    def initTagList(self):        
        self.coordSvc = self.svc('ICoordSysSvc','CoordSysSvc')
        if self.coordSvc == None:
            self.error("Failed to get ICoordSysSvc")
            return FAILURE
        
        # create ADSingleData for each predefined tag
        for tagName, dataName in self.tag_data_locations.items():
         #   make it simple by using path as name as well
            myTag = self.addTag(tagName, '').setBaseHeader(self.location)
            myData = myTag.addData(dataName , dataName)
            myData.addInt('multiplicity'
               ).addIntArray('dt_ns', 'I', 'J')
            
            # add RecHeader info
            for paraName in RecDataHelper.DATA_SUMMARY_INT:
                myData.addIntArray(paraName)
            for paraName in RecDataHelper.DATA_SUMMARY_FLOAT:
                myData.addFloatArray(paraName)
            '''
            # add CalibStats info
            for paraName in CalibDataHelper.DATA_STATS_INT:
                myData.addIntArray('calib_'+paraName)
            for paraName in CalibDataHelper.DATA_STATS_FLOAT:
                myData.addFloatArray('calib_'+paraName)
            for paraName in CalibDataHelper.DATA_DTLAST_FLOAT:
                myData.addFloatArray(paraName) 
            
            if self.isSimulation:
                # add GenHeader info
                for paraName in GenDataHelper.DATA_SUMMARY_INT:
                    myData.addIntArray(paraName)
                for paraName in GenDataHelper.DATA_SUMMARY_FLOAT:
                    myData.addFloatArray(paraName)

                # add SimHeader Info
                for paraName in SimDataHelper.DATA_SUMMARY_INT:
                    myData.addIntArray(paraName)
                for paraName in SimDataHelper.DATA_NCAP_INT:
                    myData.addIntArray(paraName)
                for paraName in SimDataHelper.DATA_NCAP_FLOAT:
                    myData.addFloatArray(paraName)
              '''       
    # -------------------------------------------
    def check(self, evt):                
        for tagName, dataName in self.tag_data_locations.items():
            tagHeader = evt[tagName]  # tagName == tagPath
            if not tagHeader:
                continue  # skip non-single events
            else:
                myData = self.getTag(tagName).getData(dataName)
                self.SaveADSingleData(myData, tagHeader)
                self.tagIt(tagName)
                break  # the AD single tags are exclusive
    
    # -------------------------------------------
    def SaveADSingleData(self, myData, tagHeader):
        '''
        Keyword arguments:
            myData -- the Data object defined in UserTagging.Models
            tagHeader -- the AD single UserTag object
        '''

        singleRecHdrs = tagHeader.inputHeaders()
        firstHdr = True
        for recHdr in singleRecHdrs:
            data = RecDataHelper.summary(recHdr)
            # save RecHeader info
            for paraName in RecDataHelper.DATA_SUMMARY_INT:
                myData.append(paraName, data[paraName])
            for paraName in RecDataHelper.DATA_SUMMARY_FLOAT:
                myData.append(paraName, data[paraName])
            if firstHdr:
                t0_s = data["t_s"]
                t0_ns = data["t_ns"]
                firstHdr = False
            myData.append('dt_ns', (data["t_s"]-t0_s)*1e9 + data["t_ns"] - t0_ns)
            '''   
            # save calib stats
            calibStatsHdrs = recHdr.findHeaders(gbl.DayaBay.UserDataHeader.classID())
            if calibStatsHdrs and len(calibStatsHdrs) == 1:
                calibStats = calibStatsHdrs[0]
                for paraName in CalibDataHelper.DATA_STATS_INT:
                    myData.append('calib_'+paraName, calibStats.getInt(paraName))
                for paraName in CalibDataHelper.DATA_STATS_FLOAT:
                    myData.append('calib_'+paraName, calibStats.getFloat(paraName))
                for paraName in CalibDataHelper.DATA_DTLAST_FLOAT:
                    myData.append(paraName, calibStats.getFloat(paraName))
            else:
                self.warning("no calibStats in recHeader's inputHeaders")

            
            # save some gen/sim info for simualtion files
            if self.isSimulation:
                # save GenHeader info
                genHdrs = recHdr.findHeaders(gbl.DayaBay.GenHeader.classID())
                data = GenDataHelper.summary(self, recHdr, genHdrs)
                for paraName in GenDataHelper.DATA_SUMMARY_INT:
                    myData.append(paraName, data[paraName])
                for paraName in GenDataHelper.DATA_SUMMARY_FLOAT:
                    myData.append(paraName, data[paraName])

                # save SimHeader info
                simHdrs = recHdr.findHeaders(gbl.DayaBay.SimHeader.classID())
                data = SimDataHelper.summary(simHdrs)
                for paraName in SimDataHelper.DATA_SUMMARY_INT:
                    myData.append(paraName, data[paraName])
                data = SimDataHelper.nCapture(self, recHdr)
                for paraName in SimDataHelper.DATA_NCAP_INT:
                    myData.append(paraName, data[paraName])
                for paraName in SimDataHelper.DATA_NCAP_FLOAT:
                    myData.append(paraName, data[paraName])
            '''          
        multiplicity = len(singleRecHdrs)
        myData.set("multiplicity", multiplicity)
        # add the pairing array trick to allow easy TTree::Draw() such as
        # tree.Draw("e[J]:e[I]")
        for i in range(0, multiplicity-1):
            for j in range(i+1, multiplicity):
                myData.append('I', i).append('J', j)
                
        
#####  Job Configuration for nuwa.py ########################################
options = None

def configure(argv=[]):
    """Configuration with command line arguments"""   
    global options 
    from optparse import OptionParser
    
    defaultLocations = ''
    for detname in ['AD1', 'AD2']:
        defaultLocations += '/Event/Tag/Single/%sSingleTag:/Event/Data/Single/%sSingleData ' % (detname, detname)

    parser = OptionParser()
    parser.add_option("-l", "--location", type="string",
        default='/Event/Rec/AdSimple',
        help="the TES path of target event header location")
    parser.add_option("-t", "--tag-data-locations", type="string",
        default=defaultLocations.rstrip(), 
        help="tagPath:dataPath pairs for AD single events")
    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 = ADSingleData("ADSingleData")
    myAlg.location = options.location
    myAlg.tag_data_locations = mapify(options.tag_data_locations)
    myAlg.isSimulation = options.isSimulation
    app.addAlgorithm(myAlg)