Changeset 1354 in lmdz_wrf

Timestamp:

Nov 25, 2016, 6:24:18 PM (8 years ago)

Author:

lfita

Message:

Fixing variable-dimension's lack in `reproject' and variable definition

File:

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

trunk/tools/nc_var_tools.py

@@ -4931 +4931 @@
       'newlatdim]|[oldlonname,oldlatname]|[oldlondim,oldlatdim]'
  
-    check_arguments(fname,values,expectargs,'|')
+    gen.check_arguments(fname,values,expectargs,'|')
 
     newprojectionfile = values.split('|')[0]
@@ -4944 +4944 @@
         ofile = 'remapnn_allvars.nc'
 
-    filexist(filename, errormsg, 'old file')
-    filexist(newprojectionfile, errormsg, 'new file')
+    gen.filexist(filename, errormsg, 'old file')
+    gen.filexist(newprojectionfile, errormsg, 'new file')
 
     oldlon=oldlonlatname.split(',')[0]
@@ -4961 +4961 @@
     varinfile(newncfile, newprojectionfile, errormsg, 'new', newlat)
 
-    oldlon2d, oldlat2d = lonlat2D(oldncfile.variables[oldlon],                       \
+    oldlon2d, oldlat2d = gen.lonlat2D(oldncfile.variables[oldlon],                   \
       oldncfile.variables[oldlat])
-    newlon2d, newlat2d = lonlat2D(newncfile.variables[newlon],                       \
+    newlon2d, newlat2d = gen.lonlat2D(newncfile.variables[newlon],                   \
       newncfile.variables[newlat])
 
@@ -4991 +4991 @@
               (newlat2d[jnew, inew] - oldlat2d10)**2.)
             mindist = np.min(dist)
-            minydist10, minxdist10 = index_mat(dist,mindist)
+            minydist10, minxdist10 = gen.index_mat(dist,mindist)
             firstmappairs[:,jnew,inew] = [int(dy10*(1+minydist10)),                  \
               int(dx10*(1+minxdist10))]
@@ -16391 +16391 @@
               varn + "' !!"
             quit(-1)
-        print '  ' + fname + ": re-projecting '" + varn + "' ..."
         ovarin = inf.variables[varn]
         if flipy:
@@ -16408 +16407 @@
         varint = varin.transpose()
 
+        # Variable's dimensions and their related variables
+        vdims = list(ovarin.dimensions)
+        for dimn in vdims:
+            dimsfile = onewnc.dimensions.keys()
+            if not gen.searchInlist(dimsfile, dimn):
+                odimn = inf.dimensions[dimn]
+                if odimn.isunlimited():
+                    newdim = onewnc.createDimension(dimn, None)
+                else:
+                    newdim = onewnc.createDimension(dimn, len(odimn))
+                if not vardimns.has_key(dimn):
+                    print errormsg
+                    print '  ' + fname + ": dimension '" + dimn + "' does not have "+\
+                      'assigned any variable !!'
+                    print '    assigned pairs (via [dimvars]) _______'
+                    gen.printing_dictionary(vardimns)
+                    quit(-1)
+                else:
+                    add_vars(inf,onewnc,[vardimns[dimn]])
+
+        # Interpolated variable
+        vardims = list(ovarin.dimensions)
+        newvardims = gen.replace_list(vardims,dimvarns[inlonn],'lon')
+        newvardims = gen.replace_list(newvardims,dimvarns[inlatn],'lat')
+        varattrs = ovarin.ncattrs()
+        if gen.searchInlist(varattrs,'_FillValue'):        
+            fillv = ovarin.getncattr('_FillValue')
+            varout = np.where(varout == fillv, gen.fillValueF, varout)
+            varattrs.remove('_FillValue')
+        newvar = onewnc.createVariable(varn, 'f4', tuple(newvardims),            \
+          fill_value=gen.fillValueF)
+        onewnc.sync()
+
         # Do we have masked values?
         if type(varin) == type(ma.asarray([1,1])):
             inmask = np.where(varin.mask, 0, 1)
+            print '  ' + fname + ": re-projecting masked '" + varn + "'",           \
+              varin.shape, " ..."
         else:
             inmask = np.ones((varin.shape), dtype=int)
+            print '  ' + fname + ": re-projecting '" +varn+ "'", varin.shape," ..."
         inmaskt = inmask.transpose()
 
@@ -16442 +16477 @@
         varout = varoutt.transpose()
         varout = np.where(varout >= 0.9*gen.fillValueF, gen.fillValueF, varout)
-
-        # Variable's dimensions and their related variables
-        vdims = list(ovarin.dimensions)
-        for dimn in vdims:
-            dimsfile = onewnc.dimensions.keys()
-            if not gen.searchInlist(dimsfile, dimn):
-                odimn = inf.dimensions[dimn]
-                if odimn.isunlimited():
-                    newdim = onewnc.createDimension(dimn, None)
-                else:
-                    newdim = onewnc.createDimension(dimn, len(odimn))
-                if not vardimns.has_key(dimn):
-                    print errormsg
-                    print '  ' + fname + ": dimension '" + dimn + "' does not have "+\
-                      'assigned any variable !!'
-                    print '    assigned pairs (via [dimvars]) _______'
-                    gen.printing_dictionary(vardimns)
-                    quit(-1)
-                else:
-                    add_vars(inf,onewnc,[vardimns[dimn]])
-
-        # Interpolated variable
-        vardims = list(ovarin.dimensions)
-        newvardims = gen.replace_list(vardims,dimvarns[inlonn],'lon')
-        newvardims = gen.replace_list(newvardims,dimvarns[inlatn],'lat')
-        varattrs = ovarin.ncattrs()
-        if gen.searchInlist(varattrs,'_FillValue'):        
-            fillv = ovarin.getncattr('_FillValue')
-            varout = np.where(varout == fillv, gen.fillValueF, varout)
-            varattrs.remove('_FillValue')
-        newvar = onewnc.createVariable(varn, 'f4', tuple(newvardims),            \
-          fill_value=gen.fillValueF)
 
         newvar[:] = varout[:]