spirq-spvasm 0.1.1

Tools for SPIR-V Assembly interaction.
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

from collections import defaultdict
import json

with open("assets/spirv/spirv.core.grammar.json") as f:
    j = json.load(f)

bit_enums = defaultdict(dict)
value_enums = defaultdict(dict)
for operand_kind in j["operand_kinds"]:
    category = operand_kind["category"]
    kind = operand_kind["kind"]

    if category == "BitEnum":
        enumerants = operand_kind["enumerants"]
        for enumerant in enumerants:
            bit_enums[kind][enumerant["enumerant"]] = enumerant["value"]
    elif category == "ValueEnum":
        enumerants = operand_kind["enumerants"]
        for enumerant in enumerants:
            value_enums[kind][enumerant["enumerant"]] = enumerant["value"]

out = []

out += [
    "use anyhow::{bail, Result};",
    "",
    "pub fn enum_from_str(ety: &str, name: &str) -> Result<u32> {",
    "    let out: u32 = match ety {",
]

# ValueEnum
for kind, enumerants in value_enums.items():
    if len(enumerants) == 0:
        continue

    out += [
        f'        "{kind}" => match name {{',
    ]
    for name, value in enumerants.items():
        out += [
            f'            "{name}" => {value},',
        ]
    out += [
        '            _ => bail!("unknown enum: {}::{}", ety, name)',
        "        }",
    ]

# BitEnum
for kind, enumerants in bit_enums.items():
    if len(enumerants) == 0:
        continue

    out += [
        f'        "{kind}" => match name {{',
    ]
    for name, value in enumerants.items():
        out += [
            f'            "{name}" => {value},',
        ]
    out += [
        '            _ => bail!("unknown enum: {}::{}", ety, name)',
        "        }",
    ]

out += [
    '        _ => bail!("unknown enum: {}::{}", ety, name),',
    "    };",
    "    Ok(out)",
    "}",
    "",
]

with open("spirq-spvasm/src/generated/enum_from_str.rs", "w") as f:
    f.write("\n".join(out))