| Geant4 Cross Reference |
1 // 1
2 // *******************************************
3 // * License and Disclaimer
4 // *
5 // * The Geant4 software is copyright of th
6 // * the Geant4 Collaboration. It is provided
7 // * conditions of the Geant4 Software License
8 // * LICENSE and available at http://cern.ch/
9 // * include a list of copyright holders.
10 // *
11 // * Neither the authors of this software syst
12 // * institutes,nor the agencies providing fin
13 // * work make any representation or warran
14 // * regarding this software system or assum
15 // * use. Please see the license in the file
16 // * for the full disclaimer and the limitatio
17 // *
18 // * This code implementation is the result
19 // * technical work of the GEANT4 collaboratio
20 // * By using, copying, modifying or distri
21 // * any work based on the software) you ag
22 // * use in resulting scientific publicati
23 // * acceptance of all terms of the Geant4 Sof
24 // *******************************************
25 //
26 //
27 //
28
29 #include "G4TheMTRayTracer.hh"
30 #include "G4SystemOfUnits.hh"
31 #include "G4RTMessenger.hh"
32 #include "G4VFigureFileMaker.hh"
33 #include "G4RTJpegMaker.hh"
34 #include "G4RTRun.hh"
35 #include "G4RTRunAction.hh"
36 #include "G4RTWorkerInitialization.hh"
37 #include "G4VRTScanner.hh"
38
39 #include "G4MTRunManager.hh"
40 #include "G4SDManager.hh"
41 #include "G4StateManager.hh"
42 #include "G4Colour.hh"
43 #include "G4VisAttributes.hh"
44 #include "G4UImanager.hh"
45 #include "G4UIcommand.hh"
46 #include "G4VVisManager.hh"
47 #include "G4RunManagerFactory.hh"
48
49 #define G4warn G4cout
50
51 G4TheMTRayTracer* G4TheMTRayTracer::theInstanc
52
53 G4TheMTRayTracer::G4TheMTRayTracer(G4VFigureFi
54 G4VRTScanner* scanner)
55 : G4TheRayTracer(figMaker,scanner)
56 {
57 if(!theInstance)
58 { theInstance = this; }
59 else
60 { G4Exception("G4TheMTRayTracer::G4TheMTRayT
61 FatalException,"G4TheMTRayTracer has
62 theUserWorkerInitialization = 0;
63 theRTWorkerInitialization = 0;
64 theUserRunAction = 0;
65 theRTRunAction = 0;
66 }
67
68 G4TheMTRayTracer* G4TheMTRayTracer::Instance()
69 {
70 if (theInstance) return theInstance;
71 else return new G4TheMTRayTracer;
72 }
73
74 G4TheMTRayTracer* G4TheMTRayTracer::Instance
75 (G4VFigureFileMaker* figMaker,G4VRTScanner* sc
76 {
77 if (theInstance) {
78 theFigMaker=figMaker;
79 theScanner=scanner;
80 return theInstance;
81 }
82 else return new G4TheMTRayTracer(figMaker,sc
83 }
84
85 G4TheMTRayTracer::~G4TheMTRayTracer()
86 {
87 if(theRTWorkerInitialization)
88 {
89 delete theRTWorkerInitialization;
90 theRTWorkerInitialization = 0;
91 }
92 if(theRTRunAction)
93 {
94 delete theRTRunAction;
95 theRTRunAction = 0;
96 }
97 }
98
99 void G4TheMTRayTracer::Trace(const G4String& f
100 {
101 G4StateManager* theStateMan = G4StateManager
102 G4ApplicationState currentState = theStateMa
103 if(currentState!=G4State_Idle)
104 {
105 G4warn << "Illegal application state <" <<
106 << "> - Trace() ignored. " << G4end
107 return;
108 }
109
110 if(!theFigMaker)
111 {
112 G4warn << "Figure file maker class is not
113 return;
114 }
115
116 G4UImanager* UI = G4UImanager::GetUIpointer(
117 G4int storeTrajectory = UI->GetCurrentIntVal
118 UI->ApplyCommand("/tracking/storeTrajectory
119
120 G4ThreeVector tmpVec = targetPosition - eyeP
121 eyeDirection = tmpVec.unit();
122 G4int nPixel = nColumn*nRow;
123 colorR = new unsigned char[nPixel];
124 colorG = new unsigned char[nPixel];
125 colorB = new unsigned char[nPixel];
126 unsigned char defR = (unsigned char)(G4int(2
127 unsigned char defG = (unsigned char)(G4int(2
128 unsigned char defB = (unsigned char)(G4int(2
129 for(G4int ii=0;ii<nPixel;++ii)
130 {
131 colorR[ii] = defR;
132 colorG[ii] = defG;
133 colorB[ii] = defB;
134 }
135
136 G4bool succeeded = CreateBitMap();
137 if(succeeded)
138 { CreateFigureFile(fileName); }
139 else
140 { G4warn << "Could not create figure file" <
141 G4warn << "You might set the eye position
142
143 G4String str = "/tracking/storeTrajectory "
144 UI->ApplyCommand(str);
145
146 delete [] colorR;
147 delete [] colorG;
148 delete [] colorB;
149 }
150
151 void G4TheMTRayTracer::StoreUserActions()
152 {
153 G4MTRunManager* mrm = G4RunManagerFa
154 theUserWorkerInitialization = mrm->GetUserWo
155 theUserRunAction = mrm->GetUserRunAction();
156
157 if(!theRTWorkerInitialization) theRTWorkerIn
158 if(!theRTRunAction) theRTRunAction = new G4R
159
160 mrm->SetUserInitialization(theRTWorkerInitia
161 mrm->SetUserAction(theRTRunAction);
162 }
163
164 void G4TheMTRayTracer::RestoreUserActions()
165 {
166 G4MTRunManager* mrm = G4RunManagerFactory::G
167 mrm->SetUserInitialization(
168 const_cast<G4UserWorkerInitialization*>(th
169 mrm->SetUserAction(const_cast<G4UserRunActio
170 }
171
172 G4bool G4TheMTRayTracer::CreateBitMap()
173 {
174 G4VVisManager* visMan = G4VVisManager::GetCo
175 visMan->IgnoreStateChanges(true);
176 StoreUserActions();
177
178 G4MTRunManager* mrm = G4RunManagerFactory::G
179
180 // Keep, then switch off any printing reques
181 auto runVerbosity = mrm->GetVerboseLeve
182 auto runPrintProgress = mrm->GetPrintProgre
183 G4UImanager::GetUIpointer()->ApplyCommand("/
184 G4UImanager::GetUIpointer()->ApplyCommand("/
185
186 // Event loop
187 G4int nEvent = nRow*nColumn;
188 //// mrm->BeamOn(nEvent);
189 //// Temporary work-around until direct invok
190 G4String str = "/run/beamOn " + G4UIcommand:
191 G4UImanager::GetUIpointer()->ApplyCommand(st
192
193 // Restore printing requests
194 str = "/run/verbose " + G4UIcommand::Convert
195 G4UImanager::GetUIpointer()->ApplyCommand(st
196 str = "/run/printProgress " + G4UIcommand::C
197 G4UImanager::GetUIpointer()->ApplyCommand(st
198
199 RestoreUserActions();
200 visMan->IgnoreStateChanges(false);
201
202 const G4RTRun* theRun = static_cast<const G4
203 if(!theRun) return false;
204
205 G4THitsMap<G4Colour>* colMap = theRun->GetMa
206 auto itr = colMap->GetMap()->cbegin();
207 for(;itr!=colMap->GetMap()->cend();++itr)
208 {
209 G4int key = itr->first;
210 G4Colour* col = itr->second;
211 colorR[key] = (unsigned char)(G4int(255*co
212 colorG[key] = (unsigned char)(G4int(255*co
213 colorB[key] = (unsigned char)(G4int(255*co
214 }
215
216 theScanner->Initialize(nRow,nColumn);
217 G4int iRow, iColumn;
218 while (theScanner->Coords(iRow,iColumn))
219 {
220 G4int iCoord = iRow * nColumn + iColumn;
221 theScanner->Draw(colorR[iCoord],colorG[iCo
222 }
223
224 return true;
225 }
226
227