mitkBilateralFilter.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 __mitkBilateralFilter_h
#define __mitkBilateralFilter_h

#include "MitkImageDenoisingExports.h"

// MITK
#include "mitkImageToImageFilter.h"
#include <itkImage.h>
#include <mitkImage.h>

namespace mitk
{
  class MITKIMAGEDENOISING_EXPORT BilateralFilter : public ImageToImageFilter
  {
  public:
    mitkClassMacro(BilateralFilter, ImageToImageFilter);
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);
    itkSetMacro(DomainSigma, float);
    itkSetMacro(RangeSigma, float);
    itkSetMacro(AutoKernel, bool);
    itkSetMacro(KernelRadius, unsigned int);

    itkGetMacro(DomainSigma, float);
    itkGetMacro(RangeSigma, float);
    itkGetMacro(AutoKernel, bool);
    itkGetMacro(KernelRadius, unsigned int);

  protected:
    BilateralFilter();
    ~BilateralFilter() override;
    void GenerateOutputInformation() override;
    void GenerateData() override;

    template <typename TPixel, unsigned int VImageDimension>
    void ItkImageProcessing(const itk::Image<TPixel, VImageDimension> *itkImage);

    float m_DomainSigma; 
    float m_RangeSigma;  
    bool m_AutoKernel;   // true: kernel size is calculated from DomainSigma. See ITK Doc; false: set by m_KernelRadius
    unsigned int m_KernelRadius; // use in combination with m_AutoKernel = true
  };
} // END mitk namespace
#endif