Changeset 361 in lmdz_wrf for trunk/tools/drawing_tools.py

Timestamp:

Mar 18, 2015, 4:13:46 PM (10 years ago)

Author:

lfita

Message:

Adding the version of 'variables_values' reading the information from a file
Keeping the old one as 'variables_values_old'

File:

• trunk/tools/drawing_tools.py (modified) (1 diff)

trunk/tools/drawing_tools.py

@@ -698 +698 @@
 
 def variables_values(varName):
+    """ Function to provide values to plot the different variables values from ASCII file
+      'variables_values.dat'
+    variables_values(varName)
+      [varName]= name of the variable
+        return: [var name], [std name], [minimum], [maximum], 
+          [long name]('|' for spaces), [units], [color palette] (following: 
+          http://matplotlib.org/1.3.1/examples/color/colormaps_reference.html)
+     [varn]: original name of the variable
+       NOTE: It might be better doing it with an external ASII file. But then we 
+         got an extra dependency...
+    >>> variables_values('WRFght')
+    ['z', 'geopotential_height', 0.0, 80000.0, 'geopotential|height', 'm2s-2', 'rainbow']
+    """
+    fname='variables_values'
+
+    if varName == 'h':
+        print fname + '_____________________________________________________________'
+        print variables_values.__doc__
+        quit()
+
+    infile = 'variables_values.dat'
+
+    if not os.path.isfile(infile):
+        print errormsg
+        print '  ' + fname + ": File '" + infile + "' does not exist !!"
+        quit(-1)
+
+# Variable name might come with a statistical surname...
+    stats=['min','max','mean','stdv', 'sum']
+
+    ifst = False
+    for st in stats:
+        if varName.find(st) > -1:
+            print '    '+ fname + ": varibale '" + varName + "' with a statistical "+\
+              " surname: '",st,"' !!"
+            Lst = len(st)
+            LvarName = len(varName)
+            varn = varName[0:LvarName - Lst]
+            ifst = True
+            break
+    if not ifst:
+        varn = varName
+
+    ncf = open('variables_values.dat', 'r')
+
+    for line in ncf:
+        if line[0:1] != '#':
+            values = line.replace('\n','').split(',')
+            if len(values) != 8:
+                print errormsg
+                print "problem in varibale:'", values[0],                            \
+                  'it should have 8 values and it has',len(values)
+                quit(-1)
+
+            if varn[0:6] == 'varDIM': 
+# Variable from a dimension (all with 'varDIM' prefix)
+                Lvarn = len(varn)
+                varvals = [varn[6:Lvarn+1], varn[6:Lvarn+1], 0., 1.,                 \
+                  "variable|from|size|of|dimension|'" + varn[6:Lvarn+1] + "'", '1',  \
+                   'rainbow']
+            else:
+                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 values[0] == varn:
+                print varvals
+                ncf.close()
+                return varvals
+                break
+
+    print errormsg
+    print '  ' + fname + ": variable '" + varn + "' not defined !!!"
+    quit(-1)
+
+    return 
+
+def variables_values_old(varName):
     """ Function to provide values to plot the different variables
     variables_values(varName)