| 1 | # $Id: runSequence.py,v 1.4 2004/06/09 15:04:36 daquinog Exp $
|
|---|
| 2 | # -------------------------------------------------------------------
|
|---|
| 3 | # GEANT4 tag $Name: $
|
|---|
| 4 | # -------------------------------------------------------------------
|
|---|
| 5 | #
|
|---|
| 6 | import Tiara
|
|---|
| 7 | import myUtils
|
|---|
| 8 | import shelve
|
|---|
| 9 | import os
|
|---|
| 10 | import string
|
|---|
| 11 |
|
|---|
| 12 | class RunConfig(object):
|
|---|
| 13 | def __init__(self):
|
|---|
| 14 | self.basePath = ""
|
|---|
| 15 | self.tApp = ""
|
|---|
| 16 | self.tiaraSpecs = ""
|
|---|
| 17 | self.impGeo = ""
|
|---|
| 18 | self.impScorer = ""
|
|---|
| 19 | self.totalTime = ""
|
|---|
| 20 | self.comment = ""
|
|---|
| 21 | def getConfInfo(self):
|
|---|
| 22 | return myUtils.\
|
|---|
| 23 | getConfigurationInfo(self.impGeo,
|
|---|
| 24 | self.tiaraSpecs.experiment,
|
|---|
| 25 | self.tApp.physicsList,
|
|---|
| 26 | self.totalTime,
|
|---|
| 27 | self.comment)
|
|---|
| 28 |
|
|---|
| 29 |
|
|---|
| 30 |
|
|---|
| 31 |
|
|---|
| 32 | class RunSequence(object):
|
|---|
| 33 | # methods to be used public
|
|---|
| 34 | def __init__(self, runConfig, usePI = True):
|
|---|
| 35 | self.rc = runConfig
|
|---|
| 36 | self.usePI = usePI
|
|---|
| 37 | self.runNum = -1
|
|---|
| 38 | self.confInfo = self.rc.getConfInfo()
|
|---|
| 39 | self.storeName = myUtils.getStoreName()
|
|---|
| 40 | self.xmlStore = ""
|
|---|
| 41 | self.pathXMLName = ""
|
|---|
| 42 | self.shelveName = ""
|
|---|
| 43 | self.pathShelveName = ""
|
|---|
| 44 | self.randomNumberFileName = ""
|
|---|
| 45 | self.path = self.mkPath()
|
|---|
| 46 |
|
|---|
| 47 | def runNevents(self, events):
|
|---|
| 48 | self.runNum += 1
|
|---|
| 49 | self.mkNames()
|
|---|
| 50 | self.mkShelve()
|
|---|
| 51 | self.report()
|
|---|
| 52 | self.rc.tApp.tiaraSim.BeamOn (events)
|
|---|
| 53 | Tiara.saveRandomStatus(self.randomNumberFileName)
|
|---|
| 54 | myUtils.saveResults(self.rc.tApp,
|
|---|
| 55 | self.path,
|
|---|
| 56 | self.shelveName,
|
|---|
| 57 | self.rc.impScorer,
|
|---|
| 58 | self.rc.impGeo)
|
|---|
| 59 |
|
|---|
| 60 | def runLoop(self):
|
|---|
| 61 | while ( not \
|
|---|
| 62 | (self.rc.tApp.eventAction.\
|
|---|
| 63 | GetTotalProcessedTime() > self.rc.totalTime)):
|
|---|
| 64 | self.runNevents(10000000) # dummy num. events
|
|---|
| 65 |
|
|---|
| 66 |
|
|---|
| 67 |
|
|---|
| 68 |
|
|---|
| 69 |
|
|---|
| 70 |
|
|---|
| 71 |
|
|---|
| 72 | # methods used privately
|
|---|
| 73 |
|
|---|
| 74 | def report(self):
|
|---|
| 75 | print "\n\nRunSequence.report:"
|
|---|
| 76 | print self.confInfo
|
|---|
| 77 | if self.usePI:
|
|---|
| 78 | print "the xml store will be named: "
|
|---|
| 79 | print " ", self.pathXMLName
|
|---|
| 80 | print "the shelve name: "
|
|---|
| 81 | print " ", self.pathShelveName
|
|---|
| 82 | print "\n\n"
|
|---|
| 83 |
|
|---|
| 84 |
|
|---|
| 85 |
|
|---|
| 86 | def mkPath(self):
|
|---|
| 87 | if not os.path.exists(self.rc.basePath):
|
|---|
| 88 | os.mkdir(self.rc.basePath)
|
|---|
| 89 | path = self.rc.basePath + "/" + self.storeName
|
|---|
| 90 | if not os.path.exists(path):
|
|---|
| 91 | os.mkdir(path)
|
|---|
| 92 | return path
|
|---|
| 93 |
|
|---|
| 94 | def mkNames(self):
|
|---|
| 95 | rn = self.runNum
|
|---|
| 96 | rns = "%(rn)d" % vars()
|
|---|
| 97 | rns = string.rjust(rns, 5)
|
|---|
| 98 | rns = string.replace(rns,' ','0')
|
|---|
| 99 | rId = "_run" + rns
|
|---|
| 100 | if self.usePI:
|
|---|
| 101 | self.xmlStore = self.storeName + rId + ".xml"
|
|---|
| 102 | self.pathXMLName = self.path + "/" + self.xmlStore
|
|---|
| 103 |
|
|---|
| 104 | self.shelveName = self.storeName + rId + ".shelve"
|
|---|
| 105 | self.pathShelveName = self.path + "/" + self.shelveName
|
|---|
| 106 | self.randomNumberFileName = self.path + "/randomNumberFile" + rId
|
|---|
| 107 |
|
|---|
| 108 |
|
|---|
| 109 | def mkShelve(self):
|
|---|
| 110 | myShelve = shelve.open(self.pathShelveName)
|
|---|
| 111 | if self.usePI:
|
|---|
| 112 | myShelve["xmlStoreName"] = self.xmlStore
|
|---|
| 113 | for info in self.confInfo:
|
|---|
| 114 | myShelve[info] = self.confInfo[info]
|
|---|
| 115 | myShelve.close()
|
|---|
| 116 |
|
|---|
| 117 |
|
|---|
| 118 | # end of RunSeuence
|
|---|
| 119 |
|
|---|
