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 bool operator!=( const ComplexToModulus & ) const
00040 {
00041 return false;
00042 }
00043 bool operator==( const ComplexToModulus & other ) const
00044 {
00045 return !(*this != other);
00046 }
00047 inline TOutput operator()( const TInput & A )
00048 { return (TOutput)( vcl_sqrt(A.real() * A.real() +
00049 A.imag() * A.imag() ) ); }
00050 };
00051 }
00052
00053 template <class TInputImage, class TOutputImage>
00054 class ITK_EXPORT ComplexToModulusImageFilter :
00055 public
00056 UnaryFunctorImageFilter<TInputImage,TOutputImage,
00057 Function::ComplexToModulus<
00058 typename TInputImage::PixelType,
00059 typename TOutputImage::PixelType> >
00060 {
00061 public:
00063 typedef ComplexToModulusImageFilter Self;
00064 typedef UnaryFunctorImageFilter<TInputImage,TOutputImage,
00065 Function::ComplexToModulus< typename TInputImage::PixelType,
00066 typename TOutputImage::PixelType> > Superclass;
00067 typedef SmartPointer<Self> Pointer;
00068 typedef SmartPointer<const Self> ConstPointer;
00069
00071 itkNewMacro(Self);
00072
00074 itkTypeMacro(ComplexToModulusImageFilter,
00075 UnaryFunctorImageFilter);
00076
00077 typedef typename TInputImage::PixelType InputPixelType;
00078 typedef typename TOutputImage::PixelType OutputPixelType;
00079 typedef typename NumericTraits< InputPixelType >::ValueType InputPixelValueType;
00080
00081
00082 #ifdef ITK_USE_CONCEPT_CHECKING
00083
00084 itkConceptMacro(InputMultiplyOperatorCheck,
00085 (Concept::MultiplyOperator<InputPixelValueType>));
00086
00088 #endif
00089
00090 protected:
00091 ComplexToModulusImageFilter() {}
00092 virtual ~ComplexToModulusImageFilter() {}
00093
00094 private:
00095 ComplexToModulusImageFilter(const Self&);
00096 void operator=(const Self&);
00097
00098 };
00099
00100 }
00101
00102
00103 #endif
00104