resize transforms to feat at least the sizes/angles length

mgoacolou · mgoacolou · commit ff5d36e84968 · 2014-02-26T13:52:55.000+01:00
diff --git a/lib/matplotlib/collections.py b/lib/matplotlib/collections.py
@@ -709,6 +709,9 @@ class _CollectionWithSizes(Collection):
     """
     Base class for collections that have an array of sizes.
     """
+    def __init__(self):
+        self._sizes = np.array([])
+        
     def get_sizes(self):
         return self._sizes
 
@@ -745,6 +748,9 @@ def set_sizes(self, sizes, dpi=72.0):
                 s[:, 0, 0] = scale
                 s[:, 1, 1] = scale
                 s[:, 2, 2] = 1.0
+                if self._transforms.shape[0] < len(self._sizes):
+                    # resize transforms to feat at least the sizes length
+                    self._transforms = np.resize(self._transforms,(len(self._sizes),3,3))
                 for i in xrange(self._transforms.shape[0]):
                     self._transforms[i,:,:] = np.dot(s[i%len(self._sizes),:,:], self._transforms[i,:,:])
 
@@ -755,6 +761,9 @@ class _CollectionWithAngles(Collection):
     """
     Base class for collections that have an array of angles.
     """
+    def __init__(self):
+        self._angles = np.array([])
+
     def get_angles(self):
         return self._angles
 
@@ -799,6 +808,9 @@ def set_angles(self, angles):
                 r[:, 1, 1] = rot_c
                 r[:, 1, 0] = rot_s
                 r[:, 2, 2] = 1.0
+                if self._transforms.shape[0] < len(self._angles):
+                    # resize transforms to feat at least the angles length
+                    self._transforms = np.resize(self._transforms,(len(self._angles),3,3))
                 for i in xrange(self._transforms.shape[0]):
                     self._transforms[i,:,:] = np.dot(r[i%len(self._angles),:,:], self._transforms[i,:,:])
 
