Changeset 1092 in lmdz_wrf for trunk

Timestamp:

Sep 9, 2016, 6:16:44 PM (9 years ago)

Author:

lfita

Message:

Addin `time_reset': Function to re-set the time axis of a file

Location:

trunk/tools

Files:

• nc_var.py (modified) (2 diffs)
• nc_var_tools.py (modified) (2 diffs)

trunk/tools/nc_var.py

@@ -52 +52 @@
   'seasmean', 'sellonlatbox', 'sellonlatlevbox', 'selvar', 'setvar_asciivalues',     \
   'sorttimesmat', 'spacemean', 'SpatialWeightedMean', 'statcompare_files',           \
-  'subbasin', 'submns', 'subyrs', 'TimeInf',                                         \
+  'subbasin', 'submns', 'subyrs', 'TimeInf', 'time_reset',                           \
   'TimeSplitmat', 'timemean', 'valmod', 'valmod_dim','varaddattrk', 'varaddattr',    \
   'varaddref',                                                                       \
@@ -273 +273 @@
 elif oper == 'TimeInf':
     ncvar.TimeInf(opts.ncfile, opts.varname)
+elif oper == 'time_reset':
+    ncvar.time_reset(opts.values, opts.ncfile, opts.varname)
 elif oper == 'TimeSplitmat':
     ncvar.TimeSplitmat(opts.values, opts.ncfile, opts.varname)

trunk/tools/nc_var_tools.py

@@ -111 +111 @@
 # time_information: Function to provide information about variable time
 # timemean: Function to retrieve a time mean series from a multidimensional variable of a file
+# time_reset: Function to re-set the time axis of a file
 # timeshiftvar: Function to temporaly shift a number of time-steps a given variable inside a netCDF file
 # TimeSplitmat: Function to transform a file with CFtimes to a matrix [Nyear,Nmonth,Nday,Nhour,Nminute,Nsecond]
@@ -16968 +16969 @@
     return
 
+def time_reset(values,ncfile,variable):
+    """ Function to re-set the time axis of a file
+      values= [kind],[value]
+        [kind]: kind of time re-set:
+          'setorigin': make time-steps start at a given date [value]=[YYYYMMDDHHMISS]
+      ncfile: netcdf file to re-set times
+      variable: name of the variable time within the file
+    """
+    import datetime as dt
+    fname = 'time_reset'
+
+    OpReady = ['setorigin']
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print time_reset.__doc__
+        quit()
+
+    arguments = '[kind],[value]'
+    gen.check_arguments(fname, values, arguments, ',')
+    
+    kind = values.split(',')[0]
+    value = values.split(',')[1]
+
+    ncf = NetCDFFile(ncfile,'a')
+    if not ncf.variables.has_key(variable):
+        print errormsg
+        print '  ' + fname + ": file '" +ncfile+ "' does not have time variable: '"+ \
+          variable + "' !!"
+        quit(-1)
+
+    otvar = ncf.variables[variable]
+    tvals = otvar[:]
+    dimt = tvals.shape[0]
+    tunits = otvar.getncattr('units')
+    tattrs = otvar.ncattrs()
+
+    if kind == 'setorigin':
+        RefDate = value
+        RefDateS = RefDate[0:4] + '-' + RefDate[4:6] + '-' + RefDate[6:8] + ' ' +    \
+          RefDate[8:10] + ':' + RefDate[10:12] + ':' + RefDate[12:14]
+
+        dt = tvals[1] - tvals[0]
+        newtvals = np.arange(dimt)*dt
+        otvar[:] = newtvals
+
+        newtunits = ' '.join(tunits.split(' ')[0:2]) + ' ' + RefDateS
+        newattr = set_attribute(otvar, 'units', newtunits)
+        if gen.searchInlist(tattrs, 'time_origin'):
+            newattr = set_attribute(otvar, 'time_origin', newtunits)
+    else:
+        print errormsg
+        print '  ' + fname + ": kind '" + kind + "' not ready !!"
+        print '    Operations ready:', OpReady
+        quit(-1)
+
+    ncf.sync()
+    ncf.close()
+
+    return
+
+
 #quit()