lib/fpa/Image/Dijkstra.h

   1 #ifndef __FPA__IMAGE__DIJKSTRA__H__
   2 #define __FPA__IMAGE__DIJKSTRA__H__
   3
   4 #include <itkFunctionBase.h>
   5 #include <itkImageToImageFilter.h>
   6 #include <fpa/Base/Dijkstra.h>
   7 #include <fpa/Image/Algorithm.h>
   8 #include <fpa/Image/Functors/ImageCostFunction.h>
   9
  10 namespace fpa
  11 {
  12   namespace Image
  13   {
  14     /**
  15      * @param I Input image type
  16      * @param O Output image type
  17      */
  18     template< class I, class O >
  19     class Dijkstra
  20       : public Algorithm< I, O, fpa::Base::Dijkstra< typename I::IndexType, typename I::PixelType, typename O::PixelType, itk::Functor::IndexLexicographicCompare< I::ImageDimension >, itk::ImageToImageFilter< I, O > > >
  21     {
  22     public:
  23       typedef fpa::Base::Dijkstra< typename I::IndexType, typename I::PixelType, typename O::PixelType, itk::Functor::IndexLexicographicCompare< I::ImageDimension >, itk::ImageToImageFilter< I, O > > TBaseAlgorithm;
  24
  25       typedef Dijkstra                          Self;
  26       typedef Algorithm< I, O, TBaseAlgorithm > Superclass;
  27       typedef itk::SmartPointer< Self >         Pointer;
  28       typedef itk::SmartPointer< const Self >   ConstPointer;
  29
  30       typedef typename Superclass::TInputImage          TInputImage;
  31       typedef typename Superclass::TOutputImage         TOutputImage;
  32       typedef typename Superclass::TVertex              TVertex;
  33       typedef typename Superclass::TValue               TValue;
  34       typedef typename Superclass::TResult              TResult;
  35       typedef typename Superclass::TMinimumSpanningTree TMinimumSpanningTree;
  36
  37       typedef typename Superclass::TStartEvent     TStartEvent;
  38       typedef typename Superclass::TStartLoopEvent TStartLoopEvent;
  39       typedef typename Superclass::TEndEvent       TEndEvent;
  40       typedef typename Superclass::TEndLoopEvent   TEndLoopEvent;
  41       typedef typename Superclass::TAliveEvent     TAliveEvent;
  42       typedef typename Superclass::TFrontEvent     TFrontEvent;
  43       typedef typename Superclass::TFreezeEvent    TFreezeEvent;
  44
  45       typedef typename Superclass::TStartBacktrackingEvent TStartBacktrackingEvent;
  46       typedef typename Superclass::TEndBacktrackingEvent TEndBacktrackingEvent;
  47       typedef typename Superclass::TBacktrackingEvent TBacktrackingEvent;
  48
  49       typedef fpa::Image::Functors::ImageCostFunction< TInputImage, TResult > TCostFunction;
  50       typedef itk::FunctionBase< TResult, TResult > TConversionFunction;
  51
  52     protected:
  53       typedef typename Superclass::_TVertices      _TVertices;
  54       typedef typename Superclass::_TCollision     _TCollision;
  55       typedef typename Superclass::_TCollisionsRow _TCollisionsRow;
  56       typedef typename Superclass::_TCollisions    _TCollisions;
  57       typedef typename Superclass::_TNode          _TNode;
  58       typedef typename Superclass::_TNodes         _TNodes;
  59
  60     public:
  61       itkNewMacro( Self );
  62       itkTypeMacro( Dijkstra, Algorithm );
  63
  64       itkGetObjectMacro( CostFunction, TCostFunction );
  65       itkGetObjectMacro( ConversionFunction, TConversionFunction );
  66
  67       itkGetConstObjectMacro( CostFunction, TCostFunction );
  68       itkGetConstObjectMacro( ConversionFunction, TConversionFunction );
  69
  70       itkSetObjectMacro( CostFunction, TCostFunction );
  71       itkSetObjectMacro( ConversionFunction, TConversionFunction );
  72
  73     protected:
  74       Dijkstra( );
  75       virtual ~Dijkstra( );
  76
  77       virtual TResult _Cost( const TVertex& v, const TVertex& p ) const;
  78
  79       virtual void _BeforeGenerateData( );
  80       virtual void _InitResults( );
  81
  82     private:
  83       // Purposely not implemented
  84       Dijkstra( const Self& other );
  85       Self& operator=( const Self& other );
  86
  87     protected:
  88       typename TCostFunction::Pointer m_CostFunction;
  89       typename TConversionFunction::Pointer m_ConversionFunction;
  90     };
  91
  92   } // ecapseman
  93
  94 } // ecapseman
  95
  96 #include <fpa/Image/Dijkstra.hxx>
  97
  98 #endif // __FPA__IMAGE__DIJKSTRA__H__
  99
 100 // eof - $RCSfile$