Changeset 1053 in lmdz_wrf

Timestamp:

Aug 27, 2016, 9:34:32 AM (9 years ago)

Author:

lfita

Message:

Adding `LMDZ' model in 'pinterp'

Location:

trunk/tools

Files:

• nc_var.py (modified) (1 diff)
• nc_var_tools.py (modified) (10 diffs)

trunk/tools/nc_var.py

@@ -18 +18 @@
 ## e.g. # nc_var.py -o LMDZ_toCF -f LMDZ/AR40/vas_histins_1-1.nc
 ## e.g. # nc_var.py -o pinterp -f wrfout_d01_2001-11-11_00\:00\:00 -S 100000.:97500.:95000.:92500.:90000.:85000.:80000.:75000.:70000.:65000.:60000.:55000.:50000.:45000.:40000.:35000.:30000.:25000.:20000.:15000.:10000.:5000.:2500.:1000.:500.:250.,1,0 -v WRFt,WRFrh
+## e.g. # nc_var.py -o pinterp -f /media/ExtDiskC_ext4/DATA/etudes/WRF_LMDZ/WaquaL_highres/short_copies/LMDZ/AR40/histins_19790101000000-19790304000000_short.nc -S 100000.:97500.:95000.:92500.:90000.:85000.:80000.:75000.:70000.:65000.:60000.:55000.:50000.:45000.:40000.:35000.:30000.:25000.:20000.:15000.:10000.:5000.:2500.:1000.:500.:250.,1,0 -v temp,ovap
 
 from optparse import OptionParser

trunk/tools/nc_var_tools.py

@@ -16187 +16187 @@
 
     CFdims = ['time', 'pres', 'lat', 'lon']
+    # Just in case there are weird units, attribute value will be taken from input file
+    CFattrs = ['standard_name', 'long_name']
 
     onc = NetCDFFile(ncfile, 'r')
@@ -16205 +16207 @@
           'XLONG', 'XLAT', 'Times']
         MODvardims = ['XLONG', 'XLAT', 'Times']
+    elif gen.searchInlist(dimsinfile,'presnivs'):
+        modname = 'LMDZ'
+        print warnmsg
+        print '  ' + fname + ": gessing that file '" + ncfile + "' comes from '" +   \
+          modname + "' !!"
+
+        # Variables to interpolate
+        MODdims = ['time_counter', 'presnivs', 'lat', 'lon']
+        newMODdims = ['time_counter', 'pres', 'lat', 'lon']
+        notCHK = []
+        MODvarrequired = ['pres', 'psol', 'geop', 'phis', 'temp', 'ovap', 'lon',     \
+          'lat', 'time_counter']
+        MODvardims = ['lon', 'lat', 'time_counter']
 
     varns = gen.str_list(variables, ',')
@@ -16229 +16244 @@
         if not gen.searchInlist(onc.variables, var):
             print errormsg
-            print '  ' + fname + ": file '" + ncfile + "' does not have variable '" +\
-              var + "' !!"
+            print '  ' + fname + ": file '" + ncfile + "' does not have required " + \
+              "variable '" + var + "' !!"
             quit(-1)
 
@@ -16249 +16264 @@
         CFdimvs = {'time': dimt, 'bottom_top': dimz, 'lat': dimy, 'lon': dimx}
         unstaggerDIM = 'west_east'
+    elif modname == 'LMDZ':
+        ovar1 = onc.variables['pres']
+        pres = ovar1[:].astype('float64')
+        dimx = pres.shape[3]
+        dimy = pres.shape[2]
+        dimz = pres.shape[1]
+        dimt = pres.shape[0]
+        MODdimvs = {'time_counter':dimt, 'presnivs':dimz, 'lat':dimy, 'lon': dimx}
+        CFdimvs = {'time': dimt, 'presnivs': dimz, 'lat': dimy, 'lon': dimx}
+        unstaggerDIM = 'lon'
 
     # sfc pressure
     if modname == 'WRF':
         ovar1 = onc.variables['PSFC']
+        psfc = ovar1[:].astype('float64')
+    elif modname == 'LMDZ':
+        ovar1 = onc.variables['psol']
         psfc = ovar1[:].astype('float64')
 
@@ -16264 +16292 @@
         geop = np.zeros(tuple(unstg), dtype=np.float)
         geop = 0.5*(geop0[:,1:dimz,:,:] + geop0[:,0:dimz-1,:,:]).astype('float64')
+    elif modname == 'LMDZ':
+        ovar1 = onc.variables['geop']
+        geop = ovar1[:].astype('float64')
 
     # terrain height
@@ -16269 +16300 @@
         ovar1 = onc.variables['HGT']
         hgt = ovar1[0,:,:].astype('float64')
+    elif modname == 'LMDZ':
+        grav = 9.81
+        ovar1 = onc.variables['phis']
+        hgt = (ovar1[0,:,:]/grav).astype('float64')
 
     # water vapour mixing ratio
     if modname == 'WRF':
         ovar1 = onc.variables['QVAPOR']
+        qv = ovar1[:].astype('float64')
+    elif modname == 'LMDZ':
+        ovar1 = onc.variables['ovap']
         qv = ovar1[:].astype('float64')
 
@@ -16283 +16321 @@
         ovar1 = onc.variables['T']
         temp = ((ovar1[:]+300.)*(pres[:]/p0)**RCP).astype('float64')
+    elif modname == 'LMDZ':
+        ovar1 = onc.variables['temp']
+        temp = ovar1[:].astype('float64')
 
     onewnc = NetCDFFile(ofile, 'w')
@@ -16291 +16332 @@
         newdim = onewnc.createDimension('pres', len(interplevs))
         newdim = onewnc.createDimension('Time', None)
+    elif modname == 'LMDZ':
+        newdim = onewnc.createDimension('lon', dimx)
+        newdim = onewnc.createDimension('lat', dimy)
+        newdim = onewnc.createDimension('pres', len(interplevs))
+        newdim = onewnc.createDimension('time_counter', None)
 
 # Creation of variable dimensions
@@ -16356 +16402 @@
                 newvarattr['long_name'] = varattrs[4].replace('|',' ')
                 newvarattr['units'] = varattrs[5]
-        elif not gen.searchInlist(MODvarrequired, vn) or vn == 'QVAPOR':
+            elif vn == 'pres':
+                varin = pres
+                isgeop = False
+                varattrs = gen.variables_values('pres')
+                CFvn = varattrs[0]
+                newvarattr['standard_name'] = varattrs[1]
+                newvarattr['long_name'] = varattrs[4].replace('|',' ')
+                newvarattr['units'] = varattrs[5]
+#        elif not gen.searchInlist(MODvarrequired, vn) or vn == 'QVAPOR':
+        else:
             ovarin = onc.variables[vn]
             varinf = variable_inf(ovarin)
@@ -16393 +16448 @@
 
             for attrn in newvarattr.keys():
-                attr = set_attribute(newvar, attrn, newvarattr[attrn])
+                if attrn != '_FillValue' and not gen.searchInlist(CFattrs, attrn):
+                    attr = set_attribute(newvar, attrn, newvarattr[attrn])
             onewnc.sync()