grafo 0.16.1

A GPU-accelerated rendering library for Rust
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

// Swizzle BGRA8 bytes (with padded row stride) into packed ARGB32 u32 pixels.
// Input buffer layout matches wgpu copy_texture_to_buffer with bytes_per_row alignment.

struct Params {
    width: u32,
    height: u32,
    padded_bpr: u32, // bytes per row including padding; guaranteed multiple of 256 and 4
    _pad: u32,       // std140 alignment padding
};

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

@group(0) @binding(1)
var<storage, read_write> output_argb: array<u32>;

@group(0) @binding(2)
var<uniform> params: Params;

@compute @workgroup_size(16, 16, 1)
fn cs_main(@builtin(global_invocation_id) gid: vec3<u32>) {
    if (gid.x >= params.width || gid.y >= params.height) {
        return;
    }

    let row_words = (gid.y * params.padded_bpr) / 4u;
    let word_index = row_words + gid.x; // 1 word per pixel (4 bytes)
    let px = input_words[word_index];

    let b = px & 0xffu;
    let g = (px >> 8u) & 0xffu;
    let r = (px >> 16u) & 0xffu;
    let a = (px >> 24u) & 0xffu;

    let out_val = (a << 24u) | (r << 16u) | (g << 8u) | b;
    let out_index = gid.y * params.width + gid.x;
    output_argb[out_index] = out_val;
}