Changeset 395

Timestamp:

Nov 17, 2011, 1:08:01 PM (14 years ago)

Author:

acolaitis

Message:

PYTHON

M 393 mymath.py
^{----------------- Cosmetic change}

M 393 make_netcdf.py
^{----------------- Cosmetic change}

M 393 planetoplot.py
^{----------------- corrected bug with varname and --tsat}

M 393 myplot.py
^{----------------- added possibility for the script to read ncdf files with NaN value and treat them correctly without messing with the mathematical operations.}

Location:

trunk/UTIL/PYTHON

Files:

• make_netcdf.py (modified) (2 diffs)
• mymath.py (modified) (1 diff)
• myplot.py (modified) (3 diffs)
• planetoplot.py (modified) (1 diff)

trunk/UTIL/PYTHON/make_netcdf.py

@@ -89 +89 @@
        zvarcarac[zznames[i]]=zzvardim
        i=i+1
+
+   print "creating "+zfilename
+
    print zvarcarac  
  
@@ -139 +142 @@
    file.close()
 
+   print "closing "+zfilename
+
+   return
+
 def find_key(dic, val):
     """return the key of dictionary dic given the value"""

trunk/UTIL/PYTHON/mymath.py

@@ -112 +112 @@
     m=np.ma.masked_invalid(temp,copy=False)
     zoutput=np.ma.array(output,mask=m.mask,fill_value=np.NaN)
-#    zout=zoutput.filled()
     return zoutput
 

trunk/UTIL/PYTHON/myplot.py

@@ -49 +49 @@
 def getfield (nc,var):
     ## this allows to get much faster (than simply referring to nc.variables[var])
+    import numpy as np
     dimension = len(nc.variables[var].dimensions)
     #print "   Opening variable with", dimension, "dimensions ..."
@@ -54 +55 @@
     elif dimension == 3:  field = nc.variables[var][:,:,:]
     elif dimension == 4:  field = nc.variables[var][:,:,:,:]
-    return field
+    # if there are NaNs in the ncdf, they should be loaded as a masked array which will be
+    # recasted as a regular array later in reducefield
+    if (np.isnan(np.sum(field)) and (type(field).__name__ not in 'MaskedArray')):
+       print "Warning: netcdf as nan values but is not loaded as a Masked Array."
+       print "recasting array type"
+       out=np.ma.masked_invalid(field)
+       out.set_fill_value([np.NaN])
+    else:
+       out=field
+    return out
 
 ## Author: AS + TN + AC
@@ -492 +502 @@
                               llcrnrlat=wlat[0],urcrnrlat=wlat[1],llcrnrlon=wlon[0],urcrnrlon=wlon[1])
     elif char == "npstere": m = Basemap(rsphere=radius,projection='npstere', boundinglat=blat, lon_0=0.)
-    elif char == "spstere": m = Basemap(rsphere=radius,projection='spstere', boundinglat=blat, lon_0=0.)
+    elif char == "spstere": m = Basemap(rsphere=radius,projection='spstere', boundinglat=blat, lon_0=180.)
     elif char == "nplaea":  m = Basemap(rsphere=radius,projection='nplaea', boundinglat=wlat[0], lon_0=meanlon)
     elif char == "laea":    m = Basemap(rsphere=radius,projection='laea',lon_0=meanlon,lat_0=meanlat,lat_ts=meanlat,\

trunk/UTIL/PYTHON/planetoplot.py

@@ -183 +183 @@
           else:                                 tinput=tt
           all_var[k]=get_tsat(vert,tinput,zlon=lon,zlat=lat,zalt=vert,ztime=time)
+          all_varname[k] = varname
       else:
       ##### GENERAL STUFF HERE