Allow string for offset in an AnchoredOffset

dstansby · dstansby · commit d2ab91a522a3 · 2017-01-16T23:41:41.000Z
diff --git a/lib/matplotlib/offsetbox.py b/lib/matplotlib/offsetbox.py
@@ -33,6 +33,7 @@
 from matplotlib import rcParams
 
 from matplotlib import docstring
+from matplotlib.cbook import is_string_like
 
 #from bboximage import BboxImage
 from matplotlib.image import BboxImage
@@ -986,6 +987,19 @@ class AnchoredOffsetbox(OffsetBox):
     """
     zorder = 5  # zorder of the legend
 
+    # Location codes
+    codes = {'upper right': 1,
+             'upper left': 2,
+             'lower left': 3,
+             'lower right': 4,
+             'right': 5,
+             'center left': 6,
+             'center right': 7,
+             'lower center': 8,
+             'upper center': 9,
+             'center': 10,
+             }
+
     def __init__(self, loc,
                  pad=0.4, borderpad=0.5,
                  child=None, prop=None, frameon=True,
@@ -1028,6 +1042,14 @@ def __init__(self, loc,
         self.set_bbox_to_anchor(bbox_to_anchor, bbox_transform)
         self.set_child(child)
 
+        if is_string_like(loc):
+            if loc not in self.codes:
+                raise RuntimeError('Unrecognized location "%s". Valid '
+                                   'locations are\n\t%s\n'
+                                   % (loc, '\n\t'.join(self.codes)))
+            else:
+                loc = self.codes[loc]
+
         self.loc = loc
         self.borderpad = borderpad
         self.pad = pad
