refactor: split utils.py into _geometry / _datashader / _color / _validate (#715)

Author: timtreis

src/spatialdata_plot/pl/_color.py

@@ -29,6 +29,21 @@
 
 from spatialdata_plot._accessor import register_spatial_data_accessor
 from spatialdata_plot._logging import _log_context, logger
+from spatialdata_plot.pl._color import (
+    _maybe_set_colors,
+    _prepare_cmap_norm,
+    _set_outline,
+)
+from spatialdata_plot.pl._validate import (
+    _expand_color_panels,
+    _validate_as_points_size,
+    _validate_graph_render_params,
+    _validate_image_render_params,
+    _validate_label_render_params,
+    _validate_points_render_params,
+    _validate_shape_render_params,
+    _validate_show_parameters,
+)
 from spatialdata_plot.pl.render import (
     _draw_channel_legend,
     _render_graph,
@@ -61,24 +76,13 @@
 from spatialdata_plot.pl.utils import (
     _RENDER_CMD_TO_CS_FLAG,
     _draw_scalebar,
-    _expand_color_panels,
     _get_cs_contents,
     _get_elements_to_be_rendered,
     _get_extent_fast,
     _get_valid_cs,
     _get_wanted_render_elements,
-    _maybe_set_colors,
     _mpl_ax_contains_elements,
-    _prepare_cmap_norm,
     _prepare_params_plot,
-    _set_outline,
-    _validate_as_points_size,
-    _validate_graph_render_params,
-    _validate_image_render_params,
-    _validate_label_render_params,
-    _validate_points_render_params,
-    _validate_shape_render_params,
-    _validate_show_parameters,
     _verify_plotting_tree,
     save_fig,
 )

src/spatialdata_plot/pl/_datashader.py

@@ -32,12 +32,38 @@
 from xarray import DataTree
 
 from spatialdata_plot._logging import _log_context, logger
+from spatialdata_plot.pl._color import (
+    _align_outline_vector_to_length,
+    _apply_mask_to_outline_vectors,
+    _color_vector_to_rgba,
+    _get_colors_for_categorical_obs,
+    _get_linear_colormap,
+    _make_continuous_mappable,
+    _map_color_seg,
+    _maybe_set_colors,
+    _prepare_cmap_norm,
+    _set_color_source_vec,
+)
 from spatialdata_plot.pl._datashader import (
+    _ax_show_and_transform,
     _build_ds_colorbar,
+    _datashader_canvas_from_dataframe,
+    _get_extent_and_range_for_datashader_canvas,
+    _hex_no_alpha,
+    _prepare_transformation,
     _render_ds_image,
     _render_ds_outlines,
     _shade_datashader_aggregate,
 )
+from spatialdata_plot.pl._geometry import (
+    _build_shape_patches,
+    _convert_shapes,
+    _get_collection_shape,
+    _validate_polygons,
+)
+from spatialdata_plot.pl._validate import (
+    _check_obs_var_shadow,
+)
 from spatialdata_plot.pl.render_params import (
     ChannelLegendEntry,
     CmapParams,
@@ -53,35 +79,15 @@
     _DsReduction,
 )
 from spatialdata_plot.pl.utils import (
-    _align_outline_vector_to_length,
-    _apply_mask_to_outline_vectors,
-    _ax_show_and_transform,
-    _build_shape_patches,
-    _check_obs_var_shadow,
-    _color_vector_to_rgba,
-    _convert_shapes,
-    _datashader_canvas_from_dataframe,
     _decorate_axs,
     _fast_extent,
-    _get_collection_shape,
-    _get_colors_for_categorical_obs,
-    _get_extent_and_range_for_datashader_canvas,
-    _get_linear_colormap,
-    _hex_no_alpha,
     _join_table_for_element,
-    _make_continuous_mappable,
-    _map_color_seg,
-    _maybe_set_colors,
     _mpl_ax_contains_elements,
     _multiscale_to_spatial_image,
     _pixel_to_coord,
-    _prepare_cmap_norm,
-    _prepare_transformation,
     _rasterize_if_necessary,
     _rasterize_if_necessary_datashader,
-    _set_color_source_vec,
     _stream_label_centroid_stats,
-    _validate_polygons,
 )
 
 _Normalize = Normalize | abc.Sequence[Normalize]