mitkSurfaceVtkMapper3D.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef mitkSurfaceVtkMapper3D_h
#define mitkSurfaceVtkMapper3D_h
 
#include "mitkBaseRenderer.h"
#include "mitkLocalStorageHandler.h"
#include "mitkVtkMapper.h"
#include <MitkCoreExports.h>
#include <mitkSurface.h>
 
#include <vtkActor.h>
#include <vtkDepthSortPolyData.h>
#include <vtkPlaneCollection.h>
#include <vtkPolyDataMapper.h>
#include <vtkPolyDataNormals.h>
#include <vtkSmartPointer.h>
 
namespace mitk
{
  class MITKCORE_EXPORT SurfaceVtkMapper3D : public VtkMapper
  {
  public:
    mitkClassMacro(SurfaceVtkMapper3D, VtkMapper);
 
    itkFactorylessNewMacro(Self);
 
    itkCloneMacro(Self);
 
    itkSetMacro(GenerateNormals, bool);
 
    itkGetMacro(GenerateNormals, bool);
 
    virtual const mitk::Surface *GetInput();
 
    vtkProp *GetVtkProp(mitk::BaseRenderer *renderer) override;
 
    virtual void ApplyAllProperties(mitk::BaseRenderer *renderer, vtkActor *actor);
 
    static void SetDefaultProperties(mitk::DataNode *node, mitk::BaseRenderer *renderer = nullptr, bool overwrite = false);
 
  protected:
    SurfaceVtkMapper3D();
 
    ~SurfaceVtkMapper3D() override;
 
    void GenerateDataForRenderer(mitk::BaseRenderer *renderer) override;
 
    void ResetMapper(mitk::BaseRenderer *renderer) override;
 
    virtual void CheckForClippingProperty(mitk::BaseRenderer *renderer, mitk::BaseProperty *property);
 
    bool m_GenerateNormals;
 
  public:
    class LocalStorage : public mitk::Mapper::BaseLocalStorage
    {
    public:
      vtkSmartPointer<vtkActor> m_Actor;
      vtkSmartPointer<vtkPolyDataMapper> m_VtkPolyDataMapper;
      vtkSmartPointer<vtkPolyDataNormals> m_VtkPolyDataNormals;
      vtkSmartPointer<vtkPlaneCollection> m_ClippingPlaneCollection;
      vtkSmartPointer<vtkDepthSortPolyData> m_DepthSort;
      itk::TimeStamp m_ShaderTimestampUpdate;
 
      LocalStorage()
      {
        m_VtkPolyDataMapper = vtkSmartPointer<vtkPolyDataMapper>::New();
        m_VtkPolyDataNormals = vtkSmartPointer<vtkPolyDataNormals>::New();
        m_Actor = vtkSmartPointer<vtkActor>::New();
        m_ClippingPlaneCollection = vtkSmartPointer<vtkPlaneCollection>::New();
 
        m_Actor->SetMapper(m_VtkPolyDataMapper);
 
        m_DepthSort = vtkSmartPointer<vtkDepthSortPolyData>::New();
      }
 
      ~LocalStorage() override {}
    };
 
    mitk::LocalStorageHandler<LocalStorage> m_LSH;
 
    static void ApplyMitkPropertiesToVtkProperty(mitk::DataNode *node,
                                                 vtkProperty *property,
                                                 mitk::BaseRenderer *renderer);
    static void SetDefaultPropertiesForVtkProperty(mitk::DataNode *node, mitk::BaseRenderer *renderer, bool overwrite);
  };
} // namespace mitk
 
#endif