#!/home/misha/venv/bin/python
import math as  m
import numpy as np
import matplotlib.pyplot as plt
import os
from pathlib import Path

os.chdir(Path(__file__).parent)
a=np.loadtxt("dim2.dat")
b=np.loadtxt("kiirendus.dat")


figure,(p1,p2,p3)=plt.subplots(1,3)

p1.plot(a[:,1],a[:,2])
p1.legend(["y(x)"])
p1.grid(color="black")

p2.plot(a[:,0],b[:,0],a[:,0],b[:,1],a[:,0],b[:,2],a[:,0],b[:,3])
p2.legend(["at","an","aabs","R"])
p2.grid(color="black")

p3.plot(a[:,0],b[:,4],a[:,0],b[:,5],a[:,0],b[:,6])
p3.legend(["Ekin","Epot","Etot"])
p3.grid(color="black")

plt.show()