Changeset 887 in lmdz_wrf for trunk

Timestamp:

Jun 18, 2016, 10:03:26 PM (8 years ago)

Author:

lfita

Message:

adding `generalNCvariable': Function fake netCDF varibale

File:

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

trunk/tools/nc_var_tools.py

@@ -49 +49 @@
 # gaddattr: Add a global attribute to a netCDF. Removes previous attribute if it exist
 # gaddattrk: Add a global attribute to a netCDF caring about the type. Removes previous attribute if it exist
+# generalNCvariable: Function fake netCDF varibale
 # get_1str_nc: Function to get 1 string value in a netCDF variable as a chain of 1char values
 # get_attribute: Function to get an attribute from a netCDF file
@@ -8797 +8798 @@
     return operedvals, opereddims
 
+class generalNCvariable(object):
+    """ Function fake netCDF varibale
+      d[x/y/z/t]: dimensions of the variable
+      dn[x/y/z/t]: name of the dimensions of the variable
+    """
+    def __init__(self,dx,dy,dz,dt,dnx,dny,dnz,dnt):
+        totshape = dt*dz*dy*dx
+        dimensions = {}
+        
+        self.values = np.arange((totshape), dtype=np.float).reshape(dt,dz,dy,dx)
+        self.dimensions = {dnx: dx, dny: dy, dnz: dz, dnt: dt}
+        self.shape = (dt, dz, dy, dx)
+        self.standard_name = 'general_var'
+        self.long_name = 'generic netcdf variable'
+        self.units = '-'
+
+    def __getitem__(self, val):
+        return self.values[val]
+
+#variable = generalNCvariable(5,4,3,2,'lon','lat','pres','time')
+
 def operation_alongdims(ovar,dimvals,dimsoper,opkind):
     """ Function to operate along different dimensions of a variable
@@ -8982 +9004 @@
             end = int(dvv.slice(':')[1])
             dimslice[dnv] = [beg, end, 1]
-        else
+        else:
             dimslice[dnv] = int(dvv)
 
@@ -9034 +9056 @@
             varinf = variable_inf(ov)
             if varinf.FillValue is not None:
-                 newvar = objnewnc.createVariable(vn + oper, nctype(varinf.dtype),   \
-                   tuple(finaldims), fillValue=varinf.FillValue)
+                newvar = objnewnc.createVariable(vn + oper, nctype(varinf.dtype),    \
+                  tuple(finaldims), fillValue=varinf.FillValue)
             else:
-                 newvar = objnewnc.createVariable(vn + oper, nctype(varinf.dtype),   \
-                   tuple(finaldims))
+                newvar = objnewnc.createVariable(vn + oper, nctype(varinf.dtype),    \
+                  tuple(finaldims))
                 newNvar = newNvar + 1
 
@@ -13019 +13041 @@
                     slicevals.append(slice(0,varshape[idim]))
                     slicefinaldims.append(dim)
-                else
+                else:
                     slicevals.append(dictv)
         else: