Support plotnine v0.12.1

Author: ponnhide

patchworklib/__init__.py

@@ -1,2 +1,2 @@
 from .patchworklib import *
-__version__ = "0.6.0"
+__version__ = "0.6.1"

patchworklib/modified_grid.py

@@ -25,6 +25,7 @@
     _core_docs,
 )
 
+
 #Referred to Seaborn, v0.11.2, Copyright (c) 2012-2021, Michael L. Waskom.
 def add_legend(self, legend_data=None, title=None, label_order=None,
                adjust_subtitles=False, **kwargs):
@@ -577,3 +578,4 @@ def __plot_for_clustergrid__(self, metric, method, colorbar_kws, row_cluster, co
     #mpl.rcParams["figure.subplot.hspace"] = hspace 
     #mpl.rcParams["figure.subplot.wspace"] = wspace
     return self
+

patchworklib/modified_plotnine.py

@@ -0,0 +1,94 @@
+from copy import deepcopy
+
+def draw(self, return_ggplot=False, show: bool = False):
+    """
+    Render the complete plot
+
+    Parameters
+    ----------
+    show : bool (default: False)
+        Whether to show the plot.
+
+    Returns
+    -------
+    fig : ~matplotlib.figure.Figure
+        Matplotlib figure
+    """
+    import matplotlib as mpl 
+    from plotnine._mpl.layout_engine import PlotnineLayoutEngine
+    from plotnine.ggplot import plot_context
+    plot_context.__exit__ = __exit__
+    plot_context.__enter__ = __enter__
+    # Do not draw if drawn already.
+    # This prevents a needless error when reusing
+    # figure & axes in the jupyter notebook.
+    if hasattr(self, "figure"):
+        return self.figure
+
+    # Prevent against any modifications to the users
+    # ggplot object. Do the copy here as we may/may not
+    # assign a default theme
+    self = deepcopy(self)
+    
+    with plot_context(self, show=show):
+        self._build()
+
+        # setup
+        figure, axs = self._create_figure()
+        self._setup_parameters()
+        self.theme.setup()
+        self.facet.strips.generate()
+
+        # Drawing
+        self._draw_layers()
+        self._draw_breaks_and_labels()
+        self._draw_legend()
+        self._draw_figure_texts()
+        self._draw_watermarks()
+
+        # Artist object theming
+        self.theme.apply()  
+        figure.set_layout_engine(PlotnineLayoutEngine(self))
+
+    if return_ggplot == True:
+        return figure, self 
+    else:
+        return figure 
+
+def __enter__(self):
+    """
+    Enclose in matplolib & pandas environments
+    """
+    import pandas as pd 
+    import matplotlib as mpl
+
+    self.plot.theme._targets = {}
+    self.rc_context = mpl.rc_context(self.plot.theme.rcParams)
+    # Pandas deprecated is_copy, and when we create new dataframes
+    # from slices we do not want complaints. We always uses the
+    # new frames knowing that they are separate from the original.
+    self.pd_option_context = pd.option_context(
+        "mode.chained_assignment", None
+    )
+    self.rc_context.__enter__()
+    self.pd_option_context.__enter__()
+    return self
+
+def __exit__(self, exc_type, exc_value, exc_traceback):
+    """
+    Exit matplotlib & pandas environments
+    """
+    import matplotlib.pyplot as plt
+
+    if exc_type is None:
+        if self.show:
+            plt.show()
+        else:
+            plt.close(self.plot.figure)
+    else:
+        # There is an exception, close any figure
+        if hasattr(self.plot, "figure"):
+            plt.close(self.plot.figure)
+
+    self.rc_context.__exit__(exc_type, exc_value, exc_traceback)
+    self.pd_option_context.__exit__(exc_type, exc_value, exc_traceback)