#!/usr/bin/python
import numpy as np
import matplotlib.pyplot as plt

res=np.loadtxt("harm.dat")
#******** visual ************************************
fig=plt.figure(figsize=(10,10))
p1=plt.subplot(221)
p1.set_xlabel("t")
p1.set_ylabel("x")
p1.plot(res[:,0],res[:,1],color="red",label="x")
p1.grid(color="black")
p1.legend(loc='upper right')

p2=plt.subplot(222)
p2.set_xlabel("t")
p2.set_ylabel("v")
p2.plot(res[:,0],res[:,2],color="black",label="kiirus")
p2.grid(color="black")
p2.legend(loc='upper right')

p3=plt.subplot(223)
p3.set_xlabel("t")
p3.set_ylabel("enegy")
p3.plot(res[:,0],res[:,3],color="blue",label="ekin")
p3.plot(res[:,0],res[:,4],color="red",label="epot")
p3.plot(res[:,0],res[:,5],color="green",label="etot")
p3.plot(res[:,0],res[:,6],color="black",label="ekorr")
p3.grid(color="black")
p3.legend(loc='upper right')

plt.show()