from __future__ import (absolute_import, division, print_function)

from .util import extract_vars


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