...

[cpPlugins.git] / lib / cpExtensions / DataStructures / Skeleton.h

diff --git a/lib/cpExtensions/DataStructures/Skeleton.h b/lib/cpExtensions/DataStructures/Skeleton.h
index f30b601714b59a490e03d33be62836041087af4c..671b8acc529f282c7fd1427dddb5d497ec1d1a25 100644 (file)
--- a/lib/cpExtensions/DataStructures/Skeleton.h
+++ b/lib/cpExtensions/DataStructures/Skeleton.h
@@ -3,6 +3,7 @@
 
 #include <cpExtensions/DataStructures/Graph.h>
 #include <cpExtensions/DataStructures/PolyLineParametricPath.h>
+#include <vector>
 
 namespace cpExtensions
 {
@@ -20,6 +21,8 @@ namespace cpExtensions
       typedef typename TIndex::LexicographicCompare TIndexCompare;
       typedef typename TPath::Pointer TPathPointer;
 
+      itkStaticConstMacro( Dimension, unsigned int, _VDim );
+
       typedef Graph< TIndex, TPathPointer, TIndex, TIndexCompare > Superclass;
       typedef Skeleton                        Self;
       typedef itk::SmartPointer< Self >       Pointer;
@@ -33,6 +36,9 @@ namespace cpExtensions
       void AddBranch( TPath* path );
       const TPath* GetBranch( const TIndex& a, const TIndex& b ) const;
 
+      std::vector< TIndex > GetEndPoints( ) const;
+      std::vector< TIndex > GetBifurcations( ) const;
+
     protected:
       Skeleton( );
       virtual ~Skeleton( );