mkimage.py
View Options

			from numpy import *
			from pNbody import ic
			from pNbody import Nbody

			import os


			def initnbody():
			nb = ic.box(1000,1,1,1)
			return nb

			def createnbodyfromvec(pos):
			nb = Nbody(pos=pos)
			return nb


			def dumpimage(nb,directory,bname,i):
			name = "%s%06d.png"%(bname,i)
			file = os.path.join(directory,name)
			nb.display(save=file,size=(2,2))
			print "write %s"%file




			class ImageMaker():

			def __init__(self,directory='png',bname='img'):
			self.directory=directory
			self.bname=bname
			self.counter = 0

			def dump(self,pos):
			nb = createnbodyfromvec(pos)
			dumpimage(nb,self.directory,self.bname,self.counter)
			self.counter=self.counter+1

			def info(self):
			print "info "
			print "---- "
			print "counter =",self.counter


			# to test
			#pos = random.random((1000,3))*10
			#print pos
			#nb=initnbodyfromvec(pos)
			#dumpimage(nb)

pNbody/examples/embedding/mkimage.pya069a81b93b8master