Stg_Sto_Ar.py

# -*- coding: utf-8 -*-
"""
Created on Wed Oct 21 16:37:03 2020

@author: osamatarabih
"""
    
# To return storage or surface area given stage (i=0) else (i = any number) will return stage given storage or surface area.
class Stg_Sto_Ar:
    import sys,os
    from Model_Config import Model_Config 
    Working_Path = Model_Config.Working_Path
    os.chdir('%s'%Working_Path) 
    def stg2sto(v, i):
        import pandas as pd
        from scipy import interpolate
        stgsto_data = pd.read_csv('./Data/StgSto_data.csv')
        #NOTE: We Can use cubic interpolation instead of linear 
        x = stgsto_data['Stage']
        y = stgsto_data['Storage']
        if i == 0:
        #return storage given stage
            return interpolate.interp1d(x,y, fill_value='extrapolate', kind = 'linear')(v)
        else:
        #return stage given storage
            return interpolate.interp1d(y,x, fill_value='extrapolate', kind = 'linear')(v)
    #Calculate the Stage_Area relationship for interpolation!
    def stg2ar(v, i):
        import pandas as pd
        from scipy import interpolate
        stgar_data = pd.read_csv('./Data/Stgar_data.csv')
        #NOTE: We Can use cubic interpolation instead of linear 
        x = stgar_data['Stage']
        y = stgar_data['Surf_Area']
        if i == 0:
            #return surface area given stage
            return interpolate.interp1d(x,y, fill_value='extrapolate', kind = 'linear')(v)
        else:
            #return stage given surface area
            return interpolate.interp1d(y,x, fill_value='extrapolate', kind = 'linear')(v)
    #Calculate the Stage_MarshArea relationship for interpolation!
    def stg2mar(v, i):
        import pandas as pd
        from scipy import interpolate
        stgmar_data = pd.read_csv('./Data/Stgmar_data.csv')
        #NOTE: We Can use cubic interpolation instead of linear 
        x = stgmar_data['Stage']
        y = stgmar_data['Marsh_Ar']
        if i == 0:
            #return marsh area given stage
            return interpolate.interp1d(x,y, fill_value='extrapolate', kind = 'linear')(v)
        else:
            #return stage given marsh area
            return interpolate.interp1d(y,x, fill_value='extrapolate', kind = 'linear')(v)