mitkOtsuSegmentationFilter.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 mitkOtsuSegmentationFilter_h_Included
#define mitkOtsuSegmentationFilter_h_Included

//#include "MitkSBExports.h"
#include "mitkITKImageImport.h"
#include "mitkImage.h"
#include "mitkImageToImageFilter.h"

#include "itkImage.h"

#include <MitkSegmentationExports.h>

namespace mitk
{
  class MITKSEGMENTATION_EXPORT OtsuSegmentationFilter : public ImageToImageFilter
  {
  public:
    typedef unsigned char OutputPixelType;
    typedef itk::Image<OutputPixelType, 3> itkOutputImageType;
    typedef mitk::ITKImageImport<itkOutputImageType> ImageConverterType;

    mitkClassMacro(OtsuSegmentationFilter, ImageToImageFilter);
    itkFactorylessNewMacro(Self) itkCloneMacro(Self)

      itkGetMacro(NumberOfThresholds, unsigned int);

    void SetNumberOfThresholds(unsigned int number)
    {
      if (number < 1)
      {
        MITK_WARN << "Tried to set an invalid number of thresholds in the OtsuSegmentationFilter.";
        return;
      }
      m_NumberOfThresholds = number;
    }

    void SetValleyEmphasis(bool useValley) { m_ValleyEmphasis = useValley; }
    void SetNumberOfBins(unsigned int number)
    {
      if (number < 1)
      {
        MITK_WARN << "Tried to set an invalid number of bins in the OtsuSegmentationFilter.";
        return;
      }
      m_NumberOfBins = number;
    }

  protected:
    OtsuSegmentationFilter();
    virtual ~OtsuSegmentationFilter();
    virtual void GenerateData() override;
    // virtual void GenerateOutputInformation();

  private:
    unsigned int m_NumberOfThresholds;
    bool m_ValleyEmphasis;
    unsigned int m_NumberOfBins;

  }; // class

} // namespace
#endif