add ArrowStyle |-|

leejjoon · leejjoon · commit 7b2a3ecee3a7 · 2009-12-11T16:57:53.000Z
svn path=/trunk/matplotlib/; revision=8023
diff --git a/examples/pylab_examples/fancyarrow_demo.py b/examples/pylab_examples/fancyarrow_demo.py
@@ -4,7 +4,7 @@
 styles = mpatches.ArrowStyle.get_styles()
 
 ncol=2
-nrow = len(styles) // ncol + 1
+nrow = (len(styles)+1) // ncol
 figheight = (nrow+0.5)
 fig1 = plt.figure(1, (4.*ncol/1.5, figheight/1.5))
 fontsize = 0.2 * 70
@@ -15,13 +15,19 @@
 ax.set_xlim(0, 4*ncol)
 ax.set_ylim(0, figheight)
 
+def to_texstring(s):
+    s = s.replace("<", r"$<$")
+    s = s.replace(">", r"$>$")
+    s = s.replace("|", r"$|$")
+    return s
+
 for i, (stylename, styleclass) in enumerate(sorted(styles.items())):
     x = 3.2 + (i//nrow)*4
     y = (figheight - 0.7 - i%nrow) # /figheight
     p = mpatches.Circle((x, y), 0.2, fc="w")
     ax.add_patch(p)
 
-    ax.annotate(stylename, (x, y),
+    ax.annotate(to_texstring(stylename), (x, y),
                 (x-1.2, y),
                 #xycoords="figure fraction", textcoords="figure fraction",
                 ha="right", va="center",
diff --git a/lib/matplotlib/patches.py b/lib/matplotlib/patches.py
@@ -3299,6 +3299,41 @@ def __init__(self, widthB=1., lengthB=0.2, angleB=None):
     _style_list["-["] = BracketB
 
 
+    class BarAB(_Bracket):
+        """
+        An arrow with a bracket(])  at both ends.
+        """
+
+        def __init__(self,
+                     widthA=1., angleA=None,
+                     widthB=1., angleB=None):
+            """
+            *widthA*
+              width of the bracket
+
+            *lengthA*
+              length of the bracket
+
+            *angleA*
+              angle between the bracket and the line
+
+            *widthB*
+              width of the bracket
+
+            *lengthB*
+              length of the bracket
+
+            *angleB*
+              angle between the bracket and the line
+            """
+
+            super(ArrowStyle.BarAB, self).__init__(True, True, \
+                        widthA=widthA, lengthA=0, angleA=angleA,
+                        widthB=widthB, lengthB=0, angleB=angleB)
+
+    _style_list["|-|"] = BarAB
+
+
     class Simple(_Base):
         """
         A simple arrow. Only works with a quadratic bezier curve.
