rule_emergency.py

Go to the documentation of this file.

# -*- coding: utf-8 -*-
"""
Created on Fri May  8 05:59:04 2020
 
@author: cd
"""
 
 
import abc
import numpy as np
from efo.rule import RuleBase
from efo.lookup import LkupTblElev
from efo.time import TimeBase
 
 
class RuleEmgcBase(RuleBase):
    def __init__(self, name, time):
        super().__init__(name, time)
 
    @classmethod
    def __subclasshook__(cls,C):
        if cls is RuleEmgcBase:
            attrs = set(dir(C))
            if set(cls.__abstractmethods__) <= attrs:
                return True
        return NotImplemented
        
        
class RuleEmgc(RuleEmgcBase):
    def __init__(self, name, time, constants):
        # Call super class constructor
        super().__init__(name, time)
        # Class properties
        self.constantsconstants = constants
        self.rlsEmgcSchedrlsEmgcSched = []
 
    def set_emergency_table(self, name, hypso, elev, rlsEmgcVals):
        # TODO: Use error handling to check values passed
        self.rlsEmgcSchedrlsEmgcSched = LkupTblElev(name, hypso, elev, rlsEmgcVals)
 
    def calc_release(self, rlsProposed, *, stor, rlsUnCtrl, rlsPrev=None, qIn=None):
        rlsProposed = super().calc_release(
            rlsProposed, rlsPrev=rlsPrev, rlsUnCtrl=rlsUnCtrl, stor=stor, qIn=qIn)
        if stor > self.rlsEmgcSchedrlsEmgcSched.minIdx:
            rlsEmgc = self.rlsEmgcSchedrlsEmgcSched.get_val(stor)
            # Calculate storage after spill and emergency releases
            stor = stor - (rlsUnCtrl + rlsEmgc)*self.constantsconstants.cfs2af
            # If storage is below the emergency pool then adjust emergency release
            if rlsUnCtrl > 0 and rlsEmgc > 0 and stor < self.rlsEmgcSchedrlsEmgcSched.minIdx:
                rlsEmgc = rlsEmgc + (stor - self.rlsEmgcSchedrlsEmgcSched.minIdx)*self.constantsconstants.af2cfs
                if rlsEmgc < 0:
                    rlsEmgc = 0
        else:
            rlsEmgc = 0
        self.release[self.T.step] = rlsEmgc
        return max(rlsEmgc, rlsProposed)