import fpuCuda
from pylab import *
fpu=fpuCuda.FpuCuda(64)
fpu.init(1)
h=0.1
n1=2000
n2=1000
b=0.1
n=2
nm=10
fpu.run(h,n1,n2,b,n,nm)
            
figure(0)
for m in range(nm):
    plot(fpu.emodemat[:,m])
axis([0,n2*1.5,0,0.05])
legend(("mode 1","mode 2","mode 3","mode4","mode 5","mode 6","mode 7",),loc='upper right')
xlabel("t/(n1h)")
ylabel("Em")

            
figure(1)
for t in range(10):
    plot(fpu.qmat[100+t,:])
xlabel("k")
ylabel("qk")