]>
Commit | Line | Data |
---|---|---|
2817d3e2 | 1 | // $Id$ |
2 | // Category: visualization | |
3 | // | |
499b353a | 4 | // Author: I. Hrivnacova, A. Gheata |
5 | // | |
6 | // Class TG4VisManager | |
7 | // ------------------- | |
8 | // See the class description in the header file. | |
2817d3e2 | 9 | // According to visualization/management/include/MyVisManager.* |
10 | // John Allison 24th January 1998. | |
11 | // I. Hrivnacova 12.5.98 | |
12 | // | |
13 | // Renamed to TG4VisManager | |
14 | // I. Hrivnacova, 3.9.99 | |
15 | // | |
16 | // Added AliMC implementation | |
17 | // A. Gheata, 22.2.00 | |
18 | // | |
19 | // Added OpenGL*Win32, RayTracer (as replacement of RayX) drivers | |
20 | // based on G4 suggestions. | |
21 | // I. Gonzalez, 4.4.2000 | |
22 | // | |
23 | // Example Visualization Manager description: | |
24 | // Implements virtual function RegisterGraphicsSystems. | |
25 | // Exploits C-pre-processor variables G4VIS_USE_DAWN, etc., | |
26 | // which are set by the G4 makefiles if | |
27 | // environment variables of the same name are set. | |
28 | // | |
29 | // So all you have to do is set environment variables and compile and | |
30 | // instantiate this in your main(). | |
31 | // | |
32 | // Alternatively, you can implement an empty function here and just | |
33 | // register the systems you want in your main(), e.g.: | |
34 | // G4VisManager* myVisManager = new MyVisManager; | |
35 | // myVisManager -> RegisterGraphicsSystem (new MyGraphicsSystem); | |
36 | ||
37 | ||
304cd517 | 38 | #ifdef G4VIS_USE |
39 | ||
40 | #include "TG4VisManager.h" | |
2817d3e2 | 41 | #include "TG4Globals.h" |
42 | ||
43 | #include <G4LogicalVolumeStore.hh> | |
44 | #include <G4PhysicalVolumeStore.hh> | |
2817d3e2 | 45 | #include <G4TransportationManager.hh> |
46 | #include <G4Material.hh> | |
304cd517 | 47 | #include <G4PhysicalVolumeModel.hh> |
2817d3e2 | 48 | #include <G4VVisManager.hh> |
49 | ||
50 | // Supported drivers... | |
51 | ||
52 | #ifdef G4VIS_USE_DAWN | |
53 | #include <G4FukuiRenderer.hh> | |
54 | #endif | |
55 | ||
56 | #ifdef G4VIS_USE_DAWNFILE | |
57 | #include <G4DAWNFILE.hh> | |
58 | #endif | |
59 | ||
60 | #ifdef G4VIS_USE_OPACS | |
61 | #include <G4Wo.hh> | |
62 | #include <G4Xo.hh> | |
63 | #endif | |
64 | ||
65 | #ifdef G4VIS_USE_OPENGLX | |
66 | #include <G4OpenGLImmediateX.hh> | |
67 | #include <G4OpenGLStoredX.hh> | |
68 | #endif | |
69 | ||
70 | #ifdef G4VIS_USE_OPENGLXM | |
71 | #include <G4OpenGLImmediateXm.hh> | |
72 | #include <G4OpenGLStoredXm.hh> | |
73 | #endif | |
74 | ||
75 | #ifdef G4VIS_USE_OPENGLWIN32 | |
76 | #include "G4OpenGLImmediateWin32.hh" | |
77 | #include "G4OpenGLStoredWin32.hh" | |
78 | #endif | |
79 | ||
80 | #ifdef G4VIS_USE_OIX | |
81 | #include <G4OpenInventorX.hh> | |
82 | #endif | |
83 | ||
84 | #ifdef G4VIS_USE_OIWIN32 | |
85 | #include <G4OpenInventorWin32.hh> | |
86 | #endif | |
87 | ||
88 | #ifdef G4VIS_USE_RAYTRACER | |
89 | #include "G4RayTracer.hh" | |
90 | #endif | |
91 | ||
92 | #ifdef G4VIS_USE_VRML | |
93 | #include <G4VRML1.hh> | |
94 | #include <G4VRML2.hh> | |
95 | #endif | |
96 | ||
97 | #ifdef G4VIS_USE_VRMLFILE | |
98 | #include <G4VRML1File.hh> | |
99 | #include <G4VRML2File.hh> | |
100 | #endif | |
101 | ||
72095f7c | 102 | //_____________________________________________________________________________ |
2817d3e2 | 103 | TG4VisManager::TG4VisManager(G4int verboseLevel) { |
104 | // | |
105 | fVerbose = verboseLevel; | |
106 | fColourFlag = true; | |
107 | } | |
108 | ||
72095f7c | 109 | //_____________________________________________________________________________ |
2817d3e2 | 110 | TG4VisManager::TG4VisManager(const TG4VisManager& right) { |
111 | // | |
112 | TG4Globals::Exception( | |
113 | "Attempt to copy TG4VisManager singleton."); | |
114 | } | |
115 | ||
72095f7c | 116 | //_____________________________________________________________________________ |
2817d3e2 | 117 | TG4VisManager::~TG4VisManager() { |
118 | // | |
119 | } | |
120 | ||
121 | // operators | |
122 | ||
72095f7c | 123 | //_____________________________________________________________________________ |
2817d3e2 | 124 | TG4VisManager& TG4VisManager::operator=(const TG4VisManager& right) |
125 | { | |
126 | // check assignement to self | |
127 | if (this == &right) return *this; | |
128 | ||
129 | TG4Globals::Exception( | |
130 | "Attempt to assign TG4VisManager singleton."); | |
131 | ||
132 | return *this; | |
133 | } | |
134 | ||
135 | // private methods | |
136 | ||
72095f7c | 137 | //_____________________________________________________________________________ |
2817d3e2 | 138 | void TG4VisManager::RegisterGraphicsSystems() |
139 | { | |
140 | // Registers the graphics systems. | |
141 | // --- | |
142 | ||
143 | // all created graphics system instances are | |
144 | // deleted in G4VisManager::~G4VisManager() | |
145 | #ifdef G4VIS_USE_DAWN | |
146 | RegisterGraphicsSystem(new G4FukuiRenderer); | |
147 | #endif | |
148 | ||
149 | #ifdef G4VIS_USE_DAWNFILE | |
150 | RegisterGraphicsSystem(new G4DAWNFILE); | |
151 | #endif | |
152 | ||
153 | #ifdef G4VIS_USE_OPACS | |
154 | RegisterGraphicsSystem(new G4Wo); | |
155 | RegisterGraphicsSystem(new G4Xo); | |
156 | #endif | |
157 | ||
158 | #ifdef G4VIS_USE_OPENGLX | |
159 | RegisterGraphicsSystem(new G4OpenGLImmediateX); | |
160 | RegisterGraphicsSystem(new G4OpenGLStoredX); | |
161 | #endif | |
162 | ||
163 | #ifdef G4VIS_USE_OPENGLXM | |
164 | RegisterGraphicsSystem(new G4OpenGLImmediateXm); | |
165 | RegisterGraphicsSystem(new G4OpenGLStoredXm); | |
166 | #endif | |
167 | ||
168 | #ifdef G4VIS_USE_OPENGLWIN32 | |
169 | RegisterGraphicsSystem (new G4OpenGLImmediateWin32); | |
170 | RegisterGraphicsSystem (new G4OpenGLStoredWin32); | |
171 | #endif | |
172 | ||
173 | #ifdef G4VIS_USE_OIX | |
174 | RegisterGraphicsSystem(new G4OpenInventorX); | |
175 | #endif | |
176 | ||
177 | #ifdef G4VIS_USE_OIWIN32 | |
178 | RegisterGraphicsSystem(new G4OpenInventorWin32); | |
179 | #endif | |
180 | ||
181 | #ifdef G4VIS_USE_RAYTRACER | |
182 | RegisterGraphicsSystem (new G4RayTracer); | |
183 | #endif | |
184 | ||
185 | #ifdef G4VIS_USE_VRML | |
186 | RegisterGraphicsSystem(new G4VRML1); | |
187 | RegisterGraphicsSystem(new G4VRML2); | |
188 | #endif | |
189 | ||
190 | #ifdef G4VIS_USE_VRMLFILE | |
191 | RegisterGraphicsSystem(new G4VRML1File); | |
192 | RegisterGraphicsSystem(new G4VRML2File); | |
193 | #endif | |
194 | ||
195 | if (fVerbose > 0) { | |
3c7cd15a | 196 | G4cout << |
2817d3e2 | 197 | "\nYou have successfully chosen to use the following graphics systems." |
3c7cd15a | 198 | << G4endl; |
2817d3e2 | 199 | PrintAvailableGraphicsSystems(); |
200 | } | |
201 | } | |
202 | ||
203 | //--------------------------------------------------------------- | |
204 | // private methods | |
205 | //--------------------------------------------------------------- | |
206 | ||
207 | ||
72095f7c | 208 | //_____________________________________________________________________________ |
2817d3e2 | 209 | G4RWTPtrOrderedVector<G4LogicalVolume> TG4VisManager::GetLVList(G4String name) |
210 | { | |
211 | // Get function returning the list of logical volumes | |
212 | // associated to NAME; G4 built clones of a G3 volume (identified | |
213 | // with NAME_NUMBER will be added to the list) | |
214 | // NAME can be the name of a logical or physical volume | |
215 | // --- | |
216 | ||
217 | G4RWTPtrOrderedVector <G4LogicalVolume> lvList; | |
218 | G4LogicalVolumeStore* pLVStore = G4LogicalVolumeStore::GetInstance(); | |
219 | G4LogicalVolume* pLV = 0; | |
220 | if (pLVStore) | |
221 | { | |
765c040d | 222 | for (G4int i=0; i<pLVStore->size(); i++) |
2817d3e2 | 223 | { |
765c040d | 224 | pLV = (*pLVStore)[i]; |
2817d3e2 | 225 | if (CaseInsensitiveEqual(name,pLV->GetName())) |
226 | { | |
227 | if (!lvList.contains(pLV)) lvList.append(pLV); | |
228 | } | |
229 | } | |
230 | } | |
231 | if (!lvList.isEmpty()) return lvList; | |
232 | G4PhysicalVolumeStore* pPVStore = G4PhysicalVolumeStore::GetInstance(); | |
233 | G4VPhysicalVolume* pPV = 0; | |
234 | if (pPVStore) | |
235 | { | |
765c040d | 236 | for (G4int i=0; i<pPVStore->size(); i++) |
2817d3e2 | 237 | { |
765c040d | 238 | pPV = (*pPVStore)[i]; |
2817d3e2 | 239 | if (CaseInsensitiveEqual(name,pPV->GetName())) |
240 | { | |
241 | pLV = pPV->GetLogicalVolume(); | |
242 | if (!lvList.contains(pLV)) lvList.append(pLV); | |
243 | } | |
244 | } | |
245 | } | |
246 | return lvList; | |
247 | } | |
248 | ||
249 | ||
72095f7c | 250 | //_____________________________________________________________________________ |
2817d3e2 | 251 | G4RWTPtrOrderedVector<G4VPhysicalVolume> TG4VisManager::GetPVList(G4String name) |
252 | { | |
253 | // Get function returning the physical volume pointer for NAME | |
254 | // --- | |
255 | ||
256 | G4RWTPtrOrderedVector <G4VPhysicalVolume> pvList; | |
257 | G4PhysicalVolumeStore* pPVStore = G4PhysicalVolumeStore::GetInstance(); | |
258 | if (!pPVStore) | |
259 | { | |
260 | TG4Globals::Warning("TG4VisManager::Gdraw : No volume store !"); | |
261 | return pvList; | |
262 | } | |
263 | G4VPhysicalVolume* pPV = 0; | |
765c040d | 264 | for (G4int i=0; i<pPVStore->size(); i++) |
2817d3e2 | 265 | { |
765c040d | 266 | pPV = (*pPVStore)[i]; |
2817d3e2 | 267 | if (CaseInsensitiveEqual(name,pPV->GetName())) |
268 | { | |
269 | if (!pvList.contains(pPV)) pvList.append(pPV); | |
270 | } | |
271 | } | |
272 | return pvList; | |
273 | } | |
274 | ||
275 | ||
72095f7c | 276 | //_____________________________________________________________________________ |
2817d3e2 | 277 | G4bool TG4VisManager::CaseInsensitiveEqual(const G4String string1, |
278 | const G4String string2) | |
279 | { | |
280 | // Case insensitive comparison of 2 strings | |
281 | // --- | |
282 | ||
283 | G4int i; | |
284 | if (string1 == string2) return true; | |
285 | // if (string1.length() != string2.length()) return false; | |
286 | for (i=0; i<string1.length(); i++) | |
287 | { | |
288 | G4int diff = abs((G4int)string1(i)-(G4int)string2(i)); | |
289 | if (diff && (diff!=32)) return false; | |
290 | } | |
291 | if (string2.length() > string1.length()) | |
292 | { | |
293 | if (string2(i) == '_') return true; | |
294 | return false; | |
295 | } | |
296 | return true; | |
297 | } | |
298 | ||
299 | ||
72095f7c | 300 | //_____________________________________________________________________________ |
2817d3e2 | 301 | void TG4VisManager::SetAtt4Daughters(G4LogicalVolume* const lv, |
5317affd | 302 | const TG4G3Attribute att, const G4int val) |
2817d3e2 | 303 | { |
304 | // Iterator for setting a visual attribute for all daughters | |
305 | // --- | |
306 | ||
307 | SetG4Attribute(lv,att,val); | |
308 | ||
309 | G4String lvName = lv->GetName(); | |
310 | G4int nOfDaughters = lv->GetNoDaughters(); | |
311 | if (nOfDaughters>0) | |
312 | { | |
313 | G4String previousName = ""; | |
314 | for (G4int i=0; i<nOfDaughters; i++) | |
315 | { | |
316 | G4LogicalVolume* lvd = lv->GetDaughter(i)->GetLogicalVolume(); | |
317 | G4String currentName = lvd->GetName(); | |
318 | if (currentName != lvName && currentName != previousName) | |
319 | { | |
320 | SetAtt4Daughters(lvd, att, val); | |
321 | previousName = currentName; | |
322 | } | |
323 | } | |
324 | } | |
325 | } | |
326 | ||
327 | ||
72095f7c | 328 | //_____________________________________________________________________________ |
2817d3e2 | 329 | G4bool TG4VisManager::IsSharedVisAttributes(const G4LogicalVolume* pLV) |
330 | { | |
331 | // Function seeking if the volume's visible attributes are shared with | |
332 | // other volumes | |
333 | // --- | |
334 | ||
335 | G4LogicalVolumeStore* pLVStore = G4LogicalVolumeStore::GetInstance(); | |
336 | G4LogicalVolume* pLVCurrent = 0; | |
337 | const G4VisAttributes* pVisAtt = pLV->GetVisAttributes(); | |
338 | if (!pVisAtt) return false; | |
765c040d | 339 | for (G4int i=0; i<pLVStore->size(); i++) |
2817d3e2 | 340 | { |
765c040d | 341 | pLVCurrent = (*pLVStore)[i]; |
2817d3e2 | 342 | if (pLVCurrent != pLV) |
343 | { | |
344 | if (pLVCurrent->GetVisAttributes() == pVisAtt) | |
345 | { | |
346 | return true; | |
347 | } | |
348 | } | |
349 | } | |
350 | return false; | |
351 | } | |
352 | ||
353 | ||
72095f7c | 354 | //_____________________________________________________________________________ |
2817d3e2 | 355 | void TG4VisManager::SetG4Attribute(G4LogicalVolume* const lv, |
5317affd | 356 | const TG4G3Attribute att, const G4int val) |
2817d3e2 | 357 | { |
358 | // Set the G4 attribute fo volume LV accordingly to the G3 description | |
359 | // of (att- val) | |
360 | // -- | |
361 | ||
362 | if (!lv) return; | |
363 | // Dupplicating old vis. attributes | |
364 | const G4VisAttributes* visAttributes = lv->GetVisAttributes(); | |
365 | G4VisAttributes* newVisAttributes; | |
366 | if (!visAttributes) | |
367 | newVisAttributes = new G4VisAttributes(false); | |
0e8db0fe | 368 | else { |
369 | G4bool visibility = visAttributes->IsVisible(); | |
370 | G4Colour colour = visAttributes->GetColour(); | |
371 | newVisAttributes = new G4VisAttributes(visibility, colour); | |
372 | } | |
2817d3e2 | 373 | |
b15a2a93 | 374 | const G4int kAbsVal = abs(val); // the functionality is given by the abs value |
2817d3e2 | 375 | |
376 | // Default visible attributes | |
377 | G4double red(0),green(0),blue(0); // default is black | |
378 | G4bool isVisible(false); | |
379 | G4bool isDaughtersInvisible(false); | |
380 | G4VisAttributes::LineStyle lineStyle = G4VisAttributes::unbroken; | |
381 | G4double lineWidth = 1.0; | |
382 | G4bool isForceDrawingStyle(false); | |
383 | G4VisAttributes::ForcedDrawingStyle drawingStyle = G4VisAttributes::wireframe; | |
384 | ||
385 | // a 'hardcopy' of old vis attributes is needed because the copy constructor | |
386 | // resets to defaults some of the data members of G4VisAttributes class | |
387 | if (visAttributes) | |
388 | { | |
389 | isVisible = visAttributes->IsVisible(); | |
390 | isDaughtersInvisible = visAttributes->IsDaughtersInvisible(); | |
391 | red = visAttributes->GetColour().GetRed(); | |
392 | green = visAttributes->GetColour().GetGreen(); | |
393 | blue = visAttributes->GetColour().GetBlue(); // old RGB components | |
394 | lineStyle = visAttributes->GetLineStyle(); | |
395 | lineWidth = visAttributes->GetLineWidth(); | |
396 | isForceDrawingStyle = visAttributes->IsForceDrawingStyle(); | |
397 | if (isForceDrawingStyle) | |
398 | drawingStyle = visAttributes->GetForcedDrawingStyle(); | |
399 | ||
400 | } | |
401 | G4double luminosityBin(0.04), // bin for luminosity | |
402 | luminosity(0); // colour luminosity | |
403 | ||
404 | // Delete old vis. attributes if they are not shared | |
405 | if (visAttributes && !IsSharedVisAttributes(lv)) delete visAttributes; | |
406 | ||
407 | // Set the required attribute | |
408 | switch (att) | |
409 | { | |
410 | case kSEEN: | |
411 | switch (val) | |
412 | { | |
413 | case 0: | |
414 | isVisible = false; | |
415 | break; | |
416 | case 1: | |
417 | isVisible = true; | |
418 | break; | |
419 | case -1: | |
420 | isVisible = false; | |
421 | break; | |
422 | case -2: | |
423 | isVisible = false; | |
424 | break; | |
425 | default: | |
426 | isVisible = false; | |
427 | } | |
428 | break; | |
429 | case kLSTY: | |
b15a2a93 | 430 | switch (kAbsVal) |
2817d3e2 | 431 | { |
432 | case 1: | |
433 | lineStyle = G4VisAttributes::unbroken; break; | |
434 | case 2: | |
435 | lineStyle = G4VisAttributes::dashed; break; | |
436 | case 3: | |
437 | lineStyle = G4VisAttributes::dotted; break; | |
438 | default: | |
439 | if (fVerbose > 0) | |
3c7cd15a | 440 | G4cout << "TG4VisManager::Gsatt() Usage of LSTY :" << G4endl |
441 | << "ATT = 1,2,3 means line unbroken, dashed or dotted" << G4endl | |
442 | << "any other value resets to the default : unbroken" << G4endl; | |
2817d3e2 | 443 | lineStyle = G4VisAttributes::unbroken; |
444 | } | |
445 | break; | |
446 | case kLWID: | |
b15a2a93 | 447 | lineWidth = kAbsVal; |
2817d3e2 | 448 | if (lineWidth > 7) lineWidth = 7; |
449 | if (fVerbose > 0) | |
3c7cd15a | 450 | G4cout << "TG4VisManager::Gsatt() Usage for LWID :" << G4endl |
2817d3e2 | 451 | << " The VAL you supply means the width of lines in pixels " |
3c7cd15a | 452 | << "for the screen and in 0.1*mm for paper." << G4endl |
2817d3e2 | 453 | << " Negative values means the same, but for all daughters" |
3c7cd15a | 454 | << G4endl; |
2817d3e2 | 455 | break; |
456 | case kCOLO: | |
b15a2a93 | 457 | if (kAbsVal < 8) // G3 base colours |
2817d3e2 | 458 | { |
b15a2a93 | 459 | switch (kAbsVal) |
2817d3e2 | 460 | { |
461 | case 1: | |
462 | red=0; green=0; blue=0; //black | |
463 | break; | |
464 | case 2: | |
465 | red=1; green=0; blue=0; //red | |
466 | break; | |
467 | case 3: | |
468 | red=0; green=1; blue=0; //green | |
469 | break; | |
470 | case 4: | |
471 | red=0; green=0; blue=1; //blue | |
472 | break; | |
473 | case 5: | |
474 | red=1; green=1; blue=0; //yellow | |
475 | break; | |
476 | case 6: | |
477 | red=1; green=0; blue=1; //violet | |
478 | break; | |
479 | case 7: | |
480 | red=0; green=1; blue=1; //lightblue (almost !) | |
481 | } | |
482 | luminosity = 0.; | |
483 | } | |
b15a2a93 | 484 | if (kAbsVal>=8 && kAbsVal<=16) |
2817d3e2 | 485 | { |
486 | red=0; green=0; blue=0; | |
b15a2a93 | 487 | luminosity = (kAbsVal-7)*luminosityBin; |
2817d3e2 | 488 | } |
b15a2a93 | 489 | if (kAbsVal>=17 && kAbsVal<=41) |
2817d3e2 | 490 | { |
491 | red=1; green=0; blue=0; | |
b15a2a93 | 492 | luminosity = (kAbsVal-16)*luminosityBin; |
2817d3e2 | 493 | } |
b15a2a93 | 494 | if (kAbsVal>=67 && kAbsVal<=91) |
2817d3e2 | 495 | { |
496 | red=0; green=1; blue=0; | |
b15a2a93 | 497 | luminosity = (kAbsVal-66)*luminosityBin; |
2817d3e2 | 498 | } |
b15a2a93 | 499 | if (kAbsVal>=117 && kAbsVal<=141) |
2817d3e2 | 500 | { |
501 | red=0; green=0; blue=1; | |
b15a2a93 | 502 | luminosity = (kAbsVal-116)*luminosityBin; |
2817d3e2 | 503 | } |
b15a2a93 | 504 | if (kAbsVal>=42 && kAbsVal<=66) |
2817d3e2 | 505 | { |
506 | red=1; green=1; blue=0; | |
b15a2a93 | 507 | luminosity = (kAbsVal-41)*luminosityBin; |
2817d3e2 | 508 | } |
b15a2a93 | 509 | if (kAbsVal>=142 && kAbsVal<=166) |
2817d3e2 | 510 | { |
511 | red=1; green=0; blue=1; | |
b15a2a93 | 512 | luminosity = (kAbsVal-141)*luminosityBin; |
2817d3e2 | 513 | } |
b15a2a93 | 514 | if (kAbsVal>=92 && kAbsVal<=116) |
2817d3e2 | 515 | { |
516 | red=0; green=1; blue=1; | |
b15a2a93 | 517 | luminosity = (kAbsVal-91)*luminosityBin; |
2817d3e2 | 518 | } |
519 | if (red < luminosityBin) red += luminosity; | |
520 | if (green < luminosityBin) green += luminosity; | |
521 | if (blue < luminosityBin) blue += luminosity; | |
522 | break; | |
523 | case kFILL: | |
524 | isForceDrawingStyle = true; | |
b15a2a93 | 525 | switch (kAbsVal) |
2817d3e2 | 526 | { |
527 | case 0: | |
528 | drawingStyle = G4VisAttributes::wireframe; | |
529 | break; | |
530 | case 1: | |
531 | drawingStyle = G4VisAttributes::solid; | |
532 | break; | |
533 | default: | |
534 | if (fVerbose > 0) | |
3c7cd15a | 535 | G4cout << "TG4VisManager::Gsatt() FILL usage :" << G4endl |
536 | << " The FILL values you can supply are only :" << G4endl | |
537 | << "+/- 1 : forces wireframe drawing (default)" << G4endl | |
538 | << "+/- 2 : forces solid drawing" << G4endl | |
2817d3e2 | 539 | << "other values sets the drawing style to solid" |
3c7cd15a | 540 | << G4endl; |
2817d3e2 | 541 | drawingStyle = G4VisAttributes::solid; |
542 | } | |
543 | } | |
544 | // Register vis. attributes | |
545 | newVisAttributes->SetVisibility(isVisible); | |
546 | newVisAttributes->SetDaughtersInvisible(isDaughtersInvisible); | |
547 | newVisAttributes->SetColour(red,green,blue); | |
548 | newVisAttributes->SetLineStyle(lineStyle); | |
549 | newVisAttributes->SetLineWidth(lineWidth); | |
550 | if (drawingStyle == G4VisAttributes::wireframe) | |
551 | newVisAttributes->SetForceWireframe(isForceDrawingStyle); | |
552 | if (drawingStyle == G4VisAttributes::solid) | |
553 | newVisAttributes->SetForceSolid(isForceDrawingStyle); | |
554 | ||
555 | lv->SetVisAttributes(newVisAttributes); | |
556 | } | |
557 | ||
558 | //----------------------------------------------------------------- | |
559 | // functions for drawing | |
560 | //----------------------------------------------------------------- | |
561 | ||
72095f7c | 562 | |
563 | //_____________________________________________________________________________ | |
2817d3e2 | 564 | void TG4VisManager::DrawOneSpec(const char* name) |
565 | { | |
566 | // Function called when one double-clicks on a volume name | |
567 | // in a TPaveLabel drawn by Gdtree | |
568 | // --- | |
569 | ||
570 | G4cout << "TG4VisManager::DrawOneSpec() Not yet implemented"; | |
571 | } | |
572 | ||
573 | ||
72095f7c | 574 | //_____________________________________________________________________________ |
2817d3e2 | 575 | void TG4VisManager::SetColors() |
576 | { | |
577 | // Function for setting default volume colours | |
578 | // --- | |
579 | ||
580 | G4LogicalVolumeStore* pLVStore = G4LogicalVolumeStore::GetInstance(); | |
581 | const G4LogicalVolume* pLV = 0; | |
582 | if (!pLVStore) | |
583 | { | |
584 | TG4Globals::Warning("TG4VisManager::SetColors : Ignored, geometry not built."); | |
585 | return; | |
586 | } | |
587 | // parse the LV tree and set colours according to material density | |
765c040d | 588 | for (G4int i=0; i<pLVStore->size(); i++) |
2817d3e2 | 589 | { |
765c040d | 590 | pLV = (*pLVStore)[i]; |
3c7cd15a | 591 | // G4cout << "VOLUME : " << pLV->GetName() << G4endl; |
2817d3e2 | 592 | const G4Material* pMaterial = pLV->GetMaterial(); |
b15a2a93 | 593 | const G4State kState = pMaterial->GetState(); |
2817d3e2 | 594 | G4double density = (pMaterial->GetDensity())*cm3/g; |
595 | G4String nState = "Undefined"; | |
596 | G4int colour = 1; //black by default | |
597 | G4double luminosity = 0.; | |
b15a2a93 | 598 | if (kState == kStateUndefined) |
2817d3e2 | 599 | { |
600 | nState = "Undefined"; | |
601 | } | |
b15a2a93 | 602 | if (kState == kStateSolid) |
2817d3e2 | 603 | { |
604 | nState = "Solid"; | |
605 | if (density < 2) | |
606 | { | |
607 | colour = 17; //red | |
608 | luminosity = 25 - 25*density/2; | |
609 | } | |
610 | else if (density < 3) | |
611 | { | |
612 | colour = 117; //blue | |
613 | luminosity = 25 - 25*(density-2); | |
614 | } | |
615 | else if (density < 10) | |
616 | { | |
617 | colour = 67; //green | |
618 | luminosity = 25 - 25*(density-5)/5; | |
619 | } | |
620 | else if (density < 15) | |
621 | { | |
622 | colour = 92; //cyan | |
623 | luminosity = 25 - 25*(density-10)/5; | |
624 | } | |
625 | else if (density < 20) | |
626 | { | |
627 | colour = 8; //black | |
628 | luminosity = 9 - 9*(density-15)/5; | |
629 | } | |
630 | } | |
b15a2a93 | 631 | if (kState == kStateLiquid) |
2817d3e2 | 632 | { |
633 | nState = "Liquid"; | |
634 | colour = 142; //violet | |
635 | luminosity = 25 - 25*density/2; | |
636 | } | |
b15a2a93 | 637 | if (kState == kStateGas) |
2817d3e2 | 638 | { |
639 | nState = "Gas"; | |
640 | if (density < 0.001) {colour = 42;} //yellow | |
641 | else if (density < 0.002) {colour = 27;} //light red | |
642 | else if (density < 0.003) {colour = 77;} //light green | |
643 | else {colour = 102;} //light cyan | |
644 | luminosity = 0; | |
645 | } | |
646 | if (luminosity < 0) luminosity=0; | |
647 | colour += (G4int)luminosity; | |
648 | // Setting the corresponding colour | |
649 | Gsatt(pLV->GetName(),"COLO",colour); | |
650 | } | |
651 | } | |
652 | ||
653 | ||
72095f7c | 654 | //_____________________________________________________________________________ |
2817d3e2 | 655 | void TG4VisManager::Gsatt(const char* name, const char* att, Int_t val) |
656 | { | |
657 | // Geant3 description : | |
658 | // | |
659 | // NAME Volume name | |
660 | // IOPT Name of the attribute to be set | |
661 | // IVAL Value to which the attribute is to be set | |
662 | // | |
663 | // name= "*" stands for all the volumes. | |
664 | // iopt can be chosen among the following : | |
665 | // kWORK, kSEEN, kLSTY, kLWID, kCOLO, kFILL, kSET, kDET, kDTYP | |
666 | // --- | |
667 | ||
668 | G4int ival = val; | |
669 | G4LogicalVolume* lv = 0; | |
670 | G4RWTPtrOrderedVector<G4LogicalVolume> lvList; | |
671 | G4String sname(name), | |
672 | satt(att); | |
673 | ||
674 | // seek for known attributes | |
5317affd | 675 | TG4G3Attribute attribute = kUNKNOWN; |
2817d3e2 | 676 | if (CaseInsensitiveEqual(att,"WORK")) |
677 | { | |
678 | G4String message = "TG4VisManager::Gsatt: G3Attribute "; | |
679 | message += satt + " not used in G4"; | |
680 | TG4Globals::Warning(message); | |
681 | return; | |
682 | } | |
683 | if (CaseInsensitiveEqual(att,"SEEN")) attribute = kSEEN; | |
684 | if (CaseInsensitiveEqual(att,"LSTY")) attribute = kLSTY; | |
685 | if (CaseInsensitiveEqual(att,"LWID")) attribute = kLWID; | |
686 | if (CaseInsensitiveEqual(att,"COLO")) attribute = kCOLO; | |
687 | if (CaseInsensitiveEqual(att,"FILL")) attribute = kFILL; | |
688 | if (CaseInsensitiveEqual(att,"SET")) | |
689 | { | |
690 | G4String message = "TG4VisManager::Gsatt: G3Attribute "; | |
691 | message += satt + " not used in G4"; | |
692 | TG4Globals::Warning(message); | |
693 | return; | |
694 | } | |
695 | if (CaseInsensitiveEqual(att,"DET")) | |
696 | { | |
697 | G4String message = "TG4VisManager::Gsatt: G3Attribute "; | |
698 | message += satt + " not used in G4"; | |
699 | TG4Globals::Warning(message); | |
700 | return; | |
701 | } | |
702 | if (CaseInsensitiveEqual(att,"DTYP")) | |
703 | { | |
704 | G4String message = "TG4VisManager::Gsatt: G3Attribute "; | |
705 | message += satt + " not used in G4"; | |
706 | TG4Globals::Warning(message); | |
707 | return; | |
708 | } | |
709 | if (attribute == kUNKNOWN) | |
710 | { | |
711 | G4String message = "TG4VisManager::Gsatt: G3Attribute "; | |
712 | message += satt + " unknown"; | |
713 | TG4Globals::Warning(message); | |
714 | return; | |
715 | } | |
716 | G4bool doForDaughters(false), // tree iterator flag | |
717 | doForAll(false), // activated if NAME is "*" | |
718 | topVisible(false); // activated for kSEEN/-2 | |
719 | if (sname == "*") doForAll = true; | |
720 | if (val < 0 && sname!="*") doForDaughters = true; | |
721 | if (attribute==kSEEN && val==-2) topVisible = true; | |
722 | ||
723 | // parse all the tree | |
724 | if (doForAll) | |
725 | { | |
726 | G4LogicalVolumeStore* pLVStore = G4LogicalVolumeStore::GetInstance(); | |
765c040d | 727 | for (G4int i=0; i<pLVStore->size(); i++) |
2817d3e2 | 728 | { |
765c040d | 729 | lv = (*pLVStore)[i]; |
2817d3e2 | 730 | SetG4Attribute(lv,attribute,ival); |
731 | } | |
732 | return; | |
733 | } | |
734 | ||
735 | // get the logical volume pointer corresponding to NAME | |
736 | lvList = GetLVList(name); | |
737 | if (lvList.isEmpty()) | |
738 | { | |
739 | G4String message = "TG4VisManager::Gsatt(): Ignored\n"; | |
740 | message += " Logical volume " + sname + " has not been found.\n"; | |
741 | TG4Globals::Warning(message); | |
742 | return; | |
743 | } | |
744 | // set attribute for all descendents | |
745 | if (doForDaughters) | |
746 | { | |
747 | for (G4int i=0; i<lvList.entries(); i++) | |
748 | { | |
749 | lv = lvList[i]; | |
750 | SetAtt4Daughters(lv,attribute,ival); | |
751 | } | |
752 | } | |
753 | else | |
754 | { | |
755 | for (G4int i=0; i<lvList.entries(); i++) | |
756 | { | |
757 | lv = lvList[i]; | |
758 | SetG4Attribute(lv,attribute,ival); | |
759 | } | |
760 | } | |
761 | if (topVisible) | |
762 | { | |
763 | for (G4int i=0; i<lvList.entries(); i++) | |
764 | { | |
765 | lv = lvList[i]; | |
766 | SetG4Attribute(lv,attribute,1); | |
767 | } | |
768 | } | |
769 | } | |
770 | ||
771 | ||
72095f7c | 772 | //_____________________________________________________________________________ |
2817d3e2 | 773 | void TG4VisManager::Gdraw(const char *name,Float_t theta, Float_t phi, Float_t psi, |
774 | Float_t u0,Float_t v0,Float_t ul,Float_t vl) | |
775 | { | |
776 | // Draw the physical volume NAME and all descendents; | |
777 | // Mandatory : the graphics system, scene and view must be | |
778 | // initialized, e.g. "/vis~/create_view/new_graphics_system OGLSX"; | |
779 | // Any call of Gdraw() will use the current graphics system and | |
780 | // the current window. | |
781 | // The result will be a centered view drawing of the designated volume, | |
782 | // lights moving with camera, viewpoint direction given by theta/phi | |
783 | // and rotation on the screen given by psi; | |
784 | // The u0, v0, ul, vl factors are ignored since the object will be | |
785 | // automatically centered and will be confortable in the window | |
786 | // at any viewing angle. | |
787 | // | |
788 | // check if G4 graphics is ready for drawing | |
789 | // --- | |
790 | ||
791 | G4VVisManager* pVVisManager = G4VVisManager::GetConcreteInstance(); | |
792 | if (!pVVisManager) { | |
793 | TG4Globals::Warning( | |
794 | "TG4VisManager::Gdraw: Ignored - No graphics driver is built. "); | |
795 | return; | |
796 | } | |
797 | if (NeedSetColours()) | |
798 | { | |
799 | SetColors(); | |
800 | SetColourFlag(false); | |
801 | } | |
802 | ||
b15a2a93 | 803 | const G4double kRad = M_PI/180.; |
2817d3e2 | 804 | G4RWTPtrOrderedVector<G4VPhysicalVolume> pvList; |
805 | G4String sname(name); | |
806 | G4bool successful = false; | |
807 | ||
808 | pvList = GetPVList(name); | |
809 | if (pvList.isEmpty()) | |
810 | { | |
811 | G4String message = "TG4VisManager::Gdraw() :\n"; | |
812 | message += "Volume " + sname + " not found. Bailing out"; | |
813 | TG4Globals::Warning(message); | |
814 | return; | |
815 | } | |
816 | ||
817 | G4VPhysicalVolume *pPV = 0; | |
818 | ||
819 | // clear the current scene if not empty | |
820 | if (!fpScene->IsEmpty()) fpScene->Clear(); | |
821 | ||
822 | // create and add object's model list to the runtime-duration model | |
823 | // list and draw it | |
824 | // (it is deleted in the VisManager destructor within | |
825 | // all the RWTPtrOrderedVectors of the scene) | |
826 | for (G4int i=0; i<pvList.entries(); i++) | |
827 | { | |
828 | pPV = pvList[i]; | |
829 | G4LogicalVolume* pLV = pPV->GetLogicalVolume(); | |
830 | G4VSolid* pSolid = pLV->GetSolid(); | |
2817d3e2 | 831 | successful = fpScene->AddRunDurationModel(new G4PhysicalVolumeModel(pPV)); |
832 | if (!successful) | |
833 | { | |
834 | G4String message = "TG4VisManager::Gdraw() Could not add "; | |
835 | message += pPV->GetName() + " to the drawing list. Probably "; | |
836 | message += "it is already in the list."; | |
837 | TG4Globals::Warning(message); | |
838 | } | |
839 | } | |
840 | // get the standard target point of the scene | |
b15a2a93 | 841 | const G4Point3D kTargetPoint = fpScene->GetStandardTargetPoint(); |
2817d3e2 | 842 | |
843 | // set the viewpoint and the rotation on the screen | |
b15a2a93 | 844 | G4Vector3D viewpointDirection(sin(theta*kRad)*cos(phi*kRad), |
845 | sin(theta*kRad)*sin(phi*kRad), cos(theta*kRad)); | |
846 | G4Vector3D upVector(sin(psi*kRad), cos(psi*kRad),0); | |
2817d3e2 | 847 | |
848 | // set and register view parameters to the viewer | |
849 | ||
850 | fVP.SetLightsMoveWithCamera(true); | |
851 | fVP.SetViewGeom(); | |
852 | fVP.UnsetViewHits(); | |
853 | fVP.UnsetViewDigis(); | |
854 | fVP.SetNoOfSides(48); | |
b15a2a93 | 855 | fVP.SetCurrentTargetPoint(kTargetPoint); |
2817d3e2 | 856 | fVP.SetViewpointDirection(viewpointDirection); |
857 | fVP.SetUpVector(upVector); | |
858 | fVP.SetDensityCulling(true); | |
859 | fpViewer->SetViewParameters(fVP); | |
860 | ||
861 | if (IsValidView()) | |
862 | { | |
863 | fpSceneHandler->SetScene(fpScene); | |
864 | fpSceneHandler->SetCurrentViewer(fpViewer); | |
865 | fpViewer->DrawView(); | |
866 | fpViewer->ShowView(); | |
867 | } | |
868 | else TG4Globals::Warning( | |
869 | "TG4VisManager::Gdraw: Ignored - Failed to register volume"); | |
870 | } | |
871 | #endif //G4VIS_USE |