mitkLabelOverlay3D.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 LabelOverlay3D_H
#define LabelOverlay3D_H

#include "MitkOverlaysExports.h"
#include <mitkLocalStorageHandler.h>
#include <mitkVtkOverlay3D.h>
#include <vtkSmartPointer.h>

class vtkStringArray;
class vtkPolyDataMapper;
class vtkPolyData;
class vtkActor2D;
class vtkProperty2D;
class vtkPointSetToLabelHierarchy;
class vtkLabelPlacementMapper;
class vtkIntArray;

namespace mitk
{
  class PointSet;

  class MITKOVERLAYS_EXPORT LabelOverlay3D : public mitk::VtkOverlay3D
  {
  public:
    class LocalStorage : public mitk::Overlay::BaseLocalStorage
    {
    public:
      vtkSmartPointer<vtkPolyData> m_Points;
      vtkSmartPointer<vtkActor2D> m_LabelsActor;
      vtkSmartPointer<vtkIntArray> m_Sizes;
      vtkSmartPointer<vtkStringArray> m_Labels;
      vtkSmartPointer<vtkLabelPlacementMapper> m_LabelMapper;
      vtkSmartPointer<vtkPointSetToLabelHierarchy> m_PointSetToLabelHierarchyFilter;

      itk::TimeStamp m_LastUpdateTime;

      LocalStorage();
      ~LocalStorage();
    };

    mitkClassMacro(LabelOverlay3D, mitk::VtkOverlay3D);
    itkFactorylessNewMacro(Self) itkCloneMacro(Self)

      void SetLabelVector(const std::vector<std::string> &LabelVector);

    void SetPriorityVector(const std::vector<int> &PriorityVector);

    void SetLabelCoordinates(itk::SmartPointer<PointSet> LabelCoordinates);

    void PointSetModified(const itk::Object *, const itk::EventObject &);

  protected:
    mutable mitk::LocalStorageHandler<LocalStorage> m_LSH;

    virtual vtkProp *GetVtkProp(BaseRenderer *renderer) const override;
    void UpdateVtkOverlay(mitk::BaseRenderer *renderer) override;

    explicit LabelOverlay3D();

    virtual ~LabelOverlay3D();

  private:
    std::vector<std::string> m_LabelVector;

    std::vector<int> m_PriorityVector;

    itk::SmartPointer<PointSet> m_LabelCoordinates;

    unsigned long m_PointSetModifiedObserverTag;

    LabelOverlay3D(const LabelOverlay3D &);

    LabelOverlay3D &operator=(const LabelOverlay3D &);
  };

} // namespace mitk
#endif // LabelOverlay3D_H