// =========================================================================
// @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/LabelledSeedsInterface.h>
#include <fpa/Image/Functors/Dijkstra/Function.h>

namespace fpa
{
  namespace Image
  {
    /**
     */
    template< class _TInputImage, class _TLabelImage, class _TScalar >
    class RandomWalker
      : 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::PointType, typename _TInputImage::PixelType, _TScalar, typename _TLabelImage::PixelType, typename _TInputImage::IndexType::LexicographicCompare > > >
    {
    public:
      typedef _TInputImage TInputImage;
      typedef _TLabelImage TLabelImage;
      typedef _TScalar     TScalar;
      typedef typename TInputImage::IndexType        TVertex;
      typedef typename TInputImage::PointType        TPoint;
      typedef typename TVertex::LexicographicCompare TVertexCompare;
      typedef typename TInputImage::PixelType        TInputValue;
      typedef typename TLabelImage::PixelType        TFrontId;

      typedef itk::Image< TScalar, _TInputImage::ImageDimension > TOutputImage;
      typedef fpa::Base::MarksInterface< TVertex > TMarksInterface;
      typedef fpa::Image::LabelledSeedsInterface< TVertex, TPoint, TInputValue, _TScalar, TFrontId, TVertexCompare > TSeedsInterface;
      typedef fpa::Image::Algorithm< _TInputImage, TOutputImage, TMarksInterface, TSeedsInterface > TAlgorithm;

      typedef RandomWalker                          Self;
      typedef fpa::Base::DijkstraBase< TAlgorithm > Superclass;
      typedef itk::SmartPointer< Self >             Pointer;
      typedef itk::SmartPointer< const Self >       ConstPointer;

      typedef fpa::Image::Functors::Dijkstra::Function< TInputImage, TScalar > TWeightFunction;
      typedef typename TSeedsInterface::TNode  TNode;
      typedef typename TSeedsInterface::TNodes TNodes;

    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$