Changeset 875 in lmdz_wrf for trunk

Timestamp:

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

Author:

lfita

Message:

Generalizing `file_oper_alongdims' now only the variable dimensions with some dimension are added

File:

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

trunk/tools/nc_var_tools.py

@@ -8981 +8981 @@
                     newdimvarv, newdimvardimns=operation_alongdims(objvdim,dimvals,dimsoper,operkind)
 
+# Introducing only that dimension variables with any dimension after the operation
+                    if len(newdimvardimns) > 0:
+
 # Old method introducing new dimension if needed
 ##                    dimsvdim = objvdim.dimensions
@@ -9005 +9008 @@
 ##                            dimvdims.append(dimsvdim[idmvd])
 
-                    newvar = objnewnc.createVariable(vardims[ivdim], vdimtype,       \
-                      tuple(newdimvardimns))
-                    newvar[:] = newdimvarv
-                    dimvattrs = objvdim.ncattrs()
-                    for attrn in dimvattrs:
-                        attrv = objvdim.getncattr(attrn)
-                        newattr = set_attribute(newvar,attrn,attrv)
-
-                    newattr = set_attribute(newvar, 'operation:',  operkind +        \
-                      ' along ' + dimsoperS)
+                        newvar = objnewnc.createVariable(vardims[ivdim], vdimtype,   \
+                          tuple(newdimvardimns))
+                        newvar[:] = newdimvarv
+                        dimvattrs = objvdim.ncattrs()
+                        for attrn in dimvattrs:
+                            attrv = objvdim.getncattr(attrn)
+                            newattr = set_attribute(newvar,attrn,attrv)
+
+                        newattr = set_attribute(newvar, 'operation:',  operkind +    \
+                          ' along ' + dimsoperS)
 
 # new variable
             oldvarattr = objvar.ncattrs()
     
-            varname = variables_values(vn)[0]
+            varname = gen.variables_values(vn)[0]
             if gen.searchInlist(oldvarattr, 'standard_name'):
                 stdname = objvar.getncattr('standard_name')
             else:
-                stdname = variables_values(vn)[1]
+                stdname = gen.variables_values(vn)[1]
         
             if gen.searchInlist(oldvarattr, 'long_name'):
                 lname = objvar.getncattr('long_name')
             else:
-                lname = variables_values(vn)[4].replace('|',' ')
+                lname = gen.variables_values(vn)[4].replace('|',' ')
 
             if gen.searchInlist(oldvarattr, 'units'):
                 uname = objvar.getncattr('units')
             else:
-                uname = variables_values(vn)[5]
+                uname = gen.variables_values(vn)[5]
     
             newvar = objnewnc.createVariable(varname + operkind, 'f4', 
@@ -9042 +9045 @@
             newvar[:] = newvarv
             for idim in range(len(objvar.shape)):
-                dimn = dimvals.split('|')[idim].split(':')[0]
+                if dimvals.find('|') != -1:
+                    dimn = dimvals.split('|')[idim].split(':')[0]
+                else:
+                    dimn = dimvals.split(':')[0]
                 dimv = '{:d}'.format(len(objnc.dimensions[dimn]))
                 if idim == 0: