In This Package:

TestLoader.py

Go to the documentation of this file.

#!/usr/bin/env python
'''
Test the DybInputLoadTool and DybEntryPolicyTool

nuwa.py -n 10 -o output.root -m 'TestLoader [opts] input.root ...'
nuwa.py -n 10 -o output.root -m "TestLoader [-t test_type] -I {'input1.root':1,'dd.root':'default','input2.root':2}"

'''

from DybPython.Tools import mapify

def configure(argv=None):
    if argv is None: argv = list()

    # Use nuwa.py style -I option
    from optparse import OptionParser
    parser = OptionParser(usage=__doc__)
    parser.add_option("-I", "--input-streams",default="{}",
                      help="Input file map")
    parser.add_option("-t","--test-type",type="int",default=0,
                      help="Test type [0-4]");

    opts,args = parser.parse_args(args=argv)
    
    # Turn dict string-rep into dict
    print opts.input_streams
    opts.input_streams = eval(opts.input_streams)
    print opts.input_streams

    # Regularize default streams, args are assumed to be .root files
    try:
        default = opts.input_streams['default']
    except KeyError:
        default = []
        default += args
    opts.input_streams['default'] = default

    from Gaudi.Configuration import ApplicationMgr
    theApp = ApplicationMgr()

    # Do a minimal version of RootIOSvc.Configure()
    from RootIOSvc.RootIOSvcConf import RootIOCnvSvc
    rio = RootIOCnvSvc()
    theApp.ExtSvc.append(rio)

    from RootIOSvc import wash_streams
    print opts.input_streams
    rio.InputStreams = wash_streams(opts.input_streams)

    from DybIO.DybIOConf import TestInputLoadAlg, DybInputLoadTool, DybNextEntryPolicyTool
    alg = TestInputLoadAlg()

    loader = DybInputLoadTool()
    loader.RootIOSvc = rio
    loader.ConversionSvc = rio
    alg.Loader = loader

    policy = DybNextEntryPolicyTool()
    policy.Start = 0
    policy.Mode = "sequential"
    alg.Policy = policy

    alg.Path = '/Event/Sim/SimHeader'
    alg.TestType = int(opts.test_type)

    theApp.TopAlg.append(alg);
    return

---