fix: Fixed image saving with missing metadata

MIC-DKFZ · Aug 13, 2024 · bd542f1 · bd542f1
1 parent c99a609
commit bd542f1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/medvol/__init__.py b/medvol/__init__.py
@@ -1,3 +1,3 @@
-__version__ = "0.0.5"
+__version__ = "0.0.6"
 
 from medvol.medvol import MedVol
diff --git a/medvol/medvol.py b/medvol/medvol.py
@@ -83,9 +83,13 @@ def _load(self, filepath):
 
     def save(self, filepath):
         image_sitk = sitk.GetImageFromArray(self.array)
-        image_sitk.SetSpacing(self.spacing.tolist()[::-1])
-        image_sitk.SetOrigin(self.origin.tolist()[::-1])
-        image_sitk.SetDirection(self.direction.flatten().tolist()[::-1])
-        for key, value in self.header.items():
-            image_sitk.SetMetaData(key, value) 
+        if self.spacing is not None:
+            image_sitk.SetSpacing(self.spacing.tolist()[::-1])
+        if self.origin is not None:
+            image_sitk.SetOrigin(self.origin.tolist()[::-1])
+        if self.direction is not None:
+            image_sitk.SetDirection(self.direction.flatten().tolist()[::-1])
+        if self.header is not None:
+            for key, value in self.header.items():
+                image_sitk.SetMetaData(key, value) 
         sitk.WriteImage(image_sitk, filepath)