mitkComputeContourSetNormalsFilter.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 mitkComputeContourSetNormalsFilter_h_Included
#define mitkComputeContourSetNormalsFilter_h_Included

#include "mitkProgressBar.h"
#include "mitkSurface.h"
#include "mitkSurfaceToSurfaceFilter.h"
#include <MitkSurfaceInterpolationExports.h>

#include "vtkCellArray.h"
#include "vtkCellData.h"
#include "vtkDoubleArray.h"
#include "vtkLine.h"
#include "vtkMath.h"
#include "vtkPolyData.h"
#include "vtkSmartPointer.h"

#include "mitkImage.h"

namespace mitk
{
  class MITKSURFACEINTERPOLATION_EXPORT ComputeContourSetNormalsFilter : public SurfaceToSurfaceFilter
  {
  public:
    mitkClassMacro(ComputeContourSetNormalsFilter, SurfaceToSurfaceFilter);
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);

      /*
         \brief Returns the computed normals as a surface
     */
      mitk::Surface::Pointer GetNormalsAsSurface();

    // Resets the filter, i.e. removes all inputs and outputs
    void Reset();

    void SetMaxSpacing(double);

    void SetUseProgressBar(bool);

    void SetProgressStepSize(unsigned int stepSize);

    void SetSegmentationBinaryImage(mitk::Image *segmentationImage) { m_SegmentationBinaryImage = segmentationImage; }
  protected:
    ComputeContourSetNormalsFilter();
    ~ComputeContourSetNormalsFilter() override;
    void GenerateData() override;
    void GenerateOutputInformation() override;

  private:
    // The segmentation out of which the contours were extracted. Can be used to determine the direction of the normals
    mitk::Image::Pointer m_SegmentationBinaryImage;
    double m_MaxSpacing;

    unsigned int m_NegativeNormalCounter;
    unsigned int m_PositiveNormalCounter;

    bool m_UseProgressBar;
    unsigned int m_ProgressStepSize;

  }; // class

} // namespace
#endif