Changeset 2347 in lmdz_wrf

Timestamp:

Feb 18, 2019, 4:48:28 PM (6 years ago)

Author:

lfita

Message:

Adding on `compute_slices_stats_areaweighted' 2D lon,lat variables

File:

• trunk/tools/nc_var_tools.py (modified) (4 diffs)

trunk/tools/nc_var_tools.py

@@ -11308 +11308 @@
  
             newvar = onewnc.createVariable(ivdim, refvdimv.dtype, (refvdimv.dimensions))
+            print 'Lluis newvar:', newvar
             varattrs = refvdimv.ncattrs()
             newvar[:] = refvdimvals
             for attrn in varattrs:
                 attrv = refvdimv.getncattr(attrn)
-                newattr = set_attribute(newvar, attrn, attrv)
+                if attrn != '_FillValue':
+                    newattr = set_attribute(newvar, attrn, attrv)
 
 # Variable
@@ -27723 +27725 @@
     iid = 0
     for idn in dimv:
-        if gen.searchInlist(dimns,idn): slicevar.append(dimvs[idn])
+        if gen.searchInlist(dimns,idn): slicevar.append(dimvs[iid])
         else:
             ldim = ovar.shape[iid]
@@ -27865 +27867 @@
     print '    slicespacedim:', slicespacedim, 'sdimy, sdimx:', sdimy, sdimx
 
-    olon1D = onc.variables[gen.dictionary_key(dimvars, slicespacedim[1])] 
-    olat1D = onc.variables[gen.dictionary_key(dimvars, slicespacedim[0])] 
+    print 'Lluis: dimvars ________'
+    gen.printing_dictionary(dimvars)
+    print 'slicespacedim:', slicespacedim
+    #olon1D = onc.variables[gen.dictionary_key(dimvars, slicespacedim[1])] 
+    #olat1D = onc.variables[gen.dictionary_key(dimvars, slicespacedim[0])] 
+    olon1D = onc.variables[dimvars[slicespacedim[1]]]
+    olat1D = onc.variables[dimvars[slicespacedim[0]]]
+
     # Re-shaping them in case of 1D dimensions...
     if len(olon1D.shape) == 1 and len(olat1D.shape) == 1:
@@ -27913 +27921 @@
         new2Dvars = {slicespacedim[1]: olon2D, slicespacedim[0]: olat2D,             \
           slicespacedim[1]+'_bnds': olonbnds2D, slicespacedim[0]+'_bnds': olatbnds2D}
+        slicebndsvar[slicespacedim[1]] = [slicespacedim[1]+'_bnds',                  \
+          slicespacedim[0]+'_bnds']
+        slicebndsvar[slicespacedim[0]] = [slicespacedim[1]+'_bnds',                  \
+          slicespacedim[0]+'_bnds']
+    else:
+        lon2D, lat2D = gen.lonlat2D(olon1D[:], olat1D[:])
+        olonbnds1D = onc.variables[slicebndsdim[slicespacedim[1]]] 
+        olatbnds1D = onc.variables[slicebndsdim[slicespacedim[0]]] 
+
+        new2Dvars = {slicespacedim[1]: olon1D, slicespacedim[0]: olat1D,             \
+          slicespacedim[1]+'_bnds': olonbnds1D, slicespacedim[0]+'_bnds': olatbnds1D}
         slicebndsvar[slicespacedim[1]] = [slicespacedim[1]+'_bnds',                  \
           slicespacedim[0]+'_bnds']