Changeset 810 in lmdz_wrf

Timestamp:

Jun 7, 2016, 5:36:20 PM (9 years ago)

Author:

lfita

Message:

`faddvar': Adding updating of variable if it already exists in the file

File:

• trunk/tools/nc_var_tools.py (modified) (1 diff)

trunk/tools/nc_var_tools.py

@@ -1173 +1173 @@
       print '    ' + fname + ': Adding refvar:', refvar, 'shape: ', refvarv.shape
       var = ncf.createVariable(refvar, vartype, vardims, fill_value=varfil)
-
-      varshape = refvarv.shape
-      varvals = np.zeros(tuple(varshape), dtype=vartype)      
-      Nvars = len(varshape)
+  else:
+      print warnmsg
+      print '    ' + fname + ': updating var:', refvar, 'shape: ', refvarv.shape
+      var = ncf.variables[refvar]
+
+  varshape = refvarv.shape
+  varvals = np.zeros(tuple(varshape), dtype=vartype)      
+  Nvars = len(varshape)
    
 # Allocating all the necessary memory (just in case)
+  var[:] = varvals
+
+  if Nvars <= 2:
+      varvals = refvarv[:]
       var[:] = varvals
-
-      if Nvars <= 2:
-          varvals = refvarv[:]
-          var[:] = varvals
-      elif Nvars == 3:
-          for i in range(varshape[0]):
-              varvals[i,:,:] = refvarv[i,:,:]
-              var[i,:,:] = varvals[i,:,:]
-      elif Nvars == 4:
-          for i in range(varshape[0]):
-              for j in range(varshape[1]):
-                  varvals[i,j,:,:] = refvarv[i,j,:,:]
-                  var[i,j,:,:] = varvals[i,j,:,:]
-      elif Nvars == 5:
-          for i in range(varshape[0]):
-              for j in range(varshape[1]):
-                  for k in range(varshape[2]):
-                      varvals[i,j,k,:,:] = refvarv[i,j,k,:,:]
-                      var[i,j,k,:,:] = varvals[i,j,k,:,:]
-      elif Nvars == 6:
-          for i in range(varshape[0]):
-              for j in range(varshape[1]):
-                  for k in range(varshape[2]):
-                      for l in range(varshape[3]):
-                          varvals[i,j,k,l,:,:] = refvarv[i,j,k,l,:,:]
-                          var[i,j,k,l,:,:] = varvals[i,j,k,l,:,:]
-
-      newvar = ncf.variables[refvar]
-      for attr in varattr:
-          newvarattrs = newvar.ncattrs()
-          attrv = refvarv.getncattr(attr)
-          if not gen.searchInlist(newvarattrs, attr):     
-              newvar.setncattr(attr, attrv)
-
-      ncf.sync()
+  elif Nvars == 3:
+      for i in range(varshape[0]):
+          varvals[i,:,:] = refvarv[i,:,:]
+          var[i,:,:] = varvals[i,:,:]
+  elif Nvars == 4:
+      for i in range(varshape[0]):
+          for j in range(varshape[1]):
+              varvals[i,j,:,:] = refvarv[i,j,:,:]
+              var[i,j,:,:] = varvals[i,j,:,:]
+  elif Nvars == 5:
+      for i in range(varshape[0]):
+          for j in range(varshape[1]):
+              for k in range(varshape[2]):
+                  varvals[i,j,k,:,:] = refvarv[i,j,k,:,:]
+                  var[i,j,k,:,:] = varvals[i,j,k,:,:]
+  elif Nvars == 6:
+      for i in range(varshape[0]):
+          for j in range(varshape[1]):
+              for k in range(varshape[2]):
+                  for l in range(varshape[3]):
+                      varvals[i,j,k,l,:,:] = refvarv[i,j,k,l,:,:]
+                      var[i,j,k,l,:,:] = varvals[i,j,k,l,:,:]
+
+  newvar = ncf.variables[refvar]
+  for attr in varattr:
+      newvarattrs = newvar.ncattrs()
+      attrv = refvarv.getncattr(attr)
+      if not gen.searchInlist(newvarattrs, attr):     
+          newvar.setncattr(attr, attrv)
+
+  ncf.sync()
   ncf.close()
   ncref.close()