mitkEnhancedPointSetVtkMapper3D.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 mitkEnhancedPointSetVtkMapper3D_h
#define mitkEnhancedPointSetVtkMapper3D_h
 
#include "MitkMapperExtExports.h"
#include "mitkBaseRenderer.h"
#include "mitkCommon.h"
#include "mitkPointSet.h"
#include "mitkVector.h"
#include "mitkVtkMapper.h"
 
class vtkActor;
class vtkAssembly;
class vtkSphereSource;
class vtkCubeSource;
class vtkConeSource;
class vtkCylinderSource;
class vtkTubeFilter;
class vtkProp;
 
namespace mitk
{
  class MITKMAPPEREXT_EXPORT EnhancedPointSetVtkMapper3D : public VtkMapper
  {
  public:
    mitkClassMacro(EnhancedPointSetVtkMapper3D, VtkMapper);
 
    itkFactorylessNewMacro(Self);
 
    itkCloneMacro(Self);
 
      virtual const mitk::PointSet *GetInput();
 
    vtkProp *GetVtkProp(mitk::BaseRenderer *renderer) override;
 
    void UpdateVtkTransform(mitk::BaseRenderer *renderer) override;
 
    static void SetDefaultProperties(mitk::DataNode *node, mitk::BaseRenderer *renderer = nullptr, bool overwrite = false);
 
    LocalStorageHandler<BaseLocalStorage> m_LSH;
 
  protected:
    EnhancedPointSetVtkMapper3D();
 
    ~EnhancedPointSetVtkMapper3D() override;
 
    void RemoveEntryFromSourceMaps(mitk::PointSet::PointIdentifier pointID);
    void DeleteVtkObject(vtkObject *o); // functor for stl_each in destructor
 
    // update all vtk sources, mappers, actors with current data and properties
    void UpdateVtkObjects();
 
    void GenerateDataForRenderer(mitk::BaseRenderer *renderer) override;
    void ApplyColorAndOpacityProperties(mitk::BaseRenderer *renderer, vtkActor *actor) override;
 
    typedef mitk::PointSet::PointIdentifier PointIdentifier;
    typedef std::map<PointIdentifier, vtkSphereSource *> SphereSourceMap;
    typedef std::map<PointIdentifier, vtkCubeSource *> CubeSourceMap;
    typedef std::map<PointIdentifier, vtkConeSource *> ConeSourceMap;
    typedef std::map<PointIdentifier, vtkCylinderSource *> CylinderSourceMap;
 
    typedef std::pair<vtkActor *, mitk::PointSpecificationType> ActorAndPointType;
    typedef std::map<PointIdentifier, ActorAndPointType> ActorMap;
 
    SphereSourceMap m_SphereSources;     // stores all sphere sources
    CubeSourceMap m_CubeSources;         // stores all cube sources
    ConeSourceMap m_ConeSources;         // stores all cone sources
    CylinderSourceMap m_CylinderSources; // stores all cylinder sources
    ActorMap m_PointActors; // stores an actor for each point(referenced by its ID) and the currently used pointspec =
                            // which source type is generating the polydata
 
    vtkActor *m_Contour;
    vtkTubeFilter *m_ContourSource;
 
    vtkAssembly *m_PropAssembly; // this contains everything, this will be returned by GetVtkProp()
  };
} // namespace mitk
 
#endif