Changeset 1892 in lmdz_wrf for trunk

Timestamp:

Apr 4, 2018, 10:14:04 PM (7 years ago)

Author:

lfita

Message:

Adding:

• `genericNCvariable_Dict': Class to fake a netCDF varibale using dictionary values for
• rlon', rlat' to 'CFvars'

Fixing various issues related to `CFmorzization'

Location:

trunk/tools

Files:

• generic_tools.py (modified) (2 diffs)
• nc_var_tools.py (modified) (11 diffs)

trunk/tools/generic_tools.py

@@ -12723 +12723 @@
      'vdimn': 'time', 'axis': 'T', 'length': -1}
     """ 
-    fname = 'CFDIMvalues'
+    fname = 'CFcorValues'
 
     # Dictionary for each dimension
@@ -12748 +12748 @@
 
     # names to search for
-    lonn = ['lon', 'X', 'Longitude', 'longitude', 'x', 'west_east']
-    latn = ['lat', 'Y', 'Latitude', 'latitude', 'y', 'south_north']
+    lonn = ['lon', 'X', 'Longitude', 'longitude', 'x', 'west_east', 'rlon']
+    latn = ['lat', 'Y', 'Latitude', 'latitude', 'y', 'south_north', 'rlat']
     pressn = ['pres', 'press', 'P', 'Press', 'Pressure', 'pressure', 'p']
     heightn = ['height', 'Z', 'Height', 'z']

trunk/tools/nc_var_tools.py

@@ -75 +75 @@
 # 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
+# genericNCvariable_Dict: Class to fake a netCDF varibale using dictionary values for 
+#   dimensions
 # genericNCvariable: Class to fake a netCDF varibale
 # get_1str_nc: Function to get 1 string value in a netCDF variable as a chain of 1char values
@@ -195 +197 @@
 # Radius of the Earth (m)
 EarthR = 6378000.
+
+class genericNCvariable_Dict(object):
+    """ Class to fake a netCDF varibale using dictionary values for dimensions
+      dicdim: dictionary with name of the dimension as key and its length as valu
+      dimlist: list of dimensions to give the shape of the variable
+      name: name of the variable
+      stdname: standard_name of the variable
+      longname: longname of the variable
+      units: units of the variable
+      values: values for the variable
+    >>> variable = genericNCvariable_Dict({'lon':25, 'lat':30, 'time': 2}, ['time', 'lat', 'lon'], 'tas', 'air_temperature', 'air temperature', 'K', np.ones((2,30,25),dtype=np.float))
+    """
+    def __init__(self, dicdim, dimlist, name, stdname, longname, units, values):
+        from collections import OrderedDict
+        totshape = np.prod(dicdim.values())
+        dv = []
+        for idd in dimlist: dv.append(dicdim[idd])
+        self.values = values[:]
+        self.dimensions = tuple(list(dicdim.keys()))
+        self.shape = tuple(dv)
+        self.name = name
+        self.standard_name = stdname
+        self.long_name = longname
+        self.units = units
+
+    def __getitem__(self, val):
+        return self.values[val]
+#variable = genericNCvariable_Dict({'lon':25, 'lat':30, 'time': 2}, ['time', 'lat', 'lon'], 'tas', 'air_temperature', 'air temperature', 'K', np.ones((2,30,25), dtype=np.float))
+#print variable
+#print variable.dimensions
+#print variable.shape
+#print variable.units
+#print variable[:].shape
 
 class genericNCvariable(object):
@@ -21063 +21098 @@
       ncfile + ' _______'
     rvarns = ronc.variables.keys()
-    print 'Lluis varns:', varns
     for vn in varns:
        rdimns = list(ronc.dimensions)
@@ -23287 +23321 @@
 
     if not CFvars.has_key(cfvarn):
-        print errormsg
+        print infmsg
         print '  ' + fname + ": file '" + inCFvarfile + "' does not have CF " +      \
           "variable: '" + cfvarn + "' !!"
@@ -23645 +23679 @@
             tvals, urefvals = compute_WRFtime(timewrfv, refdate=refT,                \
               tunitsval='minutes')
-            oaxisv = tvals
+            #oaxisv = tvals 
+            oaxisv = genericNCvariable_Dict({'time':len(tvals)}, ['time'], 'time',   \
+              'time', 'Time', urefvals, tvals[:])
 
         # CF values of axis
@@ -23666 +23702 @@
         if len(oaxisv.shape) == CFdimvalues['maxrank']:
             axisv = oaxisv[:]
-            if CFdimvalues['length'] != -1: CFdimvalues['length'] = oaxisv.shape[0]
-            CFvardimvalues.append(dimn)
+            vardimdimn = []
+            vardimdimL = []
+            for dn in oaxisv.dimensions:
+                vardimdimn.append(dn)
+            if CFdimvalues['length'] != -1: 
+                CFdimvalues['length'] = len(getdim_listonc(dimn, afNs, allonc))
+            CFvardimvalues = vardimdimn
         elif len(oaxisv.shape) == CFdimvalues['maxrank'] + 1:
             print warnmsg
@@ -23926 +23967 @@
                     varv = np.where(varv == fill_value, gen.fillValueF, varv)
             else:
-                print 'Lluis: ', newVdims
                 newvar=onewnc.createVariable(cfvarn, 'f4', tuple(newVdims),          \
                   fill_value=gen.fillValueF)
@@ -23972 +24012 @@
                     # Re-setting time values to the center of the bounds
                     ovtime[:] = tvals[:]
-                elif vtb[0:10] == 'Ctime_bnds':
+                elif vtb[0:11] == 'CFtime_bnds':
                     vtbvals = vtb.split('|')
                     if len(vtbvals) != 4:
@@ -23992 +24032 @@
                     print infmsg
                     print '      ' + fname + ": creation of variable 'time_bnds' " + \
-                      "from variable '" + tvtbvals[1] + "' using period '" + period +\
+                      "from variable '" + vtbvals[1] + "' using period '" + period +\
                       "' !!"
-                    odimvar = onc.variables[tvtbvals[1]]
+                    odimvar = onc.variables[vtbvals[1]]
                     timev = odimvar[:]
                     ttu = odimvar.getncattr('units')
@@ -24001 +24041 @@
                       'Y-m-dTH:M:SZ', 'YmdHMS')
                     tbvals, tvals, urefvals = compute_time_bnds(timev, ttu, period,  \
-                      kindWRFt='begperiod', refdate=refT, tunitsval='minutes')
+                      kindt='begperiod', refdate=refT, tunitsval='minutes')
                     # Re-setting time values to the center of the bounds
                     ovtime[:] = tvals[:]
@@ -24272 +24312 @@
         tfmt = tkind.split(',')[1]
         tvals = []
+        utime = 'minutes since 1949-12-01 00:00:00'
         mattvals = np.zeros((itvals.shape[0],6), dtype=int)
         for it in range(itvals.shape[0]):
             mattvals[it,:] = dt.datetime.string(''.join(itvals[it,:]),tfmt)
             tvals[it] = gen.datetimeStr_conversion(mattvals[it,:].strftime("%Y%m%d"+ \
-              "%H%M%S"), 'YmdHMS', 'cfTime,minutes since 1949-12-01 00:00:00')
+              "%H%M%S"), 'YmdHMS', 'cfTime,'+utime)
+        
     else:
         print errormsg