EMH: add ability to return just data from _plot_var_args

tacaswell · tacaswell · commit f69580e1cfa7 · 2024-10-24T15:46:48.000-04:00
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -214,6 +214,7 @@ class _process_plot_var_args:
     """
 
     def __init__(self, command='plot'):
+        _api.check_in_list(['plot', 'fill', 'mirror'], command=command)
         self.command = command
         self.set_prop_cycle(None)
 
@@ -335,6 +336,11 @@ def _makeline(self, axes, x, y, kw, kwargs):
         seg = mlines.Line2D(x, y, **kw)
         return seg, kw
 
+    def _mirror(self, axes, x, y, kw, kwargs):
+        kw = {**kw, **kwargs}  # Don't modify the original kw.
+        self._setdefaults(self._getdefaults(kw), kw)
+        return (x, y), kw
+
     def _makefill(self, axes, x, y, kw, kwargs):
         # Polygon doesn't directly support unitized inputs.
         x = axes.convert_xunits(x)
@@ -495,9 +501,13 @@ def _plot_args(self, axes, tup, kwargs, *,
 
         if self.command == 'plot':
             make_artist = self._makeline
-        else:
+        elif self.command == 'fill':
             kw['closed'] = kwargs.get('closed', True)
             make_artist = self._makefill
+        elif self.command == 'mirror':
+            make_artist = self._mirror
+        else:
+            _api.check_in_list(['plot', 'fill', 'mirror'], command=self.command)
 
         ncx, ncy = x.shape[1], y.shape[1]
         if ncx > 1 and ncy > 1 and ncx != ncy:
