/usr/lib/python2.7/dist-packages/PySPH-1.0a4.dev0-py2.7-linux-x86_64.egg/pysph/examples/rigid_body/bouncing_cube.py is in python-pysph 0~20160514.git91867dc-4build1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 | """A cube bouncing inside a box. (5 seconds)
This is used to test the rigid body equations.
"""
import numpy as np
from pysph.base.kernels import CubicSpline
from pysph.base.utils import get_particle_array_rigid_body
from pysph.sph.equation import Group
from pysph.sph.integrator import EPECIntegrator
from pysph.solver.application import Application
from pysph.solver.solver import Solver
from pysph.sph.rigid_body import (BodyForce, RigidBodyCollision,
RigidBodyMoments, RigidBodyMotion, RK2StepRigidBody)
dim = 3
dt = 5e-3
tf = 5.0
gz = -9.81
hdx = 1.0
dx = dy = 0.02
rho0 = 10.0
class BouncingCube(Application):
def create_particles(self):
nx, ny, nz = 10, 10, 10
dx = 1.0/(nx-1)
x, y, z = np.mgrid[0:1:nx*1j, 0:1:ny*1j, 0:1:nz*1j]
x = x.flat
y = y.flat
z = (z - 1).flat
m = np.ones_like(x)*dx*dx*rho0
h = np.ones_like(x)*hdx*dx
body = get_particle_array_rigid_body(
name='body', x=x, y=y, z=z, h=h, m=m,
)
body.vc[0] = -5.0
body.vc[2] = -5.0
# Create the tank.
nx, ny, nz = 40, 40, 40
dx = 1.0/(nx-1)
xmin, xmax, ymin, ymax, zmin, zmax = -2, 2, -2, 2, -2, 2
x, y, z = np.mgrid[xmin:xmax:nx*1j, ymin:ymax:ny*1j, zmin:zmax:nz*1j]
interior = ((x < 1.8) & (x > -1.8)) & ((y < 1.8) & (y> -1.8)) & ((z > -1.8) & (z <= 2))
tank = np.logical_not(interior)
x = x[tank].flat
y = y[tank].flat
z = z[tank].flat
m = np.ones_like(x)*dx*dx*rho0
h = np.ones_like(x)*hdx*dx
tank = get_particle_array_rigid_body(
name='tank', x=x, y=y, z=z, h=h, m=m,
)
tank.total_mass[0] = np.sum(m)
return [body, tank]
def create_solver(self):
kernel = CubicSpline(dim=dim)
integrator = EPECIntegrator(body=RK2StepRigidBody())
solver = Solver(kernel=kernel, dim=dim, integrator=integrator,
dt=dt, tf=tf, adaptive_timestep=False)
solver.set_print_freq(10)
return solver
def create_equations(self):
equations = [
Group(equations=[
BodyForce(dest='body', sources=None, gz=gz),
RigidBodyCollision(
dest='body', sources=['tank'], k=1.0, d=2.0, eta=0.1, kt=0.1
)]
),
Group(equations=[RigidBodyMoments(dest='body', sources=None)]),
Group(equations=[RigidBodyMotion(dest='body', sources=None)]),
]
return equations
if __name__ == '__main__':
app = BouncingCube()
app.run()
|