Changeset 2272 in lmdz_wrf for trunk

Timestamp:

Dec 27, 2018, 10:05:14 PM (7 years ago)

Author:

lfita

Message:

Adding realistic example on `area_weighted'
Adding:

• `compute_WRFbnds': Function to compute vertices of the cells using WRF 'XLONG_[U/V]', 'XLAT_[U/V]'

Location:

trunk/tools

Files:

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

trunk/tools/nc_var.py

@@ -68 +68 @@
 ## e.g. # nc_var.py -o dimrename -f /media/lluis/ExtDiskC_ext3/DATA/estudios/FPS_Alps/additional/IOP/select/out_irene/simcdx_vars_cape_120lev_cdxwrf2.nc -S Time -v time0
 ## e.g. # nc_var.py -o area_weighted -f 'reference_data.nc:lon;lon;lon_bnds;-1;lat;lat;lat_bnds;-1,get_data.nc:lon;lon;lon_bnds;-1;lat;lat;lat_bnds;-1' -S 'yes:min,max,mean,stddev,count' -v ct_values,xband_values,box_values,mosaic_values
+## e.g. # nc_var.py -o area_weighted -f '/media/lluis/ExtDiskC_ext4/bkup/llamp/estudios/dominios/SA150k/geo_em.d01.nc:west_east;XLONG_M;WRFxbnds;-1;south_north;XLAT_M;WRFybnds;-1,/media/lluis/ExtDiskC_ext4/bkup/llamp/estudios/dominios/SA50k/geo_em.d01.nc:west_east;XLONG_M;WRFxbnds;-1;south_north;XLAT_M;WRFybnds;-1' -S 'no:mean' -v HGT_M
 
 from optparse import OptionParser

trunk/tools/nc_var_tools.py

@@ -27179 +27179 @@
     return
 
+def compute_WRFbnds(lonu, latu, lonv, latv):
+    """ Function to compute vertices of the cells using WRF 'XLONG_[U/V]', 'XLAT_[U/V]'
+      lonu: longitudes on the u-staggered matrix
+      latu: latitudes on the u-staggered matrix
+      lonv: longitudes on the v-staggered matrix
+      latv: latitudes on the v-staggered matrix
+    """
+    fname = 'compute_WRFbnds'
+
+    dx = lonv.shape[1]
+    dy = lonu.shape[0]
+    sdx = lonu.shape[1]
+    sdy = lonv.shape[0]
+
+    lonbnds = np.zeros((4, dy, dx), dtype=np.float)
+    latbnds = np.zeros((4, dy, dx), dtype=np.float)
+
+    for j in range(dy-1):
+        lonbnds[0,j,:] = 0.5*(lonu[j,0:sdx-1]+lonu[j+1,0:sdx-1])
+        lonbnds[1,j,:] = 0.5*(lonu[j,1:sdx]+lonu[j+1,1:sdx])
+        lonbnds[2,j,:] = 0.5*(lonu[j-1,1:sdx]+lonu[j,1:sdx])
+        lonbnds[3,j,:] = 0.5*(lonu[j-1,0:sdx-1]+lonu[j,0:sdx-1])
+
+    for i in range(dx-1):
+        latbnds[0,:,i] = 0.5*(latv[1:sdy,i-1]+latv[1:sdy,i])
+        latbnds[1,:,i] = 0.5*(latv[1:sdy,i]+latv[1:sdy,i+1])
+        latbnds[2,:,i] = 0.5*(latv[0:sdy-1,i]+latv[0:sdy-1,i+1])
+        latbnds[3,:,i] = 0.5*(latv[0:sdy-1,i-1]+latv[0:sdy-1,i])
+
+    return lonbnds, latbnds
+
 def area_weighted(values, ncfiles, variable):
     """ Function to provide an area weighted statistics within two different values 
@@ -27204 +27235 @@
             [dimxn]: name of the dimension along the x-axis
             [varxn]: name of the variable with the values of dimxn
-            [varxbndsn]: name of the variable with the bounds of the values of varxn
+            [varxbndsn]: name of the variable with the bounds of the values of varxn 
+              ('WRFxbnds' to be computed from WRF output using XLONG_[U/V], XLAT_[U/V])
             [dimyn]: name of the dimension along the y-axis
             [varyn]: name of the variable with the values of dimyn
             [varybndsn]: name of the variable with the bounds of the values of varyn
+              ('WRFybnds' to be computed from WRF output using XLONG_[U/V], XLAT_[U/V])
             [slicedim]: series of values to define the slicing of the variable
               * [integer]: which value of the dimension
@@ -27309 +27342 @@
                 quit(-1)
 
-            if not oncref.variables.has_key(refvarxbndsn):
+            if refvarxbndsn != 'WRFxbnds' and                                        \
+              not oncref.variables.has_key(refvarxbndsn):
                 print errormsg
                 print '  ' + fname + ": reference file '" + netcdfrefn + "' does " + \
@@ -27336 +27370 @@
                 quit(-1)
 
-            if not oncref.variables.has_key(refvarybndsn):
+            if refvarybndsn != 'WRFybnds' and                                        \
+              not oncref.variables.has_key(refvarybndsn):
                 print errormsg
                 print '  ' + fname + ": reference file '" + netcdfrefn + "' does " + \
@@ -27347 +27382 @@
             slicedict = {refdimxn: slicedimx, refdimyn: slicedimy}
             ovarx = oncref.variables[refvarxn]
-            ovarxbnds = oncref.variables[refvarxbndsn]
             ovary = oncref.variables[refvaryn]
-            ovarybnds = oncref.variables[refvarybndsn]
-
             srefvarx, dims = SliceVarDict(ovarx,slicedict)
-            srefvarbndsx, dims = SliceVarDict(ovarxbnds,slicedict)
             srefvary, dims = SliceVarDict(ovary,slicedict)
-            srefvarbndsy, dims = SliceVarDict(ovarybnds,slicedict)
-
             reflon, reflat =  gen.lonlat2D(ovarx[tuple(srefvarx)], ovary[tuple(srefvary)])
-            refvarbndsx = ovarxbnds[tuple(srefvarbndsx)]
-            refvarbndsy = ovarybnds[tuple(srefvarbndsy)]
+
+            if refvarxbndsn != 'WRFxbnds':
+                ovarxbnds = oncref.variables[refvarxbndsn]
+                ovarybnds = oncref.variables[refvarybndsn]
+                srefvarbndsx, dims = SliceVarDict(ovarxbnds,slicedict)
+                srefvarbndsy, dims = SliceVarDict(ovarybnds,slicedict)
+                refvarbndsx = ovarxbnds[tuple(srefvarbndsx)]
+                refvarbndsy = ovarybnds[tuple(srefvarbndsy)]
+            else:
+                oulon = oncref.variables['XLONG_U']
+                oulat = oncref.variables['XLAT_U']
+                ovlon = oncref.variables['XLONG_V']
+                ovlat = oncref.variables['XLAT_V']
+                refvarbndsx, refvarbndsy= compute_WRFbnds(oulon[0,:,:], oulat[0,:,:],\
+                  ovlon[0,:,:], ovlat[0,:,:])
 
             refdx = reflon.shape[1]
@@ -27412 +27454 @@
                 quit(-1)
 
-            if not oncget.variables.has_key(getvarxbndsn):
+            if getvarxbndsn != 'WRFxbnds' and                                        \
+              not oncget.variables.has_key(getvarxbndsn):
                 print errormsg
                 print '  ' + fname + ": get file '" + netcdfgetn + "' does " +       \
@@ -27439 +27482 @@
                 quit(-1)
 
-            if not oncget.variables.has_key(getvarybndsn):
+            if getvarybndsn != 'WRFybnds' and                                        \
+              not oncget.variables.has_key(getvarybndsn):
                 print errormsg
                 print '  ' + fname + ": get file '" + netcdfgetn + "' does " +       \
@@ -27450 +27494 @@
             getslicedict = {getdimxn: slicedimx, getdimyn: slicedimy}
             ovarx = oncget.variables[getvarxn]
-            ovarxbnds = oncget.variables[getvarxbndsn]
             ovary = oncget.variables[getvaryn]
-            ovarybnds = oncget.variables[getvarybndsn]
-
             sgetvarx, dims = SliceVarDict(ovarx,getslicedict)
-            sgetvarbndsx, dims = SliceVarDict(ovarxbnds,getslicedict)
             sgetvary, dims = SliceVarDict(ovary,getslicedict)
-            sgetvarbndsy, dims = SliceVarDict(ovarybnds,getslicedict)
-
             getlon, getlat =  gen.lonlat2D(ovarx[tuple(sgetvarx)], ovary[tuple(sgetvary)])
-            getvarbndsx = ovarxbnds[tuple(sgetvarbndsx)]
-            getvarbndsy = ovarybnds[tuple(sgetvarbndsy)]
+
+            if getvarxbndsn != 'WRFxbnds':
+                ovarxbnds = oncget.variables[getvarxbndsn]
+                ovarybnds = oncget.variables[getvarybndsn]
+                sgetvarbndsx, dims = SliceVarDict(ovarxbnds,getslicedict)
+                sgetvarbndsy, dims = SliceVarDict(ovarybnds,getslicedict)
+                getvarbndsx = ovarxbnds[tuple(sgetvarbndsx)]
+                getvarbndsy = ovarybnds[tuple(sgetvarbndsy)]
+            else:
+                oulon = oncget.variables['XLONG_U']
+                oulat = oncget.variables['XLAT_U']
+                ovlon = oncget.variables['XLONG_V']
+                ovlat = oncget.variables['XLAT_V']
+                getvarbndsx, getvarbndsy= compute_WRFbnds(oulon[0,:,:], oulat[0,:,:],\
+                  ovlon[0,:,:], ovlat[0,:,:])
+
             getdx = getlon.shape[1]
             getdy = getlon.shape[0]