brunelWrapper.py

Go to the documentation of this file.

#! /usr/bin/env python

"""
This script acts as a wrapper in order to generate Gaudi configs for running 
with different parameters in order to perform performance measurements.
It takes command line arguments.
"""

import optparse
"""
NumberOfEvents = 10
NumberOfEventsInFlight = 5
NumberOfAlgosInFlight = 5
NumberOfThreads = 10
CloneAlgos = False
DumpQueues = False
"""

def createParser():
    usage="%prog [options]"
    parser = optparse.OptionParser(usage)
    parser.add_option('-n', help='Number of events', dest='nevts', default=15)
    parser.add_option('--eif', help='Events in flight', dest='eif', default=5)    
    parser.add_option('--aif', help='Algos in flight', dest='aif', default=10)
    parser.add_option('--nthreads', help='Number of threads', dest='nthreads', default=10)
    parser.add_option('--clone', help='Clone Algos',dest='clone', action='store_true', default=False)
    parser.add_option('--dumpQueues', help='Dump Queues',dest='dumpqueues', action='store_true', default=False)
    parser.add_option('-v', help='Verbosity level', dest='verbosity', default=5)
    parser.add_option('--exec', help='ExecuteWorkflow',dest='execbrunel', action='store_true', default=False)
    parser.add_option('--bg', help='Launch in background',dest='bg', action='store_true', default=False)
    parser.add_option('--scale', help='Scale Algorithms time',dest='scale', default=1)    
    options,args = parser.parse_args()
    return options


def replaceValues(cfg_name,n,eif,aif,nthreads,scale,clone,dumpQueues,verbosity):

    newcfglines=open(cfg_name,"r").readlines()
    cfg_name=cfg_name.replace(".py","")    
    verb=""
    if verbosity!=6:
        verb="_v%s"%verbosity
    scale_s=""
    if scale != 1:
      scale_s="_s%s"%scale
    newcfgname="measurement_%s_n%s_eif%s_aif%s_nthreads%s_c%s_dq%s%s%s.py" %(cfg_name,n,eif,aif,nthreads,clone,dumpQueues,scale_s,verb)
    newcfg=open(newcfgname,"w")
    for line in newcfglines:
        if ('NumberOfEvents' in line and 'NUMBEROFEVENTS' in line and not "FLIGHT" in line):
            line=line.replace("NUMBEROFEVENTS",str(n))
        if ('NumberOfEventsInFlight' in line and 'NUMBEROFEVENTSINFLIGHT' in line):
            line=line.replace("NUMBEROFEVENTSINFLIGHT",str(eif))
        if ('NumberOfAlgosInFlight' in line and 'NUMBEROFALGOSINFLIGHT' in line):
            line=line.replace("NUMBEROFALGOSINFLIGHT",str(aif))
        if ('NumberOfThreads' in line and 'NUMBEROFTHREADS' in line):
            line=line.replace("NUMBEROFTHREADS",str(nthreads))
        if ('DumpQueues'in line and 'DUMPQUEUES' in line):
            line=line.replace("DUMPQUEUES",str(dumpQueues))
        if ('CloneAlgos'in line and 'CLONEALGOS' in line):
            line=line.replace("CLONEALGOS",str(clone))
        if ('Verbosity'in line and 'VERBOSITY' in line):
            line=line.replace("VERBOSITY",str(verbosity))
        if ('Scale'in line and 'SCALE' in line):
            line=line.replace("SCALE",str(scale))
        newcfg.write(line)
    newcfg.close()
    return newcfgname


if __name__ == "__main__":
    options = createParser()
    newcfg = replaceValues("BrunelScenario.py",
                  options.nevts,
                  options.eif,
                  options.aif,
                  options.nthreads,
                  float(options.scale),
                  options.clone,
                  options.dumpqueues,
                  options.verbosity)
    
    logfile = newcfg.replace(".py",".log")
    gaudirun="`alias gaudirun`"
    gaudirun = "/afs/cern.ch/user/d/dpiparo/Gaudi/build.x86_64-slc5-gcc46-opt/run gaudirun.py"
    command = "/usr/bin/time -f %%S -o timing_%s %s %s >& %s  " %(logfile, gaudirun, newcfg,logfile)
    if options.bg:
      command+=" &"

    print command
    if options.execbrunel :
      import os
      os.system(command)