forked from 3rdparty/wrf-python
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
2.0 KiB
58 lines
2.0 KiB
from __future__ import (absolute_import, division, print_function, |
|
unicode_literals) |
|
|
|
from .util import extract_vars |
|
from .latlonutils import (_lat_varname, _lon_varname, _ll_to_xy, _xy_to_ll) |
|
from .metadecorators import set_latlon_metadata |
|
|
|
|
|
def get_lat(wrfnc, timeidx=0, method="cat", squeeze=True, |
|
cache=None, meta=True, |
|
stagger=None): |
|
|
|
varname = _lat_varname(wrfnc, stagger) |
|
lat_var = extract_vars(wrfnc, timeidx, varname, method, squeeze, cache, |
|
meta) |
|
|
|
return lat_var[varname] |
|
|
|
|
|
def get_lon(wrfnc, timeidx=0, method="cat", squeeze=True, |
|
cache=None, meta=True, |
|
stagger=None): |
|
|
|
varname = _lon_varname(wrfnc, stagger) |
|
lon_var = extract_vars(wrfnc, timeidx, varname, method, squeeze, cache, |
|
meta) |
|
|
|
return lon_var[varname] |
|
|
|
|
|
# Can either use wrfnc as a single file or sequence, or provide |
|
# projection parameters (which don't allow for moving domains) |
|
@set_latlon_metadata(xy=True) |
|
def ll_to_xy(wrfnc, latitude, longitude, timeidx=0, stagger=None, method="cat", |
|
squeeze=True, cache=None, meta=True): |
|
return _ll_to_xy(latitude, longitude, wrfnc, timeidx, stagger, method, |
|
squeeze, cache, **{}) |
|
|
|
|
|
@set_latlon_metadata(xy=True) |
|
def ll_to_xy_proj(latitude, longitude, meta=True, squeeze=True, **projparams): |
|
return _ll_to_xy(latitude, longitude, None, 0, squeeze, "cat", True, None, |
|
**projparams) |
|
|
|
|
|
@set_latlon_metadata(xy=False) |
|
def xy_to_ll(wrfnc, x, y, timeidx=0, stagger=None, method="cat", squeeze=True, |
|
cache=None, meta=True): |
|
return _xy_to_ll(x, y, wrfnc, timeidx, stagger, method, squeeze, cache, |
|
**{}) |
|
|
|
|
|
@set_latlon_metadata(xy=False) |
|
def xy_to_ll_proj(x, y, meta=True, squeeze=True, **projparams): |
|
return _xy_to_ll(x, y, None, 0, None, "cat", squeeze, None, **projparams) |
|
|
|
|
|
|