Changeset 1889 in lmdz_wrf for trunk

Timestamp:

Apr 4, 2018, 5:20:47 PM (7 years ago)

Author:

lfita

Message:

Adding:

• `itime': Function to provide information of the time from a netCDF file

Location:

trunk/tools

Files:

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

trunk/tools/nc_var.py

@@ -23 +23 @@
 ## e.g. # nc_var.py -o ifile -f ../PY/wrfout_d01_1995-01-01_00:00:00
 ## e.g. # nc_var.py -o splitfile_dim -S bottom_top:ZNU:QVAPOR_ZNU:.4f -v QVAPOR,XLONG,XLAT,Times -f ~/PY/wrfout_d01_1995-01-01_00:00:00
+## e.g. # nc_var.py -o itime -S WRFtime -f ../PY/wrfout_d01_1995-01-01_00:00:00 -v Times
 
 ## e.g. ccrc468-17 # ./nc_var.py -v time -f 123/CCRC_NARCliM_Sydney_All_1990-1999_pr10max.nc -o out -S 1:-1
@@ -119 +120 @@
 # isgattrs: Give a single global attribute of a file and its type
 # isvattrs: Give a single attribute of a variable
+# itime: Function to provide information of the time from a netCDF file
 # ivars: Give all the variable names of a file
 # ivattrs: Give all the attributes of a variable and its type
@@ -197 +199 @@
   'getvalues_lonlat', 'getvars_tofile', 'grattr',                                    \
   'grmattr', 'idims', 'ifile', 'igattrs', 'increaseDimvar', 'isgattrs',              \
-  'isvattrs', 'ivars', 'ivattrs', 'LMDZ_toCF', 'lonlat_polygon', 'maskvar',          \
+  'isvattrs', 'itime', 'ivars', 'ivattrs', 'LMDZ_toCF', 'lonlat_polygon', 'maskvar', \
   'merge_files', 'model_characteristics',                                            \
   'mthDYNAMICO_toCF', 'ncreplace', 'ncstepdiff', 'netcdf_concatenation',             \
@@ -394 +396 @@
 elif oper == 'isvattrs':
     ncvar.isvattrs(opts.values, opts.ncfile, opts.varname)
+elif oper == 'itime':
+    ncvar.itime(opts.values, opts.ncfile, opts.varname)
 elif oper == 'ivars':
     ncvar.ivars(opts.ncfile)

trunk/tools/nc_var_tools.py

@@ -96 +96 @@
 # isgattrs: Give a single global attribute of a file and its type
 # isvattrs: Give a single attribute of a variable
+# itime: Function to provide information of the time from a netCDF file
 # ivars: Give all the variable names of a file
 # ivattrs: Give all the attributes of a variable and its type
@@ -24215 +24216 @@
 
     return 
+
+def itime(values, ncfile, variable):
+    """ Function to provide information of the time from a netCDF file
+      values=[tkind]
+        [tkind]: kind of variable time
+          'WRFtime': for WRF derived time values (String variable)
+          'CFtime': time following CF-conventions (as [tunits] since [refdate])
+          'string',[tfmt]: time values as a given string format [tfmt] (following python specification)
+      ncfile= file from which provide the information
+      variable= name of the variable time in the file
+    """
+    import datetime as dt
+    fname = 'itime'
+
+    availtkind = ['WRFtime', 'CFtime', 'string,[tfmt]']
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print itime.__doc__
+        quit()
+
+    expectargs = '[tkind]'
+    gen.check_arguments(fname, values, expectargs, ':')
+
+    tkind = values.split(':')[0]
+
+    onc = NetCDFFile(ncfile, 'r')
+
+    if not onc.variables.has_key(variable):
+        ncvars = list(onc.variables.keys())
+        ncvars.sort()
+        print errormsg
+        print '  ' + fname + ": file '" + ncfile + "' does not have variable '" +    \
+          variable + "' !!"
+        print '    available ones:', ncvars
+        onc.close()
+        quit(-1)
+
+    otvar = onc.variables[variable]
+    itvals = otvar[:]
+    if tkind[0:7] == 'WRFtime':
+        tvals, utime = compute_WRFtime(itvals)
+        mattvals = np.zeros((itvals.shape[0],6), dtype=int)
+        for it in range(len(tvals)):
+            mattvals[it,:]= gen.datetimeStr_conversion(str(tvals[it]),               \
+              'cfTime,minutes since 1949-12-01 00:00:00', 'matYmdHMS')
+    elif tkind[0:6] == 'CFtime':
+        tvals = itvals.copy()
+        utime = otvar.getncattr('units')
+        mattvals = np.zeros((itvals.shape[0],6), dtype=int)
+        for it in range(len(tvals)):
+            mattvals[it,:]= gen.datetimeStr_conversion(str(tvals[it]),               \
+              'cfTime,'+utime, 'matYmdHMS')
+    elif tkind[0:6] == 'string':
+        tfmt = tkind.split(',')[1]
+        tvals = []
+        mattvals = np.zeros((itvals.shape[0],6), dtype=int)
+        for it in range(itvals.shape[0]):
+            mattvals[it,:] = dt.datetime.string(''.join(itvals[it,:]),tfmt)
+            tvals[it] = gen.datetimeStr_conversion(mattvals[it,:].strftime("%Y%m%d"+ \
+              "%H%M%S"), 'YmdHMS', 'cfTime,minutes since 1949-12-01 00:00:00')
+    else:
+        print errormsg
+        print '  ' + fname + ": time kind '" + tkind + "' not ready !!"
+        print '    available ones:', availtkind
+        quit(-1)
+
+    # Getting information
+    dimt = mattvals.shape[0]
+    print 'First_date:', mattvals[0,:]
+    print 'Last_date:', mattvals[dimt-1,:]
+    print 'dimt:', dimt
+    date1 = dt.datetime(mattvals[0,0], mattvals[0,1], mattvals[0,2], mattvals[0,3],  \
+      mattvals[0,4], mattvals[0,5])
+    date2 = dt.datetime(mattvals[1,0], mattvals[1,1], mattvals[1,2], mattvals[1,3],  \
+      mattvals[1,4], mattvals[1,5])
+    diffdate = date2 - date1
+    if gen.searchInlist(dir(diffdate),'total_seconds'):
+        totsecs = diffdate.total_seconds()
+    else:
+        totsecs = diffdate.days()*24*3600. + diffdate.seconds()
+
+    print 'diffdate_2nd-1st:', diffdate
+    print 'diff_seconds_2nd-1st:', totsecs
+
+    for it in range(1,dimt-1):
+        it1 = it + 1
+        date1 = dt.datetime(mattvals[it,0], mattvals[it,1], mattvals[it,2],          \
+          mattvals[it,3], mattvals[it,4], mattvals[it,5])
+        date2 = dt.datetime(mattvals[it1,0], mattvals[it1,1], mattvals[it1,2],       \
+          mattvals[it1,3], mattvals[it1,4], mattvals[it1,5])
+        diffdate = date2 - date1
+        if gen.searchInlist(dir(diffdate),'total_seconds'):
+            newtotsecs = diffdate.total_seconds()
+        else:
+            newtotsecs = diffdate.days()*24*3600. + diffdate.seconds()
+        if newtotsecs != totsecs:
+            print '  different_diff_seconds_between_it_' + str(it) + '_and_' +       \
+              str(it1) + ':', newtotsecs
+
+    # attributes
+    for attrn in otvar.ncattrs():
+        attrv = otvar.getncattr(attrn)
+        print 'Tattr_@' + attrn + '@:', attrv.replace(' ','!')
+
+    onc.close()
+
+    return
+
+#itime('WRFtime', '../PY/wrfout_d01_1995-01-01_00:00:00', 'Times')
+
 #quit()