Changeset 674 in lmdz_wrf for trunk

Timestamp:

Jan 18, 2016, 4:37:28 PM (9 years ago)

Author:

lfita

Message:

Adding 'variables' in `DataSetSection_multidims'

Location:

trunk/tools

Files:

• nc_var.py (modified) (1 diff)
• nc_var_tools.py (modified) (4 diffs)

trunk/tools/nc_var.py

@@ -133 +133 @@
     ncvar.DataSetSection(opts.values, opts.ncfile)
 elif oper == 'DataSetSection_multidims':
-    ncvar.DataSetSection_multidims(opts.values, opts.ncfile)
+    ncvar.DataSetSection_multidims(opts.values, opts.ncfile, opts.varname)
 elif oper == 'DataSetSection_multivars':
     ncvar.DataSetSection_multivars(opts.values, opts.ncfile, opts.varname)

trunk/tools/nc_var_tools.py

@@ -8766 +8766 @@
     return
 
-def DataSetSection_multidims(values, filen):
+def DataSetSection_multidims(values, filen, varn):
     """ Function to get a section (values along multiple dimensions) of a given data-set
       values= [dimn1],[beg1],[end1],[int1]@[...[[dimnM],[begM],[endM],[intM]]]
@@ -8774 +8774 @@
           NOTE: dimensions without values are taken allover their size
       filen= netCDF with the data-set
-      DataSetSection_multidims('Time,-1,-1,1@bottom_top,6,6,1','wrfout_d01_1979-12-01_00:00:00')
+      varn= ',' list of variables, 'all' for all variables in data-set
+      DataSetSection_multidims('Time,-1,-1,1@bottom_top,6,6,1','wrfout_d01_1979-12-01_00:00:00','all')
     """
     fname = 'DataSetSection_multidims'
@@ -8835 +8836 @@
 
             if begvs[dimid] != endvs[dimid]:
-                dimsize = (endvs[dimid]-begvs[dimid]+1)/intvs[dimid]
-                dimslices[dims] = [begvs[dimid],endvs[dimid]+1,intvs[dimid]]
+                if endvs[dimid] != -1:
+                    dimsize = (endvs[dimid]-begvs[dimid]+1)/intvs[dimid]
+                    dimslices[dims] = [begvs[dimid],endvs[dimid]+1,intvs[dimid]]
+                else:
+                    print 'here!'
+                    dimsize = len(objdim)
+                    dimslices[dims] = [begvs[dimid],len(objdim),1]
             else:
                 dimsize = 1
@@ -8856 +8862 @@
 ## 
     print '  ' + fname + ': adding variables...'
-    for varns in nciobj.variables:
-#        print '    ' + fname + ': adding variable "' + varns + '"...'
+    if varn == 'all':
+        getvarns = nciobj.variables
+    else:
+        getvarns = varn.split(',')
+
+    for varns in getvarns:
+        print '    ' + fname + ': adding variable "' + varns + '"...'
         varobj = nciobj.variables[varns]