maidr 1.4.1__tar.gz → 1.4.2__tar.gz

{maidr-1.4.1 → maidr-1.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: maidr
-Version: 1.4.1
+Version: 1.4.2
 Summary: Multimodal Access and Interactive Data Representations
 License: GPL-3.0-or-later
 Keywords: accessibility,visualization,sonification,braille,tactile,multimodal,data representation,blind,low vision,visual impairments

{maidr-1.4.1 → maidr-1.4.2}/maidr/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "1.4.1"
+__version__ = "1.4.2"
 
 from .api import close, render, save_html, show, stacked
 from .core import Maidr

{maidr-1.4.1 → maidr-1.4.2}/maidr/core/enum/smooth_keywords.py

@@ -2,8 +2,11 @@
 SMOOTH_KEYWORDS = [
     "smooth",
     "lowess",
+    "loess",
     "regression",
     "linear regression",
+    "linear fit",
+    "fit",
     "kde",
     "density",
     "gaussian",

{maidr-1.4.1 → maidr-1.4.2}/maidr/patch/regplot.py

@@ -5,11 +5,10 @@ from matplotlib.axes import Axes
 from matplotlib.lines import Line2D
 from maidr.core.enum import PlotType
 from maidr.patch.common import common
-import numpy as np
 from maidr.core.context_manager import ContextManager
 import uuid
 from maidr.core.enum.smooth_keywords import SMOOTH_KEYWORDS
-from maidr.util.regression_line_utils import find_regression_line
+from maidr.util.regression_line_utils import find_smooth_lines_by_label
 
 
 def regplot(wrapped, instance, args, kwargs) -> Axes:
@@ -23,27 +22,34 @@ def regplot(wrapped, instance, args, kwargs) -> Axes:
         # Prevent any MAIDR layer registration during plotting when scatter=False
         with ContextManager.set_internal_context():
             ax = wrapped(*args, **kwargs)
+
     axes = ax if isinstance(ax, Axes) else ax.axes if hasattr(ax, "axes") else None
     if axes is not None:
-        regression_line = find_regression_line(axes)
-        if regression_line is not None:
-            # ---
-            # Assign a unique gid to the regression line if not already set.
-            # This is necessary because the SVG output may contain many <g> and <path> tags,
-            # and only the regression line should be uniquely selectable for accessibility and highlighting.
-            # By setting a unique gid, we ensure the backend and frontend can generate a reliable selector
-            # (e.g., g[id='maidr-...'] path) that matches only the intended regression line.
-            # ---
-            if regression_line.get_gid() is None:
+        # Find and register all smooth lines
+        smooth_lines = find_smooth_lines_by_label(axes)
+        for line in smooth_lines:
+            # If line doesn't have a gid yet, assign one and register
+            if line.get_gid() is None:
                 new_gid = f"maidr-{uuid.uuid4()}"
-                regression_line.set_gid(new_gid)
-            common(
-                PlotType.SMOOTH,
-                lambda *a, **k: ax,
-                instance,
-                args,
-                dict(kwargs, regression_line=regression_line),
-            )
+                line.set_gid(new_gid)
+                common(
+                    PlotType.SMOOTH,
+                    lambda *a, **k: ax,
+                    instance,
+                    args,
+                    dict(kwargs, regression_line=line),
+                )
+            else:
+                # Even if it has a gid, register it as a smooth layer
+                # This handles the case where patched_plot already assigned a gid
+                common(
+                    PlotType.SMOOTH,
+                    lambda *a, **k: ax,
+                    instance,
+                    args,
+                    dict(kwargs, regression_line=line),
+                )
+
     return ax
 
 
@@ -53,7 +59,8 @@ def patched_plot(wrapped, instance, args, kwargs):
     """
     # Call the original plot function
     lines = wrapped(*args, **kwargs)
-    # lines can be a list of Line2D objects
+
+    # Check each line for smooth keywords and register if found
     for line in lines:
         if isinstance(line, Line2D):
             label = line.get_label() or ""
@@ -72,6 +79,7 @@ def patched_plot(wrapped, instance, args, kwargs):
                     args,
                     dict(kwargs, regression_line=line),
                 )
+
     return lines
 
 

maidr-1.4.2/maidr/util/regression_line_utils.py

@@ -0,0 +1,69 @@
+from matplotlib.lines import Line2D
+from matplotlib.axes import Axes
+import numpy as np
+from typing import List, Union
+from maidr.core.enum.smooth_keywords import SMOOTH_KEYWORDS
+
+
+def find_regression_line(axes: Axes) -> Union[Line2D, None]:
+    """
+    Helper to find the regression line (Line2D) in the given axes.
+    """
+    return next(
+        (
+            artist
+            for artist in axes.get_children()
+            if isinstance(artist, Line2D)
+            and artist.get_label() not in (None, "", "_nolegend_")
+            and artist.get_xydata() is not None
+            and np.asarray(artist.get_xydata()).size > 0
+        ),
+        None,
+    )
+
+
+def find_smooth_lines_by_label(axes: Axes) -> List[Line2D]:
+    """
+    Helper to find all smooth/regression lines (Line2D) in the given axes by checking their labels.
+
+    This function detects smooth lines by examining their labels for smooth-related keywords
+    or generic '_child' labels that are commonly created by seaborn's regplot function.
+
+    Parameters
+    ----------
+    axes : matplotlib.axes.Axes
+        The matplotlib axes object to search for smooth lines.
+
+    Returns
+    -------
+    List[Line2D]
+        List of Line2D objects that have labels matching smooth keywords or generic '_child' labels.
+
+    Examples
+    --------
+    >>> import matplotlib.pyplot as plt
+    >>> import seaborn as sns
+    >>>
+    >>> fig, ax = plt.subplots()
+    >>> # Create a regplot with smooth line
+    >>> sns.regplot(x=[1, 2, 3], y=[1, 2, 3], ax=ax, lowess=True)
+    >>>
+    >>> # Find smooth lines
+    >>> smooth_lines = find_smooth_lines_by_label(ax)
+    >>> print(f"Found {len(smooth_lines)} smooth lines")
+    """
+    smooth_lines = []
+    for line in axes.get_lines():
+        if isinstance(line, Line2D):
+            label = line.get_label() or ""
+            label_str = str(label)
+
+            # Check if label matches smooth keywords
+            if any(key in label_str.lower() for key in SMOOTH_KEYWORDS):
+                smooth_lines.append(line)
+            # Also check for seaborn regplot lines with generic labels (like '_child0', '_child1')
+            elif label_str.startswith("_child"):
+                # Lines with _child labels are likely smooth lines from seaborn regplot
+                smooth_lines.append(line)
+
+    return smooth_lines

{maidr-1.4.1 → maidr-1.4.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "maidr"
-version = "1.4.1"
+version = "1.4.2"
 description = "Multimodal Access and Interactive Data Representations"
 authors = [
     "JooYoung Seo <jseo1005@illinois.edu>",

maidr-1.4.1/maidr/util/regression_line_utils.py

@@ -1,19 +0,0 @@
-from matplotlib.lines import Line2D
-import numpy as np
-
-
-def find_regression_line(axes):
-    """
-    Helper to find the regression line (Line2D) in the given axes.
-    """
-    return next(
-        (
-            artist
-            for artist in axes.get_children()
-            if isinstance(artist, Line2D)
-            and artist.get_label() not in (None, "", "_nolegend_")
-            and artist.get_xydata() is not None
-            and np.asarray(artist.get_xydata()).size > 0
-        ),
-        None,
-    )