// =========================================================================
// @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$