import numpy
import pyfits
import matplotlib
import matplotlib.image as mpimg
import os

abspath = os.path.abspath(__file__)
dname = os.path.dirname(abspath)
os.chdir(dname)


prefix = 'frame_'

nmax = 45
n = 0

x = 400
y = 362
z = nmax

image = numpy.zeros((z,y,x))

for i in range(n,nmax):
	print i
	img = mpimg.imread(prefix+str(i).zfill(3)+'.png')
	
	#for j in range(0,x):
	#	for k in range(0,y):
	#		image[i,j,k] = img[k,j,0]
						# Greyscale	#RGB data
	image[i,:,:] = img[:,:] 	#img[:,:,0]	# 	

fitsfile = prefix+'.fits'

pyfits.core.writeto(fitsfile,image, clobber=True)