particle.wgsl.h

Go to the documentation of this file.

#pragma once
 
#include <string>
 
shader_source = R"(
struct Particle {
    id: i32,
    shape_id: i32,
    enable_bound_aabb: i32,
    need_update_linked_list: i32,
 
    material_type: i32,
    density: f32,
    damp_viscous: f32,
    damp_numerical: f32,
 
    bound_min: vec3<f32>,
    padding1: f32, 
    bound_max: vec3<f32>,
    margin: f32,
 
    bound_disp: vec3<f32>,
    padding2: f32, 
 
    pos: vec3<f32>,
    padding3: f32, 
    quaternion: vec4<f32>,
 
    vel: vec3<f32>,
    padding4: f32, 
    spin: vec3<f32>,
    padding5: f32, 
 
    vel_m0p5: vec3<f32>,
    padding6: f32, 
    spin_principal: vec3<f32>,
    padding7: f32, 
 
    force: vec3<f32>,
    padding8: f32, 
    moment: vec3<f32>,
    padding9: f32, 
};
 
@group(0) @binding(0) var<storage, read_write> particles : array<Particle>;
 
@compute @workgroup_size(32)
fn main(@builtin(global_invocation_id) GlobalInvocationID: vec3<u32>) {
    let index = GlobalInvocationID.x;
    if index >= arrayLength(&particles) {
        return;
    }
 
    let dt = 1.0e-6f;  //time step
    particles[index].vel.z += -10.0f * dt;
    particles[index].pos += particles[index].vel * dt;
}
 
)";