hypso.py

Go to the documentation of this file.

# -*- coding: utf-8 -*-
"""
Created on Wed May  6 10:52:54 2020
 
@author: cd
"""
 
 
import abc
import numpy as np
from scipy.interpolate import interp1d
 
class HypsoBase(metaclass=abc.ABCMeta):
    def __init__(self, name):
        self.namename = name
        super().__init__()  
    @abc.abstractmethod
    def stor2elev(self, stor):
        pass
    @abc.abstractmethod
    def stor2area(self, stor):
        pass
    @classmethod
    def __subclasshook__(cls,C):
        if cls is HypsoBase:
            attrs = set(dir(C))
            if set(cls.__abstractmethods__) <= attrs:
                return True
        return NotImplemented
        
        
class Hypso(HypsoBase):
    def __init__(self, name, elev, stor, area):
        # Call super class constructor
        super().__init__(name)
        self.elevelev = elev
        self.storstor = stor
        self.areaarea = area
        self.interpFncElevinterpFncElev = interp1d(self.storstor, self.elevelev, kind='cubic', fill_value='extrapolate')
        self.interpFncAreainterpFncArea = interp1d(self.storstor, self.areaarea, kind='cubic', fill_value='extrapolate')
        self.interpFncStorinterpFncStor = interp1d(self.elevelev, self.storstor, kind='cubic', fill_value='extrapolate')
 
    def stor2elev(self, stor):
        # if isinstance(stor, np.ndarray):
        #     elevReturn = np.empty(stor.size)
        #     for i, curStor in enumerate(stor):
        #         elevReturn[i] = np.interp(curStor, self.stor, self.elev)
        #     return elevReturn
        # else:
        #     return np.interp(stor, self.stor, self.elev)
        return self.interpFncElevinterpFncElev(stor)
    
    def stor2area(self, stor):
        # if isinstance(stor, np.ndarray):
        #     areaReturn = np.empty(stor.size)
        #     for i, curStor in enumerate(stor):
        #         areaReturn[i] = np.interp(curStor, self.stor, self.area)
        #     return areaReturn
        # else:
        #     return np.interp(stor, self.stor, self.area)
        return self.interpFncAreainterpFncArea(stor)
    
    def elev2stor(self, elev):
        # if isinstance(elev, np.ndarray):
        #     storReturn = np.empty(elev.size)
        #     for i, curElev in enumerate(elev):
        #         storReturn[i] = np.interp(curElev, self.elev, self.stor)
        #     return storReturn
        # else:
        #     return np.interp(elev, self.elev, self.stor)
        return self.interpFncStorinterpFncStor(elev)