Changeset 2765 in lmdz_wrf for trunk/tools

Timestamp:

Nov 16, 2019, 10:51:32 PM (5 years ago)

Author:

lfita

Message:

Adding:

• `1st order 2D horizontal gradient'

Location:

trunk/tools

Files:

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

trunk/tools/diag_tools.py

@@ -1037 +1037 @@
 
     return mslpv, mslpdims, mslpvdims
+
+def Forcompute_gradient2Dh(var, x, y, dimns, dimvns):
+    """ Function to compute 1st order horizontal 2D gradient of any given variable
+    Forcompute_gradient(var, x, y, dimns, dimvns)
+      [var]= variable (assuming [[t],z,y,x]) [??]
+      [x]= distance along x-axis (assuming [y,x]) [?]
+      [y]= distance along y-axis (assuming [y,x]) [?]
+      [dimns]= list of the name of the dimensions of [pa]
+      [dimvns]= list of the name of the variables with the values of the 
+        dimensions
+    """
+    fname = 'Forcompute_gradient'
+
+    vardims = dimns[:]
+    varvdims = dimvns[:]
+
+    if len(var.shape) == 2:
+        dx = var.shape[1]
+        dy = var.shape[0]
+        gradt=fdin.module_fordiagnostics.compute_gradient2dh1rk(                     \
+          var=var[:].transpose(), x=x[:].transpose(), y=y[:].transpose(), d1=dx,     \
+          d2=dy)
+    elif len(var.shape) == 3:
+        dx = var.shape[2]
+        dy = var.shape[1]
+        dt = var.shape[0]
+        gradt=fdin.module_fordiagnostics.compute_gradient2dh3rk(                     \
+          var=var[:].transpose(), x=x[:].transpose(), y=y[:].transpose(), d1=dx,     \
+          d2=dy, d3=dt)
+    elif len(var.shape) == 4:
+        dx = var.shape[3]
+        dy = var.shape[2]
+        dz = var.shape[1]
+        dt = var.shape[0]
+        gradt=fdin.module_fordiagnostics.compute_gradient2dh4rk( \
+          var=var[:].transpose(), x=x[:].transpose(), y=y[:].transpose(), d1=dx,      \
+          d2=dy, d3=dt)
+    else:
+        print errormsg
+        print '  ' + fname + ': rank', len(var.shape), 'not ready !!'
+        print '  it only computes 2D [y,x], 3D [t,y,x], or 4D [t,y,x] rank values'
+        quit(-1)
+    grad = gradt.transpose()
+
+    return grad, vardims, varvdims
 
 def Forcompute_psl_ecmwf(ps, hgt, ta1, pa2, unpa1, dimns, dimvns):

trunk/tools/diagnostics.py

@@ -91 +91 @@
     #######
 availdiags = ['ACRAINTOT', 'accum', 'clt', 'cllmh', 'convini', 'deaccum', 'fog_K84', \
-  'fog_RUC', 'front_R04', 'frontogenesis', 'rhs_tas_tds', 'LMDZrh', 'mslp', 'OMEGAw',\
-  'RAINTOT', 'range_faces',                                                          \
+  'fog_RUC', 'front_R04', 'frontogenesis', 'gradient2Dh', 'rhs_tas_tds', 'LMDZrh',   \
+  'mslp', 'OMEGAw', 'RAINTOT', 'range_faces',                                        \
   'rvors', 'td', 'timemax', 'timeoverthres', 'turbulence', 'tws', 'uavaFROMwswd',    \
   'WRFbnds', 'WRFcape_afwa', 'WRFclivi', 'WRFclwvi', 'WRF_denszint', 'WRFgeop',      \
@@ -1181 +1181 @@
         newnc.sync()
 
+# gradient2Dh: 1st order Horizontal 2D-gradient of any variable [variable], [distx], 
+#   [disty]
+    elif diagn == 'gradient2Dh':
+
+        var0 = ncobj.variables[depvars[0]]
+        var1 = ncobj.variables[depvars[1]]
+        var2 = ncobj.variables[depvars[2]]
+
+        dnamesvar = var0.dimensions
+        dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
+
+        diagout, diagoutd, diagoutvd = diag.Forcompute_gradient2Dh(var0, var1, var2, \
+          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)
+
+        CFvarn = ncvar.variables_values(depvars[0])[0]
+        ncvar.insert_variable(ncobj, depvars[0], diagout, diagoutd, diagoutvd,       \
+          newnc, gen.fillValueF)
+        newnc.sync()
+        ovar = newnc.variables[depvars[0]]
+        varu = ovar.units
+        newnc.renameVariable(depvars[0], depvars[0]+'2Dhgrad')
+        ovar.setncattr('units', varu+'ds-1')
+
 # LMDZrh (pres, t, r)
     elif diagn == 'LMDZrh':

trunk/tools/module_ForDiagnostics.f90

@@ -34 +34 @@
 ! compute_front_R04d3: Subroutine to compute presence of a front following Rodrigues et al.(2004)
 ! compute_frontogenesis: Subroutine to compute the frontogenesis
+! compute_gradient2Dh4RK: calculation of 1st order horizontal 2D gradient on 4D RK variable
+! compute_gradient2Dh3RK: calculation of 1st order horizontal 2D gradient on 3D RK variable
+! compute_gradient2Dh2RK: calculation of 1st order horizontal 2D gradient on 2D RK variable
 ! compute_massvertint1D: Subroutine to vertically integrate a 1D variable in eta vertical coordinates
 ! compute_psl_ecmwf: Compute sea level pressure using ECMWF method following Mats Hamrud and Philippe Courtier [Pa]
@@ -1635 +1638 @@
   END SUBROUTINE compute_frontogenesis
 
+  SUBROUTINE compute_gradient2Dh4RK(v,x,y,d1,d2,d3,d4,grad)
+! calculation of 1st order horizontal 2D gradient on 4D RK variable
+
+    IMPLICIT NONE
+
+    INTEGER, INTENT(in)                                  :: d1, d2, d3, d4
+    REAL(r_k), DIMENSION(d1,d2,d3,d4), INTENT(in)        :: v
+    REAL(r_k), DIMENSION(d1,d2), INTENT(in)              :: x, y
+    REAL(r_k), DIMENSION(d1,d2,d3,d4), INTENT(out)       :: grad
+
+! Local
+    INTEGER                                              :: k,l
+
+!!!!!!! Variables
+! v: variable to compute its gradient
+! x: distances along x-axis
+! y: distances along y-axis
+
+    fname = 'compute_gradiend2Dh4RK'
+
+    DO k=1, d3
+      DO l=1, d4
+        CALL gradient2D_1o(d1, d2, v(:,:,k,l), x, y, grad(:,:,k,l))
+      END DO
+    END DO
+
+  END SUBROUTINE compute_gradient2Dh4RK
+
+  SUBROUTINE compute_gradient2Dh3RK(v,x,y,d1,d2,d3,grad)
+! calculation of 1st order horizontal 2D gradient on 3D RK variable
+
+    IMPLICIT NONE
+
+    INTEGER, INTENT(in)                                  :: d1, d2, d3
+    REAL(r_k), DIMENSION(d1,d2,d3), INTENT(in)           :: v
+    REAL(r_k), DIMENSION(d1,d2), INTENT(in)              :: x, y
+    REAL(r_k), DIMENSION(d1,d2,d3), INTENT(out)          :: grad
+
+! Local
+    INTEGER                                              :: k
+
+!!!!!!! Variables
+! v: variable to compute its gradient
+! x: distances along x-axis
+! y: distances along y-axis
+
+    fname = 'compute_gradiend2Dh3RK'
+
+    DO k=1, d3
+      CALL gradient2D_1o(d1, d2, v(:,:,k), x, y, grad(:,:,k))
+    END DO
+
+  END SUBROUTINE compute_gradient2Dh3RK
+
+  SUBROUTINE compute_gradient2Dh2RK(v,x,y,d1,d2,grad)
+! calculation of 1st order horizontal 2D gradient on 2D RK variable
+
+    IMPLICIT NONE
+
+    INTEGER, INTENT(in)                                  :: d1, d2
+    REAL(r_k), DIMENSION(d1,d2), INTENT(in)              :: v
+    REAL(r_k), DIMENSION(d1,d2), INTENT(in)              :: x, y
+    REAL(r_k), DIMENSION(d1,d2), INTENT(out)             :: grad
+
+! Local
+    INTEGER                                              :: k
+
+!!!!!!! Variables
+! v: variable to compute its gradient
+! x: distances along x-axis
+! y: distances along y-axis
+
+    fname = 'compute_gradiend2Dh2RK'
+
+    CALL gradient2D_1o(d1, d2, v, x, y, grad)
+
+  END SUBROUTINE compute_gradient2Dh2RK
+
 END MODULE module_ForDiagnostics