...

[cpPlugins.git] / lib / cpExtensions / Algorithms / MFluxMedialness.h

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