output 2D as png file

[STMS.git] / Lib / PrePostProcessing / itkSTMS_TemporalSetToImageSequence.h

diff --git a/Lib/PrePostProcessing/itkSTMS_TemporalSetToImageSequence.h b/Lib/PrePostProcessing/itkSTMS_TemporalSetToImageSequence.h
index 996d65b8ce5e1e2546f573ac6d00c563fc5cd73d..1b53b3e57910ff07c5dfac7db346483cb867142b 100755 (executable)
--- a/Lib/PrePostProcessing/itkSTMS_TemporalSetToImageSequence.h
+++ b/Lib/PrePostProcessing/itkSTMS_TemporalSetToImageSequence.h
@@ -77,14 +77,14 @@
 
 namespace itkSTMS
 {
-    template < class ImageType, class ClassImageType >
+    template < class ImageType, class ClassImageType, class OutputImageType=ImageType>
     class itkSTMS_TemporalSetToImageSequence
     {
     public:
         // Reader and writer typedefs
         typedef itk::ImageFileReader< ImageType >       ReaderType;
         typedef typename ReaderType::Pointer            ReaderPointer;
-        typedef itk::ImageFileWriter< ImageType >       WriterType;
+        typedef itk::ImageFileWriter< OutputImageType > WriterType;
         typedef typename WriterType::Pointer            WriterPointer;
         typedef itk::ImageFileWriter< ClassImageType >  ClassWriterType;
         typedef typename ClassWriterType::Pointer       ClassWriterPointer;