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 7c5e5866f736f3329f6fca009090ae5d93b7c90b..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;
@@ -121,7 +121,10 @@ namespace itkSTMS
                                             itkSTMS::ParserOutputType*         desc   );
 
         void GenerateImageSequence();
-        void GenerateCSVFile();
+
+        //! To generate CSV File from all classes. The file contains the class ID, the number of pixels of this class and then all filtered values along time.
+        //! You can pass a delimiter (ie a record separator) to this function.
+        void GenerateCSVFile(const std::string &delimiter = std::string(", "));
 
     private:
         // Attributes