libprocessing/crates/processing_pyo3/examples/particles_gpu_surface_lit.py at main · processing/libprocessing · GitHub

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
from processing_py import *
from math import cos, sin

NX, NY = 200, 200
EXTENT = 30.0

p = None
gen = None


def setup():
    global p, gen
    size(900, 700)
    window_title("GPU surface — color + normals generated on the GPU")
    mode_3d()

    p = create_particles(
        NX * NY,
        attributes=[Attribute.position(), Attribute.color(), Attribute.normal()],
    )
    idx = p.index_buffer((NX - 1) * (NY - 1) * 6)
    gen = create_compute(load_shader("shaders/gen_colored_surface.wesl"))
    gen.set(indices=idx)


def draw():
    background(6, 8, 14)

    t = elapsed_time
    d = EXTENT * 1.3
    camera_position(cos(t * 0.08) * d, EXTENT * 0.6, sin(t * 0.08) * d)
    camera_look_at(0.0, 0.0, 0.0)

    gen.set(nx=NX, ny=NY, time=t, extent=EXTENT)
    p.apply(gen)

    particles(p, topology=TRIANGLES)


run()