Changeset 1939 in lmdz_wrf for trunk

Timestamp:

Jul 17, 2018, 9:22:53 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `mindist': Function to proivde the minimum distance toa pair of lon,lat
• `get_point': Function to provide the closest grid point to a given lat,lon
• `get_time': Function to provide the closest time-step to a given time

Location:

trunk/tools

Files:

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

trunk/tools/nc_var.py

@@ -24 +24 @@
 ## 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. # nc_var.py -o get_point -S 'XLONG:XLAT:Time|0' -f /home/lluis/PY/wrfout_d01_1995-01-01_00:00:00 -v -45.,-30.
+## e.g. # nc_var.py -o get_time -S '1422759600;seconds!since!1949-12-01!00:00:00' -f /home/lluis/PY/wrfout_d01_1995-01-01_00:00:00 -v WRFtime
 
 ## e.g. ccrc468-17 # ./nc_var.py -v time -f 123/CCRC_NARCliM_Sydney_All_1990-1999_pr10max.nc -o out -S 1:-1
@@ -107 +109 @@
 # gaddattr: Add a global attribute to a netCDF. Removes previous attribute if it exist
 # get_attribute: Function to get an attribute from a netCDF file
+# get_point: Function to provide the closest grid point to a given lat,lon
+# get_time: Function to provide the closest time-step to a given time
 # get_namelist_vars: Function to get namelist-like  values ([varname] = [value]) 
 # get_Variables: Function to get a list of variables from an existing file
@@ -196 +200 @@
   'file_oper_alongdims', 'filter_2dim',                                              \
   'flipdim', 'fvaradd', 'gaddattrk', 'gaddattr', 'get_attribute',                    \
-  'get_namelist_vars', 'get_Variables',                                              \
+  'get_point', 'get_time', 'get_namelist_vars', 'get_Variables',                     \
   'getvalues_lonlat', 'getvars_tofile', 'grattr',                                    \
   'grmattr', 'idims', 'ifile', 'igattrs', 'increaseDimvar', 'isgattrs',              \
@@ -374 +378 @@
 elif oper == 'get_namelist_vars':
     ncvar.get_namelist_vars(opts.values, opts.ncfile)
+elif oper == 'get_point':
+    ncvar.get_point(opts.values, opts.ncfile, opts.varname)
+elif oper == 'get_time':
+    ncvar.get_time(opts.values, opts.ncfile, opts.varname)
 elif oper == 'get_Variables':
     ncvar.get_Variables(opts.values, opts.ncfile, opts.varname)

trunk/tools/nc_var_tools.py

@@ -82 +82 @@
 # getdim_listonc: Function to get a dimension object from a list of netCDF file object
 # getvar_listonc: Function to get a variable object from a list of netCDF file object
+# get_time: Function to provide the closest time-step to a given time
+# get_point: Function to provide the closest grid point to a given lat,lon
 # get_namelist_vars: Function to get namelist-like  values ([varname] = [value])
 # get_str_nc: Function to get string values in a netCDF variable as a chains of 1char values
@@ -106 +108 @@
 # maskvar: Function to mask a variable using a mask. It is assumed that mask[...,dimM,dimJ,dimK] and var[...,dimM,dimJ,dimK] share the last dimensions
 # merge_files: Function to merge variables from two files
+# mindist: Function to proivde the minimum distance toa pair of lon,lat
 # ModelChar: Class object to determine major characterisitcs of a given model output
 # model_characteristics: Functino to provide major characterisitcs of a given model output
@@ -24433 +24436 @@
 #itime('WRFtime', '../PY/wrfout_d01_1995-01-01_00:00:00', 'Times')
 
+def mindist(lonv,latv,lonvalue,latvalue):
+    """ Function to proivde the minimum distance toa pair of lon,lat
+     lonv: matrix with longitudes
+     latv: matrix with longitudes
+     lonvalue: longitude to look for
+     latvalue: latitude to look for
+    """
+    fname = 'mindist'
+
+    if len(lonv.shape) == 1:
+        distlon = np.abs(lonv - lonvalue)
+        distlat = np.abs(latv - latvalue)
+
+        mindifflon = np.min(distlon)
+        mindifflat = np.min(distlat)
+        ix = gen.index_vec(distlon, mindistlon)
+        iy = gen.index_vec(distlat, mindistlat)
+    else:
+        dist = np.sqrt((lonv - lonvalue)**2 + (latv - latvalue)**2)
+        mindist = np.min(dist)
+        [iy, ix] = gen.index_mat(dist, mindist)
+
+    return [ix, iy]
+
+def get_point(values, ncfile, variable):
+    """ Function to provide the closest grid point to a given lat,lon
+      values=[lonvarn]:[latvarn]:[dimvals]
+        [lonvarn]: name of the variable with the longitudes
+        [latvarn]: name of the variable with the latitudes
+        [dimvals]: ',' list of [dimname]|[value] telling at which dimension of the 
+          variable a given value is required:
+            * [integer]: which value of the dimension
+            * -1: all along the dimension
+            * -9: last value of the dimension
+            * [beg]@[end]@[inc] slice from [beg] to [end] every [inc]
+            * NOTE, no dim name all the dimension size        
+      ncfile= file from which provide the information
+      variable= [lonval],[latval] longitude and latitude values to look for
+    """
+    fname = 'get_point'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print get_point.__doc__
+        quit()
+
+    expectargs = '[lonvarn]:[latvarn]:[dimvals]'
+    gen.check_arguments(fname, values, expectargs, ':')
+
+    lonvarn = values.split(':')[0]
+    latvarn = values.split(':')[1]
+    dimvals = values.split(':')[2].replace('|',':')
+
+    [lonvalue, latvalue] = gen.str_list_k(variable, ',', 'R')
+
+    onc = NetCDFFile(ncfile, 'r')
+
+    if not onc.variables.has_key(lonvarn):
+        print errormsg
+        print '  ' + fname + ": file '" + ncfile + "' does not have longitude " +    \
+          "varible '" + lonvarn + "' !!"
+        print '    available ones:', onc.variables.keys()
+        quit(-1)
+    if not onc.variables.has_key(latvarn):
+        print errormsg
+        print '  ' + fname + ": file '" + ncfile + "' does not have latitude " +     \
+          "varible '" + latvarn + "' !!"
+        print '    available ones:', onc.variables.keys()
+        quit(-1)
+
+    olon = onc.variables[lonvarn]
+    olat = onc.variables[latvarn]
+
+    lonv, londims = slice_variable(olon, dimvals.replace(',','|'))
+    latv, latdims = slice_variable(olat, dimvals.replace(',','|'))
+
+    if len(lonv.shape) > 2:
+        print errormsg
+        print '  ' + fname + ': wrong shape for longitudes. It must be <= 2D!!'
+        print '    provided shape after slice:', lonv.shape, 'slice', dimvals
+        quit(-1)
+    if len(latv.shape) > 2:
+        print errormsg
+        print '  ' + fname + ': wrong shape for latitudes. It must be <= 2D!!'
+        print '    provided shape after slice:', latv.shape, 'slice', dimvals
+        quit(-1)
+
+    onc.close()
+    # looking for minimum distance
+    gridpoint = mindist(lonv,latv,lonvalue,latvalue)
+
+    print str(gridpoint[0])+','+str(gridpoint[1])
+
+    return 
+
+#print get_point('XLONG:XLAT:Time|0', '/home/lluis/PY/wrfout_d01_1995-01-01_00:00:00', '-45.,-30.')
+
+def get_time(values, ncfile, variable):
+    """ Function to provide the closest time-step to a given time
+      values= [timevalue];[units]
+        [timevalue]: CF value of time to look for
+        [units]: CF units of time [Tunit] since [date] ('!' for spaces)
+      ncfile= file from which provide the information
+      variable= [timevarn] name of the variable with the CF-times ('WRFtime' for WRF
+        Times variable)
+    """
+    fname = 'get_time'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print get_time.__doc__
+        quit()
+
+    expectargs = '[timevalue];[units]'
+    gen.check_arguments(fname, values, expectargs, ';')
+
+    timevalue = np.float(values.split(';')[0])
+    uTunits = values.split(';')[1].replace('!',' ')
+
+    onc = NetCDFFile(ncfile, 'r')
+
+    if not variable == 'WRFtime' and not onc.variables.has_key(variable):
+        print errormsg
+        print '  ' + fname + ": file '" + ncfile + "' does not have time " +         \
+          "varible '" + variable + "' !!"
+        print '    available ones:', onc.variables.keys()
+        quit(-1)
+
+    if variable == 'WRFtime':
+        otime = onc.variables['Times']
+        timev0, timeu = compute_WRFtime(otime[:])
+    else:
+        otime = onc.variables[variable]
+        timev0 = otime[:]
+        timeu = otime.units
+
+    onc.close()
+
+    # Referring to the same units
+    reftvals = gen.coincident_CFtimes(timev0, uTunits, timeu)
+
+    # looking for minimum distance
+    dist = np.abs(reftvals - timevalue)
+    mindist = np.min(dist)
+    it = gen.index_vec(dist, mindist)
+
+    print it
+
+    return 
+
+#get_time('1422759600;seconds since 1949-12-01 00:00:00', '/home/lluis/PY/wrfout_d01_1995-01-01_00:00:00', 'WRFtime')
+
 #quit()