...

[FrontAlgorithms.git] / lib / fpa / Filters / Image / RandomWalker.h

diff --git a/lib/fpa/Filters/Image/RandomWalker.h b/lib/fpa/Filters/Image/RandomWalker.h
new file mode 100644 (file)
index 0000000..41fba2c
--- /dev/null
+++ b/lib/fpa/Filters/Image/RandomWalker.h
@@ -0,0 +1,105 @@
+// =========================================================================
+// @author Leonardo Florez Valencia
+// @email florez-l@javeriana.edu.co
+// =========================================================================
+#ifndef __fpa__Filters__Image__RandomWalker__h__
+#define __fpa__Filters__Image__RandomWalker__h__
+
+#include <itkImage.h>
+#include <fpa/Filters/DijkstraBase.h>
+#include <fpa/Filters/Image/Algorithm.h>
+#include <fpa/Filters/Image/DefaultTraits.h>
+#include <fpa/Filters/MarksInterface.h>
+#include <fpa/Filters/Image/LabelsSeedInterface.h>
+
+namespace fpa
+{
+  namespace Filters
+  {
+    namespace Image
+    {
+      /**
+       */
+      template< class _TInputImage, class _TInputLabels, class _TCostsImage >
+      class RandomWalkerTraits
+        : public fpa::Filters::Image::DefaultTraits< _TInputImage, _TCostsImage, typename _TInputLabels::PixelType >
+      {
+      public:
+        typedef _TInputImage  TInputImage;
+        typedef _TInputLabels TInputLabels;
+        typedef _TCostsImage  TCostsImage;
+        typedef fpa::Filters::Image::DefaultTraits< TInputImage, TCostsImage, typename _TInputLabels::PixelType > Superclass;
+
+        typedef typename Superclass::TInternalTraits TInternalTraits;
+        typedef typename Superclass::TFilter TFilter;
+        typedef fpa::Filters::MarksInterface< TInternalTraits >  TMarksInterface;
+        typedef fpa::Filters::Image::LabelsSeedInterface< TInternalTraits > TSeedsInterface;
+      };
+
+      /**
+       */
+      template< class _TInputImage, class _TInputLabels, class _TCostsImage = itk::Image< float, _TInputImage::ImageDimension >, class _TTraits = fpa::Filters::Image::RandomWalkerTraits< _TInputImage, _TInputLabels, _TCostsImage > >
+      class RandomWalker
+        : public fpa::Filters::Image::Algorithm< fpa::Filters::DijkstraBase< _TTraits > >
+      {
+      public:
+        typedef _TInputImage  TInputImage;
+        typedef _TInputLabels TInputLabels;
+        typedef _TCostsImage  TCostsImage;
+        typedef _TTraits      TTraits;
+        fpaTraitsMacro( typename, TTraits );
+
+        typedef fpa::Filters::DijkstraBase< TTraits >        TAlgorithm;
+        typedef fpa::Filters::Image::Algorithm< TAlgorithm > Superclass;
+        typedef RandomWalker                                 Self;
+        typedef itk::SmartPointer< Self >                    Pointer;
+        typedef itk::SmartPointer< const Self >              ConstPointer;
+
+        typedef typename Superclass::TMarksImage TOutputLabels;
+
+      public:
+        itkNewMacro( Self );
+        itkTypeMacro(
+          fpa::Filters::Image::RandomWalker, fpa::Filters::Image::Algorithm
+          );
+
+        fpaFilterInputMacro( InputLabels, TInputLabels );
+
+      public:
+        TCostsImage* GetOutputCosts( );
+        const TCostsImage* GetOutputCosts( ) const;
+
+        TOutputLabels* GetOutputLabels( );
+        const TOutputLabels* GetOutputLabels( ) const;
+
+        TInputImage* GetInputImage( );
+        const TInputImage* GetInputImage( ) const;
+        void SetInputImage( TInputImage* image );
+
+      protected:
+        RandomWalker( );
+        virtual ~RandomWalker( );
+
+        virtual const itk::DataObject* _GetReferenceInput( ) const override;
+        virtual void _PrepareSeeds( const itk::DataObject* input ) override;
+
+      private:
+        // Purposely not implemented.
+        RandomWalker( const Self& other );
+        Self& operator=( const Self& other );
+
+      protected:
+        unsigned int m_InputLabelsIdx;
+      };
+
+    } // ecapseman
+
+  } // ecapseman
+
+} // ecapseman
+
+#ifndef ITK_MANUAL_INSTANTIATION
+#  include <fpa/Filters/Image/RandomWalker.hxx>
+#endif // ITK_MANUAL_INSTANTIATION
+#endif // __fpa__Filters__Image__RandomWalker__h__
+// eof - $RCSfile$