#!/usr/bin/env python
#GUI application

from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
import PySimpleGUI as sg
import pandas as pd
# from Q_test import run_qtest
import matplotlib.pyplot as plt
import ctypes

lib = ctypes.CDLL("/media/syatov430/Files/UNN/3_course/ЧМ/lab1/lab1/Q_test.so")

# In[]:
df = pd.DataFrame({
    '    Xi    ': [],
    '    Vi    ': [],
    '    V2i    ': [],
    '    ОЛП    ': [],
    '    Hi    ': [],
    '    C1    ': [],
    '    C2    ': [],
})

fig, graf = plt.subplots(figsize=(5, 5))
graf = plt.plot([1, 2, 3, 4], [1, 2, 3, 4])

list_q = ['Тестовая', 'Основная1', 'Основная2']

frame_start_condition = [
    [sg.Radio('1', default=True, group_id=1, key="-U1-"), sg.Radio('-1', group_id=1, key="-U-1-")],
]

frame_output_data = [
    [sg.Output(size=(55, 15), key="-DATA-")],
]

# In[]:
column1 = [
    [sg.DropDown(list_q, default_value=list_q[0], size=(25, 1))],
    [sg.Frame("Начальные Условия", frame_start_condition, size=(180, 50))],
    [sg.Submit(size=(24, 1))],
    [sg.Exit(size=(24, 1))]
]

column2 = [
    [sg.Text("Точность выхода на границу", size=(50, 1))],
    [sg.InputText(default_text="0.0000001", size=(30, 1), key='-EPSOUT-')],
    [sg.Text("Контроль погрешности")],
    [sg.InputText(default_text="0.0001", size=(30, 1), key='-EPS-')],
    [sg.Text("Число шагов")],
    [sg.InputText(default_text="1000", size=(30, 1), key='-NMAX-')],
    [sg.Text("Начальный шаг")],
    [sg.InputText(default_text="0.1", size=(30, 1), key='-HSTART-')],
    [sg.Text("Максимальный X")],
    [sg.InputText(default_text="20", size=(30, 1), key='-XMAX-')],
]

column_table = [
    [sg.Table(values=df.values.tolist(), headings=df.columns.tolist(),
            alternating_row_color='darkblue', key='-TABLE-',
            row_height = 25, vertical_scroll_only=False, size=(200, 100),
            justification='left', auto_size_columns=True)],
]

column_graf = [
    [sg.Canvas(key="-CANVAS-")],
]

# In[]:
layout = [
    [sg.Column(column1), sg.VerticalSeparator(), sg.Column(column2), sg.Frame("Данные", frame_output_data, element_justification='right')],
    [sg.HorizontalSeparator()],
    [sg.Frame("Таблица точек", column_table, size=(500, 500)), sg.VerticalSeparator(), sg.Column(column_graf, size=(500, 500), justification='right')],
]

# In[]:
window = sg.Window('LAB1', layout, finalize=True, resizable=True, grab_anywhere=True, )

canvas_elem = window["-CANVAS-"]
canvas = FigureCanvasTkAgg(fig, master=canvas_elem.Widget)
canvas.get_tk_widget().pack(fill='both', expand=1)

while True:                             # The Event Loop
    event, values = window.read()
    window.FindElement('-DATA-').Update('')
    print(event, values) #debug
    if event in (None, 'Exit', 'Cancel'):
        break
    if event == 'Submit':
        u0 = 1. if window["-U1-"].Get() == True else -1.
        # df = run_qtest(float(window["-HSTART-"].Get()), u0, 1000, 0.001, 0.0001)
        lib.rungeKuttaAdaptive(float(window["-HSTART-"].Get()), u0, 1000, 0.001, 0.0001)
        window["-TABLE-"].Update(values = df.values.tolist())

window.close()