anyplotlib 0.1.0b1__py3-none-any.whl

anyplotlib/plotxy/_plotxy.py

@@ -0,0 +1,273 @@
+"""
+_plotxy.py — ``PlotXY``: a blank **data-coordinate 2-D axis**.
+
+Where :class:`~anyplotlib.plot1d.Plot1D` is a curve over a (monotonic) x-axis,
+``PlotXY`` is a coordinate canvas in the matplotlib sense: you set ``xlim`` /
+``ylim`` (+ optional ``aspect="equal"``) and draw ``scatter`` / ``plot`` /
+``fill`` / ``text`` as **collection-style artists** in data coordinates — the
+surface orix's ``StereographicPlot`` (and an IPF triangle) needs.
+
+It is built on ``Plot1D`` because the 1-D marker collections
+(``add_points`` / ``add_lines`` / ``add_polygons`` / ``add_texts``) already map
+through the 1-D data→canvas transform, which is exactly matplotlib's
+``transLimits → transAxes`` (``_axisValToFrac`` for x + ``_valToPy1d`` for y →
+unit box → panel rect). The primary curve is hidden (``alpha=0``); the markers
+ARE the content. ``aspect="equal"`` is honoured by the renderer's xy-aspect step
+(``state["aspect"]``); without it the data simply fills the panel (auto aspect).
+
+Mirrors matplotlib: ``scatter`` → a ``PathCollection`` (offsets + per-point
+colours/sizes), ``plot`` → ``Line2D``, ``fill`` → ``Polygon``, ``text`` →
+``Text``.
+"""
+from __future__ import annotations
+
+import numpy as np
+
+from anyplotlib._utils import _build_colormap_lut
+from anyplotlib.plot1d._plot1d import Plot1D
+
+
+def _regular_grid_edges(x, y, rtol=1e-6):
+    """If ``x``/``y`` corner grids are separable + axis-aligned, return the 1-D
+    edge vectors ``(xe, ye)``; otherwise ``None``.
+
+    "Separable + axis-aligned" means every row of ``x`` is identical and every
+    column of ``y`` is identical (the usual ``np.meshgrid`` of 1-D edges) — the
+    case a single stretched raster reproduces exactly.  Spacing may be
+    non-uniform; ``drawImage`` only needs a common bounding box, and each cell
+    is one pixel, so non-uniform *edges* would distort — hence we additionally
+    require uniform spacing for the raster fast path (checked by the caller).
+    """
+    if x.shape != y.shape or x.ndim != 2:
+        return None
+    xe = x[0, :]
+    ye = y[:, 0]
+    if not np.allclose(x, xe[None, :], rtol=rtol, atol=0.0):
+        return None
+    if not np.allclose(y, ye[:, None], rtol=rtol, atol=0.0):
+        return None
+    return xe, ye
+
+
+def _is_uniform(v, rtol=1e-4):
+    """True if 1-D ``v`` is monotonic with (near-)constant spacing."""
+    d = np.diff(np.asarray(v, float))
+    if d.size == 0 or not np.all(np.isfinite(d)):
+        return False
+    if np.any(d > 0) and np.any(d < 0):
+        return False
+    step = d.mean()
+    if step == 0:
+        return False
+    return bool(np.allclose(d, step, rtol=rtol, atol=abs(step) * rtol))
+
+
+def _colseq(c, n):
+    """A matplotlib-style colour arg → ``list[str]`` of length ``n`` (or None)."""
+    if c is None:
+        return None
+    if isinstance(c, str):
+        return [c] * n
+    return [str(x) for x in c]
+
+
+class PlotXY(Plot1D):
+    """Coordinate-only 2-D axis: ``scatter`` / ``plot`` / ``fill`` / ``text`` in
+    data coordinates, with ``set_xlim`` / ``set_ylim`` / ``set_aspect``."""
+
+    #: Heavy state keys routed to the geometry channel (see Figure._push).
+    #: ``raster_geom`` holds the base64 RGBA bytes of any ``add_raster`` /
+    #: ``pcolormesh`` raster (keyed by marker-set id), so re-aiming the view
+    #: never re-transmits the image.  Declared here rather than on Plot1D so
+    #: plain line/bar panels keep the single-trait path (no empty geom trait).
+    _GEOM_KEYS = frozenset({"raster_geom"})
+
+    def __init__(self, *, xlim=(0.0, 1.0), ylim=(0.0, 1.0), aspect=None,
+                 units: str = "", y_units: str = ""):
+        super().__init__(
+            np.zeros(2, dtype=float),
+            x_axis=np.asarray([xlim[0], xlim[1]], dtype=float),
+            units=units, y_units=y_units, alpha=0.0,        # hidden primary curve
+        )
+        s = self._state
+        s["line_alpha"] = 0.0
+        s["data_min"] = float(ylim[0])
+        s["data_max"] = float(ylim[1])
+        s["y_range"] = [float(ylim[0]), float(ylim[1])]
+        s["aspect"] = "equal" if aspect == "equal" else None
+
+    # ── view (data limits = the transData domain) ────────────────────────────
+    def set_xlim(self, xmin: float, xmax: float) -> None:
+        self._state["x_axis"] = np.asarray([float(xmin), float(xmax)], dtype=float)
+        self._push()
+
+    def set_ylim(self, ymin: float, ymax: float) -> None:
+        self._state["y_range"] = [float(ymin), float(ymax)]
+        self._state["data_min"] = float(ymin)
+        self._state["data_max"] = float(ymax)
+        self._push()
+
+    def get_xlim(self) -> tuple:
+        x = self._state["x_axis"]
+        return (float(x[0]), float(x[-1]))
+
+    def get_ylim(self) -> tuple:
+        yr = self._state["y_range"]
+        return (float(yr[0]), float(yr[1]))
+
+    def set_aspect(self, aspect) -> None:
+        """``"equal"`` → one data unit is the same pixel length on x and y
+        (matplotlib ``apply_aspect``); ``"auto"`` / ``None`` → fill the panel."""
+        self._state["aspect"] = "equal" if aspect == "equal" else None
+        self._push()
+
+    def get_aspect(self):
+        return self._state.get("aspect")
+
+    # ── matplotlib-parity artists (each returns its collection MarkerGroup) ──
+    def scatter(self, x, y, *, s=8, c=None, facecolors=None,
+                edgecolors="#1f77b4", alpha=1.0, name=None):
+        """Data-coord scatter → a ``PathCollection``-style points group."""
+        x = np.asarray(x, float).ravel()
+        y = np.asarray(y, float).ravel()
+        offs = np.column_stack([x, y])
+        face = _colseq(c if c is not None else facecolors, len(x))
+        return self.add_points(offs, name=name, sizes=s, color=edgecolors,
+                               facecolors=face, alpha=alpha)
+
+    def plot(self, x, y, *, color="#1f77b4", linewidth=1.5, name=None):
+        """Data-coord polyline → a ``Line2D``-style lines group."""
+        pts = np.column_stack([np.asarray(x, float).ravel(),
+                               np.asarray(y, float).ravel()])
+        segs = [[pts[i].tolist(), pts[i + 1].tolist()] for i in range(len(pts) - 1)]
+        return self.add_lines(segs, name=name, edgecolors=color, linewidths=linewidth)
+
+    def fill(self, x, y, *, facecolor=None, edgecolor="#1f77b4", alpha=0.3,
+             linewidth=1.5, name=None):
+        """Data-coord filled polygon → a ``Polygon``-style group."""
+        verts = np.column_stack([np.asarray(x, float).ravel(),
+                                 np.asarray(y, float).ravel()]).tolist()
+        return self.add_polygons([verts], name=name, facecolors=facecolor,
+                                 edgecolors=edgecolor, alpha=alpha,
+                                 linewidths=linewidth)
+
+    def text(self, x, y, s, *, color="#000000", fontsize=12, name=None):
+        """Data-coord text → a ``Text``-style group (one label)."""
+        return self.add_texts([[float(x), float(y)]], [str(s)], name=name,
+                              color=color, fontsize=fontsize)
+
+    def pcolormesh(self, x, y, c, *, cmap="viridis", vmin=None, vmax=None,
+                   edgecolor=None, alpha=1.0, clip_path=None, smooth=False,
+                   name=None):
+        """Data-coord quad mesh — matplotlib ``pcolormesh``.
+
+        ``x``/``y`` are the ``(N+1, M+1)`` cell-corner grids and ``c`` the
+        ``(N, M)`` field: either a scalar array (mapped through *cmap* between
+        *vmin*/*vmax*) or an array of CSS colour strings. Masked / non-finite
+        cells are skipped — so an ``orix`` pole-density histogram (masked
+        outside the fundamental sector) clips itself to the sector. Drawn as
+        one polygon ``MarkerGroup`` with per-cell face colours (a
+        ``PathCollection``); the edges default to the face colour for a
+        seamless heatmap.
+
+        ``clip_path`` is an optional ``(K, 2)`` data-coord polygon the mesh is
+        clipped to (matplotlib ``set_clip_path``) — pass the curved sector
+        boundary so the edge cells don't overflow it.
+
+        **Fast path.** When the grid is a regular, axis-aligned, uniformly
+        spaced mesh of a scalar field (the common heatmap / density-histogram
+        case), the whole mesh is rendered as one stretched RGBA raster
+        (:meth:`add_raster`) instead of one polygon per cell — orders of
+        magnitude less data to serialise and one ``drawImage`` to draw.
+        Irregular meshes, colour-string ``c``, or an explicit ``edgecolor``
+        fall back to the per-cell polygon path.  ``smooth=True`` bilinearly
+        interpolates the raster for a smooth heat field (default is crisp
+        nearest-neighbour cells); it applies only on the raster fast path.
+        """
+        x = np.asarray(x, float)
+        y = np.asarray(y, float)
+        cm = np.ma.asarray(c)
+        nr, nc = cm.shape
+        mask = np.ma.getmaskarray(cm)
+
+        if cm.dtype.kind in "fiub":               # scalar field → LUT
+            vals = np.ma.filled(cm.astype(float), np.nan)
+            finite = vals[np.isfinite(vals)]
+            lo = float(vmin) if vmin is not None else (
+                float(finite.min()) if finite.size else 0.0)
+            hi = float(vmax) if vmax is not None else (
+                float(finite.max()) if finite.size else 1.0)
+            lut = _build_colormap_lut(cmap)
+            span = (hi - lo) or 1.0
+
+            # ── Raster fast path ────────────────────────────────────────────
+            edges = _regular_grid_edges(x, y) if edgecolor is None else None
+            if edges is not None:
+                xe, ye = edges
+                if _is_uniform(xe) and _is_uniform(ye):
+                    return self._pcolormesh_raster(
+                        xe, ye, vals, mask, lut, lo, span,
+                        alpha=alpha, clip_path=clip_path, smooth=smooth,
+                        name=name)
+
+            def _color(i, j):
+                v = vals[i, j]
+                if not np.isfinite(v):
+                    return None
+                t = min(1.0, max(0.0, (v - lo) / span))
+                r, g, b = lut[int(round(t * 255))]
+                return f"#{r:02x}{g:02x}{b:02x}"
+        else:                                     # already colour strings
+            def _color(i, j):
+                return None if mask[i, j] else str(cm[i, j])
+
+        verts, faces = [], []
+        for i in range(nr):
+            for j in range(nc):
+                if mask[i, j]:
+                    continue
+                col = _color(i, j)
+                if col is None:
+                    continue
+                verts.append([[x[i, j],     y[i, j]],
+                              [x[i + 1, j], y[i + 1, j]],
+                              [x[i + 1, j + 1], y[i + 1, j + 1]],
+                              [x[i, j + 1], y[i, j + 1]]])
+                faces.append(col)
+
+        edges = faces if edgecolor is None else edgecolor
+        return self.add_polygons(verts, name=name, facecolors=faces,
+                                 edgecolors=edges, alpha=alpha, linewidths=0.5,
+                                 clip_path=clip_path)
+
+    def _pcolormesh_raster(self, xe, ye, vals, mask, lut, lo, span, *,
+                           alpha, clip_path, smooth, name):
+        """Render a regular scalar mesh as one stretched RGBA raster.
+
+        ``vals``/``mask`` are ``(nr, nc)``; ``xe``/``ye`` the ``nc+1``/``nr+1``
+        uniform edge vectors.  Each cell maps to one pixel; colours come from
+        the same LUT / lo / span math as the polygon path so the two are
+        pixel-equivalent (up to per-cell quantisation).
+        """
+        lut_arr = np.asarray(lut, dtype=np.uint8)            # (256, 3)
+        opaque = (~mask) & np.isfinite(vals)
+        # NaN/masked cells become alpha 0; give them a safe index (0) so the
+        # cast never overflows — their RGB is invisible regardless.
+        t = np.clip((np.where(opaque, vals, lo) - lo) / span, 0.0, 1.0)
+        idx = np.rint(t * 255).astype(np.intp)               # matches int(round(...))
+        rgba = np.zeros((*vals.shape, 4), dtype=np.uint8)
+        rgba[..., :3] = lut_arr[idx]
+        rgba[..., 3] = np.where(opaque, 255, 0).astype(np.uint8)
+
+        # Orient so image row 0 = highest data-y, col 0 = lowest data-x —
+        # add_raster stretches the image across ``extent`` with row 0 at the
+        # top (max y) and col 0 at the left (min x).
+        if ye[0] <= ye[-1]:        # ascending y → flip rows so top = max y
+            rgba = rgba[::-1, :, :]
+        if xe[0] > xe[-1]:         # descending x → flip cols so left = min x
+            rgba = rgba[:, ::-1, :]
+
+        extent = (float(min(xe[0], xe[-1])), float(max(xe[0], xe[-1])),
+                  float(min(ye[0], ye[-1])), float(max(ye[0], ye[-1])))
+        return self.add_raster(np.ascontiguousarray(rgba), extent=extent,
+                               clip_path=clip_path, smooth=smooth, name=name)

anyplotlib/sphinx_anywidget/__init__.py

@@ -0,0 +1,177 @@
+"""
+sphinx_anywidget
+================
+
+A generic Sphinx extension that makes any ``anywidget.AnyWidget``-based
+figure interactive in documentation pages — powered by Pyodide, with no
+server or Jupyter kernel required.
+
+Quick start (any project)
+-------------------------
+
+In your ``conf.py``::
+
+    extensions = [
+        "anyplotlib.sphinx_anywidget",
+    ]
+
+    # Package whose wheel is built and served to Pyodide at runtime.
+    anywidget_pyodide_package = "mypackage"
+
+
+The extension:
+* builds a pure-Python wheel at docs-build time;
+* injects ``anywidget_bridge.js`` (per-figure ⚡ badges + Pyodide boot);
+* provides ``AnywidgetScraper`` for Sphinx Gallery (``# Interactive`` tag);
+* registers ``.. anywidget-figure::`` RST directive.
+
+Monkey-patch approach
+---------------------
+``anywidget_bridge.js`` patches ``AnyWidget.__init__`` in Pyodide to add a
+``traitlets.observe(names=All)`` observer.  When any ``sync=True`` trait
+changes and the widget has ``_anywidget_fig_id`` set, the observer calls
+``window._anywidgetPush(fig_id, name, value_str)`` which postMessages the
+new state into the matching iframe — no library-side Pyodide code needed.
+"""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+from anyplotlib.sphinx_anywidget._scraper import AnywidgetScraper, ViewerScraper  # noqa: F401
+
+_HERE = Path(__file__).parent
+_STATIC_SRC = _HERE / "static"
+
+
+def setup(app):
+    """Register sphinx_anywidget with Sphinx."""
+    app.add_config_value("anywidget_pyodide_package", default=None, rebuild="html")
+
+    from anyplotlib.sphinx_anywidget._directive import AnywidgetFigureDirective
+    app.add_directive("anywidget-figure", AnywidgetFigureDirective)
+
+    app.connect("builder-inited", _copy_static_assets)
+    app.connect("builder-inited", _build_pyodide_wheel)
+
+    # anywidget_config.js is written dynamically by _build_pyodide_wheel;
+    # it must load BEFORE anywidget_bridge.js so _inferPackageName sees the name.
+    app.add_js_file("anywidget_config.js", loading_method="defer", priority=490)
+    app.add_js_file("anywidget_bridge.js", loading_method="defer", priority=500)
+    app.add_css_file("anywidget_overlay.css")
+
+    return {
+        "version": "0.1.0",
+        "parallel_read_safe": True,
+        "parallel_write_safe": True,
+    }
+
+
+def _copy_static_assets(app):
+    """Add the extension's static/ dir to html_static_path."""
+    src_str = str(_STATIC_SRC)
+    if hasattr(app.config, "html_static_path"):
+        if src_str not in app.config.html_static_path:
+            app.config.html_static_path.append(src_str)
+
+
+def _build_pyodide_wheel(app):
+    """Build the configured package wheel for the Pyodide bridge."""
+    pkg = getattr(app.config, "anywidget_pyodide_package", None)
+    if not pkg:
+        pkg = _infer_package_name(app)
+
+    conf_dir   = Path(app.confdir)
+    static_dir = conf_dir / "_static"
+    static_dir.mkdir(parents=True, exist_ok=True)
+
+    import json as _json
+    import re as _re
+
+    if not pkg:
+        print(
+            "[sphinx_anywidget] WARNING: anywidget_pyodide_package not set; "
+            "Pyodide interactive mode disabled."
+        )
+        # Write a config that explicitly disables interactive mode so the
+        # bridge's heuristic package detection never runs.
+        config_js = (
+            "window._anywidgetPackage = null;\n"
+            "window._anywidgetInteractiveDisabled = true;\n"
+        )
+        (static_dir / "anywidget_config.js").write_text(config_js, encoding="utf-8")
+        return
+
+    # Write a tiny config script so anywidget_bridge.js can find the package
+    # name without fragile heuristics.  Loaded before anywidget_bridge.js.
+    config_js = f"window._anywidgetPackage = {_json.dumps(pkg)};\n"
+    (static_dir / "anywidget_config.js").write_text(config_js, encoding="utf-8")
+
+    # If a stable wheel was already placed here (e.g. by the CI ``uv build``
+    # step that runs before sphinx-build), reuse it — but only when it is
+    # newer than every source file, otherwise the ⚡ interactive mode would
+    # run stale code.
+    normalised = _re.sub(r"[-.]", "_", pkg)
+    wheels_dir = static_dir / "wheels"
+    stable = wheels_dir / f"{normalised}-0.0.0-py3-none-any.whl"
+    if stable.exists():
+        project_root = _find_project_root(conf_dir)
+        pkg_dir = project_root / pkg
+        newest_src = 0.0
+        if pkg_dir.is_dir():
+            newest_src = max(
+                (f.stat().st_mtime for f in pkg_dir.rglob("*")
+                 if f.suffix in (".py", ".js", ".css") and "tests" not in f.parts),
+                default=0.0,
+            )
+        if stable.stat().st_mtime >= newest_src:
+            # ASCII only: Windows consoles (cp1252) can't print '→'
+            print(f"[sphinx_anywidget] wheel up to date -> {stable}")
+            return
+        print("[sphinx_anywidget] wheel older than source -> rebuilding")
+        stable.unlink()
+
+    from anyplotlib.sphinx_anywidget._wheel_builder import build_wheel
+    project_root = _find_project_root(conf_dir)
+    build_wheel(static_dir, pkg, project_root)
+
+
+def _find_project_root(start: Path) -> Path:
+    """Walk up from *start* to find the directory containing pyproject.toml or setup.py.
+
+    Falls back to ``start.parent`` for backwards compatibility so behaviour is
+    unchanged for the common ``docs/conf.py`` layout where the project root is
+    directly above the docs directory.
+    """
+    markers = ("pyproject.toml", "setup.py", "setup.cfg")
+    current = start.resolve()
+    # Check start itself and each parent up to the filesystem root.
+    for directory in [current, *current.parents]:
+        if any((directory / m).exists() for m in markers):
+            return directory
+    # No marker found — fall back to conf_dir's parent (original behaviour).
+    return start.parent
+
+
+def _infer_package_name(app) -> str | None:
+    """Infer package name from pyproject.toml near conf.py."""
+    try:
+        import tomllib
+    except ImportError:
+        try:
+            import tomli as tomllib  # type: ignore[no-redef]
+        except ImportError:
+            return None
+    conf_dir = Path(app.confdir)
+    for candidate in [conf_dir / "pyproject.toml", conf_dir.parent / "pyproject.toml"]:
+        if candidate.exists():
+            with open(candidate, "rb") as fh:
+                data = tomllib.load(fh)
+            name = (
+                data.get("project", {}).get("name")
+                or data.get("tool", {}).get("poetry", {}).get("name")
+            )
+            if name:
+                return name
+    return None
+