def test_with_cube():
z_cnt = 8
spacing = 0.1
rho0 = 1000.
visc = 0.01
kappa = 0.000
dt = 0.04
g = 0.
pos = create_pos_cube(z_cnt, spacing=spacing)
mass_part = spacing*spacing*spacing*rho0
maxneighs = 200
avgneighs = 100
h = pow(3.*avgneighs*mass_part/(4.*PI*rho0), 1./3.)
solver = IISPHSolver()
solver.particles.pos.value = pos
solver.particles.rho0.value = np.ones((len(pos),1))*rho0
solver.particles.mass.value = np.ones((len(pos),1))*mass_part
solver.particles.info.value = np.ones((len(pos),1))*solver.consts['MAT_FLUID']
solver.particles.visc.value = np.ones((len(pos),1))*visc
solver.particles.kappa.value = np.ones((len(pos),1))*kappa
solver.h = h
solver.dt = dt
solver.g = g
solver.maxneighs = maxneighs
solver.avgneighs = avgneighs
solver.wkrn = 'WendlandC2'
solver.setup()
display = GL_Window()
display.register_points_display(solver.update, solver.particles.pos, solver.particles.rho)
display.start()
def test_with_dambreak():
z_cnt = 20
spacing = 0.1
rho0 = 1000.
visc = 10.
kappa = 0.001
dt = 0.002
g = 9.81
slipcoeff = 1.
avgdvar = 0.001
maxdvar = 0.01
dt_min = 0.01
auto_dt_coeff = 0.03
pos, info = create_dam_break(z_cnt, lfact=1., wallthick=2, walldensfact=1, spacing=spacing, fill=0.4)
mass_part = spacing*spacing*spacing*rho0
maxneighs = 60
avgneighs = 40
h = pow(3.*avgneighs*mass_part/(4.*PI*rho0), 1./3.)
solver = IISPHSolver()
solver.particles.pos.value = pos
solver.particles.rho0.value = np.ones((len(pos),1))*rho0
solver.particles.mass.value = np.ones((len(pos),1))*mass_part
solver.particles.info.value = info
solver.particles.visc.value = np.ones((len(pos),1))*visc
solver.particles.kappa.value = np.ones((len(pos),1))*kappa
solver.h = h
solver.dt = dt
solver.g = g
solver.slipcoeff = slipcoeff
solver.maxneighs = maxneighs
solver.avgneighs = avgneighs
solver.maxdvar = maxdvar
solver.avgdvar = avgdvar
solver.dt_min = dt_min
solver.auto_dt_coeff = auto_dt_coeff
solver.wkrn = 'WendlandC2'
solver.setup()
def update_and_count_neighs():
count = 10
for i in range(count):
solver.update()
numcl.analyze.dump()
neighs = solver.neighsearch.neighbors.value
np.place(neighs, neighs == -1, [0])
nnz = np.zeros(len(solver.neighsearch.neighbors))
for i in range(len(nnz)):
nnz[i] = np.count_nonzero(neighs[i])
print 'neigh_count ', np.amax(nnz)
display = GL_Window()
display.register_points_display(update_and_count_neighs, solver.particles.pos, solver.particles.rho, z=-z_cnt*spacing*1.8)
display.start()
def test_neighsearch():
z_cnt = 10
spacing = 0.1
rho0 = 1000.
visc = 0.01
kappa = 0.000
dt = 0.04
g = 0.
pos = create_pos_cube(z_cnt, spacing=spacing)
mass_part = spacing*spacing*spacing*rho0
maxneighs = 200
avgneighs = 100
h = pow(3.*avgneighs*mass_part/(4.*PI*rho0), 1./3.)
solver = IISPHSolver()
solver.particles.pos.value = pos
solver.particles.rho0.value = np.ones((len(pos),1))*rho0
solver.particles.mass.value = np.ones((len(pos),1))*mass_part
solver.particles.info.value = np.ones((len(pos),1))*solver.consts['MAT_FLUID']
solver.particles.visc.value = np.ones((len(pos),1))*visc
solver.particles.kappa.value = np.ones((len(pos),1))*kappa
solver.h = h
solver.dt = dt
solver.g = g
solver.maxneighs = maxneighs
solver.avgneighs = avgneighs
solver.wkrn = 'WendlandC2'
solver.setup()
solver.update()
h = solver.h
p = solver.particles.pos.value
n = np.zeros((len(p), maxneighs), np.int32) - 1
for i in range(len(p)):
count = 0
for j in range(len(p)):
if not(i == j):
pi = np.array(p[i,:3])
pj = np.array(p[j,:3])
d = np.linalg.norm(pi - pj)
if d < h:
n[i,count] = j
count += 1
n_cl = solver.neighsearch.neighbors.value
err = 0
errind = []
for i in range(len(p)):
nu = np.sort(n[i][(n[i] > -1)])
nc = np.sort(n_cl[i][(n_cl[i] > -1)])
#print nu, nc
if not np.all((nu == nc)):
err += 1
errind.append(i)
#print nu, '\n', nc, '\n', '\n'
print 'neigh search errors: ' + str(err)
def test_with_dambreak_perform(steps):
z_cnt = 50
spacing = 0.1
rho0 = 1000.
visc = 0.00
kappa = 0.000
dt = 0.01
g = 0.
pos, info = create_dam_break(z_cnt, lfact=2., wallthick=1, walldensfact=1, spacing=spacing)
mass_part = spacing*spacing*spacing*rho0
maxneighs = 200
avgneighs = 100
h = pow(3.*avgneighs*mass_part/(4.*PI*rho0), 1./3.)
solver = IISPHSolver()
solver.particles.pos.value = pos
solver.particles.rho0.value = np.ones((len(pos),1))*rho0
solver.particles.mass.value = np.ones((len(pos),1))*mass_part
solver.particles.info.value = info
solver.particles.visc.value = np.ones((len(pos),1))*visc
solver.particles.kappa.value = np.ones((len(pos),1))*kappa
solver.h = h
solver.dt = dt
solver.g = g
solver.maxneighs = maxneighs
solver.avgneighs = avgneighs
solver.wkrn = 'WendlandC2'
solver.setup()
for i in range(steps):
solver.update()
def test_with_dambreak_onerun():
z_cnt = 8
spacing = 0.1
rho0 = 1000.
visc = 0.00
kappa = 0.000
dt = 0.01
g = 0.
pos, info = create_dam_break(z_cnt, lfact=2., wallthick=1, walldensfact=1, spacing=spacing)
mass_part = spacing*spacing*spacing*rho0
maxneighs = 200
avgneighs = 100
h = pow(3.*avgneighs*mass_part/(4.*PI*rho0), 1./3.)
solver = IISPHSolver()
solver.particles.pos.value = pos
solver.particles.rho0.value = np.ones((len(pos),1))*rho0
solver.particles.mass.value = np.ones((len(pos),1))*mass_part
solver.particles.info.value = info
solver.particles.visc.value = np.ones((len(pos),1))*visc
solver.particles.kappa.value = np.ones((len(pos),1))*kappa
solver.h = h
solver.dt = dt
solver.g = g
solver.maxneighs = maxneighs
solver.avgneighs = avgneighs
solver.wkrn = 'WendlandC2'
solver.setup()
solver.precompute_values()
solver.update()
print 'rho_max ', np.amax(solver.particles.rho.value)
print 'rho_min ', np.amin(solver.particles.rho.value)
neighs = solver.neighsearch.neighbors.value
np.place(neighs, neighs == -1, [0])
nnz = np.zeros(len(solver.neighsearch.neighbors))
for i in range(len(nnz)):
nnz[i] = np.count_nonzero(neighs[i])
print 'neigh_count ', np.amax(nnz)
print 'force_max ', np.amax(solver.particles.force.value)
print 'force_min ', np.amin(solver.particles.force.value)
print 'vel_adv_max ', np.amax(solver.particles.vel_adv.value)
print 'vel_adv_min ', np.amin(solver.particles.vel_adv.value)
print 'dii_max ', np.amax(solver.particles.dii.value)
print 'dii_min ', np.amin(solver.particles.dii.value)
print 'rho_adv_max ', np.amax(solver.particles.rho_adv.value)
print 'rho_adv_min ', np.amin(solver.particles.rho_adv.value)
print 'aii_max ', np.amax(solver.particles.aii.value)
print 'aii_min ', np.amin(solver.particles.aii.value)
print 'sum_dijpjl_max ', np.amax(solver.particles.sum_dijpjl.value)
print 'sum_dijpjl_min ', np.amin(solver.particles.sum_dijpjl.value)
print 'new_press_max ', np.amax(solver.particles.new_press.value)
print 'new_press_min ', np.amin(solver.particles.new_press.value)
print 'press_max ', np.amax(solver.particles.press.value)
print 'press_min ', np.amin(solver.particles.press.value)
print 'new_rho_max ', np.amax(solver.particles.new_rho.value)
print 'new_rho_min ', np.amin(solver.particles.new_rho.value)
print 'rho_err_max ', np.amax(solver.particles.rho_err.value)
print 'rho_err_min ', np.amin(solver.particles.rho_err.value)
print 'pforce_max ', np.amax(solver.particles.pforce.value)
print 'pforce_min ', np.amin(solver.particles.pforce.value)
print 'vel_max ', np.amax(solver.particles.vel.value)
print 'vel_min ', np.amin(solver.particles.vel.value)