mitkLabeledImageToSurfaceFilter.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 _mitk_labeled_image_to_surface_filter__h_
#define _mitk_labeled_image_to_surface_filter__h_

#include "MitkAlgorithmsExtExports.h"
#include <map>
#include <mitkImageToSurfaceFilter.h>
#include <vtkImageData.h>

namespace mitk
{
  class MITKALGORITHMSEXT_EXPORT LabeledImageToSurfaceFilter : public ImageToSurfaceFilter
  {
  public:
    mitkClassMacro(LabeledImageToSurfaceFilter, ImageToSurfaceFilter);

    itkFactorylessNewMacro(Self);

    itkCloneMacro(Self);

      typedef int LabelType;

    typedef std::map<LabelType, unsigned long> LabelMapType;

    typedef std::map<unsigned int, LabelType> IdxToLabelMapType;

    itkSetMacro(GenerateAllLabels, bool);

    itkGetMacro(GenerateAllLabels, bool);

    itkBooleanMacro(GenerateAllLabels);

    itkSetMacro(Label, LabelType);

    itkGetMacro(Label, LabelType);

    itkSetMacro(BackgroundLabel, LabelType);

    itkGetMacro(BackgroundLabel, LabelType);

    itkSetMacro(GaussianStandardDeviation, double);

    itkGetMacro(GaussianStandardDeviation, double);

    LabelType GetLabelForNthOutput(const unsigned int &i);

    mitk::ScalarType GetVolumeForNthOutput(const unsigned int &i);

    mitk::ScalarType GetVolumeForLabel(const LabelType &label);

  protected:
    double m_GaussianStandardDeviation;

    bool m_GenerateAllLabels;

    LabelType m_Label;

    LabelType m_BackgroundLabel;

    LabelMapType m_AvailableLabels;

    IdxToLabelMapType m_IdxToLabels;

    void GenerateData() override;

    void GenerateOutputInformation() override;

    virtual void CreateSurface(int time, vtkImageData *vtkimage, mitk::Surface *surface, LabelType label);

    virtual LabelMapType GetAvailableLabels();

    LabeledImageToSurfaceFilter();

    ~LabeledImageToSurfaceFilter() override;

  private:
    virtual void CreateSurface(int time, vtkImageData *vtkimage, mitk::Surface *surface, const ScalarType threshold);
  };

} // end of namespace mitk

#endif