103 lines
3.1 KiB
C++
103 lines
3.1 KiB
C++
/*//////////////////////////////////////////////////////////////////////////////
|
|
// CMT Cosmic Muon Tomography project //////////////////////////////////////////
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
Copyright (c) 2014, Universita' degli Studi di Padova, INFN sez. di Padova
|
|
All rights reserved
|
|
|
|
Authors: Andrea Rigoni Garola < andrea.rigoni@pd.infn.it >
|
|
|
|
------------------------------------------------------------------
|
|
This library is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU Lesser General Public
|
|
License as published by the Free Software Foundation; either
|
|
version 3.0 of the License, or (at your option) any later version.
|
|
|
|
This library is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public
|
|
License along with this library.
|
|
|
|
//////////////////////////////////////////////////////////////////////////////*/
|
|
|
|
#include "vtkGeantScene.h"
|
|
#include "vtkGeantSolid.h"
|
|
#include "vtkBoxSolid.h"
|
|
#include "vtkTessellatedSolid.h"
|
|
#include "Vtk/vtkViewport.h"
|
|
|
|
namespace uLib {
|
|
namespace Vtk {
|
|
|
|
GeantScene::GeantScene(Geant::Scene *scene)
|
|
: m_Scene(scene), m_WorldProp3D(nullptr) {
|
|
if (!m_Scene)
|
|
return;
|
|
|
|
// 1. Create the world box wireframe prop3d
|
|
uLib::ContainerBox *worldBox = m_Scene->GetWorldBox();
|
|
if (worldBox) {
|
|
m_WorldProp3D = new ContainerBox(worldBox);
|
|
m_WorldProp3D->SetRepresentation(Prop3D::Wireframe);
|
|
m_WorldProp3D->ShowScaleMeasures(true);
|
|
}
|
|
|
|
// 2. Create prop3ds for each non-world solid
|
|
const Geant::Solid *world = m_Scene->GetWorld();
|
|
const Vector<Geant::Solid *> &solids = m_Scene->GetSolids();
|
|
|
|
for (Geant::Solid *solid : solids) {
|
|
// Skip the world volume itself — it's already shown as the wireframe box
|
|
if (solid == world)
|
|
continue;
|
|
|
|
// Only create a prop3d if the solid has a valid G4VSolid
|
|
if (solid->GetG4Solid()) {
|
|
GeantSolid *vtkSolid = nullptr;
|
|
if (auto *box = dynamic_cast<Geant::BoxSolid *>(solid)) {
|
|
vtkSolid = new BoxSolid(box);
|
|
} else if (auto *tess = dynamic_cast<Geant::TessellatedSolid *>(solid)) {
|
|
vtkSolid = new TessellatedSolid(tess);
|
|
} else {
|
|
vtkSolid = new GeantSolid(solid);
|
|
vtkSolid->Update();
|
|
}
|
|
|
|
if (vtkSolid) {
|
|
m_SolidProp3Ds.push_back(vtkSolid);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
GeantScene::~GeantScene() {
|
|
delete m_WorldProp3D;
|
|
for (auto *p : m_SolidProp3Ds) {
|
|
delete p;
|
|
}
|
|
}
|
|
|
|
void GeantScene::AddToViewer(Viewport &viewer) {
|
|
if (m_WorldProp3D) {
|
|
viewer.AddProp3D(*m_WorldProp3D);
|
|
}
|
|
for (auto *p : m_SolidProp3Ds) {
|
|
viewer.AddProp3D(*p);
|
|
}
|
|
}
|
|
|
|
void GeantScene::RemoveFromViewer(Viewport &viewer) {
|
|
if (m_WorldProp3D) {
|
|
viewer.RemoveProp3D(*m_WorldProp3D);
|
|
}
|
|
for (auto *p : m_SolidProp3Ds) {
|
|
viewer.RemoveProp3D(*p);
|
|
}
|
|
}
|
|
|
|
} // namespace Vtk
|
|
} // namespace uLib
|