...

[FrontAlgorithms.git] / lib / fpa / Common / SliceBySliceRandomWalker.h

diff --git a/lib/fpa/Common/SliceBySliceRandomWalker.h b/lib/fpa/Common/SliceBySliceRandomWalker.h
index f16fe647870bf84a48461c082767d5ff307e2186..ea88fe3c2cd65001372b1a31848fef91fed93bd5 100644 (file)
--- a/lib/fpa/Common/SliceBySliceRandomWalker.h
+++ b/lib/fpa/Common/SliceBySliceRandomWalker.h
@@ -38,6 +38,15 @@ namespace fpa
         fpa::Common::SliceBySliceRandomWalker, itk::ImageToImageFilter
         );
 
+      itkGetConstMacro( Epsilon, TScalar );
+      itkSetMacro( Epsilon, TScalar );
+
+      itkGetConstMacro( Beta, TScalar );
+      itkSetMacro( Beta, TScalar );
+
+      itkGetConstMacro( VesselnessThreshold, TScalar );
+      itkSetMacro( VesselnessThreshold, TScalar );
+
       ivqITKInputMacro( InputLabels, TLabels );
       ivqITKInputMacro( InputVesselness, TScalarImage );
 
@@ -50,7 +59,8 @@ namespace fpa
       void _Composite(
         typename TScalarImage::Pointer& composite,
         const TLabels* labels,
-        const TScalarImage* vesselness
+        const TScalarImage* vesselness,
+        const TScalar& maxVess
         );
 
       template< class _TSlicePtr, class _TInput >
@@ -72,6 +82,13 @@ namespace fpa
       // Purposely not implemented
       SliceBySliceRandomWalker( const Self& other );
       Self& operator=( const Self& other );
+
+    protected:
+      TScalar m_Epsilon;
+      TScalar m_Beta;
+      TScalar m_VesselnessThreshold;
+
+      TScalar m_VesselnessValue;
     };
 
   } // ecapseman