compute_water_budget function

Compute the water budget