Sorted fixImages in case there is no image tag in the xml

matty-jones · Apr 24, 2018 · 555449d · 555449d
1 parent e094cf9
commit 555449d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/utils/fixImages/fixImages.py b/utils/fixImages/fixImages.py
@@ -17,6 +17,9 @@ def checkBonds(morphology, bondDict):
 
 
 def zeroOutImages(morphology):
+    if len(morphology['image']) == 0:
+        for _ in morphology['position']:
+            morphology['image'].append([0, 0, 0])
     for atomID, image in enumerate(morphology['image']):
         if image != [0, 0, 0]:
             morphology['image'][atomID] = [0, 0, 0]