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
772 B
31 lines
772 B
|
|
import numpy as n |
|
|
|
from wrf.var.util import extract_vars |
|
|
|
__all__ = ["get_accum_precip", "get_precip_diff"] |
|
|
|
def get_accum_precip(wrfnc, timeidx=0): |
|
vars = extract_vars(wrfnc, timeidx, vars=("RAINC", "RAINNC")) |
|
rainc = vars["RAINC"] |
|
rainnc = vars["RAINNC"] |
|
|
|
rainsum = rainc + rainnc |
|
|
|
return rainsum |
|
|
|
def get_precip_diff(wrfnc1, wrfnc2, timeidx=0): |
|
vars1 = extract_vars(wrfnc, timeidx, vars=("RAINC", "RAINNC")) |
|
vars2 = extract_vars(wrfnc, timeidx, vars=("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
|
|
|