Changeset 2643 in lmdz_wrf for trunk/tools/diag_tools.py

Timestamp:

Jun 28, 2019, 9:42:25 PM (6 years ago)

Author:

lfita

Message:

Adding:

• Computing of front following Rodrigues et al 2004

File:

• trunk/tools/diag_tools.py (modified) (2 diffs)

trunk/tools/diag_tools.py

@@ -41 +41 @@
 # Forcompute_fog_RUC: Computation of fog and visibility following RUC method Smirnova, (2000)
 # Forcompute_fog_FRAML50: fog and visibility following Gultepe and Milbrandt, (2010)
+# Forcompute_front_R04: Function to compute front following Rodrigues et al.(2004)
 # Forcompute_potevap_orPM: Function to compute potential evapotranspiration following
 #   Penman-Monteith formulation implemented in ORCHIDEE
@@ -1531 +1532 @@
     return hgtmax, pthgtmax, dhgt, peaks, valleys, origfaces, filtfaces, vardims,    \
       varvdims, ranges, rangeshgtmax, ptrangeshgtmax
+
+def Forcompute_front_R04(tas, uas, vas, dxs, dys, dimns, dimvns):
+    """ Function to compute front following Rodrigues et al.(2004), Rev. Bras. 
+          Geofis. 22, 135-151
+    Forcompute_front_R04(tas, vas, dimns, dimvns)
+      [tas]= surface air-temperature values (assuming [[t],y,x]) [K]
+      [uas]= latitudinal wind (assuming [[t],y,x]) [ms-1]
+      [vas]= meridional wind (assuming [[t],y,x]) [ms-1]
+      [dxs]= grid spacing between grid points along x-axis (assuming [y,x]) [m]
+      [dys]= grid spacing between grid points along y-axis (assuming [y,x]) [m]
+      [dimns]= list of the name of the dimensions of [pa]
+      [dimvns]= list of the name of the variables with the values of the 
+        dimensions of [pa]
+    """
+    fname = 'Forcompute_front_R04'
+
+    psldims = dimns[:]
+    pslvdims = dimvns[:]
+
+    if len(pa.shape) == 3:
+        front = np.zeros((tas.shape[0],tas.shape[1],tas.shape[2]), dtype=np.float)
+
+        dx = tas.shape[2]
+        dy = tas.shape[1]
+        dt = tas.shape[0]
+
+        pfront=fdin.module_fordiagnostics.compute_front_R04d3(tas=tas[:].transpose(),\
+          uas=uas[:].transpose(), vas=vas[:].transpose(), dxs=dxs[:].transpose(),    \
+          dys=dys[:].transpose(), d1=dx, d2=dy, d3=dt)
+        front = pfront.transpose()
+    else:
+        print errormsg
+        print '  ' + fname + ': rank', len(pa.shape), 'not ready !!'
+        print '  it only computes 3D [t,y,x] rank values'
+        quit(-1)
+
+    return front
 
 ####### ###### ##### #### ### ## # END Fortran diagnostics