In This Package:

TestHeaderObjects.py

Go to the documentation of this file.

#!/usr/bin/env python
'''
Test header objects are correctly sent through I/O

usage: nuwa.py -m "RootIOTest.TestHeaderObjects [filebase]"

filebase.txt will be filled with a text representation of the header
objects

filebase.pkl will be filled with a Python Pickled representation of
the header objects

The latter can be compared by running this module directly like:

python TestHeaderObjects.py statefile1.pkl statefiel2.pkl

'''


from GaudiPython.GaudiAlgs import GaudiAlgo
from GaudiPython import SUCCESS, FAILURE, Bindings
import os, pickle
from DybPython.Util import irange


class PythonHeaderObject(object):
    '''
    A purely Python header object
    '''

    def __init__(self,ho):
        self.defloc = ho.defLoc()


        ctx = ho.context()
        self.site = ctx.GetSite()
        self.flag = ctx.GetSimFlag()
        self.detid = ctx.GetDetId()

        #print 'Getting timestamp'
        ts = ctx.GetTimeStamp()
        self.seconds = ts.GetSec()
        self.nanosec = ts.GetNanoSec()

        self.execNumber = ho.execNumber()
        self.randomState = []

        # print 'Getting random state'
        # rs = ho.randomState()
        # for rn in irange(rs.begin(), rs.end()):
        #     print rn
        #     self.randomState.append(rn)
        #     continue

        self.inputHeaders = []
        ihs = ho.inputHeaders()
        siz = ihs.size()
        #print 'Recursing on %d input headers'%siz
        for ind in range(siz):
            ih = ihs[ind]
            self.inputHeaders.append(PythonHeaderObject(ih))
            continue
        return

    def __cmp__(self,other):
        return cmp(self.__dict__,other.__dict__)

    def __str__(self):
        s = '%s exec=%d sec.ns=%d.%09d site=%d mcflag=%d detid=%d with %d inputHeaders:'%\
            (self.defloc,
             self.execNumber, self.seconds, self.nanosec,
             self.site, self.flag, self.detid, 
             len(self.inputHeaders))
        if self.inputHeaders:
            l = []
            for ih in self.inputHeaders:
                l.append('\t%s'%ih)
            s += '\n' + '\n'.join(l)

        return s
        

class HeaderDump(GaudiAlgo):
    '''
    A gaudi alg that dumps header objects into a pickle file
    '''
    def __init__(self,name='InputHeaderDump',filebase='input_header_dump',headers=None):
        GaudiAlgo.__init__(self,name)
        self.pkl = open(filebase+'.pkl',"w")
        self.txt = open(filebase+'.txt',"w")
        if headers:
            self.paths = headers
        else:
            self.paths = ['/Event/Sim/SimHeader','/Event/Gen/GenHeader']
        return

    def execute(self):
        evt = self.evtSvc()
        for path in self.paths:
            hdr = evt[path]
            if not hdr:
                print 'Failed to get HeaderObject from "%s"'%path
                return FAILURE
            self.txt.write("%s"%hdr)
            phdr = PythonHeaderObject(hdr)
            pickle.dump(phdr,self.pkl)
            continue
        return SUCCESS

    def finalize(self):
        self.pkl.close()
        self.txt.close()
        return SUCCESS
        

filebase = None

def configure(argv=None):
    global filebase
    try:
        filebase = argv[0]
    except IndexError:
        filebase = "test_input_headers.pkl"

def run(app):
    global filebase
    alg = HeaderDump(filebase=filebase)
    app.addAlgorithm(alg)
    return

def compare(file1,file2):

    print 'Comparing "%s" and "%s"'%(file1,file2)
    
    file1 = open(file1,"r")
    file2 = open(file2,"r")

    count = 0
    ndiffer = 0
    while True:
        count +=1 
        try:
            o1 = pickle.load(file1)
            o2 = pickle.load(file2)
        except EOFError:
            break

        if o1 == o2: 
            print 'Same:'
            print 'First: ',o1
            print 'Second:',o2
            continue

        ndiffer += 1

        print 'Objects #%d differ'%count
        print 'Object1:%s'%o1
        print 'Object2:%s'%o2
        continue
    print "Number of differing objects: %d"%ndiffer
    return ndiffer

if __name__ == '__main__':
    import sys
    compare(sys.argv[1],sys.argv[2])

---