In This Package:

TestUserData.py

Go to the documentation of this file.

#!/usr/bin/env python
'''
Make some user data


CAUTION: don't use this as an example for your own stuff.  It does
things at a low level.  Better to look at the
RootIOTest.TestUserTagging module or even better at the
UserData/Tagging documentation.

'''

from GaudiPython.GaudiAlgs import GaudiAlgo
from GaudiPython import SUCCESS, FAILURE, Bindings, gbl
import ROOT

def spew(udh):
    s = '#%d @ %s count=%d aFloat=%f anInt=%d'%(\
        udh.execNumber(), udh.timeStamp().AsString(),
        udh.getInt("count"), udh.getFloat("aFloat"),udh.getInt("anInt"))
    return s


class SomeUserData(GaudiAlgo):
    '''
    Place a UserDataHeader into the data stream
    '''
    def __init__(self,name='SomeUserData'):
        GaudiAlgo.__init__(self,name)
        self.count = 0;
        self.cache = None
        print 'SomeUserData: making some'
        return

    def execute(self):
        tes = self.evtSvc()

        self.count += 1

        print 'making UserDataHeader'
        udh = gbl.DayaBay.UserDataHeader()
        ROOT.SetOwnership(udh,False)

        print 'saving to TES'
        tes['/Event/UserData/SomeUserData'] = udh

        print 'setting basic values'
        udh.setExecNumber(self.count)
        now = gbl.TimeStamp(self.count*3600,self.count)
        udh.setEarliest(now)
        now = gbl.TimeStamp(self.count*3600,self.count+1)
        udh.setTimeStamp(now)
        now = gbl.TimeStamp(self.count*3600,self.count+2)
        udh.setLatest(now)

        print 'setting user data'
        udh.setInt("count",self.count)
        udh.setFloat("aFloat",6.9*self.count)
        udh.setInt("anInt",self.count*self.count)

        print 'SomeData: %s'%spew(udh)

        print 'Deleting Python-side UserDataHeader'
        del (udh)

        print 'bye bye'
        return SUCCESS

class SomeMoreUserData(GaudiAlgo):
    '''
    Read some user data
    '''
    def __init__(self,name='SomeMoreUserData'):
        GaudiAlgo.__init__(self,name)
        print 'SomeMoreUserData: readin some more'
        return

    def execute(self):
        tes = self.evtSvc()

        udh = tes['/Event/UserData/SomeUserData']

        print 'SomeMoreUserData: %s'%spew(udh)

        return SUCCESS
    pass


do_write = True
do_read = True
def configure(argv=None):
    global do_write
    global do_read

    if argv and argv[0] == "read":
        do_write = False
    if argv and argv[0] == "write":
        do_read = False
        

def run(app):
    global do_write
    global do_read

    print 'TestUserData: configured with write="%s" and read="%s"'%(do_write,do_read)

    if do_write:
        alg = SomeUserData()
        app.addAlgorithm(alg)
    if do_read:
        alg = SomeMoreUserData()
        app.addAlgorithm(alg)

    return
    

---