mitkContourModelMapper3D.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 _MITK_CONTOURMODEL_MAPPER_3D_H_
#define _MITK_CONTOURMODEL_MAPPER_3D_H_

#include "mitkCommon.h"
#include <MitkContourModelExports.h>

#include "mitkBaseRenderer.h"
#include "mitkVtkMapper.h"

#include "mitkContourModel.h"
//#include "mitkContourModelToVtkPolyDataFilter.h"

#include <vtkActor.h>
#include <vtkPolyData.h>
#include <vtkPolyDataMapper.h>
#include <vtkProp.h>
#include <vtkSmartPointer.h>
#include <vtkTubeFilter.h>

namespace mitk
{
  class MITKCONTOURMODEL_EXPORT ContourModelMapper3D : public VtkMapper
  {
  public:
    mitkClassMacro(ContourModelMapper3D, VtkMapper);

    itkFactorylessNewMacro(Self) itkCloneMacro(Self)

      const mitk::ContourModel *GetInput(void);

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

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

    /*+++ END methods of MITK-VTK rendering pipeline +++*/

    class MITKCONTOURMODEL_EXPORT LocalStorage : public mitk::Mapper::BaseLocalStorage
    {
    public:
      vtkSmartPointer<vtkActor> m_Actor;
      vtkSmartPointer<vtkPolyDataMapper> m_Mapper;

      vtkSmartPointer<vtkTubeFilter> m_TubeFilter;

      // mitk::ContourModelToVtkPolyDataFilter::Pointer m_contourToPolyData;

      vtkSmartPointer<vtkPolyData> m_OutlinePolyData;

      itk::TimeStamp m_LastUpdateTime;

      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);

  protected:
    ContourModelMapper3D();
    virtual ~ContourModelMapper3D();

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

    virtual vtkSmartPointer<vtkPolyData> CreateVtkPolyDataFromContour(mitk::ContourModel *inputContour);

    virtual void ApplyContourProperties(mitk::BaseRenderer *renderer);
  };
}
#endif