In This Package:

OutputFrobicator.py

Go to the documentation of this file.

#!/usr/bin/env python
'''
Muck with RootIOCnvSvc's output file map, just to piss it off.
'''

from GaudiPython.GaudiAlgs import GaudiAlgo
from GaudiPython import SUCCESS, FAILURE, Bindings
import os

class OutFrob(GaudiAlgo):
    '''
    An algorithm that will change the RootIOCnvSvc's output file map.
    '''
    def __init__(self,name="OutFrob"):
        GaudiAlgo.__init__(self,name)
        self.count = 0
        return

    def initialize(self):
        sc = GaudiAlgo.initialize(self)
        if sc.isFailure(): 
            print 'Failed to initialize GaudiAlgo base class'
            return sc
        self.rio = self.svc('IRootIOSvc','RootIOCnvSvc')
        if not self.rio:
            print 'Failed to get RootIOCnvSvc, bogus!'
            return FAILURE
        self.riop = Bindings.iProperty('RootIOCnvSvc',self.rio)
        return SUCCESS

    def execute(self):
        self.count += 1 
        print self.count

        outstreams = self.riop.OutputStreams
        for k,v in outstreams.iteritems():
            print '\t%s --> %s'%(k,v)
            basenum,ext = os.path.splitext(v)
            base,num = basenum.split("_")
            newname = '%s_%03d%s'%(base,self.count,ext)
            print 'Changing output stream "%s" output file from "%s" to "%s"'%(k,v,newname)
            self.rio.associateOutput(newname,k)
            continue

        #for k,v in self.riop.properties().iteritems():
        #    print '\t"%s" --> (%s)%s [%s]'%(k,v.ptype(),v.value(),v.documentation())

        return SUCCESS        


def configure(argv=None):
    from RootIOSvc.RootIOSvcConf import RootIOCnvSvc
    rio = RootIOCnvSvc()

    from Gaudi.Configuration import ApplicationMgr
    theApp = ApplicationMgr()
    theApp.ExtSvc.append(rio)

    return

def run(app):
    alg = OutFrob()
    app.addAlgorithm(alg)
    return

---