spirv-webgpu-transform 0.1.5

Transform SPIRV to be webgpu friendly
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
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

//! # SPIRV WebGPU Transforms
//!
//! See [https://github.com/davnotdev/spirv-webgpu-transform](https://github.com/davnotdev/spirv-webgpu-transform) for more details.
//!
//! ## Features
//!
//! At the moment, the following transformations are supported:
//!
//! | Feature                           | `spirv-val` | `naga` | `tint` |
//! | --------------------------------- | ----------- | ------ | ------ |
//! | Combined Image Samplers           | ✅          | ✅     | ✅     |
//! | Immediates (Push Constants)       | ✅          | ✅\*   | ✅     |
//! | Binding Arrays                    | ✅          | ✅     | ✅     |
//! | Mixed Depth / Comparison          | ✅          | ⚠️\*   | ❌     |
//! | isnan / isinf Patching            | ✅          | ✅     | ✅     |
//! | Storage Cube Patching             | ✅          | ✅     | ✅     |
//! | Unused Image Sampler Pruning      | ✅          | ✅     | ✅     |
//!
//! > (1)\* 99% OK, just one very specific padding related `naga` bug.
//!
//! > (2)\* Simple cases are OK.
//! > With some [special patches](https://github.com/davnotdev/wgpu/tree/trunk-naga-patches), `naga` can process these.
//!
//! ## Using the result
//!
//! After running an individual shader through one or multiple transformations, you will want to:
//!
//! 1. Know which set bindings were affected, use the output [`CorrectionMap`] for this purpose.
//! 2. Ensure that your vertex and fragment shaders shader the same binding layout, use [`mirrorpatch`] for this purpose
//!

use std::collections::{HashMap, HashSet};

mod correction;
mod immediatespatch;
mod isnanisinfpatch;
mod mirrorpatch;
mod pruneunuseddref;
mod splitbindingarray;
mod splitcombined;
mod splitdref;
mod spv;
mod storagecubepatch;
mod util;

#[cfg(test)]
mod test;

use spv::*;
use util::*;

pub use correction::*;
pub use immediatespatch::*;
pub use isnanisinfpatch::*;
pub use mirrorpatch::*;
pub use pruneunuseddref::*;
pub use splitbindingarray::*;
pub use splitcombined::*;
pub use splitdref::*;
pub use storagecubepatch::*;

#[derive(Debug, Clone)]
struct InstructionInsert {
    previous_spv_idx: usize,
    instruction: Vec<u32>,
}

#[derive(Debug, Clone)]
struct WordInsert {
    idx: usize,
    word: u32,
    head_idx: usize,
}

/// Helper to convert a `&[u8]` into a `Vec<u32>`.
pub fn u8_slice_to_u32_vec(vec: &[u8]) -> Vec<u32> {
    assert_eq!(
        vec.len() % 4,
        0,
        "Input slice length must be a multiple of 4."
    );

    vec.chunks_exact(4)
        .map(|chunk| {
            (chunk[0] as u32)
                | ((chunk[1] as u32) << 8)
                | ((chunk[2] as u32) << 16)
                | ((chunk[3] as u32) << 24)
        })
        .collect::<Vec<_>>()
}

/// Helper to convert a `&[u32]` into a `Vec<u8>`.
pub fn u32_slice_to_u8_vec(vec: &[u32]) -> Vec<u8> {
    vec.iter()
        .flat_map(|&num| {
            vec![
                (num & 0xFF) as u8,
                ((num >> 8) & 0xFF) as u8,
                ((num >> 16) & 0xFF) as u8,
                ((num >> 24) & 0xFF) as u8,
            ]
        })
        .collect::<Vec<u8>>()
}