Changeset 2711 in lmdz_wrf

Timestamp:

Sep 26, 2019, 3:08:03 PM (5 years ago)

Author:

lfita

Message:

Adding

• `equal_units': Function to provide the same units of two datasetsequal_units: Function to provide the same units of two datasets

File:

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

trunk/tools/generic_tools.py

@@ -123 +123 @@
 #   360 days/yr (or 12 30-days months) calendar
 # dtsec360dyr: Class to operate a number of seconds to a date in a 360 days/yr (or 12 30-days months) calendar
+# equal_units: Function to provide the same units of two datasets
 # files_folder_HMT: Function to retrieve a list of files from a folder [fold] and files with [head]*[middle]*[tail]
 # fill_Narray: Function to fill a n-dimensional array with an arrary of lesser rank
@@ -17122 +17123 @@
     return lu
 
-def euqal_units(units1, units2):
+def equal_units(units1, units2, vals2):
     """ Function to provide the same units of two datasets
       units1: units of reference (from units_lunits)
-    """
+      units2: units to change
+    >>> equal_units('$gkg^{-1}$','$kgkg^{-1}$', 0.023)
+    ['mulc', 1000.0], 23.0
+    >>> equal_units('$K$','$^{\circ}C$', 23)
+    ['addc', 273.15], 296.15
+    """
+    fname = 'equal_units'
+    availunits = ['$gkg^{-1}$', 'K']
+
+    if units1 == '$gkg^{-1}$':
+        availunits1 = ['$kgkg^{-1}$']
+        if units2 == '$kgkg^{-1}$': factor = ['mulc', 1000.]
+        else:
+            print errormsg
+            print '  ' + fname + ": for ref. units '" + units1 + "' not available '"+\
+              units2 + "' !!"
+            print '    available ones:', availunits1
+            quit()
+    elif units1 == '$K$':
+        availunits1 = ['$^{\circ}C$']
+        if units2 == '$^{\circ}C$': factor = ['addc', 273.15]
+        else:
+            print errormsg
+            print '  ' + fname + ": for ref. units '" + units1 + "' not available '"+\
+              units2 + "' !!"
+            print '    available ones:', availunits1
+            quit()
+    else:
+        print errormsg
+        print '  ' + fname + ": reference units '" + units1 + "' not available !!"
+        print '    available ones:', availunits
+        quit()
+
+    # Changing values
+    if factor[0] == 'mulc': vals2 = vals2*factor[1]
+    if factor[0] == 'addc': vals2 = vals2 + factor[1]
+
+    return factor, vals2
 
 #quit()