Changeset 909 in lmdz_wrf for trunk/tools

Timestamp:

Jun 20, 2016, 1:12:53 PM (8 years ago)

Author:

lfita

Message:

Adding:

• de-staggering of the variable
• come-back to the WRF dimension names

File:

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

trunk/tools/nc_var_tools.py

@@ -15987 +15987 @@
     newWRFdims = ['Time', 'pres', 'south_north', 'west_east']
     notCHK = ['WRFght', 'WRFt', 'WRFrh']
+    CFdims = ['time', 'pres', 'lat', 'lon']
 
     varns = gen.str_list(variables, ',')
@@ -16009 +16010 @@
     # looking for WRF required variables
     WRFvarrequired = ['P', 'PB', 'PSFC', 'PH', 'PHB', 'HGT', 'T', 'QVAPOR', 'XLONG', \
-      'XLAT', 'Times' , 'P_TOP']
+      'XLAT', 'Times']
     for var in WRFvarrequired:
         if not gen.searchInlist(onc.variables, var):
@@ -16058 +16059 @@
     temp = ((ovar1[:]+300.)*(pres[:]/p0)**RCP).astype('float64')
 
+    WRFdimvs = {'Time':dimt, 'bottom_top':dimz, 'south_north':dimy, 'west_east': dimx}
+
     onewnc = NetCDFFile(ofile, 'w')
 # Creation of dimensions
@@ -16070 +16073 @@
         ovar = onc.variables[var]
         varinf = variable_inf(ovar)
+        newvard = []
         for vd in varinf.dimns:
-            if not gen.searchInlist(onewnc.dimensions, vd):
+            if not gen.searchInlist(onewnc.dimensions, vd) and                       \
+              not gen.searchInlist(CFdims, vd):
                 newdim = onewnc.createDimension(vd, len(onc.dimensions[vd]))
+            if gen.searchInlist(CFdims,vd):
+                icfdim = CFdims.index(vd)
+                newvard.append(WRFdims[icfdim])
+            else:
+                newvard.append(vd)
+
         newvar = onewnc.createVariable(var, nctype(varinf.dtype),                    \
-          tuple(varinf.dimns))
+          tuple(newvard))
         newvar[:] = ovar[:]
         for attrn in ovar.ncattrs():
@@ -16126 +16137 @@
             varinf = variable_inf(ovarin)
             varin = ovarin[:]
+            # de-staggering
+            if gen.searchInlist(ovarin.dimensions, 'west_east'):
+                varint = gen.stagger_unstagger(varin,list(ovarin.dimensions),WRFdimvs)
+            else:
+                CFdimvs = {'time': dimt, 'bottom_top': dimz, 'lat': dimy, 'lon': dimx}
+                varint = gen.stagger_unstagger(varin,list(ovarin.dimensions),CFdimvs)
             isgeop = False
+            if vn == 'z': isgeop = True
             varattrs = gen.variables_values(vn)
             CFvn = varattrs[0]
@@ -16143 +16161 @@
             tempt = temp.transpose()
             qvt = qv.transpose()
+            print fname + ' Lluis shapes varint:', varint.shape, 'prest:', prest.shape, \
+              'psfct:', psfct.shape, 'hgtt:', hgtt.shape,' tempt:', tempt.shape, 'qvt:',qvt.shape
             varinterp = fin.module_forinterpolate.interp( data_in=varint,            \
               pres_field=prest, interp_levels=interplevs, psfc=psfct,                \