Fully fixes matplotlib#2 by fixing the scale bar too

francisco-dlp · francisco-dlp · commit 7b7fcbb73c0b · 2011-09-20T14:50:31.000+01:00
diff --git a/hyperspy/signal.py b/hyperspy/signal.py
@@ -257,10 +257,12 @@ def plot(self, axes_manager=None):
             if self.axes_manager._non_slicing_axes:
                 self._plot.image_data_function = self._get_explorer
                 self._plot.image_title = ''
-                self._plot.pixel_size = \
-                self.axes_manager._non_slicing_axes[0].scale
-                self._plot.pixel_units = \
-                self.axes_manager._non_slicing_axes[0].units
+                if self.axes_manager.navigation_dimension == 1:
+                    scalebar_axis = self.axes_manager._slicing_axes[0]
+                else:
+                    scalebar_axis = self.axes_manager._non_slicing_axes[-1]
+                self._plot.pixel_size = scalebar_axis.scale
+                self._plot.pixel_units = scalebar_axis.units
             self._plot.plot()
 
         elif axes_manager.signal_dimension == 2:
