G4VtkSceneHandler.hh

Go to the documentation of this file.

//
// ********************************************************************
// * 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