mitkSurfaceInterpolationController.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 mitkSurfaceInterpolationController_h
#define mitkSurfaceInterpolationController_h
 
#include <mitkDataStorage.h>
#include <mitkLabelSetImage.h>
#include <mitkLabel.h>
#include <mitkSurface.h>
 
#include <MitkSurfaceInterpolationExports.h>
 
namespace mitk
{
  class ComputeContourSetNormalsFilter;
  class CreateDistanceImageFromSurfaceFilter;
  class MultiLabelSegmentation;
  class ReduceContourSetFilter;
 
  class MITKSURFACEINTERPOLATION_EXPORT SurfaceInterpolationController : public itk::Object
  {
  public:
    mitkClassMacroItkParent(SurfaceInterpolationController, itk::Object);
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);
 
    struct MITKSURFACEINTERPOLATION_EXPORT ContourPositionInformation
    {
      Surface::ConstPointer Contour;
      PlaneGeometry::ConstPointer Plane;
      Label::PixelType LabelValue;
      TimeStepType TimeStep;
 
      ContourPositionInformation()
        : Plane(nullptr),
          LabelValue(std::numeric_limits<Label::PixelType>::max()),
          TimeStep(std::numeric_limits<TimeStepType>::max())
      {
      }
 
      ContourPositionInformation(Surface::ConstPointer contour,
        PlaneGeometry::ConstPointer plane,
        Label::PixelType labelValue,
        TimeStepType timeStep)
        :
        Contour(contour),
        Plane(plane),
        LabelValue(labelValue),
        TimeStep(timeStep)
      {
      }
 
      bool IsPlaceHolder() const
      {
        return Contour.IsNull();
      }
    };
 
    typedef std::vector<ContourPositionInformation> CPIVector;
 
    static SurfaceInterpolationController *GetInstance();
 
    void AddNewContours(const std::vector<ContourPositionInformation>& newCPIs, bool reinitializeAction = false, bool silent = false);
 
    bool RemoveContour(ContourPositionInformation contourInfo, bool keepPlaceholderForUndo = false);
 
    void RemoveObservers();
 
    void Interpolate(const MultiLabelSegmentation* segmentationImage, MultiLabelSegmentation::LabelValueType labelValue, TimeStepType timeStep);
 
    mitk::Surface::Pointer GetInterpolationResult(const MultiLabelSegmentation* segmentationImage, MultiLabelSegmentation::LabelValueType labelValue, TimeStepType timeStep);
 
    void SetMinSpacing(double minSpacing);
 
    void SetMaxSpacing(double maxSpacing);
 
    void SetDistanceImageVolume(unsigned int distImageVolume);
 
    mitk::MultiLabelSegmentation* GetCurrentSegmentation();
 
    void SetDataStorage(DataStorage::Pointer ds);
 
    void SetCurrentInterpolationSession(MultiLabelSegmentation* currentSegmentationImage);
 
    void RemoveInterpolationSession(const MultiLabelSegmentation* segmentationImage);
 
    void RemoveAllInterpolationSessions();
 
    CPIVector* GetContours(MultiLabelSegmentation::LabelValueType labelValue, TimeStepType timeStep);
 
    std::vector<MultiLabelSegmentation::LabelValueType> GetAffectedLabels(const MultiLabelSegmentation* seg, TimeStepType timeStep, const PlaneGeometry* plane) const;
 
    void CompleteReinitialization(const std::vector<ContourPositionInformation>& newCPIs);
 
    void RemoveContours(const MultiLabelSegmentation* segmentationImage, mitk::Label::PixelType label, TimeStepType timeStep);
 
    void RemoveContours(const MultiLabelSegmentation* segmentationImage, mitk::Label::PixelType label);
 
    unsigned int GetNumberOfInterpolationSessions();
 
    mitk::DataNode* GetSegmentationImageNode() const;
 
  protected:
    SurfaceInterpolationController();
 
    ~SurfaceInterpolationController() override;
 
    template <typename TPixel, unsigned int VImageDimension>
    void GetImageBase(itk::Image<TPixel, VImageDimension> *input, itk::ImageBase<3>::Pointer &result);
 
  private:
 
    void OnSegmentationDeleted(const itk::Object *caller, const itk::EventObject &event);
 
    void OnRemoveLabel(const itk::Object* caller, const itk::EventObject& event);
 
    void AddPlaneGeometryNodeToDataStorage(const ContourPositionInformation& contourInfo) const;
 
    DataStorage::SetOfObjects::ConstPointer GetPlaneGeometryNodeFromDataStorage(const DataNode* segNode) const;
    DataStorage::SetOfObjects::ConstPointer GetPlaneGeometryNodeFromDataStorage(const DataNode* segNode, MultiLabelSegmentation::LabelValueType labelValue) const;
    DataStorage::SetOfObjects::ConstPointer GetPlaneGeometryNodeFromDataStorage(const DataNode* segNode, MultiLabelSegmentation::LabelValueType labelValue, TimeStepType timeStep) const;
    DataStorage::SetOfObjects::ConstPointer GetPlaneGeometryNodeFromDataStorage(const mitk::MultiLabelSegmentation* seg) const;
 
    void AddActiveLabelContoursForInterpolation(ReduceContourSetFilter* reduceFilter, const MultiLabelSegmentation* segmentationImage, MultiLabelSegmentation::LabelValueType labelValue, TimeStepType timeStep);
 
    void ClearInterpolationSession();
 
    void RemoveObserversInternal(const mitk::MultiLabelSegmentation* segmentationImage);
 
    void AddToCPIMap(ContourPositionInformation& contourInfo, bool reinitializationAction = false);
 
    unsigned int m_DistanceImageVolume;
    mitk::DataStorage::Pointer m_DataStorage;
 
    WeakPointer<MultiLabelSegmentation> m_SelectedSegmentation;
  };
}
 
#endif