Geant4 Cross Reference

Cross-Referencing   Geant4
Geant4/visualization/Tree/src/G4ASCIITreeSceneHandler.cc

Version: [ ReleaseNotes ] [ 1.0 ] [ 1.1 ] [ 2.0 ] [ 3.0 ] [ 3.1 ] [ 3.2 ] [ 4.0 ] [ 4.0.p1 ] [ 4.0.p2 ] [ 4.1 ] [ 4.1.p1 ] [ 5.0 ] [ 5.0.p1 ] [ 5.1 ] [ 5.1.p1 ] [ 5.2 ] [ 5.2.p1 ] [ 5.2.p2 ] [ 6.0 ] [ 6.0.p1 ] [ 6.1 ] [ 6.2 ] [ 6.2.p1 ] [ 6.2.p2 ] [ 7.0 ] [ 7.0.p1 ] [ 7.1 ] [ 7.1.p1 ] [ 8.0 ] [ 8.0.p1 ] [ 8.1 ] [ 8.1.p1 ] [ 8.1.p2 ] [ 8.2 ] [ 8.2.p1 ] [ 8.3 ] [ 8.3.p1 ] [ 8.3.p2 ] [ 9.0 ] [ 9.0.p1 ] [ 9.0.p2 ] [ 9.1 ] [ 9.1.p1 ] [ 9.1.p2 ] [ 9.1.p3 ] [ 9.2 ] [ 9.2.p1 ] [ 9.2.p2 ] [ 9.2.p3 ] [ 9.2.p4 ] [ 9.3 ] [ 9.3.p1 ] [ 9.3.p2 ] [ 9.4 ] [ 9.4.p1 ] [ 9.4.p2 ] [ 9.4.p3 ] [ 9.4.p4 ] [ 9.5 ] [ 9.5.p1 ] [ 9.5.p2 ] [ 9.6 ] [ 9.6.p1 ] [ 9.6.p2 ] [ 9.6.p3 ] [ 9.6.p4 ] [ 10.0 ] [ 10.0.p1 ] [ 10.0.p2 ] [ 10.0.p3 ] [ 10.0.p4 ] [ 10.1 ] [ 10.1.p1 ] [ 10.1.p2 ] [ 10.1.p3 ] [ 10.2 ] [ 10.2.p1 ] [ 10.2.p2 ] [ 10.2.p3 ] [ 10.3 ] [ 10.3.p1 ] [ 10.3.p2 ] [ 10.3.p3 ] [ 10.4 ] [ 10.4.p1 ] [ 10.4.p2 ] [ 10.4.p3 ] [ 10.5 ] [ 10.5.p1 ] [ 10.6 ] [ 10.6.p1 ] [ 10.6.p2 ] [ 10.6.p3 ] [ 10.7 ] [ 10.7.p1 ] [ 10.7.p2 ] [ 10.7.p3 ] [ 10.7.p4 ] [ 11.0 ] [ 11.0.p1 ] [ 11.0.p2 ] [ 11.0.p3, ] [ 11.0.p4 ] [ 11.1 ] [ 11.1.1 ] [ 11.1.2 ] [ 11.1.3 ] [ 11.2 ] [ 11.2.1 ] [ 11.2.2 ] [ 11.3.0 ]

  1 //
  2 // ********************************************************************
  3 // * License and Disclaimer                                           *
  4 // *                                                                  *
  5 // * The  Geant4 software  is  copyright of the Copyright Holders  of *
  6 // * the Geant4 Collaboration.  It is provided  under  the terms  and *
  7 // * conditions of the Geant4 Software License,  included in the file *
  8 // * LICENSE and available at  http://cern.ch/geant4/license .  These *
  9 // * include a list of copyright holders.                             *
 10 // *                                                                  *
 11 // * Neither the authors of this software system, nor their employing *
 12 // * institutes,nor the agencies providing financial support for this *
 13 // * work  make  any representation or  warranty, express or implied, *
 14 // * regarding  this  software system or assume any liability for its *
 15 // * use.  Please see the license in the file  LICENSE  and URL above *
 16 // * for the full disclaimer and the limitation of liability.         *
 17 // *                                                                  *
 18 // * This  code  implementation is the result of  the  scientific and *
 19 // * technical work of the GEANT4 collaboration.                      *
 20 // * By using,  copying,  modifying or  distributing the software (or *
 21 // * any work based  on the software)  you  agree  to acknowledge its *
 22 // * use  in  resulting  scientific  publications,  and indicate your *
 23 // * acceptance of all terms of the Geant4 Software license.          *
 24 // ********************************************************************
 25 //
 26 //
 27 //
 28 // 
 29 // John Allison  5th April 2001
 30 // A scene handler to dump geometry hierarchy.
 31 // Based on a provisional G4ASCIITreeGraphicsScene (was in modeling).
 32 
 33 #include "G4ASCIITreeSceneHandler.hh"
 34 
 35 #include "G4ASCIITree.hh"
 36 #include "G4ASCIITreeMessenger.hh"
 37 #include "G4VSolid.hh"
 38 #include "G4PhysicalVolumeModel.hh"
 39 #include "G4VPhysicalVolume.hh"
 40 #include "G4LogicalVolume.hh"
 41 #include "G4VPVParameterisation.hh"
 42 #include "G4Polyhedron.hh"
 43 #include "G4UnitsTable.hh"
 44 #include "G4Material.hh"
 45 #include "G4Scene.hh"
 46 #include "G4ModelingParameters.hh"
 47 #include "G4PhysicalVolumeMassScene.hh"
 48 #include "G4VSensitiveDetector.hh"
 49 #include "G4VReadOutGeometry.hh"
 50 #include "G4TransportationManager.hh"
 51 #include "G4AttCheck.hh"
 52 #include "G4AttValue.hh"
 53 
 54 G4ASCIITreeSceneHandler::G4ASCIITreeSceneHandler
 55 (G4VGraphicsSystem& system,
 56  const G4String& name):
 57   G4VTreeSceneHandler(system, name),
 58   fpOutFile(0),
 59   fpLastPV(0),
 60   fLastCopyNo(-99),
 61   fLastNonSequentialCopyNo(-99)
 62 {}
 63 
 64 G4ASCIITreeSceneHandler::~G4ASCIITreeSceneHandler () {}
 65 
 66 void G4ASCIITreeSceneHandler::BeginModeling () {
 67 
 68   G4VTreeSceneHandler::BeginModeling ();  // To re-use "culling off" code.
 69 
 70   const G4ASCIITree* pSystem = (G4ASCIITree*)GetGraphicsSystem();
 71   const G4String& outFileName = pSystem -> GetOutFileName();
 72   if (outFileName == "G4cout") {
 73     fpOutFile = &G4cout;
 74   } else {
 75     fOutFile.open (outFileName);
 76     fpOutFile = &fOutFile;
 77   }
 78 
 79   static G4bool firstTime = true;
 80   if (firstTime) {
 81     firstTime = false;
 82     G4cout << "G4ASCIITreeSceneHandler::BeginModeling: writing to ";
 83     if (outFileName == "G4cout") {
 84       G4cout << "G4 standard output (G4cout)";
 85     } else {
 86       G4cout << "file \"" << outFileName << "\"";
 87     }
 88     G4cout << G4endl;
 89 
 90     WriteHeader (G4cout); G4cout << G4endl;
 91   }
 92   
 93   if (outFileName != "G4cout") {
 94     WriteHeader (fOutFile); fOutFile << std::endl;
 95   }
 96 }
 97 
 98 void G4ASCIITreeSceneHandler::WriteHeader (std::ostream& os)
 99 {
100   const G4ASCIITree* pSystem = (G4ASCIITree*)GetGraphicsSystem();
101   const G4int verbosity = pSystem->GetVerbosity();
102   const G4int detail = verbosity % 10;
103   os << "#  Set verbosity with \"/vis/ASCIITree/verbose <verbosity>\":";
104   for (size_t i = 0;
105        i < G4ASCIITreeMessenger::fVerbosityGuidance.size(); ++i) {
106     os << "\n#  " << G4ASCIITreeMessenger::fVerbosityGuidance[i];
107   }
108   os << "\n#  Now printing with verbosity " << verbosity;
109   os << "\n#  Format is: PV:n";
110   if (detail >= 1) os << " / LV (SD,RO)";
111   if (detail >= 2) os << " / Solid(type)";
112   if (detail >= 3) os << ", volume, density";
113   if (detail >= 5) os << ", daughter-subtracted volume and mass";
114   if (detail >= 6) os << ", physical volume dump";
115   if (detail >= 7) os << ", polyhedron dump";
116   os <<
117     "\n#  Abbreviations: PV = Physical Volume,     LV = Logical Volume,"
118     "\n#                 SD = Sensitive Detector,  RO = Read Out Geometry.";
119 }
120 
121 void G4ASCIITreeSceneHandler::EndModeling () {
122   const G4ASCIITree* pSystem = (G4ASCIITree*) GetGraphicsSystem();
123   const G4int verbosity = pSystem->GetVerbosity();
124   const G4int detail = verbosity % 10;
125   const G4String& outFileName = pSystem -> GetOutFileName();
126 
127   // Output left over copy number, if any...
128   if (fLastCopyNo != fLastNonSequentialCopyNo) {
129     if (fLastCopyNo == fLastNonSequentialCopyNo + 1) *fpOutFile << ',';
130     else *fpOutFile << '-';
131     *fpOutFile << fLastCopyNo;
132   }
133   // Output outstanding rest of line, if any...
134   if (!fRestOfLine.str().empty()) *fpOutFile << fRestOfLine.str();
135   fRestOfLine.str("");
136   fpLastPV = 0;
137   fLastPVName.clear();
138   fLastCopyNo = -99;
139   fLastNonSequentialCopyNo = -99;
140 
141   // This detail to G4cout regardless of outFileName...
142   if (detail >= 4) {
143     G4cout << "Calculating mass(es)..." << G4endl;
144     const std::vector<G4Scene::Model>& models = fpScene->GetRunDurationModelList();
145     std::vector<G4Scene::Model>::const_iterator i;
146     for (i = models.begin(); i != models.end(); ++i) {
147       G4PhysicalVolumeModel* pvModel =
148   dynamic_cast<G4PhysicalVolumeModel*>(i->fpModel);
149       if (pvModel) {
150         const G4ModelingParameters* tempMP =
151         pvModel->GetModelingParameters();
152         G4ModelingParameters mp;  // Default - no culling.
153         pvModel->SetModelingParameters (&mp);
154         G4PhysicalVolumeMassScene massScene(pvModel);
155         pvModel->DescribeYourselfTo (massScene);
156         G4double volume = massScene.GetVolume();
157         G4double mass = massScene.GetMass();
158 
159         G4cout << "Overall volume of \""
160         << pvModel->GetTopPhysicalVolume()->GetName()
161         << "\":"
162         << pvModel->GetTopPhysicalVolume()->GetCopyNo()
163         << ", is "
164         << G4BestUnit (volume, "Volume")
165         << " and the daughter-included mass";
166         G4int requestedDepth = pvModel->GetRequestedDepth();
167         if (requestedDepth == G4PhysicalVolumeModel::UNLIMITED) {
168           G4cout << " to unlimited depth";
169         } else {
170           G4cout << ", ignoring daughters at depth "
171           << requestedDepth
172           << " and below,";
173         }
174         G4cout << " is " << G4BestUnit (mass, "Mass")
175         << G4endl;
176         
177         pvModel->SetModelingParameters (tempMP);
178       }
179     }
180   }
181 
182   if (outFileName != "G4cout") {
183     fOutFile.close();
184     G4cout << "Output file \"" << outFileName << "\" closed." << G4endl;
185   }
186   fLVSet.clear();
187   fReplicaSet.clear();
188   G4cout << "G4ASCIITreeSceneHandler::EndModeling" << G4endl;
189   G4VTreeSceneHandler::EndModeling ();  // To re-use "culling off" code.
190 }
191 
192 void G4ASCIITreeSceneHandler::RequestPrimitives(const G4VSolid& solid) {
193   
194   G4PhysicalVolumeModel* pPVModel =
195     dynamic_cast<G4PhysicalVolumeModel*>(fpModel);
196   if (!pPVModel) return;  // Not from a G4PhysicalVolumeModel.
197 
198   // This call comes from a G4PhysicalVolumeModel.  drawnPVPath is
199   // the path of the current drawn (non-culled) volume in terms of
200   // drawn (non-culled) ancestors.  Each node is identified by a
201   // PVNodeID object, which is a physical volume and copy number.  It
202   // is a vector of PVNodeIDs corresponding to the geometry hierarchy
203   // actually selected, i.e., not culled.
204   // The following typedef's already set in header file...
205   //typedef G4PhysicalVolumeModel::G4PhysicalVolumeNodeID PVNodeID;
206   //typedef std::vector<PVNodeID> PVPath;
207   const PVPath& drawnPVPath = pPVModel->GetDrawnPVPath();
208   //G4int currentDepth = pPVModel->GetCurrentDepth();
209   G4VPhysicalVolume* pCurrentPV = pPVModel->GetCurrentPV();
210   const G4String& currentPVName = pCurrentPV->GetName();
211   const G4int currentCopyNo = pCurrentPV->GetCopyNo();
212   G4LogicalVolume* pCurrentLV = pPVModel->GetCurrentLV();
213   G4Material* pCurrentMaterial = pPVModel->GetCurrentMaterial();
214 
215   G4ASCIITree* pSystem = (G4ASCIITree*)GetGraphicsSystem();
216   G4int verbosity = pSystem->GetVerbosity();
217   G4int detail = verbosity % 10;
218 
219   // If verbosity < 10 suppress unnecessary repeated printing.
220   // Repeated simple replicas can always be suppressed.
221   // Paramaterisations can only be suppressed if verbosity < 3, since their
222   // size, density, etc., are in principle different.
223   const G4bool isParameterised = pCurrentPV->GetParameterisation();
224   const G4bool isSimpleReplica = pCurrentPV->IsReplicated() && !isParameterised;
225   const G4bool isAmenableToSupression =
226   (verbosity < 10 && isSimpleReplica) || (verbosity < 3 && isParameterised);
227   if (isAmenableToSupression) {
228     // See if this has been found before with same mother LV...
229     PVPath::const_reverse_iterator thisID = drawnPVPath.rbegin();
230     PVPath::const_reverse_iterator motherID = ++drawnPVPath.rbegin();
231     G4bool ignore = false;
232     for (ReplicaSetIterator i = fReplicaSet.begin(); i != fReplicaSet.end();
233          ++i) {
234       if (i->back().GetPhysicalVolume()->GetLogicalVolume() ==
235           thisID->GetPhysicalVolume()->GetLogicalVolume()) {
236         // For each one previously found (if more than one, they must
237         // have different mothers)...
238         // To avoid compilation errors on VC++ .Net 7.1...
239         // Previously:
240         //   PVNodeID previousMotherID = ++(i->rbegin());
241         // (Should that have been: PVNodeID::const_iterator previousMotherID?)
242         // Replace
243         //   previousMotherID == i->rend()
244         // by
245         //   i->size() <= 1
246         // Replace
247         //   previousMotherID != i->rend()
248         // by
249         //   i->size() > 1
250         // Replace
251         //   previousMotherID->
252         // by
253         //   (*i)[i->size() - 2].
254         if (motherID == drawnPVPath.rend() &&
255             i->size() <= 1)
256           ignore = true;  // Both have no mother.
257         if (motherID != drawnPVPath.rend() &&
258             i->size() > 1 &&
259             motherID->GetPhysicalVolume()->GetLogicalVolume() ==
260             (*i)[i->size() - 2].GetPhysicalVolume()->GetLogicalVolume())
261           ignore = true;  // Same mother LV...
262       }
263     }
264     if (ignore) {
265       pPVModel->CurtailDescent();
266       return;
267     }
268   }
269 
270   // Now suppress printing for volumes with the same name but different
271   // copy number - but not if they are parameterisations that have not been
272   // taken out above (those that are "amenable to suppression" will have been
273   // taken out).
274   if (verbosity < 10 && !isParameterised &&
275       currentPVName == fLastPVName &&
276       currentCopyNo != fLastCopyNo) {
277     // Check...
278     if (isAmenableToSupression) {
279       G4Exception("G4ASCIITreeSceneHandler::RequestPrimitives",
280       "vistree0001",
281       JustWarning,
282       "Volume amenable to suppressed printing unexpected");
283     }
284     // Check mothers are identical...
285     else if (pCurrentLV == (fpLastPV? fpLastPV->GetLogicalVolume(): 0)) {
286       if (currentCopyNo != fLastCopyNo + 1) {
287   // Non-sequential copy number...
288   *fpOutFile << ',' << currentCopyNo;
289   fLastNonSequentialCopyNo = currentCopyNo;
290       }
291       fLastCopyNo = currentCopyNo;
292       pPVModel->CurtailDescent();
293       return;
294     }
295   }
296   fpLastPV = pCurrentPV;
297 
298   // High verbosity or a new or otherwise non-amenable volume...
299   // Output copy number, if any, from previous invocation...
300   if (fLastCopyNo != fLastNonSequentialCopyNo) {
301     if (fLastCopyNo == fLastNonSequentialCopyNo + 1) *fpOutFile << ',';
302     else *fpOutFile << '-';
303     *fpOutFile << fLastCopyNo;
304   }
305   // Output rest of line, if any, from previous invocation...
306   if (!fRestOfLine.str().empty()) *fpOutFile << fRestOfLine.str();
307   fRestOfLine.str("");
308   fLastPVName = currentPVName;
309   fLastCopyNo = currentCopyNo;
310   fLastNonSequentialCopyNo = currentCopyNo;
311   // Start next line...
312   // Indent according to drawn path depth...
313   for (size_t i = 0; i < drawnPVPath.size(); i++ ) *fpOutFile << "  ";
314   *fpOutFile << "\"" << currentPVName
315        << "\":" << currentCopyNo;
316 
317   if (pCurrentPV->IsReplicated()) {
318     if (verbosity < 10) {
319       // Add printing for replicas (when replicas are ignored)...
320       EAxis axis;
321       G4int nReplicas;
322       G4double width;
323       G4double offset;
324       G4bool consuming;
325       pCurrentPV->GetReplicationData(axis,nReplicas,width,offset,consuming);
326       G4VPVParameterisation* pP = pCurrentPV->GetParameterisation();
327       if (pP) {
328   if (detail < 3) {
329     fReplicaSet.insert(drawnPVPath);
330     if (nReplicas > 2) fRestOfLine << '-';
331     else fRestOfLine << ',';
332     fRestOfLine << nReplicas - 1
333           << " (" << nReplicas << " parametrised volumes)";
334   }
335       }
336       else {
337   fReplicaSet.insert(drawnPVPath);
338   if (nReplicas > 2) fRestOfLine << '-';
339   else fRestOfLine << ',';
340   fRestOfLine << nReplicas - 1
341         << " (" << nReplicas << " replicas)";
342       }
343     }
344   } else {
345     if (fLVSet.find(pCurrentLV) != fLVSet.end()) {
346       if (verbosity <  10) {
347   // Add printing for repeated LV (if it has daughters)...
348   if (pCurrentLV->GetNoDaughters()) fRestOfLine << " (repeated LV)";
349   // ...and curtail descent.
350   pPVModel->CurtailDescent();
351       }
352     }
353   }
354 
355   if (detail >= 1) {
356     fRestOfLine << " / \""
357     << pCurrentLV->GetName() << "\"";
358     G4VSensitiveDetector* sd = pCurrentLV->GetSensitiveDetector();
359     if (sd) {
360       fRestOfLine << " (SD=\"" << sd->GetName() << "\"";
361       G4VReadOutGeometry* roGeom = sd->GetROgeometry();
362       if (roGeom) {
363   fRestOfLine << ",RO=\"" << roGeom->GetName() << "\"";
364       }
365       fRestOfLine << ")";
366     }
367   }
368 
369   if (detail >= 2) {
370     fRestOfLine << " / \""
371          << solid.GetName()
372          << "\"("
373          << solid.GetEntityType() << ")";
374   }
375 
376   if (detail >= 3) {
377     fRestOfLine << ", "
378     << G4BestUnit(((G4VSolid&)solid).GetCubicVolume(),"Volume")
379     << ", ";
380     if (pCurrentMaterial) {
381       fRestOfLine
382   << G4BestUnit(pCurrentMaterial->GetDensity(), "Volumic Mass")
383   << " (" << pCurrentMaterial->GetName() << ")";
384     } else {
385       fRestOfLine << "(No material)";
386     }
387   }
388 
389   if (detail >= 5) {
390     if (pCurrentMaterial) {
391       G4Material* pMaterial = const_cast<G4Material*>(pCurrentMaterial);
392       // ...and find daughter-subtracted mass...
393       G4double daughter_subtracted_mass = pCurrentLV->GetMass
394   (pCurrentPV->IsParameterised(),  // Force if parametrised.
395    false,  // Do not propagate - calculate for this volume minus
396    // volume of daughters.
397    pMaterial);
398       G4double daughter_subtracted_volume =
399   daughter_subtracted_mass / pCurrentMaterial->GetDensity();
400       fRestOfLine << ", "
401       << G4BestUnit(daughter_subtracted_volume,"Volume")
402       << ", "
403       << G4BestUnit(daughter_subtracted_mass,"Mass");
404     }
405   }
406 
407   if (detail >= 6) {
408     std::vector<G4AttValue>* attValues = pPVModel->CreateCurrentAttValues();
409     const std::map<G4String,G4AttDef>* attDefs = pPVModel->GetAttDefs();
410     fRestOfLine << '\n' << G4AttCheck(attValues, attDefs);
411     delete attValues;
412   }
413 
414   if (detail >= 7) {
415     G4Polyhedron* polyhedron = solid.GetPolyhedron();
416     fRestOfLine << "\nLocal polyhedron coordinates:\n" << *polyhedron;
417     const G4Transform3D& transform = pPVModel->GetCurrentTransform();
418     polyhedron->Transform(transform);
419     fRestOfLine << "\nGlobal polyhedron coordinates:\n" << *polyhedron;
420   }
421 
422   if (fLVSet.find(pCurrentLV) == fLVSet.end()) {
423     fLVSet.insert(pCurrentLV);  // Record new logical volume.
424   }
425 
426   fRestOfLine << std::endl;
427 
428   return;
429 }
430