Changeset 2800 in lmdz_wrf for trunk/tools

Timestamp:

Feb 11, 2020, 7:22:25 PM (5 years ago)

Author:

lfita

Message:

Improving `stringS_dictvar'

Location:

trunk/tools

Files:

• create_OBSnetcdf.py (modified) (6 diffs)
• generic_tools.py (modified) (1 diff)

trunk/tools/create_OBSnetcdf.py

@@ -150 +150 @@
     return finalist
 
+def stringS_dictvar(stringv, Dc=',', DVs=':'):
+    """ Function to provide a dictionary from a String which list of DVs separated 
+          [key] [value] couples
+      stringv: String with a dictionary content as [key1][DVs][val1][Dc][key2][DVs][Val2][Dc]...
+      Sc: character to separate key values
+      DVs: character to separate key-value couples
+    >>> stringS_dictvar('i:1,iv:4,vii:7',',',':')
+    {'i': '1', 'vii': '7', 'iv': '4'}
+    """
+    fname = 'stringS_dictvar'
+
+    if stringv.count(Dc) != 0:
+        strvv = stringv.split(Dc)
+    else:
+        strvv = [stringv]
+
+    dictv = {}
+    for Svals in strvv:
+        keyn = Svals.split(DVs)[0]
+        valn = Svals.split(DVs)[1]
+        dictv[keyn] = valn
+
+    return dictv
 
 def set_attribute(ncvar, attrname, attrvalue):
@@ -311 +334 @@
           divc, divide by [val]
           rmchar,[val],[pos], remove [val] characters from [pos]='B', beginning, 'E', end
+          repl,[char1]:[val1];...;[charM]:[valM], replace a given character [chari] 
+            by a value [vali]
       NAMElon=name of the variable with the longitude (x position)
       NAMElat=name of the variable with the latitude (y position)
@@ -389 +414 @@
     fname = 'value_fmt'
 
-    aopers = ['sumc','subc','mulc','divc', 'rmchar']
+    aopers = ['sumc','subc','mulc','divc', 'rmchar', 'repl']
 
     fmts = ['D', 'F', 'I', 'I64', 'S']
@@ -434 +459 @@
                               " work with '" + op.split(',')[2] + "' !!"
                             quit(-1)
+
                 elif fmt == 'F':
                     opv = np.float(operv)
@@ -454 +480 @@
 
                 elif fmt == 'I':
-                    opv = int(operv)
-                    if opern == 'sumc': newval = int(val) + opv
-                    elif opern == 'subc': newval = int(val) - opv
-                    elif opern == 'mulc': newval = int(val) * opv
-                    elif opern == 'divc': newval = int(val) / opv
+                    if opern == 'sumc': newval = int(val) + int(operv)
+                    elif opern == 'subc': newval = int(val) - int(operv)
+                    elif opern == 'mulc': newval = int(val) * int(operv)
+                    elif opern == 'divc': newval = int(val) / int(operv)
                     elif opern == 'rmchar': 
                         opv = int(operv)
@@ -471 +496 @@
                               " work with '" + op.split(',')[2] + "' !!"
                             quit(-1)
+                    elif opern == 'repl':
+                        replvals = stringS_dictvar(operv, Dc=';', DVs=':')
+                        print replvals
+                        replaced = False
+                        for repls in replvals.keys():
+                            print operv, repls, operv == repls
+                            if operv == repls: 
+                                newval = int(replvals[operv])
+                                replaced = True
+                                break
+                        print operv, 'replaced', replaced
+                        if not replaced: newval = int(operv)
+                        
                 elif fmt == 'I64':
                     opv = np.int64(operv)

trunk/tools/generic_tools.py

@@ -10593 +10593 @@
     """
     fname = 'stringS_dictvar'
+
+    if stringv.count(Dc) != 0:
+        strvv = stringv.split(Dc)
+    else:
+        strvv = [stringv]
+
     dictv = {}
+    for Svals in strvv:
     for Svals in stringv.split(Dc):
         keyn = Svals.split(DVs)[0]