# -*- coding: utf-8 -*-
"""
Created on Wed Apr 02 18:01:39 2014

@author: teuser

List all function versions
"""
import numpy
import pdb
from copy import copy as copylist

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


def selectSiR(i):
    """
    not all functions are still in this file, the older functions can be found
    (with the same numbering) in h:\My Documents\memo's\python scripts\wflow\
    """
    if i == 1:
        name = 'interception_overflow'
    if i == 2:
        name = 'interception_overflow2'
    if i == 3:
        name = 'interception_overflow_Ep'        
    return name

def interception_no_reservoir(self, k):
    """
    Effective rainfall = rainfall
    Interception evaporation = 0.
    Storage in interception = 0.
    """
    self.Pe_[k] = max(self.Precipitation, 0)
    self.Ei_[k] = 0.
    self.Si_[k] = 0.
    self.wbSi_[k] = self.Precipitation - self.Ei - self.Pe - self.Si[k] + self.Si_t[k]

def interception_overflow2(self, k):
    """
    - Effective rainfall is all that does not fit into the interception reservoir
    - Outgoing fluxes are determined separately
    - Code for ini-file: 2
    """
    
    self.Pe = max(self.Precipitation - (self.imax[k] - self.Si_t[k]),0)
    self.Si[k] = self.Si_t[k] + (self.Precipitation - self.Pe)
    self.Ei = ifthenelse(self.Sw[k] == 0, min(self.PotEvaporation, self.Si[k]), 0)           #ifstatement added on 3-11-2015 for snow module
    self.Si[k] = self.Si[k] - self.Ei    
    
    self.wbSi_[k] = self.Precipitation - self.Ei - self.Pe - self.Si[k] + self.Si_t[k]        
    
    self.Pe = self.Pe + self.Qw         #added on 3-11-2015 for snow module
    self.Ei = self.Ei + self.Ew
    
    self.Ei_[k]=self.Ei
    self.Pe_[k]=self.Pe
    self.PotEvaporation_ = self.PotEvaporation    
    
    if self.URFR_L:
         self.Ei = areatotal(self.Ei * self.percentArea, nominal(self.TopoId))
         self.Pe = areatotal(self.Pe * self.percentArea, nominal(self.TopoId))
         self.PotEvaporation = areatotal(self.PotEvaporation * self.percentArea, nominal(self.TopoId))
         self.Si[k] = areatotal(self.Si[k] * self.percentArea, nominal(self.TopoId))


def interception_overflow_Ep(self,k):
    """
    - Effective rainfall is all that does not fit into the interception reservoir
    - Outgoing fluxes are determined separately
    - this version cannot be run with Su averaged (for the current code)
    - Code for ini-file: 3
    """

    JarvisCoefficients.calcEp(self,k)
    self.PotEvaporation = cover(ifthenelse(self.EpHour >= 0, self.EpHour, 0),0)
    
    self.Pe = max(self.Precipitation - (self.imax[k] - self.Si_t[k]),0)
    self.Si[k] = self.Si_t[k] + (self.Precipitation - self.Pe)
    self.Ei = ifthenelse(self.Sw[k] == 0, min(self.PotEvaporation, self.Si[k]), 0)           #ifstatement added on 3-11-2015 for snow module
    self.Si[k] = self.Si[k] - self.Ei    
    
    self.wbSi_[k] = self.Precipitation - self.Ei - self.Pe - self.Si[k] + self.Si_t[k]        
    
    self.Pe = self.Pe + self.Qw         #added on 3-11-2015 for snow module
    self.Ei = self.Ei + self.Ew
    
    self.Ei_[k]=self.Ei
    self.Pe_[k]=self.Pe         
    self.Ep_[k]=self.EpHour    
    
    if self.URFR_L:
         self.Ei = areatotal(self.Ei * self.percentArea, nominal(self.TopoId))
         self.Pe = areatotal(self.Pe * self.percentArea, nominal(self.TopoId))
         self.PotEvaporation = areatotal(self.PotEvaporation * self.percentArea, nominal(self.TopoId))
         self.Si[k] = areatotal(self.Si[k] * self.percentArea, nominal(self.TopoId))