Changeset 507 in lmdz_wrf for trunk/tools

Timestamp:

Jun 17, 2015, 3:09:46 PM (10 years ago)

Author:

lfita

Message:

Adding `DatesFiles?', Function to find different time values on a series of WRF files in a folder

Location:

trunk/tools

Files:

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

trunk/tools/nc_var.py

@@ -22 +22 @@
   'chgtimestep', 'chvarname', 'compute_deaccum', 'compute_opersvarsfiles',           \
   'compute_opervaralltime', 'compute_opervartimes', 'compute_tevolboxtraj',          \
+  'DatesFiles',                                                                      \
   'DataSetSection', 'DataSetSection_multidims', 'dimToUnlimited', 'dimVar_creation', \
    'fattradd',                                                                       \
@@ -127 +128 @@
 elif oper == 'DataSetSection_multidims':
     ncvar.DataSetSection_multidims(opts.values, opts.ncfile)
+elif oper == 'DatesFiles':
+    ncvar.DatesFiles(opts.values, opts.ncfile, opts.varname)
 elif oper == 'dimToUnlimited':
     ncvar.dimToUnlimited(opts.values, opts.ncfile)

trunk/tools/nc_var_tools.py

@@ -15866 +15866 @@
     return 
 
+def DatesFiles(values, ncfile, varn):
+    """ Function to find different time values on a series of WRF files in a folder
+      [values]= [dates],[folder]
+        [dates]: ':' list of [YYYY][MM][DD][HH][MI][SS] dates
+        [fold]: folder with the location of the netCDF files
+      [ncfile]= header of the name of the files to concatenate [ncfile]*
+      [varn]= name of the variable Time ('WRFt', for WRF time)
+    """
+    import subprocess as sub
+    fname='DatesFiles'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print DatesFiles.__doc__
+        quit()
+
+    arguments = '[dates],[fold]'
+    check_arguments(fname, len(values.split(',')), arguments, ',',                   \
+      len(argumenst.split(',')))
+
+    dates = values.split(',')[0].split(',')
+    fold = values.split(',')[1]
+
+    confiles = files_folder(fold,ncfile)
+    Nfiles = len(confiles)
+
+    print '  ' + fname +': looking on:',Nfiles,'files'
+    print confiles
+
+# Opening all files and getting times
+    ncobjs = {}
+    for filen in confiles:
+        print 'charging: ',filen
+        filename = fold + '/' + filen
+        ncf = NetCDFFile(fname, 'r')
+        if varn == 'WRFt':
+            fdates = ncf.variables['Times'][:]
+            datesv = datetimeStr_conversion(fdates, 'WRFdatetime', 'YmdHMS')
+        else:
+            tunits = ncf.variables[varn].getncattr('units')
+            fdates = ncf.variables[varn][:]
+            datesv = datetimeStr_conversion(fdates, 'cfTime,' + tunits , 'YmdHMS')
+
+        ncobjs[fname] = datesv
+    
+    founddates = {}
+    for dt in dates:
+        for fn in ncobjs.keys():
+            if searchInlist(ncobjs[fn], dt):
+                founddates[dt] = fn
+                istep = ncobjs[fn].index(dt)
+
+                print fname + ':',dt,fn,istep
+
+    return
 
 #quit()