Changeset 2678 in lmdz_wrf for trunk

Timestamp:

Jul 13, 2019, 9:52:45 PM (5 years ago)

Author:

lfita

Message:

Starting to work 'frontogenesis'

Location:

trunk/tools

Files:

• diag_tools.py (modified) (2 diffs)
• diagnostics.py (modified) (18 diffs)

trunk/tools/diag_tools.py

@@ -1624 +1624 @@
         dyst = dys.transpose()
 
-        print '  ' + fname + ' Lluis shapes dzs', dzs.shape, 'dz:', dz
- 
         pxdiab, pydiab, pzdiab, pxdef, pydef, pzdef, pxtilt, pytilt, pzdiv, pf =     \
           fdin.module_fordiagnostics.compute_frontogenesis(theta=thetat, ua=uat,     \
@@ -1646 +1644 @@
         quit(-1)
 
-    return xdiab, ydiab, zdiab, xdef, ydef, zdef, xtilt, ytilt, zdiv, f
+    return xdiab, ydiab, zdiab, xdef, ydef, zdef, xtilt, ytilt, zdiv, f, frontdims, frontvdims
 
 ####### ###### ##### #### ### ## # END Fortran diagnostics

trunk/tools/diagnostics.py

@@ -104 +104 @@
 NONcheckingvars = ['accum', 'cllmh', 'deaccum', 'face', 'LONLATdxdy',                \
   'params', 'reglonlatbnds', 'TSrhs', 'TStd', 'TSwds', 'TSwss',                      \
+  'UNua', 'UNva', 'UNwa',                                                            \
   'WRFbils',  'WRFbnds',                                                             \
   'WRFclivi', 'WRFclwvi', 'WRFdens', 'WRFdx', 'WRFdxdy', 'WRFdxdywps', 'WRFdy',      \
@@ -206 +207 @@
     if dnv == 'WRFdxdywps':WRFdxdywps_compute = True
     if dnv == 'LONLATdxdy':LONLATdxdy_compute = True
+    if dnv[0:4] == 'UNua':
+        UNua_compute = True
+        vUnua = dnv.split(',')[1]
+    if dnv[0:4] == 'UNva':
+        UNva_compute = True
+        vUnva = dnv.split(',')[1]
+    if dnv[0:4] == 'UNwa':
+        UNwa_compute = True
+        vUnwa = dnv.split(',')[1]
 
 # diagnostics to compute
@@ -223 +233 @@
         if depvars == 'WRFtime': WRFtime_compute = True
         if depvars == 'WRFz': WRFz_compute = True
+        if depvars == 'WRFdx': WRFdx_compute = True
+        if depvars == 'WRFdy': WRFdy_compute = True
+        if depvars == 'WRFdz': WRFdz_compute = True
+        if depvars == 'WRFdxdy': WRFdxdy_compute = True
+        if depvars == 'WRFdxdywps': WRFdxdywps_compute = True
+        if depvars == 'LONLATdxdy': LONLATdxdy_compute = True
+        if depvars[0:4] == 'UNua': 
+            UNua_compute = True
+            vUnua = dnv.split(',')[1]
+        if depvars[0:4] == 'UNva': 
+            UNva_compute = True
+            vUnva = dnv.split(',')[1]
+        if depvars[0:4] == 'UNwa': 
+            UNwa_compute = True
+            vUnwa = dnv.split(',')[1]
     else:
         depvars = diags[idiag].split('|')[1].split('@')
@@ -240 +265 @@
         if gen.searchInlist(depvars, 'WRFdxdywps'): WRFdxdywps_compute = True
         if gen.searchInlist(depvars, 'LONLATdxdy'): LONLATdxdy_compute = True
+        if gen.searchInlist_Strsec(depvars, 0, 3, 'UNua'): 
+            UNua_compute = True
+            vals, ind = gen.search_sec_list(depvars,'UNua')
+            dnv = depvars[ind[0]]
+            vUNua = dnv.split(':')[1]
+        if gen.searchInlist_Strsec(depvars, 0, 3, 'UNva'): 
+            UNva_compute = True
+            vals, ind = gen.search_sec_list(depvars,'UNva')
+            dnv = depvars[ind[0]]
+            vUNva = dnv.split(':')[1]
+        if gen.searchInlist_Strsec(depvars, 0, 3, 'UNwa'): 
+            UNwa_compute = True
+            vals, ind = gen.search_sec_list(depvars,'UNwa')
+            dnv = depvars[ind[0]]
+            vUNwa = dnv.split(':')[1]
 
 # Dictionary with the new computed variables to be able to add them
@@ -379 +419 @@
     refdateS = yrref + '-' + monref + '-' + dayref + ' ' + horref + ':' + minref +   \
       ':' + secref
-
 
     if len(timeobj.shape) == 2:
@@ -537 +576 @@
     # Attributes of the variable
     Vvals = gen.variables_values('WRFdx')
-    dictcompvars['WRFdx'] = {'name': Vvals[0], 'standard_name': Vvals[1],             \
+    dictcompvars['WRFdx'] = {'name': Vvals[0], 'standard_name': Vvals[1],            \
       'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
     Vvals = gen.variables_values('WRFdy')
-    dictcompvars['WRFdy'] = {'name': Vvals[0], 'standard_name': Vvals[1],             \
+    dictcompvars['WRFdy'] = {'name': Vvals[0], 'standard_name': Vvals[1],            \
       'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
     Vvals = gen.variables_values('WRFds')
-    dictcompvars['WRFds'] = {'name': Vvals[0], 'standard_name': Vvals[1],             \
+    dictcompvars['WRFds'] = {'name': Vvals[0], 'standard_name': Vvals[1],            \
+      'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
+
+if UNua_compute:
+    print '  ' + main + ": un-staggering '" + vUNua + "': as 0.5*(ua[0:dimx-1]+" +   \
+      "ua[1:dimx])"
+    vunua = ncobj.variables[vUNua][:]
+    dimv = ncobj.variables[vUNua].shape
+
+    dimx = dimv[3]
+    dimy = dimv[2]
+    dimz = dimv[1]
+    dimt = dimv[0]
+
+    undimx = 'unx'
+
+    unua = np.zeros((dimt,dimz,dimy,dimx-1), dtype=np.float)
+    unua[...,0:dimx-1] = 0.5*(vunua[...,1:dimx]+vunua[...,0:dimx-1])
+
+    # Attributes of the variable
+    Vvals = gen.variables_values('ua')
+    dictcompvars['unua'] = {'name': Vvals[0], 'standard_name': Vvals[1],            \
+      'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
+
+if UNva_compute:
+    print '  ' + main + ": un-staggering '" + vUNva + "': as 0.5*(va[0:dimy-1]+" +   \
+      "va[1:dimy])"
+    vunva = ncobj.variables[vUNva][:]
+    dimv = ncobj.variables[vUNva].shape
+
+    dimx = dimv[3]
+    dimy = dimv[2]
+    dimz = dimv[1]
+    dimt = dimv[0]
+
+    undimy = 'uny'
+
+    unva = np.zeros((dimt,dimz,dimy-1,dimx), dtype=np.float)
+    unva[...,0:dimy-1,:] = 0.5*(vunva[...,1:dimy,:]+vunva[...,0:dimy-1,:])
+
+    # Attributes of the variable
+    Vvals = gen.variables_values('va')
+    dictcompvars['unva'] = {'name': Vvals[0], 'standard_name': Vvals[1],            \
+      'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
+
+if UNwa_compute:
+    print '  ' + main + ": un-staggering '" + vUNwa + "': as 0.5*(wa[0:dimz-1]+" +   \
+      "wa[1:dimz])"
+    vunwa = ncobj.variables[vUNwa][:]
+    dimv = ncobj.variables[vUNwa].shape
+
+    dimx = dimv[3]
+    dimy = dimv[2]
+    dimz = dimv[1]
+    dimt = dimv[0]
+
+    undimz = 'unz'
+
+    unwa = np.zeros((dimt,dimz-1,dimy,dimx), dtype=np.float)
+    unwa[...,0:dimz-1,:,:] = 0.5*(vunwa[...,1:dimz,:,:]+vunwa[...,0:dimz-1,:,:])
+
+    # Attributes of the variable
+    Vvals = gen.variables_values('wa')
+    dictcompvars['unwa'] = {'name': Vvals[0], 'standard_name': Vvals[1],            \
       'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
 
@@ -881 +983 @@
 # frontogenesis (theta, ua, va, wa, press, dxs, dys, dzs, time)
     elif diagn == 'frontogenesis':
-            
         if depvars[0] == 'WRFt': var0 = WRFt
         else: var0 = ncobj.variables[depvars[0]][:]
@@ -888 +989 @@
         dz = var0.shape[1]
         dt = var0.shape[0]
-        var1 = ncobj.variables[depvars[1]][:]
-        var2 = ncobj.variables[depvars[2]][:]
-        var3 = ncobj.variables[depvars[3]][:]
+        if depvars[1][0:4] == 'UNua': var1 = unua
+        else: var1 = ncobj.variables[depvars[1]][:]
+        if depvars[2][0:4] == 'UNva': var2 = unva
+        else: var2 = ncobj.variables[depvars[2]][:]
+        if depvars[3][0:4] == 'UNwa': var3 = unwa
+        else: var3 = ncobj.variables[depvars[3]][:]
         if depvars[4] == 'WRFp': var4 = WRFp
         else: var4 = ncobj.variables[depvars[4]]
@@ -909 +1013 @@
           var4, var5, var6, var7, var8, dnames, dvnames)
 
+        # 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)
+
         ncvar.insert_variable(ncobj, 'diabh', diag1, diagoutd, diagoutvd, newnc,     \
           gen.fillValueF)
-        ovar = newnc.variables['diabh']
+        newnc.renameVariable('diabh', 'xdiabh')
+        newnc.sync()
+        ovar = newnc.variables['xdiabh']
         stdn = ovar.getncattr('standard_name')
         ovar.setncattr('standard_name', 'x'+stdn)
@@ -921 +1034 @@
         ncvar.insert_variable(ncobj, 'diabh', diag2, diagoutd, diagoutvd, newnc,     \
           gen.fillValueF)
-        ovar = newnc.variables['diabh']
+        newnc.renameVariable('diabh', 'ydiabh')
+        newnc.sync()
+        ovar = newnc.variables['ydiabh']
         stdn = ovar.getncattr('standard_name')
         ovar.setncattr('standard_name', 'y'+stdn)
@@ -931 +1046 @@
         ncvar.insert_variable(ncobj, 'diabh', diag3, diagoutd, diagoutvd, newnc,     \
           gen.fillValueF)
-        ovar = newnc.variables['diabh']
+        newnc.renameVariable('diabh', 'zdiabh')
+        newnc.sync()
+        ovar = newnc.variables['zdiabh']
         stdn = ovar.getncattr('standard_name')
         ovar.setncattr('standard_name', 'z'+stdn)
@@ -942 +1059 @@
         ncvar.insert_variable(ncobj, 'def', diag4, diagoutd, diagoutvd, newnc,       \
           gen.fillValueF)
-        ovar = newnc.variables['def']
+        newnc.renameVariable('def', 'xdef')
+        newnc.sync()
+        ovar = newnc.variables['xdef']
         stdn = ovar.getncattr('standard_name')
         ovar.setncattr('standard_name', 'thetax'+stdn)
@@ -952 +1071 @@
         ncvar.insert_variable(ncobj, 'def', diag5, diagoutd, diagoutvd, newnc,       \
           gen.fillValueF)
-        ovar = newnc.variables['def']
+        newnc.renameVariable('def', 'ydef')
+        newnc.sync()
+        ovar = newnc.variables['ydef']
         stdn = ovar.getncattr('standard_name')
         ovar.setncattr('standard_name', 'thetay'+stdn)
@@ -962 +1083 @@
         ncvar.insert_variable(ncobj, 'def', diag6, diagoutd, diagoutvd, newnc,       \
           gen.fillValueF)
-        ovar = newnc.variables['def']
+        newnc.renameVariable('def', 'zdef')
+        newnc.sync()
+        ovar = newnc.variables['zdef']
         stdn = ovar.getncattr('standard_name')
         ovar.setncattr('standard_name', 'thetaz'+stdn)
@@ -973 +1096 @@
         ncvar.insert_variable(ncobj, 'tilt', diag7, diagoutd, diagoutvd, newnc,      \
           gen.fillValueF)
-        ovar = newnc.variables['tilt']
+        newnc.renameVariable('tilt', 'xtilt')
+        newnc.sync()
+        ovar = newnc.variables['xtilt']
         stdn = ovar.getncattr('standard_name')
         ovar.setncattr('standard_name', 'thetax'+stdn)
@@ -984 +1109 @@
         ncvar.insert_variable(ncobj, 'tilt', diag8, diagoutd, diagoutvd, newnc,      \
           gen.fillValueF)
-        ovar = newnc.variables['tilt']
+        newnc.renameVariable('tilt', 'ytilt')
+        newnc.sync()
+        ovar = newnc.variables['ytilt']
         stdn = ovar.getncattr('standard_name')
         ovar.setncattr('standard_name', 'thetay'+stdn)
@@ -995 +1122 @@
         ncvar.insert_variable(ncobj, 'div', diag9, diagoutd, diagoutvd, newnc,       \
           gen.fillValueF)
-        ovar = newnc.variables['div']
+        newnc.renameVariable('div', 'zdiv')
+        newnc.sync()
+        ovar = newnc.variables['zdiv']
         stdn = ovar.getncattr('standard_name')
         ovar.setncattr('standard_name', 'thetaz'+stdn)
@@ -1007 +1136 @@
         newvar = newnc.createVariable('e', 'i', ('e'))
         newvar[:] = range(3)
-        ncvar.newvar.basicvardef(newvar, 'component', 'axis component: x, y, z', '-')
+        ncvar.basicvardef(newvar, 'component', 'axis component: x, y, z', '-')
         newnc.sync()