00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017 #ifndef __itkComplexToModulusImageFilter_h
00018 #define __itkComplexToModulusImageFilter_h
00019
00020 #include "itkUnaryFunctorImageFilter.h"
00021 #include "vnl/vnl_math.h"
00022
00023 namespace itk
00024 {
00025
00031 namespace Function {
00032
00033 template< class TInput, class TOutput>
00034 class ComplexToModulus
00035 {
00036 public:
00037 ComplexToModulus() {}
00038 ~ComplexToModulus() {}
00039 inline TOutput operator()( const TInput & A )
00040 { return (TOutput)( sqrt( A.real() * A.real() +
00041 A.imag() * A.imag() ) ); }
00042 };
00043 }
00044
00045 template <class TInputImage, class TOutputImage>
00046 class ITK_EXPORT ComplexToModulusImageFilter :
00047 public
00048 UnaryFunctorImageFilter<TInputImage,TOutputImage,
00049 Function::ComplexToModulus<
00050 typename TInputImage::PixelType,
00051 typename TOutputImage::PixelType> >
00052 {
00053 public:
00055 typedef ComplexToModulusImageFilter Self;
00056 typedef UnaryFunctorImageFilter<TInputImage,TOutputImage,
00057 Function::ComplexToModulus< typename TInputImage::PixelType,
00058 typename TOutputImage::PixelType> > Superclass;
00059 typedef SmartPointer<Self> Pointer;
00060 typedef SmartPointer<const Self> ConstPointer;
00061
00063 itkNewMacro(Self);
00064
00065 protected:
00066 ComplexToModulusImageFilter() {}
00067 virtual ~ComplexToModulusImageFilter() {}
00068
00069 private:
00070 ComplexToModulusImageFilter(const Self&);
00071 void operator=(const Self&);
00072
00073 };
00074
00075 }
00076
00077
00078 #endif