itkAnalyticSignalImageFilter.h

Go to the documentation of this file.

#ifndef __itkAnalyticSignalImageFilter_h
#define __itkAnalyticSignalImageFilter_h

#include <complex>

#include "itkFFT1DComplexToComplexImageFilter.h"
#include "itkFFT1DRealToComplexConjugateImageFilter.h"
#include "itkImageToImageFilter.h"
#include "itkImageRegionSplitterDirection.h"

namespace itk
{
template <class TPixel, unsigned int VDimension = 3>
class ITK_EXPORT AnalyticSignalImageFilter:
  public ImageToImageFilter< Image< TPixel, VDimension >,
           Image< std::complex< TPixel >, VDimension > >
{
public:
  typedef Image<TPixel,VDimension>                              InputImageType;
  typedef Image< std::complex< TPixel > , VDimension>           OutputImageType;
  typedef typename OutputImageType::RegionType                  OutputImageRegionType;

  typedef AnalyticSignalImageFilter                             Self;
  typedef ImageToImageFilter< InputImageType, OutputImageType > Superclass;
  typedef SmartPointer<Self>                                    Pointer;
  typedef SmartPointer<const Self>                              ConstPointer;

  itkTypeMacro( AnalyticSignalImageFilter, ImageToImageFilter );
  itkNewMacro( Self );

  virtual unsigned int GetDirection() const
    {
    return this->m_FFTRealToComplexFilter->GetDirection();
    }

  virtual void SetDirection( const unsigned int direction )
    {
    if( this->m_FFTRealToComplexFilter->GetDirection() != direction )
      {
      this->m_FFTRealToComplexFilter->SetDirection( direction );
      this->m_FFTComplexToComplexFilter->SetDirection( direction );
      this->Modified();
      }
    }

protected:
  AnalyticSignalImageFilter();
  virtual ~AnalyticSignalImageFilter() {}
  void PrintSelf(std::ostream& os, Indent indent) const;

  // These behave like their analogs in FFT1DRealToComplexConjugateImageFilter.
  virtual void GenerateInputRequestedRegion();
  virtual void EnlargeOutputRequestedRegion(DataObject *output);

  virtual void BeforeThreadedGenerateData ();
  virtual void ThreadedGenerateData( const OutputImageRegionType& outputRegionForThread, ThreadIdType threadId );
  virtual void AfterThreadedGenerateData ();

  typedef FFT1DRealToComplexConjugateImageFilter< TPixel, VDimension > FFTRealToComplexType;
  typedef FFT1DComplexToComplexImageFilter< TPixel, VDimension > FFTComplexToComplexType;

  typename FFTRealToComplexType::Pointer m_FFTRealToComplexFilter;
  typename FFTComplexToComplexType::Pointer m_FFTComplexToComplexFilter;

  virtual const ImageRegionSplitterBase* GetImageRegionSplitter(void) const;

private:
  AnalyticSignalImageFilter( const Self& ); // purposely not implemented
  void operator=( const Self& ); // purposely not implemented

  ImageRegionSplitterDirection::Pointer m_ImageRegionSplitter;
};
}

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkAnalyticSignalImageFilter.hxx"
#endif

#endif // __itkAnalyticSignalImageFilter_h