lib/fpa/Image/RandomWalker.h

   1 // =========================================================================
   2 // @author Leonardo Florez Valencia
   3 // @email florez-l@javeriana.edu.co
   4 // =========================================================================
   5
   6 #ifndef __fpa__Image__RandomWalker__h__
   7 #define __fpa__Image__RandomWalker__h__
   8
   9 #include <itkImage.h>
  10
  11 #include <fpa/Base/DijkstraBase.h>
  12 #include <fpa/Base/MarksInterface.h>
  13 #include <fpa/Image/Algorithm.h>
  14 #include <fpa/Image/LabelledSeedsInterface.h>
  15 #include <fpa/Image/Functors/Dijkstra/Function.h>
  16
  17 namespace fpa
  18 {
  19   namespace Image
  20   {
  21     /**
  22      */
  23     template< class _TInputImage, class _TLabelImage, class _TScalar >
  24     class RandomWalker
  25       : public fpa::Base::DijkstraBase< fpa::Image::Algorithm< _TInputImage, itk::Image< _TScalar, _TInputImage::ImageDimension >, fpa::Base::MarksInterface< typename _TInputImage::IndexType >, fpa::Image::LabelledSeedsInterface< typename _TInputImage::IndexType, typename _TInputImage::PixelType, _TScalar, typename _TLabelImage::PixelType, typename _TInputImage::IndexType::LexicographicCompare > > >
  26     {
  27     public:
  28       typedef _TInputImage TInputImage;
  29       typedef _TLabelImage TLabelImage;
  30       typedef _TScalar     TScalar;
  31       typedef typename TInputImage::IndexType        TVertex;
  32       typedef typename TVertex::LexicographicCompare TVertexCompare;
  33       typedef typename TInputImage::PixelType        TInputValue;
  34       typedef typename TLabelImage::PixelType        TFrontId;
  35
  36       typedef itk::Image< TScalar, _TInputImage::ImageDimension > TOutputImage;
  37       typedef fpa::Base::MarksInterface< TVertex > TMarksInterface;
  38       typedef fpa::Image::LabelledSeedsInterface< TVertex, TInputValue, _TScalar, TFrontId, TVertexCompare > TSeedsInterface;
  39       typedef fpa::Image::Algorithm< _TInputImage, TOutputImage, TMarksInterface, TSeedsInterface > TAlgorithm;
  40
  41       typedef RandomWalker                          Self;
  42       typedef fpa::Base::DijkstraBase< TAlgorithm > Superclass;
  43       typedef itk::SmartPointer< Self >             Pointer;
  44       typedef itk::SmartPointer< const Self >       ConstPointer;
  45
  46       typedef fpa::Image::Functors::Dijkstra::Function< TInputImage, TScalar > TWeightFunction;
  47
  48     public:
  49       itkNewMacro( Self );
  50       itkTypeMacro( fpa::Image::RandomWalker, fpa::Base::RandomWalker );
  51
  52     public:
  53       virtual itk::ModifiedTimeType GetMTime( ) const override;
  54
  55     protected:
  56       RandomWalker( );
  57       virtual ~RandomWalker( );
  58
  59       virtual void _BeforeGenerateData( ) override;
  60       virtual void _QueueInit( ) override;
  61
  62     private:
  63       // Purposely not implemented.
  64       RandomWalker( const Self& other );
  65       Self& operator=( const Self& other );
  66     };
  67
  68   } // ecapseman
  69
  70 } // ecapseman
  71
  72 #ifndef ITK_MANUAL_INSTANTIATION
  73 #  include <fpa/Image/RandomWalker.hxx>
  74 #endif // ITK_MANUAL_INSTANTIATION
  75
  76 #endif // __fpa__Image__RandomWalker__h__
  77
  78 // eof - $RCSfile$