iplotx 1.1.0__py3-none-any.whl → 1.2.0__py3-none-any.whl

iplotx/__init__.py

@@ -11,6 +11,7 @@ from .plotting import (
     network,
     graph,
     tree,
+    doubletree,
     plot,
 )
 import iplotx.artists as artists
@@ -20,6 +21,7 @@ import iplotx.style as style
 __all__ = [
     "network",
     "tree",
+    "doubletree",
     "plot",
     "graph",
     "artists",

iplotx/edge/__init__.py

@@ -274,6 +274,12 @@ class EdgeCollection(mpl.collections.PatchCollection):
         """Return mappable for colorbar."""
         return self
 
+    def shift(self, x: float, y: float) -> None:
+        """Shift the cascade by a certain amount."""
+        for path in self._paths:
+            path.vertices[:, 0] += x
+            path.vertices[:, 1] += y
+
     def _get_adjacent_vertices_info(self):
         index = self._vertex_collection.get_index()
         index = pd.Series(

iplotx/plotting.py

@@ -1,4 +1,8 @@
-from typing import Optional, Sequence
+from typing import (
+    Optional,
+    Sequence,
+    Any,
+)
 from contextlib import nullcontext
 import numpy as np
 import pandas as pd
@@ -269,8 +273,86 @@ def tree(
     return artist
 
 
+def doubletree(
+    tree_left: Optional[TreeType] = None,
+    tree_right: Optional[TreeType] = None,
+    kwargs_left: Optional[dict[Any]] = None,
+    kwargs_right: Optional[dict[Any]] = None,
+    gap: float = 0,
+    ax: Optional[mpl.axes.Axes] = None,
+    title: Optional[str] = None,
+    aspect: Optional[str | float] = None,
+    margins: float | tuple[float, float] = 0,
+    strip_axes: bool = True,
+    figsize: Optional[tuple[float, float]] = None,
+) -> tuple[TreeArtist, TreeArtist]:
+    """Visualize two trees facing each other.
+
+    Parameters:
+        tree_left: The tree to plot on the left side.
+        tree_right: The tree to plot on the right side.
+        kwargs_left: Additional keyword arguments passed to the left tree plotting function.
+        kwargs_right: Additional keyword arguments passed to the right tree plotting function.
+        ax: The axis to plot on. If None, a new figure and axis will be created. Defaults to
+            None.
+        title: If not None, set the axes title to this value.
+        aspect: If not None, set the aspect ratio of the axis to this value. The most common
+            value is 1.0, which proportionates x- and y-axes.
+        margins: How much margin to leave around the plot. A higher value (e.g. 0.1) can be
+            used as a quick fix when some vertex shapes reach beyond the plot edge. This is
+            a fraction of the data limits, so 0.1 means 10% of the data limits will be left
+            as margin.
+        strip_axes: If True, remove axis spines and ticks.
+        figsize: If ax is None, a new matplotlib Figure is created. This argument specifies
+            the (width, height) dimension of the figure in inches. If ax is not None, this
+            argument is ignored. If None, the default matplotlib figure size is used.
+    Returns:
+        A tuple with the left and right TreeArtist objects.
+    """
+    artist1 = tree(
+        tree_left,
+        layout="horizontal",
+        layout_orientation="right",
+        ax=ax,
+        strip_axes=False,
+        figsize=figsize,
+        **kwargs_left or {},
+    )
+
+    ax = artist1.axes
+
+    if kwargs_right is None:
+        kwargs_right = {}
+
+    had_layout_start = "layout_start" in kwargs_right
+
+    artist2 = tree(
+        tree_right,
+        layout="horizontal",
+        layout_orientation="left",
+        ax=ax,
+        title=title,
+        aspect=aspect,
+        strip_axes=False,
+        margins=margins,
+        **kwargs_right,
+    )
+
+    if not had_layout_start:
+        x2min = artist2.get_layout().values[:, 0].min()
+        x1max = artist1.get_layout().values[:, 0].max()
+        xshift = x1max - x2min + gap
+
+        artist2.shift(0.5 * xshift, 0)
+        artist1.shift(-0.5 * xshift, 0)
+
+    _postprocess_axes(ax, [artist1, artist2], strip=strip_axes, ignore_previous=True)
+
+    return (artist1, artist2)
+
+
 # INTERNAL ROUTINES
-def _postprocess_axes(ax, artists, strip=True, had_data=None):
+def _postprocess_axes(ax, artists, strip=True, had_data=None, ignore_previous=False):
     """Postprocess axis after plotting."""
 
     if strip:
@@ -309,7 +391,11 @@ def _postprocess_axes(ax, artists, strip=True, had_data=None):
         for art in artists:
             bboxes.append(art.get_datalim(ax.transData))
         bbox = mpl.transforms.Bbox.union(bboxes)
-        ax.update_datalim(bbox)
+
+        if not ignore_previous:
+            ax.update_datalim(bbox)
+        else:
+            ax.dataLim.update_from_data_xy(bbox.corners(), ignore=True)
 
         # Autoscale for x/y axis limits
         ax.autoscale_view()

iplotx/tree/__init__.py

@@ -219,6 +219,31 @@ class TreeArtist(mpl.artist.Artist):
         else:
             raise ValueError(f"Unknown layout kind: {kind}. Use 'vertex' or 'edge'.")
 
+    def shift(self, x: float, y: float) -> None:
+        """Shift layout coordinates for all tree elements.
+
+        Paramerers:
+            x: The shift in x direction.
+            y: The shift in y direction.
+        """
+        layout_columns = [f"_ipx_layout_{i}" for i in range(self._ipx_internal_data["ndim"])]
+        self._ipx_internal_data["vertex_df"][layout_columns[0]] += x
+        self._ipx_internal_data["vertex_df"][layout_columns[1]] += y
+
+        self.get_vertices()._layout.values[:, 0] += x
+        self.get_vertices()._layout.values[:, 1] += y
+        self.get_vertices()._update_offsets_from_layout()
+
+        self.get_edges().shift(x, y)
+
+        if hasattr(self, "_leaf_vertices"):
+            self.get_leaf_vertices()._layout.values[:, 0] += x
+            self.get_leaf_vertices()._layout.values[:, 1] += y
+            self.get_leaf_vertices()._update_offsets_from_layout()
+
+        if hasattr(self, "_cascades"):
+            self._cascades.shift(x, y)
+
     def get_datalim(self, transData, pad=0.15):
         """Get limits on x/y axes based on the graph layout data.
 

iplotx/tree/cascades.py

@@ -158,6 +158,12 @@ class CascadeCollection(mpl.collections.PatchCollection):
             zorder=zorder,
         )
 
+    def shift(self, x: float, y: float) -> None:
+        """Shift the cascade by a certain amount."""
+        for path in self._paths:
+            path.vertices[:, 0] += x
+            path.vertices[:, 1] += y
+
     def get_maxdepth(self) -> float:
         """Get the maxdepth of the cascades.
 

iplotx/version.py

@@ -2,4 +2,4 @@
 iplotx version information module.
 """
 
-__version__ = "1.1.0"
+__version__ = "1.2.0"

{iplotx-1.1.0.dist-info → iplotx-1.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iplotx
-Version: 1.1.0
+Version: 1.2.0
 Summary: Plot networkx from igraph and networkx.
 Project-URL: Homepage, https://github.com/fabilab/iplotx
 Project-URL: Documentation, https://readthedocs.org/iplotx
@@ -78,8 +78,7 @@ import iplotx as ipx
 
 g = nx.Graph([(0, 1), (1, 2), (2, 3), (3, 4), (4, 0)])
 layout = nx.layout.circular_layout(g)
-fig, ax = plt.subplots(figsize=(3, 3))
-ipx.plot(g, ax=ax, layout=layout)
+ipx.plot(g, layout)
 ```
 
 ![Quick start image](/docs/source/_static/graph_basic.png)

{iplotx-1.1.0.dist-info → iplotx-1.2.0.dist-info}/RECORD

@@ -1,16 +1,16 @@
-iplotx/__init__.py,sha256=RzSct91jO8abrxOIn33rKEnDUgYpu1oj4olbObgX_hs,489
+iplotx/__init__.py,sha256=RKlRSSEAv2qECd6rCiovdLDu-4k1eXMGCOCPt0xwpFA,523
 iplotx/artists.py,sha256=2dBDT240zGwKb6tIc_y9pXeyU3LuYeF9wjj2tvi4KJo,730
 iplotx/label.py,sha256=7eS8ByadrhdIFOZz19U4VrS-oXY_ndFYNB-D4RZbFqI,9573
 iplotx/layout.py,sha256=S-iFxHaIOzhBDG2JUzl9_oDBRP5TYY1hXnEOs0h1Rck,5588
-iplotx/plotting.py,sha256=FvV33DCuEjJwO9ytiYJuQmfOywgF-cDANd6nEE5s8R0,13211
+iplotx/plotting.py,sha256=RyAdvaHSpuyJkf8DF3SJBvEXBrPmJEdovUyAlBWQvqU,16228
 iplotx/typing.py,sha256=QLdzV358IiD1CFe88MVp0D77FSx5sSAVUmM_2WPPE8I,1463
-iplotx/version.py,sha256=mrnbNTnvhHXH5H10Drqi5HLqPXOwoyNlRYIAv7vTAy0,66
+iplotx/version.py,sha256=t6hulw8HK7i9Hmcp81rtaFx81Xw1b0UWhjmKsoS2Ld8,66
 iplotx/vertex.py,sha256=_yYyvusn4vYvi6RBEW6CHa3vnbv43GnZylnMIaK4bG0,16040
 iplotx/art3d/vertex.py,sha256=Xf8Um30X2doCd8KdNN7332F6BxC4k72Mb_GeRAuzQfQ,2545
 iplotx/art3d/edge/__init__.py,sha256=uw1U_mMXqcZAvea-7JbU1PUKULQD1CMMrbwY02tiWRQ,8529
 iplotx/art3d/edge/arrow.py,sha256=14BFXY9kDOUGPZl2fMD9gRVGyaaN5kyd-l6ikBg6WHU,3601
 iplotx/art3d/edge/geometry.py,sha256=76VUmpPG-4Mls7x_994dMwdDPrWWnjT7nHJsHfwK_hA,2467
-iplotx/edge/__init__.py,sha256=lkrMkQFx9PNzorKc9trQ8MggC-nZSyALhvP78DMvhN4,26363
+iplotx/edge/__init__.py,sha256=GlY_CmlMVD1DWJeOb5cFQEP_4K1pPneS_fRLqAxNGMk,26573
 iplotx/edge/arrow.py,sha256=U7vvBo7IMwo1qiyU9cyUEwraOaBcJLgdu9oU2OyoHL4,17453
 iplotx/edge/geometry.py,sha256=jkTMvQC5425GjB_fmGLIPJeSDAr_7NZF8zZDLTrSj34,15541
 iplotx/edge/leaf.py,sha256=SyGMv2PIOoH0pey8-aMVaZheK3hNe1Qz_okcyWbc4E4,4268
@@ -33,13 +33,13 @@ iplotx/network/groups.py,sha256=E_eYVXRHjv1DcyA4RupTkMa-rRFrIKkt9Rxn_Elw9Nc,6796
 iplotx/style/__init__.py,sha256=rf1GutrE8hHUhCoe4FGKYX-aNtHuu_U-kYQnqUxZNrY,10282
 iplotx/style/leaf_info.py,sha256=3xBn7xv9Uy2KAqdhM9S6ew5ZBJrGRTXRL3xXb8atfLw,1018
 iplotx/style/library.py,sha256=58Y8BlllGLsR4pQM7_PVCP5tH6_4GkchXZvJpqGHlcg,8534
-iplotx/tree/__init__.py,sha256=mFCgXSusB1dSMc-xN_zOc5PojpEf_XyP8yR3xeZVQMY,30065
-iplotx/tree/cascades.py,sha256=on5GyqbWasl1zgK7bYXYQE0LOSfHc1z-1hnm0GWd6aw,8184
+iplotx/tree/__init__.py,sha256=nV2iWrtpPhpGZnkEJqK8ydimMi7pJtRQ1Eqn5dgqUyA,31043
+iplotx/tree/cascades.py,sha256=Wwqhy46QGeb4LNGUuz_-bgNWUMz6PFzs_dIxIb1dtqc,8394
 iplotx/tree/scalebar.py,sha256=Yxt_kF8JdTwKGa8Jzqt3qVePPK5ZBG8P0EiONrsh3E8,11863
 iplotx/utils/geometry.py,sha256=6RrC6qaB0-1vIk1LhGA4CfsiMd-9JNniSPyL_l9mshE,9245
 iplotx/utils/internal.py,sha256=WWfcZDGK8Ut1y_tOHRGg9wSqY1bwSeLQO7dHM_8Tvwo,107
 iplotx/utils/matplotlib.py,sha256=p_53Oamof0RI4mtV8HrdDtZbgVqUxeUZ_KDvLZSiBUQ,8604
 iplotx/utils/style.py,sha256=vyNP80nDYVinqm6_9ltCJCtjK35ZcGlHvOskNv3eQBc,4225
-iplotx-1.1.0.dist-info/METADATA,sha256=BBWbFLjqgs00VaB3DXNSex39oA0iDplDgBeCmrO6hKw,5053
-iplotx-1.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-iplotx-1.1.0.dist-info/RECORD,,
+iplotx-1.2.0.dist-info/METADATA,sha256=9QR6MWdU5XcKIB6JTwEX58Gw5FQInscAdB27AKKidi4,5000
+iplotx-1.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+iplotx-1.2.0.dist-info/RECORD,,