import Equation
from Simulator import Simulator
from DimensionalityFinder import DimensionalityFinder
import matplotlib.pyplot as plt
import math
import numpy as np
from scipy.integrate import odeint
from scipy.integrate import solve_ivp
eq = Equation.Fitz()
sim = Simulator(eq)
# data = sim.states(duration=100, split = 0.01) # 40 -> 1.369, 400 -> 1.497
state0 = [0.01, 0.01, 0.01, 0.01]
t = np.linspace(0, 10000, 50000)
# sim = Simulator(eq)
#
# data = sim.states(duration = 300011, split = 0.25)
# data = sim.interpolateCurve()[10000:]
# print(data.shape)
n = 2
a = [-0.025794, -0.025794]
b = [0.0065, 0.0135]
c = [0.02, 0.02]
k = 0.128
A = [[0, 1], [1, 0]]
def f(state, t):
x = state[0:n]
y = state[n:2 * n] # unpack the state vector
