...

[FrontAlgorithms.git] / lib / fpa / Base / SingleSeedInterface.h

diff --git a/lib/fpa/Base/SingleSeedInterface.h b/lib/fpa/Base/SingleSeedInterface.h
index dc45fe9b49e7998f4d3fe90fbf18e8754eba3185..02950325cbf0cac1200211ba1e9f3b3a8f8b0e94 100644 (file)
--- a/lib/fpa/Base/SingleSeedInterface.h
+++ b/lib/fpa/Base/SingleSeedInterface.h
@@ -6,10 +6,7 @@
 #ifndef __fpa__Base__SingleSeedInterface__h__
 #define __fpa__Base__SingleSeedInterface__h__
 
-#include <functional>
-#include <set>
-#include <vector>
-
+#include <fpa/Config.h>
 #include <itkConceptChecking.h>
 #include <itkProcessObject.h>
 
@@ -19,58 +16,22 @@ namespace fpa
   {
     /**
      */
-    template< class _TVertex, class _TPoint, class _TInputValue, class _TOutputValue, class _TFrontId, class _TCompare = std::greater< _TVertex > >
+    template< class _TTraits >
     class SingleSeedInterface
     {
     public:
+      typedef SingleSeedInterface Self;
+      typedef _TTraits            TTraits;
+      fpa_Base_TraitTypes( typename TTraits );
+
+    private:
       itkConceptMacro(
         Check_TFrontId,
-        ( itk::Concept::IsUnsignedInteger< _TFrontId > )
+        ( itk::Concept::IsUnsignedInteger< TFrontId > )
         );
 
     public:
-      typedef _TVertex       TVertex;
-      typedef _TPoint        TPoint;
-      typedef _TInputValue   TInputValue;
-      typedef _TOutputValue  TOutputValue;
-      typedef _TFrontId      TFrontId;
-      typedef _TCompare      TCompare;
-      typedef SingleSeedInterface Self;
-
-      struct TSeed
-      {
-        TVertex  Vertex;
-        TPoint   Point;
-        bool     IsPoint;
-        bool     IsUnified;
-        TFrontId FrontId;
-        TSeed( )
-          : IsUnified( false ),
-            FrontId( TFrontId( 0 ) )
-          { }
-      };
-      typedef std::vector< TSeed > TSeeds;
-
-      struct TNode
-      {
-        TVertex  Vertex;
-        TVertex  Parent;
-        TFrontId FrontId;
-
-        // Hack to hide the fact that seed values need to be initialized
-        mutable TOutputValue Value;
-      };
-      struct TNodeCompare
-      {
-        bool operator()( const TNode& a, const TNode& b ) const
-          {
-            TCompare cmp;
-            return( cmp( a.Vertex, b.Vertex ) );
-          }
-      };
-      typedef std::set< TNode, TNodeCompare > TNodes;
-
-    public:
+      TSeeds& GetSeeds( );
       const TSeeds& GetSeeds( ) const;
 
       virtual void SetSeed( const TVertex& seed );