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
120
121
122
123
124
125
126
127
|
"""
GPU Version of https://github.com/salt-die/ripple
Hold left mouse button to place drop in the surface
"""
import random
from pathlib import Path
import moderngl
import numpy as np
import moderngl_window
from moderngl_window import geometry, screenshot
class Water(moderngl_window.WindowConfig):
title = "Water"
resource_dir = (Path(__file__) / "../../resources").absolute()
aspect_ratio = None # We'll do manual viewport for now
window_size = 1280, 720
resizable = False
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.size = self.wnd.buffer_size
self.viewport = (0, 0, self.size[0], self.size[1])
self.quad_fs = geometry.quad_fs()
self.sprite = geometry.quad_2d(size=(9 / self.wnd.size[0], 9 / self.wnd.size[1]))
self.texture_1 = self.ctx.texture(self.size, components=3)
self.texture_2 = self.ctx.texture(self.size, components=3)
self.fbo_1 = self.ctx.framebuffer(color_attachments=[self.texture_1])
self.fbo_1.viewport = self.viewport
self.fbo_2 = self.ctx.framebuffer(color_attachments=[self.texture_2])
self.fbo_2.viewport = self.viewport
# fmt: off
drop = np.array([
[0.0, 0.0, 1/6, 1/5, 1/4, 1/5, 1/6, 0.0, 0.0],
[0.0, 1/6, 1/5, 1/4, 1/3, 1/4, 1/5, 1/6, 0.0],
[1/6, 1/5, 1/4, 1/3, 1/2, 1/3, 1/4, 1/5, 1/6],
[1/5, 1/4, 1/3, 1/2, 1.0, 1/2, 1/3, 1/4, 1/5],
[1/4, 1/3, 1/2, 1.0, 1.0, 1.0, 1/2, 1/3, 1/4],
[1/5, 1/4, 1/3, 1/2, 1.0, 1/2, 1/3, 1/4, 1/5],
[1/6, 1/5, 1/4, 1/3, 1/2, 1/3, 1/4, 1/5, 1/6],
[0.0, 1/6, 1/5, 1/4, 1/3, 1/4, 1/5, 1/6, 0.0],
[0.0, 0.0, 1/6, 1/5, 1/4, 1/5, 1/6, 0.0, 0.0],
])
# fmt: on
self.drops_texture = self.ctx.texture((9, 9), components=1, dtype="f4")
self.drops_texture.write(drop.astype("f4").tobytes())
# programs
self.drop_program = self.load_program("programs/water/drop.glsl")
self.wave_program = self.load_program("programs/water/wave.glsl")
self.texture_program = self.load_program("programs/water/texture.glsl")
self.wave_program["texture0"].value = 0
self.wave_program["texture1"].value = 1
self.mouse_pos = 0, 0
self.wnd.fbo.viewport = self.viewport
def on_render(self, time, frame_time):
# randomize color
self.drop_program["color"].value = random.random(), random.random(), random.random()
self.fbo_2.use()
# Render drop (with additive blending) when mouse is pressed
if self.wnd.mouse_states.any:
self.ctx.enable(moderngl.BLEND)
self.ctx.blend_func = moderngl.ONE, moderngl.ONE
self.drops_texture.use()
self.drop_program["pos"].value = self.mouse_pos
self.sprite.render(self.drop_program)
self.ctx.disable(moderngl.BLEND)
# HACK: Just draw 100 new drops per frame (copy paste from above)
# This is pretty terrible and slow!
self.ctx.enable(moderngl.BLEND)
self.ctx.blend_func = moderngl.ONE, moderngl.ONE
self.drops_texture.use()
for i in range(10):
self.drop_program["pos"].value = random.random() * 2 - 1.0, random.random() * 2 - 1
self.sprite.render(self.drop_program)
self.ctx.disable(moderngl.BLEND)
self.fbo_1.use()
# Process the water
self.texture_2.use(location=0)
self.texture_1.use(location=1)
self.quad_fs.render(self.wave_program)
# Render the result to the screen.
# We can blit only when the texture format matches the default framebuffer
self.wnd.fbo.use()
self.texture_1.use()
self.quad_fs.render(self.texture_program)
# Swap texture and fbo
self.texture_1, self.texture_2 = self.texture_2, self.texture_1
self.fbo_1, self.fbo_2 = self.fbo_2, self.fbo_1
def on_mouse_position_event(self, x, y, dx, dy):
xx = x * 2 / self.wnd.size[0] - 1.0
yy = -y * 2 / self.wnd.size[1] + 1.0
self.mouse_pos = xx, yy
def on_mouse_drag_event(self, x, y, dx, dy):
self.mouse_position_event(x, y, dx, dy)
def on_key_event(self, key, action, modifiers):
keys = self.wnd.keys
# Key presses
if action == keys.ACTION_PRESS:
if key == keys.F1:
screenshot.create(self.fbo_1)
if __name__ == "__main__":
moderngl_window.run_window_config(Water)
|
