Changeset 1462 in lmdz_wrf for trunk/tools

Timestamp:

Mar 4, 2017, 7:01:07 PM (8 years ago)

Author:

lfita

Message:

Adding:

CF_gribequiv: Function to provide the GRIB variable code number from a CF name
grib_CFequiv: Function to provide the CF name of a GRIB variable code number

File:

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

trunk/tools/generic_tools.py

@@ -47 +47 @@
 # auto_val: Function to provide a value following an 'auto' configuration
 # Capturing: Class to capture the standard output from a function
+# CF_gribequiv: Function to provide the GRIB variable code number from a CF name
 # CFmonthU_daysU: Function to transform from a CF date series with units as 'months since [DATE]' to 'days since [DATE]'
 # CFvar_DIAGvar: Function to provide which model diagnostic values can provide a CF-variable from ASCII file
@@ -74 +75 @@
 # get_configuration: Function to get the configuration from an ASCII external file
 # get_specdictionary_HMT: Function to get specific values from a dictionary by selcting that keys with H*M*T
+# grib_CFequiv: Function to provide the CF name of a GRIB variable code number
 # ijlonlat: Function to provide the imin,jmin imax,jmax of a lon,lat box
 # incomming_flow: Function to determine if a fgrid-flow inflows to the central grid point
@@ -11363 +11365 @@
     return timelab
 
+def grib_CFequiv(gribcode, table):
+    """ Function to provide the CF name of a GRIB variable code number
+      gridcode= grib number to search for the reference
+      table= gribd table reference
+      returns list with [CFname], [GRistdbname], [Griblong_name], [Gribunits]]
+    >>> grib_CFequiv(129, '128ECMWF')
+    ['z', 'Z', 'Geopotential', 'm2s-2']
+    """
+    fname = 'grib_CFequiv'
+
+    availtables = ['128ECMWF']
+
+    if table == '128ECMWF':
+        # Source: https://rda.ucar.edu/cgi-bin/transform?xml=/metadata/ParameterTables/WMO_GRIB1.98-0.128.xml&view=gribdoc
+        
+        gribCF = {129: ['z', 'z', 'Geopotential', 'm2 s-2'],                         \
+          130: ['ta', 't', 'Temperature', 'K'],                                      \
+          131: ['ua', 'u', 'U velocity', 'm s-1'],                                   \
+          132: ['va', 'v', 'U velocity', 'm s-1'],                                   \
+          157: ['hur', 'r', 'Relative humidity', '%']}
+    
+    else:
+        print errormsg
+        print '  ' + fname + ": grib table '" + table + "' not ready !!"
+        print '    available ones:', availtables
+        quit(-1)
+
+    if not gribCF.has_key(gribcode):
+        print errormsg
+        print '  ' + fname + ": table '" + table + "' does not have grib code :",    \
+          gribcode, '!!'
+        print '   available ones:', gribCF.keys() 
+        quit(-1)
+
+    CFname = gribCF[gribcode]
+
+    return CFname
+
+def CF_gribequiv(CFname, table):
+    """ Function to provide the GRIB variable code number from a CF name
+      CFname= CF-variable name
+      table= gribd table reference
+    >>> CF_gribequiv('ua', '128ECMWF')
+    [131, 'U', 'U velocity', 'm s-1']
+    """
+    fname = 'CF_gribequiv'
+
+    availtables = ['128ECMWF']
+
+    if table == '128ECMWF':
+        # Source: https://rda.ucar.edu/cgi-bin/transform?xml=/metadata/ParameterTables/WMO_GRIB1.98-0.128.xml&view=gribdoc
+        gribCF = {129: ['z', 'z', 'Geopotential', 'm2 s-2'],                         \
+          130: ['ta', 't', 'Temperature', 'K'],                                      \
+          131: ['ua', 'u', 'U velocity', 'm s-1'],                                   \
+          132: ['va', 'v', 'U velocity', 'm s-1'],                                   \
+          157: ['hur', 'r', 'Relative humidity', '%']}
+    
+    else:
+        print errormsg
+        print '  ' + fname + ": grib table '" + table + "' not ready !!"
+        print '    available ones:', availtables
+        quit(-1)
+
+    # Inverting dictionaries (no need to hand write two dictionaries!)
+    # For single values
+    # FROM: http://stackoverflow.com/questions/483666/python-reverse-invert-a-mapping
+    # CFgrib = dict((gribCF[k], k) for k in gribCF)
+    CFgrib = {}
+    for kv in gribCF.keys():
+        vals = gribCF[kv]
+        newkv = vals[0]
+        vals[0] = kv
+        CFgrib[newkv] = vals
+
+    if not CFgrib.has_key(CFname):
+        print errormsg
+        print '  ' + fname + ": table '" + table + "' does not have CF name :" +     \
+          CFname + "' !!"
+        print '   available ones:', CFgrib.keys() 
+        quit(-1)
+
+    gribcode = CFgrib[CFname]
+
+    return gribcode
+
 #quit()