1#!/usr/bin/env python3
2# Copyright (c) 2016 Google Inc.
3
4# Licensed under the Apache License, Version 2.0 (the "License");
5# you may not use this file except in compliance with the License.
6# You may obtain a copy of the License at
7#
8#     http://www.apache.org/licenses/LICENSE-2.0
9#
10# Unless required by applicable law or agreed to in writing, software
11# distributed under the License is distributed on an "AS IS" BASIS,
12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13# See the License for the specific language governing permissions and
14# limitations under the License.
15"""Generates the vendor tool table from the SPIR-V XML registry."""
16
17import errno
18import io
19import os.path
20from xml.etree.ElementTree import XML, XMLParser, TreeBuilder
21
22
23def mkdir_p(directory):
24    """Make the directory, and all its ancestors as required.  Any of the
25    directories are allowed to already exist.
26    This is compatible with Python down to 3.0.
27    """
28
29    if directory == "":
30        # We're being asked to make the current directory.
31        return
32
33    try:
34        os.makedirs(directory)
35    except OSError as e:
36        if e.errno == errno.EEXIST and os.path.isdir(directory):
37            pass
38        else:
39            raise
40
41
42def generate_vendor_table(registry):
43    """Returns a list of C style initializers for the registered vendors
44    and their tools.
45
46    Args:
47      registry: The SPIR-V XMLregistry as an xml.ElementTree
48    """
49
50    lines = []
51    for ids in registry.iter('ids'):
52        if 'vendor' == ids.attrib['type']:
53            for an_id in ids.iter('id'):
54                value = an_id.attrib['value']
55                vendor = an_id.attrib['vendor']
56                if 'tool' in an_id.attrib:
57                    tool = an_id.attrib['tool']
58                    vendor_tool = vendor + ' ' + tool
59                else:
60                    tool = ''
61                    vendor_tool = vendor
62                line = '{' + '{}, "{}", "{}", "{}"'.format(value,
63                                                           vendor,
64                                                           tool,
65                                                           vendor_tool) + '},'
66                lines.append(line)
67    return '\n'.join(lines)
68
69
70def main():
71    import argparse
72    parser = argparse.ArgumentParser(description=
73                                     'Generate tables from SPIR-V XML registry')
74    parser.add_argument('--xml', metavar='<path>',
75                        type=str, required=True,
76                        help='SPIR-V XML Registry file')
77    parser.add_argument('--generator-output', metavar='<path>',
78                        type=str, required=True,
79                        help='output file for SPIR-V generators table')
80    args = parser.parse_args()
81
82    with io.open(args.xml, encoding='utf-8') as xml_in:
83      parser = XMLParser(target=TreeBuilder(), encoding='utf-8')
84      registry = XML(xml_in.read(), parser=parser)
85
86    mkdir_p(os.path.dirname(args.generator_output))
87    with open(args.generator_output, 'w') as f:
88      f.write(generate_vendor_table(registry))
89
90
91if __name__ == '__main__':
92    main()
93