test03.py
View Options


			from pNbody import *
			import time
			import treelib

			import Ptools as pt

			ErrTolTheta = 0.8
			eps = 0.1

			params = {}
			params['ErrTolTheta'] = ErrTolTheta

			# open a model
			nb = Nbody("../../examples/snap.dat",ftype='gadget')
			#nb.set_tpe(0)
			#nb = nb.selectc(nb.rxyz()<2)

			# create a tree object
			nb.Tree = treelib.Tree(npart=array(nb.npart),pos=nb.pos,vel=nb.vel,mass=nb.mass,num=nb.num,tpe=nb.tpe,params=params)
			# make the tree
			nb.Tree.BuildTree()






			R = arange(-20,20,0.1)
			print len(R)
			pos = concatenate((R,zeros(len(R)),zeros(len(R))))
			pos.shape = (3,len(R))
			pos = transpose(pos)
			pos = pos.astype(float32)


			hsml = nb.Tree.InitHsml(33,3)
			density,hsml = nb.Tree.Density(pos,hsml,33,3)

			pt.plot(R,density)
			pt.show()

pNbody/src/new.treelib/test03.py6c0821250383master