wrld 1.0.0

A wgpu derive macro to help with buffer description
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
106
107
108

const F16 : usize = 2;

#[derive(Copy, Clone, Debug)]
pub struct TypeToWGPU {
    pub offset: u64,
    pub ty: wgpu::VertexFormat,
}

#[derive(Copy, Clone)]
pub struct WGPUData {
    pub wgpu_type: TypeToWGPU,
    pub shader_location: u32
}

const TYPE_MAPPER : [(&str, TypeToWGPU); 34] = [
    ("u32", TypeToWGPU { offset: std::mem::size_of::<u32>() as u64, ty: wgpu::VertexFormat::Uint32 }),
    ("f32", TypeToWGPU { offset: std::mem::size_of::<f32>() as u64, ty: wgpu::VertexFormat::Float32 }),
    ("s32", TypeToWGPU { offset: std::mem::size_of::<i32>() as u64, ty: wgpu::VertexFormat::Sint32 }),
    ("f64", TypeToWGPU { offset: std::mem::size_of::<f64>() as u64, ty: wgpu::VertexFormat::Float64 }),
    ("u8x2", TypeToWGPU { offset: std::mem::size_of::<[u8; 2]>() as u64, ty: wgpu::VertexFormat::Uint8x2 }),
    ("u8x4", TypeToWGPU { offset: std::mem::size_of::<[u8; 4]>() as u64, ty: wgpu::VertexFormat::Uint8x4 }),
    ("s8x2", TypeToWGPU { offset: std::mem::size_of::<[i8; 2]>() as u64, ty: wgpu::VertexFormat::Sint8x2 }),
    ("s8x4", TypeToWGPU { offset: std::mem::size_of::<[i8; 4]>() as u64, ty: wgpu::VertexFormat::Sint8x4 }),
    ("un8x2", TypeToWGPU { offset: std::mem::size_of::<[u8; 2]>() as u64, ty: wgpu::VertexFormat::Unorm8x2 }),
    ("un8x4", TypeToWGPU { offset: std::mem::size_of::<[u8; 4]>() as u64, ty: wgpu::VertexFormat::Unorm8x4 }),
    ("sn8x2", TypeToWGPU { offset: std::mem::size_of::<[i8; 2]>() as u64, ty: wgpu::VertexFormat::Snorm8x2 }),
    ("sn8x4", TypeToWGPU { offset: std::mem::size_of::<[i8; 4]>() as u64, ty: wgpu::VertexFormat::Snorm8x4 }),
    ("u16x2", TypeToWGPU { offset: std::mem::size_of::<[u16; 2]>() as u64, ty: wgpu::VertexFormat::Uint16x2 }),
    ("u16x4", TypeToWGPU { offset: std::mem::size_of::<[u16; 4]>() as u64, ty: wgpu::VertexFormat::Uint16x4 }),
    ("s16x2", TypeToWGPU { offset: std::mem::size_of::<[i16; 2]>() as u64, ty: wgpu::VertexFormat::Sint16x2 }),
    ("s16x4", TypeToWGPU { offset: std::mem::size_of::<[i16; 4]>() as u64, ty: wgpu::VertexFormat::Sint16x4 }),
    ("un16x2", TypeToWGPU { offset: std::mem::size_of::<[u16; 2]>() as u64, ty: wgpu::VertexFormat::Unorm16x2 }),
    ("un16x4", TypeToWGPU { offset: std::mem::size_of::<[u16; 4]>() as u64, ty: wgpu::VertexFormat::Unorm16x4 }),
    ("sn16x2", TypeToWGPU { offset: std::mem::size_of::<[i16; 2]>() as u64, ty: wgpu::VertexFormat::Snorm16x2 }),
    ("sn16x4", TypeToWGPU { offset: std::mem::size_of::<[i16; 4]>() as u64, ty: wgpu::VertexFormat::Snorm16x4 }),
    ("f16x2", TypeToWGPU { offset: (F16 * 2) as u64, ty: wgpu::VertexFormat::Float16x2 }),
    ("f16x4", TypeToWGPU { offset: (F16 * 4) as u64, ty: wgpu::VertexFormat::Float16x4 }),
    ("f32x2", TypeToWGPU { offset: std::mem::size_of::<[f32; 2]>() as u64, ty: wgpu::VertexFormat::Float32x2 }),
    ("f32x3", TypeToWGPU { offset: std::mem::size_of::<[f32; 3]>() as u64, ty: wgpu::VertexFormat::Float32x3 }),
    ("f32x4", TypeToWGPU { offset: std::mem::size_of::<[f32; 4]>() as u64, ty: wgpu::VertexFormat::Float32x4 }),
    ("u32x2", TypeToWGPU { offset: std::mem::size_of::<[u32; 2]>() as u64, ty: wgpu::VertexFormat::Uint32x2 }),
    ("u32x3", TypeToWGPU { offset: std::mem::size_of::<[u32; 3]>() as u64, ty: wgpu::VertexFormat::Uint32x3 }),
    ("u32x4", TypeToWGPU { offset: std::mem::size_of::<[u32; 4]>() as u64, ty: wgpu::VertexFormat::Uint32x4 }),
    ("s32x2", TypeToWGPU { offset: std::mem::size_of::<[i32; 2]>() as u64, ty: wgpu::VertexFormat::Sint32x2 }),
    ("s32x3", TypeToWGPU { offset: std::mem::size_of::<[i32; 3]>() as u64, ty: wgpu::VertexFormat::Sint32x3 }),
    ("s32x4", TypeToWGPU { offset: std::mem::size_of::<[i32; 4]>() as u64, ty: wgpu::VertexFormat::Sint32x4 }),
    ("f64x2", TypeToWGPU { offset: std::mem::size_of::<[f64; 2]>() as u64, ty: wgpu::VertexFormat::Float64x2 }),
    ("f64x3", TypeToWGPU { offset: std::mem::size_of::<[f32; 3]>() as u64, ty: wgpu::VertexFormat::Float64x3 }),
    ("f64x4", TypeToWGPU { offset: std::mem::size_of::<[f64; 4]>() as u64, ty: wgpu::VertexFormat::Float64x4 })
];

fn get_type(name: &str) -> Result<TypeToWGPU, String> {
    for i in TYPE_MAPPER {
        if i.0 == name {
            return Ok(i.1)
        }
    }

    Err(format!("Cannot get type for {:?}", name))
}

pub fn get_allowed_type(name: &str) -> std::vec::Vec<&str> {
    let mut vec: std::vec::Vec::<&str> = std::vec::Vec::new();
    for i in TYPE_MAPPER {
        if i.0.contains(name) {
            vec.push(i.0);
        }
    }
    vec
}

pub fn convert_mat_type_to_wgou(name: &String, shader_location: u32, ty: &mut String) -> std::vec::Vec<WGPUData> {
    let mut wgpu_vec: Vec<WGPUData> = std::vec::Vec::new();
    let mut row = ty.clone();
    row.push_str(&name[name.len() - 2..name.len()]);

    let column: &u32 = &name[name.len() - 3..name.len() - 2].parse().unwrap();
    let mut final_ty = convert_type_to_wgpu(&row, shader_location).unwrap_or_else(|_| {

        let allowed_types = get_allowed_type(ty)
            .join(" or ")
            .replace(ty.as_str(), format!("mat{:?}", column).as_str());

        panic!("Matrix {} cannot be use with {} ! Available matrix are {} for {}", name, ty, allowed_types, ty);
    });

    for i in 0..*column {
        final_ty.shader_location += if i == 0 { 0 } else { 1 };
        wgpu_vec.push(final_ty);
    }

    wgpu_vec
}

pub fn convert_type_to_wgpu(name: &String, shader_location: u32) -> Result<WGPUData, String> {
    let wgpu_type = get_type(name.as_str())?;

    Ok(WGPUData { wgpu_type, shader_location })
}

pub fn has_type(name: &str) -> bool {
    for i in TYPE_MAPPER {
        if i.0 == name {
            return true
        }
    }
    false
}