behaviz 0.4.0__py3-none-any.whl

behaviz/__init__.py

@@ -0,0 +1,110 @@
+from importlib.metadata import version as _version, PackageNotFoundError
+
+try:
+    __version__ = _version("behaviz")
+except PackageNotFoundError:  # not installed (e.g. running from source tree)
+    __version__ = "0.0.0+unknown"
+
+# Plot functions
+from behaviz.core.core import (
+    plot_line,
+    plot_scatter,
+    plot_errorbar,
+    plot_violin,
+    plot_step,
+    plot_bar,
+    plot_vertical,
+    plot_horizontal,
+    plot_image,
+    plot_fill_between,
+    plot_pie,
+    plot_hexbin,
+)
+
+from behaviz.core.auxiliary import plot_pval
+
+# Errors (malformed data arguments, and unsupported save formats)
+from behaviz.core.errors import BehavizDataError, BehavizSaveError
+
+# Unified figure output (save + canvas context manager)
+from behaviz.io import save, canvas
+
+# Spec classes (users need these to configure plots)
+from behaviz.spec.plot_spec import PlotSpec
+from behaviz.spec.axis_spec import AxisSpec, ScaleType
+from behaviz.spec.figure_spec import FigureSpec, LegendPosition
+from behaviz.spec.colorbar_spec import ColorbarSpec
+
+# Preset save/load system
+from behaviz.presets import (
+    load_preset,
+    save_preset,
+    list_presets,
+    delete_preset,
+    export_preset,
+    import_preset,
+    init_home,
+    presets_dir,
+    examples_dir,
+)
+
+# composite plot styling helper
+from behaviz.composite_plots.styling import split_styles
+
+# Visual manipulations
+import behaviz.manipulations
+
+# Backend switcher
+import behaviz.backends
+from behaviz.backends.renderer_manager import set_renderer, get_renderer
+
+
+__all__ = [
+    # plot functions
+    "plot_line",
+    "plot_scatter",
+    "plot_errorbar",
+    "plot_violin",
+    "plot_step",
+    "plot_bar",
+    "plot_rain",
+    "plot_vertical",
+    "plot_horizontal",
+    "plot_image",
+    "plot_fill_between",
+    "plot_pie",
+    "plot_hexbin",
+    "plot_distribution",
+    "plot_psychometric",
+    "plot_impact",
+    "plot_pval",
+    # errors
+    "BehavizDataError",
+    "BehavizSaveError",
+    # figure output
+    "save",
+    "canvas",
+    # specs
+    "PlotSpec",
+    "AxisSpec",
+    "ScaleType",
+    "FigureSpec",
+    "LegendPosition",
+    "ColorbarSpec",
+    # presets
+    "load_preset",
+    "save_preset",
+    "list_presets",
+    "delete_preset",
+    "export_preset",
+    "import_preset",
+    "init_home",
+    "presets_dir",
+    "examples_dir",
+    # backend
+    "set_renderer",
+    "get_renderer",
+    "__version__",
+]
+
+set_renderer("matplotlib")

behaviz/backends/__init__.py

@@ -0,0 +1,11 @@
+import behaviz.backends.matplotlib
+import behaviz.backends.seaborn
+import behaviz.backends.bokeh  # noqa: F401
+
+# Validate that every registered plot type is fully implemented across all
+# backends and overriders. This runs once at import and raises immediately if
+# anything is missing, so errors surface during development rather than at
+# call time.
+from behaviz.core.registry_validation import validate_registry
+
+validate_registry()

behaviz/backends/_legend.py

@@ -0,0 +1,22 @@
+"""Shared legend helper for the matplotlib-based backends."""
+
+from __future__ import annotations
+
+
+def dedup_legend(handles_labels):
+    """Drop repeated labels (keeping the first) from ``get_legend_handles_labels()``.
+
+    Grouped/hued plots can attach the same label to several artists — e.g. every
+    rectangle of a bar container, or one line per subject sharing a hue label —
+    which would otherwise produce duplicate legend entries.
+    """
+    handles, labels = handles_labels
+    seen: set = set()
+    h_out, l_out = [], []
+    for h, l in zip(handles, labels):
+        if l in seen:
+            continue
+        seen.add(l)
+        h_out.append(h)
+        l_out.append(l)
+    return h_out, l_out

behaviz/backends/_save.py

@@ -0,0 +1,40 @@
+"""Shared figure-saving helpers for the matplotlib-based backends.
+
+``SeabornRenderer`` is not a subclass of ``MatplotlibRenderer``, so the common
+``fig.savefig`` logic lives here as free functions both can call.
+"""
+
+from __future__ import annotations
+
+import os
+
+from behaviz.core.errors import BehavizSaveError
+
+# Raster/vector formats matplotlib's Agg/PDF/SVG/PS writers handle.
+MPL_FORMATS = frozenset({".png", ".pdf", ".svg", ".jpg", ".jpeg", ".tif", ".tiff", ".eps", ".ps"})
+
+
+def extension(path) -> str:
+    """Lower-cased file extension of ``path`` (including the dot)."""
+    return os.path.splitext(os.fspath(path))[1].lower()
+
+
+def save_matplotlib(fig, path, **kwargs) -> str:
+    """Write a matplotlib ``Figure`` to ``path``, dispatching on its extension."""
+    ext = extension(path)
+    if ext in MPL_FORMATS:
+        kwargs.setdefault("bbox_inches", "tight")
+        fig.savefig(path, **kwargs)
+        return os.fspath(path)
+
+    if ext in (".html", ".htm"):
+        raise BehavizSaveError(
+            "The matplotlib/seaborn backend cannot write HTML. Use a raster or "
+            "vector format (.png, .pdf, .svg), or switch to the bokeh backend "
+            "for interactive HTML output."
+        )
+
+    raise BehavizSaveError(
+        f"Unsupported file format '{ext or path}' for the matplotlib/seaborn backend. "
+        f"Supported: {', '.join(sorted(MPL_FORMATS))}."
+    )

behaviz/backends/bokeh/__init__.py

@@ -0,0 +1,12 @@
+from behaviz.backends.bokeh.backend import (
+    BokehRenderer,
+)
+
+from behaviz.backends.renderer_registry import (
+    register_renderer,
+)
+
+register_renderer(
+    "bokeh",
+    BokehRenderer,
+)