...

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

diff --git a/lib/cpExtensions/Algorithms/PolyLineParametricPathWriter.h b/lib/cpExtensions/Algorithms/PolyLineParametricPathWriter.h
index f699602d1583689012a911585f0058e089c51e12..deb6b30e68b1c713968e36cf11d927009226ac36 100644 (file)
--- 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: