scitex 2.7.0__py3-none-any.whl → 2.8.1__py3-none-any.whl

scitex/plt/_subplots/_AxisWrapperMixins/_RawMatplotlibMixin.py

@@ -0,0 +1,321 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# Timestamp: "2025-12-13 (ywatanabe)"
+# File: /home/ywatanabe/proj/scitex-code/src/scitex/plt/_subplots/_AxisWrapperMixins/_RawMatplotlibMixin.py
+
+"""
+Matplotlib aliases (mpl_xxx) for explicit matplotlib-style API.
+
+Provides consistent naming convention:
+- stx_xxx: scitex-specific methods (ArrayLike input, tracked)
+- sns_xxx: seaborn wrappers (DataFrame input, tracked)
+- mpl_xxx: matplotlib methods (matplotlib-style input, tracked)
+
+All three API layers track data for reproducibility.
+
+Usage:
+    ax.stx_line(y)              # ArrayLike input
+    ax.sns_boxplot(data=df, x="group", y="value")  # DataFrame input
+    ax.mpl_plot(x, y)           # matplotlib-style input
+    ax.plot(x, y)               # Same as mpl_plot
+"""
+
+import os
+
+__FILE__ = __file__
+__DIR__ = os.path.dirname(__FILE__)
+
+
+class RawMatplotlibMixin:
+    """Mixin providing mpl_xxx aliases for matplotlib-style API.
+
+    These methods are identical to calling ax.plot(), ax.scatter(), etc.
+    They go through SciTeX's __getattr__ wrapper and are fully tracked.
+
+    The mpl_* prefix provides:
+    - Explicit naming convention (mpl_* vs stx_* vs sns_*)
+    - Programmatic access via MPL_METHODS registry
+    - Same tracking and styling as regular matplotlib calls
+    """
+
+    # =========================================================================
+    # Helper to call through __getattr__ wrapper (enables tracking)
+    # =========================================================================
+    def _mpl_call(self, method_name, *args, **kwargs):
+        """Call matplotlib method through __getattr__ wrapper for tracking."""
+        # Use object.__getattribute__ to get the __getattr__ from AxisWrapper
+        # Then call it with the method name to get the tracked wrapper
+        wrapper_class = type(self)
+        # Walk up MRO to find __getattr__ in AxisWrapper
+        for cls in wrapper_class.__mro__:
+            if "__getattr__" in cls.__dict__:
+                return cls.__getattr__(self, method_name)(*args, **kwargs)
+        # Fallback to direct call if no __getattr__ found
+        return getattr(self._axes_mpl, method_name)(*args, **kwargs)
+
+    # =========================================================================
+    # Line plots
+    # =========================================================================
+    def mpl_plot(self, *args, **kwargs):
+        """Matplotlib plot() - tracked, identical to ax.plot()."""
+        return self._mpl_call("plot", *args, **kwargs)
+
+    def mpl_step(self, *args, **kwargs):
+        """Matplotlib step() - tracked, identical to ax.step()."""
+        return self._mpl_call("step", *args, **kwargs)
+
+    def mpl_stem(self, *args, **kwargs):
+        """Matplotlib stem() - tracked, identical to ax.stem()."""
+        return self._mpl_call("stem", *args, **kwargs)
+
+    # =========================================================================
+    # Scatter plots
+    # =========================================================================
+    def mpl_scatter(self, *args, **kwargs):
+        """Matplotlib scatter() - tracked, identical to ax.scatter()."""
+        return self._mpl_call("scatter", *args, **kwargs)
+
+    # =========================================================================
+    # Bar plots
+    # =========================================================================
+    def mpl_bar(self, *args, **kwargs):
+        """Matplotlib bar() - tracked, identical to ax.bar()."""
+        return self._mpl_call("bar", *args, **kwargs)
+
+    def mpl_barh(self, *args, **kwargs):
+        """Matplotlib barh() - tracked, identical to ax.barh()."""
+        return self._mpl_call("barh", *args, **kwargs)
+
+    def mpl_bar3d(self, *args, **kwargs):
+        """Matplotlib bar3d() (3D axes) - tracked."""
+        return self._mpl_call("bar3d", *args, **kwargs)
+
+    # =========================================================================
+    # Histograms
+    # =========================================================================
+    def mpl_hist(self, *args, **kwargs):
+        """Matplotlib hist() - tracked, identical to ax.hist()."""
+        return self._mpl_call("hist", *args, **kwargs)
+
+    def mpl_hist2d(self, *args, **kwargs):
+        """Matplotlib hist2d() - tracked, identical to ax.hist2d()."""
+        return self._mpl_call("hist2d", *args, **kwargs)
+
+    def mpl_hexbin(self, *args, **kwargs):
+        """Matplotlib hexbin() - tracked, identical to ax.hexbin()."""
+        return self._mpl_call("hexbin", *args, **kwargs)
+
+    # =========================================================================
+    # Statistical plots
+    # =========================================================================
+    def mpl_boxplot(self, *args, **kwargs):
+        """Matplotlib boxplot() - tracked, identical to ax.boxplot()."""
+        return self._mpl_call("boxplot", *args, **kwargs)
+
+    def mpl_violinplot(self, *args, **kwargs):
+        """Matplotlib violinplot() - tracked, identical to ax.violinplot()."""
+        return self._mpl_call("violinplot", *args, **kwargs)
+
+    def mpl_errorbar(self, *args, **kwargs):
+        """Matplotlib errorbar() - tracked, identical to ax.errorbar()."""
+        return self._mpl_call("errorbar", *args, **kwargs)
+
+    def mpl_eventplot(self, *args, **kwargs):
+        """Matplotlib eventplot() - tracked, identical to ax.eventplot()."""
+        return self._mpl_call("eventplot", *args, **kwargs)
+
+    # =========================================================================
+    # Fill and area plots
+    # =========================================================================
+    def mpl_fill(self, *args, **kwargs):
+        """Matplotlib fill() - tracked, identical to ax.fill()."""
+        return self._mpl_call("fill", *args, **kwargs)
+
+    def mpl_fill_between(self, *args, **kwargs):
+        """Matplotlib fill_between() - tracked, identical to ax.fill_between()."""
+        return self._mpl_call("fill_between", *args, **kwargs)
+
+    def mpl_fill_betweenx(self, *args, **kwargs):
+        """Matplotlib fill_betweenx() - tracked, identical to ax.fill_betweenx()."""
+        return self._mpl_call("fill_betweenx", *args, **kwargs)
+
+    def mpl_stackplot(self, *args, **kwargs):
+        """Matplotlib stackplot() - tracked, identical to ax.stackplot()."""
+        return self._mpl_call("stackplot", *args, **kwargs)
+
+    # =========================================================================
+    # Contour and heatmap plots
+    # =========================================================================
+    def mpl_contour(self, *args, **kwargs):
+        """Matplotlib contour() - tracked, identical to ax.contour()."""
+        return self._mpl_call("contour", *args, **kwargs)
+
+    def mpl_contourf(self, *args, **kwargs):
+        """Matplotlib contourf() - tracked, identical to ax.contourf()."""
+        return self._mpl_call("contourf", *args, **kwargs)
+
+    def mpl_imshow(self, *args, **kwargs):
+        """Matplotlib imshow() - tracked, identical to ax.imshow()."""
+        return self._mpl_call("imshow", *args, **kwargs)
+
+    def mpl_pcolormesh(self, *args, **kwargs):
+        """Matplotlib pcolormesh() - tracked, identical to ax.pcolormesh()."""
+        return self._mpl_call("pcolormesh", *args, **kwargs)
+
+    def mpl_pcolor(self, *args, **kwargs):
+        """Matplotlib pcolor() - tracked, identical to ax.pcolor()."""
+        return self._mpl_call("pcolor", *args, **kwargs)
+
+    def mpl_matshow(self, *args, **kwargs):
+        """Matplotlib matshow() - tracked, identical to ax.matshow()."""
+        return self._mpl_call("matshow", *args, **kwargs)
+
+    # =========================================================================
+    # Vector field plots
+    # =========================================================================
+    def mpl_quiver(self, *args, **kwargs):
+        """Matplotlib quiver() - tracked, identical to ax.quiver()."""
+        return self._mpl_call("quiver", *args, **kwargs)
+
+    def mpl_streamplot(self, *args, **kwargs):
+        """Matplotlib streamplot() - tracked, identical to ax.streamplot()."""
+        return self._mpl_call("streamplot", *args, **kwargs)
+
+    def mpl_barbs(self, *args, **kwargs):
+        """Matplotlib barbs() - tracked, identical to ax.barbs()."""
+        return self._mpl_call("barbs", *args, **kwargs)
+
+    # =========================================================================
+    # Pie and polar plots
+    # =========================================================================
+    def mpl_pie(self, *args, **kwargs):
+        """Matplotlib pie() - tracked, identical to ax.pie()."""
+        return self._mpl_call("pie", *args, **kwargs)
+
+    # =========================================================================
+    # Text and annotations
+    # =========================================================================
+    def mpl_text(self, *args, **kwargs):
+        """Matplotlib text() - tracked, identical to ax.text()."""
+        return self._mpl_call("text", *args, **kwargs)
+
+    def mpl_annotate(self, *args, **kwargs):
+        """Matplotlib annotate() - tracked, identical to ax.annotate()."""
+        return self._mpl_call("annotate", *args, **kwargs)
+
+    # =========================================================================
+    # Lines and spans
+    # =========================================================================
+    def mpl_axhline(self, *args, **kwargs):
+        """Matplotlib axhline() - tracked, identical to ax.axhline()."""
+        return self._mpl_call("axhline", *args, **kwargs)
+
+    def mpl_axvline(self, *args, **kwargs):
+        """Matplotlib axvline() - tracked, identical to ax.axvline()."""
+        return self._mpl_call("axvline", *args, **kwargs)
+
+    def mpl_axhspan(self, *args, **kwargs):
+        """Matplotlib axhspan() - tracked, identical to ax.axhspan()."""
+        return self._mpl_call("axhspan", *args, **kwargs)
+
+    def mpl_axvspan(self, *args, **kwargs):
+        """Matplotlib axvspan() - tracked, identical to ax.axvspan()."""
+        return self._mpl_call("axvspan", *args, **kwargs)
+
+    # =========================================================================
+    # Patches and shapes
+    # =========================================================================
+    def mpl_add_patch(self, patch, **kwargs):
+        """Matplotlib add_patch() - tracked, identical to ax.add_patch()."""
+        return self._mpl_call("add_patch", patch, **kwargs)
+
+    def mpl_add_artist(self, artist, **kwargs):
+        """Matplotlib add_artist() - tracked, identical to ax.add_artist()."""
+        return self._mpl_call("add_artist", artist, **kwargs)
+
+    def mpl_add_collection(self, collection, **kwargs):
+        """Matplotlib add_collection() - tracked, identical to ax.add_collection()."""
+        return self._mpl_call("add_collection", collection, **kwargs)
+
+    # =========================================================================
+    # 3D plotting (if available)
+    # =========================================================================
+    def mpl_plot_surface(self, *args, **kwargs):
+        """Matplotlib plot_surface() (3D axes) - tracked."""
+        return self._mpl_call("plot_surface", *args, **kwargs)
+
+    def mpl_plot_wireframe(self, *args, **kwargs):
+        """Matplotlib plot_wireframe() (3D axes) - tracked."""
+        return self._mpl_call("plot_wireframe", *args, **kwargs)
+
+    def mpl_contour3D(self, *args, **kwargs):
+        """Matplotlib contour3D() (3D axes) - tracked."""
+        return self._mpl_call("contour3D", *args, **kwargs)
+
+    def mpl_scatter3D(self, *args, **kwargs):
+        """Matplotlib scatter3D() (3D axes) - tracked."""
+        return self._mpl_call("scatter3D", *args, **kwargs)
+
+    # =========================================================================
+    # Utility method to get raw axes
+    # =========================================================================
+    @property
+    def mpl_axes(self):
+        """Direct access to underlying matplotlib axes object."""
+        return self._axes_mpl
+
+    def mpl_raw(self, method_name, *args, **kwargs):
+        """Call any matplotlib method by name without scitex processing.
+
+        Parameters
+        ----------
+        method_name : str
+            Name of matplotlib axes method to call
+        *args, **kwargs
+            Arguments to pass to the method
+
+        Returns
+        -------
+        result
+            Result from matplotlib method
+
+        Example
+        -------
+        >>> ax.mpl_raw("tricontour", x, y, z, levels=10)
+        """
+        method = getattr(self._axes_mpl, method_name)
+        return method(*args, **kwargs)
+
+
+# Registry of mpl_xxx methods for programmatic access
+MPL_METHODS = [
+    # Line plots
+    "mpl_plot", "mpl_step", "mpl_stem",
+    # Scatter
+    "mpl_scatter",
+    # Bar
+    "mpl_bar", "mpl_barh", "mpl_bar3d",
+    # Histograms
+    "mpl_hist", "mpl_hist2d", "mpl_hexbin",
+    # Statistical
+    "mpl_boxplot", "mpl_violinplot", "mpl_errorbar", "mpl_eventplot",
+    # Fill/area
+    "mpl_fill", "mpl_fill_between", "mpl_fill_betweenx", "mpl_stackplot",
+    # Contour/heatmap
+    "mpl_contour", "mpl_contourf", "mpl_imshow", "mpl_pcolormesh", "mpl_pcolor", "mpl_matshow",
+    # Vector fields
+    "mpl_quiver", "mpl_streamplot", "mpl_barbs",
+    # Pie
+    "mpl_pie",
+    # Text/annotations
+    "mpl_text", "mpl_annotate",
+    # Lines/spans
+    "mpl_axhline", "mpl_axvline", "mpl_axhspan", "mpl_axvspan",
+    # Patches
+    "mpl_add_patch", "mpl_add_artist", "mpl_add_collection",
+    # 3D
+    "mpl_plot_surface", "mpl_plot_wireframe", "mpl_contour3D", "mpl_scatter3D",
+]
+
+
+# EOF

scitex/plt/_subplots/_AxisWrapperMixins/_SeabornMixin/__init__.py

@@ -0,0 +1,33 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# Timestamp: "2025-12-13 (ywatanabe)"
+# File: __init__.py - SeabornMixin package
+
+"""
+SeabornMixin - Modular seaborn integration mixin for AxisWrapper.
+
+This package provides seaborn plotting functionality:
+- _base: Helper methods and data preparation
+- _wrappers: Individual seaborn plot wrappers (sns_xxx)
+"""
+
+import os
+
+__FILE__ = __file__
+__DIR__ = os.path.dirname(__FILE__)
+
+from ._base import SeabornBaseMixin, sns_copy_doc
+from ._wrappers import SeabornWrappersMixin
+
+
+class SeabornMixin(SeabornBaseMixin, SeabornWrappersMixin):
+    """Mixin class for seaborn plotting integration.
+
+    Combines:
+    - SeabornBaseMixin: Helper methods for tracking and data preparation
+    - SeabornWrappersMixin: Individual sns_ prefixed seaborn wrappers
+    """
+    pass
+
+
+# EOF

scitex/plt/_subplots/_AxisWrapperMixins/_SeabornMixin/_base.py

@@ -0,0 +1,152 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# Timestamp: "2025-12-13 (ywatanabe)"
+# File: _base.py - Base seaborn functionality
+
+"""Base seaborn mixin with helper methods for tracking and data preparation."""
+
+import os
+from functools import wraps
+
+import scitex
+import numpy as np
+import pandas as pd
+import seaborn as sns
+
+__FILE__ = __file__
+__DIR__ = os.path.dirname(__FILE__)
+
+
+def sns_copy_doc(func):
+    """Decorator to copy docstring from seaborn function."""
+    @wraps(func)
+    def wrapper(self, *args, **kwargs):
+        return func(self, *args, **kwargs)
+
+    sns_method_name = func.__name__.split("sns_")[-1]
+    wrapper.__doc__ = getattr(sns, sns_method_name).__doc__
+    return wrapper
+
+
+class SeabornBaseMixin:
+    """Base mixin for seaborn integration with tracking support."""
+
+    def _sns_base(
+        self, method_name, *args, track=True, track_obj=None, id=None, **kwargs
+    ):
+        """Execute seaborn plot method with tracking support."""
+        sns_method_name = method_name.split("sns_")[-1]
+
+        with self._no_tracking():
+            sns_plot_fn = getattr(sns, sns_method_name)
+
+            if kwargs.get("hue_colors"):
+                kwargs = scitex.gen.alternate_kwarg(
+                    kwargs, primary_key="palette", alternate_key="hue_colors"
+                )
+
+            import warnings
+            from scitex import logging
+
+            mpl_logger = logging.getLogger("matplotlib")
+            original_level = mpl_logger.level
+            mpl_logger.setLevel(logging.WARNING)
+
+            try:
+                with warnings.catch_warnings():
+                    warnings.filterwarnings(
+                        "ignore",
+                        message=".*categorical units.*parsable as floats or dates.*",
+                        category=UserWarning,
+                    )
+                    warnings.filterwarnings(
+                        "ignore",
+                        message=".*Using categorical units.*",
+                        module="matplotlib.*",
+                    )
+                    warnings.simplefilter("ignore", UserWarning)
+
+                    self._axis_mpl = sns_plot_fn(ax=self._axis_mpl, *args, **kwargs)
+            finally:
+                mpl_logger.setLevel(original_level)
+
+            # Post-processing for histplot with kde=True
+            if sns_method_name == "histplot" and kwargs.get("kde", False):
+                from scitex.plt.utils import mm_to_pt
+                kde_lw = mm_to_pt(0.2)
+                for line in self._axis_mpl.get_lines():
+                    line.set_linewidth(kde_lw)
+                    line.set_color("black")
+                    line.set_linestyle("--")
+
+            # Post-processing for histplot alpha
+            if sns_method_name == "histplot" and "alpha" not in kwargs:
+                for patch in self._axis_mpl.patches:
+                    patch.set_alpha(1.0)
+
+        track_obj = track_obj if track_obj is not None else args
+        tracked_dict = {
+            "data": track_obj,
+            "args": args,
+        }
+        self._track(track, id, method_name, tracked_dict, kwargs)
+
+    def _sns_base_xyhue(self, method_name, *args, track=True, id=None, **kwargs):
+        """Execute seaborn plot with x/y/hue data preparation."""
+        df = kwargs.get("data")
+        x, y, hue = kwargs.get("x"), kwargs.get("y"), kwargs.get("hue")
+
+        track_obj = self._sns_prepare_xyhue(df, x, y, hue) if df is not None else None
+        self._sns_base(
+            method_name,
+            *args,
+            track=track,
+            track_obj=track_obj,
+            id=id,
+            **kwargs,
+        )
+
+    def _sns_prepare_xyhue(self, data=None, x=None, y=None, hue=None, **kwargs):
+        """Prepare data for tracking based on x/y/hue configuration."""
+        data = data.reset_index()
+
+        if hue is not None:
+            if x is None and y is None:
+                return data
+            elif x is None:
+                agg_dict = {}
+                for hh in data[hue].unique():
+                    agg_dict[hh] = data.loc[data[hue] == hh, y]
+                df = scitex.pd.force_df(agg_dict)
+                return df
+            elif y is None:
+                df = pd.concat(
+                    [data.loc[data[hue] == hh, x] for hh in data[hue].unique()],
+                    axis=1,
+                )
+                return df
+            else:
+                pivoted_data = data.pivot_table(
+                    values=y,
+                    index=data.index,
+                    columns=[x, hue],
+                    aggfunc="first",
+                )
+                pivoted_data.columns = [
+                    f"{col[0]}-{col[1]}" for col in pivoted_data.columns
+                ]
+                return pivoted_data
+        else:
+            if x is None and y is None:
+                return data
+            elif x is None:
+                return data[[y]]
+            elif y is None:
+                return data[[x]]
+            else:
+                return data.pivot_table(
+                    values=y, index=data.index, columns=x, aggfunc="first"
+                )
+
+
+# EOF