source: trunk/examples/advanced/Tiara/source/py_modules/tiaraApplication.py @ 807

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

update
File size: 9.0 KB
Line 
1# $Id: tiaraApplication.py,v 1.7 2006/06/26 10:13:14 ahoward Exp $
2# -------------------------------------------------------------------
3# GEANT4 tag $Name:  $
4# -------------------------------------------------------------------
5#
6import string
7import CLHEP
8import G4Kernel
9import Tiara
10
11import tiaraSpecifications
12import tallyData
13
14
15
16class TiaraApplet(object):
17    tiaraSim = None
18    def __init__(self, tiaraSpecs, tSim = None, usePI = True):
19        if usePI:
20            import myPI
21            self.tree = myPI.tf.create()
22            self.hf = myPI.af.createHistogramFactory (self.tree)
23            print self.hf
24            print "hf"
25            hf_ptr = str(self.hf._theObject)
26            print hf_ptr
27            print "hf_ptr before split"
28            hf_ptr = string.split(str(self.hf._theObject))[-1]
29            print hf_ptr
30            print "hf_ptr after split" 
31            hf_ptr2 = hf_ptr.split('x')[1]
32            for i in range(len(hf_ptr2),8) :
33                hf_ptr2 = '0'+hf_ptr2
34               
35            self.hf.this='_' + hf_ptr2 + '_p_AIDA__IHistogramFactory'
36            self.hf.thisown = 1
37            print self.hf.this
38            print "hf.this after split"
39        else:
40            self.tree = None
41            self.hf = None
42           
43        if (not TiaraApplet.tiaraSim):
44            if not tSim:
45                print "Error: TiaraApplet: argument is empty, and no tiaraSim exists!"
46            else:
47                TiaraApplet.tiaraSim = tSim
48        self.tiaraSpecs = tiaraSpecs
49        self.tiaraHall = Tiara.TiaraGeometry(self.tiaraSpecs.\
50                                             materials)
51        self.cellScorerStore = Tiara.TiaraCellScorerStore()
52        self.sampler = G4Kernel.G4MassGeometrySampler("neutron")
53        self.eventAction = None
54        self.scorer = None
55        self.primGen = None
56        self.scSrc = None
57        self.physicsList = None
58        self.nameExt = ""
59        self.scoreDets = []
60        self.scoreDetectorCreator = None
61        self.noComponents = 0
62        self.createdDPS = 0
63        self.tiaraSim.SetGeometry(self.tiaraHall)
64
65    def specifyPhysicsList(self, pl, particleCut):
66        self.particleCut = particleCut
67        self.physicsList = pl
68       
69    def setScoreDetectorCreator(self, creator):
70        self.scoreDetectorCreator=creator
71       
72    def visMode(self):
73        if self.eventAction:
74            print "TiaraApplet.visMode(): event action exists already"
75        else:
76            self.eventAction = Tiara.TiaraVisEventAction()
77# now later            self.tiaraSim.AddVisRunAction()
78            self.tiaraSim.AddTiaraEventAction(self.eventAction)
79        return
80
81    def timedMode(self, time, randomNumFileName = ""):
82        if self.eventAction:
83            print "TiaraApplet.timedMode(): event action exists already"
84        else:
85            self.eventAction = Tiara.TiaraTimedEventAction(time)
86            if randomNumFileName:
87                self.eventAction.SetRnadomNumFilename(randomNumFileName)
88            self.tiaraSim.AddTiaraEventAction(self.eventAction)
89        return
90   
91   
92    def buildColimator(self):
93        posColi = \
94                CLHEP.Hep3Vector(0,0, self.tiaraSpecs.
95                                 dimensions.targetPosZ + 
96                                 self.tiaraSpecs.
97                                 dimensions.distTargetExperiment +
98                                 self.tiaraSpecs.\
99                                 experiment.colWidth/2)
100       
101        logCol = self.tiaraHall.BuildCollimator(self.tiaraSpecs.\
102                                                experiment.colWidth,
103                                                "iron",
104                                                "air")
105        self.tiaraHall.PlaceExpComponent(posColi,
106                                         logCol,
107                                         "colimator")
108        return
109
110    def buildShield(self):
111        log = self.tiaraHall.BuildShield(self.tiaraSpecs.\
112                                         experiment.shieldWidth,
113                                         self.tiaraSpecs.\
114                                         experiment.shieldMaterial)
115        posShield = CLHEP.Hep3Vector(0,0, self.tiaraSpecs.
116                                     dimensions.targetPosZ + 
117                                     self.tiaraSpecs.
118                                     dimensions.
119                                     distTargetExperiment + 
120                                     self.tiaraSpecs.experiment.
121                                     colWidth +
122                                     self.tiaraSpecs.experiment.\
123                                     shieldWidth/2)
124        self.tiaraHall.PlaceExpComponent(posShield, log, "shield")
125        return
126
127    def buildGeometry(self):
128        self.tiaraHall.BuildGeometry(self.tiaraSpecs.dimensions)
129        if self.noComponents != 1:
130            self.tiaraHall.CreateComponents()
131        if self.tiaraSpecs.experiment.colWidth > 0:
132            self.buildColimator()
133        self.buildShield()
134
135    def createCellScorers(self):
136        self.cellScorer = []
137        tally = Tiara.TiaraTally()
138        tally.setBinEdges(tiaraSpecifications.\
139                          tallyBinEdges[self.tiaraSpecs.\
140                                        experiment.energy])
141        for det in self.scoreDets:
142            det.scorer = None
143            if self.hf:
144                det.scorer = \
145                           Tiara.\
146                           TiaraCellScorer(self.hf,
147                                           det.name,
148                                           self.tiaraSpecs.
149                                           experiment.binEdgesScinti,
150                                           self.tiaraSpecs.
151                                           experiment.binEdgesBonner,
152                                           tally)
153            else:
154                det.scorer = \
155                           Tiara.\
156                           TiaraCellScorer(det.name,
157                                           tally)
158            self.cellScorer.append(det.scorer)
159           
160                                   
161    def buildDetectors(self):
162        tally = Tiara.TiaraTally()
163        tally.setBinEdges(tiaraSpecifications.\
164                          tallyBinEdges[self.tiaraSpecs.
165                                        experiment.energy])
166
167        physSrcDet = self.tiaraHall.AddSourceDetector()
168        self.scSrc = None
169        if self.hf:
170            self.scSrc = Tiara.\
171                         TiaraCellScorer(self.hf,
172                                         "source_detector",
173                                         self.tiaraSpecs.
174                                         experiment.binEdgesScinti,
175                                         self.tiaraSpecs.
176                                         experiment.binEdgesBonner,
177                                         tally)
178
179        else:
180            self.scSrc = Tiara.\
181                         TiaraCellScorer("source_detector",
182                                         tally)
183           
184
185        self.scoreDets = self.scoreDetectorCreator.\
186                         createScoreDetectors(self.tiaraHall)
187
188        self.createCellScorers()
189
190        self.cellScorerStore.AddTiaraCellScorer(self.scSrc,
191                                      G4Kernel.G4GeometryCell(physSrcDet,
192                                                                  0))
193       
194        for det in self.scoreDets:
195            self.cellScorerStore.AddTiaraCellScorer(det.scorer,
196                                                    G4Kernel.\
197                                                    G4GeometryCell(det.
198                                                                   phys,
199                                                                   0))
200
201        self.scorer = G4Kernel.G4CellStoreScorer(self.cellScorerStore.
202                                                 GetG4VCellScorerStore())
203
204        self.eventAction.SetScorerStore(self.cellScorerStore)
205           
206        return
207
208    def setPhysics(self):
209        self.tiaraSim.SetPhysicsList(self.physicsList)
210        return
211
212    def visAdd(self):
213        self.tiaraSim.AddVisRunAction()
214        return
215
216    def setPrimaryGenerator(self, primGen):
217        self.primGen = primGen
218        self.tiaraSim.SetPrimaryGenerator(self.primGen)
219        return
220
221    def fillShelve(self, shelveDB):
222        t = self.primGen.GetTally()
223        shelveDB["generatorTally"] = tallyData.createTallyDat(t)
224        t = self.scSrc.GetTally()
225        shelveDB["sourceDetectorTally"] = tallyData.createTallyDat(t)
226        for det in self.scoreDets:
227            t = det.scorer.GetTally()
228            shelveDB[det.name + "Tally"] = tallyData.createTallyDat(t)
229        shelveDB["runTime"] = self.eventAction.GetTotalProcessedTime()
230       
231    def config(self):
232
233        self.buildDetectors()
234        self.tiaraSim.initialize()
235
236        if len(self.particleCut) > 0:
237            for p in self.particleCut:
238                self.tiaraSim.AddParticleCut(p, self.particleCut[p]);
239       
240        self.sampler.PrepareScoring(self.scorer)
241        self.sampler.Configure()
242
243 
Note: See TracBrowser for help on using the repository browser.