ENH : add look-up dictionary for step functions

tacaswell · tacaswell · commit 024fc291f8bc · 2015-07-16T00:04:08.000-04:00
Add dict to cbook to map strings -&gt; functions
diff --git a/lib/matplotlib/cbook.py b/lib/matplotlib/cbook.py
@@ -2551,6 +2551,12 @@ def pts_to_midstep(x, *args):
     # convert 2D array back to tuple
     return tuple(steps)
 
+STEP_LOOKUP_MAP = {'pre': pts_to_prestep,
+                   'post': pts_to_poststep,
+                   'mid': pts_to_midstep,
+                   'step-pre': pts_to_prestep,
+                   'step-post': pts_to_poststep,
+                   'step-mid': pts_to_midstep}
 
 # Numpy > 1.6.x deprecates putmask in favor of the new copyto.
 # So long as we support versions 1.6.x and less, we need the
