iplotx 0.7.0__py3-none-any.whl → 0.8.0__py3-none-any.whl

iplotx/network.py

@@ -162,6 +162,8 @@ class NetworkArtist(mpl.artist.Artist):
         """Get VertexCollection artist."""
         return self._vertices
 
+    get_nodes = get_vertices
+
     def get_edges(self):
         """Get EdgeCollection artist."""
         return self._edges
@@ -170,6 +172,8 @@ class NetworkArtist(mpl.artist.Artist):
         """Get list of vertex label artists."""
         return self._vertices.get_labels()
 
+    get_node_labels = get_vertex_labels
+
     def get_edge_labels(self):
         """Get list of edge label artists."""
         return self._edges.get_labels()
@@ -211,6 +215,10 @@ class NetworkArtist(mpl.artist.Artist):
         return vertex_layout_df
 
     def _get_label_series(self, kind):
+        # Equivalence vertex/node
+        if kind == "node":
+            kind = "vertex"
+
         if "label" in self._ipx_internal_data[f"{kind}_df"].columns:
             return self._ipx_internal_data[f"{kind}_df"]["label"]
         else:

iplotx/style/__init__.py

@@ -1,16 +1,24 @@
+"""
+Main style module for iplotx.
+"""
+
 from typing import (
     Any,
     Iterable,
     Optional,
     Sequence,
 )
-from collections import defaultdict
 from collections.abc import Hashable
 from contextlib import contextmanager
 import numpy as np
 import pandas as pd
 
-from ..utils.style import copy_with_deep_values
+from ..utils.style import (
+    copy_with_deep_values,
+    sanitize_leaves,
+    update_style,
+    sanitize_ambiguous,
+)
 from .library import style_library
 from .leaf_info import (
     style_leaves,
@@ -83,94 +91,16 @@ def merge_styles(
     Returns:
         The composite style as a dict.
     """
-    try:
-        import networkx as nx
-    except ImportError:
-        nx = None
-
-    def _sanitize_leaves(style: dict):
-        for key, value in style.items():
-            if key in style_leaves:
-                # Networkx has a few lazy data structures
-                # TODO: move this code to provider
-                if nx is not None:
-                    if isinstance(value, nx.classes.reportviews.NodeView):
-                        style[key] = dict(value)
-                    elif isinstance(value, nx.classes.reportviews.EdgeViewABC):
-                        style[key] = [v for *e, v in value]
-
-            elif isinstance(value, dict):
-                _sanitize_leaves(value)
-
-    def _update(style: dict, current: dict):
-        for key, value in style.items():
-            if key not in current:
-                current[key] = value
-                continue
-
-            # Style non-leaves are either recurred into or deleted
-            if key not in style_leaves:
-                if isinstance(value, dict):
-                    _update(value, current[key])
-                elif value is None:
-                    del current[key]
-                else:
-                    raise ValueError(
-                        f"Setting non-leaf style value to a non-dict: {key}, {value}",
-                    )
-            else:
-                # Style leaves could be incomplete, ensure a sensible default
-                if value is None:
-                    del current[key]
-                    continue
-
-                if not isinstance(value, dict):
-                    current[key] = value
-                    continue
-
-                if hasattr(value, "default_factory"):
-                    current[key] = value
-                    continue
-
-                if hasattr(current[key], "default_factory"):
-                    default_value = current[key].default_factory()
-                else:
-                    default_value = current[key]
-                current[key] = defaultdict(
-                    lambda: default_value,
-                    value,
-                )
-
-    def _sanitize_ambiguous(style: dict):
-        """Fix a few ambiguous cases in the style dict."""
-
-        # Accept "node" style as "vertex" style for user flexibility
-        if "node" in style:
-            style_node = style.pop("node")
-            if "vertex" not in style:
-                style["vertex"] = style_node
-            else:
-                # "node" style applies on TOP of "vertex" style
-                _update(style_node, style["vertex"])
-
-        # NOTE: Deprecate edge_padding for edge_shrink
-        for edgekey in ["edge", "leafedge"]:
-            if "padding" in style.get(edgekey, {}):
-                # shrink takes over
-                if "shrink" in style[edgekey]:
-                    del style[edgekey]["padding"]
-                else:
-                    style[edgekey]["shrink"] = style[edgekey].pop("padding")
 
     merged = {}
     for style in styles:
         if isinstance(style, str):
             style = get_style(style)
         else:
-            _sanitize_leaves(style)
+            sanitize_leaves(style)
             unflatten_style(style)
-            _sanitize_ambiguous(style)
-        _update(style, merged)
+            sanitize_ambiguous(style)
+        update_style(style, merged)
 
     return merged
 

iplotx/tree.py

@@ -193,6 +193,10 @@ class TreeArtist(mpl.artist.Artist):
         """Get vertex or edge layout."""
         layout_columns = [f"_ipx_layout_{i}" for i in range(self._ipx_internal_data["ndim"])]
 
+        # Equivalence vertex <-> node
+        if kind == "node":
+            kind = "vertex"
+
         if kind == "vertex":
             layout = self._ipx_internal_data["vertex_df"][layout_columns]
             return layout
@@ -245,6 +249,10 @@ class TreeArtist(mpl.artist.Artist):
         return bbox
 
     def _get_label_series(self, kind: str) -> Optional[pd.Series]:
+        # Equivalence vertex <-> node
+        if kind == "node":
+            kind = "vertex"
+
         if "label" in self._ipx_internal_data[f"{kind}_df"].columns:
             return self._ipx_internal_data[f"{kind}_df"]["label"]
         else:
@@ -254,6 +262,8 @@ class TreeArtist(mpl.artist.Artist):
         """Get VertexCollection artist."""
         return self._vertices
 
+    get_nodes = get_vertices
+
     def get_edges(self) -> EdgeCollection:
         """Get EdgeCollection artist."""
         return self._edges
@@ -262,6 +272,8 @@ class TreeArtist(mpl.artist.Artist):
         """Get leaf VertexCollection artist."""
         return self._leaf_vertices
 
+    get_leaf_nodes = get_leaf_vertices
+
     def get_leaf_edges(self) -> Optional[LeafEdgeCollection]:
         """Get LeafEdgeCollection artist if present."""
         if hasattr(self, "_leaf_edges"):
@@ -272,6 +284,8 @@ class TreeArtist(mpl.artist.Artist):
         """Get list of vertex label artists."""
         return self._vertices.get_labels()
 
+    get_node_labels = get_vertex_labels
+
     def get_edge_labels(self) -> LabelCollection:
         """Get list of edge label artists."""
         return self._edges.get_labels()

iplotx/utils/style.py

@@ -1,6 +1,15 @@
 import copy
 from collections import defaultdict
 
+from ..style.leaf_info import (
+    style_leaves,
+)
+
+try:
+    import networkx as nx
+except ImportError:
+    nx = None
+
 
 def copy_with_deep_values(style):
     """Make a deep copy of the style dict but do not create copies of the keys."""
@@ -15,3 +24,105 @@ def copy_with_deep_values(style):
         else:
             newdict[key] = copy.copy(value)
     return newdict
+
+
+def sanitize_leaves(style: dict):
+    """Sanitize the leaves of a style dictionary.
+
+    Parameters:
+        style (dict): A style dictionary.
+    Returns:
+        None: The style dictionary is modified in place.
+    """
+    for key, value in style.items():
+        if key in style_leaves:
+            # Networkx has a few lazy data structures
+            # TODO: move this code to provider
+            if nx is not None:
+                if isinstance(value, nx.classes.reportviews.NodeView):
+                    style[key] = dict(value)
+                elif isinstance(value, nx.classes.reportviews.EdgeViewABC):
+                    style[key] = [v for *e, v in value]
+
+        elif isinstance(value, dict):
+            sanitize_leaves(value)
+
+
+def update_style(new_style: dict, current_style: dict):
+    """Update the current style with a new style.
+
+    Parameters:
+        new_style (dict): A new style dictionary.
+        current_style (dict): The current style dictionary.
+    Returns:
+        None: The current style dictionary is modified in place.
+    """
+    for key, value in new_style.items():
+        if key not in current_style:
+            current_style[key] = value
+            continue
+
+        # Style non-leaves are either recurred into or deleted
+        if key not in style_leaves:
+            if isinstance(value, dict):
+                update_style(value, current_style[key])
+            elif value is None:
+                del current_style[key]
+            else:
+                raise ValueError(
+                    f"Setting non-leaf style value to a non-dict: {key}, {value}",
+                )
+        else:
+            # Style leaves could be incomplete, ensure a sensible default
+            if value is None:
+                del current_style[key]
+                continue
+
+            if not isinstance(value, dict):
+                current_style[key] = value
+                continue
+
+            if hasattr(value, "default_factory"):
+                current_style[key] = value
+                continue
+
+            if hasattr(current_style[key], "default_factory"):
+                default_value = current_style[key].default_factory()
+            else:
+                default_value = current_style[key]
+            current_style[key] = defaultdict(
+                lambda: default_value,
+                value,
+            )
+
+
+def sanitize_ambiguous(style: dict):
+    """Fix a few ambiguous cases in the style dict.
+
+    Parameters:
+        style (dict): A style dictionary. This must be unflattened beforehand.
+    Returns:
+        None: The style dictionary is modified in place.
+
+    NOTE: This function exists by design, not accident. It is useful for purposeful
+    (e.g. node vs vertex) or historical (e.g. edge_padding vs edge_shrink) reasons.
+    Either way, it softens the user experience without complicating the API.
+    """
+
+    # Accept "node" style as "vertex" style for user flexibility
+    if "node" in style:
+        style_node = style.pop("node")
+        if "vertex" not in style:
+            style["vertex"] = style_node
+        else:
+            # "node" style applies on TOP of "vertex" style
+            update_style(style_node, style["vertex"])
+
+    # NOTE: Deprecate edge_padding for edge_shrink
+    for edgekey in ["edge", "leafedge"]:
+        if "padding" in style.get(edgekey, {}):
+            # shrink takes over
+            if "shrink" in style[edgekey]:
+                del style[edgekey]["padding"]
+            else:
+                style[edgekey]["shrink"] = style[edgekey].pop("padding")

iplotx/version.py

@@ -2,4 +2,4 @@
 iplotx version information module.
 """
 
-__version__ = "0.7.0"
+__version__ = "0.8.0"

{iplotx-0.7.0.dist-info → iplotx-0.8.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iplotx
-Version: 0.7.0
+Version: 0.8.0
 Summary: Plot networkx from igraph and networkx.
 Project-URL: Homepage, https://github.com/fabilab/iplotx
 Project-URL: Documentation, https://readthedocs.org/iplotx

{iplotx-0.7.0.dist-info → iplotx-0.8.0.dist-info}/RECORD

@@ -4,11 +4,11 @@ iplotx/cascades.py,sha256=OPqF7Huls-HFmDA5MCF6DEZlUeRVaXsbQcHBoKAgNJs,8182
 iplotx/groups.py,sha256=_9KdIiTAi1kXtd2mDywgBJCbqoRq2z-5fzOPf76Wgb8,6287
 iplotx/label.py,sha256=6am3a0ejcW_bWEXSOODE1Ke3AyCU1lJ45RfnXNbHAQw,8923
 iplotx/layout.py,sha256=KxmRLqjo8AYCBAmXez8rIiLU2sM34qhb6ox9AHYwRyE,4839
-iplotx/network.py,sha256=LaW9zQZ4sKiDVb25_icnquGNnN7HrKC7NO07o6PSmGI,11527
+iplotx/network.py,sha256=SGmXXrFxqgOoQcEJSZdL3WiI6rHDlPOGg5loGPrYpDk,11688
 iplotx/plotting.py,sha256=IEUxW1xzTljLjBfsVP2BNsOPCDpj5bEPZ99bzvD5-mo,10066
-iplotx/tree.py,sha256=-69lpPjisRfigY4fDQcmVxzOAf1tGwMXviUcW6mZU6U,28508
+iplotx/tree.py,sha256=TxbNoBHS0CfswrcMIWCNtnOl_3e4-PwCrVo0goywC0U,28807
 iplotx/typing.py,sha256=QLdzV358IiD1CFe88MVp0D77FSx5sSAVUmM_2WPPE8I,1463
-iplotx/version.py,sha256=8BIMMcy91d9_BPlUuoNyMNeyj04-gejk1STd-SLsrT0,66
+iplotx/version.py,sha256=tH4d2ah5uIPOD3rmxSuQMJ0PfIEEE0SOFEJKo0zwzXU,66
 iplotx/vertex.py,sha256=hqdlD9fRBSwH5bRvlpaaPu7jgUR4z9nob1SYfPWDxtI,14966
 iplotx/edge/__init__.py,sha256=AVnLsrDWWCkix1LVhrjpWKEKDxOp8joM4tF6RqEHC8I,27115
 iplotx/edge/arrow.py,sha256=ZKt3UNZ7XRa2S3KxpoQfd4q_6eSUHOS476BZNqlf2pw,16462
@@ -26,13 +26,13 @@ iplotx/ingest/providers/tree/cogent3.py,sha256=JmELbDK7LyybiJzFNbmeqZ4ySJoDajvFf
 iplotx/ingest/providers/tree/ete4.py,sha256=D7usSq0MOjzrk3EoLi834IlaDGwv7_qG6Qt0ptfKqfI,928
 iplotx/ingest/providers/tree/simple.py,sha256=aV9wGqBomJ5klM_aJQeuL_Q_J1pLCv6AFN98BPDiKUw,2593
 iplotx/ingest/providers/tree/skbio.py,sha256=O1KUr8tYi28pZ3VVjapgO4Uj-YpMuix3GhOH5je8Lv4,822
-iplotx/style/__init__.py,sha256=_LhOJ9WJzC3of_nbaJyAJX3k4paI8utrigTzYDZRSB8,13138
+iplotx/style/__init__.py,sha256=rf1GutrE8hHUhCoe4FGKYX-aNtHuu_U-kYQnqUxZNrY,10282
 iplotx/style/leaf_info.py,sha256=mcd6ewZl3jC0CPshmbeUkNp2geoihJW9515roGy2T8o,1000
 iplotx/style/library.py,sha256=58Y8BlllGLsR4pQM7_PVCP5tH6_4GkchXZvJpqGHlcg,8534
 iplotx/utils/geometry.py,sha256=6RrC6qaB0-1vIk1LhGA4CfsiMd-9JNniSPyL_l9mshE,9245
 iplotx/utils/internal.py,sha256=WWfcZDGK8Ut1y_tOHRGg9wSqY1bwSeLQO7dHM_8Tvwo,107
 iplotx/utils/matplotlib.py,sha256=wELE73quQv10-1w9uA5eDTgkZkylJvjg7pd3K5tZPOo,6294
-iplotx/utils/style.py,sha256=wMWxJykxBD-JmcN8-rSKlWcV6pMfwKgR4EzSpk_NX8k,547
-iplotx-0.7.0.dist-info/METADATA,sha256=vbdamou2U35GeUBn4_fG8kzeC9fTwYRzKeg2Ik9212Q,4908
-iplotx-0.7.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-iplotx-0.7.0.dist-info/RECORD,,
+iplotx/utils/style.py,sha256=vyNP80nDYVinqm6_9ltCJCtjK35ZcGlHvOskNv3eQBc,4225
+iplotx-0.8.0.dist-info/METADATA,sha256=L756OyuMisG3j2Y_OsgvcqUcCavwDbVAPvfwBg7S9dQ,4908
+iplotx-0.8.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+iplotx-0.8.0.dist-info/RECORD,,