Changeset 1050 in lmdz_wrf for trunk

Timestamp:

Aug 27, 2016, 8:57:29 AM (9 years ago)

Author:

lfita

Message:

Generalizing 'pinterp' for multiple input files. Version only for 'WRF' input

Location:

trunk/tools

Files:

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

trunk/tools/nc_var.py

@@ -17 +17 @@
 ## e.g. # nc_var.py -o cleaning_varsfile -f ~/PY/wrfout_d01_2001-11-11_00:00:00 -S T2,XLONG,XLAT,Times
 ## 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
 
 from optparse import OptionParser

trunk/tools/nc_var_tools.py

@@ -16186 +16186 @@
     ofile = 'pinterp.nc'
 
+    CFdims = ['time', 'pres', 'lat', 'lon']
+
     onc = NetCDFFile(ncfile, 'r')
 
-    # Variables to interpolate
-    WRFdims = ['Time', 'bottom_top', 'south_north', 'west_east']
-    newWRFdims = ['Time', 'pres', 'south_north', 'west_east']
-    notCHK = ['WRFght', 'WRFt', 'WRFrh']
-    CFdims = ['time', 'pres', 'lat', 'lon']
+    # Gessing orgin of the file
+    dimsinfile = onc.dimensions.keys()
+    if gen.searchInlist(dimsinfile,'bottom_top'):
+        modname = 'WRF'
+        print warnmsg
+        print '  ' + fname + ": gessing that file '" + ncfile + "' comes from '" +   \
+          modname + "' !!"
+
+        # Variables to interpolate
+        MODdims = ['Time', 'bottom_top', 'south_north', 'west_east']
+        newMODdims = ['Time', 'pres', 'south_north', 'west_east']
+        notCHK = ['WRFght', 'WRFt', 'WRFrh']
+        MODvarrequired = ['P', 'PB', 'PSFC', 'PH', 'PHB', 'HGT', 'T', 'QVAPOR',      \
+          'XLONG', 'XLAT', 'Times']
+        MODvardims = ['XLONG', 'XLAT', 'Times']
 
     varns = gen.str_list(variables, ',')
@@ -16201 +16213 @@
             coincdims = set(WRFdims) & set(ovn.dimensions)
            
-            if len(WRFdims) == len(ovn.dimensions) and len(coincdims) == len(WRFdims):
+            if len(MODdims) == len(ovn.dimensions) and len(coincdims) == len(MODdims):
                 varns.append(vn)
         varns = varns + notCHK
@@ -16209 +16221 @@
           not gen.searchInlist(notCHK, vn):
             print errormsg
-            print '  ' + fname + "': WRF file '" + ncfile +                          \
-              "' does not have variable '" + vn + "' !!"
+            print '  ' + fname + "': file '" + ncfile + "' does not have variable '"+\
+              vn + "' !!"
             quit(-1)
 
-    # looking for WRF required variables
-    WRFvarrequired = ['P', 'PB', 'PSFC', 'PH', 'PHB', 'HGT', 'T', 'QVAPOR', 'XLONG', \
-      'XLAT', 'Times']
-    for var in WRFvarrequired:
+    # looking for model required variables
+    for var in MODvarrequired:
         if not gen.searchInlist(onc.variables, var):
             print errormsg
@@ -16225 +16235 @@
 # Computing necessary variables
     # pressure
-    ovar1 = onc.variables['P']
-    ovar2 = onc.variables['PB']
-
-    pres = (ovar1[:] + ovar2[:]).astype('float64')
-
-    dimx = pres.shape[3]
-    dimy = pres.shape[2]
-    dimz = pres.shape[1]
-    dimt = pres.shape[0]
+    if modname == 'WRF':
+        ovar1 = onc.variables['P']
+        ovar2 = onc.variables['PB']
+
+        pres = (ovar1[:] + ovar2[:]).astype('float64')
+
+        dimx = pres.shape[3]
+        dimy = pres.shape[2]
+        dimz = pres.shape[1]
+        dimt = pres.shape[0]
+        MODdimvs = {'Time':dimt, 'bottom_top':dimz, 'south_north':dimy,              \
+          'west_east': dimx}
+        CFdimvs = {'time': dimt, 'bottom_top': dimz, 'lat': dimy, 'lon': dimx}
+        unstaggerDIM = 'west_east'
 
     # sfc pressure
-    ovar1 = onc.variables['PSFC']
-    psfc = ovar1[:].astype('float64')
+    if modname == 'WRF':
+        ovar1 = onc.variables['PSFC']
+        psfc = ovar1[:].astype('float64')
 
     # geopotential height
-    ovar1 = onc.variables['PH']
-    ovar2 = onc.variables['PHB']
-    geop0 = ovar1[:] + ovar2[:]
-    unstg = list(geop0.shape)
-    unstg[1] = unstg[1] - 1
-    geop = np.zeros(tuple(unstg), dtype=np.float)
-    geop = 0.5*(geop0[:,1:dimz,:,:] + geop0[:,0:dimz-1,:,:]).astype('float64')
+    if modname == 'WRF':
+        ovar1 = onc.variables['PH']
+        ovar2 = onc.variables['PHB']
+        geop0 = ovar1[:] + ovar2[:]
+        unstg = list(geop0.shape)
+        unstg[1] = unstg[1] - 1
+        geop = np.zeros(tuple(unstg), dtype=np.float)
+        geop = 0.5*(geop0[:,1:dimz,:,:] + geop0[:,0:dimz-1,:,:]).astype('float64')
 
     # terrain height
-    ovar1 = onc.variables['HGT']
-    hgt = ovar1[0,:,:].astype('float64')
-
-    # water vapour micing ratio
-    ovar1 = onc.variables['QVAPOR']
-    qv = ovar1[:].astype('float64')
+    if modname == 'WRF':
+        ovar1 = onc.variables['HGT']
+        hgt = ovar1[0,:,:].astype('float64')
+
+    # water vapour mixing ratio
+    if modname == 'WRF':
+        ovar1 = onc.variables['QVAPOR']
+        qv = ovar1[:].astype('float64')
 
     # temperature
-    Rd = 287.04
-    Cp = 7.*Rd/2.
-    RCP = Rd/Cp
-    p0 = 100000.
-    ovar1 = onc.variables['T']
-    temp = ((ovar1[:]+300.)*(pres[:]/p0)**RCP).astype('float64')
-
-    WRFdimvs = {'Time':dimt, 'bottom_top':dimz, 'south_north':dimy, 'west_east': dimx}
+    if modname == 'WRF':
+        Rd = 287.04
+        Cp = 7.*Rd/2.
+        RCP = Rd/Cp
+        p0 = 100000.
+        ovar1 = onc.variables['T']
+        temp = ((ovar1[:]+300.)*(pres[:]/p0)**RCP).astype('float64')
 
     onewnc = NetCDFFile(ofile, 'w')
 # Creation of dimensions
-    newdim = onewnc.createDimension('west_east', dimx)
-    newdim = onewnc.createDimension('south_north', dimy)
-    newdim = onewnc.createDimension('pres', len(interplevs))
-    newdim = onewnc.createDimension('Time', None)
+    if modname == 'WRF':
+        newdim = onewnc.createDimension('west_east', dimx)
+        newdim = onewnc.createDimension('south_north', dimy)
+        newdim = onewnc.createDimension('pres', len(interplevs))
+        newdim = onewnc.createDimension('Time', None)
 
 # Creation of variable dimensions
-    WRFvardims = ['XLONG', 'XLAT', 'Times']
-    for var in WRFvardims:
+    for var in MODvardims:
         ovar = onc.variables[var]
         varinf = variable_inf(ovar)
@@ -16285 +16303 @@
             if gen.searchInlist(CFdims,vd):
                 icfdim = CFdims.index(vd)
-                newvard.append(WRFdims[icfdim])
+                newvard.append(MODdims[icfdim])
             else:
                 newvard.append(vd)
@@ -16338 +16356 @@
                 newvarattr['long_name'] = varattrs[4].replace('|',' ')
                 newvarattr['units'] = varattrs[5]
-        elif not gen.searchInlist(WRFvarrequired, vn) or vn == 'QVAPOR':
+        elif not gen.searchInlist(MODvarrequired, vn) or vn == 'QVAPOR':
             ovarin = onc.variables[vn]
             varinf = variable_inf(ovarin)
             varin = ovarin[:]
             # de-staggering
-            if gen.searchInlist(ovarin.dimensions, 'west_east'):
-                varint = gen.stagger_unstagger(varin,list(ovarin.dimensions),WRFdimvs)
+            if gen.searchInlist(ovarin.dimensions, unstaggerDIM):
+                varint = gen.stagger_unstagger(varin,list(ovarin.dimensions),MODdimvs)
             else:
-                CFdimvs = {'time': dimt, 'bottom_top': dimz, 'lat': dimy, 'lon': dimx}
                 varint = gen.stagger_unstagger(varin,list(ovarin.dimensions),CFdimvs)
             isgeop = False
@@ -16371 +16388 @@
               extrapolate=extrap, geopt=isgeop, missing=gen.fillValueF)
 
-            newvar = onewnc.createVariable(CFvn, 'f4', tuple(newWRFdims),            \
+            newvar = onewnc.createVariable(CFvn, 'f4', tuple(newMODdims),            \
               fill_value=gen.fillValueF)
             newvar[:] = varinterp.transpose()