SF.net SVN: matplotlib:[7103] trunk/matplotlib/lib/matplotlib/quiver.py

[<ef...@us...>]

Revision: 7103
          http://matplotlib.svn.sourceforge.net/matplotlib/?rev=7103&view=rev
Author:   efiring
Date:     2009-05-16 01:01:49 +0000 (Sat, 16 May 2009)

Log Message:
-----------
Fixed bugs in quiver affecting angles passed in as a sequence

Modified Paths:
--------------
    trunk/matplotlib/lib/matplotlib/quiver.py

Modified: trunk/matplotlib/lib/matplotlib/quiver.py
===================================================================
--- trunk/matplotlib/lib/matplotlib/quiver.py	2009-05-14 06:27:58 UTC (rev 7102)
+++ trunk/matplotlib/lib/matplotlib/quiver.py	2009-05-16 01:01:49 UTC (rev 7103)
@@ -479,11 +479,12 @@
         length = a/(self.scale*self.width)
         X, Y = self._h_arrows(length)
         if self.angles == 'xy':
-            theta = self._angles(U, V).filled(0)[:,np.newaxis]
+            theta = self._angles(U, V).filled(0)
         elif self.angles == 'uv':
-            theta = np.angle(ma.asarray(uv[..., np.newaxis]).filled(0))
+            theta = np.angle(uv.filled(0))
         else:
-            theta = ma.asarray(self.angles*np.pi/180.0).filled(0)
+            theta = ma.asarray(self.angles).filled(0)*np.pi/180.0
+        theta.shape = (theta.shape[0], 1) # for broadcasting
         xy = (X+Y*1j) * np.exp(1j*theta)*self.width
         xy = xy[:,:,np.newaxis]
         XY = ma.concatenate((xy.real, xy.imag), axis=2)


This was sent by the SourceForge.net collaborative development platform, the world's largest Open Source development site.