Geant4 Cross Reference |
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 #include "G4RayTracerSceneHandler.hh" 29 30 #include "G4VisManager.hh" 31 32 #define G4warn G4cout 33 34 G4int G4RayTracerSceneHandler::fSceneIdCount = 0; 35 36 G4RayTracerSceneHandler::G4RayTracerSceneHandler(G4VGraphicsSystem& system, 37 const G4String& name) 38 : G4VSceneHandler(system, fSceneIdCount++, name) 39 { 40 // Keep vis manager happy when someone opens a ray tracer with "/vis/open 41 // RayTracer" but uses the ray tracer with "/vis/rayTracer" commands 42 // before creating any scenes, for example, instead of using 43 // "/vis/drawVolume"... 44 G4VisManager* visManager = G4VisManager::GetInstance(); 45 if(visManager) { 46 G4Scene* pScene = visManager->GetCurrentScene(); 47 if (!pScene) { 48 // Create new scene like /vis/scene/create... 49 fpScene = new G4Scene("dummy-ray-tracer-scene"); 50 // Add dummy run-duration model to avoid world being added and 51 // notifyHandler being invoked... 52 fpScene->AddWorldIfEmpty(); 53 // Add to vis manager list; ownership thereby passes to vis manager... 54 visManager->SetSceneList().push_back(fpScene); 55 // ...and make current... 56 visManager->SetCurrentScene(fpScene); 57 } 58 } 59 } 60 61 G4RayTracerSceneHandler::~G4RayTracerSceneHandler() 62 {} 63 64 void G4RayTracerSceneHandler::ClearStore() 65 { 66 fSceneVisAttsMap.clear(); 67 } 68 69 G4bool G4RayTracerSceneHandler::PathLessThan::operator() 70 (const G4ModelingParameters::PVPointerCopyNoPath& a, 71 const G4ModelingParameters::PVPointerCopyNoPath& b) const 72 { 73 if (a.size() != b.size()) return a.size() < b.size(); 74 auto ia = a.begin(); 75 auto ib = b.begin(); 76 for (; ia != a.end(); ++ia, ++ib) { 77 if (ia->GetPVPointer() < ib->GetPVPointer()) return true; 78 if (ia->GetPVPointer() > ib->GetPVPointer()) return false; 79 // Pointers equal 80 if (ia->GetCopyNo() < ib->GetCopyNo()) return true; 81 if (ia->GetCopyNo() > ib->GetCopyNo()) return false; 82 // Both pointers and copy no are equal - continue 83 } 84 // Equality 85 return false; 86 } 87 88 void G4RayTracerSceneHandler::BuildVisAttsMap (const G4VSolid&) 89 { 90 // Build map of vis attributes 91 92 G4PhysicalVolumeModel* fpPVModel = dynamic_cast<G4PhysicalVolumeModel*>(fpModel); 93 if (fpPVModel) { 94 G4ModelingParameters::PVPointerCopyNoPath temp; 95 for (const auto& nodeID: fpPVModel->GetFullPVPath()) { 96 // Build an element from the nodeid. 97 temp.push_back(G4ModelingParameters::PVPointerCopyNo(nodeID.GetPhysicalVolume(),nodeID.GetCopyNo())); 98 } 99 const G4VisAttributes* pVisAtts = fpVisAttribs; 100 if (!pVisAtts) { 101 // Shouldn't happen. 102 if (G4VisManager::GetInstance()->GetVerbosity() >= G4VisManager::warnings) { 103 G4warn << 104 "WARNING: G4RayTracerSceneHandler::BuildVisAttsMap: null vis atts pointer." 105 "\n Using a default vis atts." 106 << G4endl; 107 } 108 static const G4VisAttributes defaultVisAtts; 109 pVisAtts = &defaultVisAtts; 110 } 111 // Copy vis atts into the vis atts map 112 fSceneVisAttsMap[temp] = *pVisAtts; 113 } 114 } 115 116