# -*- coding: utf-8 -*-
"""
Created on Thu Apr 03 17:06:26 2014

@author: TEuser

List all function versions

not all functions are still in this file, the older functions can be found
in h:\My Documents\memo's\python scripts\wflow\
"""

import numpy
import pdb
from copy import copy as copylist

try:
    from  wflow.wf_DynamicFramework import *
except ImportError:
    from  wf_DynamicFramework import *
import scipy

def groundWater_no_reservoir(self):
    """
    Recharge = Qu * D (per class)
    QsinClass = sum of recharge, percolation and capillary rise per class
    Qsin = QsinClass, summed over areal fraction and all classes

    """

    self.Recharge = [x*y for x,y in zip(self.Qu_, self.D)]    
    self.QsinClass = [a+b-c for a,b,c in zip(self.Recharge, self.Perc_, self.Cap_)]  # all fluxes are summed per class
    self.QsinTemp = sum([a*b for a,b in zip(self.QsinClass ,self.percent)])     #fluxes are summed per cell according to percentage of class
    self.Qsin = areatotal(self.QsinTemp * self.percentArea ,nominal(self.TopoId))  #areatotal is taken, according to area percentage of cell
 
    self.Qs = self.Qsin
    self.Ss = 0.
    self.wbSs = self.Qsin - self.Qs - self.Ss + self.Ss_t
    
def groundWaterCombined3(self):
    """
    - When using this option, the declaration of Ss and Qs should not be divided into separate classes
    - Stocks are lumped for the entire catchment
    - Differences with 'groundWaterCombined' are the order of calculations to try to decrease the numerical error --> turns out there is no difference
    - cannot be used combined with Sus for wetland areas
    """
    
    self.Recharge = [x*y for x,y in zip(self.Qu_, self.D)]    
    self.QsinClass = [a+b-c for a,b,c in zip(self.Recharge, self.Perc_, self.Cap_)]  # all fluxes are summed per class
    self.QsinTemp = sum([a*b for a,b in zip(self.QsinClass ,self.percent)])     #fluxes are summed per cell according to percentage of class
    self.Qsin = areatotal(self.QsinTemp * self.percentArea ,nominal(self.TopoId))  #areatotal is taken, according to area percentage of cell
 
    self.Qs = self.Ss * self.Ks    
    self.Ss = self.Ss * exp(-self.Ks) + self.Qsin
    
    self.wbSs = self.Qsin - self.Qs - self.Ss + self.Ss_t    
    
    self.Qs_ = self.Qs
    self.Recharge_ = self.Recharge
    self.QsinClass_ = self.QsinClass
    self.QsinTemp_ = self.QsinTemp
    self.Qsin_ = self.Qsin