Homec4science
Diffusion Gear (master) Browse

Gear/PyGear/examples.tessel/test_memorybound.py
f616dc0c9bedmaster

Last Change
Raw File

test_memorybound.py
View Options

#!/usr/bin/env python
from pNbody import *
from pNbody import ic
from pNbody import libutil
from numpy import *
import sys
import time
import copy
from PyGear import gadget
def SaveFile(gadget,i):
nb.pos = gadget.GetAllPositions()
nb.vel = gadget.GetAllVelocities()
nb.mass = gadget.GetAllMasses()
nb.num = gadget.GetAllID()
nb.tpe = None
#nb.u = gadget.GetAllEnergySpec() # not defined
nb.rho = gadget.GetAllvDensities() # note the v
nb.rsp = gadget.GetAllvVolumes()
nb.u = gadget.GetAllvEnergySpec()
nb.init()
# save
nb.rename('snap/snap%05d.dat'%i)
nb.set_pio("yes")
nb.write()
def SaveEnergy(gadget,time,fdEnergy):
nb.pos = gadget.GetAllPositions()
nb.vel = gadget.GetAllVelocities()
nb.mass = gadget.GetAllMasses()
nb.num = gadget.GetAllID()
nb.tpe = None
#nb.u = gadget.GetAllEnergySpec() # not defined
nb.rho = gadget.GetAllvDensities() # note the v
nb.rsp = gadget.GetAllvVolumes()
nb.u = gadget.GetAllvEnergySpec()
nb.init()
Ekin = nb.Ekin()
Eint = sum(nb.u)
Epot = 0
fdEnergy.write( "%g %g %g %g\n"%(time,Eint,Epot,Ekin) )
nb = Nbody("snap.dat",ftype="gadget")
###################################
# init PyGear
gadget.InitMPI() # init MPI
gadget.InitDefaultParameters() # init default parameters
gadget.Info()
params = {}
params['ErrTolTheta'] = 0.7
params['DesNumNgb'] = 15
params['MaxNumNgbDeviation'] = 3
params['UnitLength_in_cm'] = 3.085e+21
params['UnitMass_in_g'] = 4.435693e+44
params['UnitVelocity_in_cm_per_s'] = 97824708.2699
params['BoxSize'] = 1.0
params['PartAllocFactor'] = 10
gadget.SetParameters(params)
###################################
# load particles
gadget.LoadParticles(array(nb.npart),nb.pos,nb.vel,nb.mass,nb.num,nb.tpe,nb.u)
###################################
# compute Delaunay
t1 = time.time()
gadget.ConstructDelaunay()
t2 = time.time()
print "Delaunay","in",(t2-t1),"s"
c4science · Help