nilyin
/
wrf-python
forked from 3rdparty/wrf-python
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

Now squeezes single element multidim arrays. Timeidx made a positional argument and removed from kargs check

main
Bill Ladwig 10 years ago
parent
5135d89896
commit
5b89a89c2e
19 changed files with 120 additions and 100 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 88
      wrf_open/var/src/python/wrf/var/__init__.py
  2. 4
      wrf_open/var/src/python/wrf/var/cape.py
  3. 2
      wrf_open/var/src/python/wrf/var/ctt.py
  4. 4
      wrf_open/var/src/python/wrf/var/dbz.py
  5. 6
      wrf_open/var/src/python/wrf/var/destag.py
  6. 4
      wrf_open/var/src/python/wrf/var/dewpoint.py
  7. 2
      wrf_open/var/src/python/wrf/var/extension.py
  8. 8
      wrf_open/var/src/python/wrf/var/geoht.py
  9. 4
      wrf_open/var/src/python/wrf/var/helicity.py
  10. 8
      wrf_open/var/src/python/wrf/var/interp.py
  11. 2
      wrf_open/var/src/python/wrf/var/pressure.py
  12. 2
      wrf_open/var/src/python/wrf/var/slp.py
  13. 10
      wrf_open/var/src/python/wrf/var/temp.py
  14. 2
      wrf_open/var/src/python/wrf/var/terrain.py
  15. 11
      wrf_open/var/src/python/wrf/var/times.py
  16. 4
      wrf_open/var/src/python/wrf/var/units.py
  17. 35
      wrf_open/var/src/python/wrf/var/util.py
  18. 14
      wrf_open/var/src/python/wrf/var/uvmet.py
  19. 10
      wrf_open/var/src/python/wrf/var/wind.py

88
wrf_open/var/src/python/wrf/var/__init__.py
Unescape View File

@ -124,43 +124,43 @@ _FUNC_MAP = {"cape2d" : get_2dcape,
"ctt" : get_ctt "ctt" : get_ctt
} }
_VALID_ARGS = {"cape2d" : ["missing", "timeidx"], _VALID_KARGS = {"cape2d" : ["missing"],
"cape3d" : ["missing", "timeidx"], "cape3d" : ["missing"],
"dbz" : ["do_variant", "do_liqskin", "timeidx"], "dbz" : ["do_variant", "do_liqskin"],
"maxdbz" : ["do_variant", "do_liqskin", "timeidx"], "maxdbz" : ["do_variant", "do_liqskin"],
"dp" : ["timeidx", "units"], "dp" : ["units"],
"dp2m" : ["timeidx", "units"], "dp2m" : ["units"],
"height" : ["msl", "units", "timeidx"], "height" : ["msl", "units"],
"geopt" : ["timeidx"], "geopt" : [],
"srh" : ["top", "timeidx"], "srh" : ["top"],
"uhel" : ["bottom", "top", "timeidx"], "uhel" : ["bottom", "top"],
"omega" : ["timeidx"], "omega" : [],
"pw" : ["timeidx"], "pw" : [],
"rh" : ["timeidx"], "rh" : [],
"rh2m" : ["timeidx"], "rh2m" : [],
"slp" : ["units", "timeidx"], "slp" : ["units"],
"temp" : ["units", "timeidx"], "temp" : ["units"],
"theta" : ["units", "timeidx"], "theta" : ["units"],
"theta_e" : ["timeidx", "units"], "theta_e" : ["units"],
"tv" : ["units", "timeidx"], "tv" : ["units"],
"twb" : ["units", "timeidx"], "twb" : ["units"],
"terrain" : ["units", "timeidx"], "terrain" : ["units"],
"times" : ["timeidx"], "times" : [],
"uvmet" : ["units", "timeidx"], "uvmet" : ["units"],
"uvmet10" : ["units", "timeidx"], "uvmet10" : ["units"],
"avo" : ["timeidx"], "avo" : [],
"pvo" : ["timeidx"], "pvo" : [],
"ua" : ["units", "timeidx"], "ua" : ["units"],
"va" : ["units", "timeidx"], "va" : ["units"],
"wa" : ["units", "timeidx"], "wa" : ["units"],
"lat" : ["timeidx"], "lat" : [],
"lon" : ["timeidx"], "lon" : [],
"pressure" : ["units", "timeidx"], "pressure" : ["units"],
"wspddir" : ["units", "timeidx"], "wspddir" : ["units"],
"wspddir_uvmet" : ["units", "timeidx"], "wspddir_uvmet" : ["units"],
"wspddir_uvmet10" : ["units", "timeidx"], "wspddir_uvmet10" : ["units"],
"ctt" : ["timeidx"], "ctt" : [],
"default" : ["timeidx"] "default" : []
} }
_ALIASES = {"cape_2d" : "cape2d", _ALIASES = {"cape_2d" : "cape2d",
@ -198,25 +198,21 @@ def _undo_alias(alias):
def _check_kargs(var, kargs): def _check_kargs(var, kargs):
for arg in kargs.iterkeys(): for arg in kargs.iterkeys():
if arg not in _VALID_ARGS[var]: if arg not in _VALID_KARGS[var]:
raise ArgumentError("'%s' is an invalid keyword " raise ArgumentError("'%s' is an invalid keyword "
"argument for '%s" % (arg, var)) "argument for '%s'" % (arg, var))
def getvar(wrfnc, var, **kargs): def getvar(wrfnc, var, timeidx=0, **kargs):
if is_standard_wrf_var(wrfnc, var): if is_standard_wrf_var(wrfnc, var):
if "timeidx" in kargs:
timeidx = kargs["timeidx"]
else:
timeidx = 0
return extract_vars(wrfnc, timeidx, var)[var] return extract_vars(wrfnc, timeidx, var)[var]
actual_var = _undo_alias(var) actual_var = _undo_alias(var)
if actual_var not in _VALID_ARGS: if actual_var not in _VALID_KARGS:
raise ArgumentError("'%s' is not a valid variable name" % (var)) raise ArgumentError("'%s' is not a valid variable name" % (var))
_check_kargs(actual_var, kargs) _check_kargs(actual_var, kargs)
return _FUNC_MAP[actual_var](wrfnc,**kargs) return _FUNC_MAP[actual_var](wrfnc,timeidx,**kargs)

4
wrf_open/var/src/python/wrf/var/cape.py
Unescape View File

@ -8,7 +8,7 @@ from wrf.var.util import extract_vars
__all__ = ["get_2dcape", "get_3dcape"] __all__ = ["get_2dcape", "get_3dcape"]
def get_2dcape(wrfnc, missing=-999999.0, timeidx=0): def get_2dcape(wrfnc, timeidx=0, missing=-999999.0):
"""Return the 2d fields of cape, cin, lcl, and lfc""" """Return the 2d fields of cape, cin, lcl, and lfc"""
ncvars = extract_vars(wrfnc, timeidx, vars=("T", "P", "PB", "QVAPOR", "PH", ncvars = extract_vars(wrfnc, timeidx, vars=("T", "P", "PB", "QVAPOR", "PH",
"PHB", "HGT", "PSFC")) "PHB", "HGT", "PSFC"))
@ -60,7 +60,7 @@ def get_2dcape(wrfnc, missing=-999999.0, timeidx=0):
return ma.masked_values(res,missing) return ma.masked_values(res,missing)
def get_3dcape(wrfnc, missing=-999999.0, timeidx=0): def get_3dcape(wrfnc, timeidx=0, missing=-999999.0):
"""Return the 3d fields of cape and cin""" """Return the 3d fields of cape and cin"""
ncvars = extract_vars(wrfnc, timeidx, vars=("T", "P", "PB", "QVAPOR", ncvars = extract_vars(wrfnc, timeidx, vars=("T", "P", "PB", "QVAPOR",

2
wrf_open/var/src/python/wrf/var/ctt.py
Unescape View File

@ -10,7 +10,7 @@ from wrf.var.util import extract_vars
__all__ = ["get_ctt"] __all__ = ["get_ctt"]
@convert_units("temp", "c") @convert_units("temp", "c")
def get_ctt(wrfnc, units="c", timeidx=0): def get_ctt(wrfnc, timeidx=0, units="c"):
"""Return the cloud top temperature. """Return the cloud top temperature.
""" """

4
wrf_open/var/src/python/wrf/var/dbz.py
Unescape View File

@ -6,7 +6,7 @@ from wrf.var.util import extract_vars
__all__ = ["get_dbz", "get_max_dbz"] __all__ = ["get_dbz", "get_max_dbz"]
def get_dbz(wrfnc, do_varint=False, do_liqskin=False, timeidx=0): def get_dbz(wrfnc, timeidx=0, do_varint=False, do_liqskin=False):
""" Return the dbz """ Return the dbz
do_varint - do variable intercept (if False, constants are used. Otherwise, do_varint - do variable intercept (if False, constants are used. Otherwise,
@ -58,7 +58,7 @@ def get_dbz(wrfnc, do_varint=False, do_liqskin=False, timeidx=0):
return computedbz(full_p,tk,qv,qr,qs,qg,sn0,ivarint,iliqskin) return computedbz(full_p,tk,qv,qr,qs,qg,sn0,ivarint,iliqskin)
def get_max_dbz(wrfnc, do_varint=False, do_liqskin=False, timeidx=0): def get_max_dbz(wrfnc, timeidx=0, do_varint=False, do_liqskin=False):
return n.amax(get_dbz(wrfnc, do_varint, do_liqskin, timeidx), return n.amax(get_dbz(wrfnc, do_varint, do_liqskin, timeidx),
axis=-3) axis=-3)

6
wrf_open/var/src/python/wrf/var/destag.py
Unescape View File

@ -15,7 +15,7 @@ def destagger(var, stagger_dim):
""" """
var_shape = var.shape var_shape = var.shape
num_dims = len(var_shape) num_dims = var.ndim
stagger_dim_size = var_shape[stagger_dim] stagger_dim_size = var_shape[stagger_dim]
# Dynamically building the range slices to create the appropriate # Dynamically building the range slices to create the appropriate
@ -29,8 +29,8 @@ def destagger(var, stagger_dim):
slice2 = slice(1, stagger_dim_size, 1) slice2 = slice(1, stagger_dim_size, 1)
# default to full slices # default to full slices
dim_ranges_1 = [full_slice for x in xrange(num_dims)] dim_ranges_1 = [full_slice] * num_dims
dim_ranges_2 = [full_slice for x in xrange(num_dims)] dim_ranges_2 = [full_slice] * num_dims
# for the stagger dim, insert the appropriate slice range # for the stagger dim, insert the appropriate slice range
dim_ranges_1[stagger_dim] = slice1 dim_ranges_1[stagger_dim] = slice1

4
wrf_open/var/src/python/wrf/var/dewpoint.py
Unescape View File

@ -5,7 +5,7 @@ from wrf.var.util import extract_vars
__all__ = ["get_dp", "get_dp_2m"] __all__ = ["get_dp", "get_dp_2m"]
@convert_units("temp", "c") @convert_units("temp", "c")
def get_dp(wrfnc, units="c", timeidx=0): def get_dp(wrfnc, timeidx=0, units="c"):
ncvars = extract_vars(wrfnc, timeidx, vars=("P", "PB", "QVAPOR")) ncvars = extract_vars(wrfnc, timeidx, vars=("P", "PB", "QVAPOR"))
@ -21,7 +21,7 @@ def get_dp(wrfnc, units="c", timeidx=0):
return td return td
@convert_units("temp", "c") @convert_units("temp", "c")
def get_dp_2m(wrfnc, units="c", timeidx=0): def get_dp_2m(wrfnc, timeidx=0, units="c"):
ncvars = extract_vars(wrfnc, timeidx, vars=("PSFC", "Q2")) ncvars = extract_vars(wrfnc, timeidx, vars=("PSFC", "Q2"))
# Algorithm requires hPa # Algorithm requires hPa

2
wrf_open/var/src/python/wrf/var/extension.py
Unescape View File

@ -23,6 +23,7 @@ class FortranException(Exception):
def __call__(self, message): def __call__(self, message):
raise self.__class__(message) raise self.__class__(message)
@handle_left_iter(2,3,0)
@handle_casting(arg_idxs=(0,1)) @handle_casting(arg_idxs=(0,1))
def interpz3d(data3d,zdata,desiredloc,missingval): def interpz3d(data3d,zdata,desiredloc,missingval):
res = f_interpz3d(data3d.T, res = f_interpz3d(data3d.T,
@ -32,6 +33,7 @@ def interpz3d(data3d,zdata,desiredloc,missingval):
return res.T return res.T
@handle_casting(arg_idxs=(0,1)) @handle_casting(arg_idxs=(0,1))
@handle_left_iter(2,2,0)
def interpz2d(data3d,xy): def interpz2d(data3d,xy):
res = f_interp2dxy(data3d.T, res = f_interp2dxy(data3d.T,
xy.T) xy.T)

8
wrf_open/var/src/python/wrf/var/geoht.py
Unescape View File

@ -5,7 +5,7 @@ from wrf.var.util import extract_vars
__all__ = ["get_geopt", "get_height"] __all__ = ["get_geopt", "get_height"]
def _get_geoht(wrfnc, height=True, msl=True, timeidx=0): def _get_geoht(wrfnc, timeidx, height=True, msl=True):
"""Return the geopotential in units of m2 s-2 if height is False, """Return the geopotential in units of m2 s-2 if height is False,
otherwise return the geopotential height in meters. If height is True, otherwise return the geopotential height in meters. If height is True,
then if msl is True the result will be in MSL, otherwise AGL (the terrain then if msl is True the result will be in MSL, otherwise AGL (the terrain
@ -40,10 +40,10 @@ def _get_geoht(wrfnc, height=True, msl=True, timeidx=0):
return geopt_unstag return geopt_unstag
def get_geopt(wrfnc, timeidx=0): def get_geopt(wrfnc, timeidx=0):
return _get_geoht(wrfnc, False, timeidx=timeidx) return _get_geoht(wrfnc, timeidx, False)
@convert_units("height", "m") @convert_units("height", "m")
def get_height(wrfnc, msl=True, units="m", timeidx=0): def get_height(wrfnc, timeidx=0, msl=True, units="m"):
z = _get_geoht(wrfnc, True, msl, timeidx) z = _get_geoht(wrfnc, timeidx, True, msl)
return z return z

4
wrf_open/var/src/python/wrf/var/helicity.py
Unescape View File

@ -6,7 +6,7 @@ from wrf.var.util import extract_vars, extract_global_attrs
__all__ = ["get_srh", "get_uh"] __all__ = ["get_srh", "get_uh"]
def get_srh(wrfnc, top=3000.0, timeidx=0): def get_srh(wrfnc, timeidx=0, top=3000.0):
# Top can either be 3000 or 1000 (for 0-1 srh or 0-3 srh) # Top can either be 3000 or 1000 (for 0-1 srh or 0-3 srh)
ncvars = extract_vars(wrfnc, timeidx, vars=("HGT", "PH", "PHB")) ncvars = extract_vars(wrfnc, timeidx, vars=("HGT", "PH", "PHB"))
@ -53,7 +53,7 @@ def get_srh(wrfnc, top=3000.0, timeidx=0):
return srh return srh
def get_uh(wrfnc, bottom=2000.0, top=5000.0, timeidx=0): def get_uh(wrfnc, timeidx=0, bottom=2000.0, top=5000.0):
ncvars = extract_vars(wrfnc, timeidx, vars=("W", "PH", "PHB", "MAPFAC_M")) ncvars = extract_vars(wrfnc, timeidx, vars=("W", "PH", "PHB", "MAPFAC_M"))

8
wrf_open/var/src/python/wrf/var/interp.py
Unescape View File

@ -98,7 +98,7 @@ def _get_xy(xdim, ydim, pivot_point=None, angle=None,
x1 = (y1 - intercept)/slope x1 = (y1 - intercept)/slope
if( y1 > ydim-1):# intersect outside of top boundary if( y1 > ydim-1):# intersect outside of top boundary
y1 = ydim1 y1 = ydim-1
x1 = (y1 - intercept)/slope x1 = (y1 - intercept)/slope
elif start_point is not None and end_point is not None: elif start_point is not None and end_point is not None:
x0 = start_point[0] x0 = start_point[0]
@ -118,7 +118,7 @@ def _get_xy(xdim, ydim, pivot_point=None, angle=None,
npts = int(distance) npts = int(distance)
dxy = distance/npts dxy = distance/npts
xz = n.zeros((npts,2), "float") xy = n.zeros((npts,2), "float")
dx = dx/npts dx = dx/npts
dy = dy/npts dy = dy/npts
@ -134,8 +134,8 @@ def _get_xy(xdim, ydim, pivot_point=None, angle=None,
def get_vertcross(data3d, z, missingval=-99999, def get_vertcross(data3d, z, missingval=-99999,
pivot_point=None,angle=None,start_point=None,end_point=None): pivot_point=None,angle=None,start_point=None,end_point=None):
xdim = z.shape[2] xdim = z.shape[-1]
ydim = z.shape[1] ydim = z.shape[-2]
xy = _get_xy(xdim, ydim, pivot_point, angle, start_point, end_point) xy = _get_xy(xdim, ydim, pivot_point, angle, start_point, end_point)

2
wrf_open/var/src/python/wrf/var/pressure.py
Unescape View File

@ -6,7 +6,7 @@ from wrf.var.util import extract_vars
__all__ = ["get_pressure"] __all__ = ["get_pressure"]
@convert_units("pressure", "pa") @convert_units("pressure", "pa")
def get_pressure(wrfnc, units="hpa", timeidx=0): def get_pressure(wrfnc, timeidx=0, units="hpa"):
try: try:
p_vars = extract_vars(wrfnc, timeidx, vars=("P", "PB")) p_vars = extract_vars(wrfnc, timeidx, vars=("P", "PB"))

2
wrf_open/var/src/python/wrf/var/slp.py
Unescape View File

@ -7,7 +7,7 @@ from wrf.var.util import extract_vars
__all__ = ["get_slp"] __all__ = ["get_slp"]
@convert_units("pressure", "hpa") @convert_units("pressure", "hpa")
def get_slp(wrfnc, units="hpa", timeidx=0): def get_slp(wrfnc, timeidx=0, units="hpa"):
ncvars = extract_vars(wrfnc, timeidx, vars=("T", "P", "PB", "QVAPOR", ncvars = extract_vars(wrfnc, timeidx, vars=("T", "P", "PB", "QVAPOR",
"PH", "PHB")) "PH", "PHB"))

10
wrf_open/var/src/python/wrf/var/temp.py
Unescape View File

@ -7,7 +7,7 @@ from wrf.var.util import extract_vars
__all__ = ["get_theta", "get_temp", "get_eth", "get_tv", "get_tw"] __all__ = ["get_theta", "get_temp", "get_eth", "get_tv", "get_tw"]
@convert_units("temp", "k") @convert_units("temp", "k")
def get_theta(wrfnc, units="k", timeidx=0): def get_theta(wrfnc, timeidx=0, units="k"):
ncvars = extract_vars(wrfnc, timeidx, vars="T") ncvars = extract_vars(wrfnc, timeidx, vars="T")
t = ncvars["T"] t = ncvars["T"]
full_t = t + Constants.T_BASE full_t = t + Constants.T_BASE
@ -15,7 +15,7 @@ def get_theta(wrfnc, units="k", timeidx=0):
return full_t return full_t
@convert_units("temp", "k") @convert_units("temp", "k")
def get_temp(wrfnc, units="k", timeidx=0): def get_temp(wrfnc, timeidx=0, units="k"):
"""Return the temperature in Kelvin or Celsius""" """Return the temperature in Kelvin or Celsius"""
ncvars = extract_vars(wrfnc, timeidx, vars=("T", "P", "PB")) ncvars = extract_vars(wrfnc, timeidx, vars=("T", "P", "PB"))
@ -30,7 +30,7 @@ def get_temp(wrfnc, units="k", timeidx=0):
return tk return tk
@convert_units("temp", "k") @convert_units("temp", "k")
def get_eth(wrfnc, units="k", timeidx=0): def get_eth(wrfnc, timeidx=0, units="k"):
"Return equivalent potential temperature (Theta-e) in Kelvin" "Return equivalent potential temperature (Theta-e) in Kelvin"
ncvars = extract_vars(wrfnc, timeidx, vars=("T", "P", "PB", "QVAPOR")) ncvars = extract_vars(wrfnc, timeidx, vars=("T", "P", "PB", "QVAPOR"))
@ -48,7 +48,7 @@ def get_eth(wrfnc, units="k", timeidx=0):
return eth return eth
@convert_units("temp", "k") @convert_units("temp", "k")
def get_tv(wrfnc, units="k", timeidx=0): def get_tv(wrfnc, timeidx=0, units="k"):
"Return the virtual temperature (tv) in Kelvin or Celsius" "Return the virtual temperature (tv) in Kelvin or Celsius"
ncvars = extract_vars(wrfnc, timeidx, vars=("T", "P", "PB", "QVAPOR")) ncvars = extract_vars(wrfnc, timeidx, vars=("T", "P", "PB", "QVAPOR"))
@ -68,7 +68,7 @@ def get_tv(wrfnc, units="k", timeidx=0):
@convert_units("temp", "k") @convert_units("temp", "k")
def get_tw(wrfnc, units="k", timeidx=0): def get_tw(wrfnc, timeidx=0, units="k"):
"Return the wetbulb temperature (tw)" "Return the wetbulb temperature (tw)"
ncvars = extract_vars(wrfnc, timeidx, vars=("T", "P", "PB", "QVAPOR")) ncvars = extract_vars(wrfnc, timeidx, vars=("T", "P", "PB", "QVAPOR"))

2
wrf_open/var/src/python/wrf/var/terrain.py
Unescape View File

@ -5,7 +5,7 @@ from wrf.var.util import extract_vars
__all__ = ["get_terrain"] __all__ = ["get_terrain"]
@convert_units("height", "m") @convert_units("height", "m")
def get_terrain(wrfnc, units="m", timeidx=0): def get_terrain(wrfnc, timeidx=0, units="m"):
try: try:
hgt_vars = extract_vars(wrfnc, timeidx, vars="HGT") hgt_vars = extract_vars(wrfnc, timeidx, vars="HGT")

11
wrf_open/var/src/python/wrf/var/times.py
Unescape View File

@ -1,12 +1,7 @@
import datetime as dt from wrf.var.util import extract_times
__all__ = ["get_times"] __all__ = ["get_times"]
def _make_time(timearr): def get_times(wrfnc,timeidx=0):
return dt.datetime.strptime("".join(timearr[:]), "%Y-%m-%d_%H:%M:%S") return extract_times(wrfnc,timeidx)
# TODO: handle list of files and multiple times
def get_times(wrfnc):
times = wrfnc.variables["Times"][:,:]
return [_make_time(times[i,:]) for i in xrange(times.shape[0])]

4
wrf_open/var/src/python/wrf/var/units.py
Unescape View File

@ -108,9 +108,9 @@ _TEMP_CONV_METHODS = {"k" : _c_to_k,
"f" : _c_to_f "f" : _c_to_f
} }
def check_units(unit, type): def check_units(unit, unit_type):
unitl = unit.lower() unitl = unit.lower()
if unitl not in _VALID_UNITS[type]: if unitl not in _VALID_UNITS[unit_type]:
raise ValueError("invalid unit type '%s'" % unit) raise ValueError("invalid unit type '%s'" % unit)
def do_conversion(var, vartype, var_unit, dest_unit): def do_conversion(var, vartype, var_unit, dest_unit):

35
wrf_open/var/src/python/wrf/var/util.py
Unescape View File

@ -1,9 +1,10 @@
from collections import Iterable from collections import Iterable
import datetime as dt
import numpy as n import numpy as n
__all__ = ["extract_vars", "extract_global_attrs", "extract_dim", __all__ = ["extract_vars", "extract_global_attrs", "extract_dim",
"combine_files", "is_standard_wrf_var"] "combine_files", "is_standard_wrf_var", "extract_times"]
def _is_multi_time(timeidx): def _is_multi_time(timeidx):
if timeidx == -1: if timeidx == -1:
@ -53,14 +54,14 @@ def combine_files(wrflist, var, timeidx):
else: else:
time_idx_or_slice = slice(None, None, None) time_idx_or_slice = slice(None, None, None)
first_var = wrflist[0].variables[var][time_idx_or_slice, :] first_var = n.squeeze(wrflist[0].variables[var][time_idx_or_slice, :])
outdims = [numfiles] outdims = [numfiles]
outdims += first_var.shape outdims += first_var.shape
outarr = n.zeros(outdims, first_var.dtype) outarr = n.zeros(outdims, first_var.dtype)
outarr[0,:] = first_var[:] outarr[0,:] = first_var[:]
for idx, wrfnc in enumerate(wrflist[1:], start=1): for idx, wrfnc in enumerate(wrflist[1:], start=1):
outarr[idx,:] = wrfnc.variables[var][time_idx_or_slice, :] outarr[idx,:] = n.squeeze(wrfnc.variables[var][time_idx_or_slice, :])
return outarr return outarr
@ -77,10 +78,36 @@ def extract_vars(wrfnc, timeidx, vars):
if not multitime and not multifile: if not multitime and not multifile:
return {var:wrfnc.variables[var][timeidx,:] for var in varlist} return {var:wrfnc.variables[var][timeidx,:] for var in varlist}
elif multitime and not multifile: elif multitime and not multifile:
return {var:wrfnc.variables[var][:] for var in varlist} return {var:n.squeeze(wrfnc.variables[var][:]) for var in varlist}
else: else:
return {var:combine_files(wrfnc, var, timeidx) for var in varlist} return {var:combine_files(wrfnc, var, timeidx) for var in varlist}
def _make_time(timearr):
return dt.datetime.strptime("".join(timearr[:]), "%Y-%m-%d_%H:%M:%S")
def _file_times(wrfnc,timeidx):
multitime = _is_multi_time(timeidx)
if multitime:
times = wrfnc.variables["Times"][:,:]
for i in xrange(times.shape[0]):
yield _make_time(times[i,:])
else:
times = wrfnc.variables["Times"][timeidx,:]
yield _make_time(times)
def extract_times(wrfnc,timeidx):
multi_file = _is_multi_file(wrfnc)
if not multi_file:
wrf_list = [wrfnc]
else:
wrf_list = wrfnc
return [file_time
for wrf_file in wrf_list
for file_time in _file_times(wrf_file,timeidx) ]
def is_standard_wrf_var(wrfnc, var): def is_standard_wrf_var(wrfnc, var):
multifile = _is_multi_file(wrfnc) multifile = _is_multi_file(wrfnc)
if multifile: if multifile:

14
wrf_open/var/src/python/wrf/var/uvmet.py
Unescape View File

@ -11,7 +11,7 @@ __all__=["get_uvmet", "get_uvmet10", "get_uvmet_wspd_wdir",
"get_uvmet10_wspd_wdir"] "get_uvmet10_wspd_wdir"]
@convert_units("wind", "mps") @convert_units("wind", "mps")
def get_uvmet(wrfnc, ten_m=False, units ="mps", timeidx=0): def get_uvmet(wrfnc, timeidx=0, ten_m=False, units ="mps"):
""" Return a tuple of u,v with the winds rotated in to earth space""" """ Return a tuple of u,v with the winds rotated in to earth space"""
if not ten_m: if not ten_m:
@ -144,15 +144,15 @@ def get_uvmet(wrfnc, ten_m=False, units ="mps", timeidx=0):
return res return res
def get_uvmet10(wrfnc, units="mps", timeidx=0): def get_uvmet10(wrfnc, timeidx=0, units="mps"):
return get_uvmet(wrfnc, True, units, timeidx) return get_uvmet(wrfnc, timeidx, True, units)
def get_uvmet_wspd_wdir(wrfnc, units="mps", timeidx=0): def get_uvmet_wspd_wdir(wrfnc, timeidx=0, units="mps"):
u,v = get_uvmet(wrfnc, False, units, timeidx) u,v = get_uvmet(wrfnc, timeidx, False, units)
return _calc_wspd_wdir(u, v, units) return _calc_wspd_wdir(u, v, units)
def get_uvmet10_wspd_wdir(wrfnc, units="mps", timeidx=0): def get_uvmet10_wspd_wdir(wrfnc, timeidx=0, units="mps"):
u,v = get_uvmet10(wrfnc, units="mps", timeidx=0) u,v = get_uvmet10(wrfnc, timeidx, units="mps")
return _calc_wspd_wdir(u, v, units) return _calc_wspd_wdir(u, v, units)

10
wrf_open/var/src/python/wrf/var/wind.py
Unescape View File

@ -16,25 +16,25 @@ def _calc_wdir(u, v):
wdir = 270.0 - n.arctan2(v,u) * (180.0/Constants.PI) wdir = 270.0 - n.arctan2(v,u) * (180.0/Constants.PI)
return n.remainder(wdir, 360.0) return n.remainder(wdir, 360.0)
def _calc_wspd_wdir(u, v, units="mps"): def _calc_wspd_wdir(u, v, units):
return (_calc_wspd(u,v, units), _calc_wdir(u,v)) return (_calc_wspd(u,v, units), _calc_wdir(u,v))
@convert_units("wind", "mps") @convert_units("wind", "mps")
def get_u_destag(wrfnc, units="mps", timeidx=0): def get_u_destag(wrfnc, timeidx=0, units="mps"):
u = destagger_windcomp(wrfnc,"u", timeidx) u = destagger_windcomp(wrfnc,"u", timeidx)
return u return u
@convert_units("wind", "mps") @convert_units("wind", "mps")
def get_v_destag(wrfnc, units="mps", timeidx=0): def get_v_destag(wrfnc, timeidx=0, units="mps"):
v = destagger_windcomp(wrfnc,"v", timeidx) v = destagger_windcomp(wrfnc,"v", timeidx)
return v return v
@convert_units("wind", "mps") @convert_units("wind", "mps")
def get_w_destag(wrfnc, units="mps", timeidx=0): def get_w_destag(wrfnc, timeidx=0, units="mps"):
w = destagger_windcomp(wrfnc,"w", timeidx) w = destagger_windcomp(wrfnc,"w", timeidx)
return w return w
def get_destag_wspd_wdir(wrfnc, units="mps", timeidx=0): def get_destag_wspd_wdir(wrfnc, timeidx=0, units="mps"):
u = destagger_windcomp(wrfnc,"u", timeidx) u = destagger_windcomp(wrfnc,"u", timeidx)
v = destagger_windcomp(wrfnc,"v", timeidx) v = destagger_windcomp(wrfnc,"v", timeidx)

Write Preview
Loading…
Cancel
Save