niftkFastPointSetVtkMapper3D.h

Go to the documentation of this file.

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

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). 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 in the top level directory for details.

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

#ifndef niftkFastPointSetVtkMapper3D_h
#define niftkFastPointSetVtkMapper3D_h

#include "niftkCoreExports.h"

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

#include <vtkSmartPointer.h>

class vtkPoints;
class vtkIdTypeArray;
class vtkFloatArray;
class vtkCellArray;
class vtkPolyData;
class vtkPolyDataMapper;
class vtkActor;

namespace mitk
{
class PointSet;
}

namespace niftk
{

class NIFTKCORE_EXPORT FastPointSetVtkMapper3D : public mitk::VtkMapper
{
public:
  mitkClassMacro(FastPointSetVtkMapper3D, mitk::VtkMapper)
  itkNewMacro(Self)

  virtual vtkProp* GetVtkProp(mitk::BaseRenderer* renderer) override;
  virtual const mitk::PointSet* GetInput();

protected:
  FastPointSetVtkMapper3D();
  virtual ~FastPointSetVtkMapper3D();

  class LocalStorage : public mitk::Mapper::BaseLocalStorage
  {
    public:

    LocalStorage();
    ~LocalStorage();

    vtkSmartPointer<vtkIdTypeArray>    m_Indicies;
    vtkSmartPointer<vtkFloatArray>     m_Array;
    vtkSmartPointer<vtkPoints>         m_Points;
    vtkSmartPointer<vtkCellArray>      m_CellArray;
    vtkSmartPointer<vtkPolyData>       m_PolyData;
    vtkSmartPointer<vtkPolyDataMapper> m_PolyDataMapper;
    vtkSmartPointer<vtkActor>          m_Actor;

    unsigned long int                  m_NumberOfPoints;

    itk::TimeStamp m_ShaderTimestampUpdate;
  };

  mitk::LocalStorageHandler<LocalStorage> m_LocalStorage;

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

};

}

#endif