Changeset 1480 in lmdz_wrf for trunk

Timestamp:

Mar 31, 2017, 5:51:21 PM (8 years ago)

Author:

lfita

Message:

Adding 'time' awareness on file `netcdf_fold_concatenation_HMT'

File:

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

trunk/tools/nc_var_tools.py

@@ -10592 +10592 @@
         [fold]: folder with the location of the netCDF files
         [dimname]: dimension along which files should be concatenated
-        [tail]: tail of the files to concatenate (optional)
+        [vardimname]: name of the dimension variable with the values
       [ncfile]= [header],[middle],[tail]
         [header] = header of the name of the files to concatenate [ncfile]*[middle]*[tail]
@@ -10613 +10613 @@
     fold = values.split(',')[0]
     condim = values.split(',')[1]
+    vardimname = values.split(',')[2]
 
     HMT = ncfile.split(',')
@@ -10639 +10640 @@
         print errormsg
         print '  ' + fname + ": files do not have dimensions '" + condim + "' !!!"
+        quit(-1)
+    if not ncobjs[0].variables.has_key(vardimname):
+        print errormsg
+        print '  ' + fname + ": files do not have variable '" + vardimname + "' !!!"
         quit(-1)
 
@@ -10677 +10682 @@
     newvar = newnc.createVariable('int' + condim, 'i4', tuple(dims))
     newvar[:] = range(totcondim)
+
+# Dimension variable for the concatenation
+    
+    odimvar = ncobjs[0].variables[vardimname]
+    odimvardims = odimvar.dimensions
+    odimattrs = odimvar.ncattrs
+    for dimn in odimvardims:
+        if not gen.searchInlist(newnc.dimensions,dimn):
+            Ldim = len(odimvar.dimensions[dimn])
+    newvar = newnc.crateVarieble(vardimname, odimvar.dtype, odimvardims)
+    for attrn in odimattrs:
+        attrv = odimvar.getncattr(attrn)
+        newattr = set_attribute(newvar, attrn, attrv)
+        if attrn == 'units': urefvals = attrv
+
+    # Checking units of condimvar
+    uvalsecs = urefvals.split(' ')
+    if uvalsecs[1] == 'since': 
+        tunits = True
+        if len(uvalsecs) == 4:
+            date=uvalsecs[2]
+            time=uvalsecs[3]
+        else:
+            if uvalsecs[2].find('_') != -1:
+                date=uvalsecs[2]
+                time='00:00:00'
+            else:
+                date=uvalsecs[2].split('_')[0]
+                time=uvalsecs[2].split('_')[1]
+        refdate = gen.datetimeStr_datetime(date+'_'+time)
+        print warnmsg
+        print '  ' + fname + ': concatenating dimension-variable has time units!'
+        print '    shifting temporal-values since the date of the first file', refdate
+
+    for ifile in range(Nfiles):
+        slicevartot = []
+        objvar = ncobjs[ifile].variables[vardimname]
+        for dimn in objvar.dimensions:
+            ldimfile = len(ncobjs[ifile].dimensions[dimn])
+            if dimn == condim:
+                slicevartot.append(slice(begslicetot,begslicetot+ldimfile))
+                begslicetot = begslicetot + ldimfile
+            else:
+                slicevartot.append(slice(0,ldimfile))
+            if tunits:
+                uvals = objvar.getncattr('units')
+                newtimes = gen.coincident_CFtimes(objvar[:], uvals, urefvals)
+                newvar[tuple(slicevartot)] = objvar[:]
+            else:
+                newvar[tuple(slicevartot)] = objvar[:]
+            newnc.sync()
 
 # Looping variables