forked from 3rdparty/wrf-python
Bill Ladwig
9 years ago
37 changed files with 1287 additions and 1200 deletions
@ -0,0 +1,32 @@ |
|||||||
|
SUBROUTINE testfunc(a, b, c, nx, ny, nz, errstat, errmsg) |
||||||
|
USE wrf_constants, ONLY : ALGERR |
||||||
|
IMPLICIT NONE |
||||||
|
|
||||||
|
!threadsafe |
||||||
|
!f2py intent(in,out) :: b |
||||||
|
|
||||||
|
INTEGER, INTENT(IN) :: nx, ny, nz |
||||||
|
|
||||||
|
REAL(KIND=8), DIMENSION(nx,ny,nz), INTENT(IN) :: a |
||||||
|
REAL(KIND=8), DIMENSION(nx,ny,nz), INTENT(OUT) :: b |
||||||
|
CHARACTER(LEN=*), INTENT(IN) :: c |
||||||
|
INTEGER, INTENT(INOUT) :: errstat |
||||||
|
REAL(KIND=8), PARAMETER :: blah=123.45 |
||||||
|
CHARACTER(LEN=*), INTENT(INOUT) :: errmsg |
||||||
|
|
||||||
|
INTEGER :: i,j,k |
||||||
|
|
||||||
|
DO k=1,nz |
||||||
|
DO j=1,ny |
||||||
|
DO i=1,nx |
||||||
|
b(i,j,k) = a(i,j,k) |
||||||
|
END DO |
||||||
|
END DO |
||||||
|
END DO |
||||||
|
|
||||||
|
errstat = ALGERR |
||||||
|
WRITE(errmsg, *) c(1:20), blah |
||||||
|
|
||||||
|
RETURN |
||||||
|
|
||||||
|
END SUBROUTINE testfunc |
||||||
@ -0,0 +1,580 @@ |
|||||||
|
from math import fabs, log, tan, sin, cos |
||||||
|
|
||||||
|
import unittest as ut |
||||||
|
import numpy.testing as nt |
||||||
|
import numpy as np |
||||||
|
import numpy.ma as ma |
||||||
|
import os, sys |
||||||
|
import subprocess |
||||||
|
|
||||||
|
from netCDF4 import Dataset as nc |
||||||
|
|
||||||
|
from wrf import * |
||||||
|
|
||||||
|
NCL_EXE = "/Users/ladwig/nclbuild/6.3.0/bin/ncl" |
||||||
|
TEST_FILE = "/Users/ladwig/Documents/wrf_files/wrfout_d01_2010-06-13_21:00:00" |
||||||
|
OUT_NC_FILE = "/tmp/wrftest.nc" |
||||||
|
NCFILE = nc(TEST_FILE) |
||||||
|
NCGROUP = [NCFILE, NCFILE, NCFILE] |
||||||
|
|
||||||
|
# Python 3 |
||||||
|
if sys.version_info > (3,): |
||||||
|
xrange = range |
||||||
|
|
||||||
|
|
||||||
|
ROUTINE_MAP = {"avo" : avo, |
||||||
|
"eth" : eth, |
||||||
|
"cape_2d" : cape_2d, |
||||||
|
"cape_3d" : cape_3d, |
||||||
|
"ctt" : ctt, |
||||||
|
"dbz" : dbz, |
||||||
|
"helicity" : srhel, |
||||||
|
"omg" : omega, |
||||||
|
"pvo" : pvo, |
||||||
|
"pw" : pw, |
||||||
|
"rh" : rh, |
||||||
|
"slp" : slp, |
||||||
|
"td" : td, |
||||||
|
"tk" : tk, |
||||||
|
"tv" : tvirtual, |
||||||
|
"twb" : wetbulb, |
||||||
|
"updraft_helicity" : udhel, |
||||||
|
"uvmet" : uvmet, |
||||||
|
"cloudfrac" : cloudfrac} |
||||||
|
|
||||||
|
class ProjectionError(RuntimeError): |
||||||
|
pass |
||||||
|
|
||||||
|
def get_args(varname, wrfnc, timeidx, method, squeeze): |
||||||
|
if varname == "avo": |
||||||
|
ncvars = extract_vars(wrfnc, timeidx, ("U", "V", "MAPFAC_U", |
||||||
|
"MAPFAC_V", "MAPFAC_M", |
||||||
|
"F"), |
||||||
|
method, squeeze, cache=None, meta=True) |
||||||
|
|
||||||
|
attrs = extract_global_attrs(wrfnc, attrs=("DX", "DY")) |
||||||
|
u = ncvars["U"] |
||||||
|
v = ncvars["V"] |
||||||
|
msfu = ncvars["MAPFAC_U"] |
||||||
|
msfv = ncvars["MAPFAC_V"] |
||||||
|
msfm = ncvars["MAPFAC_M"] |
||||||
|
cor = ncvars["F"] |
||||||
|
|
||||||
|
dx = attrs["DX"] |
||||||
|
dy = attrs["DY"] |
||||||
|
|
||||||
|
return (u, v, msfu, msfv, msfm, cor, dx, dy) |
||||||
|
|
||||||
|
if varname == "pvo": |
||||||
|
ncvars = extract_vars(wrfnc, timeidx, ("U", "V", "T", "P", |
||||||
|
"PB", "MAPFAC_U", |
||||||
|
"MAPFAC_V", "MAPFAC_M", |
||||||
|
"F"), |
||||||
|
method, squeeze, cache=None, meta=True) |
||||||
|
attrs = extract_global_attrs(wrfnc, attrs=("DX", "DY")) |
||||||
|
|
||||||
|
u = ncvars["U"] |
||||||
|
v = ncvars["V"] |
||||||
|
t = ncvars["T"] |
||||||
|
p = ncvars["P"] |
||||||
|
pb = ncvars["PB"] |
||||||
|
msfu = ncvars["MAPFAC_U"] |
||||||
|
msfv = ncvars["MAPFAC_V"] |
||||||
|
msfm = ncvars["MAPFAC_M"] |
||||||
|
cor = ncvars["F"] |
||||||
|
|
||||||
|
dx = attrs["DX"] |
||||||
|
dy = attrs["DY"] |
||||||
|
|
||||||
|
full_t = t + 300 |
||||||
|
full_p = p + pb |
||||||
|
|
||||||
|
return (u, v, full_t, full_p, msfu, msfv, msfm, cor, dx, dy) |
||||||
|
|
||||||
|
if varname == "eth": |
||||||
|
varnames=("T", "P", "PB", "QVAPOR") |
||||||
|
ncvars = extract_vars(wrfnc, timeidx, varnames, method, squeeze, |
||||||
|
cache=None, meta=True) |
||||||
|
t = ncvars["T"] |
||||||
|
p = ncvars["P"] |
||||||
|
pb = ncvars["PB"] |
||||||
|
qv = ncvars["QVAPOR"] |
||||||
|
|
||||||
|
full_t = t + Constants.T_BASE |
||||||
|
full_p = p + pb |
||||||
|
tkel = tk(full_p, full_t, meta=False) |
||||||
|
|
||||||
|
return (qv, tkel, full_p) |
||||||
|
|
||||||
|
if varname == "cape_2d": |
||||||
|
varnames = ("T", "P", "PB", "QVAPOR", "PH","PHB", "HGT", "PSFC") |
||||||
|
ncvars = extract_vars(wrfnc, timeidx, varnames, method, squeeze, |
||||||
|
cache=None, meta=True) |
||||||
|
|
||||||
|
t = ncvars["T"] |
||||||
|
p = ncvars["P"] |
||||||
|
pb = ncvars["PB"] |
||||||
|
qv = ncvars["QVAPOR"] |
||||||
|
ph = ncvars["PH"] |
||||||
|
phb = ncvars["PHB"] |
||||||
|
ter = ncvars["HGT"] |
||||||
|
psfc = ncvars["PSFC"] |
||||||
|
|
||||||
|
full_t = t + Constants.T_BASE |
||||||
|
full_p = p + pb |
||||||
|
tkel = tk(full_p, full_t, meta=False) |
||||||
|
|
||||||
|
geopt = ph + phb |
||||||
|
geopt_unstag = destagger(geopt, -3) |
||||||
|
z = geopt_unstag/Constants.G |
||||||
|
|
||||||
|
# Convert pressure to hPa |
||||||
|
p_hpa = ConversionFactors.PA_TO_HPA * full_p |
||||||
|
psfc_hpa = ConversionFactors.PA_TO_HPA * psfc |
||||||
|
|
||||||
|
i3dflag = 0 |
||||||
|
ter_follow = 1 |
||||||
|
|
||||||
|
return (p_hpa, tkel, qv, z, ter, psfc_hpa, ter_follow) |
||||||
|
|
||||||
|
if varname == "cape_3d": |
||||||
|
varnames = ("T", "P", "PB", "QVAPOR", "PH", "PHB", "HGT", "PSFC") |
||||||
|
ncvars = extract_vars(wrfnc, timeidx, varnames, method, squeeze, |
||||||
|
cache=None, meta=True) |
||||||
|
t = ncvars["T"] |
||||||
|
p = ncvars["P"] |
||||||
|
pb = ncvars["PB"] |
||||||
|
qv = ncvars["QVAPOR"] |
||||||
|
ph = ncvars["PH"] |
||||||
|
phb = ncvars["PHB"] |
||||||
|
ter = ncvars["HGT"] |
||||||
|
psfc = ncvars["PSFC"] |
||||||
|
|
||||||
|
full_t = t + Constants.T_BASE |
||||||
|
full_p = p + pb |
||||||
|
tkel = tk(full_p, full_t, meta=False) |
||||||
|
|
||||||
|
geopt = ph + phb |
||||||
|
geopt_unstag = destagger(geopt, -3) |
||||||
|
z = geopt_unstag/Constants.G |
||||||
|
|
||||||
|
# Convert pressure to hPa |
||||||
|
p_hpa = ConversionFactors.PA_TO_HPA * full_p |
||||||
|
psfc_hpa = ConversionFactors.PA_TO_HPA * psfc |
||||||
|
|
||||||
|
i3dflag = 1 |
||||||
|
ter_follow = 1 |
||||||
|
|
||||||
|
return (p_hpa, tkel, qv, z, ter, psfc_hpa, ter_follow) |
||||||
|
|
||||||
|
if varname == "ctt": |
||||||
|
varnames = ("T", "P", "PB", "PH", "PHB", "HGT", "QVAPOR") |
||||||
|
ncvars = extract_vars(wrfnc, timeidx, varnames, method, squeeze, |
||||||
|
cache=None, meta=True) |
||||||
|
t = ncvars["T"] |
||||||
|
p = ncvars["P"] |
||||||
|
pb = ncvars["PB"] |
||||||
|
ph = ncvars["PH"] |
||||||
|
phb = ncvars["PHB"] |
||||||
|
ter = ncvars["HGT"] |
||||||
|
qv = ncvars["QVAPOR"] * 1000.0 # g/kg |
||||||
|
|
||||||
|
haveqci = 1 |
||||||
|
try: |
||||||
|
icevars = extract_vars(wrfnc, timeidx, "QICE", |
||||||
|
method, squeeze, cache=None, meta=False) |
||||||
|
except KeyError: |
||||||
|
qice = np.zeros(qv.shape, qv.dtype) |
||||||
|
haveqci = 0 |
||||||
|
else: |
||||||
|
qice = icevars["QICE"] * 1000.0 #g/kg |
||||||
|
|
||||||
|
try: |
||||||
|
cldvars = extract_vars(wrfnc, timeidx, "QCLOUD", |
||||||
|
method, squeeze, cache=None, meta=False) |
||||||
|
except KeyError: |
||||||
|
raise RuntimeError("'QCLOUD' not found in NetCDF file") |
||||||
|
else: |
||||||
|
qcld = cldvars["QCLOUD"] * 1000.0 #g/kg |
||||||
|
|
||||||
|
full_p = p + pb |
||||||
|
p_hpa = full_p * ConversionFactors.PA_TO_HPA |
||||||
|
full_t = t + Constants.T_BASE |
||||||
|
tkel = tk(full_p, full_t, meta=False) |
||||||
|
|
||||||
|
geopt = ph + phb |
||||||
|
geopt_unstag = destagger(geopt, -3) |
||||||
|
ght = geopt_unstag / Constants.G |
||||||
|
|
||||||
|
return (p_hpa, tkel, qv, qcld, ght, ter, qice) |
||||||
|
|
||||||
|
if varname == "dbz": |
||||||
|
varnames = ("T", "P", "PB", "QVAPOR", "QRAIN") |
||||||
|
ncvars = extract_vars(wrfnc, timeidx, varnames, method, squeeze, |
||||||
|
cache=None, meta=True) |
||||||
|
t = ncvars["T"] |
||||||
|
p = ncvars["P"] |
||||||
|
pb = ncvars["PB"] |
||||||
|
qv = ncvars["QVAPOR"] |
||||||
|
qr = ncvars["QRAIN"] |
||||||
|
|
||||||
|
try: |
||||||
|
snowvars = extract_vars(wrfnc, timeidx, "QSNOW", |
||||||
|
method, squeeze, cache=None, meta=False) |
||||||
|
except KeyError: |
||||||
|
qs = np.zeros(qv.shape, qv.dtype) |
||||||
|
else: |
||||||
|
qs = snowvars["QSNOW"] |
||||||
|
|
||||||
|
try: |
||||||
|
graupvars = extract_vars(wrfnc, timeidx, "QGRAUP", |
||||||
|
method, squeeze, cache=None, meta=False) |
||||||
|
except KeyError: |
||||||
|
qg = np.zeros(qv.shape, qv.dtype) |
||||||
|
else: |
||||||
|
qg = graupvars["QGRAUP"] |
||||||
|
|
||||||
|
full_t = t + Constants.T_BASE |
||||||
|
full_p = p + pb |
||||||
|
tkel = tk(full_p, full_t, meta=False) |
||||||
|
|
||||||
|
return (full_p, tkel, qv, qr, qs, qg) |
||||||
|
|
||||||
|
if varname == "helicity": |
||||||
|
# Top can either be 3000 or 1000 (for 0-1 srh or 0-3 srh) |
||||||
|
|
||||||
|
ncvars = extract_vars(wrfnc, timeidx, ("HGT", "PH", "PHB"), |
||||||
|
method, squeeze, cache=None, meta=True) |
||||||
|
|
||||||
|
ter = ncvars["HGT"] |
||||||
|
ph = ncvars["PH"] |
||||||
|
phb = ncvars["PHB"] |
||||||
|
|
||||||
|
# As coded in NCL, but not sure this is possible |
||||||
|
varname = "U" |
||||||
|
u_vars = extract_vars(wrfnc, timeidx, varname, method, squeeze, |
||||||
|
cache=None, meta=False) |
||||||
|
u = destagger(u_vars[varname], -1) |
||||||
|
|
||||||
|
varname = "V" |
||||||
|
v_vars = extract_vars(wrfnc, timeidx, varname, method, squeeze, |
||||||
|
cache=None, meta=False) |
||||||
|
v = destagger(v_vars[varname], -2) |
||||||
|
|
||||||
|
geopt = ph + phb |
||||||
|
geopt_unstag = destagger(geopt, -3) |
||||||
|
|
||||||
|
z = geopt_unstag / Constants.G |
||||||
|
|
||||||
|
return (u, v, z, ter) |
||||||
|
|
||||||
|
if varname == "updraft_helicity": |
||||||
|
ncvars = extract_vars(wrfnc, timeidx, ("W", "PH", "PHB", "MAPFAC_M"), |
||||||
|
method, squeeze, cache=None, meta=True) |
||||||
|
|
||||||
|
wstag = ncvars["W"] |
||||||
|
ph = ncvars["PH"] |
||||||
|
phb = ncvars["PHB"] |
||||||
|
mapfct = ncvars["MAPFAC_M"] |
||||||
|
|
||||||
|
attrs = extract_global_attrs(wrfnc, attrs=("DX", "DY")) |
||||||
|
dx = attrs["DX"] |
||||||
|
dy = attrs["DY"] |
||||||
|
|
||||||
|
# As coded in NCL, but not sure this is possible |
||||||
|
varname = "U" |
||||||
|
u_vars = extract_vars(wrfnc, timeidx, varname, method, squeeze, |
||||||
|
cache=None, meta=True) |
||||||
|
u = destagger(u_vars[varname], -1, meta=True) |
||||||
|
|
||||||
|
varname = "V" |
||||||
|
v_vars = extract_vars(wrfnc, timeidx, varname, method, squeeze, |
||||||
|
cache=None, meta=True) |
||||||
|
v = destagger(v_vars[varname], -2, meta=True) |
||||||
|
|
||||||
|
zstag = ph + phb |
||||||
|
|
||||||
|
return (zstag, mapfct, u, v, wstag, dx, dy) |
||||||
|
|
||||||
|
if varname == "omg": |
||||||
|
varnames=("T", "P", "W", "PB", "QVAPOR") |
||||||
|
ncvars = extract_vars(wrfnc, timeidx, varnames, method, squeeze, |
||||||
|
cache=None, meta=True) |
||||||
|
t = ncvars["T"] |
||||||
|
p = ncvars["P"] |
||||||
|
w = ncvars["W"] |
||||||
|
pb = ncvars["PB"] |
||||||
|
qv = ncvars["QVAPOR"] |
||||||
|
|
||||||
|
wa = destagger(w, -3) |
||||||
|
full_t = t + Constants.T_BASE |
||||||
|
full_p = p + pb |
||||||
|
tkel = tk(full_p, full_t, meta=False) |
||||||
|
|
||||||
|
return (qv, tkel, wa, full_p) |
||||||
|
|
||||||
|
if varname == "pw": |
||||||
|
varnames=("T", "P", "PB", "PH", "PHB", "QVAPOR") |
||||||
|
ncvars = extract_vars(wrfnc, timeidx, varnames, method, squeeze, |
||||||
|
cache=None, meta=True) |
||||||
|
|
||||||
|
t = ncvars["T"] |
||||||
|
p = ncvars["P"] |
||||||
|
pb = ncvars["PB"] |
||||||
|
ph = ncvars["PH"] |
||||||
|
phb = ncvars["PHB"] |
||||||
|
qv = ncvars["QVAPOR"] |
||||||
|
|
||||||
|
# Change this to use real virtual temperature! |
||||||
|
full_p = p + pb |
||||||
|
ht = (ph + phb)/Constants.G |
||||||
|
full_t = t + Constants.T_BASE |
||||||
|
|
||||||
|
tkel = tk(full_p, full_t, meta=False) |
||||||
|
|
||||||
|
return (full_p, tkel, qv, ht) |
||||||
|
|
||||||
|
if varname == "rh": |
||||||
|
varnames=("T", "P", "PB", "QVAPOR") |
||||||
|
ncvars = extract_vars(wrfnc, timeidx, varnames, method, squeeze, |
||||||
|
cache=None, meta=True) |
||||||
|
t = ncvars["T"] |
||||||
|
p = ncvars["P"] |
||||||
|
pb = ncvars["PB"] |
||||||
|
qvapor = npvalues(ncvars["QVAPOR"]) |
||||||
|
|
||||||
|
full_t = t + Constants.T_BASE |
||||||
|
full_p = p + pb |
||||||
|
qvapor[qvapor < 0] = 0 |
||||||
|
tkel = tk(full_p, full_t, meta=False) |
||||||
|
|
||||||
|
return (qvapor, full_p, tkel) |
||||||
|
|
||||||
|
if varname == "slp": |
||||||
|
varnames=("T", "P", "PB", "QVAPOR", "PH", "PHB") |
||||||
|
ncvars = extract_vars(wrfnc, timeidx, varnames, method, squeeze, |
||||||
|
cache=None, meta=True) |
||||||
|
|
||||||
|
t = ncvars["T"] |
||||||
|
p = ncvars["P"] |
||||||
|
pb = ncvars["PB"] |
||||||
|
qvapor = npvalues(ncvars["QVAPOR"]) |
||||||
|
ph = ncvars["PH"] |
||||||
|
phb = ncvars["PHB"] |
||||||
|
|
||||||
|
full_t = t + Constants.T_BASE |
||||||
|
full_p = p + pb |
||||||
|
qvapor[qvapor < 0] = 0. |
||||||
|
|
||||||
|
full_ph = (ph + phb) / Constants.G |
||||||
|
|
||||||
|
destag_ph = destagger(full_ph, -3) |
||||||
|
|
||||||
|
tkel = tk(full_p, full_t, meta=False) |
||||||
|
|
||||||
|
return (destag_ph, tkel, full_p, qvapor) |
||||||
|
|
||||||
|
if varname == "td": |
||||||
|
varnames=("P", "PB", "QVAPOR") |
||||||
|
ncvars = extract_vars(wrfnc, timeidx, varnames, method, squeeze, |
||||||
|
cache=None, meta=True) |
||||||
|
|
||||||
|
p = ncvars["P"] |
||||||
|
pb = ncvars["PB"] |
||||||
|
qvapor = npvalues(ncvars["QVAPOR"]) |
||||||
|
|
||||||
|
# Algorithm requires hPa |
||||||
|
full_p = .01*(p + pb) |
||||||
|
qvapor[qvapor < 0] = 0 |
||||||
|
|
||||||
|
return (full_p, qvapor) |
||||||
|
|
||||||
|
if varname == "tk": |
||||||
|
varnames=("T", "P", "PB") |
||||||
|
ncvars = extract_vars(wrfnc, timeidx, varnames, method, squeeze, |
||||||
|
cache=None, meta=True) |
||||||
|
t = ncvars["T"] |
||||||
|
p = ncvars["P"] |
||||||
|
pb = ncvars["PB"] |
||||||
|
|
||||||
|
full_t = t + Constants.T_BASE |
||||||
|
full_p = p + pb |
||||||
|
|
||||||
|
return (full_p, full_t) |
||||||
|
|
||||||
|
if varname == "tv": |
||||||
|
varnames=("T", "P", "PB", "QVAPOR") |
||||||
|
ncvars = extract_vars(wrfnc, timeidx, varnames, method, squeeze, |
||||||
|
cache=None, meta=True) |
||||||
|
|
||||||
|
t = ncvars["T"] |
||||||
|
p = ncvars["P"] |
||||||
|
pb = ncvars["PB"] |
||||||
|
qv = ncvars["QVAPOR"] |
||||||
|
|
||||||
|
full_t = t + Constants.T_BASE |
||||||
|
full_p = p + pb |
||||||
|
tkel = tk(full_p, full_t) |
||||||
|
|
||||||
|
return (tkel, qv) |
||||||
|
|
||||||
|
if varname == "twb": |
||||||
|
varnames=("T", "P", "PB", "QVAPOR") |
||||||
|
ncvars = extract_vars(wrfnc, timeidx, varnames, method, squeeze, |
||||||
|
cache=None, meta=True) |
||||||
|
t = ncvars["T"] |
||||||
|
p = ncvars["P"] |
||||||
|
pb = ncvars["PB"] |
||||||
|
qv = ncvars["QVAPOR"] |
||||||
|
|
||||||
|
full_t = t + Constants.T_BASE |
||||||
|
full_p = p + pb |
||||||
|
|
||||||
|
tkel = tk(full_p, full_t) |
||||||
|
|
||||||
|
return (full_p, tkel, qv) |
||||||
|
|
||||||
|
if varname == "uvmet": |
||||||
|
varname = "U" |
||||||
|
u_vars = extract_vars(wrfnc, timeidx, varname, method, squeeze, |
||||||
|
cache=None, meta=True) |
||||||
|
|
||||||
|
u = destagger(u_vars[varname], -1, meta=True) |
||||||
|
|
||||||
|
varname = "V" |
||||||
|
v_vars = extract_vars(wrfnc, timeidx, varname, method, squeeze, |
||||||
|
cache=None, meta=True) |
||||||
|
v = destagger(v_vars[varname], -2, meta=True) |
||||||
|
|
||||||
|
map_proj_attrs = extract_global_attrs(wrfnc, attrs="MAP_PROJ") |
||||||
|
map_proj = map_proj_attrs["MAP_PROJ"] |
||||||
|
|
||||||
|
if map_proj in (0,3,6): |
||||||
|
raise ProjectionError("Map projection does not need rotation") |
||||||
|
elif map_proj in (1,2): |
||||||
|
lat_attrs = extract_global_attrs(wrfnc, attrs=("TRUELAT1", |
||||||
|
"TRUELAT2")) |
||||||
|
radians_per_degree = Constants.PI/180.0 |
||||||
|
# Rotation needed for Lambert and Polar Stereographic |
||||||
|
true_lat1 = lat_attrs["TRUELAT1"] |
||||||
|
true_lat2 = lat_attrs["TRUELAT2"] |
||||||
|
|
||||||
|
try: |
||||||
|
lon_attrs = extract_global_attrs(wrfnc, attrs="STAND_LON") |
||||||
|
except AttributeError: |
||||||
|
try: |
||||||
|
cen_lon_attrs = extract_global_attrs(wrfnc, attrs="CEN_LON") |
||||||
|
except AttributeError: |
||||||
|
raise RuntimeError("longitude attributes not found in NetCDF") |
||||||
|
else: |
||||||
|
cen_lon = cen_lon_attrs["CEN_LON"] |
||||||
|
else: |
||||||
|
cen_lon = lon_attrs["STAND_LON"] |
||||||
|
|
||||||
|
|
||||||
|
varname = "XLAT" |
||||||
|
xlat_var = extract_vars(wrfnc, timeidx, varname, |
||||||
|
method, squeeze, cache=None, meta=True) |
||||||
|
lat = xlat_var[varname] |
||||||
|
|
||||||
|
varname = "XLONG" |
||||||
|
xlon_var = extract_vars(wrfnc, timeidx, varname, |
||||||
|
method, squeeze, cache=None, meta=True) |
||||||
|
lon = xlon_var[varname] |
||||||
|
|
||||||
|
if map_proj == 1: |
||||||
|
if((fabs(true_lat1 - true_lat2) > 0.1) and |
||||||
|
(fabs(true_lat2 - 90.) > 0.1)): |
||||||
|
cone = (log(cos(true_lat1*radians_per_degree)) |
||||||
|
- log(cos(true_lat2*radians_per_degree))) |
||||||
|
cone = (cone / |
||||||
|
(log(tan((45.-fabs(true_lat1/2.))*radians_per_degree)) |
||||||
|
- log(tan((45.-fabs(true_lat2/2.))*radians_per_degree)))) |
||||||
|
else: |
||||||
|
cone = sin(fabs(true_lat1)*radians_per_degree) |
||||||
|
else: |
||||||
|
cone = 1 |
||||||
|
|
||||||
|
return (u, v, lat, lon, cen_lon, cone) |
||||||
|
|
||||||
|
if varname == "cloudfrac": |
||||||
|
vars = extract_vars(wrfnc, timeidx, ("P", "PB", "QVAPOR", "T"), |
||||||
|
method, squeeze, cache=None, meta=True) |
||||||
|
|
||||||
|
p = vars["P"] |
||||||
|
pb = vars["PB"] |
||||||
|
qv = vars["QVAPOR"] |
||||||
|
t = vars["T"] |
||||||
|
|
||||||
|
full_p = p + pb |
||||||
|
full_t = t + Constants.T_BASE |
||||||
|
|
||||||
|
tkel = tk(full_p, full_t) |
||||||
|
relh = rh(qv, full_p, tkel) |
||||||
|
|
||||||
|
return (full_p, relh) |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
class WRFVarsTest(ut.TestCase): |
||||||
|
longMessage = True |
||||||
|
|
||||||
|
def make_func(varname, wrfnc, timeidx, method, squeeze, meta): |
||||||
|
def func(self): |
||||||
|
|
||||||
|
try: |
||||||
|
args = get_args(varname, wrfnc, timeidx, method, squeeze) |
||||||
|
except ProjectionError: # Don't fail for this |
||||||
|
return |
||||||
|
|
||||||
|
routine = ROUTINE_MAP[varname] |
||||||
|
|
||||||
|
kwargs = {"meta" : meta} |
||||||
|
result = routine(*args, **kwargs) |
||||||
|
|
||||||
|
ref = getvar(wrfnc, varname, timeidx, method, squeeze, cache=None, |
||||||
|
meta=meta) |
||||||
|
|
||||||
|
nt.assert_allclose(npvalues(result), npvalues(ref)) |
||||||
|
|
||||||
|
if meta: |
||||||
|
self.assertEqual(result.dims, ref.dims) |
||||||
|
|
||||||
|
return func |
||||||
|
|
||||||
|
if __name__ == "__main__": |
||||||
|
varnames = ["avo", "eth", "cape_2d", "cape_3d", "ctt", "dbz", "mdbz", |
||||||
|
"geopt", "helicity", "lat", "lon", "omg", "p", "pressure", |
||||||
|
"pvo", "pw", "rh2", "rh", "slp", "ter", "td2", "td", "tc", |
||||||
|
"theta", "tk", "tv", "twb", "updraft_helicity", "ua", "va", |
||||||
|
"wa", "uvmet10", "uvmet", "z", "cloudfrac"] |
||||||
|
|
||||||
|
#varnames = ["helicity"] |
||||||
|
varnames=["avo", "pvo", "eth", "dbz", "helicity", "updraft_helicity", |
||||||
|
"omg", "pw", "rh", "slp", "td", "tk", "tv", "twb", "uvmet", |
||||||
|
"cloudfrac"] |
||||||
|
|
||||||
|
# Turn this one off when not needed, since it's slow |
||||||
|
#varnames += ["cape_2d", "cape_3d"] |
||||||
|
|
||||||
|
for varname in varnames: |
||||||
|
for i,wrfnc in enumerate((NCFILE, NCGROUP)): |
||||||
|
for j,timeidx in enumerate((0, ALL_TIMES)): |
||||||
|
for method in ("cat", "join"): |
||||||
|
for squeeze in (True, False): |
||||||
|
for meta in (True, False): |
||||||
|
func = make_func(varname, wrfnc, timeidx, method, |
||||||
|
squeeze, meta) |
||||||
|
ncname = "single" if i == 0 else "multi" |
||||||
|
timename = "t0" if j == 0 else "all" |
||||||
|
squeeze_name = "squeeze" if squeeze else "nosqueeze" |
||||||
|
meta_name = "meta" if meta else "nometa" |
||||||
|
test_name = "test_{}_{}_{}_{}_{}_{}".format(varname, |
||||||
|
ncname, timename, method, |
||||||
|
squeeze_name, meta_name) |
||||||
|
|
||||||
|
setattr(WRFVarsTest, test_name, func) |
||||||
|
|
||||||
|
ut.main() |
||||||
|
|
||||||
|
|
||||||
Loading…
Reference in new issue