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.
31 lines
876 B
31 lines
876 B
from __future__ import (absolute_import, division, print_function, |
|
unicode_literals) |
|
|
|
from .util import extract_vars |
|
|
|
__all__ = ["get_accum_precip", "get_precip_diff"] |
|
|
|
def get_accum_precip(wrfin, timeidx=0): |
|
ncvars = extract_vars(wrfin, timeidx, varnames=("RAINC", "RAINNC")) |
|
rainc = ncvars["RAINC"] |
|
rainnc = ncvars["RAINNC"] |
|
|
|
rainsum = rainc + rainnc |
|
|
|
return rainsum |
|
|
|
def get_precip_diff(wrfin1, wrfin2, timeidx=0): |
|
vars1 = extract_vars(wrfin1, timeidx, varnames=("RAINC", "RAINNC")) |
|
vars2 = extract_vars(wrfin2, timeidx, varnames=("RAINC", "RAINNC")) |
|
rainc1 = vars1["RAINC"] |
|
rainnc1 = vars1["RAINNC"] |
|
|
|
rainc2 = vars2["RAINC"] |
|
rainnc2 = vars2["RAINNC"] |
|
|
|
rainsum1 = rainc1 + rainnc1 |
|
rainsum2 = rainc2 + rainnc2 |
|
|
|
return (rainsum1 - rainsum2) |
|
|
|
# TODO: Handle bucket flipping
|
|
|