#!/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("kiirendus.dat")
b=np.loadtxt("dim2.dat")
scalep=30
widthp=0.002
n=len(b[:,0])
t=np.arange(0,n,100)
plt.quiver(b[t,1],b[t,2],b[t,7],b[t,8],angles="xy",color="black",scale_units="xy",scale=scalep,width=widthp,label="at")
plt.quiver(b[t,1],b[t,2],b[t,9],b[t,10],angles="xy",color="green",scale_units="xy",scale=scalep,width=widthp,label="aa")
plt.quiver(b[t,1],b[t,2],b[t,11],b[t,12],angles="xy",color="red",scale_units="xy",scale=scalep,width=widthp,label="atot")
plt.quiver(b[t,1],b[t,2],b[t,5],b[t,6],angles="xy",color="blue",scale_units="xy",scale=scalep,width=widthp,label="ar")
plt.plot(b[:,1],b[:,2],color="gray",label="traj")
plt.legend()
plt.grid()
plt.show()