-
Notifications
You must be signed in to change notification settings - Fork 0
/
gui.py
119 lines (93 loc) · 4.2 KB
/
gui.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
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
import taichi as ti
from simulator import Simulator
import time
import math
@ti.data_oriented
class SimulationGUI(object):
def __init__(self, sim: Simulator, title: str = "Simulator", resolution=(800, 600), output_img=False) -> None:
super().__init__()
self.sim = sim
self.filename = f'../output/{self.sim.mode}_{self.sim.scene_init}/%04d.png' if output_img else None
self.frame = 0
self.bound_min = ti.Vector([0.0,0.0,0.0], dt=ti.f32)
self.bound_max = self.sim.grid_extent
self.bound_extent = self.bound_max - self.bound_min
self.bound_center = 0.5 * self.bound_extent + self.bound_min
# Particle visualization radius
self.p_radius = self.sim.cell_extent * 0.5
self.resolution = resolution
self.window = ti.ui.Window(title, resolution, vsync=True, show_window=not output_img)
self.canvas = self.window.get_canvas()
self.scene = ti.ui.Scene()
self.camera = ti.ui.Camera()
self.water_color = (0.0, 0.0, 1.0)
self.reset_camera()
self.reset()
def reset(self):
self.t_sim = 0.0
self.t_render = 0.0
self.export_mesh = False
def reset_camera(self):
# Offset the camera position from the center of the bound along -Y
self.camera.position(self.bound_center.x, self.bound_center.y - self.bound_extent.y, self.bound_center.z)
self.camera.lookat(self.bound_center.x, self.bound_center.y, self.bound_center.z)
self.camera.up(0.0, 0.0, 1.0)
self.camera.fov(math.degrees(math.atan2(self.bound_extent.y, self.bound_extent.x)*2.0)) # todo: compute this
# self.camera.projection_mode()
self.scene.set_camera(self.camera)
# def draw_center(self):
# center
def render(self):
# This seems to be a bug... If no point light exists then interal error would occur
self.scene.point_light(pos=tuple(self.bound_center), color=(5, 5, 5))
# Use the ambient light to see the particles
self.scene.ambient_light((1, 1, 1))
# Draw the particles
self.scene.particles(self.sim.particles_position, radius=0.25*self.p_radius, color=self.water_color)
self.canvas.scene(self.scene)
if self.filename is not None:
self.frame += 1
if self.frame % 10 == 0:
print(self.frame)
self.window.write_image(self.filename % self.frame)
else:
self.window.show()
def run(self):
while self.window.running:
# self.camera.track_user_inputs(self.window, movement_speed=0.02, hold_key=ti.ui.RMB)
if self.filename is None:
# handle user input
for e in self.window.get_events(tag='Press'):
#if e.action == ti.ui.PRESS:
self.gui_event_callback(e)
# Export the mesh if required
if self.export_mesh:
self.sim.reconstruct_mesh()
# Simulation step
self.sim.step()
self.t_sim = self.sim.t
# Render
while self.t_render < self.t_sim: # In the case that simulation step is large
t_render_beg = time.time()
# update gui
self.render()
t_render_end = time.time()
self.t_render += t_render_end - t_render_beg
else:
for i in range(3):
self.sim.step()
self.render()
if self.frame >= 300:
break
def gui_event_callback(self, event):
if event.key == "Space": # Export mesh only for one frame
if not self.export_mesh:
self.sim.reconstruct_mesh()
elif event.key == "m": # Toggle the flag to export mesh continuously
self.export_mesh = not self.export_mesh
@ti.func
def compute_screen_pos(self, p):
# 使用ti.cast确保整数类型，以匹配屏幕坐标的要求
screen_x = ti.cast(p.x * self.res_x, ti.i32)
screen_y = ti.cast(p.y * self.res_y, ti.i32)
return ti.Vector([screen_x, screen_y])