...

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

diff --git a/lib/fpa/Image/ExtractEndPointsAndBifurcationsFromMinimumSpanningTree.h b/lib/fpa/Image/ExtractEndPointsAndBifurcationsFromMinimumSpanningTree.h
index 312da367f18ad8d2bdc0c47828843410aabb9bf6..002afb4dba3898339455580c24d1d403830f7300 100644 (file)
--- a/lib/fpa/Image/ExtractEndPointsAndBifurcationsFromMinimumSpanningTree.h
+++ b/lib/fpa/Image/ExtractEndPointsAndBifurcationsFromMinimumSpanningTree.h
@@ -2,6 +2,7 @@
 #define __FPA__IMAGE__EXTRACTENDPOINTSANDBIFURCATIONSFROMMINIMUMSPANNINGTREE__H__
 
 #include <fpa/Base/ExtractEndPointsAndBifurcationsFromMinimumSpanningTree.h>
+#include <fpa/Base/ImageSkeleton.h>
 
 namespace fpa
 {
@@ -26,6 +27,7 @@ namespace fpa
       typedef typename Superclass::TVertex              TVertex;
 
       typedef itk::Image< unsigned short, TImage::ImageDimension > TMarkImage;
+      typedef fpa::Base::ImageSkeleton< _TMST > TSkeleton;
 
     public:
       itkNewMacro( Self );
@@ -41,6 +43,7 @@ namespace fpa
     public:
       const TImage* GetCostsImage( );
       const TImage* GetDistanceMap( );
+      TSkeleton* GetSkeleton( );
       void SetCostsImage( TImage* image );
       void SetDistanceMap( TImage* image );