X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=lib%2FcpExtensions%2FAlgorithms%2FPolyLineParametricPathWriter.h;fp=lib%2FcpExtensions%2FAlgorithms%2FPolyLineParametricPathWriter.h;h=deb6b30e68b1c713968e36cf11d927009226ac36;hb=9af34b55ea1d60eef3608aede00f9ef5a16dccdf;hp=f699602d1583689012a911585f0058e089c51e12;hpb=d976f28fd3120d75c30a521bb5987d25e78f72c2;p=cpPlugins.git

diff --git a/lib/cpExtensions/Algorithms/PolyLineParametricPathWriter.h b/lib/cpExtensions/Algorithms/PolyLineParametricPathWriter.h
index f699602..deb6b30 100644
--- a/lib/cpExtensions/Algorithms/PolyLineParametricPathWriter.h
+++ b/lib/cpExtensions/Algorithms/PolyLineParametricPathWriter.h
@@ -6,6 +6,7 @@
 #define __cpExtensions__Algorithms__PolyLineParametricPathWriter__h__
 
 #include <itkProcessObject.h>
+#include <itkImage.h>
 
 // -------------------------------------------------------------------------
 namespace cpExtensions
@@ -14,7 +15,7 @@ namespace cpExtensions
   {
     /**
      */
-    template< class _TPolyLine >
+    template< class _TPolyLine, class _TImage = itk::Image< unsigned char, _TPolyLine::PathDimension > >
     class PolyLineParametricPathWriter
       : public itk::ProcessObject
     {
@@ -38,6 +39,10 @@ namespace cpExtensions
     public:
       void SetInput( const _TPolyLine* input );
       const _TPolyLine* GetInput( );
+
+      void SetImage( const _TImage* image );
+      const _TImage* GetImage( );
+
       virtual void Update( );
 
     protected: