// =========================================================================
// @author Leonardo Florez Valencia
// @email florez-l@javeriana.edu.co
// =========================================================================

#ifndef __fpa__Image__RandomWalker__h__
#define __fpa__Image__RandomWalker__h__

#include <itkImage.h>
#include <fpa/Base/DijkstraBase.h>
#include <fpa/Base/MarksInterface.h>
#include <fpa/Image/Algorithm.h>
#include <fpa/Image/DefaultTraits.h>
#include <fpa/Image/LabelledSeedsInterface.h>
#include <fpa/Image/Functors/Dijkstra/Function.h>

namespace fpa
{
  namespace Image
  {
    /**
     */
    template< class _TInputImage, class _TLabelImage, class _TScalar, class _TTraits = fpa::Image::DefaultTraits< _TInputImage, itk::Image< _TScalar, _TInputImage::ImageDimension >, typename _TLabelImage::PixelType > >
    class RandomWalker
      : public fpa::Base::DijkstraBase< fpa::Image::Algorithm< _TTraits, fpa::Base::MarksInterface< _TTraits >, fpa::Image::LabelledSeedsInterface< _TTraits > > >
    {
    public:
      typedef _TInputImage TInputImage;
      typedef _TLabelImage TLabelImage;
      typedef _TScalar     TScalar;
      typedef _TTraits     TTraits;
      fpa_Base_TraitTypes( typename TTraits );

      typedef fpa::Base::MarksInterface< TTraits > TMarksInterface;
      typedef fpa::Image::LabelledSeedsInterface< TTraits > TSeedsInterface;
      typedef fpa::Image::Algorithm< TTraits, TMarksInterface, TSeedsInterface > TAlgorithm;

      typedef fpa::Base::DijkstraBase< TAlgorithm > Superclass;
      typedef RandomWalker                          Self;
      typedef itk::SmartPointer< Self >             Pointer;
      typedef itk::SmartPointer< const Self >       ConstPointer;
      
      typedef fpa::Image::Functors::Dijkstra::Function< TInputImage, TScalar > TWeightFunction;

    public:
      itkNewMacro( Self );
      itkTypeMacro( fpa::Image::RandomWalker, fpa::Base::RandomWalker );

    public:
      virtual itk::ModifiedTimeType GetMTime( ) const override;

    protected:
      RandomWalker( );
      virtual ~RandomWalker( );

      virtual void _BeforeGenerateData( ) override;
      virtual TNodes _UnifySeeds( ) override;

    private:
      // Purposely not implemented.
      RandomWalker( const Self& other );
      Self& operator=( const Self& other );
    };

  } // ecapseman

} // ecapseman

#ifndef ITK_MANUAL_INSTANTIATION
#  include <fpa/Image/RandomWalker.hxx>
#endif // ITK_MANUAL_INSTANTIATION

#endif // __fpa__Image__RandomWalker__h__

// eof - $RCSfile$