Changeset 2277 in lmdz_wrf for trunk/tools

Timestamp:

Dec 28, 2018, 4:03:45 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `reglonlatbnds': cellboundaries using lon, lat from a reglar lon/lat projection as intersection of their related parallels and meridians

Location:

trunk/tools

Files:

• diag_tools.py (modified) (2 diffs)
• diagnostics.inf (modified) (1 diff)
• diagnostics.py (modified) (2 diffs)
• module_ForDiagnostics.f90 (modified) (2 diffs)

trunk/tools/diag_tools.py

@@ -32 +32 @@
 # Forcompute_cellbnds: Function to compute cellboundaries using wind-staggered lon, 
 #   lats as intersection of their related parallels and meridians
+# Forcompute_cellbndsreg: Function to compute cellboundaries using lon, lat from a 
+#   reglar lon/lat projection as intersection of their related parallels and meridians
 # Forcompute_cllmh: Function to compute cllmh: low/medium/hight cloud fraction 
 #   following newmicro.F90 from LMDZ via Fortran subroutine
@@ -566 +568 @@
         print '    vlon shape:', vlon.shape
         print '    vlat shape:', vlat.shape
+        quit(-1)
+    
+    xbnds = xbndst.transpose()
+    ybnds = ybndst.transpose()
+
+    return xbnds, ybnds, dims, vdims
+
+
+def Forcompute_cellbndsreg(lon, lat, dimns, dimvns):
+    """ Function to compute cellboundaries using lon, lat from a reglar lon/lat 
+      projection as intersection of their related parallels and meridians
+    compute_cellbnds(ulon, ulat, vlon, vlat, dimns, dimvns)
+      [ulon]= x-staggered longitudes (assuming [y,x+1])
+      [ulat]= x-staggered latitudes (assuming [y,x+1])
+      [vlon]= y-staggered longitudes (assuming [y+1,x])
+      [vlat]= y-staggered latitudes (assuming [y+1,x])
+      [dimns]= list of the name of the dimensions of [cldfra]
+      [dimvns]= list of the name of the variables with the values of the 
+        dimensions of [ulon]
+    """
+    fname = 'Forcompute_cellbndsreg'
+
+    dims = dimns[:]
+    vdims = dimvns[:]
+
+    if len(lon.shape) == 2:
+        dy = lon.shape[0]
+        dx = lon.shape[1]
+
+        lont = lon.transpose()
+        latt = lat.transpose()
+
+        xbndst, ybndst = fdin.module_fordiagnostics.compute_cellbndsreg(lon=lont,    \
+          lat=latt, dx=dx, dy=dy)
+    else:
+        print errormsg
+        print '  ' + fname + ": wrong rank of variables !!"
+        print '    2D matrices are expected and its found instead'
+        print '    lon shape:', lon.shape
+        print '    lat shape:', lat.shape
         quit(-1)
     

trunk/tools/diagnostics.inf

@@ -22 +22 @@
 hurs, TSrhs, psfc@t@q
 hurs, WRFrhs, PSFC@T2@Q2
+lat_bnds, reglonlatbnds, lon@lat
+lon_bnds, reglonlatbnds, lon@lat
 lat_bnds, WRFbnds, XLONG_U@XLAT_U@XLONG_V@XLAT_V
 lon_bnds, WRFbnds, XLONG_U@XLAT_U@XLONG_V@XLAT_V

trunk/tools/diagnostics.py

@@ -101 +101 @@
 
 # Variables not to check
-NONcheckingvars = ['accum', 'cllmh', 'deaccum', 'face', 'LONLATdxdy', 'TSrhs',       \
-  'TStd', 'TSwds', 'TSwss',                                                          \
+NONcheckingvars = ['accum', 'cllmh', 'deaccum', 'face', 'LONLATdxdy',                \
+  'reglonlatbnds', 'TSrhs', 'TStd', 'TSwds', 'TSwss',                                \
   'WRFbils',  'WRFbnds',                                                             \
   'WRFclivi', 'WRFclwvi', 'WRFdens', 'WRFdx', 'WRFdxdy', 'WRFdxdywps', 'WRFdy',      \
@@ -891 +891 @@
         ncvar.set_attributek(ovar, 'h_valley_newrange', hvalleyrange, 'F')
 
+# cell_bnds: grid cell bounds from lon, lat from a reglar lon/lat projection  as 
+#   intersection of their related parallels and meridians
+    elif diagn == 'reglonlatbnds':
+            
+        var00 = ncobj.variables[depvars[0]][:]
+        var01 = ncobj.variables[depvars[1]][:]
+
+        var0, var1 = gen.lonlat2D(var00,var01)
+
+        dnamesvar = []
+        dnamesvar.append('bnds')
+        if (len(var00.shape) == 3):
+            dnamesvar.append(ncobj.variables[depvars[0]].dimensions[1])
+            dnamesvar.append(ncobj.variables[depvars[0]].dimensions[2])
+        elif (len(var00.shape) == 2):
+            dnamesvar.append(ncobj.variables[depvars[0]].dimensions[0])
+            dnamesvar.append(ncobj.variables[depvars[0]].dimensions[1])
+        elif (len(var00.shape) == 1):
+            dnamesvar.append(ncobj.variables[depvars[0]].dimensions[0])
+            dnamesvar.append(ncobj.variables[depvars[1]].dimensions[0])
+        dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
+
+        cellbndsx, cellbndsy, diagoutd, diagoutvd = diag.Forcompute_cellbndsreg(var0,\
+          var1, dnamesvar, dvnamesvar)
+
+        # Removing the nonChecking variable-dimensions from the initial list
+        varsadd = []
+        diagoutvd = list(dvnames)
+        for nonvd in NONchkvardims:
+            if gen.searchInlist(dvnames,nonvd): diagoutvd.remove(nonvd)
+            varsadd.append(nonvd)
+        # creation of bounds dimension
+        newdim = newnc.createDimension('bnds', 4)
+
+        ncvar.insert_variable(ncobj, 'lon_bnds', cellbndsx, diagoutd, diagoutvd, newnc)
+        ncvar.insert_variable(ncobj, 'lat_bnds', cellbndsy, diagoutd, diagoutvd, newnc)
+
 # cell_bnds: grid cell bounds from XLONG_U, XLAT_U, XLONG_V, XLAT_V as intersection 
 #   of their related parallels and meridians

trunk/tools/module_ForDiagnostics.f90

@@ -18 +18 @@
 ! compute_cellbnds: Subroutine to compute cellboundaries using wind-staggered lon, lats as 
 !   intersection of their related parallels and meridians
+! compute_cellbndsreg: Subroutine to compute cellboundaries using lon, lat from a reglar lon/lat 
+!   projection as intersection of their related parallels and meridians
 ! compute_cllmh4D3: Computation of low, medium and high cloudiness from a 4D CLDFRA and pressure being 
 !   3rd dimension the z-dim 
@@ -1158 +1160 @@
     END DO
 
+  END SUBROUTINE compute_cellbnds
+
+  SUBROUTINE compute_cellbndsreg(dx, dy, lon, lat, xbnds, ybnds)
+! Subroutine to compute cellboundaries using lon, lat from a reglar lon/lat projection as intersection 
+!   of their related parallels and meridians
+
+    IMPLICIT NONE
+
+    INTEGER, INTENT(in)                                  :: dx, dy
+    REAL(r_k), DIMENSION(dx, dy), INTENT(in)             :: lon, lat
+    REAL(r_k), DIMENSION(dx, dy, 4), INTENT(out)         :: xbnds, ybnds
+
+! Local
+    INTEGER                                              :: i,j,iv
+    INTEGER                                              :: ix,ex,iy,ey
+    CHARACTER(len=2), DIMENSION(4)                       :: Svertex
+    INTEGER, DIMENSION(4,2,2,2)                          :: indices
+    REAL(r_k), DIMENSION(2)                              :: ptintsct
+    REAL(r_k), DIMENSION(2,2)                            :: merid, paral
+    LOGICAL                                              :: intsct
+
+!!!!!!! Variables
+! dx, dy: un-staggered dimensions
+! lon, lat: longitudes and latitudes
+! xbnds, ybnds: x and y cell boundaries
+
+    fname = 'compute_cellbndsreg'
+
+    ! Indices to use indices[SW/NW/NE/SE, m/p, x/y, i/e]
+    Svertex = (/ 'SW', 'NW', 'NE', 'SE' /)
+
+    ! SW
+    indices(1,1,1,1) = 0
+    indices(1,1,1,2) = 0
+    indices(1,1,2,1) = -1
+    indices(1,1,2,2) = 0
+    indices(1,2,1,1) = -1
+    indices(1,2,1,2) = 0
+    indices(1,2,2,1) = -1
+    indices(1,2,2,2) = -1
+    ! NW
+    indices(2,1,1,1) = 0
+    indices(2,1,1,2) = 0
+    indices(2,1,2,1) = 0
+    indices(2,1,2,2) = 1
+    indices(2,2,1,1) = -1
+    indices(2,2,1,2) = 0
+    indices(2,2,2,1) = 1
+    indices(2,2,2,2) = 1
+    ! NE
+    indices(3,1,1,1) = 1
+    indices(3,1,1,2) = 1
+    indices(3,1,2,1) = 0
+    indices(3,1,2,2) = 1
+    indices(3,2,1,1) = 0
+    indices(3,2,1,2) = 1
+    indices(3,2,2,1) = 1
+    indices(3,2,2,2) = 1
+    ! SE
+    indices(4,1,1,1) = 1
+    indices(4,1,1,2) = 1
+    indices(4,1,2,1) = -1
+    indices(4,1,2,2) = 0
+    indices(4,2,1,1) = 0
+    indices(4,2,1,2) = 1
+    indices(4,2,2,1) = -1
+    indices(4,2,2,2) = -1
+
+    DO i=1,dx
+      DO j=1,dy
+        DO iv=1,4
+
+          ix = MAX(i+indices(iv,1,1,1),1)
+          !ex = MIN(i+indices(iv,1,1,2),dx)
+          ex = i+indices(iv,1,1,2)
+          iy = MAX(j+indices(iv,1,2,1),1)
+          ey = MIN(j+indices(iv,1,2,2),dy)
+ 
+          merid(1,1) = lon(ix,iy)
+          merid(1,2) = lat(ix,iy)
+          merid(2,1) = lon(ex,ey)
+          merid(2,2) = lat(ex,ey)
+
+          ix = MAX(i+indices(iv,2,1,1),1)
+          ex = MIN(i+indices(iv,2,1,2),dx)
+          iy = MAX(j+indices(iv,2,2,1),1)
+          !ey = MIN(i+indices(iv,2,2,2),dy)
+          ey = j+indices(iv,2,2,2)
+          paral(1,1) = lon(ix,iy)
+          paral(1,2) = lat(ix,iy)
+          paral(2,1) = lon(ex,ey)
+          paral(2,2) = lat(ex,ey)
+
+          CALL intersection_2Dlines(merid, paral, intsct, ptintsct)
+          IF (.NOT.intsct) THEN
+            msg = 'not interection found for ' // Svertex(iv) // ' vertex'
+            CALL ErrMsg(msg, fname, -1)
+          END IF
+          xbnds(i,j,iv) = ptintsct(1)
+          ybnds(i,j,iv) = ptintsct(2)
+        END DO
+      END DO
+    END DO
+
   END SUBROUTINE