Changeset 1853 in lmdz_wrf for trunk

Timestamp:

Mar 25, 2018, 4:53:59 PM (7 years ago)

Author:

lfita

Message:

Working version with projection information

File:

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

trunk/tools/nc_var_tools.py

@@ -994 +994 @@
     attrvalue = value of the attribute
     atrtrkind = kind of attribute: 'S', string ('!' as spaces); 'U', unicode ('!' as spaces); 'I', integer; 
-      'Inp32', numpy integer 32; 'R', real; ' npfloat', np.float; 'D', np.float64
+      'Inp32', numpy integer 32; 'R', ot 'F' real; ' npfloat', np.float; 'D', np.float64
     """
     fname = 'set_attributek'
     validk = {'S': 'string', 'U': 'unicode', 'I': 'integer',                         \
-      'Inp32': 'integer long (np.int32)', 'R': 'float', 'npfloat': 'np.float',       \
-      'D': 'float long (np.float64)'}
+      'Inp32': 'integer long (np.int32)', 'F': 'float', 'R': 'float',                \
+      'npfloat': 'np.float', 'D': 'float long (np.float64)'}
 
     if type(attrkind) == type('s'):
@@ -1008 +1008 @@
         elif attrkind == 'I':
             attrvalue = np.int(attrval)
-        elif attrkind == 'R':
+        elif attrkind == 'R' or attrkind == 'F' :
             attrvalue = float(attrval)
         elif attrkind == 'npfloat':
@@ -22722 +22722 @@
              (...) (specific projection attributes and values)
              [name] [value] [kind] ('I': integer, 'R': float, 'D': double float, 'S': string)
+             Add three more attributes for the resolution of the projection:
+               x_resolution [value] (resolution along x-axis)
+               y_resoltuion [value] (resolution along y-axis)
+               proj_units [value] (units of the resoltion of the projection)
         ncfile= netCDF file to use
         variable= ',' list of variables to CF transform ('all' for all variables). A separated file will be created
@@ -22787 +22791 @@
 
     pattr = {}
+    ppn = []
     opinff = open(projfilen, 'r')
     for line in opinff:
@@ -22800 +22805 @@
                 quit(-1)
             pattr[linevals[0]] = [linevals[1].replace('!',' '), linevals[2]]
+            ppn.append(linevals[0])
     opinff.close()
+    print '  ' + fname + ": values for projection '" + gattr['grid'] + "' _______"
+    for iatn in ppn:
+        pattrv = pattr[iatn]
+        print  iatn + ': ', gen.typemod(pattrv[0], pattrv[1])
 
     # Getting variables to process
@@ -22916 +22926 @@
             quit(-1)
         if CFdimvalues['length'] == -1: CFdimvalues['length'] = None
-        CFdimvals[axn] = CFdimvalues
-        filedimvals[axn] = axsiv[:]
-        CFaxisvardimvals[axn] = CFvardimvalues
 
         # Some extra axis-stuff...
@@ -22931 +22938 @@
                 tunits = urefvals
                 tcal = 'standard'
+            if gattr['calendar'] == '[unknown]':  gattr['calendar']= tcal
+            CFdimvalues['units'] = tunits
+            
             Sinit = gen.datetimeStr_conversion(str(CFitime), 'cfTime,'+tunits,       \
               'Y-m-dTH:M:SZ')
@@ -22970 +22980 @@
                 Sfreq = gattr['frequency']
 
+        CFdimvals[axn] = CFdimvalues
+        filedimvals[axn] = axsiv[:]
+        CFaxisvardimvals[axn] = CFvardimvalues
+
     # Renaming dimensions for 2D lon,lat variable-dimensions
     if len(filedimvals['X'].shape) == 2 or len(filedimvals['Y'].shape) == 2:
@@ -22975 +22989 @@
         print '    because Xvals have rank 2:', filedimvals['X'].shape,              \
           'renaming lon,lat dimensions to x,y'
+        addCFdimvals = {}
         axv = CFdimvals['X']
         axv['dimn'] = 'x'
+        axv['vdimn'] = 'lon'
         CFdimvals['X'] = axv
+        addCFdimvals['x'] = {'dimn': 'x', 'vdimn': 'x',
+          'stdn': 'projection_x_coordinate','longname': 'x coordinate of projection',\
+          'units': pattr['proj_units'][0], 'maxrank': 1,                             \
+          'length': filedimvals['X'].shape[1]}
+        Xres = np.float(pattr['x_resolution'][0])
+        filedimvals['x'] = np.arange((filedimvals['X'].shape[1]),dtype=np.float)*Xres
         axv = CFdimvals['Y']
         axv['dimn'] = 'y'
+        axv['vdimn'] = 'lat'
         CFdimvals['Y'] = axv
+        addCFdimvals['y'] = {'dimn': 'y', 'vdimn': 'y',
+          'stdn': 'projection_y_coordinate','longname': 'y coordinate of projection',\
+          'units': pattr['proj_units'][0], 'maxrank': 1,                             \
+          'length': filedimvals['X'].shape[0]}
+        Yres = np.float(pattr['y_resolution'][0])
+        filedimvals['y'] = np.arange((filedimvals['X'].shape[0]),dtype=np.float)*Yres
         #axiv = axisinf['X']
         #filecfdimn[axiv[0]] = 'x'
@@ -23076 +23105 @@
         for axn in varaxes:
             dvals = CFdimvals[axn]
-            if not gen.searchInlist(onewnc.variables.keys(), dvals['dimn']):
-                newvar = onewnc.createVariable(dvals['dimn'], 'f8',                  \
-                  tuple(CFaxisvardimvals[axn]))
-                basicvardef(newvar, dvals['stdn'], dvals['longname'], dvals['units'])
-                for ivn in dvals.keys():
-                    if ivn != 'dimn' and ivn != 'stdn' and ivn != 'longname' and     \
-                      ivn != 'units' and ivn != 'maxrank' and ivn != 'length':
-                        set_attribute(newvar,ivn,dvals[ivn])
+            if dvals['dimn'] == dvals['vdimn']:
+                if not gen.searchInlist(onewnc.variables.keys(), dvals['dimn']):
+                    newvar = onewnc.createVariable(dvals['dimn'], 'f8',              \
+                      tuple(CFaxisvardimvals[axn]))
+                    basicvardef(newvar, dvals['stdn'], dvals['longname'],            \
+                      dvals['units'])
+                    newvar[:] = filedimvals[axn]
+                    for ivn in dvals.keys():
+                        if ivn != 'dimn' and ivn != 'stdn' and ivn != 'longname' and \
+                          ivn != 'units' and ivn != 'maxrank' and ivn != 'length':
+                            set_attribute(newvar,ivn,dvals[ivn])
+            else:
+                # Dimensions with a different name for the variable dimension
+                if not gen.searchInlist(onewnc.variables.keys(), dvals['dimn']):
+                    ddvals = addCFdimvals[dvals['dimn']]
+                    newvar = onewnc.createVariable(dvals['dimn'], 'f8',              \
+                      tuple(dvals['dimn']))
+                    basicvardef(newvar, ddvals['stdn'], ddvals['longname'],          \
+                      ddvals['units'])
+                    newvar[:] = filedimvals[dvals['dimn']]
+                if not gen.searchInlist(onewnc.variables.keys(), dvals['vdimn']):
+                    newvar = onewnc.createVariable(dvals['vdimn'], 'f4',             \
+                      tuple(CFaxisvardimvals[axn]))
+                    basicvardef(newvar, dvals['stdn'], dvals['longname'],            \
+                      dvals['units'])
+                    newvar[:] = filedimvals[axn]
+                    for ivn in dvals.keys():
+                        if ivn != 'dimn' and ivn != 'stdn' and ivn != 'longname' and \
+                          ivn != 'units' and ivn != 'maxrank' and ivn != 'length':
+                            set_attribute(newvar,ivn,dvals[ivn])
 
         # Variable
@@ -23093 +23144 @@
               fill_value=gen.fillValueF)
             basicvardef(newvar, stdvarn, longvarn, utsvarn)
-            set_attribute(newvar, 'coordinates', ' '.join(CFvardims))
+
+            # Providing the right coordinates
+            coorv = []
+            for cdn in CFvardims:
+                if cdn == 'x': coorv.append('lon')
+                elif cdn == 'y': coorv.append('lat')
+                else: coorv.append(cdn)
+            set_attribute(newvar, 'coordinates', ' '.join(coorv))
+
             # Setting values, but taking into account pre-existing masked values
             varv = ovar[:]
@@ -23101 +23160 @@
  
             newvar[:] = varv[:]
-        onewnc.sync()
+            set_attribute(newvar, 'grid_mapping', gattr['grid'])
+            onewnc.sync()
+
+        # Adding projection variable
+        print infmsg
+        print '  '+fname+": adding projection information of '" + gattr['grid'] + "'"
+        newvar = onewnc.createVariable(gattr['grid'], 'i')
+        for attrn in ppn:
+            pvattr = pattr[attrn]
+            set_attributek(newvar, attrn, pvattr[0], pvattr[1])
+            onewnc.sync()
 
         # Global attributes