Implement geo plotting overrides with metaclasses

Author: lukelbd

proplot/axes/base.py

@@ -3,7 +3,9 @@
 The base axes class used for all ProPlot figures.
 """
 import copy
+import functools
 import re
+import types
 from numbers import Integral, Number
 
 import matplotlib.axes as maxes
@@ -57,7 +59,7 @@
 }
 
 
-docstring.snippets['axes.other'] = """
+_format_other_docstring = """
 rc_kw : dict, optional
     Dictionary containing `~proplot.config.rc` settings applied to
     this axes using `~proplot.config.RcConfigurator.context`.
@@ -66,24 +68,16 @@
     and used to update axes `~proplot.config.rc` settings. For example,
     ``abcstyle='A.'`` modifies the :rcraw:`abc.style` setting.
 """
+docstring.snippets['axes.format_other'] = _format_other_docstring
 
-docstring.snippets['axes.patch_kw'] = """
+_patch_kw_docstring = """
 patch_kw : dict-like, optional
     Keyword arguments used to update the background patch. This can
     be used e.g. to apply background hatching with ``patch_kw={'hatch': 'xxx'}``.
 """
+docstring.snippets['axes.patch_kw'] = _patch_kw_docstring
 
-docstring.snippets['axes.proj'] = """
-The map projection specification(s). If ``'cartesian'`` (the default), a
-`~proplot.axes.CartesianAxes` is created. If ``'polar'``, a
-`~proplot.axes.PolarAxes` is created. Otherwise, the argument is
-interpreted by `~proplot.constructor.Proj`, and the result is used
-to make a `~proplot.axes.GeoAxes` (in this case the argument can be
-a `cartopy.crs.Projection` instance, a `~mpl_toolkits.basemap.Basemap`
-instance, or a projection name listed in :ref:`this table <proj_table>`).
-"""
-
-docstring.snippets['axes.inset'] = """
+_inset_docstring = """
 Return an inset `CartesianAxes`. This is similar to the builtin
 `~matplotlib.axes.Axes.inset_axes` but includes some extra options.
 
@@ -129,9 +123,10 @@
 ----------------
 **kwargs
     Passed to `CartesianAxes`.
-""" % docstring.snippets
+"""
+docstring.snippets['axes.inset'] = _inset_docstring
 
-docstring.snippets['axes.panel'] = """
+_panel_docstring = """
 Return a panel drawn along the edge of this axes.
 
 Parameters
@@ -166,9 +161,34 @@
 `~proplot.axes.CartesianAxes`
     The panel axes.
 """
+docstring.snippets['axes.panel'] = _panel_docstring
+
+
+class _MetaAxes(type):
+    """
+    Redirect internal plotting calls to native matplotlib methods.
+    """
+    # NOTE: This obviates the need for basemap-specific redirection to mpl.axes.Axes.
+    def __new__(cls, name, bases, dct_orig):
+        dct = dct_orig.copy()
+        for attr, func_proplot in tuple(dct_orig.items()):
+            if attr[:1] == '_':
+                continue
+            func_matplotlib = getattr(maxes.Axes, attr, None)
+            if not isinstance(func_proplot, types.FunctionType):
+                continue
+            if not isinstance(func_matplotlib, types.FunctionType):
+                continue
+            dct[attr] = functools.wraps(func_proplot)(
+                lambda self, *args, _func_proplot=func_proplot, _func_matplotlib=func_matplotlib, **kwargs:  # noqa: E501
+                _func_matplotlib(self, *args, **kwargs)
+                if getattr(self, '_internal_call', None) else
+                _func_proplot(self, *args, **kwargs)
+            )
+        return super().__new__(cls, name, bases, dct)
 
 
-class Axes(maxes.Axes):
+class Axes(maxes.Axes, metaclass=_MetaAxes):
     """
     Lowest-level axes subclass. Handles titles and axis
     sharing. Adds several new methods and overrides existing ones.