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_HEADER_INCLUDED_C1907273
#define MITKEnhancedPointSetVtkMapper3D_H_HEADER_INCLUDED_C1907273

#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 /* MITKEnhancedPointSetVtkMapper3D_H_HEADER_INCLUDED_C1907273 */