source: trunk/examples/advanced/Tiara/source/py_modules/myUtils.py@ 1283

Last change on this file since 1283 was 807, checked in by garnier, 17 years ago

update
File size: 7.7 KB
Line 
1# $Id: myUtils.py,v 1.6 2004/12/08 15:37:14 daquinog Exp $
2# -------------------------------------------------------------------
3# GEANT4 tag $Name: $
4# -------------------------------------------------------------------
5#
6import CLHEP
7import G4Kernel
8
9import string
10import os
11import time
12import shelve
13
14G4analysisUse = os.environ.has_key("G4ANALYSIS_USE")
15if G4analysisUse:
16 import myPI
17
18
19def createParallelSampler(impGeo, impScorer):
20 parallelSampler = G4Kernel. \
21 G4ParallelGeometrySampler(\
22 impGeo.getWorldVolume(), "neutron")
23 parallelSampler.PrepareScoring(impScorer)
24 if impGeo.base > 1:
25 istore = impGeo.getImportanceStore()
26 parallelSampler.PrepareImportanceSampling(istore,None)
27
28 parallelSampler.Configure()
29 return parallelSampler
30
31
32
33def printImpTable(impScorer, istore = ""):
34 iScMap = impScorer.GetMapGeometryCellCellScorer()
35 table = ""
36 if istore:
37 table = G4Kernel.G4ScoreTable(istore)
38 else:
39 table = G4Kernel.G4ScoreTable()
40 table.Print(iScMap)
41
42
43
44def saveResults(tApp, path, shelveName, impScorer, impGeo):
45 myShelve = shelve.open(path + "/" + shelveName)
46 printImpTable(impScorer, impGeo.getImportanceStore())
47 table = G4Kernel.G4ScoreTable()
48 table.Print(tApp.cellScorerStore.GetMapGeometryCellCellScorer())
49 if G4analysisUse:
50 storePath = path + "/" + myShelve["xmlStoreName"]
51 persistantStore = myPI.tf.create(storePath,"xml",0,1)
52 coppyTrees(tApp.tree, persistantStore)
53 persistantStore.commit()
54 persistantStore.close()
55 print "wrote to store: ", storePath
56
57 tApp.fillShelve(myShelve)
58 myShelve.close()
59 print "wrote shelve: ", shelveName
60
61
62def coppyTrees(srcTree, cpTree):
63 mntPoint = "cpTreeMountPoint"
64 srcTree.mkdir("/" + mntPoint)
65 srcTree.mount("/" + mntPoint, cpTree, "/")
66 objectNames = srcTree.listObjectNames()
67 objectTypes = srcTree.listObjectTypes()
68 for i in range(len(objectNames)):
69 name = objectNames[i]
70 type = objectTypes[i]
71 if string.find(name, mntPoint) < 0:
72 srcTree.cp (name,"/" + mntPoint +"/")
73
74min = 6000
75hour = 60 * min
76day = 24 * hour
77
78
79def getTotalTime(totalTime):
80 days = totalTime / day
81 dayRest = totalTime - days*day
82 hours = dayRest / hour
83 minrest = dayRest - hours*hour
84 minutes = minrest / min
85
86 tString = ""
87 if days > 0:
88 tString += "%(days)dd" % vars()
89 if hours > 0:
90 tString += "%(hours)dh" % vars()
91 if minutes > 0:
92 tString += "%(minutes)dm" % vars()
93
94 return tString
95
96
97def getStoreName():
98
99 Y, M, D, h, m, s, wd, jd, ds = time.localtime()
100 hostName = os.environ["HOST"]
101
102 storeName = "tiara-" + \
103 "%(Y)d" % vars() + "_" + \
104 "%(M)d" % vars() + "_" + \
105 "%(D)d" % vars() + "_" + \
106 "%(h)d" % vars() + "_" + \
107 "%(m)d" % vars() + "_" + \
108 "%(s)d" % vars() + "_" + \
109 hostName
110
111 return storeName
112
113
114def getConfigurationInfo(impGeo, experiment, physicsList, totalTime,
115 comment):
116 configInfo = {}
117
118
119 configInfo["energy"] = experiment.energy
120
121 configInfo["shieldMaterial"] = experiment.shieldMaterial
122
123 width = experiment.shieldWidth / CLHEP.cm
124 s_width = "%(width)d" % vars()
125 configInfo["shieldWidth"] = s_width
126
127
128 base = impGeo.base
129 simp = "no"
130 if base > 1:
131 simp = "ImpBase_%(base)f" % vars()
132
133
134 configInfo["biasing"] = simp
135
136 configInfo["physListName"] = physicsList.getName()
137
138 particles = ""
139
140 configInfo["minEnergyCut"] = experiment.particleCut
141
142
143 stime = "no"
144 if totalTime > 0:
145 stime = getTotalTime(totalTime)
146
147
148 configInfo["timeLimit"] = stime
149
150
151 configInfo["impGeoName"] = impGeo.nameExt
152
153
154 configInfo["hostName"] = os.environ["HOST"]
155
156
157
158 configInfo["comment"] = comment
159
160
161 return configInfo
162
163
164
165
166def setConfigInfo(tree, confInfo, af):
167 hf = af.createHistogramFactory(tree)
168 h = hf.createHistogram1D("configInfo","configInfo",1,0,1)
169 anno = h.annotation()
170 for k in confInfo:
171 anno.addItem(k, confInfo[k])
172
173
174def getConfigInfoFromTree(tree):
175 configInfo = {}
176 h = tree.findH1D("configInfo")
177 anno = h.annotation ()
178 for i in range(8, anno.size()):
179 configInfo[anno.key(i)] = anno.value(i)
180
181 return configInfo
182
183if G4analysisUse:
184 def addToXML(mergedXMLStore, xmlStore):
185 objNames = xmlStore.listObjectNames()
186 objTypes = xmlStore.listObjectTypes()
187 hf = myPI.af.createHistogramFactory(mergedXMLStore)
188 for i in range(len(objNames)):
189 oName = objNames[i]
190 oType = objTypes[i]
191 if oType == "IHistogram1D":
192 h1 = mergedXMLStore.findH1D(oName)
193 h2 = xmlStore.findH1D(oName)
194 hf.add(oName,h1,h2)
195
196if G4analysisUse:
197 def comparableShelves(she, mergedXMLStore):
198 comp = 1
199 if mergedShelve["energy"] != she["energy"]:
200 comp = 0
201 if mergedShelve["shieldWidth"] != she["shieldWidth"]:
202 comp = 0
203
204
205def setUpMergedShelve(she, mergedShelve, mergedXMLname, shelveNameToBeAdded):
206 mergedShelve["xmlStoreName"] = mergedXMLname
207 mergedShelve["energy"] = she["energy"]
208 mergedShelve["shieldWidth"] = she["shieldWidth"]
209 mergedShelve["mergedFiles"] = shelveNameToBeAdded
210 mergedShelve["runTime"] = she["runTime"]
211 mergedShelve["generatorTally"] = she["generatorTally"]
212 mergedShelve["sourceDetectorTally"] = she["sourceDetectorTally"]
213 mergedShelve["detector_00Tally"] = she["detector_00Tally"]
214 mergedShelve["detector_20Tally"] = she["detector_20Tally"]
215 mergedShelve["detector_40Tally"] = she["detector_40Tally"]
216
217
218def addToShelve(mergedShelve, she, shelveNameToBeAdded):
219 mergedShelve["mergedFiles"] += shelveNameToBeAdded + " "
220 mergedShelve["runTime"] += she["runTime"]
221 mergedShelve["generatorTally"].addMeasures(she["generatorTally"])
222 mergedShelve["sourceDetectorTally"].addMeasures(she["sourceDetectorTally"])
223 mergedShelve["detector_00Tally"].addMeasures(she["detector_00Tally"])
224 mergedShelve["detector_20Tally"].addMeasures(she["detector_20Tally"])
225 mergedShelve["detector_40Tally"].addMeasures(she["detector_40Tally"])
226
227if G4analysisUse:
228 def mergeData(mergedName, shelveList):
229 mergedShelveName = mergedName + ".shelve"
230 print "creating merged shelve: ", mergedShelveName
231 mergedShelve = shelve.open(mergedShelveName)
232 mergedXMLname = mergedName + ".xml"
233 print "creating merged XML store: ", mergedXMLname
234 mergedXMLStore = myPI.tf.create(mergedXMLname, "xml", 0, 1)
235
236
237 for i in range(len(shelveList)):
238 shelveName = shelveList[i]
239 print "adding data from: ", shelveName
240 she = shelve.open(shelveName,"r")
241 xmlStoreName = she["xmlStoreName"]
242 print "opening: ", xmlStoreName
243 xmlStore = myPI.tf.create(xmlStoreName,"xml",1,0)
244 if i == 0:
245 if mergedXMLStore.listObjectNames() == ():
246 setUpMergedShelve(she, mergedShelve, mergedXMLname, shelveName)
247 coppyTrees(xmlStore, mergedXMLStore)
248 else:
249 if comparableShelves(she, mergedXMLStore):
250 addToShelve(mergedShelve, she, shelveName)
251 addToXML(mergedXMLStore, xmlStore)
252 else:
253 addToShelve(mergedShelve, she, shelveName)
254 addToXML(mergedXMLStore, xmlStore)
255
256 mergedShelve.close()
257 mergedXMLStore.commit()
258 mergedXMLStore.close()
259
260
261def rmPath(she):
262 xmlFile = she["xmlStoreName"]
263 n = string.rfind(xmlFile,"/")
264 if n > -1:
265 xmlFile = xmlFile[n+1:]
266 she["xmlStoreName"] = xmlFile
267
268
269
Note: See TracBrowser for help on using the repository browser.