Changeset 1055 in lmdz_wrf for trunk/tools/nc_var_tools.py

Timestamp:

Aug 27, 2016, 10:43:17 AM (8 years ago)

Author:

lfita

Message:

Right values for `cfLMDZ' in 'pinterp'

File:

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

trunk/tools/nc_var_tools.py

@@ -16194 +16194 @@
     # Gessing orgin of the file
     dimsinfile = onc.dimensions.keys()
+    varsinfile = onc.variables.keys()
+ 
     if gen.searchInlist(dimsinfile,'bottom_top'):
         modname = 'WRF'
@@ -16220 +16222 @@
           'lat', 'time_counter']
         MODvardims = ['lon', 'lat', 'time_counter']
+    elif gen.searchInlist(dimsinfile,'pres') and gen.searchInlist(varsinfile,        \
+      'psol'):
+        modname = 'cfLMDZ'
+        print warnmsg
+        print '  ' + fname + ": gessing that file '" + ncfile + "' comes from '" +   \
+          modname + "' ('pres, time_counter' variables renamed as 'p, time') !!"
+
+        # Variables to interpolate
+        MODdims = ['time', 'pres', 'lat', 'lon']
+        newMODdims = ['time', 'pres', 'lat', 'lon']
+        notCHK = []
+        MODvarrequired = ['p', 'psol', 'geop', 'phis', 'temp', 'ovap', 'lon',     \
+          'lat', 'time']
+        MODvardims = ['lon', 'lat', 'time']
     else:
         modname = 'CF'
@@ -16286 +16302 @@
         CFdimvs = {'time': dimt, 'presnivs': dimz, 'lat': dimy, 'lon': dimx}
         unstaggerDIM = 'lon'
+    elif modname == 'cfLMDZ':
+        ovar1 = onc.variables['p']
+        pres = ovar1[:].astype('float64')
+        dimx = pres.shape[3]
+        dimy = pres.shape[2]
+        dimz = pres.shape[1]
+        dimt = pres.shape[0]
+        MODdimvs = {'time':dimt, 'pres':dimz, 'lat':dimy, 'lon': dimx}
+        CFdimvs = {'time': dimt, 'pres': dimz, 'lat': dimy, 'lon': dimx}
+        unstaggerDIM = 'lon'
     else:
         ovar1 = onc.variables['p']
@@ -16301 +16327 @@
         ovar1 = onc.variables['PSFC']
         psfc = ovar1[:].astype('float64')
-    elif modname == 'LMDZ':
+    elif modname == 'LMDZ' or modname == 'cfLMDZ':
         ovar1 = onc.variables['psol']
         psfc = ovar1[:].astype('float64')
@@ -16317 +16343 @@
         geop = np.zeros(tuple(unstg), dtype=np.float)
         geop = 0.5*(geop0[:,1:dimz,:,:] + geop0[:,0:dimz-1,:,:]).astype('float64')
-    elif modname == 'LMDZ':
+    elif modname == 'LMDZ' or modname == 'cfLMDZ':
         ovar1 = onc.variables['geop']
         geop = ovar1[:].astype('float64')
@@ -16328 +16354 @@
         ovar1 = onc.variables['HGT']
         hgt = ovar1[0,:,:].astype('float64')
-    elif modname == 'LMDZ':
+    elif modname == 'LMDZ' or modname == 'cfLMDZ':
         grav = 9.81
         ovar1 = onc.variables['phis']
@@ -16340 +16366 @@
         ovar1 = onc.variables['QVAPOR']
         qv = ovar1[:].astype('float64')
-    elif modname == 'LMDZ':
+    elif modname == 'LMDZ' or modname == 'cfLMDZ':
         ovar1 = onc.variables['ovap']
         qv = ovar1[:].astype('float64')
@@ -16355 +16381 @@
         ovar1 = onc.variables['T']
         temp = ((ovar1[:]+300.)*(pres[:]/p0)**RCP).astype('float64')
-    elif modname == 'LMDZ':
+    elif modname == 'LMDZ' or modname == 'cfLMDZ':
         ovar1 = onc.variables['temp']
         temp = ovar1[:].astype('float64')
@@ -16699 +16725 @@
     ncvars = ncf.variables.keys()
 
-#    # CFing dimension-variables
-#    CFvardims = {'time_counter': 'time', 'presnivs': 'pres'}
-#    for varn in CFvardims.keys():
-#        if gen.searchInlist(ncvars,varn):
-#            # Making sure CF vardimension is not in the file
-#            if not gen.searchInlist(ncvars,CFvardims[varn]):
-#                newname = ncf.renameVariable(varn,CFvardims[varn])
-
-#    ncf.sync()
+    # CFing variable 'pres'
+    if gen.searchInlist(ncvars,'pres'):
+        print warnmsg
+        print fname + ": renaming variable 'pres' as 'p' !!"
+        newname = ncf.renameVariable('pres','p')
+    ncf.sync()
 
     # CFing dimensions
@@ -16716 +16739 @@
             if not gen.searchInlist(ncdims,CFdims[dimn]):
                 newname = ncf.renameDimension(dimn,CFdims[dimn])
-
     ncf.sync()
 
@@ -16732 +16754 @@
 
     ncf.sync()
-
     ncf.close()