Changeset 840 in lmdz_wrf for trunk/tools

Timestamp:

Jun 16, 2016, 5:19:47 PM (9 years ago)

Author:

lfita

Message:

Adding:

• `CFvar_DIAGvar': Function to provide which model diagnostic values can provide a CF-variable from ASCII file

File:

• trunk/tools/generic_tools.py (modified) (3 diffs)

trunk/tools/generic_tools.py

@@ -19 +19 @@
 
 ####### Content
+# CFvar_DIAGvar: Function to provide which model diagnostic values can provide a CF-variable from ASCII file
 # CFvar_MODvar: Function to provide which model values can provide a CF-variable from ASCII file
 # chainSnumHierarchy: Class to provide the structure of a `ChainStrNum' hierarchy
@@ -569 +570 @@
     """ Function to provide which model values can provide a CF-variable from ASCII file
       'variables_values.dat'
-    CFvar_MODvar(varName)
-      [varName]= name of the variable
+    CFvar_MODvar(varn)
+      [varn]= CF name of the variable
     >>> CFvar_MODvar('hfss')
     ['hfss', 'LSENS', 'sens', 'HFX', 'hfx']
@@ -615 +616 @@
         return MODvars
 
-print CFvar_MODvar('hfss')
-print CFvar_MODvar('pr')
+def CFvar_DIAGvar(varn):
+    """ Function to provide which model diagnostic values can provide a CF-variable from ASCII file
+      'diagnostics.inf' (reference for diagnostics.py')
+       [CFname], [moddiag], [varcombo]
+         [moddiag]: name of the diagnosted variable
+         [varcombo]: combnination of variables to be used to compute diagnostic
+    CFvar_DIAGvar(varn)
+      [varn]= CF name of the variable
+    >>> CFvar_MODvar('pr')
+    [[' RAINC', 'RAINNC']]
+    >>> CFvar_MODvar('hurs')
+    [[' psol', 't2m', 'q2m'], [' psfc', 't', 'q'], [' PSFC', 'T2', 'Q2']]
+    """
+    import subprocess as sub
+
+    fname='CFvar_DIAGvar'
+
+    if varn == 'h':
+        print fname + '_____________________________________________________________'
+        print CFvar_DIAGvar.__doc__
+        quit()
+
+    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')
+
+    MODvars = []
+    for line in ncf:
+        if line[0:1] != '#':
+            values = line.replace('\n','').split(',')
+            varvals = [values[0], values[1], values[2]]
+            if varvals[0] == varn: MODvars.append(values[2].split('@'))
+
+    if len(MODvars) == 0:
+        print errormsg
+        print '  ' + fname + ": variable '" + varn + "' not defined !!!"
+        ncf.close()
+        return None
+    else:
+        return MODvars
+
+print CFvar_DIAGvar('pr')
+print CFvar_DIAGvar('hurs')
 
 quit()