mirror of
https://github.com/KhronosGroup/SPIRV-Tools
synced 2024-11-22 03:30:06 +00:00
6c8b460eb1
Those scripts are running on Python2.7 if build with the NDK tools. Under python2.7, io.open will load as "utf-8" since we give the encoding option, and return a "unicode" string. Under Python3, the open() function will return a UTF-8 string. This means the XMLParser needs to have a different 'encoding' option depending on the python version. For some reason I don't know, the XMLParser still fails if we use 'unicode' under Python2.7. But converting the unicode string to utf-8 does work. Signed-off-by: Nathan Gauër <brioche@google.com>
101 lines
3.5 KiB
Python
Executable File
101 lines
3.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Copyright (c) 2016 Google Inc.
|
|
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
"""Generates the vendor tool table from the SPIR-V XML registry."""
|
|
|
|
import errno
|
|
import io
|
|
import os.path
|
|
import platform
|
|
from xml.etree.ElementTree import XML, XMLParser, TreeBuilder
|
|
|
|
|
|
def mkdir_p(directory):
|
|
"""Make the directory, and all its ancestors as required. Any of the
|
|
directories are allowed to already exist.
|
|
This is compatible with Python down to 3.0.
|
|
"""
|
|
|
|
if directory == "":
|
|
# We're being asked to make the current directory.
|
|
return
|
|
|
|
try:
|
|
os.makedirs(directory)
|
|
except OSError as e:
|
|
if e.errno == errno.EEXIST and os.path.isdir(directory):
|
|
pass
|
|
else:
|
|
raise
|
|
|
|
|
|
def generate_vendor_table(registry):
|
|
"""Returns a list of C style initializers for the registered vendors
|
|
and their tools.
|
|
|
|
Args:
|
|
registry: The SPIR-V XMLregistry as an xml.ElementTree
|
|
"""
|
|
|
|
lines = []
|
|
for ids in registry.iter('ids'):
|
|
if 'vendor' == ids.attrib['type']:
|
|
for an_id in ids.iter('id'):
|
|
value = an_id.attrib['value']
|
|
vendor = an_id.attrib['vendor']
|
|
if 'tool' in an_id.attrib:
|
|
tool = an_id.attrib['tool']
|
|
vendor_tool = vendor + ' ' + tool
|
|
else:
|
|
tool = ''
|
|
vendor_tool = vendor
|
|
line = '{' + '{}, "{}", "{}", "{}"'.format(value,
|
|
vendor,
|
|
tool,
|
|
vendor_tool) + '},'
|
|
lines.append(line)
|
|
return '\n'.join(lines)
|
|
|
|
|
|
def main():
|
|
import argparse
|
|
parser = argparse.ArgumentParser(description=
|
|
'Generate tables from SPIR-V XML registry')
|
|
parser.add_argument('--xml', metavar='<path>',
|
|
type=str, required=True,
|
|
help='SPIR-V XML Registry file')
|
|
parser.add_argument('--generator-output', metavar='<path>',
|
|
type=str, required=True,
|
|
help='output file for SPIR-V generators table')
|
|
args = parser.parse_args()
|
|
|
|
with io.open(args.xml, encoding='utf-8') as xml_in:
|
|
# Python3 default str to UTF-8. But Python2.7 (in case of NDK build,
|
|
# don't be fooled by the shebang) is returning a unicode string.
|
|
# So depending of the version, we need to make sure the correct
|
|
# encoding is used.
|
|
content = xml_in.read()
|
|
if platform.python_version_tuple()[0] == '2':
|
|
content = content.encode('utf-8')
|
|
parser = XMLParser(target=TreeBuilder(), encoding='utf-8')
|
|
registry = XML(content, parser=parser)
|
|
|
|
mkdir_p(os.path.dirname(args.generator_output))
|
|
with open(args.generator_output, 'w') as f:
|
|
f.write(generate_vendor_table(registry))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|