itkNonLocalMeansDenoisingFilter.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 __itkNonLocalMeansDenoisingFilter_h_
#define __itkNonLocalMeansDenoisingFilter_h_

#include "itkImageToImageFilter.h"
#include "itkVectorImage.h"


namespace itk{
  template< class TPixelType >
  class NonLocalMeansDenoisingFilter :
    public ImageToImageFilter< VectorImage < TPixelType, 3 >, VectorImage < TPixelType, 3 > >
  {
  public:

    typedef NonLocalMeansDenoisingFilter                                                          Self;
    typedef SmartPointer<Self>                                                                    Pointer;
    typedef SmartPointer<const Self>                                                              ConstPointer;
    typedef ImageToImageFilter< VectorImage < TPixelType, 3 >, VectorImage < TPixelType, 3 > >    Superclass;
    typedef typename Superclass::InputImageType                                                   InputImageType;
    typedef typename Superclass::OutputImageType                                                  OutputImageType;
    typedef typename Superclass::OutputImageRegionType                                            OutputImageRegionType;
    typedef Image <TPixelType, 3>                                                                 MaskImageType;

    itkFactorylessNewMacro(Self)
    itkCloneMacro(Self)

    itkTypeMacro(NonLocalMeansDenoisingFilter, ImageToImageFilter)

    itkSetMacro(UseJointInformation, bool)
    itkSetMacro(SearchRadius, int)
    itkSetMacro(ComparisonRadius, int)
    itkSetMacro(Variance, double)
    itkSetMacro(UseRicianAdaption, bool)
    itkGetMacro(CurrentVoxelCount, unsigned int)


    void SetInputImage(const InputImageType* image);
    void SetInputMask(MaskImageType* mask);

  protected:
    NonLocalMeansDenoisingFilter();
    ~NonLocalMeansDenoisingFilter() {}

    void BeforeThreadedGenerateData();

    void ThreadedGenerateData( const OutputImageRegionType &outputRegionForThread, ThreadIdType);



  private:

    int m_SearchRadius;                               
    int m_ComparisonRadius;                           
    bool m_UseJointInformation;                       
    bool m_UseRicianAdaption;                         
    unsigned int m_CurrentVoxelCount;                 
    double m_Variance;                                
    typename MaskImageType::Pointer m_Mask;           
  };
}

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkNonLocalMeansDenoisingFilter.txx"
#endif

#endif //__itkNonLocalMeansDenoisingFilter_h_