import numpy as np
from core_functions import *


masses = [5.972e24, 419725, 7.34767309e22]

u_0 = np.array([[0,0,0,0,0,0], [414000 + 6378100, 0, 0, 0, 7662, 0], [363300e3, 0, 0, 0, 1081, 0]])
u_0 = np.array([[0,0,0,0,0,0], [4214929.7, 0, 5326067, 0, 7662, 0], [361836242.8, 0, 32579493.6, 0, 1081, 0]])

T = 2360600

delta_t = 10

u, times = ivp_RK4(u_0, T, delta_t, masses)

rxsat = []
rysat = []
rzsat = []

for i in range(len(times)):
    rxsat.append(u[i][1][0])
    rysat.append(u[i][1][1])
    rzsat.append(u[i][1][2])

rxmoon = []
rymoon = []
rzmoon = []

for i in range(len(times)):
    rxmoon.append(u[i][2][0])
    rymoon.append(u[i][2][1])
    rzmoon.append(u[i][2][2])

import matplotlib.pyplot as plt

plt.rcParams['figure.figsize'] = [20, 5]
plt.plot(rxmoon, rymoon, label = 'Moon')
plt.plot(rxsat, rysat, label = 'Satellite')

plt.legend()
plt.show()