itkNLMFilter.h

Go to the documentation of this file.

/*=============================================================================

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). 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 in the top level directory for details.

=============================================================================*/
#ifndef __itkNLMFilter_h
#define __itkNLMFilter_h

#include "itkImageToImageFilter.h"
#include "LSDerivatives/itkLSDerivatives.h"
#include "itkImage.h"

namespace itk
{

template <class TInputImage, class TOutputImage>
class ITK_EXPORT NLMFilter : public ImageToImageFilter< TInputImage, TOutputImage >
{
public:
  typedef NLMFilter                             Self;
  typedef TInputImage                           InputImageType;
  typedef typename InputImageType::Pointer      InputImagePointer;
  typedef typename InputImageType::ConstPointer InputImageConstPointer;
  typedef TOutputImage                          OutputImageType;
  typedef typename OutputImageType::Pointer     OutputImagePointer;
  
  typedef ImageToImageFilter< InputImageType, OutputImageType> Superclass;
  typedef SmartPointer<Self>                                   Pointer;
  typedef SmartPointer<const Self>                             ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro( NLMFilter, ImageToImageFilter );
  
  typedef typename InputImageType::PixelType           InputPixelType;
  typedef typename OutputImageType::PixelType          OutputPixelType;
  typedef typename InputImageType::RegionType          InputImageRegionType;
  typedef typename InputImageType::SizeType            InputImageSizeType;
  typedef typename InputImageType::IndexType           InputImageIndexType;
  typedef typename OutputImageType::RegionType         OutputImageRegionType;
  
  
  typedef itk::Image< LSGradientsL2, TInputImage::ImageDimension > FeaturesMapType;
  typedef typename FeaturesMapType::Pointer                        FeaturesMapPointer;
  typedef itk::LSDerivativesL0<InputImageType>                     L0Type;
  typedef typename L0Type::Pointer                                 L0Pointer;
  typedef itk::LSDerivativesL1<TInputImage::ImageDimension>        L1Type;
  typedef typename L1Type::Pointer                                 L1Pointer;
  typedef itk::LSDerivativesL2<TInputImage::ImageDimension>        L2Type;
  typedef typename L2Type::Pointer                                 L2Pointer;
  
  itkSetMacro( Sigma,      float              );
  itkGetMacro( Sigma,      float              );
  itkSetMacro( H,          float              );
  itkGetMacro( H,          float              );
  itkSetMacro( PSTh,       float              );
    itkGetMacro( PSTh,       float              );
  itkSetMacro( RSearch,    InputImageSizeType );
  itkGetMacro( RSearch,    InputImageSizeType );
  itkSetMacro( RComp,      InputImageSizeType );
  itkGetMacro( RComp,      InputImageSizeType );
  
protected:
  NLMFilter();
  virtual ~NLMFilter();
  // Threaded filter:
#if ITK_VERSION_MAJOR < 4
    void ThreadedGenerateData( const OutputImageRegionType & outputRegionForThread, int threadId );
    
#else
    void ThreadedGenerateData( const OutputImageRegionType & outputRegionForThread, ThreadIdType threadId );
    
#endif
  void GenerateInputRequestedRegion();
  void BeforeThreadedGenerateData( void );
  void PrintSelf( std::ostream &os, Indent indent ) const;
private:
  NLMFilter(const Self&);         // purposely not implemented
  void operator=(const Self&);    // purposely not implemented
  float ComputeTraceMO0( const InputImageSizeType& rcomp );
  float ComputeTraceMO1( const InputImageSizeType& rcomp );
  // The standard deviation of noise (in the complex domain)
  float                m_Sigma;
  // The true parameteres of NLM:
  float                m_H;
  float                m_PSTh;
  InputImageSizeType   m_RSearch;
  InputImageSizeType   m_RComp;
  FeaturesMapPointer   m_Features;
};


 
} // end namespace itk

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

#endif