Changeset 895 in lmdz_wrf for trunk

Timestamp:

Jun 19, 2016, 1:51:04 PM (8 years ago)

Author:

lfita

Message:

Right variable dimension to add in `file_operation_alongdims'

File:

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

trunk/tools/nc_var_tools.py

@@ -9025 +9025 @@
     objnewnc = NetCDFFile(ofile, 'w')
 
+    # Dimensions which should appear at the final file (from the computed variables)
+    dimfinalfile = []
+
     print '  ' + fname + ": '" + operkind + "' with:",dimsoper,'_______'
     newNdim = 0
     newNvar = 0
-    for vn in varns + dimvn:
+    for vn in varns:
         print "'" + vn + "' ... .. ."
         if not objnc.variables.has_key(vn):
@@ -9036 +9039 @@
             quit(-1)
 
-        # Do we have to compute this variable?
+        # Do we have to compute this variable? Only to compute that variables with 
+        #   coincidence on the dimensions to operate
         ov = objnc.variables[vn]
         tocompute = False
@@ -9043 +9047 @@
 
         if tocompute:
+            # Dimensions which need to appear at the final file: that ones not opered
+            #   from all opered variables
+            for vd in ov.dimensions:
+                if not gen.searchInlist(dimfinalfile, vd) and                        \
+                 not gen.searchInlist(dimsoper, vd): dimfinalfile.append(vd)
+
             varslice, varsliceddims = SliceVarDict(ov,dimslice)
             varvals0 = ov[tuple(varslice)]
@@ -9121 +9131 @@
             objnewnc.sync()
     
+# operating variable dimension
+    print '  ' + fname + ': Opering slice of variable diension _______'
+    for vn in dimvn:
+        print "' " + vn + "' ... .. ."
+        if not objnc.variables.has_key(vn):
+            print warnmsg
+            print '  ' + fname + ': netCDF file "' + ncfile +                        \
+              '" does not have variable "' + vn + '" skipping it !!'
+            continue
+
+        # Do we have to compute this variable?
+        ov = objnc.variables[vn]
+        tocompute = False
+        for vd in ov.dimensions:
+            if gen.searchInlist(dimfinalfile, vd): tocompute = True
+
+        print fname + 'Lluis:', vn , tocompute, vd, dimfinalfile
+        if tocompute:
+            varslice, varsliceddims = SliceVarDict(ov,dimslice)
+            varvals0 = ov[tuple(varslice)]
+            varvals = np.squeeze(varvals0)
+
+            finalvarvals, finaldims = operdim(varvals, varsliceddims, dimsoper, operkind)
+
+            # Adding dimensions
+            idn = 0
+            for dn in finaldims:
+                if newNdim == 0:
+                    if not objnc.dimensions[dn].isunlimited():
+                        objnewnc.createDimension(dn, finalvarvals.shape[idn])
+                    else:
+                        objnewnc.createDimension(dn, None)
+                    newNdim = newNdim + 1
+                else:
+                    if not gen.searchInlist(objnewnc.dimensions, dn):
+                        if not objnc.dimensions[dn].isunlimited():
+                            objnewnc.createDimension(dn, finalvarvals.shape[idn])
+                        else:
+                            objnewnc.createDimension(dn, None)
+                        newNdim = newNdim + 1
+                idn = idn + 1
+
+            # Adding variable
+            varinf = variable_inf(ov)
+            if varinf.FillValue is not None:
+                newvar = objnewnc.createVariable(vn+operkind, nctype(varinf.dtype),  \
+                  tuple(finaldims), fillValue=varinf.FillValue)
+            else:
+                newvar = objnewnc.createVariable(vn+operkind, nctype(varinf.dtype),  \
+                  tuple(finaldims))
+                newNvar = newNvar + 1
+            newvar[:] = finalvarvals[:]
+
+# new variable
+            oldvarattr = ov.ncattrs()
+    
+            varname = gen.variables_values(vn)[0]
+            if gen.searchInlist(oldvarattr, 'standard_name'):
+                stdname = ov.getncattr('standard_name')
+            else:
+                stdname = gen.variables_values(vn)[1]
+        
+            if gen.searchInlist(oldvarattr, 'long_name'):
+                lname = ov.getncattr('long_name')
+            else:
+                lname = gen.variables_values(vn)[4].replace('|',' ')
+
+            if gen.searchInlist(oldvarattr, 'units'):
+                uname = ov.getncattr('units')
+            else:
+                uname = gen.variables_values(vn)[5]
+    
+            newattr = basicvardef(newvar, stdname + operkind, lname + ' '+ operkind +\
+              ' along ' + dimsoperS, uname)
+
+            for idim in ov.dimensions:
+                for dv in dimslice.keys():
+                    if dv == idim:
+                        if type(dimslice[dv]) == type([2]):
+                            Sdv = gen.numVector_Sring(dimslice[dv],', ') + ', 1'
+                        elif type(dimslice[dv]) == type(int(2)):
+                            if dimslice[dv] == -1:
+                                Sdv = str(len(objnc.dimensions[idim]))
+                            else:
+                                Sdv = str(dimslice[dv])
+                    else:
+                        Sdv = str(len(objnc.dimensions[idim]))
+ 
+                if idim == ov.dimensions[0]:
+                    origdimsS = idim + '(' + Sdv + ')'
+                else:
+                    origdimsS = origdimsS + ' ' + idim + '(' + Sdv + ')'
+
+            newattr = set_attribute(newvar, 'orig_dimensions', origdimsS)
+            objnewnc.sync()
+
 # Including operation dimensions on the new slice for the variables dimension
 #   but, which value along the operated dimension should be taken?
@@ -9130 +9236 @@
         sliceaddvars[cdn] = dimsliceval
 
-# variable dimension
+# slicing variable dimension
+    print '  ' + fname + ': Adding slice of variable diension _______'
     for vn in dimvn:
-        print "'" + vn + "' ... .. ."
+        print "' " + vn + "' ... .. ."
         if not objnc.variables.has_key(vn):
-            print errormsg
+            print warnmsg
             print '  ' + fname + ': netCDF file "' + ncfile +                        \
-              '" does not have variable "' + vn + '" !!'
-            quit(-1)
+              '" does not have variable "' + vn + '" skippping it!!'
 
         # Do we have to add this dimension-variable? (only that ones with same 
@@ -9144 +9250 @@
         toadd = False
         for vd in ov.dimensions:
-            if gen.searchInlist(dimsoper, vd): toadd = True
+            if gen.searchInlist(dimfinalfile, vd): toadd = True
    
         if toadd: