import functools
import math
import os
-import xml.etree.cElementTree as et
+import xml.etree.ElementTree as et
from collections import OrderedDict, namedtuple
from mako.template import Template
# function and a power-of-two size table. The prime numbers are determined
# experimentally.
-# We currently don't use layers in radv, but keeping the ability for anv
-# anyways, so we can use it for device groups.
LAYERS = [
- 'radv'
+ 'radv',
+ 'sqtt'
]
TEMPLATE_H = Template("""\
/* This file generated from ${filename}, don't edit directly. */
-struct radv_dispatch_table {
+struct radv_instance_dispatch_table {
union {
- void *entrypoints[${len(entrypoints)}];
+ void *entrypoints[${len(instance_entrypoints)}];
struct {
- % for e in entrypoints:
+ % for e in instance_entrypoints:
% if e.guard is not None:
#ifdef ${e.guard}
PFN_${e.name} ${e.name};
};
};
-% for e in entrypoints:
+struct radv_physical_device_dispatch_table {
+ union {
+ void *entrypoints[${len(physical_device_entrypoints)}];
+ struct {
+ % for e in physical_device_entrypoints:
+ % if e.guard is not None:
+#ifdef ${e.guard}
+ PFN_${e.name} ${e.name};
+#else
+ void *${e.name};
+# endif
+ % else:
+ PFN_${e.name} ${e.name};
+ % endif
+ % endfor
+ };
+ };
+};
+
+struct radv_device_dispatch_table {
+ union {
+ void *entrypoints[${len(device_entrypoints)}];
+ struct {
+ % for e in device_entrypoints:
+ % if e.guard is not None:
+#ifdef ${e.guard}
+ PFN_${e.name} ${e.name};
+#else
+ void *${e.name};
+# endif
+ % else:
+ PFN_${e.name} ${e.name};
+ % endif
+ % endfor
+ };
+ };
+};
+
+extern const struct radv_instance_dispatch_table radv_instance_dispatch_table;
+%for layer in LAYERS:
+extern const struct radv_physical_device_dispatch_table ${layer}_physical_device_dispatch_table;
+%endfor
+%for layer in LAYERS:
+extern const struct radv_device_dispatch_table ${layer}_device_dispatch_table;
+%endfor
+
+% for e in instance_entrypoints:
+ % if e.alias and e.alias.enabled:
+ <% continue %>
+ % endif
+ % if e.guard is not None:
+#ifdef ${e.guard}
+ % endif
+ ${e.return_type} ${e.prefixed_name('radv')}(${e.decl_params()});
+ % if e.guard is not None:
+#endif // ${e.guard}
+ % endif
+% endfor
+
+% for e in physical_device_entrypoints:
% if e.alias:
<% continue %>
% endif
#endif // ${e.guard}
% endif
% endfor
+
+% for e in device_entrypoints:
+ % if e.alias and e.alias.enabled:
+ <% continue %>
+ % endif
+ % if e.guard is not None:
+#ifdef ${e.guard}
+ % endif
+ % for layer in LAYERS:
+ ${e.return_type} ${e.prefixed_name(layer)}(${e.decl_params()});
+ % endfor
+ % if e.guard is not None:
+#endif // ${e.guard}
+ % endif
+% endfor
""", output_encoding='utf-8')
TEMPLATE_C = Template(u"""\
#include "radv_private.h"
+#include "util/macros.h"
+
struct string_map_entry {
uint32_t name;
uint32_t hash;
* store the index into this big string.
*/
-static const char strings[] =
+<%def name="strmap(strmap, prefix)">
+static const char ${prefix}_strings[] =
% for s in strmap.sorted_strings:
"${s.string}\\0"
% endfor
;
-static const struct string_map_entry string_map_entries[] = {
+static const struct string_map_entry ${prefix}_string_map_entries[] = {
% for s in strmap.sorted_strings:
{ ${s.offset}, ${'{:0=#8x}'.format(s.hash)}, ${s.num} }, /* ${s.string} */
% endfor
/* Hash table stats:
* size ${len(strmap.sorted_strings)} entries
* collisions entries:
-% for i in xrange(10):
+% for i in range(10):
* ${i}${'+' if i == 9 else ' '} ${strmap.collisions[i]}
% endfor
*/
#define none 0xffff
-static const uint16_t string_map[${strmap.hash_size}] = {
+static const uint16_t ${prefix}_string_map[${strmap.hash_size}] = {
% for e in strmap.mapping:
${ '{:0=#6x}'.format(e) if e >= 0 else 'none' },
% endfor
};
+static int
+${prefix}_string_map_lookup(const char *str)
+{
+ static const uint32_t prime_factor = ${strmap.prime_factor};
+ static const uint32_t prime_step = ${strmap.prime_step};
+ const struct string_map_entry *e;
+ uint32_t hash, h;
+ uint16_t i;
+ const char *p;
+
+ hash = 0;
+ for (p = str; *p; p++)
+ hash = hash * prime_factor + *p;
+
+ h = hash;
+ while (1) {
+ i = ${prefix}_string_map[h & ${strmap.hash_mask}];
+ if (i == none)
+ return -1;
+ e = &${prefix}_string_map_entries[i];
+ if (e->hash == hash && strcmp(str, ${prefix}_strings + e->name) == 0)
+ return e->num;
+ h += prime_step;
+ }
+
+ return -1;
+}
+
+static const char *
+${prefix}_entry_name(int num)
+{
+ for (int i = 0; i < ARRAY_SIZE(${prefix}_string_map_entries); i++) {
+ if (${prefix}_string_map_entries[i].num == num)
+ return &${prefix}_strings[${prefix}_string_map_entries[i].name];
+ }
+ return NULL;
+}
+</%def>
+
+${strmap(instance_strmap, 'instance')}
+${strmap(physical_device_strmap, 'physical_device')}
+${strmap(device_strmap, 'device')}
+
/* Weak aliases for all potential implementations. These will resolve to
* NULL if they're not defined, which lets the resolve_entrypoint() function
* either pick the correct entry point.
*/
+% for e in instance_entrypoints:
+ % if e.alias and e.alias.enabled:
+ <% continue %>
+ % endif
+ % if e.guard is not None:
+#ifdef ${e.guard}
+ % endif
+ ${e.return_type} ${e.prefixed_name('radv')}(${e.decl_params()}) __attribute__ ((weak));
+ % if e.guard is not None:
+#endif // ${e.guard}
+ % endif
+% endfor
+
+const struct radv_instance_dispatch_table radv_instance_dispatch_table = {
+% for e in instance_entrypoints:
+ % if e.guard is not None:
+#ifdef ${e.guard}
+ % endif
+ .${e.name} = ${e.prefixed_name('radv')},
+ % if e.guard is not None:
+#endif // ${e.guard}
+ % endif
+% endfor
+};
+
+% for e in physical_device_entrypoints:
+ % if e.alias and e.alias.enabled:
+ <% continue %>
+ % endif
+ % if e.guard is not None:
+#ifdef ${e.guard}
+ % endif
+ ${e.return_type} ${e.prefixed_name('radv')}(${e.decl_params()}) __attribute__ ((weak));
+ % if e.guard is not None:
+#endif // ${e.guard}
+ % endif
+% endfor
+
+const struct radv_physical_device_dispatch_table radv_physical_device_dispatch_table = {
+% for e in physical_device_entrypoints:
+ % if e.guard is not None:
+#ifdef ${e.guard}
+ % endif
+ .${e.name} = ${e.prefixed_name('radv')},
+ % if e.guard is not None:
+#endif // ${e.guard}
+ % endif
+% endfor
+};
+
+
% for layer in LAYERS:
- % for e in entrypoints:
- % if e.alias:
+ % for e in device_entrypoints:
+ % if e.alias and e.alias.enabled:
<% continue %>
% endif
% if e.guard is not None:
#ifdef ${e.guard}
% endif
- ${e.return_type} ${e.prefixed_name(layer)}(${e.decl_params()}) __attribute__ ((weak));
+ % if layer == 'radv':
+ ${e.return_type} __attribute__ ((weak))
+ ${e.prefixed_name('radv')}(${e.decl_params()})
+ {
+ % if e.params[0].type == 'VkDevice':
+ RADV_FROM_HANDLE(radv_device, radv_device, ${e.params[0].name});
+ return radv_device->dispatch.${e.name}(${e.call_params()});
+ % elif e.params[0].type == 'VkCommandBuffer':
+ RADV_FROM_HANDLE(radv_cmd_buffer, radv_cmd_buffer, ${e.params[0].name});
+ return radv_cmd_buffer->device->dispatch.${e.name}(${e.call_params()});
+ % elif e.params[0].type == 'VkQueue':
+ RADV_FROM_HANDLE(radv_queue, radv_queue, ${e.params[0].name});
+ return radv_queue->device->dispatch.${e.name}(${e.call_params()});
+ % else:
+ assert(!"Unhandled device child trampoline case: ${e.params[0].type}");
+ % endif
+ }
+ % else:
+ ${e.return_type} ${e.prefixed_name(layer)}(${e.decl_params()}) __attribute__ ((weak));
+ % endif
% if e.guard is not None:
#endif // ${e.guard}
% endif
% endfor
- const struct radv_dispatch_table ${layer}_layer = {
- % for e in entrypoints:
+ const struct radv_device_dispatch_table ${layer}_device_dispatch_table = {
+ % for e in device_entrypoints:
% if e.guard is not None:
#ifdef ${e.guard}
% endif
};
% endfor
-static void * __attribute__ ((noinline))
-radv_resolve_entrypoint(uint32_t index)
+
+/** Return true if the core version or extension in which the given entrypoint
+ * is defined is enabled.
+ *
+ * If device is NULL, all device extensions are considered enabled.
+ */
+bool
+radv_instance_entrypoint_is_enabled(int index, uint32_t core_version,
+ const struct radv_instance_extension_table *instance)
{
- return radv_layer.entrypoints[index];
+ switch (index) {
+% for e in instance_entrypoints:
+ case ${e.num}:
+ /* ${e.name} */
+ % if e.core_version:
+ return ${e.core_version.c_vk_version()} <= core_version;
+ % elif e.extensions:
+ % for ext in e.extensions:
+ % if ext.type == 'instance':
+ if (instance->${ext.name[3:]}) return true;
+ % else:
+ /* All device extensions are considered enabled at the instance level */
+ return true;
+ % endif
+ % endfor
+ return false;
+ % else:
+ return true;
+ % endif
+% endfor
+ default:
+ return false;
+ }
}
/** Return true if the core version or extension in which the given entrypoint
* is defined is enabled.
*
- * If instance is NULL, we only allow the 3 commands explicitly allowed by the vk
- * spec.
- *
* If device is NULL, all device extensions are considered enabled.
*/
-static bool
-radv_entrypoint_is_enabled(int index, uint32_t core_version,
- const struct radv_instance_extension_table *instance,
- const struct radv_device_extension_table *device)
+bool
+radv_physical_device_entrypoint_is_enabled(int index, uint32_t core_version,
+ const struct radv_instance_extension_table *instance)
{
switch (index) {
-% for e in entrypoints:
+% for e in physical_device_entrypoints:
case ${e.num}:
- % if not e.device_command:
- if (device) return false;
- % endif
- % if e.name == 'vkCreateInstance' or e.name == 'vkEnumerateInstanceExtensionProperties' or e.name == 'vkEnumerateInstanceLayerProperties':
- return !device;
- % elif e.core_version:
- return instance && ${e.core_version.c_vk_version()} <= core_version;
+ /* ${e.name} */
+ % if e.core_version:
+ return ${e.core_version.c_vk_version()} <= core_version;
% elif e.extensions:
- % for ext in e.extensions:
- % if ext.type == 'instance':
- if (instance && instance->${ext.name[3:]}) return true;
- % else:
- if (instance && (!device || device->${ext.name[3:]})) return true;
- % endif
- %endfor
+ % for ext in e.extensions:
+ % if ext.type == 'instance':
+ if (instance->${ext.name[3:]}) return true;
+ % else:
+ /* All device extensions are considered enabled at the instance level */
+ return true;
+ % endif
+ % endfor
return false;
% else:
- return instance;
+ return true;
% endif
% endfor
default:
}
}
-static int
-radv_lookup_entrypoint(const char *name)
+/** Return true if the core version or extension in which the given entrypoint
+ * is defined is enabled.
+ *
+ * If device is NULL, all device extensions are considered enabled.
+ */
+bool
+radv_device_entrypoint_is_enabled(int index, uint32_t core_version,
+ const struct radv_instance_extension_table *instance,
+ const struct radv_device_extension_table *device)
{
- static const uint32_t prime_factor = ${strmap.prime_factor};
- static const uint32_t prime_step = ${strmap.prime_step};
- const struct string_map_entry *e;
- uint32_t hash, h;
- uint16_t i;
- const char *p;
-
- hash = 0;
- for (p = name; *p; p++)
- hash = hash * prime_factor + *p;
-
- h = hash;
- while (1) {
- i = string_map[h & ${strmap.hash_mask}];
- if (i == none)
- return -1;
- e = &string_map_entries[i];
- if (e->hash == hash && strcmp(name, strings + e->name) == 0)
- return e->num;
- h += prime_step;
+ switch (index) {
+% for e in device_entrypoints:
+ case ${e.num}:
+ /* ${e.name} */
+ % if e.core_version:
+ return ${e.core_version.c_vk_version()} <= core_version;
+ % elif e.extensions:
+ % for ext in e.extensions:
+ % if ext.type == 'instance':
+ <% assert False %>
+ % else:
+ if (!device || device->${ext.name[3:]}) return true;
+ % endif
+ % endfor
+ return false;
+ % else:
+ return true;
+ % endif
+% endfor
+ default:
+ return false;
}
+}
- return -1;
+int
+radv_get_instance_entrypoint_index(const char *name)
+{
+ return instance_string_map_lookup(name);
}
-void *
-radv_lookup_entrypoint_unchecked(const char *name)
+int
+radv_get_physical_device_entrypoint_index(const char *name)
+{
+ return physical_device_string_map_lookup(name);
+}
+
+int
+radv_get_device_entrypoint_index(const char *name)
+{
+ return device_string_map_lookup(name);
+}
+
+const char *
+radv_get_instance_entry_name(int index)
+{
+ return instance_entry_name(index);
+}
+
+const char *
+radv_get_physical_device_entry_name(int index)
+{
+ return physical_device_entry_name(index);
+}
+
+const char *
+radv_get_device_entry_name(int index)
+{
+ return device_entry_name(index);
+}
+
+static void * __attribute__ ((noinline))
+radv_resolve_device_entrypoint(uint32_t index)
{
- int index = radv_lookup_entrypoint(name);
- if (index < 0)
- return NULL;
- return radv_resolve_entrypoint(index);
+ return radv_device_dispatch_table.entrypoints[index];
}
void *
-radv_lookup_entrypoint_checked(const char *name,
- uint32_t core_version,
- const struct radv_instance_extension_table *instance,
- const struct radv_device_extension_table *device)
+radv_lookup_entrypoint(const char *name)
{
- int index = radv_lookup_entrypoint(name);
- if (index < 0 || !radv_entrypoint_is_enabled(index, core_version, instance, device))
- return NULL;
- return radv_resolve_entrypoint(index);
+ int idx = radv_get_instance_entrypoint_index(name);
+ if (idx >= 0)
+ return radv_instance_dispatch_table.entrypoints[idx];
+
+ idx = radv_get_physical_device_entrypoint_index(name);
+ if (idx >= 0)
+ return radv_physical_device_dispatch_table.entrypoints[idx];
+
+ idx = radv_get_device_entrypoint_index(name);
+ if (idx >= 0)
+ return radv_resolve_device_entrypoint(idx);
+
+ return NULL;
}""", output_encoding='utf-8')
U32_MASK = 2**32 - 1
self.core_version = None
self.extensions = []
+ def prefixed_name(self, prefix):
+ assert self.name.startswith('vk')
+ return prefix + '_' + self.name[2:]
+
class Entrypoint(EntrypointBase):
def __init__(self, name, return_type, params, guard = None):
super(Entrypoint, self).__init__(name)
self.return_type = return_type
self.params = params
self.guard = guard
- self.device_command = len(params) > 0 and (params[0].type == 'VkDevice' or params[0].type == 'VkQueue' or params[0].type == 'VkCommandBuffer')
- def prefixed_name(self, prefix):
- assert self.name.startswith('vk')
- return prefix + '_' + self.name[2:]
+ def is_physical_device_entrypoint(self):
+ return self.params[0].type in ('VkPhysicalDevice', )
+
+ def is_device_entrypoint(self):
+ return self.params[0].type in ('VkDevice', 'VkCommandBuffer', 'VkQueue')
def decl_params(self):
return ', '.join(p.decl for p in self.params)
def __init__(self, name, entrypoint):
super(EntrypointAlias, self).__init__(name)
self.alias = entrypoint
- self.device_command = entrypoint.device_command
+
+ def is_physical_device_entrypoint(self):
+ return self.alias.is_physical_device_entrypoint()
+
+ def is_device_entrypoint(self):
+ return self.alias.is_device_entrypoint()
def prefixed_name(self, prefix):
- return self.alias.prefixed_name(prefix)
+ if self.alias.enabled:
+ return self.alias.prefixed_name(prefix)
+ return super(EntrypointAlias, self).prefixed_name(prefix)
+
+ @property
+ def params(self):
+ return self.alias.params
+
+ @property
+ def return_type(self):
+ return self.alias.return_type
+
+ def decl_params(self):
+ return self.alias.decl_params()
-def get_entrypoints(doc, entrypoints_to_defines, start_index):
+ def call_params(self):
+ return self.alias.call_params()
+
+def get_entrypoints(doc, entrypoints_to_defines):
"""Extract the entry points from the registry."""
entrypoints = OrderedDict()
if ext_name not in supported_exts:
continue
- if extension.attrib['supported'] != 'vulkan':
- continue
-
ext = supported_exts[ext_name]
ext.type = extension.attrib['type']
e_clone.name = e.name
entrypoints[e.name] = e_clone
- return [e for e in entrypoints.itervalues() if e.enabled]
+ return [e for e in entrypoints.values() if e.enabled]
def get_entrypoints_defines(doc):
"""Maps entry points to extension defines."""
entrypoints_to_defines = {}
- for extension in doc.findall('./extensions/extension[@protect]'):
- define = extension.attrib['protect']
-
- for entrypoint in extension.findall('./require/command'):
- fullname = entrypoint.attrib['name']
- entrypoints_to_defines[fullname] = define
+ platform_define = {}
+ for platform in doc.findall('./platforms/platform'):
+ name = platform.attrib['name']
+ define = platform.attrib['protect']
+ platform_define[name] = define
for extension in doc.findall('./extensions/extension[@platform]'):
platform = extension.attrib['platform']
- define = 'VK_USE_PLATFORM_' + platform.upper() + '_KHR'
+ define = platform_define[platform]
for entrypoint in extension.findall('./require/command'):
fullname = entrypoint.attrib['name']
return entrypoints_to_defines
-
-def gen_code(entrypoints):
- """Generate the C code."""
- strmap = StringIntMap()
- for e in entrypoints:
- strmap.add_string(e.name, e.num)
- strmap.bake()
-
- return TEMPLATE_C.render(entrypoints=entrypoints,
- LAYERS=LAYERS,
- strmap=strmap,
- filename=os.path.basename(__file__))
-
-
def main():
parser = argparse.ArgumentParser()
parser.add_argument('--outdir', help='Where to write the files.',
for filename in args.xml_files:
doc = et.parse(filename)
- entrypoints += get_entrypoints(doc, get_entrypoints_defines(doc),
- start_index=len(entrypoints))
+ entrypoints += get_entrypoints(doc, get_entrypoints_defines(doc))
+
+ device_entrypoints = []
+ physical_device_entrypoints = []
+ instance_entrypoints = []
+ for e in entrypoints:
+ if e.is_device_entrypoint():
+ device_entrypoints.append(e)
+ elif e.is_physical_device_entrypoint():
+ physical_device_entrypoints.append(e)
+ else:
+ instance_entrypoints.append(e)
+
+ device_strmap = StringIntMap()
+ for num, e in enumerate(device_entrypoints):
+ device_strmap.add_string(e.name, num)
+ e.num = num
+ device_strmap.bake()
+
+ physical_device_strmap = StringIntMap()
+ for num, e in enumerate(physical_device_entrypoints):
+ physical_device_strmap.add_string(e.name, num)
+ e.num = num
+ physical_device_strmap.bake()
- for num, e in enumerate(entrypoints):
+ instance_strmap = StringIntMap()
+ for num, e in enumerate(instance_entrypoints):
+ instance_strmap.add_string(e.name, num)
e.num = num
+ instance_strmap.bake()
# For outputting entrypoints.h we generate a radv_EntryPoint() prototype
# per entry point.
- with open(os.path.join(args.outdir, 'radv_entrypoints.h'), 'wb') as f:
- f.write(TEMPLATE_H.render(entrypoints=entrypoints,
- LAYERS=LAYERS,
- filename=os.path.basename(__file__)))
- with open(os.path.join(args.outdir, 'radv_entrypoints.c'), 'wb') as f:
- f.write(gen_code(entrypoints))
+ try:
+ with open(os.path.join(args.outdir, 'radv_entrypoints.h'), 'wb') as f:
+ f.write(TEMPLATE_H.render(instance_entrypoints=instance_entrypoints,
+ physical_device_entrypoints=physical_device_entrypoints,
+ device_entrypoints=device_entrypoints,
+ LAYERS=LAYERS,
+ filename=os.path.basename(__file__)))
+ with open(os.path.join(args.outdir, 'radv_entrypoints.c'), 'wb') as f:
+ f.write(TEMPLATE_C.render(instance_entrypoints=instance_entrypoints,
+ physical_device_entrypoints=physical_device_entrypoints,
+ device_entrypoints=device_entrypoints,
+ LAYERS=LAYERS,
+ instance_strmap=instance_strmap,
+ physical_device_strmap=physical_device_strmap,
+ device_strmap=device_strmap,
+ filename=os.path.basename(__file__)))
+ except Exception:
+ # In the event there's an error, this imports some helpers from mako
+ # to print a useful stack trace and prints it, then exits with
+ # status 1, if python is run with debug; otherwise it just raises
+ # the exception
+ if __debug__:
+ import sys
+ from mako import exceptions
+ sys.stderr.write(exceptions.text_error_template().render() + '\n')
+ sys.exit(1)
+ raise
if __name__ == '__main__':