...

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

diff --git a/lib/fpa/Image/ExtractAxisFilter.h b/lib/fpa/Image/ExtractAxisFilter.h
index 25eee47ed9315ecb2ab2b576e3c5d60fa9f8f52e..01efc75a66e163fd80662365813847c850b0e423 100644 (file)
--- a/lib/fpa/Image/ExtractAxisFilter.h
+++ b/lib/fpa/Image/ExtractAxisFilter.h
@@ -39,6 +39,7 @@ namespace fpa
 
       typedef typename TCenterness::OutputImageType TOutputImage;
       typedef typename TInputImage::IndexType       TIndex;
+      typedef typename TInputImage::PointType       TPoint;
 
       typedef fpa::Image::PolyLineParametricPath< Self::Dimension > TPath;
       typedef fpa::Image::Dijkstra< TOutputImage, TOutputImage > TDijkstra;
@@ -50,10 +51,16 @@ namespace fpa
       itkGetConstObjectMacro( Centerness, TCenterness );
       itkGetObjectMacro( Centerness, TCenterness );
 
+      itkGetConstMacro( StartIndex, TIndex );
+      itkGetConstMacro( EndIndex, TIndex );
+
+      itkSetMacro( StartIndex, TIndex );
+      itkSetMacro( EndIndex, TIndex );
+
     public:
       virtual itk::ModifiedTimeType GetMTime( ) const override;
 
-      virtual void AddSeed( const TVertex& seed );
+      virtual void AddSeed( const TIndex& seed );
       virtual void AddSeed( const TPoint& seed );
       virtual void ClearSeeds( );
 
@@ -78,6 +85,9 @@ namespace fpa
     protected:
       typename TCenterness::Pointer m_Centerness;
       typename TDijkstra::Pointer m_Dijkstra;
+
+      TIndex m_StartIndex;
+      TIndex m_EndIndex;
     };
 
   } // ecapseman