# -*- coding: utf-8 -*-
"""
Created on Wed Feb 04 14:52:30 2015

@author: teuser
"""

# -*- coding: utf-8 -*-
"""
Created on Thu Apr 03 16:31:35 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 scipy
import JarvisCoefficients


def snow_no_reservoir(self, k):
    """
    This function is used when no snow zone reservoir is used and only
    passes fluxes from the upper reservoirs to the lower
    Qw = Psnow
    Ew = 0.
    Storage in snow zone = 0.

    !!!still needs a final check!!!    
    
    """
    self.Qw_[k] = max(self.PrecipitationSnow, 0)
    self.Ew_[k] = 0.
    self.Sw[k] = 0.
    self.wbSw_[k] = self.Precipitation - self.Ew_[k] - self.Qw_[k] - self.Sw[k] + self.Sw_t[k]  
      
def snow(self,k):
    """
    - snow melt based on degree day factor and 
    """
    JarvisCoefficients.calcEpSnow(self,k)
    self.PotEvaporation = cover(ifthenelse(self.EpHour > 0, self.EpHour, 0),0)
    
    self.Sw[k] = self.Sw_t[k] + self.PrecipitationSnow
    
    self.Ew1 = max(min(self.PotEvaporation,self.Sw[k]),0)
    self.Qw1 = max(self.Fm[k] * (self.Temperature - self.Tm[k]), 0)
    
    self.Sw[k] = self.Sw_t[k] + self.PrecipitationSnow - self.Ew1 - self.Qw1

    self.Sw_diff = ifthenelse(self.Sw[k] < 0, self.Sw[k], 0)
    self.Ew = self.Ew1 + (self.Ew1/ifthenelse(self.Ew1 + self.Qw1 > 0 , self.Ew1 + self.Qw1 , 1)) * self.Sw_diff
    self.Qw = self.Qw1 + (self.Qw1/ifthenelse(self.Ew1 + self.Qw1 > 0 , self.Ew1 + self.Qw1 , 1)) * self.Sw_diff
    self.Sw[k] = self.Sw_t[k] + self.PrecipitationSnow - self.Ew - self.Qw
    self.Sw[k] = ifthenelse(self.Sw[k] < 0, 0 , self.Sw[k])    
    self.Sw_diff2 = ifthen(self.Sw[k] < 0, self.Sw[k]) 

    self.wbSw_[k] = self.PrecipitationSnow - self.Ew - self.Qw - self.Sw[k] + self.Sw_t[k]
    
    self.Ew_[k] = self.Ew
    self.Qw_[k] = self.Qw
    
def snow_rain(self,k):
    """
    - snow melt based on degree day factor and minimum surface temperature
    - meltfactor increases with temperature
    """
   
    JarvisCoefficients.calcEpSnow(self,k)
    self.PotEvaporation = cover(ifthenelse(self.EpHour > 0, self.EpHour, 0),0)
    
    self.Sw[k] = self.Sw_t[k] + self.PrecipitationSnow
    
    self.Fm2 = max(self.Fm[k] * self.Precipitation, self.Fm[k])
    self.Ew1 = max(min(self.PotEvaporation, self.Sw[k]),0)
    self.Qw1 = max(min(self.Fm2 * (self.Temperature - self.Tm[k]), self.Sw[k]), 0)
    
    self.Sw[k] = self.Sw_t[k] + self.PrecipitationSnow - self.Ew1 - self.Qw1

    self.Sw_diff = ifthenelse(self.Sw[k] < 0, self.Sw[k], 0)
    self.Ew = self.Ew1 + (self.Ew1/ifthenelse(self.Ew1 + self.Qw1 > 0 , self.Ew1 + self.Qw1 , 1)) * self.Sw_diff
    self.Qw = self.Qw1 + (self.Qw1/ifthenelse(self.Ew1 + self.Qw1 > 0 , self.Ew1 + self.Qw1 , 1)) * self.Sw_diff
    self.Sw[k] = self.Sw_t[k] + self.PrecipitationSnow - self.Ew - self.Qw
    self.Sw[k] = ifthenelse(self.Sw[k] < 0, 0 , self.Sw[k])    
    self.Sw_diff2 = ifthen(self.Sw[k] < 0, self.Sw[k]) 

    self.wbSw_[k] = self.PrecipitationSnow - self.Ew - self.Qw - self.Sw[k] + self.Sw_t[k]
    
    self.Ew_[k] = self.Ew
    self.Qw_[k] = self.Qw

def snow_rain_hourlyEp(self,k):
    """
    - snow melt based on degree day factor and minimum surface temperature
    - meltfactor increases with temperature
    """
   
    JarvisCoefficients.calcEpSnowHour(self,k)
    self.PotEvaporation = cover(ifthenelse(self.EpHour > 0, self.EpHour, 0),0)
    
    self.Sw[k] = self.Sw_t[k] + self.PrecipitationSnow
    
    self.Fm2 = max(self.Fm[k] * self.Precipitation, self.Fm[k])
    self.Ew1 = max(min(self.PotEvaporation, self.Sw[k]),0)
    self.Qw1 = max(min(self.Fm2 * (self.Temperature - self.Tm[k]), self.Sw[k]), 0)
    
    self.Sw[k] = self.Sw_t[k] + self.PrecipitationSnow - self.Ew1 - self.Qw1

    self.Sw_diff = ifthenelse(self.Sw[k] < 0, self.Sw[k], 0)
    self.Ew = self.Ew1 + (self.Ew1/ifthenelse(self.Ew1 + self.Qw1 > 0 , self.Ew1 + self.Qw1 , 1)) * self.Sw_diff
    self.Qw = self.Qw1 + (self.Qw1/ifthenelse(self.Ew1 + self.Qw1 > 0 , self.Ew1 + self.Qw1 , 1)) * self.Sw_diff
    self.Sw[k] = self.Sw_t[k] + self.PrecipitationSnow - self.Ew - self.Qw
    self.Sw[k] = ifthenelse(self.Sw[k] < 0, 0 , self.Sw[k])    
    self.Sw_diff2 = ifthen(self.Sw[k] < 0, self.Sw[k]) 

    self.wbSw_[k] = self.PrecipitationSnow - self.Ew - self.Qw - self.Sw[k] + self.Sw_t[k]
    
    self.Ew_[k] = self.Ew
    self.Qw_[k] = self.Qw

def snow_rain_Tsurf(self,k):
    """
    - snow melt based on degree day factor and minimum surface temperature
    - meltfactor increases with temperature
    """
   
    JarvisCoefficients.calcEpSnow(self,k)
    self.PotEvaporation = cover(ifthenelse(self.EpHour > 0, self.EpHour, 0),0)
    
    self.Sw[k] = self.Sw_t[k] + self.PrecipitationSnow
    
    self.Fm2 = max(self.Fm[k] * self.Precipitation, self.Fm[k])
    self.Ew1 = max(min(self.PotEvaporation, self.Sw[k]),0)
    self.Qw1 = max(min(self.Fm2 * (self.TempSurf - self.Tm[k]), self.Sw[k]), 0)
    
    self.Sw[k] = self.Sw_t[k] + self.PrecipitationSnow - self.Ew1 - self.Qw1

    self.Sw_diff = ifthenelse(self.Sw[k] < 0, self.Sw[k], 0)
    self.Ew = self.Ew1 + (self.Ew1/ifthenelse(self.Ew1 + self.Qw1 > 0 , self.Ew1 + self.Qw1 , 1)) * self.Sw_diff
    self.Qw = self.Qw1 + (self.Qw1/ifthenelse(self.Ew1 + self.Qw1 > 0 , self.Ew1 + self.Qw1 , 1)) * self.Sw_diff
    self.Sw[k] = self.Sw_t[k] + self.PrecipitationSnow - self.Ew - self.Qw
    self.Sw[k] = ifthenelse(self.Sw[k] < 0, 0 , self.Sw[k])    
    self.Sw_diff2 = ifthen(self.Sw[k] < 0, self.Sw[k]) 

    self.wbSw_[k] = self.PrecipitationSnow - self.Ew - self.Qw - self.Sw[k] + self.Sw_t[k]
    
    self.Ew_[k] = self.Ew
    self.Qw_[k] = self.Qw

def snow_rain_TsurfAir(self,k):
    """
    - snow melt based on degree day factor and minimum surface temperature
    - meltfactor increases with temperature
    """
    JarvisCoefficients.calcEpSnow(self,k)
    self.PotEvaporation = cover(ifthenelse(self.EpHour > 0, self.EpHour, 0),0)
    
    self.Sw[k] = self.Sw_t[k] + self.PrecipitationSnow
    self.Temp = (self.TempSurf + self.Temperature) / 2
    
    self.Fm2 = max(self.Fm[k] * self.Precipitation, self.Fm[k])
    self.Ew1 = max(min(self.PotEvaporation, self.Sw[k]),0)
    self.Qw1 = max(min(self.Fm2 * (self.Temp - self.Tm[k]), self.Sw[k]), 0)
    
    self.Sw[k] = self.Sw_t[k] + self.PrecipitationSnow - self.Ew1 - self.Qw1

    self.Sw_diff = ifthenelse(self.Sw[k] < 0, self.Sw[k], 0)
    self.Ew = self.Ew1 + (self.Ew1/ifthenelse(self.Ew1 + self.Qw1 > 0 , self.Ew1 + self.Qw1 , 1)) * self.Sw_diff
    self.Qw = self.Qw1 + (self.Qw1/ifthenelse(self.Ew1 + self.Qw1 > 0 , self.Ew1 + self.Qw1 , 1)) * self.Sw_diff
    self.Sw[k] = self.Sw_t[k] + self.PrecipitationSnow - self.Ew - self.Qw
    self.Sw[k] = ifthenelse(self.Sw[k] < 0, 0 , self.Sw[k])    
    self.Sw_diff2 = ifthen(self.Sw[k] < 0, self.Sw[k]) 

    self.wbSw_[k] = self.PrecipitationSnow - self.Ew - self.Qw - self.Sw[k] + self.Sw_t[k]
    
    self.Ew_[k] = self.Ew
    self.Qw_[k] = self.Qw