test03.py
No OneTemporary
Actions

Subscribers

None

File Metadata


	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()