...

[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..32b84ac
--- /dev/null
+++ b/lib/fpa/Filters/Image/RandomWalker.h
@@ -0,0 +1,77 @@
+// =========================================================================
+// @author Leonardo Florez Valencia
+// @email florez-l@javeriana.edu.co
+// =========================================================================
+#ifndef __fpa__Filters__Image__RandomWalker__h__
+#define __fpa__Filters__Image__RandomWalker__h__
+
+#include <fpa/Filters/RandomWalker.h>
+#include <fpa/Filters/Image/Interface.h>
+#include <fpa/Filters/Image/RandomWalkerTraits.h>
+
+namespace fpa
+{
+  namespace Filters
+  {
+    namespace Image
+    {
+      /**
+       */
+      template< class _TInputImage, class _TLabelImage, class _TCost = float, class _TTraits = fpa::Filters::Image::RandomWalkerTraits< _TInputImage, _TLabelImage, _TCost > >
+      class RandomWalker
+        : public fpa::Filters::RandomWalker< fpa::Filters::Image::Interface< _TTraits > >
+      {
+      public:
+        typedef _TTraits TTraits;
+        typedef typename TTraits::TInputImage TInputImage;
+        typedef typename TTraits::TLabelImage TLabelImage;
+        typedef typename TTraits::TCost TCost;
+        typedef typename TTraits::TLabel TLabel;
+        typedef typename TTraits::TOutputImage TOutputImage;
+        typedef TOutputImage TCostsImage;
+
+        typedef fpa::Filters::Image::Interface< TTraits > TInterface;
+        typedef fpa::Filters::RandomWalker< TInterface >  Superclass;
+        typedef RandomWalker                              Self;
+        typedef itk::SmartPointer< Self >                 Pointer;
+        typedef itk::SmartPointer< const Self >           ConstPointer;
+
+      public:
+        itkNewMacro( Self );
+
+        fpaFilterInputMacro( InputLabels, _TLabelImage );
+
+      public:
+        TInputImage* GetInputImage( );
+        const TInputImage* GetInputImage( ) const;
+        void SetInputImage( TInputImage* i );
+
+        TCostsImage* GetOutputCosts( );
+        const TCostsImage* GetOutputCosts( ) const;
+
+        TLabelImage* GetOutputLabels( );
+        const TLabelImage* GetOutputLabels( ) const;
+
+      protected:
+        RandomWalker( );
+        virtual ~RandomWalker( );
+
+        // Filter related methods
+        virtual const itk::DataObject* _GetReferenceInput( ) const override;
+
+      private:
+        RandomWalker( const Self& other );
+        Self& operator=( const Self& other );
+      };
+
+    } // 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$