Gabriele Cosmo
172 lines
7.0 KiB
C++
172 lines
7.0 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. *
|
|
// ********************************************************************
|
|
//
|
|
//
|
|
//
|
|
//
|
|
// John Allison 5th April 2001
|
|
// A template for a simplest possible graphics driver.
|
|
//?? Lines or sections marked like this require specialisation for your driver.
|
|
|
|
#ifndef G4VTKSCENEHANDLER_HH
|
|
#define G4VTKSCENEHANDLER_HH
|
|
|
|
|
|
#include "G4VSceneHandler.hh"
|
|
|
|
#include <map>
|
|
#include <vector>
|
|
|
|
#include "G4VtkViewer.hh"
|
|
#pragma GCC diagnostic push
|
|
#pragma GCC diagnostic ignored "-Wextra-semi"
|
|
#include "vtkSmartPointer.h"
|
|
#include "vtkPolyData.h"
|
|
#include "vtkLine.h"
|
|
#include "vtkNamedColors.h"
|
|
#include "vtkProperty.h"
|
|
#include "vtkTextProperty.h"
|
|
#include "vtkPolyDataMapper.h"
|
|
#include "vtkActor.h"
|
|
#include "vtkFollower.h"
|
|
#include "vtkTextActor.h"
|
|
#include "vtkBillboardTextActor3D.h"
|
|
#include "vtkRegularPolygonSource.h"
|
|
#include "vtkSphereSource.h"
|
|
#include "vtkFeatureEdges.h"
|
|
#include "vtkCleanPolyData.h"
|
|
#include "vtkPolyDataNormals.h"
|
|
#include "vtkGlyph2D.h"
|
|
#include "vtkGlyph3D.h"
|
|
#include "vtkVertexGlyphFilter.h"
|
|
#include "vtkMatrix4x4.h"
|
|
#include "vtkDoubleArray.h"
|
|
#include "vtkPointData.h"
|
|
#include "vtkTriangleFilter.h"
|
|
#include "vtkTensorGlyph.h"
|
|
#include "vtkTensorGlyphColor.h"
|
|
#include "vtkScalarsToColors.h"
|
|
#include "vtkImageData.h"
|
|
#include "vtkVolume.h"
|
|
#include "vtkOpenGLGPUVolumeRayCastMapper.h"
|
|
#pragma GCC diagnostic pop
|
|
|
|
class G4VtkSceneHandler: public G4VSceneHandler {
|
|
friend class G4VtkViewer;
|
|
|
|
public:
|
|
G4VtkSceneHandler(G4VGraphicsSystem& system, const G4String& name);
|
|
virtual ~G4VtkSceneHandler() = default;
|
|
|
|
////////////////////////////////////////////////////////////////
|
|
// Required implementation of pure virtual functions...
|
|
|
|
void AddPrimitive(const G4Polyline&);
|
|
void AddPrimitive(const G4Text&);
|
|
void AddPrimitive(const G4Circle&);
|
|
void AddPrimitive(const G4Square&);
|
|
void AddPrimitive(const G4Polyhedron&);
|
|
void AddPrimitiveTensorGlyph(const G4Polyhedron&);
|
|
void AddPrimitiveBakedTransform(const G4Polyhedron&);
|
|
|
|
// Further optional AddPrimitive methods. Explicitly invoke base
|
|
// class methods if not otherwise defined to avoid warnings about
|
|
// hiding of base class methods.
|
|
void AddPrimitive(const G4Polymarker& polymarker)
|
|
{
|
|
#ifdef G4VTKDEBUG
|
|
G4cout << "=================================" << G4endl;
|
|
G4cout << "G4VtkSceneHandler::AddPrimitive(const G4Polymarker& polymarker) called." << G4endl;
|
|
#endif
|
|
G4VSceneHandler::AddPrimitive (polymarker);
|
|
}
|
|
/*
|
|
void AddPrimitive(const G4Scale& scale)
|
|
{
|
|
#ifdef G4VTKDEBUG
|
|
G4cout << "=================================" << G4endl;
|
|
G4cout << "G4VtkSceneHandler::AddScale(const G4Scale& scale) called." <<
|
|
G4endl; #endif G4VSceneHandler::AddPrimitive(scale);
|
|
}
|
|
*/
|
|
|
|
void AddSolid(const G4Box& box);
|
|
void AddCompound(const G4Mesh& mesh);
|
|
|
|
void Modified();
|
|
void Clear();
|
|
|
|
protected:
|
|
static G4int fSceneIdCount; // Counter for Vtk scene handlers.
|
|
|
|
// maps for polylines
|
|
std::map<std::size_t, const G4VisAttributes*> polylineVisAttributesMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkPoints>> polylineDataMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkCellArray>> polylineLineMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkPolyData>> polylinePolyDataMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkPolyDataMapper>> polylinePolyDataMapperMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkActor>> polylinePolyDataActorMap;
|
|
|
|
// maps for circles
|
|
std::map<std::size_t, const G4VisAttributes*> circleVisAttributesMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkPoints>> circleDataMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkPolyData>> circlePolyDataMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkVertexGlyphFilter>> circleFilterMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkPolyDataMapper>> circlePolyDataMapperMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkActor>> circlePolyDataActorMap;
|
|
|
|
// maps for squares
|
|
std::map<std::size_t, const G4VisAttributes*> squareVisAttributesMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkPoints>> squareDataMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkPolyData>> squarePolyDataMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkVertexGlyphFilter>> squareFilterMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkPolyDataMapper>> squarePolyDataMapperMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkActor>> squarePolyDataActorMap;
|
|
|
|
// map for polyhedra
|
|
std::map<std::size_t, const G4VisAttributes*> polyhedronVisAttributesMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkPoints>> polyhedronDataMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkCellArray>> polyhedronPolyMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkPolyData>> polyhedronPolyDataMap;
|
|
std::map<std::size_t, std::size_t> polyhedronPolyDataCountMap;
|
|
|
|
// map for polyhedra instances
|
|
std::map<std::size_t, vtkSmartPointer<vtkPoints>> instancePositionMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkDoubleArray>> instanceRotationMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkDoubleArray>> instanceColoursMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkPolyData>> instancePolyDataMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkTensorGlyphColor>> instanceTensorGlyphMap;
|
|
std::map<std::size_t, vtkSmartPointer<vtkActor>> instanceActorMap;
|
|
|
|
|
|
private:
|
|
#ifdef G4VTKDEBUG
|
|
void PrintThings();
|
|
#endif
|
|
|
|
};
|
|
|
|
#endif
|