Changeset 3629 for palm/trunk/SCRIPTS/palm_csd_files/palm_csd_netcdf_interface.py

Timestamp:

Dec 13, 2018 12:18:54 PM (5 years ago)

Author:

maronga

Message:

palm_csd improvements

File:

• palm/trunk/SCRIPTS/palm_csd_files/palm_csd_netcdf_interface.py (modified) (3 diffs)

palm/trunk/SCRIPTS/palm_csd_files/palm_csd_netcdf_interface.py

@@ -25 +25 @@
 # -----------------
 # $Id$
+# Some new routines
+# 
+# 3567 2018-11-27 13:59:21Z maronga
 # Initial revisions
-#
-#
-# 
-#
 #
 # Description:
@@ -76 +75 @@
    nc_file = Dataset(filename, "r+", format="NETCDF4")
    tmp_array = np.array(nc_file.variables[varname][:][:], dtype=type(nc_file.variables[varname]))
+
+   return tmp_array
+
+def nc_read_from_file_3d_all(filename, varname):
+
+
+   import numpy as np
+   import sys
+   
+   try:
+      f = open(filename)
+      f.close()
+#      print("Load: " + filename + ".")
+   except FileNotFoundError:
+      print("Error: " + filename + ". No such file. Aborting...")
+      sys.exit(1)
+   
+   nc_file = Dataset(filename, "r+", format="NETCDF4")
+   tmp_array = np.array(nc_file.variables[varname][:][:][:], dtype=type(nc_file.variables[varname]))
 
    return tmp_array
@@ -212 +230 @@
    return 0
 
+def nc_overwrite_to_file_3d(filename,varname,array):
+
+   try:
+      f =  Dataset(filename, "a", format="NETCDF4")
+      #print("Opened: " + filename + ".")
+   except FileNotFoundError:
+      print("Error. Could not open file: " + filename + ". Aborting...")
+      sys.exit(1)
+
+   print("Writing array " + varname + " to file...")
+   
+   temp = f.variables[varname]
+   temp[:,:,:] = array
+   
+   f.close()
+
+   return 0
+
 def nc_write_to_file_3d(filename,varname,array,datatype,dimname1,dimname2,dimname3,fillvalue):