Changeset 712 in lmdz_wrf for trunk

Timestamp:

Apr 25, 2016, 11:22:24 AM (9 years ago)

Author:

lfita

Message:

Imporoving `addvarattrk', by splitting when a string is provided and when a dtype output

Including also the 'unicode' type (assuming RTF-* encoding)

File:

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

trunk/tools/nc_var_tools.py

@@ -2026 +2026 @@
     attrname = name of the attribute
     attrvalue = value of the attribute
-    atrtrkind = kind of attribute: 'S', string ('!' as spaces); 'I', integer; 'Inp32', numpy integer 32;
-       'R', real; ' npfloat', np.float; 'D', np.float64
+    atrtrkind = kind of attribute: 'S', string ('!' as spaces); 'U', unicode ('!' as spaces); 'I', integer; 
+      'Inp32', numpy integer 32; 'R', real; ' npfloat', np.float; 'D', np.float64
     """
-
-    if attrkind == 'S':
-        attrvalue = str(attrval.replace('!', ' '))
-    elif attrkind == 'I':
-        attrvalue = int(attrval)
-    elif attrkind == 'Inp32':
-        attrvalue = np.int32(attrval)
-    elif attrkind == 'R':
-        attrvalue = float(attrval)
-    elif attrkind == 'npfloat':
-        attrvalue = np.float(attrval)
-    elif attrkind == 'D':
-        attrvalue = np.float64(attrval)
+    fname = 'set_attributek'
+    validk = {'S': 'string', 'U': 'unicode', 'I': 'integer',                         \
+      'Inp32': 'integer long (np.int32)', 'R': 'float', 'npfloat': 'np.float',       \
+      'D': 'float long (np.float64)'}
+
+    if type(attrkind) == type('s'):
+        if attrkind == 'S':
+            attrvalue = str(attrval.replace('!', ' '))
+        elif attrkind == 'U':
+            attrvalue = unicode(attrval.replace('!',' '))
+        elif attrkind == 'I':
+            attrvalue = np.int(attrval)
+        elif attrkind == 'R':
+            attrvalue = float(attrval)
+        elif attrkind == 'npfloat':
+            attrvalue = np.float(attrval)
+        elif attrkind == 'D':
+            attrvalue = np.float64(attrval)
+        else:
+            print errormsg
+            print '  ' + fname + ": '" + attrkind + "' kind of attribute is not ready!"
+            print '  valid values: _______'
+            for key in validk.keys():
+                print '    ', key,':', validk[key]
+            quit(-1)
     else:
-        print errormsg
-        print '    set_attributek: "' + attrkind + '" kind of attribute is not ready!'
-        quit(-1)
+        if attrkind == type(str('a')):
+            attrvalue = str(attrval.replace('!', ' '))
+        elif attrkind == type(unicode('a')):
+            attrvalue = unicode(attrval.replace('!',' '))
+        elif attrkind == type(np.int(1)):
+            attrvalue = np.int(attrval)
+        elif attrkind == np.dtype('i'):
+            attrvalue = np.int32(attrval)
+        elif attrkind == type(float(1.)):
+            attrvalue = float(attrval)
+        elif attrkind == type(np.float(1.)):
+            attrvalue = np.float(attrval)
+        elif attrkind == np.dtype('float32'):
+            attrvalue = np.array(attrval, dtype='f')
+        elif attrkind == type(np.float32(1.)):
+            attrvalue = np.float32(attrval)
+        elif attrkind == type(np.float64(1.)):
+            attrvalue = np.float64(attrval)
+        else:
+            print errormsg
+            print '  ' + fname + ": '" + attrkind + "' kind of attribute is not ready!"
+            print '  valid values: _______'
+            for key in validk.keys():
+                print '    ', key,':', validk[key]
+            quit(-1)
 
     attvar = ncv.ncattrs()
@@ -2052 +2086 @@
 
     attr = ncv.setncattr(attrname, attrvalue)
-    
+        
     return attr