Changeset 1351 in lmdz_wrf

Timestamp:

Nov 22, 2016, 5:23:21 PM (8 years ago)

Author:

lfita

Message:

Adding computation of 'WRFtime' and the attributes to all complementary variables

File:

• trunk/tools/diagnostics.py (modified) (15 diffs)

trunk/tools/diagnostics.py

@@ -867 +867 @@
 parser.add_option("-f", "--netCDF_file", dest="ncfile", help="file to use", metavar="FILE")
 parser.add_option("-d", "--dimensions", dest="dimns",  
-  help="[dimtn]@[dtvn],[dimzn]@[dzvn],[...,[dimxn]@[dxvn]], ',' list with the couples [dimDn]@[dDvn], [dimDn], name of the dimension D and name of the variable [dDvn] with the values of the dimension" + comboinf, 
+  help="[dimtn]@[dtvn],[dimzn]@[dzvn],[...,[dimxn]@[dxvn]], ',' list with the couples [dimDn]@[dDvn], [dimDn], name of the dimension D and name of the variable [dDvn] with the values of the dimension ('WRFtime', for WRF time copmutation)" + comboinf, 
   metavar="LABELS")
 parser.add_option("-v", "--variables", dest="varns", 
@@ -878 +878 @@
     #######
 availdiags = ['ACRAINTOT', 'accum', 'clt', 'cllmh', 'deaccum', 'LMDZrh', 'mslp',     \
-  'OMEGAw', 'RAINTOT',   \
+  'OMEGAw', 'RAINTOT',                                                               \
   'rvors', 'td', 'turbulence', 'WRFgeop', 'WRFp', 'WRFrvors', 'ws', 'wds', 'wss',    \
   'WRFheight', 'WRFua', 'WRFva']
@@ -890 +890 @@
   'WRFpos', 'WRFprc', 'WRFprls', 'WRFrh', 'LMDZrh', 'LMDZrhs', 'WRFrhs', 'WRFrvors', \
   'WRFt', 'WRFtime', 'WRFua', 'WRFva', 'WRFwds', 'WRFwss', 'WRFheight']
+
+NONchkvardims = ['WRFtime']
 
 ofile = 'diagnostics.nc'
@@ -911 +913 @@
 
 ncobj = NetCDFFile(opts.ncfile, 'r')
-
-# File creation
-newnc = NetCDFFile(ofile,'w')
-
-# dimensions
-dimvalues = dimns.split(',')
-dnames = []
-dvnames = []
-
-for dimval in dimvalues:
-    dnames.append(dimval.split('@')[0])
-    dvnames.append(dimval.split('@')[1])
-
-# diagnostics to compute
-diags = varns.split(',')
-Ndiags = len(diags)
 
 # Looking for specific variables that might be use in more than one diagnostic
@@ -937 +923 @@
 WRFpos_compute = False
 WRFtime_compute = False
+
+# File creation
+newnc = NetCDFFile(ofile,'w')
+
+# dimensions
+dimvalues = dimns.split(',')
+dnames = []
+dvnames = []
+
+for dimval in dimvalues:
+    dn = dimval.split('@')[0]
+    dnv = dimval.split('@')[1]
+    dnames.append(dn)
+    dvnames.append(dnv)
+    # Is there any dimension-variable which should be computed?
+    if dnv == 'WRFgeop':WRFgeop_compute = True
+    if dnv == 'WRFp': WRFp_compute = True
+    if dnv == 'WRFt': WRFt_compute = True
+    if dnv == 'WRFrh': WRFrh_compute = True
+    if dnv == 'WRFght': WRFght_compute = True
+    if dnv == 'WRFdens': WRFdens_compute = True
+    if dnv == 'WRFpos': WRFpos_compute = True
+    if dnv == 'WRFtime': WRFtime_compute = True
+
+# diagnostics to compute
+diags = varns.split(',')
+Ndiags = len(diags)
 
 for idiag in range(Ndiags):
@@ -960 +973 @@
         if gen.searchInlist(depvars, 'WRFtime'): WRFtime_compute = True
 
+# Dictionary with the new computed variables to be able to add them
+dictcompvars = {}
 if WRFgeop_compute:
     print '  ' + main + ': Retrieving geopotential value from WRF as PH + PHB'
@@ -965 +980 @@
     WRFgeop = ncobj.variables['PH'][:] + ncobj.variables['PHB'][:]
 
+    # Attributes of the variable
+    Vvals = gen.variables_values('WRFgreop')
+    dictcompvars['WRFgeop'] = {'name': Vvals[0], 'standard_name': Vvals[1],          \
+      'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
+
 if WRFp_compute:
     print '  ' + main + ': Retrieving pressure value from WRF as P + PB'
@@ -970 +990 @@
     WRFp = ncobj.variables['P'][:] + ncobj.variables['PB'][:]
 
+    # Attributes of the variable
+    Vvals = gen.variables_values('WRFp')
+    dictcompvars['WRFgeop'] = {'name': Vvals[0], 'standard_name': Vvals[1],          \
+      'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
+
 if WRFght_compute:
     print '    ' + main + ': computing geopotential height from WRF as PH + PHB ...' 
     WRFght = ncobj.variables['PH'][:] + ncobj.variables['PHB'][:]
+
+    # Attributes of the variable
+    Vvals = gen.variables_values('WRFght')
+    dictcompvars['WRFgeop'] = {'name': Vvals[0], 'standard_name': Vvals[1],          \
+      'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
 
 if WRFrh_compute:
@@ -987 +1017 @@
     WRFrh = qv/data2
 
+    # Attributes of the variable
+    Vvals = gen.variables_values('WRFrh')
+    dictcompvars['WRFrh'] = {'name': Vvals[0], 'standard_name': Vvals[1],            \
+      'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
+
 if WRFt_compute:
     print '    ' + main + ': computing temperature from WRF as inv_potT(T + 300) ...'
@@ -993 +1028 @@
 
     WRFt = (ncobj.variables['T'][:] + 300.)*(p/p0)**(2./7.)
+
+    # Attributes of the variable
+    Vvals = gen.variables_values('WRFt')
+    dictcompvars['WRFt'] = {'name': Vvals[0], 'standard_name': Vvals[1],             \
+      'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
 
 if WRFdens_compute:
@@ -1016 +1056 @@
             WRFdens[it,iz,:,:] = levval
             WRFdens[it,iz,:,:] = mu[it,:,:]*WRFdens[it,iz,:,:]/grav
+
+    # Attributes of the variable
+    Vvals = gen.variables_values('WRFdens')
+    dictcompvars['WRFdens'] = {'name': Vvals[0], 'standard_name': Vvals[1],          \
+      'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
 
 if WRFpos_compute:
@@ -1075 +1120 @@
 
     tunits = tunitsval + ' since ' + refdateS
+
+    # Attributes of the variable
+    dictcompvars['WRFtime'] = {'name': 'time', 'standard_name': 'time',              \
+      'long_name': 'time', 'units': tunits, 'calendar': 'gregorian'}
 
 ### ## #
@@ -1163 +1212 @@
         diagout, diagoutd, diagoutvd = Forcompute_cllmh(var0,var1,dnames,dvnames)
 
+        # Removing the nonChecking variable-dimensions from the initial list
+        varsadd = []
+        for nonvd in NONchkvardims:
+            if gen.searchInlist(diagoutvd,nonvd): diagoutvd.remove(nonvd)
+            varsadd.append(nonvd)
+
         ncvar.insert_variable(ncobj, 'cll', diagout[0,:], diagoutd, diagoutvd, newnc)
         ncvar.insert_variable(ncobj, 'clm', diagout[1,:], diagoutd, diagoutvd, newnc)
@@ -1172 +1227 @@
         var0 = ncobj.variables[depvars]
         diagout, diagoutd, diagoutvd = Forcompute_clt(var0,dnames,dvnames)
+
+        # Removing the nonChecking variable-dimensions from the initial list
+        varsadd = []
+        for nonvd in NONchkvardims:
+            if gen.searchInlist(diagoutvd,nonvd): diagoutvd.remove(nonvd)
+            varsadd.append(nonvd)
+            
         ncvar.insert_variable(ncobj, 'clt', diagout, diagoutd, diagoutvd, newnc)
 
@@ -1598 +1660 @@
 
     newnc.sync()
+    # Adding that additional variables required to compute some diagnostics which
+    #   where not in the original file
+    for vadd in varsadd:
+        if not gen.searchInlist(newnc.variables.keys(),vadd):
+            attrs = dictcompvars[vadd]
+            vvn = attrs['name']
+            if not gen.searchInlist(newnc.variables.keys(), vvn):
+                iidvn = dvnames.index(vadd)
+                dnn = dnames[iidvn]
+                if vadd == 'WRFtime':
+                    dvarvals = WRFtime[:]
+                newvar = newnc.createVariable(vvn, 'f8', (dnn))
+                newvar[:] = dvarvals
+                for attn in attrs.keys():
+                    if attn != 'name':
+                        attv = attrs[attn]
+                        ncvar.set_attribute(newvar, attn, attv)
 
 #   end of diagnostics