Changeset 921 in lmdz_wrf for trunk/tools

Timestamp:

Jun 21, 2016, 4:14:42 PM (8 years ago)

Author:

lfita

Message:

Adding last:' -9 and first:' 0 values in 'DataSetSection?'

File:

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

trunk/tools/nc_var_tools.py

@@ -5641 +5641 @@
         [beg],[end],[int]: beginning, end and interval along the dimension-axis
           [end] = -1, values until the end
+          [beg] = -9, last value
       filen= netCDF with the data-set
       DataSetSection('time,3,15,1','test.nc')
@@ -5671 +5672 @@
         else:
             if dims == dimn:
-                if begv > len(objdim):
-                    print errormsg
-                    print '  ' + fname + ': beginning value' ,begv,                  \
-                      'is larger than dimension (', len(objdim), ')!!!!'
-                    quit(-1)
-                if endv > len(objdim):
-                    print errormsg
-                    print '  ' + fname + ': endining value' ,endv,                   \
-                      'is larger than dimension (', len(objdim), ')!!!!'
-                    quit(-1)
-
-                if endv == -1: endv = len(objdim)
-                dimsize = (endv - begv + 1)/intv - 1
+                if begv != -9 and begv != 0:
+                    if begv > len(objdim):
+                        print errormsg
+                        print '  ' + fname + ': beginning value' ,begv,              \
+                          'is larger than dimension (', len(objdim), ')!!!!'
+                        quit(-1)
+                    if endv > len(objdim):
+                        print errormsg
+                        print '  ' + fname + ': endining value' ,endv,               \
+                          'is larger than dimension (', len(objdim), ')!!!!'
+                        quit(-1)
+
+                    if endv == -1: endv = len(objdim)
+                    dimsize = (endv - begv + 1)/intv - 1
+                else:
+                    dimsize == 0                       
+                    if begv == 0:
+                        begv = 0
+                        endv = 0
+                        intv = 0
+                    elif begv == -9:
+                        begv = len(objdim)
+                        endv = len(objdim)
+                        intv = 0
+
             else:
                 dimsize = len(objdim)
 
         print '    ' + fname + ': adding dimension: ' + dims + ' size:',dimsize
-        dim = ncoobj.createDimension(dims, dimsize)
+        if dimsize != 0:
+            if begv != -9 and begv != 0:
+               dim = ncoobj.createDimension(dims, dimsize)
+            else:
+               if dims != dimn:
+                   dim = ncoobj.createDimension(dims, dimsize)
 
     ncoobj.sync()
@@ -5701 +5719 @@
         if not gen.searchInlist(list(vardims),dimn):
             varvals = varobj[:]
+            filevardims = list(vardims)
         else:
             varslice = []
+            filevardims = []
             for dimname in varobj.dimensions:
                 if dimname == dimn:
-                    varslice.append(slice(begv,endv,intv))
+                    if begv != -9 and begv != 0:
+                        varslice.append(slice(begv,endv,intv))
+                        filevardims.append(dimname)
+                    else:
+                        if begv == -9:
+                            varslice.append(len(nciobj.dimensions[dimn])-1)
+                        elif begv == 0:
+                            varslice.append(0)
+
                 else:
                     Ldim = len(nciobj.dimensions[dimname])
                     varslice.append(slice(0,Ldim))
+                    filevardims.append(dimname)
 
             varvals = varobj[tuple(varslice)]
@@ -5721 +5750 @@
         vartype = varobj.dtype
 
-        newvar = ncoobj.createVariable(varns, vartype, vardims, fill_value=varfil)
+        newvar = ncoobj.createVariable(varns, vartype, tuple(filevardims),           \
+          fill_value=varfil)
         newvar[:] = varvals