import streamlit as st
import matplotlib.pyplot as plt
import numpy as np
import sys
  

import os
absolute_path = os.path.abspath(__file__)
# print("Full path: " + absolute_path)
# print("Directory Path: " + os.path.dirname(absolute_path))

# adding /backend to use its functions here
sys.path.append("/".join(os.path.dirname(absolute_path).split('/')[:-1]))
# print("/".join(os.path.dirname(absolute_path).split('/')[:-1]))
from backend.calc import *

def plot_data(r,i):
    # unit circle
    unit_circle_x = []
    unit_circle_y = []
    for x in np.arange(-1, 1, 0.01):
        unit_circle_x.append(x)
        unit_circle_y.append((1-x**2)**0.5)

    unit_circle_x.append(1)
    unit_circle_y.append(0)

    for x in np.arange(-1, 1, 0.01)[::-1]:
        unit_circle_x.append(x)
        unit_circle_y.append(-(1-x**2)**0.5)

    fig, ax = plt.subplots()
    ax.plot(unit_circle_x, unit_circle_y)
    #

    # data
    ax.plot(r, i)
    #

    ax.grid(True)
    ax.axis('square')
    ax.set_yticks(np.arange(-1, 1.2, 0.2))
    ax.set_yticks(np.arange(-1, 1.2, 0.2))
    st.pyplot(fig)


# ../../resource/data/1_M450.MEA
# with open("/".join(os.path.dirname(absolute_path).split('/')[:-2]) + "/resource/data/1_M450.MEA") as f:
#     row = f.readlines()
#     f, r, i = [], [], []
#     for x in row:
#         a, b, c = (float(y) for y in x.split())
#         f.append(a)  # frequency
#         r.append(b)  # Re of something
#         i.append(c)  # Im of something
#     plot_data(r,i)


data = []
uploaded_file = st.file_uploader('Upload a csv')
if uploaded_file is not None:
    data = uploaded_file.readlines()


col1, col2 = st.columns(2)

select_data_format = col1.selectbox('Choose data format from a list',['Frequency, Re(S11), Im(S11)','Frequency, Re(Zin), Im(Zin)'])

select_separator = col2.selectbox('Choose separator',['","' ,'" "','";"'])


def unpack_data(data):
    f, r, i = [], [], []  
    for x in data:
        a, b, c = (float(y) for y in x.split())
        f.append(a)  # frequency
        r.append(b)  # Re of something
        i.append(c)  # Im of something
    return f, r, i, 'very nice'

validator_status = 'nice'
# calculate
if len(data) > 0:
    f,r,i,validator_status = unpack_data(data)
    
    calc_data = prepare_data(f, r, i)
    a,c,d = solution(calc_data)
    sigmaQ0=0
    for i in range(2, 10):
        calc_data = recalculation_of_data(calc_data, a, c, d)
        a, c, d = solution(calc_data)
        Ql, diam, k, Q = q_factor(a)
        sigma2A = recalculation_of_data(calc_data, a, c, d, error=True)
        sigmaQ0 = random_deviation(a, sigma2A, diam, k, Ql)
    st.write(f"Q = {Q} +- {sigmaQ0}".format(Q, sigmaQ0))
        


st.write("Status: " +validator_status)

if len(data) > 0:
    f,r,i,validator_status = unpack_data(data)
    plot_data(r,i)