Changeset 1868 in lmdz_wrf for trunk/tools

Timestamp:

Mar 26, 2018, 10:59:24 PM (7 years ago)

Author:

lfita

Message:

Fixing a variety of issues to make it happen with the use of multiple input files in CFmorzization

File:

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

trunk/tools/nc_var_tools.py

@@ -22867 +22867 @@
 #quit()
 
-def getvar_listonc(varn, listoc):
+def getvar_listonc(varn, fns, listoc):
     """ Function to get a variable object from a list of netCDF file object
       varn= name of the variable to be found
+      fns= list with the name of the files
       listoc= list of NetCDF file objects
     """
@@ -22875 +22876 @@
 
     allfs = {}
-    fns = []
     for ioc in range(len(listoc)):
         oc = listoc[ioc]
-        allfs[oc.filepath] = [oc.dimensions, oc.variables.keys()]
-        fns.append(oc.filepath)
+        # Seems that ObjNetcdf.filepath() requires netcdf >= 4.1.2....
+        #  Thus avoiding to use the functionality
+        #allfs[oc.filepath()] = ...
+        allfs[fns[ioc]] = [oc.dimensions, oc.variables.keys()]
 
     # Looping for all files
@@ -22885 +22887 @@
     for ioc in range(len(listoc)):
         oc = listoc[ioc]
-        oncvars = allfs[ioc]
+        # Seems that ObjNetcdf.filepath() requires netcdf >= 4.1.2....
+        #  Thus avoiding to use the functionality
+        oncvars = allfs[fns[ioc]]
         oncvars.sort()
         if oc.variables.has_key(varn):
@@ -22894 +22898 @@
     if not found:
         print errormsg
-        print '  ' + fname + "': any netcdf file '" + allfs.keys() + "' contain " + \
+        print '  ' + fname + "': any netcdf file '" + ','.join(fns) + "' contain " + \
           "variable '" + varn + "' !!"
         print '    available ones _______'
@@ -22902 +22906 @@
             listoc[ioc].close()
             ioc = ioc + 1
-            quit(-1)
+        quit(-1)
 
     return ov
 
-def projection2D_information(PROJn, AXinf, projinf, dx, dy, oncs):
+def projection2D_information(PROJn, AXinf, projinf, dx, dy, Fns, oncs):
     """ Function to get the basic information of the 2D-projection: new dimensions, variables, ...
       Reproducing: class Projection from wrfncxnj.py
         PROJn= name of the projection
-        AXinf: dictionary with the informatino of the axes 'X', 'Y', 'Z', 'T'
-        projinf: dictionary with all the information of the projection
+        AXinf= dictionary with the informatino of the axes 'X', 'Y', 'Z', 'T'
+        projinf= dictionary with all the information of the projection
+        Fns= list with the name of the files
         oncs= list of netCDF objects from which retrieve the information
       Returns:
@@ -22921 +22926 @@
     fname = 'projection2D_information'
 
-    availproj = ['Lambert_Conformal', 'Rotated_Pole', 'Mercator']
+    availproj = ['Lambert_Conformal', 'Rotated_Pole/Rotate_Latituded_Longitude',     \
+      'Mercator']
 
     # Explanation of certain attributes in projection file
@@ -22997 +23003 @@
         AXinf['Y'] = axv
 
-    elif PROJn == "Rotated_Pole":
+    elif PROJn == "Rotated_Pole" or PROJn == 'Rotated_Latitude_Longitude':
         proj_attr = ["grid_mapping_name", "grid_north_pole_latitude",                \
           "grid_north_pole_longitude", 'fileXvals', 'fileYvals']
@@ -23022 +23028 @@
           'degrees']
 
-        ovar = getvar_listonc(projinf['fileXvals'][0], oncs)
+        ovar = getvar_listonc(projinf['fileXvals'][0], Fns, oncs)
         oXvar = ovar[0]
         if len(oXvar.shape) == 3:
@@ -23043 +23049 @@
         newdim['rlat'] = dy
         newvar['rlat'] = ['grid_latitude', 'latitude in rotated pole grid', 'degrees']
-        ovar = getvar_listonc(projinf['fileYvals'][0], oncs)
+        ovar = getvar_listonc(projinf['fileYvals'][0], Fns, oncs)
         oYvar = ovar[0]
         if len(oYvar.shape) == 3:
@@ -23116 +23122 @@
     else:
         print errormsg
-        print '  ' + fname + ": projection '" + + "' not ready !!"
+        print '  ' + fname + ": projection '" + PROJn + "' not ready !!"
         print '    available ones:', availproj
         quit(-1)
@@ -23436 +23442 @@
 
     # Joining all netCDF files
-    allonc = exonc + [onc]
+    allonc = [onc] + exonc
+    afNs = [ncfile] + extrafiles
     allfileNs = {}
     for ionc in range(len(allonc)):
         onc = allonc[ionc]
-        allfileNs[onc.filepath] = [onc.dimensions.keys(), onc.variables.keys()]
+        allfileNs[afNs[ionc]] = [onc.dimensions.keys(), onc.variables.keys()]
 
     # Creation of new file name
@@ -23476 +23483 @@
         # Axis values from file
         if vardimn != 'WRFtime':
-            oaxisv = getvar_listonc(vardimn, allloncs)
+            oaxisv = getvar_listonc(vardimn, afNs, allonc)
         else:
             print infmsg
             print '    ' + fname + ": creation of variable 'time' from WRF 'Times' !!"
-            odimvar = getvar_listonc('Times', allloncs)
+            odimvar = getvar_listonc('Times', afNs, allonc)
             timewrfv = odimvar[:]
             refT = gen.datetimeStr_conversion(gattr['basetime'], 'Y-m-dTH:M:SZ',     \
@@ -23599 +23606 @@
         newDim, newVar, newVARv, nonCFproj = projection2D_information(gattr['grid'], \
           CFdimvals, pattr, filedimvals['X'].shape[1], filedimvals['X'].shape[0],    \
-          alloncs)
+          afNs, allonc)
 
     if len(newDim.keys()) > 0: