X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=lib%2Ffpa%2FImage%2FRegionGrow.h;fp=lib%2Ffpa%2FImage%2FRegionGrow.h;h=b12a869d87ccdca57610f351da531a9b71e30bc0;hb=d3bb16bf060b7249a9ed1a49e6b118ca9394a22a;hp=dd7c851a5bf5324ad12b72d101a9f59910e57232;hpb=a1f9f98345f6d717e908f0507f60fd2841a73a1c;p=FrontAlgorithms.git

diff --git a/lib/fpa/Image/RegionGrow.h b/lib/fpa/Image/RegionGrow.h
index dd7c851..b12a869 100644
--- a/lib/fpa/Image/RegionGrow.h
+++ b/lib/fpa/Image/RegionGrow.h
@@ -14,22 +14,22 @@ namespace fpa
     /**
      * @param I Input image type
      */
-    template< class I >
+    template< class I, class O = I, class CC = fpa::Image::Functors::CastVertexValueToCost< typename I::PixelType, typename O::PixelType > >
     class RegionGrow
-      : public Algorithm< I, fpa::Base::RegionGrow< typename I::IndexType, typename I::PixelType, typename I::PixelType, itk::Functor::IndexLexicographicCompare< I::ImageDimension >, itk::ImageToImageFilter< I, I > > >
+      : public Algorithm< I, fpa::Base::RegionGrow< typename I::IndexType, typename O::PixelType, typename I::PixelType, itk::Functor::IndexLexicographicCompare< I::ImageDimension >, itk::ImageToImageFilter< I, O > >, CC >
     {
     public:
       // Standard class typdedefs
       typedef typename I::IndexType TVertex;
-      typedef typename I::PixelType TResult;
+      typedef typename O::PixelType TResult;
       typedef typename I::PixelType TVertexValue;
-      typedef itk::ImageToImageFilter< I, I > TBaseFilter;
+      typedef itk::ImageToImageFilter< I, O > TBaseFilter;
       typedef fpa::Base::RegionGrow< TVertex, TResult, TVertexValue, itk::Functor::IndexLexicographicCompare< I::ImageDimension >, TBaseFilter > TBaseAlgorithm;
 
-      typedef RegionGrow                      Self;
-      typedef Algorithm< I, TBaseAlgorithm >  Superclass;
-      typedef itk::SmartPointer< Self >       Pointer;
-      typedef itk::SmartPointer< const Self > ConstPointer;
+      typedef RegionGrow                         Self;
+      typedef Algorithm< I, TBaseAlgorithm, CC > Superclass;
+      typedef itk::SmartPointer< Self >          Pointer;
+      typedef itk::SmartPointer< const Self >    ConstPointer;
 
       typedef
       fpa::Image::Functors::ImageFunction< I, bool >