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
109
110
111
|
COPYRIGHT = """\
/*
* Copyright (C) 2017 Intel Corporation
*
* Permission is hereby granted, free of charge, to any person obtaining a
* copy of this software and associated documentation files (the "Software"),
* to deal in the Software without restriction, including without limitation
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
* and/or sell copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice (including the next
* paragraph) shall be included in all copies or substantial portions of the
* Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
* DEALINGS IN THE SOFTWARE.
*/
"""
import argparse
import json
from sys import stdout
from mako.template import Template
def collect_data(spirv, kind):
for x in spirv["operand_kinds"]:
if x["kind"] == kind:
operands = x
break
# There are some duplicate values in some of the tables (thanks guys!), so
# filter them out.
seen = set()
values = []
for x in operands["enumerants"]:
if x["value"] not in seen:
seen.add(x["value"])
values.append(x["enumerant"])
return (kind, values)
def collect_opcodes(spirv):
seen = set()
values = []
for x in spirv["instructions"]:
# Handle aliases by choosing the first one in the grammar.
# E.g. OpDecorateString and OpDecorateStringGOOGLE share same opcode.
if x["opcode"] in seen:
continue
opcode = x["opcode"]
name = x["opname"]
assert name.startswith("Op")
values.append(name[2:])
seen.add(opcode)
return ("Op", values)
def parse_args():
p = argparse.ArgumentParser()
p.add_argument("json")
p.add_argument("out")
return p.parse_args()
TEMPLATE = Template("""\
/* DO NOT EDIT - This file is generated automatically by spirv_info_c.py script */
""" + COPYRIGHT + """\
#include "spirv_info.h"
% for kind,values in info:
const char *
spirv_${kind.lower()}_to_string(Spv${kind} v)
{
switch (v) {
% for name in values:
case Spv${kind}${name}: return "Spv${kind}${name}";
% endfor
case Spv${kind}Max: break; /* silence warnings about unhandled enums. */
}
return "unknown";
}
% endfor
""")
if __name__ == "__main__":
pargs = parse_args()
spirv_info = json.JSONDecoder().decode(open(pargs.json, "r").read())
info = [
collect_data(spirv_info, "AddressingModel"),
collect_data(spirv_info, "BuiltIn"),
collect_data(spirv_info, "Capability"),
collect_data(spirv_info, "Decoration"),
collect_data(spirv_info, "Dim"),
collect_data(spirv_info, "ExecutionMode"),
collect_data(spirv_info, "ExecutionModel"),
collect_data(spirv_info, "ImageFormat"),
collect_data(spirv_info, "StorageClass"),
collect_opcodes(spirv_info),
]
with open(pargs.out, 'w') as f:
f.write(TEMPLATE.render(info=info))
|
