Calculate aspect ratio for DataAxis

hyperspy · Aug 9, 2023 · 38c8cc0 · 38c8cc0
1 parent 29d89d4
commit 38c8cc0
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/hyperspy/drawing/image.py b/hyperspy/drawing/image.py
@@ -246,10 +246,16 @@ def _calculate_aspect(self):
                 factor = min_asp ** -1 * xaxis.size / yaxis.size
                 self._auto_scalebar = False
                 self._auto_axes_ticks = True
-        if xaxis.is_uniform and yaxis.is_uniform:
-            self._aspect = abs(factor * xaxis.scale / yaxis.scale)
+        ## TODO: implement this for functional axis
+        if xaxis.is_uniform:
+            xscale = xaxis.scale
+        else:
+            xscale = (xaxis.axis[-1] - xaxis.axis[0]) / xaxis.size
+        if yaxis.is_uniform:
+            yscale = yaxis.scale
         else:
-            self._aspect = 1.0
+            yscale = (yaxis.axis[-1] - yaxis.axis[0]) / yaxis.size
+        self._aspect = abs(factor * xscale / yscale)
 
     def _calculate_vmin_max(self, data, auto_contrast=False,
                             vmin=None, vmax=None):