Changeset 2383 in lmdz_wrf

Timestamp:

Mar 7, 2019, 10:06:44 PM (6 years ago)

Author:

lfita

Message:

Starting to add `CFfile_creation'

File:

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

trunk/tools/nc_var_tools.py

@@ -30960 +30960 @@
     newvar = onewnc.createVariable(compressvarn, 'i', (compressvarn))
     cfvarvals = gen.variables_values(compressvarn)
-    basicvardef(newvar, cfvarvals[1], cfvarvals[4], cfvarvals[5])
+    basicvardef(newvar, cfvarvals[1], cfvarvals[4].replace('|',' '), cfvarvals[5])
     newvar.setncattr('compress', compressS)
     newvar.setncattr('criteria', criteriaS)
@@ -31064 +31064 @@
 #compress_data(values, '/home/lluis/PY/wrfout_d01_1995-01-01_00:00:00', 'T2,LU_INDEX')
 
+
+def CFfile_creation(values, ncfile, variables):
+    """ Operation to create a file folowing CF-conventions 
+      values= [dimensions]:[lonlatbox]
+        [dimensions]: lon|[dsizelon],lat|[dsizelat],...,[dimnN]:[dsizeN], ',' pairs of variable 
+          name [dimn] and [size]
+          if [dsize] = 'None' (for UNLIMITED), give a third value with the real size
+        [lonlatbox]: ',' separated list of the extremes of the domain [SWlon],[SWlat],
+          [NElon],[NElat] or 'global' for a global file [assuing lon/lat regular 
+          projection]
+      ncfile= name of the file
+      variables= ',' list of variables to create [varn]:[dims]:[kind]:[varaddattributes] 
+          and its characterisitcs
+        [varn]: CF name of the variable (attributes will be retrived from 
+           'variables_values.dat')
+        [dims]: ',' python sorted list of dimensions of the variable
+        [kind]: type of variable (standard netCDF4/C-like values, 'c', 'i', 'f', 
+          'f8',...)
+        [varaddattributes]: ',' list of additional attributes and their values for 
+          the variable [attrn]|[attrv]
+    """
+    fname = 'CFfile_creation'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print CFfile_creation.__doc__
+        quit()
+
+    expectargs = '[dimensions]:[lonlatbox]'
+ 
+    gen.check_arguments(fname,values,expectargs,':')
+
+    dimensions = gen.stringS_dictvar(values.split(':')[0], ',', '|')
+    lonlatbox = values.split(':')[1]
+
+    if not dimensions.has_key('lon') or not dimensions.has_key('lat'):
+        print errormsg
+        print '  ' + fname + ": function does not deal with CF-files without 'lon' "+\
+          "and 'lat' dimensions !!"
+        print "    no value provide through variable [dimensions]= 'lon|[dimx]," +   \
+          "lat|[dimy]"
+        quit(-1)
+
+    onewnc = NetCDFFile(ncfile, 'w')
+
+    dimx = dimensions['lon']
+    dimy = dimensions['lat']
+
+    # Assuming regular lon/lat projection
+    if lonlatbox == 'global':
+        ddx = 360./(dimx-1)
+        ddx2 = ddx/2.
+        lons = np.arange(-180.+ddx2,180.+ddx2,ddx)
+        ddy = 180./(dimy-1)
+        ddy2 = ddy/2.
+        lats = np.arange(-90.+ddy2,90.+ddy2,ddy)
+    elif lonlatbox.count(',') == 3:
+        [lonSW, latSW, lonNE, latNE] = str_listk(lonlatbox, ',', 'F')
+        ddx = (lonNE-lonSW)/(dimx-1)
+        ddx2 = ddx/2.
+        lons = np.arange(lonSW-ddx2,lonNE+ddx2,ddx)
+        ddy = (latNE-latSW)/(dimy-1)
+        ddy2 = ddy/2.
+        lats = np.arange(latSW-ddy2,latNE+ddy2,ddy)
+
+    # Sorting lon/lat_bounds
+    lon_bnds = np.zeros((2,dimx), dtype=np.float64)
+    lat_bnds = np.zeros((2,dimy), dtype=np.float64)
+    for i in range(dimx):
+        lon_bnds[0,i] = lons[i]-ddx2
+        lon_bnds[1,i] = lons[i]+ddx2
+    for j in range(dimy): 
+        lat_bnds[0,j] = lats[j]-ddy2
+        lat_bnds[1,j] = lats[j]+ddy2
+
+    # Dimensions
+    newdim = onewnc.createDimension('lon', dimx)
+    newdim = onewnc.createDimension('lat', dimy)
+    newdim = onewnc.createDimension('bnds', 2)
+
+    # Additional dimensions
+    for dn in dimensions:
+        if dn != 'lon' and dn != 'lat':
+            if dn == 'time': dl = None
+            else: dl = dimensions[dn]
+            newdim = onewnc.createDimension(dn, None)
+    onewnc.sync()
+
+    # Creation of variable dimensions
+    newvar = onewnc.createVariable('lon', 'f4', ('lon'))
+    newvar[:] = lons[:]
+    cfvarvals = gen.variables_values('lon')
+    basicvardef(newvar, cfvarvals[1], cfvarvals[4].replace('|',' '), cfvarvals[5])
+    newvar.setncattr('axis', 'X')
+    newvar.setncattr('_CoordinateAxisType', 'Lon')
+
+    newvar = onewnc.createVariable('lat', 'f4', ('lat'))
+    newvar[:] = lats[:]
+    cfvarvals = gen.variables_values('lat')
+    basicvardef(newvar, cfvarvals[1], cfvarvals[4].replace('|',' '), cfvarvals[5])
+    newvar.setncattr('axis', 'Y')
+    newvar.setncattr('_CoordinateAxisType', 'Lat')
+
+    onewnc.sync()
+
+    varns = variables.split(',')
+    for varn in varns:
+        print varn
+
+    dnames = []
+    dsize = []
+    for dim in dimensions:
+#        print "  Adding: '" + dim + "' ..."
+        dimn = dim.split(':')[0]
+        dimv = dim.split(':')[1]
+        if dimv == 'None':
+            if len(dim.split(':')) != 3:
+                print errormsg
+                print '  ' + fname + ": dimension '" + dimn + "' is None but the " + \
+                  'size is requried!!'
+                quit(-1)
+            else:
+                dv = None
+                dsize.append(int(dim.split(':')[2]))
+        else:
+            dv = int(dimv)
+            dsize.append(dv)
+
+        dnames.append(dimn)
+
+        newdim = onc.createDimension(dimn, dv)
+    
+    onc.sync()
+
+# Variable
+    if kind == 'c':
+        newvar = onc.createVariable(varn, 'c', tuple(dnames))
+#        newvar[:] = np.zeros(tuple(dsize), dtype=np.float)
+    if kind == 'f' or kind == 'f4':
+        newvar = onc.createVariable(varn, 'f4', tuple(dnames),                       \
+          fill_value=gen.fillValueF)
+        newvar[:] = np.zeros(tuple(dsize), dtype=np.float)
+    elif kind == 'f8':
+        newvar = onc.createVariable(varn, 'f8', tuple(dnames),                       \
+          fill_value= gen.fillValueD)
+        newvar[:] = np.zeros(tuple(dsize), dtype=np.float64)
+    elif kind == 'i':
+        newvar = onc.createVariable(varn, 'i', tuple(dnames),                        \
+          fill_value=gen.fillValueI)
+        newvar[:] = np.zeros(tuple(dsize), dtype=int)
+    else:
+        print errormsg
+        print '  ' + fname + ": variable kind '" + kind + "' not ready!!"
+        quit(-1)
+
+    sname = attributes.split('@')[0].replace('!', ' ')
+    lname = attributes.split('@')[1].replace('!', ' ')
+    u = attributes.split('@')[2].replace('!', ' ')
+    
+    newattr = basicvardef(newvar, sname, lname, u)
+
+    onc.sync()
+
+# Global attributes
+    newattr = set_attribute(onc, 'description', "file creation using " + fname )
+    onc.sync()
+    onc.close()
+
+#file_creation('time_counter:12,sizes:24|time@time@seconds since 1949-12-01 00:00:00|f8', 'test.nc', 't_instant')
+
+
+
 #quit()