output 2D as png file

[STMS.git] / Lib / PrePostProcessing / itkSTMS_XMLFileParser.hxx

diff --git a/Lib/PrePostProcessing/itkSTMS_XMLFileParser.hxx b/Lib/PrePostProcessing/itkSTMS_XMLFileParser.hxx
index 191adb356b4928ad9cb49c53e77319e99227c513..130e518bfc5a2c342ab12cd69959bd6de81753b8 100755 (executable)
--- a/Lib/PrePostProcessing/itkSTMS_XMLFileParser.hxx
+++ b/Lib/PrePostProcessing/itkSTMS_XMLFileParser.hxx
@@ -69,6 +69,7 @@ itkSTMS_XMLFileParser::itkSTMS_XMLFileParser()
     xmlParams.experimentPath     = "";
     xmlParams.imageExtension     = "";
     xmlParams.outputImageExtension = "";
+    xmlParams.outputClassExtension = "";
     xmlParams.inputCommonRoot    = "";
     xmlParams.inputFolder        = "";
     xmlParams.maskImage          = "";
@@ -109,10 +110,18 @@ void itkSTMS_XMLFileParser::Update()
     xmlParams.outputImageExtension = DomObject->GetAttribute( "outputImageExtension" );
     if ( xmlParams.outputImageExtension == "" )
     {
-        std::cout << std::endl << std::endl <<  "outputImageExtension not set" << std::endl << std::endl;
-        std::exit( EXIT_FAILURE );
+        std::cout << std::endl << std::endl <<  "outputImageExtension not set, using same as input (imageExtension) : "<< xmlParams.imageExtension << std::endl << std::endl;
+        xmlParams.outputImageExtension = xmlParams.imageExtension;
     }
 
+    xmlParams.outputClassExtension = DomObject->GetAttribute( "outputClassExtension" );
+    if ( xmlParams.outputClassExtension == "" )
+    {
+        std::cout << std::endl << std::endl <<  "outputClassExtension not set, .nii (nifti) used by default" << std::endl << std::endl;
+        xmlParams.outputClassExtension = ".nii";
+    }
+
+
     // Read input parameters
     listInput = DomObject->GetChild( "listInput" );
     xmlParams.inputFolder = listInput->GetAttribute( "inputFolder" );