Changeset 2379 in lmdz_wrf

Timestamp:

Mar 7, 2019, 7:47:44 PM (6 years ago)

Author:

lfita

Message:

Adding Error message when passed a wrong type of slicing value in 'SliceVarDict?'

File:

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

trunk/tools/nc_var_tools.py

@@ -15389 +15389 @@
                 else:
                     slicevals.append(dictv)
+            else:
+                print errormsg
+                print '  ' + fname + ": type of value for slicing ", type(dictv),    \
+                  " not ready !!"
+                print '    available ones:', type([1]), type(int(1))
+                quit(-1)
         else:
             slicevals.append(slice(0,varshape[idim]))
@@ -30923 +30929 @@
 
     # Space Dimension variables
+    spacedims = [axisdims['Y'][0], axisdims['X'][0]]
     dnv = axisdims['X']
-    add_vars(onc,onewnc,[dnv[1]])
+    oxvar = onc.variables[dnv[1]]    
+    varslc, vard = SliceVarDict(oxvar,slicedims)
+    newvar = onewnc.createVariable(dnv[1], 'f', tuple(vard))
+    print 'oxvar dimensions:', oxvar.dimensions
+    print 'Lluis shapes: newvar', newvar.shape, 'slice', varslc, 'oxvar:', oxvar.shape, 'X', axisdims['X']
+    print 'Lluis slicedims:', slicedims
+    newvar[:] = oxvar[tuple(varslc)]
+    ncattrs = oxvar.ncattrs()
+    for attrn in ncattrs:
+        attrv = oxvar.getncattr(attrn)
+        newvar.setncattr(attrn,attrv)
+
     dnv = axisdims['Y']
-    add_vars(onc,onewnc,[dnv[1]])
+    oyvar = onc.variables[dnv[1]]    
+    varslc, vard = SliceVarDict(oyvar,slicedims)
+    newvar = onewnc.createVariable(dnv[1], 'f', tuple(vard))
+    newvar[:] = oyvar[tuple(varslc)]
+    ncattrs = oyvar.ncattrs()
+    for attrn in ncattrs:
+        attrv = oyvar.getncattr(attrn)
+        newvar.setncattr(attrn,attrv)
 
     # Variable
-    icvar = np.arange(np.prod(compressvar.shape))
+    icvar = np.arange(np.prod(compressvar.shape)).reshape(compressvar.shape)
     icc = icvar[~compressmask.mask]
 
-    newvar = onenc.createVariable(compressvar, 'i', ('compressvar'))
-    cfvarvals = gen.variablesvalues(compressvar)
+    newvar = onewnc.createVariable(compressvarn, 'i', (compressvarn))
+    cfvarvals = gen.variables_values(compressvarn)
     basicvardef(newvar, cfvarvals[1], cfvarvals[4], cfvarvals[5])
     newvar.setncattr('compress', compressS)
     newvar[:] = icc
+
+    onc.close()
+  
+    onewnc.sync()
+    onewnc.close()
     
+    print fname + ": successfully written file '" + fname + ".nc' !!"
     return
 
 
-values = 'XLAND:Time|0:single,1:X|west_east|XLONG_M,Y|south_north|XLAT_M'
+values = 'XLAND:Time|0:single,1:X|west_east|XLONG,Y|south_north|XLAT'
 
 compress_data(values, '/home/lluis/PY/wrfout_d01_1995-01-01_00:00:00', 'T2')