import numpy as np
import matplotlib.pyplot as plt

a=np.loadtxt("kepler.dat")

fig1,(p1,p2)=plt.subplots(1,2)

p1.set_xlim(-50000,10000)
p1.set_ylim(-30000,30000)
p1.set_xlabel("x")
p1.set_ylabel("y")
p1.legend(loc='upper right')
p1.set_aspect("equal")
maa=plt.Circle((0,0),6378,color="green")
p1.add_artist(maa)
maa.set(label='Maa')

t=range(0,len(a[:,0]),10000)
p1.plot(a[:,1],a[:,2],color="red",label="traj")
p1.quiver(a[t,1],a[t,2],a[t,3],a[t,4],angles="xy",color="black",scale_units="xy",scale=0.001,width=0.003,label="v")
p1.quiver(a[t,1],a[t,2],a[t,5],a[t,6],angles="xy",color="blue",scale_units="xy",scale=0.000001,width=0.003,label="a")
p1.legend()
p1.grid(color="black")

p2.grid(color="black")
p2.set_xlabel("time")
p2.set_ylabel("energy")
p2.legend(loc='upper right')
p2.plot(a[:,0],a[:,7],color="blue",label="Ekin")
p2.plot(a[:,0],a[:,8],color="red",label="Epot")
p2.plot(a[:,0],a[:,9],color="black",label="Etot")
p2.legend()

plt.show()
#fig1.savefig("out.jpg")