timedomainresonace/pmovie.py
J.A. de Jong @ vulgaris bc4a8941b7 Working code
2015-02-22 11:39:50 +01:00

56 lines
1.1 KiB
Python
Executable File

#!/usr/bin/python
from numpy import *
from matplotlib.pyplot import *
from matplotlib import animation
from run import *
import sys
inst=0
parr=load('pdata.npy')
uarr=load('udata.npy')
# save(pdat,'p.npy')
imax=parr.shape[0]
amplitude=1
fig = figure(figsize=(12,8))
ax = axes(xlim=(0,x[-1]),ylim=(-amplitude,amplitude))
grid('on')
line, = plot([],[],lw=2)
def init():
line.set_data([], [])
return line,
# print(imax)
def animatep(i):
# update_progress(int(round(100*i/imax)))
line.set_data(x,parr[i])
return line,
# animp = animation.FuncAnimation(fig, animatep, init_func=init,
# frames=imax, interval=20, blit=True,repeat=False)
def animateu(i):
# update_progress(int(round(100*i/imax)))
line.set_data(x,uarr[i])
return line,
print("Creating animation...")
animu = animation.FuncAnimation(fig, animateu, init_func=init,
frames=imax, interval=20, blit=True,repeat=False)
# print("Saving animation...")
# anim.save('p.mp4', fps=20, extra_args=['-vcodec', 'libx264'])
print("Showing animation...")
show()