mitkLabelSetImageVtkMapper2D.h

Go to the documentation of this file.

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/

#ifndef __mitkLabelSetImageVtkMapper2D_H_
#define __mitkLabelSetImageVtkMapper2D_H_

// MITK
#include "MitkMultilabelExports.h"
#include "mitkCommon.h"

// MITK Rendering
#include "mitkBaseRenderer.h"
#include "mitkExtractSliceFilter.h"
#include "mitkLabelSetImage.h"
#include "mitkVtkMapper.h"

// VTK
#include <vtkSmartPointer.h>

class vtkActor;
class vtkPolyDataMapper;
class vtkPlaneSource;
class vtkImageData;
class vtkLookupTable;
class vtkImageReslice;
class vtkPoints;
class vtkMitkThickSlicesFilter;
class vtkPolyData;
class vtkMitkLevelWindowFilter;
class vtkNeverTranslucentTexture;

namespace mitk
{

  class MITKMULTILABEL_EXPORT LabelSetImageVtkMapper2D : public VtkMapper
  {
  public:
    mitkClassMacro(LabelSetImageVtkMapper2D, VtkMapper);

    itkNewMacro(Self);

    const mitk::Image *GetInput(void);

    virtual void Update(mitk::BaseRenderer *renderer) override;

    //### methods of MITK-VTK rendering pipeline
    virtual vtkProp *GetVtkProp(mitk::BaseRenderer *renderer) override;
    //### end of methods of MITK-VTK rendering pipeline

    class MITKMULTILABEL_EXPORT LocalStorage : public mitk::Mapper::BaseLocalStorage
    {
    public:
      vtkSmartPointer<vtkPropAssembly> m_Actors;

      std::vector<vtkSmartPointer<vtkActor>> m_LayerActorVector;
      std::vector<vtkSmartPointer<vtkPolyDataMapper>> m_LayerMapperVector;
      std::vector<vtkSmartPointer<vtkImageData>> m_ReslicedImageVector;
      std::vector<vtkSmartPointer<vtkNeverTranslucentTexture>> m_LayerTextureVector;

      vtkSmartPointer<vtkPolyData> m_EmptyPolyData;
      vtkSmartPointer<vtkPlaneSource> m_Plane;

      std::vector<mitk::ExtractSliceFilter::Pointer> m_ReslicerVector;

      vtkSmartPointer<vtkPolyData> m_OutlinePolyData;
      vtkSmartPointer<vtkActor> m_OutlineActor;
      vtkSmartPointer<vtkActor> m_OutlineShadowActor;
      vtkSmartPointer<vtkPolyDataMapper> m_OutlineMapper;

      itk::TimeStamp m_LastDataUpdateTime;

      itk::TimeStamp m_LastPropertyUpdateTime;

      mitk::ScalarType *m_mmPerPixel;

      int m_NumberOfLayers;

      // vtkSmartPointer<vtkMitkLevelWindowFilter> m_LevelWindowFilter;
      std::vector<vtkSmartPointer<vtkMitkLevelWindowFilter>> m_LevelWindowFilterVector;

      LocalStorage();
      ~LocalStorage();
    };

    mitk::LocalStorageHandler<LocalStorage> m_LSH;

    LocalStorage *GetLocalStorage(mitk::BaseRenderer *renderer);

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

    void ApplyRenderingMode(mitk::BaseRenderer *renderer);

  protected:
    void TransformActor(mitk::BaseRenderer *renderer);

    void GeneratePlane(mitk::BaseRenderer *renderer, double planeBounds[6]);

    vtkSmartPointer<vtkPolyData> CreateOutlinePolyData(mitk::BaseRenderer *renderer,
                                                       vtkImageData *image,
                                                       int pixelValue = 1);

    LabelSetImageVtkMapper2D();
    virtual ~LabelSetImageVtkMapper2D();

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

    float CalculateLayerDepth(mitk::BaseRenderer *renderer);

    void ApplyLookuptable(mitk::BaseRenderer *renderer, int layer);

    void ApplyColorTransferFunction(mitk::BaseRenderer *renderer);

    void ApplyLevelWindow(mitk::BaseRenderer *renderer);

    void ApplyColor(mitk::BaseRenderer *renderer, const mitk::Color &color);

    void ApplyOpacity(mitk::BaseRenderer *renderer, int layer);

    bool RenderingGeometryIntersectsImage(const PlaneGeometry *renderingGeometry, SlicedGeometry3D *imageGeometry);
  };

} // namespace mitk

#endif // __mitkLabelSetImageVtkMapper2D_H_