slosh3d 0.4.1

Cross-platform GPU 3D Material Point Method implementation.
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

struct RigidParticleIndices_std430_0
{
    @align(16) triangle_0 : vec3<u32>,
    @align(4) collider_0 : u32,
};

@binding(0) @group(0) var<storage, read> entryPointParams_rigid_particle_indices_0 : array<RigidParticleIndices_std430_0>;

struct Quat_std430_0
{
    @align(16) coords_0 : vec4<f32>,
};

struct Sim3_std430_0
{
    @align(16) rotation_0 : Quat_std430_0,
    @align(16) translation_scale_0 : vec4<f32>,
};

@binding(1) @group(0) var<storage, read> entryPointParams_poses_0 : array<Sim3_std430_0>;

@binding(2) @group(0) var<storage, read> entryPointParams_local_pts_0 : array<vec3<f32>>;

@binding(3) @group(0) var<storage, read_write> entryPointParams_world_pts_0 : array<vec3<f32>>;

@binding(4) @group(0) var<storage, read> entryPointParams_vertex_collider_ids_0 : array<u32>;

@binding(5) @group(0) var<storage, read> entryPointParams_poses_1 : array<Sim3_std430_0>;

@binding(6) @group(0) var<storage, read> entryPointParams_local_pts_1 : array<vec3<f32>>;

@binding(7) @group(0) var<storage, read_write> entryPointParams_world_pts_1 : array<vec3<f32>>;

fn x2A_0( q_0 : ptr<function, Quat_std430_0>,  v_0 : vec3<f32>) -> vec3<f32>
{
    var _S1 : vec3<f32> = (*q_0).coords_0.xyz;
    var _S2 : vec3<f32> = cross(_S1, v_0) * vec3<f32>(2.0f);
    return _S2 * vec3<f32>((*q_0).coords_0.w) + cross(_S1, _S2) + v_0;
}

fn Sim3_mul_pt_0( this_0 : ptr<function, Sim3_std430_0>,  pt_0 : vec3<f32>) -> vec3<f32>
{
    var _S3 : Quat_std430_0 = (*this_0).rotation_0;
    var _S4 : vec3<f32> = x2A_0(&(_S3), pt_0 * vec3<f32>((*this_0).translation_scale_0.w));
    return _S4 + (*this_0).translation_scale_0.xyz;
}

fn getCount_0() -> i32
{
    var _S5 : vec2<u32> = vec2<u32>(arrayLength(&entryPointParams_local_pts_0), 16);
    return i32(_S5.x);
}

@compute
@workgroup_size(64, 1, 1)
fn transform_sample_points(@builtin(global_invocation_id) invocation_id_0 : vec3<u32>)
{
    var _S6 : u32 = invocation_id_0.x;
    var _S7 : i32 = getCount_0();
    if(_S6 < u32(_S7))
    {
        var _S8 : Sim3_std430_0 = entryPointParams_poses_0[entryPointParams_rigid_particle_indices_0[_S6].collider_0];
        var _S9 : vec3<f32> = Sim3_mul_pt_0(&(_S8), entryPointParams_local_pts_0[_S6]);
        entryPointParams_world_pts_0[_S6] = _S9;
    }
    return;
}

fn getCount_1() -> i32
{
    var _S10 : vec2<u32> = vec2<u32>(arrayLength(&entryPointParams_local_pts_1), 16);
    return i32(_S10.x);
}

@compute
@workgroup_size(64, 1, 1)
fn transform_shape_points(@builtin(global_invocation_id) invocation_id_1 : vec3<u32>)
{
    var _S11 : u32 = invocation_id_1.x;
    var _S12 : i32 = getCount_1();
    if(_S11 < u32(_S12))
    {
        var _S13 : Sim3_std430_0 = entryPointParams_poses_1[entryPointParams_vertex_collider_ids_0[_S11]];
        var _S14 : vec3<f32> = Sim3_mul_pt_0(&(_S13), entryPointParams_local_pts_1[_S11]);
        entryPointParams_world_pts_1[_S11] = _S14;
    }
    return;
}