Changeset 1491 in lmdz_wrf

Timestamp:

Apr 4, 2017, 3:11:30 PM (8 years ago)

Author:

lfita

Message:

Adding in `setvar_asciivalues':

• Abailability to read a 3D field
• '#' As commentary character

File:

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

trunk/tools/nc_var_tools.py

@@ -7605 +7605 @@
     """ Function to set de values of a variable with an ASCII file (common 
           Fortran-like format)
-        values= asciifile: ASCII file with a Fortran-like structure
+        values= asciifile: ASCII file with a Fortran-like structure ('#' for commentaries)
           [dim1]_1 [[dim2]_1 ... [dim2]_M]
           ...
@@ -7651 +7651 @@
         valfinal = np.zeros((1), dtype=vartype)
         for line in objasciif:
-            val = (line.replace('\n','').replace(' ','').replace('\t',''))
-            if vartype != '|S1':
-                valfinal[0] = gen.retype(val, vartype)
-                if it <= values.shape[0]-1:
-                    values[it] = gen.retype(valfinal[0], vartype)
-                    it = it +1
-            else:
-                for i1 in range(len(val)):
-                    values[it] = val.replace('!',' ')
+            if line[0:1] != '#':
+                val = (line.replace('\n','').replace(' ','').replace('\t',''))
+                if vartype != '|S1':
+                    valfinal[0] = gen.retype(val, vartype)
+                    if it <= values.shape[0]-1:
+                        values[it] = gen.retype(valfinal[0], vartype)
+                        it = it +1
+                else:
+                    for i1 in range(len(val)):
+                        values[it] = val.replace('!',' ')
     elif Ndims == 2:
         iline=0
         for line in objasciif:
-            vals = line.replace('\n','').replace('\t','').split(' ')
-            if vartype != '|S1':
-                if len(vals) != values.shape[1]:
-                    print errormsg
-                    print '  ' + fname + ': given: ',len(vals),' but variable ' +    \
-                      'requires: ',values.shape[1],'!!!'
-                    exit(-1)
-                for i1 in range(len(vals)):
-                    values[iline,i1] = gen.retype(vals[i1], vartype)
-            else:
-                for i1 in range(len(vals)):
-                    values[iline,i1] = vals[i1].replace('!',' ')
-            iline=iline+1
+            if line[0:1] != '#':
+                vals = line.replace('\n','').replace('\t','').split(' ')
+                if vartype != '|S1':
+                    if len(vals) != values.shape[1]:
+                        print errormsg
+                        print '  ' + fname + ': given: ',len(vals),' but variable '+ \
+                          'requires: ',values.shape[1],'!!!'
+                        exit(-1)
+                    for i1 in range(len(vals)):
+                        values[iline,i1] = gen.retype(vals[i1], vartype)
+                else:
+                    for i1 in range(len(vals)):
+                        values[iline,i1] = vals[i1].replace('!',' ')
+                iline=iline+1
+    elif Ndims == 3:
+        d0 = objvar.shape[0]
+        d1 = objvar.shape[1]
+        d2 = objvar.shape[2]
+        iline=0
+        icol=0
+        for line in objasciif:
+            if line[0:1] != '#' and len(line) > 2:
+                vals = line.replace('\n','').replace('\t','').split(' ')
+                if vals.count('') != 0: vals.remove('')
+                if vartype != '|S1':
+                    if len(vals) != values.shape[2]:
+                        print vals
+                        print errormsg
+                        print '  ' + fname + ': given: ',len(vals),' but variable '+ \
+                          'requires: ',values.shape[2],'!!!'
+                        exit(-1)
+                    for i1 in range(len(vals)):
+                        values[icol,iline,i1] = gen.retype(vals[i1], vartype)
+                else:
+                    for i1 in range(len(vals)):
+                        values[icol,iline,i1] = vals[i1].replace('!',' ')
+                iline=iline+1
+                if iline > d1-1:
+                    iline = 0
+                    icol = icol + 1
     else:
         print errormsg
-        print '  ' + fname + ': number of dimensions',Ndims,'not ready !!!!'
+        print '  ' + fname + ': number of dimensions', Ndims, 'not ready !!!!'
         quit(-1)