soorat 1.0.0

Soorat — GPU rendering engine for AGNOS
Documentation 
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

// GPU particle simulation compute shader.
// Updates particle positions/velocities each frame.

struct Particle {
    position: vec4<f32>,   // xyz = position, w = life
    velocity: vec4<f32>,   // xyz = velocity, w = size
    color: vec4<f32>,      // rgba
}

struct SimParams {
    delta_time: f32,
    gravity_y: f32,
    damping: f32,
    particle_count: u32,
}

@group(0) @binding(0) var<storage, read_write> particles: array<Particle>;
@group(0) @binding(1) var<uniform> params: SimParams;

@compute @workgroup_size(64)
fn cs_update(@builtin(global_invocation_id) id: vec3<u32>) {
    let idx = id.x;
    if idx >= params.particle_count {
        return;
    }

    var p = particles[idx];

    // Skip dead particles
    if p.position.w <= 0.0 {
        return;
    }

    // Apply gravity
    p.velocity.y += params.gravity_y * params.delta_time;

    // Apply damping
    p.velocity = p.velocity * vec4<f32>(params.damping, params.damping, params.damping, 1.0);

    // Integrate position
    p.position = vec4<f32>(
        p.position.xyz + p.velocity.xyz * params.delta_time,
        p.position.w - params.delta_time, // decrease life
    );

    particles[idx] = p;
}