mitkFeatureBasedEdgeDetectionFilter.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 mitkFeatureBasedEdgeDetectionFilter_h_Included
#define mitkFeatureBasedEdgeDetectionFilter_h_Included

#include <MitkSegmentationExports.h>
#include <mitkImageToUnstructuredGridFilter.h>

namespace mitk
{
  class MITKSEGMENTATION_EXPORT FeatureBasedEdgeDetectionFilter : public ImageToUnstructuredGridFilter
  {
  public:
    mitkClassMacro(FeatureBasedEdgeDetectionFilter, ImageToUnstructuredGridFilter)

      itkFactorylessNewMacro(Self)

      
      void SetSegmentationMask(mitk::Image::Pointer);

  protected:
    virtual void GenerateData();

    virtual void GenerateOutputInformation();

    FeatureBasedEdgeDetectionFilter();

    virtual ~FeatureBasedEdgeDetectionFilter();

    template <typename TPixel, unsigned int VImageDimension>
    void ITKThresholding(itk::Image<TPixel, VImageDimension> *originalImage,
                         double lower,
                         double upper,
                         mitk::Image::Pointer &result);

    template <typename TPixel, unsigned int VImageDimension>
    void ContourSearch(itk::Image<TPixel, VImageDimension> *originalImage, mitk::Image::Pointer &result);

    template <typename TPixel, unsigned int VImageDimension>
    void ThreadedClosing(itk::Image<TPixel, VImageDimension> *originalImage, mitk::Image::Pointer &result);

  private:
    mitk::UnstructuredGrid::Pointer m_PointGrid;

    mitk::Image::Pointer m_SegmentationMask;
  };
}
#endif