plopp 25.6.0__py3-none-any.whl → 25.6.1__py3-none-any.whl

plopp/graphics/graphicalview.py

@@ -59,7 +59,7 @@ class GraphicalView(View):
         title: str | None = None,
         figsize: tuple[float, float] | None = None,
         format: Literal['svg', 'png'] | None = None,
-        legend: bool | tuple[float, float] = False,
+        legend: bool | tuple[float, float] = True,
         camera: Camera | None = None,
         autoscale: bool = True,
         ax: Any = None,

plopp/plotting/common.py

@@ -76,64 +76,28 @@ def to_variable(obj) -> sc.Variable:
     return out
 
 
-def _ensure_data_array_coords(
-    da: sc.DataArray, coords: list[str] | None
-) -> sc.DataArray:
-    if coords is None:
-        coords = list(da.dims)
-    elif missing := set(coords) - set(da.coords.keys()):
-        raise ValueError(f"Specified coords do not exist: {missing}")
-
-    # Remove unused coords
-    da = da.drop_coords(list(set(da.coords) - set(coords)))
-    # Assign dim coords where missing. The above checks ensure that all missing
-    # coords are dim coords, i.e., that `name in out.dims`.
-    da = da.assign_coords(
-        {
-            name: sc.arange(name, da.sizes[name], unit=None)
-            for name in set(coords) - set(da.coords)
-        }
-    )
-    return da
-
-
-def to_data_array(
-    obj: Plottable | list,
-    coords: list[str] | None = None,
-) -> sc.DataArray:
+def to_data_array(obj: Plottable | list) -> sc.DataArray:
     """
-    Convert an input to a DataArray, potentially adding fake coordinates if they are
-    missing.
+    Convert an input to a DataArray.
+    Returns a shallow copy of the input if it is already a DataArray.
 
     Parameters
     ----------
     obj:
         The input object to be converted.
-    coords:
-        If supplied, use these coords instead of the input's dimension coordinates.
     """
+    if isinstance(obj, sc.DataArray):
+        return obj.copy(deep=False)
     out = _maybe_to_variable(obj)
     if isinstance(out, sc.Variable):
         out = sc.DataArray(data=out)
     out = from_compatible_lib(out)
     if not isinstance(out, sc.DataArray):
         raise TypeError(f"Cannot convert input of type {type(obj)} to a DataArray.")
-    out = _ensure_data_array_coords(out, coords)
-    for name, coord in out.coords.items():
-        if not coord.dims:
-            raise ValueError(
-                "Input data cannot be plotted: it has a scalar coordinate along "
-                f"dimension {name}. Consider dropping this coordinate before plotting. "
-                f"Use ``data.drop_coords('{name}').plot()``."
-            )
-    for name in out.coords:
-        other_dims = [dim for dim in out.coords[name].dims if dim not in out.dims]
-        for dim in other_dims:
-            out.coords[name] = out.coords[name].mean(dim)
     return out
 
 
-def _check_size(da: sc.DataArray):
+def _check_size(da: sc.DataArray) -> None:
     """
     Prevent slow figure rendering by raising an error if the data array exceeds a
     default size.
@@ -149,7 +113,7 @@ def _check_size(da: sc.DataArray):
         )
 
 
-def check_not_binned(da: sc.DataArray):
+def check_not_binned(da: sc.DataArray) -> None:
     """
     Plopp cannot plot binned data.
     This function will raise an error if the input data is binned.
@@ -169,7 +133,7 @@ def check_not_binned(da: sc.DataArray):
         )
 
 
-def check_allowed_dtypes(da: sc.DataArray):
+def check_allowed_dtypes(da: sc.DataArray) -> None:
     """
     Currently, Plopp cannot plot data that contains vector and matrix dtypes.
     This function will raise an error if the input data type is not supported.
@@ -185,10 +149,108 @@ def check_allowed_dtypes(da: sc.DataArray):
         )
 
 
-def _all_dims_sorted(var, order='ascending'):
+def _all_dims_sorted(var, order='ascending') -> bool:
+    """
+    Check if all dimensions of a variable are sorted in the specified order.
+    This is used to ensure that the coordinates are sorted before plotting.
+    """
     return all(sc.allsorted(var, dim, order=order) for dim in var.dims)
 
 
+def _rename_dims_from_coords(da: sc.DataArray, coords: Iterable[str]) -> sc.DataArray:
+    """
+    If coordinates are provided, rename the dimensions of the data array to match the
+    names of the coordinates, so that they effectively become the dimension coordinates.
+    """
+    renamed_dims = {}
+    underlying_dims = set()
+    for dim in coords:
+        underlying = da.coords[dim].dims[-1]
+        if underlying in underlying_dims:
+            raise ValueError(
+                "coords: Cannot use more than one coordinate associated with "
+                f"the same underlying dimension ({underlying})."
+            )
+        renamed_dims[underlying] = dim
+        underlying_dims.add(underlying)
+    return da.rename_dims(**renamed_dims)
+
+
+def _add_missing_dimension_coords(da: sc.DataArray) -> sc.DataArray:
+    """
+    Add missing dimension coordinates to the data array.
+    If a dimension does not have a coordinate, it will be added with a range of values.
+    """
+    return da.assign_coords(
+        {
+            dim: sc.arange(dim, da.sizes[dim], unit=None)
+            for dim in da.dims
+            if dim not in da.coords
+        }
+    )
+
+
+def _drop_non_dimension_coords(da: sc.DataArray) -> sc.DataArray:
+    """
+    Drop all coordinates that are not dimension coordinates.
+    This is useful to ensure that only the coordinates that are actually used for
+    plotting are kept.
+    """
+    return da.drop_coords([name for name in da.coords if name not in da.dims])
+
+
+def _handle_coords_with_left_over_dimensions(da: sc.DataArray) -> sc.DataArray:
+    """
+    In some rare cases, the coordinate may have dimensions that are not present in the
+    data array. This can happen for example when a 2d coord with bin edges is sliced
+    along one dimension, leaving a coordinate with the other dimension, and a dimension
+    of length 2 (because of the bin edges) in th sliced dimension.
+
+    This function will handle such cases by averaging the coordinate over the left-over
+    dimensions, effectively reducing the coordinate to a single value for each
+    dimension that is not present in the data array.
+    """
+    return da.assign_coords(
+        {
+            name: da.coords[name].mean(
+                [dim for dim in da.coords[name].dims if dim not in da.dims]
+            )
+            for name in da.coords
+        }
+    )
+
+
+def _check_coord_sanity(da: sc.DataArray) -> None:
+    """
+    Warn if any coordinate is not sorted. This can lead to unpredictable results
+    when plotting.
+    Also, raise an error if any coordinate is scalar.
+    """
+    for name, coord in da.coords.items():
+        try:
+            if not (
+                _all_dims_sorted(coord, order='ascending')
+                or _all_dims_sorted(coord, order='descending')
+            ):
+                warnings.warn(
+                    'The input contains a coordinate with unsorted values '
+                    f'({name}). The results may be unpredictable. '
+                    'Coordinates can be sorted using '
+                    '`scipp.sort(data, dim="to_be_sorted", order="ascending")`.',
+                    RuntimeWarning,
+                    stacklevel=2,
+                )
+        except sc.DTypeError:
+            pass
+
+        if not coord.dims:
+            raise ValueError(
+                "Input data cannot be plotted: it has a scalar coordinate along "
+                f"dimension {name}. Consider dropping this coordinate before plotting. "
+                f"Use ``data.drop_coords('{name}').plot()``."
+            )
+
+
 def preprocess(
     obj: Plottable | list,
     name: str | None = None,
@@ -219,7 +281,7 @@ def preprocess(
     elif coords is not None:
         coords = list(coords)
 
-    out = to_data_array(obj, coords)
+    out = to_data_array(obj)
     check_not_binned(out)
     check_allowed_dtypes(out)
     if name is not None:
@@ -227,34 +289,11 @@ def preprocess(
     if not ignore_size:
         _check_size(out)
     if coords is not None:
-        renamed_dims = {}
-        for dim in coords:
-            underlying = out.coords[dim].dims[-1]
-            if underlying in renamed_dims:
-                raise ValueError(
-                    "coords: Cannot use more than one coordinate associated with "
-                    f"the same underlying dimension ({underlying})."
-                )
-            renamed_dims[underlying] = dim
-        out = out.rename_dims(**renamed_dims)
-    for n, coord in out.coords.items():
-        if (coord.ndim == 0) or (n not in out.dims):
-            continue
-        try:
-            if not (
-                _all_dims_sorted(coord, order='ascending')
-                or _all_dims_sorted(coord, order='descending')
-            ):
-                warnings.warn(
-                    'The input contains a coordinate with unsorted values '
-                    f'({n}). The results may be unpredictable. '
-                    'Coordinates can be sorted using '
-                    '`scipp.sort(data, dim="to_be_sorted", order="ascending")`.',
-                    RuntimeWarning,
-                    stacklevel=2,
-                )
-        except sc.DTypeError:
-            pass
+        out = _rename_dims_from_coords(out, coords)
+    out = _add_missing_dimension_coords(out)
+    out = _drop_non_dimension_coords(out)
+    out = _handle_coords_with_left_over_dimensions(out)
+    _check_coord_sanity(out)
     return out
 
 

plopp/plotting/xyplot.py

@@ -23,7 +23,17 @@ def _make_data_array(x: sc.Variable, y: sc.Variable) -> sc.DataArray:
     y:
         The variable to use as the data.
     """
-    return sc.DataArray(data=y, coords={x.dim: x})
+    if x.ndim != 1 or y.ndim != 1:
+        raise sc.DimensionError(
+            f'Expected 1 dimension, got {x.ndim} for x and {y.ndim} for y.'
+        )
+    dim = x.dim
+    return sc.DataArray(
+        data=sc.array(dims=[dim], values=y.values, variances=y.variances, unit=y.unit)
+        if y.dim != dim
+        else y,
+        coords={dim: x},
+    )
 
 
 def xyplot(
@@ -48,7 +58,4 @@ def xyplot(
     """
     x = Node(to_variable, x)
     y = Node(to_variable, y)
-    dim = x().dim
-    if dim != y().dim:
-        raise sc.DimensionError("Dimensions of x and y must match")
     return linefigure(Node(_make_data_array, x=x, y=y), **kwargs)

{plopp-25.6.0.dist-info → plopp-25.6.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plopp
-Version: 25.6.0
+Version: 25.6.1
 Summary: Visualization library for Scipp
 Author: Scipp contributors
 License: BSD 3-Clause License

{plopp-25.6.0.dist-info → plopp-25.6.1.dist-info}/RECORD

@@ -43,11 +43,11 @@ plopp/graphics/bbox.py,sha256=3WeMgu5LgWIXLJTYVT7_USwaAcS7cQPNCk6B0gvuNGo,3596
 plopp/graphics/camera.py,sha256=fXc1YH7Tp-rwKkcYeE-mHrxMz3SKauj5xJCRuKv-QPs,5323
 plopp/graphics/colormapper.py,sha256=u5cADKU9lrRkJ_a969i_5z-mdT0-lBnEonc0PW_xSXA,9643
 plopp/graphics/figures.py,sha256=PLWQLnQNWz8epubYlK7HLSkvPGJmTAmHjJM2DqpvLe8,2735
-plopp/graphics/graphicalview.py,sha256=u5sz3CebhVLoEeKPpzlBlPfRrRORxDmruyfhLXDbH00,9436
+plopp/graphics/graphicalview.py,sha256=07xpJVnvuNlaWGsJshg_wL81jG57JhRxI23E7_WYI2M,9435
 plopp/graphics/tiled.py,sha256=gXYZVAs6wRFKezzN3VlEvm7_z2zx3IDYfGZQZqrpL80,1386
 plopp/plotting/__init__.py,sha256=s25RIbKplHSRn0jjtLADPQl72JLf_KLRMRCfeMDI_UA,205
 plopp/plotting/__init__.pyi,sha256=95wYGfM5PT0Y0Ov_ue7rBrHvHxrtex-2WEYtfFzBvE0,475
-plopp/plotting/common.py,sha256=7xR2aWUWzKyxshlPn8B-OWQ5Hy8wiMn-qtPywkLsNRs,9515
+plopp/plotting/common.py,sha256=_DmUEpWk2n4uz2-KwhzdctbC0WvRU9GsAMFDgfPnUo8,10965
 plopp/plotting/inspector.py,sha256=fT4v342pqTzihKaweoGD4WaQ5eDdyngK7W451Mh2Ffw,3586
 plopp/plotting/mesh3d.py,sha256=KjAtlVlLYslwgP_uE067WtJRgNURkhaz0PPUuYLlJyw,2778
 plopp/plotting/plot.py,sha256=WQzw1vHGljWY1tQU1InQWe2-M-vCW2IRiL-v7cnw46w,3886
@@ -55,7 +55,7 @@ plopp/plotting/scatter.py,sha256=ybPWFHFwxt_Bzv3Dy-yAFCRHW5HQEggcNihpzWduH7A,316
 plopp/plotting/scatter3d.py,sha256=oizd-BCgmvNu_fSV6FQt9oe7lPn4h2Zc1ohKmor-d78,3819
 plopp/plotting/slicer.py,sha256=A3g26czD6GM3vBlVS-IBOnCOlpaERqUhjT-_POYJEHQ,5877
 plopp/plotting/superplot.py,sha256=0EnzgSmNrdPHXooqj6GNR0IAApEHyfjgxYIPgPVO_5o,1416
-plopp/plotting/xyplot.py,sha256=t_0GC2TlC6QZPKN4tfgLRUUBDhJ5zoNJjsRyYxLX_xI,1448
+plopp/plotting/xyplot.py,sha256=zi-RcsgonnR747aH3zp75at4ntgZ0t0rSdKEeVlReMI,1641
 plopp/widgets/__init__.py,sha256=s25RIbKplHSRn0jjtLADPQl72JLf_KLRMRCfeMDI_UA,205
 plopp/widgets/__init__.pyi,sha256=4hVX7I3VEjE3oy2AGVvguYsIgg7cVrhKGpX7xVgGe9Y,844
 plopp/widgets/box.py,sha256=v5Gd3sY0WXcc3UfkcXxZboPG02sfOMXIo50tmpxPwVs,1806
@@ -68,8 +68,8 @@ plopp/widgets/slice.py,sha256=ihiGEGGva_2WZIug-kizg1I5SFdMzu4_8DDzuHWIKeA,4212
 plopp/widgets/style.py,sha256=Fw4S8te0xDyT8-kWu4Ut35aLgyr98qMlWZRbXbd1-dk,184
 plopp/widgets/toolbar.py,sha256=ueNeOZSC2suAG7nCJzqwl8DmtWD5c2J8VkuP-XzJRwI,4683
 plopp/widgets/tools.py,sha256=9l7LmB8qG0OTt5ZJihJzqRna1ouxJthO_Ahnq7WKTtY,7172
-plopp-25.6.0.dist-info/licenses/LICENSE,sha256=fCqnkzCwkGneGtgOidkO2b3ed6SGZT0yhuworuJG0K0,1553
-plopp-25.6.0.dist-info/METADATA,sha256=pXa9A1kJV4J_ESLKfWthwIvFzQc0sXhArjZPiBkbq3A,4533
-plopp-25.6.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-plopp-25.6.0.dist-info/top_level.txt,sha256=Lk_GLFh37eX-3PO3c6bPkDpir7GjJmbQHYza3IpIphc,6
-plopp-25.6.0.dist-info/RECORD,,
+plopp-25.6.1.dist-info/licenses/LICENSE,sha256=fCqnkzCwkGneGtgOidkO2b3ed6SGZT0yhuworuJG0K0,1553
+plopp-25.6.1.dist-info/METADATA,sha256=1rzml_3ajA-prOwWEd_8KI8EfxZiyV6DRFcl3nTvf68,4533
+plopp-25.6.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+plopp-25.6.1.dist-info/top_level.txt,sha256=Lk_GLFh37eX-3PO3c6bPkDpir7GjJmbQHYza3IpIphc,6
+plopp-25.6.1.dist-info/RECORD,,