(varname, wrfnc, timeidx, method, squeeze)
| 58 | |
| 59 | |
| 60 | def get_args(varname, wrfnc, timeidx, method, squeeze): |
| 61 | if varname == "avo": |
| 62 | varnames = ("U", "V", "MAPFAC_U", "MAPFAC_V", "MAPFAC_M", "F") |
| 63 | ncvars = extract_vars(wrfnc, timeidx, varnames, method, squeeze, |
| 64 | cache=None, meta=True) |
| 65 | |
| 66 | attrs = extract_global_attrs(wrfnc, attrs=("DX", "DY")) |
| 67 | u = ncvars["U"] |
| 68 | v = ncvars["V"] |
| 69 | msfu = ncvars["MAPFAC_U"] |
| 70 | msfv = ncvars["MAPFAC_V"] |
| 71 | msfm = ncvars["MAPFAC_M"] |
| 72 | cor = ncvars["F"] |
| 73 | |
| 74 | dx = attrs["DX"] |
| 75 | dy = attrs["DY"] |
| 76 | |
| 77 | return (u, v, msfu, msfv, msfm, cor, dx, dy) |
| 78 | |
| 79 | if varname == "pvo": |
| 80 | varnames = ("U", "V", "T", "P", "PB", "MAPFAC_U", "MAPFAC_V", |
| 81 | "MAPFAC_M", "F") |
| 82 | ncvars = extract_vars(wrfnc, timeidx, |
| 83 | varnames, |
| 84 | method, squeeze, cache=None, meta=True) |
| 85 | attrs = extract_global_attrs(wrfnc, attrs=("DX", "DY")) |
| 86 | |
| 87 | u = ncvars["U"] |
| 88 | v = ncvars["V"] |
| 89 | t = ncvars["T"] |
| 90 | p = ncvars["P"] |
| 91 | pb = ncvars["PB"] |
| 92 | msfu = ncvars["MAPFAC_U"] |
| 93 | msfv = ncvars["MAPFAC_V"] |
| 94 | msfm = ncvars["MAPFAC_M"] |
| 95 | cor = ncvars["F"] |
| 96 | |
| 97 | dx = attrs["DX"] |
| 98 | dy = attrs["DY"] |
| 99 | |
| 100 | full_t = t + 300 |
| 101 | full_p = p + pb |
| 102 | |
| 103 | return (u, v, full_t, full_p, msfu, msfv, msfm, cor, dx, dy) |
| 104 | |
| 105 | if varname == "eth": |
| 106 | varnames = ("T", "P", "PB", "QVAPOR") |
| 107 | ncvars = extract_vars(wrfnc, timeidx, varnames, method, squeeze, |
| 108 | cache=None, meta=True) |
| 109 | t = ncvars["T"] |
| 110 | p = ncvars["P"] |
| 111 | pb = ncvars["PB"] |
| 112 | qv = ncvars["QVAPOR"] |
| 113 | |
| 114 | full_t = t + Constants.T_BASE |
| 115 | full_p = p + pb |
| 116 | tkel = tk(full_p, full_t, meta=False) |
| 117 |
no test coverage detected