ggplot2-python 4.0.2.9000__py3-none-any.whl → 4.0.2.9000.post1__py3-none-any.whl

ggplot2_py/__init__.py

@@ -7,7 +7,7 @@ approach to creating statistical visualizations.
 
 from __future__ import annotations
 
-__version__ = "4.0.2.9000"
+__version__ = "4.0.2.9000.post1"
 __r_commit__ = "c02c05a"
 
 # ---------------------------------------------------------------------------

ggplot2_py/plot_render.py

@@ -55,6 +55,130 @@ def _legend_label_width_cm(labels: List[Any], fontsize: float = 6.0) -> float:
     return max(max_w, 0.3)  # minimum width 0.3 cm
 
 
+# ---------------------------------------------------------------------------
+# Layer-to-guide filtering (ports of R's matched_aes / include_layer_in_guide,
+# ``ggplot2/R/guides-.R:871-912``).
+#
+# R's GuideLegend$process_layers only forwards layers whose aesthetic
+# mapping actually maps one of the guide's aesthetics, unless the user
+# explicitly set ``show.legend=TRUE``. Without this filter, a legend
+# picks up ``draw_key`` from any convenient layer (e.g. a backbone
+# ``geom_segment`` with a fixed black colour) and renders black path
+# glyphs instead of the colour-scale dots it should show.
+# ---------------------------------------------------------------------------
+
+_AES_SYNONYMS: Dict[str, str] = {"color": "colour"}
+
+
+def _canon_aes(name: str) -> str:
+    return _AES_SYNONYMS.get(name, name)
+
+
+def _aes_key_set(obj: Any) -> set:
+    """Return the canonicalised set of aesthetic names in a mapping-like obj."""
+    if obj is None:
+        return set()
+    try:
+        keys = obj.keys() if hasattr(obj, "keys") else list(obj)
+    except Exception:
+        return set()
+    return {_canon_aes(str(k)) for k in keys}
+
+
+def _matched_aes(layer: Any, guide_aes: set) -> set:
+    """Port of R's ``matched_aes`` (``guides-.R:871-880``).
+
+    Returns the canonical aesthetic names that are *mapped* by this
+    layer's ``aes()`` and also part of the guide's key columns, excluding
+    aesthetics that are fixed (``aes_params``/``computed_geom_params``).
+    """
+    mapping_keys = _aes_key_set(getattr(layer, "computed_mapping", None)
+                                or getattr(layer, "mapping", None))
+    stat = getattr(layer, "stat", None)
+    stat_default = _aes_key_set(getattr(stat, "default_aes", None))
+    all_names = mapping_keys | stat_default
+
+    geom = getattr(layer, "geom", None)
+    geom_required = set()
+    geom_default = set()
+    if geom is not None:
+        req = getattr(geom, "required_aes", None)
+        if req is not None:
+            geom_required = {_canon_aes(str(a)) for a in req}
+        geom_default = _aes_key_set(getattr(geom, "default_aes", None))
+    geom_names = geom_required | geom_default
+    # R's rename_size shim: size-renaming geoms contribute to size
+    # legends even without mapping "size" explicitly.
+    if geom is not None and getattr(geom, "rename_size", False):
+        if "size" in all_names and "linewidth" not in all_names:
+            geom_names = geom_names | {"size"}
+
+    matched = (all_names & geom_names) & {_canon_aes(a) for a in guide_aes}
+    matched -= _aes_key_set(getattr(layer, "computed_geom_params", None))
+    matched -= _aes_key_set(getattr(layer, "aes_params", None))
+    return matched
+
+
+def _include_layer_in_guide(layer: Any, matched: set) -> bool:
+    """Port of R's ``include_layer_in_guide`` (``guides-.R:885-912``)."""
+    show = getattr(layer, "show_legend", None)
+    # Non-logical values: R warns and treats as FALSE. Python accepts
+    # None (= NA) and bool; anything else is coerced to False.
+    if show is not None and not isinstance(show, (bool, np.bool_)):
+        # Named-dict form (``show.legend=c(colour=TRUE)``) — uncommon in
+        # ggplot2_py but supported for completeness.
+        if isinstance(show, dict):
+            if not matched:
+                return False
+            picks = {_canon_aes(k): v for k, v in show.items()}
+            vals = [picks[a] for a in matched if a in picks and picks[a] is not None]
+            return len(vals) == 0 or any(vals)
+        return False
+
+    if matched:
+        # Layer maps at least one of the guide's aesthetics:
+        # include unless show.legend is explicitly FALSE.
+        if show is None:
+            return True
+        return bool(show)
+    # Layer does not map any guide aesthetic: include only if show.legend
+    # is explicitly TRUE.
+    return show is True
+
+
+def _resolve_draw_key_for_entry(
+    entry: Dict[str, Any], layers: Any,
+) -> tuple[Any, List[Any]]:
+    """Pick the ``draw_key`` and layer subset for a single legend entry.
+
+    Mirrors R's ``GuideLegend$process_layers`` filtering combined with
+    ``get_layer_key``'s first-layer-wins behaviour for glyph selection.
+    Returns ``(draw_key_fn, included_layers)`` — the included layer
+    list is forwarded to ``build_legend_decor`` so ``aes_params`` /
+    ``default_aes`` resolution also uses only qualifying layers.
+    """
+    from ggplot2_py.draw_key import draw_key_point as _draw_key_point
+
+    guide_aes = {_canon_aes(a) for a in (entry.get("aes_mapped") or {}).keys()}
+    if not guide_aes:
+        guide_aes = {_canon_aes(str(entry.get("aesthetic", "")))}
+
+    included: List[Any] = []
+    if layers:
+        for layer in layers:
+            matched = _matched_aes(layer, guide_aes)
+            if _include_layer_in_guide(layer, matched):
+                included.append(layer)
+
+    draw_key_fn = _draw_key_point
+    for layer in included:
+        geom = getattr(layer, "geom", None)
+        if geom is not None and hasattr(geom, "draw_key"):
+            draw_key_fn = geom.draw_key
+            break
+    return draw_key_fn, included
+
+
 @singledispatch
 def ggplot_gtable(data: Any) -> Any:
     """Convert a built ggplot to a gtable for rendering.
@@ -362,16 +486,11 @@ def _table_add_legends(
     PADDING_CM = 0.15  # R: legend.margin default padding
 
     # ------------------------------------------------------------------
-    # 4. Determine draw_key function from layers
+    # 4. ``draw_key`` is now resolved per-entry inside the loop below
+    # (R's ``GuideLegend$process_layers`` filters layers against each
+    # guide's aesthetics via ``matched_aes`` / ``include_layer_in_guide``,
+    # ``guide-legend.R:219-231``). See ``_resolve_draw_key_for_entry``.
     # ------------------------------------------------------------------
-    from ggplot2_py.draw_key import draw_key_point as _draw_key_point
-    draw_key_fn = _draw_key_point
-    if layers:
-        for layer in layers:
-            geom = getattr(layer, "geom", None)
-            if geom is not None and hasattr(geom, "draw_key"):
-                draw_key_fn = geom.draw_key
-                break
 
     # ------------------------------------------------------------------
     # 5. Build each guide as an independent Gtable
@@ -537,11 +656,27 @@ def _table_add_legends(
             continue
 
         # --- Legend path: discrete scales ---
-        nrow = min(n_breaks, 20)
-        ncol = 1
+        # Mirror R ``GuideLegend$setup_params`` (``guide-legend.R:286-298``):
+        # vertical direction defaults to ``ncol = ceiling(n_breaks / 20)``,
+        # then ``nrow = ceiling(n_breaks / ncol)``. Previously hardcoded
+        # ``ncol = 1`` caused any legend with more than 20 entries to pile
+        # all wrapped entries into a single physical column (multi-column
+        # positions were computed by ``arrange_legend_layout`` but only
+        # one column width was allocated in the gtable), producing
+        # overlapping key + label glyphs per row.
+        ncol = max(1, math.ceil(n_breaks / 20))
+        nrow = max(1, math.ceil(n_breaks / ncol))
+
+        # Per-entry draw_key: mirror R's ``matched_aes`` /
+        # ``include_layer_in_guide`` so ``geom_segment`` / ``geom_path``
+        # layers that don't map the guide's aesthetic can't hijack the
+        # legend key glyph.
+        entry_draw_key_fn, entry_layers = _resolve_draw_key_for_entry(
+            entry, layers,
+        )
 
         decor = build_legend_decor(
-            entry, draw_key_fn, layers,
+            entry, entry_draw_key_fn, entry_layers,
             key_width_cm=KEY_W_CM, key_height_cm=KEY_H_CM,
             theme=theme,
         )

ggplot2_py/theme_elements.py

@@ -1425,16 +1425,26 @@ class _TitleGrob(GTree):
         self._title_widths = widths
         self._title_heights = heights
 
-    # R: widthDetails.titleGrob  → sum(x$widths)
+    # R: widthDetails.titleGrob <- function(x) sum(x$widths)
+    # (``ggplot2/R/margins.R:199-201``). R's ``sum()`` on a unit vector
+    # dispatches to ``Summary.unit`` which wraps the full vector in a
+    # *single-element* L_SUM compound (``grid/R/unit.R:300-347``). That
+    # is what ``evaluateGrobUnit`` expects at ``grid/src/unit.c:535``,
+    # where it reads ``transformWidthtoINCHES(unitx, 0, ...)``. Using
+    # Python's builtin ``sum`` here would leave a multi-element unit
+    # unchanged (because ``0 + Unit`` short-circuits to identity),
+    # which makes grid_py read only the left-margin element and
+    # underestimate the grob's width.
     def width_details(self) -> Any:
         if self._title_widths is not None:
-            return sum(self._title_widths)
+            from grid_py import unit_summary_sum
+            return unit_summary_sum(self._title_widths)
         return Unit(0, "cm")
 
-    # R: heightDetails.titleGrob → sum(x$heights)
     def height_details(self) -> Any:
         if self._title_heights is not None:
-            return sum(self._title_heights)
+            from grid_py import unit_summary_sum
+            return unit_summary_sum(self._title_heights)
         return Unit(0, "cm")
 
 

{ggplot2_python-4.0.2.9000.dist-info → ggplot2_python-4.0.2.9000.post1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ggplot2-python
-Version: 4.0.2.9000
+Version: 4.0.2.9000.post1
 Summary: Python port of the R ggplot2 package (tracks R ggplot2 4.0.2.9000)
 Project-URL: Homepage, https://github.com/Bio-Babel/ggplot2-python
 Project-URL: Repository, https://github.com/Bio-Babel/ggplot2-python
@@ -50,6 +50,8 @@ Description-Content-Type: text/markdown
 
 # ggplot2_py <a href="https://github.com/R2pyBioinformatics/ggplot2_py"><img src="assets/ggplot2_py_logo.png" align="right" height="138" alt="ggplot2_py logo" /></a>
 
+[![PyPI](https://img.shields.io/pypi/v/ggplot2-python)](https://pypi.org/project/ggplot2-python/)
+
 AI-assisted Python port of the R **ggplot2** package — Create Elegant Data Visualisations Using the Grammar of Graphics.
 
 ## Overview

{ggplot2_python-4.0.2.9000.dist-info → ggplot2_python-4.0.2.9000.post1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-ggplot2_py/__init__.py,sha256=_hNuH6YGAjWUXOM-4J0ledrAIG6EI6Qj9JmIVwf6pTM,24411
+ggplot2_py/__init__.py,sha256=Ru2gTGufNQhiqAruJJkR8DjJmXBQERgYpAE93AlQD0Q,24417
 ggplot2_py/_compat.py,sha256=sLzzhq0ii2eSx1xMRaHemNnNqUd6FfT3wRierE1EXCg,10670
 ggplot2_py/_plugins.py,sha256=TLsSEfN5L5VyWtsHaV_fms3TjQezUDUWWEj8ctVxK88,4323
 ggplot2_py/_utils.py,sha256=Qyh2ORUgLlT_rgJ9IaJpzej6G7BIK13nzBR1hF5Zjtw,13612
@@ -21,7 +21,7 @@ ggplot2_py/layer.py,sha256=de6GyIir0L7GLgGQ3e4w3VTKOEwW5h34zmZvlmCg5ZA,29346
 ggplot2_py/layout.py,sha256=QkaGP3anrQYS8vIFMlsz3jKmblHUTYGSevCXIMGlDns,25341
 ggplot2_py/limits.py,sha256=e1WezeXnqPUKn6aaNgyDkLkF1JbxustqfQorKrmjPHI,8219
 ggplot2_py/plot.py,sha256=tMdrEosF9la9rK_QxcRGjFMWvcO7xBg8eG2_kMZy_fM,41102
-ggplot2_py/plot_render.py,sha256=cwdLBFFQn0flsx2L7eg6kPFQ97OMIFprIxqIo1APkIk,29149
+ggplot2_py/plot_render.py,sha256=JIatuPPdBjb6ujKjagTWOYlOzwzPw2zEw3WGGQ0Tesg,35105
 ggplot2_py/position.py,sha256=weKCXpnDtif0NZ9cZqOEs4J96GJhMxUZiBFhZgt0IFM,34339
 ggplot2_py/protocols.py,sha256=7d34T7v7NLg3N1bSEPmmPFrV-5BZvnzBsbpiqnDatZA,5353
 ggplot2_py/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -31,7 +31,7 @@ ggplot2_py/scale.py,sha256=ufYdBhtEv-_KlXYFTzX6iHjS5fd0DTmR1VpE02gSLb8,82254
 ggplot2_py/stat.py,sha256=airtZLlQMoOAMnrn62DoQSMHzAMf8zVEQBQdldl8iWQ,167192
 ggplot2_py/theme.py,sha256=23cbJ64wkfFyw4g474yBFzjFBLtKp3vv61iDdsUA64I,13546
 ggplot2_py/theme_defaults.py,sha256=iA_kJBjjzLpsG7vHzEL5ma7t_9INiy2e6XXtZCfWH68,36311
-ggplot2_py/theme_elements.py,sha256=2e6hMWbLNMPCMqzjlyS1bH6XL5l22EqxdMwKQuqTyw0,63580
+ggplot2_py/theme_elements.py,sha256=Q2z8JZOTj--ntcwY7kPa4b2uri7rR-QoPgK9Rux4E94,64255
 ggplot2_py/coords/__init__.py,sha256=EB2z4ocd6jRlf2f95-_F8xD4aLvf2QFMmqos2PUixiw,834
 ggplot2_py/geoms/__init__.py,sha256=iUCIceJbLSD4TjOtBLCn-8BK2jHVyIJBVYfjTt9BdIA,293
 ggplot2_py/guides/__init__.py,sha256=YPw0bfL6l-xpduIJuwPcq4KVtnJVgOqLcYHzxTh-Fbc,204
@@ -48,7 +48,7 @@ ggplot2_py/resources/seals.csv,sha256=VEuSEBLz3Fj14eBdM6hp6M6GpH2lqN5ekT_d2Mk1rK
 ggplot2_py/resources/txhousing.csv,sha256=RdHoH5W9bud_DzsefIc8yNOFaxMl6IDDKMiP68aoIoY,523993
 ggplot2_py/scales/__init__.py,sha256=n1NLGW0cOeFxOYkj_yS2rvec4p9N1ZJBsugPG2TOHgY,100013
 ggplot2_py/stats/__init__.py,sha256=X_WSwq3jS2iJPGy7jEu_tXSQvVrIvC-5pm4ag2qTLVI,222
-ggplot2_python-4.0.2.9000.dist-info/METADATA,sha256=XlPN5apyiqjKVSHhnC7dJyKzRxjthLK8K2BTMvCIUzY,8231
-ggplot2_python-4.0.2.9000.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
-ggplot2_python-4.0.2.9000.dist-info/licenses/LICENSE,sha256=E0GJjw07cYRmJYj_8UZKbhevWQ-Swd3nVR6qBddvz9c,39
-ggplot2_python-4.0.2.9000.dist-info/RECORD,,
+ggplot2_python-4.0.2.9000.post1.dist-info/METADATA,sha256=sx0qj7GIFOoFlxb50W4MvJS1BcryyG4cDjjfSeToufw,8336
+ggplot2_python-4.0.2.9000.post1.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+ggplot2_python-4.0.2.9000.post1.dist-info/licenses/LICENSE,sha256=E0GJjw07cYRmJYj_8UZKbhevWQ-Swd3nVR6qBddvz9c,39
+ggplot2_python-4.0.2.9000.post1.dist-info/RECORD,,