Gabriele Cosmo
116 lines
4.5 KiB
C++
116 lines
4.5 KiB
C++
//
|
|
// ********************************************************************
|
|
// * License and Disclaimer *
|
|
// * *
|
|
// * The Geant4 software is copyright of the Copyright Holders of *
|
|
// * the Geant4 Collaboration. It is provided under the terms and *
|
|
// * conditions of the Geant4 Software License, included in the file *
|
|
// * LICENSE and available at http://cern.ch/geant4/license . These *
|
|
// * include a list of copyright holders. *
|
|
// * *
|
|
// * Neither the authors of this software system, nor their employing *
|
|
// * institutes,nor the agencies providing financial support for this *
|
|
// * work make any representation or warranty, express or implied, *
|
|
// * regarding this software system or assume any liability for its *
|
|
// * use. Please see the license in the file LICENSE and URL above *
|
|
// * for the full disclaimer and the limitation of liability. *
|
|
// * *
|
|
// * This code implementation is the result of the scientific and *
|
|
// * technical work of the GEANT4 collaboration. *
|
|
// * By using, copying, modifying or distributing the software (or *
|
|
// * any work based on the software) you agree to acknowledge its *
|
|
// * use in resulting scientific publications, and indicate your *
|
|
// * acceptance of all terms of the Geant4 Software license. *
|
|
// ********************************************************************
|
|
//
|
|
//
|
|
|
|
#include "G4RayTracerSceneHandler.hh"
|
|
|
|
#include "G4VisManager.hh"
|
|
|
|
#define G4warn G4cout
|
|
|
|
G4int G4RayTracerSceneHandler::fSceneIdCount = 0;
|
|
|
|
G4RayTracerSceneHandler::G4RayTracerSceneHandler(G4VGraphicsSystem& system,
|
|
const G4String& name)
|
|
: G4VSceneHandler(system, fSceneIdCount++, name)
|
|
{
|
|
// Keep vis manager happy when someone opens a ray tracer with "/vis/open
|
|
// RayTracer" but uses the ray tracer with "/vis/rayTracer" commands
|
|
// before creating any scenes, for example, instead of using
|
|
// "/vis/drawVolume"...
|
|
G4VisManager* visManager = G4VisManager::GetInstance();
|
|
if(visManager) {
|
|
G4Scene* pScene = visManager->GetCurrentScene();
|
|
if (!pScene) {
|
|
// Create new scene like /vis/scene/create...
|
|
fpScene = new G4Scene("dummy-ray-tracer-scene");
|
|
// Add dummy run-duration model to avoid world being added and
|
|
// notifyHandler being invoked...
|
|
fpScene->AddWorldIfEmpty();
|
|
// Add to vis manager list; ownership thereby passes to vis manager...
|
|
visManager->SetSceneList().push_back(fpScene);
|
|
// ...and make current...
|
|
visManager->SetCurrentScene(fpScene);
|
|
}
|
|
}
|
|
}
|
|
|
|
G4RayTracerSceneHandler::~G4RayTracerSceneHandler()
|
|
{}
|
|
|
|
void G4RayTracerSceneHandler::ClearStore()
|
|
{
|
|
fSceneVisAttsMap.clear();
|
|
}
|
|
|
|
G4bool G4RayTracerSceneHandler::PathLessThan::operator()
|
|
(const G4ModelingParameters::PVPointerCopyNoPath& a,
|
|
const G4ModelingParameters::PVPointerCopyNoPath& b) const
|
|
{
|
|
if (a.size() != b.size()) return a.size() < b.size();
|
|
auto ia = a.begin();
|
|
auto ib = b.begin();
|
|
for (; ia != a.end(); ++ia, ++ib) {
|
|
if (ia->GetPVPointer() < ib->GetPVPointer()) return true;
|
|
if (ia->GetPVPointer() > ib->GetPVPointer()) return false;
|
|
// Pointers equal
|
|
if (ia->GetCopyNo() < ib->GetCopyNo()) return true;
|
|
if (ia->GetCopyNo() > ib->GetCopyNo()) return false;
|
|
// Both pointers and copy no are equal - continue
|
|
}
|
|
// Equality
|
|
return false;
|
|
}
|
|
|
|
void G4RayTracerSceneHandler::BuildVisAttsMap (const G4VSolid&)
|
|
{
|
|
// Build map of vis attributes
|
|
|
|
G4PhysicalVolumeModel* fpPVModel = dynamic_cast<G4PhysicalVolumeModel*>(fpModel);
|
|
if (fpPVModel) {
|
|
G4ModelingParameters::PVPointerCopyNoPath temp;
|
|
for (const auto& nodeID: fpPVModel->GetFullPVPath()) {
|
|
// Build an element from the nodeid.
|
|
temp.push_back(G4ModelingParameters::PVPointerCopyNo(nodeID.GetPhysicalVolume(),nodeID.GetCopyNo()));
|
|
}
|
|
const G4VisAttributes* pVisAtts = fpVisAttribs;
|
|
if (!pVisAtts) {
|
|
// Shouldn't happen.
|
|
if (G4VisManager::GetInstance()->GetVerbosity() >= G4VisManager::warnings) {
|
|
G4warn <<
|
|
"WARNING: G4RayTracerSceneHandler::BuildVisAttsMap: null vis atts pointer."
|
|
"\n Using a default vis atts."
|
|
<< G4endl;
|
|
}
|
|
static const G4VisAttributes defaultVisAtts;
|
|
pVisAtts = &defaultVisAtts;
|
|
}
|
|
// Copy vis atts into the vis atts map
|
|
fSceneVisAttsMap[temp] = *pVisAtts;
|
|
}
|
|
}
|
|
|