Lazify hs.plot.markers

Author: ericpre

hyperspy/drawing/markers.py

@@ -16,7 +16,6 @@
 # You should have received a copy of the GNU General Public License
 # along with HyperSpy. If not, see <https://www.gnu.org/licenses/#GPL>.
 
-import logging
 from copy import deepcopy
 
 import matplotlib.collections as mpl_collections
@@ -25,11 +24,7 @@
 from matplotlib.transforms import IdentityTransform
 
 from hyperspy.events import Event, Events
-from hyperspy.misc._markers import markers_dict_to_markers
-from hyperspy.misc.dask_utils import _get_navigation_dimension_chunk_slice
-from hyperspy.misc.utils import is_dask_array, isiterable
-
-_logger = logging.getLogger(__name__)
+from hyperspy.misc import _markers, dask_utils, utils
 
 
 def convert_positions(peaks, signal_axes):
@@ -194,16 +189,16 @@ def __init__(
         for key, value in self.kwargs.items():
             # Populate `_iterable_argument_keys`
             if (
-                isiterable(value)
+                utils.isiterable(value)
                 and not isinstance(value, str)
                 and key != self._position_key
             ):
                 self._iterable_argument_keys.append(key)
 
             # Handling dask arrays
-            if is_dask_array(value) and value.dtype == object:
+            if utils.is_dask_array(value) and value.dtype == object:
                 self.dask_kwargs[key] = self.kwargs[key]
-            elif is_dask_array(value):  # and value.dtype != object:
+            elif utils.is_dask_array(value):  # and value.dtype != object:
                 self.kwargs[key] = value.compute()
             # Patches or verts shouldn't be cast to array
             elif (
@@ -414,7 +409,7 @@ def remove_items(self, indices, keys=None, navigation_indices=None):
             # Don't remove when it doesn't have the same length as the
             # position kwargs because it is a "cycling" argument
             if (
-                isiterable(value)
+                utils.isiterable(value)
                 and not isinstance(value, str)
                 and len(value) == len(self.kwargs[self._position_key])
             ):
@@ -498,7 +493,7 @@ def _get_cache_dask_kwargs_chunk(self, indices):
 
         chunks = {key: value.chunks for key, value in self.dask_kwargs.items()}
         chunk_slices = {
-            key: _get_navigation_dimension_chunk_slice(indices, chunk)
+            key: dask_utils._get_navigation_dimension_chunk_slice(indices, chunk)
             for key, chunk in chunks.items()
         }
         to_compute = {}
@@ -612,7 +607,7 @@ def from_signal(
         return cls(**kwargs)
 
     def __deepcopy__(self, memo):
-        new_marker = markers_dict_to_markers(self._to_dictionary())
+        new_marker = _markers.markers_dict_to_markers(self._to_dictionary())
         return new_marker
 
     def _to_dictionary(self):
@@ -843,4 +838,6 @@ def plot_colorbar(self):
 
 
 def is_iterating(arg):
-    return (isinstance(arg, np.ndarray) or is_dask_array(arg)) and arg.dtype == object
+    return (
+        isinstance(arg, np.ndarray) or utils.is_dask_array(arg)
+    ) and arg.dtype == object

hyperspy/drawing/utils.py

@@ -40,7 +40,7 @@
 from hyperspy import signals
 from hyperspy.defaults_parser import preferences
 from hyperspy.docstrings.signal import HISTOGRAM_BIN_ARGS, HISTOGRAM_RANGE_ARGS
-from hyperspy.misc.utils import is_dask_array, isiterable, to_numpy
+from hyperspy.misc import utils
 
 _logger = logging.getLogger(__name__)
 
@@ -507,11 +507,11 @@ def _transpose_if_required(signal, expected_dimension):
 def _parse_array(signal, normalise=False):
     """Convenience function to parse array from a signal."""
     data = signal.data
-    if is_dask_array(data):
+    if utils.is_dask_array(data):
         data = data.compute()
     if normalise:
         data = (data - data.min()) / (data.max() - data.min())
-    return to_numpy(data)
+    return utils.to_numpy(data)
 
 
 def plot_images(
@@ -929,7 +929,7 @@ def __check_single_colorbar(cbar):
 
     # Get the figure from ax is provided
     if ax is not None:
-        if isiterable(ax):
+        if utils.isiterable(ax):
             if isinstance(ax, np.ndarray):
                 # plt.subplots can return numpy array
                 # convert and flatten to support list and array
@@ -1057,7 +1057,7 @@ def transparent_single_color_cmap(color):
                 ax = fig.add_axes([0, 0, 1, 1])
             else:
                 ax = fig.add_subplot()
-        elif isiterable(ax):
+        elif utils.isiterable(ax):
             raise ValueError(
                 "When using `overlay=True`, `ax` must be a matplotlib axis."
             )
@@ -1133,7 +1133,7 @@ def transparent_single_color_cmap(color):
     # Below is for non-overlayed images
     else:
         if ax is not None:
-            if not isiterable(ax):
+            if not utils.isiterable(ax):
                 ax = (ax,)
 
         # Loop through each image, adding subplot for each one
@@ -1633,13 +1633,13 @@ def _reverse_legend(ax_, legend_loc_):
             raise ValueError("The `ax` parameter is not supported for 'heatmap' style.")
         # To avoid ambiguity, don't support iterable with overalp and cascase style
         elif style in ["overlap", "cascade"]:
-            if isiterable(ax):
+            if utils.isiterable(ax):
                 raise ValueError(
                     "When using 'overlap' or 'cascade' style, `ax` must be a matplotlib axis."
                 )
             fig = ax.get_figure()
         else:
-            if isiterable(ax):
+            if utils.isiterable(ax):
                 if isinstance(ax, np.ndarray):
                     # plt.subplots can return numpy array
                     # convert and flatten to support list and array

hyperspy/utils/markers.py

@@ -34,18 +34,9 @@
 
 """
 
-from hyperspy.drawing._markers.arrows import Arrows
-from hyperspy.drawing._markers.circles import Circles
-from hyperspy.drawing._markers.ellipses import Ellipses
-from hyperspy.drawing._markers.horizontal_lines import HorizontalLines
-from hyperspy.drawing._markers.lines import Lines
-from hyperspy.drawing._markers.points import Points
-from hyperspy.drawing._markers.polygons import Polygons
-from hyperspy.drawing._markers.rectangles import Rectangles
-from hyperspy.drawing._markers.squares import Squares
-from hyperspy.drawing._markers.texts import Texts
-from hyperspy.drawing._markers.vertical_lines import VerticalLines
-from hyperspy.drawing.markers import Markers
+import importlib
+
+# ruff: noqa: F822
 
 __all__ = [
     "Arrows",
@@ -62,6 +53,29 @@
     "VerticalLines",
 ]
 
+_import_mapping = {
+    "Arrows": "_markers.arrows",
+    "Circles": "_markers.circles",
+    "Ellipses": "_markers.ellipses",
+    "HorizontalLines": "_markers.horizontal_lines",
+    "Lines": "_markers.lines",
+    "Markers": "markers",
+    "Points": "_markers.points",
+    "Polygons": "_markers.polygons",
+    "Rectangles": "_markers.rectangles",
+    "Squares": "_markers.squares",
+    "Texts": "_markers.texts",
+    "VerticalLines": "_markers.vertical_lines",
+}
+
 
 def __dir__():
     return sorted(__all__)
+
+
+def __getattr__(name):
+    if name in __all__:
+        import_path = f"hyperspy.drawing.{_import_mapping.get(name)}"
+        return getattr(importlib.import_module(import_path), name)
+
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")