Changeset 837 in lmdz_wrf for trunk

Timestamp:

Jun 16, 2016, 4:47:47 PM (8 years ago)

Author:

lfita

Message:

Adding `CFvar_MODvar': Function to provide which model values can provide a CF-variable from ASCII file

File:

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

trunk/tools/generic_tools.py

@@ -19 +19 @@
 
 ####### Content
+# 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
 # chainSnum_levnext: Function to provide the next value for a given level from a chainStrnum hirerarchy of numbers
@@ -41 +42 @@
 # vals_around: Function to provide the 3x3 values around a given j,i point
 
+def searchInlist(listname, nameFind):
+    """ Function to search a value within a list
+    listname = list
+    nameFind = value to find
+    >>> searInlist(['1', '2', '3', '5'], '5')
+    True
+    """
+    for x in listname:
+      if x == nameFind:
+        return True
+    return False
+
 def values_line(line, splitv, chars):
     """ Function to retrieve values from a line of ASCII text removing a series of characters
@@ -251 +264 @@
     if idate == 'h':
         print fname + '_____________________________________________________________'
-        print variables_values.__doc__
+        print period_information.__doc__
         quit()
 
@@ -552 +565 @@
 
     return 
+
+def CFvar_MODvar(varn):
+    """ 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('hfss')
+    ['hfss', 'LSENS', 'sens', 'HFX', 'hfx']
+    >>> CFvar_MODvar('clm')
+    ['clm', 'cldm', 'LCLDM', 'Mid-level cloudiness']
+    >>> CFvar_MODvar('pr')
+    ['pr', 'RAINTOT', 'precip', 'LPRECIP', 'Precip Totale liq+sol']
+    """
+    import subprocess as sub
+
+    fname='CFvar_MODvar'
+
+    if varn == 'h':
+        print fname + '_____________________________________________________________'
+        print CFvar_MODvar.__doc__
+        quit()
+
+    folder = os.path.dirname(os.path.realpath(__file__))
+
+    infile = folder + '/variables_values.dat'
+
+    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[1].replace(' ',''), values[2].replace(' ',''),         \
+              np.float(values[3]), np.float(values[4]),values[5].replace(' ',''),    \
+              values[6].replace(' ',''), values[7].replace(' ','')]
+            if varvals[0] == varn: MODvars.append(values[0])
+
+    if len(MODvars) == 0:
+        print errormsg
+        print '  ' + fname + ": variable '" + varn + "' not defined !!!"
+        ncf.close()
+        return None
+    else:
+        return MODvars
+
+print CFvar_MODvar('hfss')
+print CFvar_MODvar('pr')
+
+quit()
 
 def variables_values_old(varName):
@@ -1419 +1486 @@
       return end
 
-def searchInlist(listname, nameFind):
-    """ Function to search a value within a list
-    listname = list
-    nameFind = value to find
-    >>> searInlist(['1', '2', '3', '5'], '5')
-    True
-    """
-    for x in listname:
-      if x == nameFind:
-        return True
-    return False
-
 def datetimeStr_datetime(StringDT):
     """ Function to transform a string date ([YYYY]-[MM]-[DD]_[HH]:[MI]:[SS] format) to a date object