mitkMultiLabelSurfaceNetsExtractor.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 mitkMultiLabelSurfaceNetsExtractor_h
#define mitkMultiLabelSurfaceNetsExtractor_h
 
#include <MitkMultilabelExports.h>
#include <mitkLabelSetImage.h>
 
#include <vtkSmartPointer.h>
 
#include <map>
#include <vector>
 
class vtkImageData;
class vtkMatrix4x4;
class vtkPolyData;
class vtkPolyDataNormals;
class vtkSurfaceNets3D;
 
namespace mitk
{
  class BaseGeometry;
  class MITKMULTILABEL_EXPORT MultiLabelSurfaceNetsExtractor
  {
  public:
    using LabelValueType = MultiLabelSegmentation::LabelValueType;
 
    MultiLabelSurfaceNetsExtractor();
    ~MultiLabelSurfaceNetsExtractor();
 
    MultiLabelSurfaceNetsExtractor(const MultiLabelSurfaceNetsExtractor&) = delete;
    MultiLabelSurfaceNetsExtractor& operator=(const MultiLabelSurfaceNetsExtractor&) = delete;
 
    void SetSmoothing(bool smoothing);
    bool GetSmoothing() const;
 
    vtkSmartPointer<vtkPolyData> Extract(
      vtkImageData* groupImage,
      const std::vector<LabelValueType>& labelValues);
 
    std::map<LabelValueType, vtkSmartPointer<vtkPolyData>> ExtractPerLabel(
      vtkImageData* groupImage,
      const std::vector<LabelValueType>& labelValues);
 
    static vtkSmartPointer<vtkMatrix4x4> GetImageToWorldMatrix(const BaseGeometry* geometry);
 
  private:
    void ConfigureLabels(const std::vector<LabelValueType>& labelValues);
 
    vtkSmartPointer<vtkSurfaceNets3D> m_SurfaceNets;
    vtkSmartPointer<vtkPolyDataNormals> m_NormalsFilter;
    bool m_Smoothing;
  };
 
} // namespace mitk
 
#endif