fixed 2D image z-averaging

Author: rgerum

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

@@ -256,8 +256,8 @@ class Null:
                     self.input_arrows = QtShortCuts.QInputChoice(None, "arrows", "piv", values=["None", "piv", "target deformations", "fitted deformations", "fitted forces"])
                     self.input_arrows.valueChanged.connect(self.update_display)
                     self.input_arrows.valueChanged.connect(self.hide_arrow)
-                    self.input_average_range = QtShortCuts.QInputNumber(None, "averaging z thickness", min=0, max=0,
-                                                                        step=10)
+                    self.input_average_range = QtShortCuts.QInputNumber(None, "averaging z thickness +/-", min=0, max=0,
+                                                                        step=10, unit="µm")
                     self.input_average_range.valueChanged.connect(self.update_display)
                     QtShortCuts.currentLayout().addStretch()
 

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

@@ -132,7 +132,7 @@ def project_data(R, field, skip=1):
             max_length = scale_max * params_arrows["arrow_scale"]
 
         if is3D:
-            z_center = (params["averaging_size"] - result.stacks[0].shape[2] / 2) * display_image[1][2] * 1e-6
+            z_center = (params["stack"]["z"] - result.stacks[0].shape[2] / 2) * display_image[1][2] * 1e-6
             z_min = z_center - params["averaging_size"] * 1e-6
             z_max = z_center + params["averaging_size"] * 1e-6
 
@@ -145,12 +145,14 @@ def project_data(R, field, skip=1):
             length = np.linalg.norm(field, axis=1)
             angle = np.arctan2(field[:, 1], field[:, 0])
             field = pd.DataFrame(np.hstack((length[:, None], angle[:, None])), columns=["length", "angle"])
-        pil_image = add_quiver(pil_image, R, field.length, field.angle, max_length=max_length, cmap=colormap,
-                               alpha=alpha,
-                               scale=im_scale * aa_scale,
-                               width=params["2D_arrows"]["width"],
-                               headlength=params["2D_arrows"]["headlength"],
-                               headheight=params["2D_arrows"]["headheight"])
+        # safety check if all arrows where filtered out
+        if R.shape[0] != 0:
+            pil_image = add_quiver(pil_image, R, field.length, field.angle, max_length=max_length, cmap=colormap,
+                                   alpha=alpha,
+                                   scale=im_scale * aa_scale,
+                                   width=params["2D_arrows"]["width"],
+                                   headlength=params["2D_arrows"]["headlength"],
+                                   headheight=params["2D_arrows"]["headheight"])
     if return_scale:
         return pil_image, {"scale_max": scale_max, "colormap": colormap}
     return pil_image