evap.py

Go to the documentation of this file.

# -*- coding: utf-8 -*-
"""
Created on Thu May  7 06:13:47 2020
 
@author: cd
"""
 
import abc
import numpy as np
from efo.lookup import LkupTblAnn
from efo.time import TimeCont
 
class EvapBase(metaclass=abc.ABCMeta):
    def __init__(self, name, time):
        self.namename = name
        self.TT = time
        self.evapevap = np.empty(self.TT.nSteps)
        super().__init__()  
    @abc.abstractmethod
    def calc_evap(self, stor):
        pass
    @classmethod
    def __subclasshook__(cls,C):
        if cls is EvapBase:
            attrs = set(dir(C))
            if set(cls.__abstractmethods__) <= attrs:
                return True
        return NotImplemented
        
 
class EvapLkupTbl(EvapBase):
    def __init__(self, name, time, hypso, monDayHr, lkupVals, *,
                 typ='step', timeUnit=None):
        # Call super class constructor
        super().__init__(name, time)
        # Create lookup table
        self.evapRateTblevapRateTbl = LkupTblAnn(name+'RateTbl', time, monDayHr, lkupVals,
                                      typ=typ, timeUnit=timeUnit)
        self.hypsohypso = hypso
    def calc_evap(self, stor):
         curEvapRate = self.evapRateTblevapRateTbl.get_val(self.TT.curDT)
         area = self.hypsohypso.stor2area(stor)
         self.evapevap[self.TT.step] = curEvapRate/12 * area * self.TT.nHrs/24   
         return self.evapevap[self.TT.step]