Gabriele Cosmo
136 lines
4.6 KiB
C++
136 lines
4.6 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 "G4VtkStore.hh"
|
|
#include "G4VtkViewer.hh"
|
|
|
|
#include <map>
|
|
#include <vector>
|
|
|
|
#ifndef WIN32
|
|
#pragma GCC diagnostic push
|
|
#pragma GCC diagnostic ignored "-Wextra-semi"
|
|
#endif
|
|
|
|
#include <vtkActor.h>
|
|
#include <vtkBillboardTextActor3D.h>
|
|
#include <vtkCleanPolyData.h>
|
|
#include <vtkDoubleArray.h>
|
|
#include <vtkFeatureEdges.h>
|
|
#include <vtkFollower.h>
|
|
#include <vtkGlyph2D.h>
|
|
#include <vtkGlyph3D.h>
|
|
#include <vtkImageData.h>
|
|
#include <vtkLine.h>
|
|
#include <vtkMatrix4x4.h>
|
|
#include <vtkNamedColors.h>
|
|
#include <vtkOpenGLGPUVolumeRayCastMapper.h>
|
|
#include <vtkPointData.h>
|
|
#include <vtkPolyData.h>
|
|
#include <vtkPolyDataMapper.h>
|
|
#include <vtkPolyDataNormals.h>
|
|
#include <vtkProperty.h>
|
|
#include <vtkRegularPolygonSource.h>
|
|
#include <vtkScalarsToColors.h>
|
|
#include <vtkSmartPointer.h>
|
|
#include <vtkSphereSource.h>
|
|
#include <vtkTensorGlyph.h>
|
|
#include <vtkTensorGlyphColor.h>
|
|
#include <vtkTextActor.h>
|
|
#include <vtkTextProperty.h>
|
|
#include <vtkTriangleFilter.h>
|
|
#include <vtkVertexGlyphFilter.h>
|
|
#include <vtkVolume.h>
|
|
|
|
#ifndef WIN32
|
|
#pragma GCC diagnostic pop
|
|
#endif
|
|
|
|
class G4VtkSceneHandler : public G4VSceneHandler
|
|
{
|
|
public:
|
|
G4VtkSceneHandler(G4VGraphicsSystem& system, const G4String& name);
|
|
~G4VtkSceneHandler() override = default;
|
|
|
|
////////////////////////////////////////////////////////////////
|
|
// Required implementation of pure virtual functions...
|
|
|
|
using G4VSceneHandler::AddPrimitive;
|
|
void AddPrimitive(const G4Polyline&) override;
|
|
void AddPrimitive(const G4Text&) override;
|
|
void AddPrimitive(const G4Circle&) override;
|
|
void AddPrimitive(const G4Square&) override;
|
|
void AddPrimitive(const G4Polyhedron&) override;
|
|
|
|
// 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) override
|
|
{
|
|
G4VSceneHandler::AddPrimitive(polymarker);
|
|
}
|
|
|
|
using G4VSceneHandler::AddSolid;
|
|
void AddSolid(const G4Box& box) override;
|
|
|
|
using G4VSceneHandler::AddCompound;
|
|
void AddCompound(const G4Mesh& mesh) override;
|
|
|
|
void Modified();
|
|
void ClearStore() override;
|
|
void ClearTransientStore() override;
|
|
G4VtkVisContext MakeDefaultVisContext();
|
|
G4VtkStore& GetStore() { return store; }
|
|
G4VtkStore& GetTransientStore() { return transientStore; }
|
|
|
|
virtual void Print();
|
|
|
|
void SetPolyhedronPipeline(const G4String& str);
|
|
|
|
protected:
|
|
static G4int fSceneIdCount; // Counter for Vtk scene handlers.
|
|
|
|
G4VtkStore store = G4VtkStore("perm");
|
|
G4VtkStore transientStore = G4VtkStore("trans");
|
|
|
|
G4String polyhedronPipelineType;
|
|
|
|
private:
|
|
friend class G4VtkViewer;
|
|
};
|
|
|
|
#endif
|