Even more tests

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

diff --git a/lib/fpa/Image/DijkstraWithSphereBacktracking.h b/lib/fpa/Image/DijkstraWithSphereBacktracking.h
index 070a01a2706fb8f85189274307baa7f769a45e6b..febfcc9f134d7e82eea42dbf2dabd92c01c819b5 100644 (file)
--- a/lib/fpa/Image/DijkstraWithSphereBacktracking.h
+++ b/lib/fpa/Image/DijkstraWithSphereBacktracking.h
@@ -31,6 +31,10 @@ namespace fpa
       typedef typename Superclass::TTraits::TVertexCmp TVertexCmp;
       typedef std::map< TVertex, TVertex, TVertexCmp > TTree;
 
+      typedef unsigned short                         TMark;
+      typedef itk::Image< TMark, I::ImageDimension > TMarkImage;
+
+
       typedef typename Superclass::TEndEvent             TEndEvent;
       typedef typename Superclass::TBacktrackingEvent    TBacktrackingEvent;
       typedef typename Superclass::TEndBacktrackingEvent TEndBacktrackingEvent;
@@ -51,6 +55,11 @@ namespace fpa
       itkGetConstMacro( FinalTree, TTree );
       itkGetConstMacro( EndPoints, TVertices );
       itkGetConstMacro( BifurcationPoints, TVertices );
+      itkGetConstMacro( NumberOfBranches, TMark );
+
+    public:
+      TMarkImage* GetOutputMarkImage( );
+      const TMarkImage* GetOutputMarkImage( ) const;
 
     protected:
       DijkstraWithSphereBacktracking( );
@@ -72,6 +81,7 @@ namespace fpa
       TTree        m_FinalTree;
       TVertices    m_BifurcationPoints;
       TVertices    m_EndPoints;
+      TMark        m_NumberOfBranches;
     };
 
   } // ecapseman