Changeset 844 in lmdz_wrf for trunk

Timestamp:

Jun 16, 2016, 6:10:03 PM (8 years ago)

Author:

lfita

Message:

Adding:

• `computevar_model': Function to provide the way to compute a CF-variable providing its name
• `var_model': Function to provide the way to compute a CF-variable providing its name

File:

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

trunk/tools/nc_var_tools.py

@@ -114 +114 @@
 # var_dim_dimv: Function to bring back which variables with the values should be use for a given variable
 # varDimension: Function to find the variable with the values of a dimension. It is assumed, that given variable
+# var_model: Function to provide the way to compute a CF-variable providing its name
 # varout: Function when we want to output variable values
 # varoutold: Function when we want to output variable values
@@ -15101 +15102 @@
     return
 
+def var_model(varname, onc):
+    """ Function to provide the way to compute a CF-variable providing its name
+      varname= CF-compilant variable name
+      onc= netCDF file object from which the variable should be retrieved
+    >>> onc = NetCDFFile('/home/lluis/PY/wrfout_d01_2001-11-11_00:00:00', 'r')
+    >>> mod, diag = var_model('tas', onc)
+    ['T2'], None
+    >>> mod, diag = var_model('hurs', onc)
+    None, [['PSFC', 'T2', 'Q2']]
+    """
+    fname = 'var_model'
+
+    ncvars = onc.variables
+
+# Getting non-chceking variables
+    folder = os.path.dirname(os.path.realpath(__file__))
+    infile = folder + '/diagnostics.inf'
+    if not os.path.isfile(infile):
+        print errormsg
+        print '  ' + fname + ": File '" + infile + "' does not exist !!"
+        quit(-1)
+
+    ncf = open(infile, 'r')
+
+    for line in ncf:
+        if line[0:9] == '# NOcheck':
+            NOchk = line.split('=')[1].replace(' ','').replace('\n','').split(':')
+            ncf.close()
+            break
+
+    print NOchk
+
+# Variables from which CF variable might be computed
+    modelvars = gen.CFvar_MODvar(varname)
+    diagvars = gen.CFvar_DIAGvar(varname)
+
+    computevarmod = []
+    if modelvars is not None:
+        for var in modelvars:
+            if gen.searchInlist(ncvars,var): computevarmod.append(var)
+    if len(computevarmod) < 1: computevarmod = None
+
+    computevardiag = []
+    if diagvars is not None:
+        for combovar in diagvars:
+            alltrue = False
+            if len(combovar) > 1:
+                for cvar in combovar: 
+                    if gen.searchInlist(NOchk,cvar):
+                        alltrue = True
+                    else:
+                        if gen.searchInlist(ncvars,cvar): alltrue = True
+            else:
+                if gen.searchInlist(ncvars,combovar): alltrue = True
+
+            if alltrue: computevardiag.append(combovar)
+
+    if len(computevardiag) < 1: computevardiag = None
+
+    return computevarmod, computevardiag
+
+def computevar_model(values, ncfile):
+    """ Function to provide the way to compute a CF-variable providing its name
+      varname= CF-compilant variable name
+      ncfile= netCDF file from which the variable should be retrieved
+    """
+    fname = 'computevar_model'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print computevar_model.__doc__
+        quit()
+
+    varname = values
+    ncobj = NetCDFFile(ncfile, 'r')
+ 
+    varmod, vardiag =  var_model(varname, ncobj)
+    print '  ' + fname + ": CF variable '" + varname + "' can be computed ______"
+    print '    model:', varmod
+    print '    diagnostics:', vardiag
+    
+    ncobj.close()
+
+    return
+
 #quit()