X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=lib%2Ffpa%2FImage%2FDijkstraWithEndPointDetection.h;h=b32d68a2bbb5f8dc0cc6db563968a1321f8f654e;hb=f28d460c6d9ce93e55c073bd04dba8de16d55d3a;hp=4288da38e592d59333a2a4d6d3e8c8444f396c5b;hpb=b4ed6ddfa7e90e892f07cad9a760961bd4e84e6b;p=FrontAlgorithms.git

diff --git a/lib/fpa/Image/DijkstraWithEndPointDetection.h b/lib/fpa/Image/DijkstraWithEndPointDetection.h
index 4288da3..b32d68a 100644
--- a/lib/fpa/Image/DijkstraWithEndPointDetection.h
+++ b/lib/fpa/Image/DijkstraWithEndPointDetection.h
@@ -76,8 +76,19 @@ namespace fpa
       itkNewMacro( Self );
       itkTypeMacro( DijkstraWithEndPointDetection, Dijkstra );
 
+      itkBooleanMacro( CorrectSeeds );
+      itkBooleanMacro( CorrectEndPoints );
+
+      itkGetConstMacro( CorrectSeeds, bool );
+      itkGetConstMacro( CorrectEndPoints, bool );
+      itkGetConstMacro( SafetyNeighborhoodSize, unsigned int );
+
       itkGetConstMacro( NumberOfBranches, unsigned long );
 
+      itkSetMacro( CorrectSeeds, bool );
+      itkSetMacro( CorrectEndPoints, bool );
+      itkSetMacro( SafetyNeighborhoodSize, unsigned int );
+
     public:
       TLabelImage* GetLabelImage( );
       const TLabelImage* GetLabelImage( ) const;
@@ -125,8 +136,12 @@ namespace fpa
       unsigned int m_EndPointsIndex;
       unsigned int m_BranchesIndex;
 
-      _TCandidates    m_Candidates;
-      unsigned long   m_NumberOfBranches;
+      bool m_CorrectSeeds;
+      bool m_CorrectEndPoints;
+      unsigned int m_SafetyNeighborhoodSize;
+
+      _TCandidates  m_Candidates;
+      unsigned long m_NumberOfBranches;
     };
 
   } // ecapseman