Changeset 1961 in lmdz_wrf

Timestamp:

Jul 24, 2018, 1:32:25 AM (7 years ago)

Author:

lfita

Message:

Final version of merging single-stations netCDF

Location:

trunk/tools

Files:

• generic_tools.py (modified) (1 diff)
• nc_var_tools.py (modified) (9 diffs)

trunk/tools/generic_tools.py

@@ -9538 +9538 @@
       listA= list with values to match with
       listB= list with values
-    >>> list_coincidences(range(10),range(8,15))
+    >>> list_differences(range(10),range(8,15))
     [10, 11, 12, 13, 14], [2, 3, 4, 5, 6]
     """

trunk/tools/nc_var_tools.py

@@ -24618 +24618 @@
     latvals = []
     heightvals = []
+    timevals = []
     for fobsn in fobs:
         filen =  namefoldobs + '/' + fobsn
@@ -24632 +24633 @@
             print errormsg
             print '  ' + fname + ": file '" + filen + "' does not have any 'time'" + \
-              "dimensisn !!"
+              "dimension !!"
             print "    available ones:", onc.dimensions
             quit(-1)
@@ -24639 +24640 @@
             if variable == 'all': varns = onc.variables.keys()
             else: varns = gen.str_list(variable, ',')
+            if onc.variables.has_key('time'): 
+                timevals = list(onc.variables['time'][:])
+            elif onc.variables.has_key('CFtime'): 
+                timevals = list(onc.variables['CFtime'][:])
+            else:
+                print errormsg
+                print '  '+fname+": file '" + filen + "' does not have any 'time'" + \
+                  "expected variable ('time', 'CFtime') !!"
+                varnsorted = onc.variables.has_key(varn)
+                varnsorted.sort()
+                print '    available ones:', varnsorted
+                quit(-1)
+        else:
+            if onc.variables.has_key('time'): 
+                tvals = onc.variables['time'][:]
+            elif onc.variables.has_key('CFtime'): 
+                tvals =onc.variables['CFtime'][:]
+            else:
+                print errormsg
+                print '  ' +fname+ ": file '" +filen+ "' does not have any 'time'" + \
+                  "expected variable ('time', 'CFtime') !!"
+                varnsorted = onc.variables.has_key(varn)
+                varnsorted.sort()
+                print '    available ones:', varnsorted
+                quit(-1)
+            NOtvals, NOits = gen.list_differences(timevals, list(tvals))
+            if len(NOtvals) > 0:
+                print '  ' + fname + ": file '" + filen + "' adds additional times:",\
+                  NOtvals
+                for tv in NOtvals: timevals.append(tv)
         stn = ''
         for stS in onc.variables['station'][0,:]:
@@ -24682 +24713 @@
     basicvardef(newvar, 'height', 'height of stations', 'm')
     newvar[:] = heightvals[:]
+
+    # Time axis
+    timevalsort = timevals
+    timevalsort.sort()
+    newvart = onewnc.createVariable('ststimes', 'f', ('time'))
+    newvart[:] = timevalsort[:]
+    newvart.setncattr('standrad_name','time')
+    newvart.setncattr('long_name','Time')
 
     ifile = 0
@@ -24693 +24732 @@
 
         onc = NetCDFFile(filen, 'r')
+        ot = onc.variables['time']
+        otv = list(ot[:])
+        coint, itcoinsort, itcoinotv = gen.list_coincidences(timevalsort, otv)
+        itcoinsort.sort()
+
         if ifile == 0:
             for varn in varns:
@@ -24703 +24747 @@
                     print '    available ones:', varnsorted
                     quit(-1)
+                tu = ot.units
+                newvart.setncattr('units',tu)
+
                 ovar = onc.variables[varn]
                 attrns = ovar.ncattrs()
@@ -24724 +24771 @@
                     newvar = onewnc.createVariable(varn, vartype, tuple(newdims))
 
-                varv = ovar[:]
-                if len(newvar.shape) == 4: newvar[:,:,:,ifile] = varv[:]
-                elif len(newvar.shape) == 3: newvar[:,:,ifile] = varv[:]
-                elif len(newvar.shape) == 2: newvar[:,ifile] = varv[:]
                 for attrn in attrns:
                     if attrn != '_FillValue':
@@ -24736 +24779 @@
                     attrv = onc.getncattr(attrn)
                     onewnc.setncattr(attrn, attrv)
-
+                newvar = onewnc.variables[varn]
+                varv = ovar[:]
+                if gen.searchInlist(newvar.dimensions, 'time'):
+                    if len(newvar.shape) == 4: newvar[itcoinsort,:,:,ifile] = varv[:]
+                    elif len(newvar.shape) == 3: newvar[itcoinsort,:,ifile] = varv[:]
+                    elif len(newvar.shape) == 2: newvar[itcoinsort,ifile] = varv[:]
+                else:
+                    if len(newvar.shape) == 4: newvar[:,:,:,ifile] = varv[:]
+                    elif len(newvar.shape) == 3: newvar[:,:,ifile] = varv[:]
+                    elif len(newvar.shape) == 2: newvar[:,ifile] = varv[:]
         else:
             for varn in varns:
@@ -24747 +24799 @@
                     print '    available ones:', varnsorted
                     quit(-1)
+
                 ovar = onc.variables[varn]
                 newvar = onewnc.variables[varn]
                 varv = ovar[:]
-                if len(newvar.shape) == 4: newvar[:,:,:,ifile] = varv[:]
-                elif len(newvar.shape) == 3: newvar[:,:,ifile] = varv[:]
-                elif len(newvar.shape) == 2: newvar[:,ifile] = varv[:]
+                if gen.searchInlist(newvar.dimensions, 'time'):
+                    if len(newvar.shape) == 4: newvar[itcoinsort,:,:,ifile] = varv[:]
+                    elif len(newvar.shape) == 3: newvar[itcoinsort,:,ifile] = varv[:]
+                    elif len(newvar.shape) == 2: newvar[itcoinsort,ifile] = varv[:]
+                else:
+                    if len(newvar.shape) == 4: newvar[:,:,:,ifile] = varv[:]
+                    elif len(newvar.shape) == 3: newvar[:,:,ifile] = varv[:]
+                    elif len(newvar.shape) == 2: newvar[:,ifile] = varv[:]
  
         onewnc.sync()