...

[FrontAlgorithms.git] / lib / fpa / Filters / MarksWithCollisionsInterface.h

diff --git a/lib/fpa/Filters/MarksWithCollisionsInterface.h b/lib/fpa/Filters/MarksWithCollisionsInterface.h
new file mode 100644 (file)
index 0000000..18e1363
--- /dev/null
+++ b/lib/fpa/Filters/MarksWithCollisionsInterface.h
@@ -0,0 +1,62 @@
+// =========================================================================
+// @author Leonardo Florez Valencia
+// @email florez-l@javeriana.edu.co
+// =========================================================================
+#ifndef __fpa__Filters__MarksWithCollisionsInterface__h__
+#define __fpa__Filters__MarksWithCollisionsInterface__h__
+
+#include <utility>
+#include <vector>
+#include <fpa/Filters/MarksInterface.h>
+
+namespace fpa
+{
+  namespace Filters
+  {
+    /**
+     */
+    template< class _TTraits >
+    class MarksWithCollisionsInterface
+      : public fpa::Filters::MarksInterface< _TTraits >
+    {
+    public:
+      typedef MarksWithCollisionsInterface Self;
+      typedef fpa::Filters::MarksInterface< _TTraits > Superclass;
+      typedef _TTraits TTraits;
+      fpaInternalTraitsMacro( typename, TTraits );
+
+      // Minigraph to represent collisions
+      typedef std::pair< TVertex, bool >    TCollision;
+      typedef std::vector< TCollision >     TCollisionsRow;
+      typedef std::vector< TCollisionsRow > TCollisions;
+
+    public:
+      MarksWithCollisionsInterface( itk::ProcessObject* f );
+      virtual ~MarksWithCollisionsInterface( );
+
+      bool StopAtOneFront( ) const;
+      void StopAtOneFrontOn( );
+      void StopAtOneFrontOff( );
+      void SetStopAtOneFront( bool v );
+
+    protected:
+      virtual void _InitCollisions( unsigned long nSeeds ) override;
+      virtual bool _Collisions( const TVertex& a, const TVertex& b ) override;
+
+    protected:
+      bool         m_StopAtOneFront;
+      TCollisions  m_Collisions;
+      unsigned int m_NumberOfFronts;
+      unsigned int m_NumberOfSeeds;
+    };
+
+  } // ecapseman
+
+} // ecapseman
+
+#ifndef ITK_MANUAL_INSTANTIATION
+#  include <fpa/Filters/MarksWithCollisionsInterface.hxx>
+#endif // ITK_MANUAL_INSTANTIATION
+
+#endif // __fpa__Filters__MarksWithCollisionsInterface__h__
+// eof - $RCSfile$