Changeset 1379 in lmdz_wrf

Timestamp:

Dec 13, 2016, 1:32:29 PM (8 years ago)

Author:

lfita

Message:

Adding to 'addVar':

• 'scalar' as [dnames] for the addition of a scalar variable
• 'None' as [arguments] to avoid the addition of the standard attributes: standard_name, long_name, units

File:

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

trunk/tools/nc_var_tools.py

@@ -18151 +18151 @@
       values= [dnames]|[varattributes]|[kind]
         [dnames]: [dimn1],...,[dimnN], ',' list of name of dimensions for the new variable (already existing)
-        [attributes]: [std_name]@[long_name]@[units], standard name, long name and units of the variable
-          ('!' for spaces)
+          'scalar', for scalar variables
+        [attributes]: [std_name]@[long_name]@[units], standard name, long name and units of the variable ('!' for spaces)
+          'None' for no standard attributes
         [kind]: type of variable (standard netCDF4/C-like values, 'c', 'i', 'f', 'f8',...)
       ncfile= name of the file
@@ -18176 +18177 @@
     dimns = onc.dimensions
     varns = onc.variables.keys()
-
-    # Checking dimensions
-    dsize = []
-    for dimn in dnames:
-        if not gen.searchInlist(dimns, dimn):
-            print errormsg
-            print '  ' +fname+ ": file '" + ncfile + "' does not have dimension '" + \
-                dimn + "' !!"
-            print '    existing ones:', dimns
-            quit(-1)
-        else:
-            dsize.append(len(onc.dimensions[dimn]))
-
+    
     # Checking new varname
     if gen.searchInlist(varns, varn):
@@ -18197 +18186 @@
         quit(-1)
 
+    if values.split('|')[0].find(',') != -1:
+        # Checking dimensions
+        dsize = []
+        for dimn in dnames:
+            if not gen.searchInlist(dimns, dimn):
+                print errormsg
+                print '  ' +fname+ ": file '" + ncfile + "' does not have " +        \
+                  "dimension '" +  dimn + "' !!"
+                print '    existing ones:', dimns
+                quit(-1)
+            else:
+                dsize.append(len(onc.dimensions[dimn]))
+
 # Variable
-    if kind == 'c':
-        newvar = onc.createVariable(varn, 'c', tuple(dnames))
-#        newvar[:] = np.zeros(tuple(dsize), dtype=np.float)
-    elif kind == 'f' or kind == 'f4':
-        newvar = onc.createVariable(varn, 'f4', tuple(dnames),                       \
-          fill_value=gen.fillValueF)
-        newvar[:] = np.zeros(tuple(dsize), dtype=np.float)
-    elif kind == 'f8':
-        newvar = onc.createVariable(varn, 'f8', tuple(dnames),                       \
-          fill_value=gen.fillValueD)
-        newvar[:] = np.zeros(tuple(dsize), dtype=np.float64)
-    elif kind == 'i':
-        newvar = onc.createVariable(varn, 'i', tuple(dnames),                        \
-          fill_value=gen.fillValueI)
-        newvar[:] = np.zeros(tuple(dsize), dtype=int)
+        if kind == 'c':
+            newvar = onc.createVariable(varn, 'c', tuple(dnames))
+#            newvar[:] = np.zeros(tuple(dsize), dtype=np.float)
+        elif kind == 'f' or kind == 'f4':
+            newvar = onc.createVariable(varn, 'f4', tuple(dnames),                   \
+                  fill_value=gen.fillValueF)
+            newvar[:] = np.zeros(tuple(dsize), dtype=np.float)
+        elif kind == 'f8':
+            newvar = onc.createVariable(varn, 'f8', tuple(dnames),                   \
+              fill_value=gen.fillValueD)
+            newvar[:] = np.zeros(tuple(dsize), dtype=np.float64)
+        elif kind == 'i':
+            newvar = onc.createVariable(varn, 'i', tuple(dnames),                    \
+              fill_value=gen.fillValueI)
+            newvar[:] = np.zeros(tuple(dsize), dtype=int)
+        else:
+            print errormsg
+            print '  ' + fname + ": variable kind '" + kind + "' not ready!!"
+            quit(-1)
+
+    elif values.split('|')[0] == 'scalar':
+        print '  ' + fname + ': creation of scalar variable !!'
+
+# Variable
+        if kind == 'c':
+            newvar = onc.createVariable(varn, 'c')
+#            newvar[:] = np.zeros(tuple(dsize), dtype=np.float)
+        elif kind == 'f' or kind == 'f4':
+            newvar = onc.createVariable(varn, 'f4')
+            newvar[:] = np.float(0.)
+        elif kind == 'f8':
+            newvar = onc.createVariable(varn, 'f8')
+            newvar[:] = np.float64(0.)
+        elif kind == 'i':
+            newvar = onc.createVariable(varn, 'i')
+            newvar[:] = int(0)
+        else:
+            print errormsg
+            print '  ' + fname + ": variable kind '" + kind + "' not ready!!"
+            quit(-1)
     else:
         print errormsg
-        print '  ' + fname + ": variable kind '" + kind + "' not ready!!"
+        print '  ' + fname + ": wrong dimension values: '" + dnames + "' !!"
         quit(-1)
 
-    sname = attributes.split('@')[0]
-    lname = attributes.split('@')[1]
-    u = attributes.split('@')[2]
+    # Setting standard attributes
+    if values.split('|')[1] != 'None':
+        sname = attributes.split('@')[0]
+        lname = attributes.split('@')[1]
+        u = attributes.split('@')[2]
     
-    newattr = basicvardef(newvar, sname, lname, u)
+        newattr = basicvardef(newvar, sname, lname, u)
 
     onc.sync()
@@ -18247 +18276 @@
     if values == 'h':
         print fname + '_____________________________________________________________'
-        print addVar.__doc__
+        print setvar_nc.__doc__
         quit()