gengif.py
No OneTemporary
Actions

Subscribers

None

File Metadata

	#!/usr/bin/python

	import sys
	import numpy as np
	import matplotlib.pyplot as plt
	from matplotlib.animation import FuncAnimation, PillowWriter

	def init():
	im.set_data(updategrid[0])
	return [im]

	def update(i):
	im.set_array(updategrid[i])
	return [im]

	argv = sys.argv


	#samples = [1]
	#samples = samples + list(range(500, 5500, 500))
	if len(argv) < 3:
	print("More args plz")
	sys.exit(1)

	imsize = int(argv[1])
	samples = argv[2].split()

	updategrid = []

	for n in samples:
	# load matrix
	grid = np.loadtxt(open("output-%d.csv" % (int(n)), "rb"), delimiter=";")
	updategrid.append(grid)

	# initialize plot
	fig, ax = plt.figure(), plt.axes(xlim=(0,imsize-1), ylim=(0,imsize-1))
	im = plt.imshow(updategrid[0], cmap=plt.get_cmap('hot'), interpolation='none')

	fps = 1.42

	anim = FuncAnimation(fig, update, frames = len(samples), interval = int(1000 / fps), init_func=init)

	anim.save('temperature.gif', writer='imagemagick', fps=fps)