forked from erizmr/taichi_sph
-
Notifications
You must be signed in to change notification settings - Fork 3
/
demo.py
38 lines (31 loc) · 1.1 KB
/
demo.py
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
import taichi as ti
import numpy as np
from particle_system import ParticleSystem
from wcsph import WCSPHSolver
ti.init(arch=ti.cpu)
# Use GPU for higher peformance if available
# ti.init(arch=ti.gpu, device_memory_GB=3, packed=True)
if __name__ == "__main__":
ps = ParticleSystem((512, 512))
ps.add_cube(lower_corner=[6, 2],
cube_size=[3.0, 5.0],
velocity=[-5.0, -10.0],
density=1000.0,
color=0x956333,
material=1)
ps.add_cube(lower_corner=[3, 1],
cube_size=[2.0, 6.0],
velocity=[0.0, -20.0],
density=1000.0,
color=0x956333,
material=1)
wcsph_solver = WCSPHSolver(ps)
gui = ti.GUI(background_color=0xFFFFFF)
while gui.running:
for i in range(5):
wcsph_solver.step()
particle_info = ps.dump()
gui.circles(particle_info['position'] * ps.screen_to_world_ratio / 512,
radius=ps.particle_radius / 1.5 * ps.screen_to_world_ratio,
color=0x956333)
gui.show()