mitkSurfaceVtkMapper2D.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 mitkSurfaceVtkMapper2D_h
#define mitkSurfaceVtkMapper2D_h

#include "mitkBaseRenderer.h"
#include "mitkLocalStorageHandler.h"
#include "mitkVtkMapper.h"
#include <MitkCoreExports.h>

// VTK
#include <vtkSmartPointer.h>
class vtkAssembly;
class vtkCutter;
class vtkPlane;
class vtkLookupTable;
class vtkGlyph3D;
class vtkArrowSource;
class vtkReverseSense;

namespace mitk
{
  class Surface;

  class MITKCORE_EXPORT SurfaceVtkMapper2D : public VtkMapper
  {
  public:
    mitkClassMacro(SurfaceVtkMapper2D, VtkMapper);

    itkFactorylessNewMacro(Self);

    itkCloneMacro(Self);

      virtual const mitk::Surface *GetInput() const;

    vtkProp *GetVtkProp(mitk::BaseRenderer *renderer) override;

    static void SetDefaultProperties(mitk::DataNode *node, mitk::BaseRenderer *renderer = nullptr, bool overwrite = false);

    class LocalStorage : public mitk::Mapper::BaseLocalStorage
    {
    public:
      itk::TimeStamp m_LastUpdateTime;
      vtkSmartPointer<vtkAssembly> m_PropAssembly;

      vtkSmartPointer<vtkActor> m_Actor;
      vtkSmartPointer<vtkActor> m_NormalActor;
      vtkSmartPointer<vtkActor> m_InverseNormalActor;
      vtkSmartPointer<vtkPolyDataMapper> m_Mapper;
      vtkSmartPointer<vtkCutter> m_Cutter;
      vtkSmartPointer<vtkPlane> m_CuttingPlane;

      vtkSmartPointer<vtkPolyDataMapper> m_NormalMapper;

      vtkSmartPointer<vtkPolyDataMapper> m_InverseNormalMapper;

      vtkSmartPointer<vtkGlyph3D> m_NormalGlyph;

      vtkSmartPointer<vtkGlyph3D> m_InverseNormalGlyph;

      vtkSmartPointer<vtkArrowSource> m_ArrowSource;

      vtkSmartPointer<vtkReverseSense> m_ReverseSense;

      LocalStorage();
      ~LocalStorage() override;
    };

    mitk::LocalStorageHandler<LocalStorage> m_LSH;

    void UpdateVtkTransform(mitk::BaseRenderer * /*renderer*/) override {}
  protected:
    SurfaceVtkMapper2D();

    ~SurfaceVtkMapper2D() override;

    void GenerateDataForRenderer(mitk::BaseRenderer *renderer) override;

    void ResetMapper(BaseRenderer *renderer) override;

    virtual void FixupLegacyProperties(PropertyList *properties);

    void ApplyAllProperties(BaseRenderer *renderer);

    void Update(BaseRenderer *renderer) override;
  };
} // namespace mitk
#endif /* mitkSurfaceVtkMapper2D_h */