outlet.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 LkupTblInterp
 
 
class OutletBase(metaclass=abc.ABCMeta):
    def __init__(self, name, time):
        self.namename = name
        self.TT = time
        self.rlsOutletrlsOutlet = np.empty(self.TT.nSteps)
        super().__init__()
    @abc.abstractmethod
    def calc_release(self, stor):
        pass
    @classmethod
    def __subclasshook__(cls, C):
        if cls is OutletBase:
            attrs = set(dir(C))
            if set(cls.__abstractmethods__) <= attrs:
                return True
        return NotImplemented
        
 
# TODO: This class should keep track of its releases this way they don't have
# be tracked in the reseroir junction        
class OutletRatedElev(OutletBase):
    def __init__(self, name, time, hypso, xElev, yQ):
        # Call super class constructor
        super().__init__(name, time)
        # Convert rating table to storage
        # xStor = np.interp(xElev, hypso.elev, hypso.stor)
        xStor = hypso.elev2stor(xElev)
        self.storVsQRatingTblstorVsQRatingTbl = LkupTblInterp(name, xStor, yQ)
    def calc_release(self, stor):
        self.rlsOutletrlsOutlet[self.TT.step] = self.storVsQRatingTblstorVsQRatingTbl.get_val(stor)
        return self.rlsOutletrlsOutlet[self.TT.step].copy()