Add --mha tag in clitkDicomRTStruct2Image to allow structure with special characters...

[clitk.git] / tools / clitkDicomRTStruct2Image.cxx

diff --git a/tools/clitkDicomRTStruct2Image.cxx b/tools/clitkDicomRTStruct2Image.cxx
index 04564a016f4e63a55f9a717378ee8bfa2cec9a66..302f65ced9ed72ec44964b0650b4e5b01c480832 100644 (file)
--- a/tools/clitkDicomRTStruct2Image.cxx
+++ b/tools/clitkDicomRTStruct2Image.cxx
@@ -68,11 +68,21 @@ int main(int argc, char * argv[]) {
         filter.SetCropMaskEnabled(args_info.crop_flag);
         filter.SetImageFilename(args_info.image_arg);  // Used to get spacing + origin
         name.erase(remove_if(name.begin(), name.end(), isspace), name.end());
-        std::string n = std::string(args_info.output_arg).append
-          (clitk::toString(num)).append
-          ("_").append
-          (name).append
-          (".mhd");
+        std::string n;
+        if (args_info.mha_flag) {
+          n = std::string(args_info.output_arg).append
+            (clitk::toString(num)).append
+            ("_").append
+            (name).append
+            (".mha");
+        }
+        else {
+          n = std::string(args_info.output_arg).append
+            (clitk::toString(num)).append
+            ("_").append
+            (name).append
+            (".mhd");
+        }
         if (args_info.verbose_flag) {
           std::cout << num << " " << roi->GetName() << " num=" << num << " : " << n << std::endl;
         }