Enable creating, saving & showing empty result files

Author: davidbhr

saenopy/gui/solver/modules/exporter/ExportRenderCommon.py

@@ -19,32 +19,15 @@ def get_mesh_arrows(params, result):
     return None, None, {}, ""
 
 
-def get_mesh_extent(params, result):
-    if params["arrows"] == "piv":
-        mesh = result.mesh_piv[params["time"]["t"]]
-        if mesh is not None and mesh.displacements_measured is not None:
-            return [mesh.nodes.min(axis=0) * 1e6, mesh.nodes.max(axis=0) * 1e6]
-    elif params["arrows"] == "target deformations":
-        M = result.solvers[params["time"]["t"]]
-        if M is not None:
-            return [M.mesh.nodes.min(axis=0) * 1e6, M.mesh.nodes.max(axis=0) * 1e6]
-    elif params["arrows"] == "fitted deformations":
-        M = result.solvers[params["time"]["t"]]
-        if M is not None:
-            return [M.mesh.nodes.min(axis=0) * 1e6, M.mesh.nodes.max(axis=0) * 1e6]
-    elif params["arrows"] == "fitted forces":
-        M = result.solvers[params["time"]["t"]]
-        if M is not None:
-            return [M.mesh.nodes.min(axis=0) * 1e6, M.mesh.nodes.max(axis=0) * 1e6]
+def get_mesh_extent(params, result): 
+    mesh, field = result.get_field_data(params["arrows"], params["time"]["t"]) 
+    if mesh is None:
+        return None
     else:
-        M = result.solvers[params["time"]["t"]]
-        if M is not None:
-            return [M.mesh.nodes.min(axis=0) * 1e6, M.mesh.nodes.max(axis=0) * 1e6]
-        else:
-            M = result.mesh_piv[params["time"]["t"]]
-            if M is not None:
-                return [M.mesh.nodes.min(axis=0) * 1e6, M.mesh.nodes.max(axis=0) * 1e6]
-    return None
+        return [mesh.nodes.min(axis=0) * 1e6, mesh.nodes.max(axis=0) * 1e6]
+  
+
+
 
 
 def getVectorFieldImage(result, params, use_fixed_contrast_if_available=False, use_2D=False, exporter=None):

saenopy/gui/solver/modules/exporter/ExporterRender3D.py

@@ -72,6 +72,8 @@ def render_3d_fibers(params, result, plotter, exporter):
     t = params["time"]["t"]
     t_start = time.time()
     stack_data = None
+    if len(result.stacks) == 0:
+        return
     stack = result.stacks[t]
 
     if params["stack"]["use_reference_stack"]:

saenopy/result_file.py

@@ -429,7 +429,7 @@ def apply_delete(obj_data, rename):
             data_dict["___save_version__"] = "1.4"
         return super().from_dict(data_dict)
 
-    def __init__(self, output=None, template=None, stack=None, time_delta=None, **kwargs):
+    def __init__(self, output=None, template="", stack=None, time_delta=None, **kwargs):
         if output is not None:
             self.output = str(Path(output).absolute())
 
@@ -551,11 +551,11 @@ def get_data_structure(self):
             "dimensions": 3,
             "time_point_count": len(self.stacks),
             "has_reference": self.stack_reference is not None,
-            "z_slices_count": self.stacks[0].shape[2],
-            "im_shape": self.stacks[0].shape,
-            "voxel_size": self.stacks[0].voxel_size,
+            "z_slices_count": self.stacks[0].shape[2] if self.stacks else 0,
+            "im_shape": self.stacks[0].shape if self.stacks else (0,0,0),
+            "voxel_size": self.stacks[0].voxel_size if self.stacks else (0,0,0),
             "time_delta": self.time_delta,
-            "channels": self.stacks[0].channels,
+            "channels": self.stacks[0].channels if self.stacks else [],
             "fields": {
                 "piv": {
                     "type": "vector",
@@ -599,20 +599,23 @@ def get_image_data(self, time_point, channel="default", use_reference=False):
         return stack[:, :, :, :, channel]
 
     def get_field_data(result, name, time_point):
-        if name == "piv":
-            mesh = result.mesh_piv[time_point]
-            if mesh is not None and mesh.displacements_measured is not None:
-                return mesh, mesh.displacements_measured
-        elif name == "target deformations":
-            M = result.solvers[time_point]
-            if M is not None:
-                return M.mesh, M.mesh.displacements_target
-        elif name == "fitted deformations":
-            M = result.solvers[time_point]
-            if M is not None:
-                return M.mesh, M.mesh.displacements
-        elif name == "fitted forces":
-            M = result.solvers[time_point]
-            if M is not None:
-                return M.mesh, -M.mesh.forces * M.mesh.regularisation_mask[:, None]
+        try:
+            if name == "piv":
+                mesh = result.mesh_piv[time_point]
+                if mesh is not None and mesh.displacements_measured is not None:
+                    return mesh, mesh.displacements_measured
+            elif name == "target deformations":
+                M = result.solvers[time_point]
+                if M is not None:
+                    return M.mesh, M.mesh.displacements_target
+            elif name == "fitted deformations":
+                M = result.solvers[time_point]
+                if M is not None:
+                    return M.mesh, M.mesh.displacements
+            elif name == "fitted forces":
+                M = result.solvers[time_point]
+                if M is not None:
+                    return M.mesh, -M.mesh.forces * M.mesh.regularisation_mask[:, None]
+        except IndexError:
+            return None, None         
         return None, None