Add angle kwarg to patches.Rectangle

dmcdougall · dmcdougall · commit f404c9466179 · 2012-11-11T11:00:18.000-06:00
Allows rotation of a rectangle upon instantiation.
diff --git a/lib/matplotlib/patches.py b/lib/matplotlib/patches.py
@@ -539,9 +539,12 @@ def __str__(self):
             + "(%g,%g;%gx%g)" % (self._x, self._y, self._width, self._height)
 
     @docstring.dedent_interpd
-    def __init__(self, xy, width, height, **kwargs):
+    def __init__(self, xy, width, height, angle=0.0, **kwargs):
         """
 
+        *angle*
+          rotation in degrees (anti-clockwise)
+
         *fill* is a boolean indicating whether to fill the rectangle
 
         Valid kwargs are:
@@ -554,6 +557,7 @@ def __init__(self, xy, width, height, **kwargs):
         self._y = xy[1]
         self._width = width
         self._height = height
+        self._angle = angle
         # Note: This cannot be calculated until this is added to an Axes
         self._rect_transform = transforms.IdentityTransform()
 
@@ -574,7 +578,10 @@ def _update_patch_transform(self):
         width = self.convert_xunits(self._width)
         height = self.convert_yunits(self._height)
         bbox = transforms.Bbox.from_bounds(x, y, width, height)
+        rot_trans = transforms.Affine2D()
+        rot_trans.translate(-x, -y).rotate_deg(self._angle).translate(x, y)
         self._rect_transform = transforms.BboxTransformTo(bbox)
+        self._rect_transform += rot_trans
 
     def get_patch_transform(self):
         self._update_patch_transform()
