Changeset 1867 in lmdz_wrf for trunk/tools/nc_var_tools.py

Timestamp:

Mar 26, 2018, 8:09:39 PM (7 years ago)

Author:

lfita

Message:

Adding and using:

• `getvar_listonc': Function to get a variable object from a list of netCDF file object

File:

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

trunk/tools/nc_var_tools.py

@@ -77 +77 @@
 # 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
+# getvar_listonc: Function to get a variable object from a list of netCDF file object
 # get_namelist_vars: Function to get namelist-like  values ([varname] = [value])
 # get_str_nc: Function to get string values in a netCDF variable as a chains of 1char values
@@ -22866 +22867 @@
 #quit()
 
+def getvar_listonc(varn, listoc):
+    """ Function to get a variable object from a list of netCDF file object
+      varn= name of the variable to be found
+      listoc= list of NetCDF file objects
+    """
+    fname = 'getvar_listonc'
+
+    allfs = {}
+    fns = []
+    for ioc in range(len(listoc)):
+        oc = listoc[ioc]
+        allfs[oc.filepath] = [oc.dimensions, oc.variables.keys()]
+        fns.append(oc.filepath)
+
+    # Looping for all files
+    found = False
+    for ioc in range(len(listoc)):
+        oc = listoc[ioc]
+        oncvars = allfs[ioc]
+        oncvars.sort()
+        if oc.variables.has_key(varn):
+            ov = oc.variables[varn]
+            found = True
+            break
+
+    if not found:
+        print errormsg
+        print '  ' + fname + "': any netcdf file '" + allfs.keys() + "' contain " + \
+          "variable '" + varn + "' !!"
+        print '    available ones _______'
+        ioc = 0
+        for fn in fns:
+            print fn, ':', allfs[fn][1]
+            listoc[ioc].close()
+            ioc = ioc + 1
+            quit(-1)
+
+    return ov
+
 def projection2D_information(PROJn, AXinf, projinf, dx, dy, oncs):
     """ Function to get the basic information of the 2D-projection: new dimensions, variables, ...
@@ -22872 +22912 @@
         AXinf: dictionary with the informatino of the axes 'X', 'Y', 'Z', 'T'
         projinf: dictionary with all the information of the projection
-        oncs= netCDF objects from which retrieve the information
+        oncs= list of netCDF objects from which retrieve the information
       Returns:
         newdim: dictionary new dimensions to be added {dimn1: dimlength1, ..., dimnN: dimlengthN} 
@@ -22981 +23021 @@
         newvar['rlon'] = ['grid_longitude', 'longitude in rotated pole grid',        \
           'degrees']
-        found = Flase
-        fileNs = ''
-        for ionc in range(len(oncs)):
-            onc = oncs[ionc]
-            fileNs = fileNs + onc.filepath + ', '
-            if gen.searchInlist(onc.variables.keys(),projinf['fileXrefvals'][0]):
-                oXvar = onc.variables[projinf['fileXvals'][0]]
-                break
-        if not found:
-            print errormsg
-            print '  ' + fname + ": files: " + fileNs + " do not have variable " +   \
-              "with X ref values '" + projinf['fileXrefvals'][0] + "' !!"
-            print '    available ones:', onc.variables.keys()
-            quit(-1)
+
+        ovar = getvar_listonc(projinf['fileXvals'][0], oncs)
+        oXvar = ovar[0]
         if len(oXvar.shape) == 3:
             newvarv['rlon'] = oXvar[0,0,:]
@@ -23014 +23043 @@
         newdim['rlat'] = dy
         newvar['rlat'] = ['grid_latitude', 'latitude in rotated pole grid', 'degrees']
-        if not gen.searchInlist(onc.variables.keys(),projinf['fileYrefvals'][0]):
-            print errormsg
-            print '  ' + fname + ": file does not have variable with Y ref values '"+\
-              projinf['fileYrefvals'][0] + "' !!"
-            print '    available ones:', onc.variables.keys()
-            quit(-1)
-        oYvar = onc.variables[projinf['fileYvals'][0]]
+        ovar = getvar_listonc(projinf['fileYvals'][0], oncs)
+        oYvar = ovar[0]
         if len(oYvar.shape) == 3:
             newvarv['rlat'] = oYvar[0,0,:]
@@ -23411 +23435 @@
         Varns = gen.str_list(variable, ',')
 
+    # Joining all netCDF files
+    allonc = exonc + [onc]
+    allfileNs = {}
+    for ionc in range(len(allonc)):
+        onc = allonc[ionc]
+        allfileNs[onc.filepath] = [onc.dimensions.keys(), onc.variables.keys()]
+
     # Creation of new file name
     # This will done following CORDEX rules (getting information from the ASCII 
@@ -23430 +23461 @@
     axes = ['X', 'Y', 'Z', 'T']
 
-    oncvars = onc.variables.keys()
-    oncvars.sort()
-
     # Axes information: dimensions and variables dependency
     # CF-dimensions of a given axis
@@ -23440 +23468 @@
     # CD-fimensions of the CF-variables for each axis
     CFaxisvardimvals = {}
+
     for axn in axes:
         axisv = axisinf[axn]
@@ -23447 +23476 @@
         # Axis values from file
         if vardimn != 'WRFtime':
-            if not onc.variables.has_key(vardimn):
-                print errormsg
-                print '  '+fname+"': netcdf file '"+ ncfile + "' does not contain "+ \
-                  "variable '" + vardimn + "' !!"
-                print '    available ones:', oncvars
-                onc.close()
-                quit(-1)
-            oaxisv = onc.variables[vardimn]
+            oaxisv = getvar_listonc(vardimn, allloncs)
         else:
             print infmsg
             print '    ' + fname + ": creation of variable 'time' from WRF 'Times' !!"
-            odimvar = onc.variables['Times']
+            odimvar = getvar_listonc('Times', allloncs)
             timewrfv = odimvar[:]
             refT = gen.datetimeStr_conversion(gattr['basetime'], 'Y-m-dTH:M:SZ',     \
@@ -23577 +23599 @@
         newDim, newVar, newVARv, nonCFproj = projection2D_information(gattr['grid'], \
           CFdimvals, pattr, filedimvals['X'].shape[1], filedimvals['X'].shape[0],    \
-          exonc + [onc])
+          alloncs)
 
     if len(newDim.keys()) > 0: