X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=lib%2FcpExtensions%2FAlgorithms%2FFluxMedialness.h;h=c25f5f47031209541fc4b37e66ffed60c5dfd72b;hb=8eaa2cf759c39233848cdd5b8346faccbea4c172;hp=f71da00b5526c3677c728aec2471d741c2bb64a8;hpb=1df9496bf2f5d456616947d377f991a1022ebf6f;p=cpPlugins.git

diff --git a/lib/cpExtensions/Algorithms/FluxMedialness.h b/lib/cpExtensions/Algorithms/FluxMedialness.h
index f71da00..c25f5f4 100644
--- a/lib/cpExtensions/Algorithms/FluxMedialness.h
+++ b/lib/cpExtensions/Algorithms/FluxMedialness.h
@@ -9,15 +9,15 @@ namespace cpExtensions
   {
     /**
      */
-    template< class _TGradient >
+    template< class _TGradient, class _TMask = itk::Image< unsigned char, _TGradient::ImageDimension > >
     class FluxMedialness
-      : public GradientImageFunctionBase< _TGradient >
+      : public GradientImageFunctionBase< _TGradient, _TMask >
     {
     public:
-      typedef FluxMedialness                          Self;
-      typedef GradientImageFunctionBase< _TGradient > Superclass;
-      typedef itk::SmartPointer< Self >               Pointer;
-      typedef itk::SmartPointer< const Self >         ConstPointer;
+      typedef FluxMedialness                                  Self;
+      typedef GradientImageFunctionBase< _TGradient, _TMask > Superclass;
+      typedef itk::SmartPointer< Self >                       Pointer;
+      typedef itk::SmartPointer< const Self >                 ConstPointer;
 
       itkStaticConstMacro( Dimension, unsigned int, Superclass::Dimension );
 
@@ -47,7 +47,7 @@ namespace cpExtensions
       FluxMedialness( );
       virtual ~FluxMedialness( );
 
-      virtual TOutput _Evaluate( const TIndex& i ) const ITK_OVERRIDE;
+      virtual TOutput _Evaluate( const TIndex& i ) const cpExtensions_OVERRIDE;
 
     private:
       // Purposely not implemented.