In This Package:

__init__.py

Go to the documentation of this file.

#!/usr/bin/env python

__all__ = ['RootIOSvcConf','Configure']

def wash_streams(stream):
    'Assure that all "root:" URLs have ?mkpath=1 appended'
    ret = {}
    def appender(fname,end='?mkpath=1',scheme='root:'):
        'Append to fname the given "end" string if it is URL of given scheme and lacks it. Return fname'
        if scheme != fname[:len(scheme)]: return fname
        if end == fname[-len(end):]: return fname
        return fname + '?mkpath=1'
                           
    for name,flist in stream.iteritems():
        ret[name] = map(appender,flist)
        continue
    return ret


class Configure:
    'Configure RootIOSvc'

    def __init__(self,input_streams,output_streams,force_readout):
        '''Do default configuration of RootIOSvc.  The given input and
        output streams are maps between locations in the TES and
        files.  The special location "default" is used when locations
        are not explicitly listed.'''

        from Gaudi.Configuration import ApplicationMgr
        theApp = ApplicationMgr()

        theApp.EvtSel = "NONE"

        if not (input_streams or output_streams): 
            print "Note: No file I/O requested."
            return

        if input_streams:
            theApp.EvtSel = ""
            from DybIO.DybIOConf import DybEvtSelector
            theApp.ExtSvc += [ DybEvtSelector("EventSelector") ]

        if output_streams:              
            from DybIO.DybIOConf import DybStorageSvc
            theApp.ExtSvc += [ DybStorageSvc() ]

        from RootIOSvc.RootIOSvcConf import RootIOCnvSvc
        rio = RootIOCnvSvc()
        theApp.ExtSvc.append(rio)
        from GaudiSvc.GaudiSvcConf import EvtPersistencySvc
        per = EvtPersistencySvc()
        per.CnvServices = [ rio ];

        print 'input_streams=',str(input_streams)
        print 'output_streams=',str(output_streams)

        rio.InputStreams = wash_streams(input_streams)
        rio.OutputStreams = output_streams
        rio.ForceReadout = force_readout

        return


    pass # End Configure

---