test_acc.py
View Options

			from numarray import *
			import SM
			import treelib

			from pNbody import *

			import time



			eps = 0.1

			# open a model
			nb = Nbody('disk.dat',ftype='gadget')
			#nb = nb.selectc(nb.rxyz()<2)

			# create a tree object
			MyTree = nb.getTree()





			R = arange(0,200,1.0)
			print len(R)
			pos = concatenate((R,zeros(len(R)),zeros(len(R))))
			pos.shape = (3,len(R))
			pos = transpose(pos)
			pos = pos.astype(Float32)



			acc = MyTree.Acceleration(pos,eps)
			acc = acc.astype(Float32)



			R = R.astype(Float32)

			g = SM.plot()
			g.limits(R,acc[:,0])
			g.box()
			g.connect(R,acc[:,0])
			g.show()

pNbody/src/treelib/test_acc.pyde93a8c7a339master