lib/cpExtensions/Algorithms/GradientFunctionBase.h

   1 // -------------------------------------------------------------------------
   2 // @author Leonardo Florez-Valencia (florez-l@javeriana.edu.co)
   3 // -------------------------------------------------------------------------
   4
   5 #ifndef __CPEXTENSIONS__ALGORITHMS__GRADIENTFUNCTIONBASE__H__
   6 #define __CPEXTENSIONS__ALGORITHMS__GRADIENTFUNCTIONBASE__H__
   7
   8 #include <map>
   9 #include <itkImageFunction.h>
  10
  11 namespace cpExtensions
  12   {
  13     namespace Algorithms
  14     {
  15       /**
  16        */
  17       template< class G >
  18       class GradientFunctionBase
  19         : public itk::ImageFunction< G, typename G::PixelType::ValueType, typename G::PixelType::ValueType >
  20       {
  21       public:
  22         // Types from input arguments
  23         typedef G                           TGradient;
  24         typedef typename G::PixelType       TVector;
  25         typedef typename TVector::ValueType TScalar;
  26         itkStaticConstMacro( Dimension, unsigned int, G::ImageDimension );
  27
  28         // Standard itk types
  29         typedef GradientFunctionBase                      Self;
  30         typedef itk::ImageFunction< G, TScalar, TScalar > Superclass;
  31         typedef itk::SmartPointer< Self >                 Pointer;
  32         typedef itk::SmartPointer< const Self >           ConstPointer;
  33
  34         // Types from base itk::ImageFunction
  35         typedef typename Superclass::InputType           TInput;
  36         typedef typename Superclass::OutputType          TOutput;
  37         typedef typename Superclass::PointType           TPoint;
  38         typedef typename Superclass::ContinuousIndexType TContIndex;
  39         typedef typename Superclass::IndexType           TIndex;
  40
  41         // Sparse buffer
  42         typedef std::map< TIndex, TOutput, typename TIndex::LexicographicCompare > TBuffer;
  43
  44       public:
  45         itkTypeMacro( GradientFunctionBase, itkImageFunction );
  46
  47         itkBooleanMacro( BufferResults );
  48         itkGetConstMacro( BufferResults, bool );
  49         itkSetMacro( BufferResults, bool );
  50
  51       public:
  52         virtual void ResetBuffer( );
  53
  54         virtual TOutput Evaluate( const TPoint& p ) const;
  55         virtual TOutput EvaluateAtIndex( const TIndex& i ) const;
  56         virtual TOutput EvaluateAtContinuousIndex( const TContIndex& i ) const;
  57
  58       protected:
  59         GradientFunctionBase( );
  60         virtual ~GradientFunctionBase( );
  61
  62         virtual TOutput _Evaluate( const TIndex& i ) const = 0;
  63
  64       private:
  65         // Purposely not implemented.
  66         GradientFunctionBase( const Self& );
  67         void operator=( const Self& );
  68
  69       protected:
  70         mutable TBuffer m_Buffer;
  71         bool m_BufferResults;
  72       };
  73
  74     } // ecapseman
  75
  76 } // ecapseman
  77
  78 #include <cpExtensions/Algorithms/GradientFunctionBase.hxx>
  79
  80 #endif // __CPEXTENSIONS__ALGORITHMS__GRADIENTFUNCTIONBASE__H__
  81
  82 // eof - $RCSfile$