...

[FrontAlgorithms.git] / lib / fpa / Image / Mori.h

diff --git a/lib/fpa/Image/Mori.h b/lib/fpa/Image/Mori.h
index a81016ce6b53c4f5d86a699117970678e79a2268..7e6b0e032ce1f779d763893c3600f1010ff1916a 100644 (file)
--- a/lib/fpa/Image/Mori.h
+++ b/lib/fpa/Image/Mori.h
@@ -10,6 +10,7 @@
 #include <fpa/Base/MarksInterface.h>
 #include <fpa/Base/SingleSeedInterface.h>
 #include <fpa/Image/Algorithm.h>
+#include <fpa/Image/DefaultTraits.h>
 
 namespace fpa
 {
@@ -18,66 +19,90 @@ namespace fpa
     /**
      */
     template< class _TInputImage, class _TOutputImage >
-    class Mori
-      : public fpa::Base::Mori< fpa::Image::Algorithm< _TInputImage, _TOutputImage, fpa::Base::MarksInterface< typename _TInputImage::IndexType >, fpa::Base::SingleSeedInterface< typename _TInputImage::IndexType, typename _TInputImage::PointType, typename _TInputImage::PixelType, typename _TOutputImage::PixelType, typename _TOutputImage::PixelType, typename _TInputImage::IndexType::LexicographicCompare > > >
+    class MoriTraits
+      : public fpa::Image::DefaultTraits< _TInputImage, _TOutputImage, typename _TOutputImage::PixelType >
     {
     public:
+      typedef MoriTraits    Self;
       typedef _TInputImage  TInputImage;
       typedef _TOutputImage TOutputImage;
+      typedef typename TOutputImage::PixelType TFrontId;
+      typedef fpa::Image::DefaultTraits< TInputImage, TOutputImage, TFrontId > Superclass;
 
-      typedef typename TInputImage::IndexType        TVertex;
-      typedef typename TInputImage::PointType        TPoint;
-      typedef typename TVertex::LexicographicCompare TVertexCompare;
-      typedef typename TInputImage::PixelType        TInputValue;
-      typedef typename TOutputImage::PixelType       TOutputValue;
-      typedef typename TOutputImage::PixelType       TFrontId;
+      typedef typename Superclass::TVertex       TVertex;
+      typedef typename Superclass::TInputValue   TInputValue;
+      typedef typename Superclass::TPoint        TPoint;
+      typedef typename Superclass::TOutputValue  TOutputValue;
+      typedef typename Superclass::TCompare      TCompare;
+      typedef typename Superclass::TNeighborhood TNeighborhood;
+      typedef typename Superclass::TFilter       TFilter;
 
+      typedef fpa::Base::SingleSeedInterface< TVertex, TPoint, TInputValue, TOutputValue, TFrontId, TCompare > TSeedsInterface;
       typedef fpa::Base::MarksInterface< TVertex > TMarksInterface;
-      typedef fpa::Base::SingleSeedInterface< TVertex, TPoint, TInputValue, TOutputValue, TFrontId, TVertexCompare > TSeedsInterface;
-      typedef fpa::Image::Algorithm< TInputImage, TOutputImage, TMarksInterface, TSeedsInterface > TAlgorithm;
+
+      typedef typename TSeedsInterface::TNode  TNode;
+      typedef typename TSeedsInterface::TNodes TNodes;
+      typedef typename TSeedsInterface::TSeed  TSeed;
+      typedef typename TSeedsInterface::TSeeds TSeeds;
+
+    private:
+      // Purposely not implemented.
+      MoriTraits( );
+      MoriTraits( const Self& other );
+      virtual ~MoriTraits( );
+      Self& operator=( const Self& other );
+    };
+
+    /**
+     */
+    template< class _TInputImage, class _TOutputImage >
+    class Mori
+      : public fpa::Base::Mori< fpa::Image::Algorithm< fpa::Image::MoriTraits< _TInputImage, _TOutputImage > > >
+    {
+    public:
+      typedef _TInputImage  TInputImage;
+      typedef _TOutputImage TOutputImage;
+      typedef fpa::Image::MoriTraits< TInputImage, TOutputImage > TTraits;
+      typedef fpa::Image::Algorithm< TTraits >                 TAlgorithm;
 
       typedef Mori                            Self;
       typedef fpa::Base::Mori< TAlgorithm >   Superclass;
       typedef itk::SmartPointer< Self >       Pointer;
       typedef itk::SmartPointer< const Self > ConstPointer;
 
-      typedef typename TSeedsInterface::TNode  TNode;
-      typedef typename TSeedsInterface::TNodes TNodes;
-      typedef typename TAlgorithm::TMarks      TMarks;
+      typedef typename TAlgorithm::TMarks TMarks;
 
     public:
       itkNewMacro( Self );
       itkTypeMacro( fpa::Image::Mori, fpa::Base::Mori );
 
     public:
-      TMarks* GetOutputLevels( )
-        {
-          return( this->GetMarks( ) );
-        }
-      const TMarks* GetOutputLevels( ) const
-        {
-          return( this->GetMarks( ) );
-        }
+      TOutputImage* GetThresholdedOutput( );
+      const TOutputImage* GetThresholdedOutput( ) const;
 
     protected:
-      Mori( )
-        : Superclass( )
-        {
-        }
-      virtual ~Mori( )
-        {
-        }
+      Mori( );
+      virtual ~Mori( );
+
+      virtual void _AfterGenerateData( ) override;
 
     private:
       // Purposely not implemented.
       Mori( const Self& other );
       Self& operator=( const Self& other );
+
+    protected:
+      unsigned long m_ThresholdedOutputIdx;
     };
 
   } // ecapseman
 
 } // ecapseman
 
+#ifndef ITK_MANUAL_INSTANTIATION
+#  include <fpa/Image/Mori.hxx>
+#endif // ITK_MANUAL_INSTANTIATION
+
 #endif // __fpa__Image__Mori__h__
 
 // eof - $RCSfile$