X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=lib%2FcpExtensions%2FAlgorithms%2FGulsunTekMedialness.h;h=15add9eacd806149d0b399f32cc60a1d8c5e2b7a;hb=a422d88602e9b49085ccfab0ef0f9f2d93fb6a1e;hp=979c41f6df6c207c83b434685ff7c1c88de0f147;hpb=3393941bf8f26babc7b592db434b40c1b747a687;p=cpPlugins.git

diff --git a/lib/cpExtensions/Algorithms/GulsunTekMedialness.h b/lib/cpExtensions/Algorithms/GulsunTekMedialness.h
index 979c41f..15add9e 100644
--- a/lib/cpExtensions/Algorithms/GulsunTekMedialness.h
+++ b/lib/cpExtensions/Algorithms/GulsunTekMedialness.h
@@ -9,15 +9,15 @@ namespace cpExtensions
   {
     /**
      */
-    template< class _TGradient >
+    template< class _TGradient, class _TMask = itk::Image< unsigned char, _TGradient::ImageDimension > >
     class GulsunTekMedialness
-      : public GradientImageFunctionBase< _TGradient >
+      : public GradientImageFunctionBase< _TGradient, _TMask >
     {
     public:
-      typedef GulsunTekMedialness                     Self;
-      typedef GradientImageFunctionBase< _TGradient > Superclass;
-      typedef itk::SmartPointer< Self >               Pointer;
-      typedef itk::SmartPointer< const Self >         ConstPointer;
+      typedef GulsunTekMedialness                             Self;
+      typedef GradientImageFunctionBase< _TGradient, _TMask > Superclass;
+      typedef itk::SmartPointer< Self >                       Pointer;
+      typedef itk::SmartPointer< const Self >                 ConstPointer;
 
       itkStaticConstMacro( Dimension, unsigned int, Superclass::Dimension );