SF.net SVN: matplotlib: [3887] trunk/toolkits/basemap/lib/matplotlib/ toolkits/basemap/basemap.py

[<js...@us...>]

Revision: 3887
          http://matplotlib.svn.sourceforge.net/matplotlib/?rev=3887&view=rev
Author:   jswhit
Date:     2007-09-25 05:14:28 -0700 (Tue, 25 Sep 2007)

Log Message:
-----------
fix addcyclic so it can handle masked arrays

Modified Paths:
--------------
    trunk/toolkits/basemap/lib/matplotlib/toolkits/basemap/basemap.py

Modified: trunk/toolkits/basemap/lib/matplotlib/toolkits/basemap/basemap.py
===================================================================
--- trunk/toolkits/basemap/lib/matplotlib/toolkits/basemap/basemap.py	2007-09-24 17:33:03 UTC (rev 3886)
+++ trunk/toolkits/basemap/lib/matplotlib/toolkits/basemap/basemap.py	2007-09-25 12:14:28 UTC (rev 3887)
@@ -2917,10 +2917,16 @@
     """
     nlats = arrin.shape[0]
     nlons = arrin.shape[1]
-    arrout  = NX.zeros((nlats,nlons+1),arrin.dtype)
+    if hasattr(arrin,'mask'):
+        arrout  = ma.zeros((nlats,nlons+1),arrin.dtype)
+    else:
+        arrout  = NX.zeros((nlats,nlons+1),arrin.dtype)
     arrout[:,0:nlons] = arrin[:,:]
     arrout[:,nlons] = arrin[:,0]
-    lonsout = NX.zeros(nlons+1,lonsin.dtype)
+    if hasattr(lonsin,'mask'):
+        lonsout = ma.zeros(nlons+1,lonsin.dtype)
+    else:
+        lonsout = NX.zeros(nlons+1,lonsin.dtype)
     lonsout[0:nlons] = lonsin[:]
     lonsout[nlons]  = lonsin[-1] + lonsin[1]-lonsin[0]
     return arrout,lonsout


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