blob: 3f7e7b3b7eacde04ce0166f98a9421fd6cdf76c9 [file] [log] [blame]
Tobin Ehlis12076fc2014-10-22 09:06:33 -06001#!/usr/bin/env python3
2#
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -06003# VK
Tobin Ehlis12076fc2014-10-22 09:06:33 -06004#
5# Copyright (C) 2014 LunarG, Inc.
6#
7# Permission is hereby granted, free of charge, to any person obtaining a
8# copy of this software and associated documentation files (the "Software"),
9# to deal in the Software without restriction, including without limitation
10# the rights to use, copy, modify, merge, publish, distribute, sublicense,
11# and/or sell copies of the Software, and to permit persons to whom the
12# Software is furnished to do so, subject to the following conditions:
13#
14# The above copyright notice and this permission notice shall be included
15# in all copies or substantial portions of the Software.
16#
17# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
20# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
22# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
23# DEALINGS IN THE SOFTWARE.
24#
25# Authors:
26# Chia-I Wu <olv@lunarg.com>
27
28import sys
Tobin Ehlis14ff0852014-12-17 17:44:50 -070029import os
Mark Lobodzinski7c75b852015-05-05 15:01:37 -050030import re
Tobin Ehlis12076fc2014-10-22 09:06:33 -060031
Courtney Goeltzenleuchterf53c3cb2015-04-14 14:55:44 -060032import vulkan
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -060033import vk_helper
Tobin Ehlis08fafd02015-06-12 12:49:01 -060034from source_line_info import sourcelineinfo
Tobin Ehlis12076fc2014-10-22 09:06:33 -060035
Jon Ashburn95a77ba2015-05-15 15:09:35 -060036def proto_is_global(proto):
Courtney Goeltzenleuchter110fdf92015-06-29 15:39:26 -060037 if proto.params[0].ty == "VkInstance" or proto.params[0].ty == "VkPhysicalDevice" or proto.name == "CreateInstance" or proto.name == "GetGlobalLayerProperties" or proto.name == "GetGlobalExtensionProperties" or proto.name == "GetPhysicalDeviceLayerProperties" or proto.name == "GetPhysicalDeviceExtensionProperties":
Jon Ashburn95a77ba2015-05-15 15:09:35 -060038 return True
39 else:
40 return False
41
Mike Stroyan938c2532015-04-03 13:58:35 -060042def generate_get_proc_addr_check(name):
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -060043 return " if (!%s || %s[0] != 'v' || %s[1] != 'k')\n" \
44 " return NULL;" % ((name,) * 3)
Mike Stroyan938c2532015-04-03 13:58:35 -060045
Mark Lobodzinski7c75b852015-05-05 15:01:37 -050046def ucc_to_U_C_C(CamelCase):
47 temp = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', CamelCase)
48 return re.sub('([a-z0-9])([A-Z])', r'\1_\2', temp).upper()
49
Tobin Ehlis12076fc2014-10-22 09:06:33 -060050class Subcommand(object):
51 def __init__(self, argv):
52 self.argv = argv
Courtney Goeltzenleuchterf53c3cb2015-04-14 14:55:44 -060053 self.headers = vulkan.headers
54 self.protos = vulkan.protos
Mike Stroyan3e3a1eb2015-04-03 17:13:23 -060055 self.no_addr = False
56 self.layer_name = ""
Tobin Ehlis08fafd02015-06-12 12:49:01 -060057 self.lineinfo = sourcelineinfo()
Tobin Ehlis12076fc2014-10-22 09:06:33 -060058
59 def run(self):
Tobin Ehlis12076fc2014-10-22 09:06:33 -060060 print(self.generate())
61
62 def generate(self):
63 copyright = self.generate_copyright()
64 header = self.generate_header()
65 body = self.generate_body()
66 footer = self.generate_footer()
67
68 contents = []
69 if copyright:
70 contents.append(copyright)
71 if header:
72 contents.append(header)
73 if body:
74 contents.append(body)
75 if footer:
76 contents.append(footer)
77
78 return "\n\n".join(contents)
79
80 def generate_copyright(self):
81 return """/* THIS FILE IS GENERATED. DO NOT EDIT. */
82
83/*
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -060084 * Vulkan
Tobin Ehlis12076fc2014-10-22 09:06:33 -060085 *
86 * Copyright (C) 2014 LunarG, Inc.
87 *
88 * Permission is hereby granted, free of charge, to any person obtaining a
89 * copy of this software and associated documentation files (the "Software"),
90 * to deal in the Software without restriction, including without limitation
91 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
92 * and/or sell copies of the Software, and to permit persons to whom the
93 * Software is furnished to do so, subject to the following conditions:
94 *
95 * The above copyright notice and this permission notice shall be included
96 * in all copies or substantial portions of the Software.
97 *
98 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
99 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
100 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
101 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
102 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
103 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
104 * DEALINGS IN THE SOFTWARE.
105 */"""
106
107 def generate_header(self):
108 return "\n".join(["#include <" + h + ">" for h in self.headers])
109
110 def generate_body(self):
111 pass
112
113 def generate_footer(self):
114 pass
115
116 # Return set of printf '%' qualifier and input to that qualifier
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -0600117 def _get_printf_params(self, vk_type, name, output_param, cpp=False):
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600118 # TODO : Need ENUM and STRUCT checks here
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -0600119 if vk_helper.is_type(vk_type, 'enum'):#"_TYPE" in vk_type: # TODO : This should be generic ENUM check
Courtney Goeltzenleuchter95487bc2015-04-14 18:48:46 -0600120 return ("%s", "string_%s(%s)" % (vk_type.replace('const ', '').strip('*'), name))
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -0600121 if "char*" == vk_type:
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600122 return ("%s", name)
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -0600123 if "uint64" in vk_type:
124 if '*' in vk_type:
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600125 return ("%lu", "*%s" % name)
126 return ("%lu", name)
Tobin Ehlisa30e7e52015-07-06 14:02:36 -0600127 if vk_type.strip('*') in vulkan.object_non_dispatch_list:
128 if '*' in vk_type:
129 return ("%lu", "%s->handle" % name)
130 return ("%lu", "%s.handle" % name)
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -0600131 if "size" in vk_type:
132 if '*' in vk_type:
Chia-I Wu99ff89d2014-12-27 14:14:50 +0800133 return ("%zu", "*%s" % name)
134 return ("%zu", name)
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -0600135 if "float" in vk_type:
136 if '[' in vk_type: # handle array, current hard-coded to 4 (TODO: Make this dynamic)
Tobin Ehlis99f88672015-01-10 12:42:41 -0700137 if cpp:
138 return ("[%i, %i, %i, %i]", '"[" << %s[0] << "," << %s[1] << "," << %s[2] << "," << %s[3] << "]"' % (name, name, name, name))
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600139 return ("[%f, %f, %f, %f]", "%s[0], %s[1], %s[2], %s[3]" % (name, name, name, name))
140 return ("%f", name)
Courtney Goeltzenleuchtercd2a0992015-07-09 11:44:38 -0600141 if "bool" in vk_type.lower() or 'xcb_randr_crtc_t' in vk_type:
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600142 return ("%u", name)
Tobin Ehlisb870cbb2015-04-15 07:46:12 -0600143 if True in [t in vk_type.lower() for t in ["int", "flags", "mask", "xcb_window_t"]]:
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -0600144 if '[' in vk_type: # handle array, current hard-coded to 4 (TODO: Make this dynamic)
Tobin Ehlis99f88672015-01-10 12:42:41 -0700145 if cpp:
146 return ("[%i, %i, %i, %i]", "%s[0] << %s[1] << %s[2] << %s[3]" % (name, name, name, name))
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600147 return ("[%i, %i, %i, %i]", "%s[0], %s[1], %s[2], %s[3]" % (name, name, name, name))
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -0600148 if '*' in vk_type:
Tobin Ehlisd2b88e82015-02-04 15:15:11 -0700149 if 'pUserData' == name:
150 return ("%i", "((pUserData == 0) ? 0 : *(pUserData))")
Tobin Ehlisc62cb892015-04-17 13:26:33 -0600151 if 'const' in vk_type.lower():
152 return ("%p", "(void*)(%s)" % name)
Jon Ashburn52f79b52014-12-12 16:10:45 -0700153 return ("%i", "*(%s)" % name)
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600154 return ("%i", name)
Tobin Ehlis3a1cc8d2014-11-11 17:28:22 -0700155 # TODO : This is special-cased as there's only one "format" param currently and it's nice to expand it
Courtney Goeltzenleuchterfb4efc62015-04-10 08:34:15 -0600156 if "VkFormat" == vk_type:
Tobin Ehlis99f88672015-01-10 12:42:41 -0700157 if cpp:
158 return ("%p", "&%s" % name)
Courtney Goeltzenleuchterc729b3a2015-07-12 12:40:29 -0600159 return ("{%s.channelFormat = %%s, %s.numericFormat = %%s}" % (name, name), "string_VK_CHANNEL_FORMAT(%s.channelFormat), string_VK_FORMAT_NUM(%s.numericFormat)" % (name, name))
Tobin Ehlise7271572014-11-19 15:52:46 -0700160 if output_param:
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600161 return ("%p", "(void*)*%s" % name)
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -0600162 if vk_helper.is_type(vk_type, 'struct') and '*' not in vk_type:
Courtney Goeltzenleuchterd462fba2015-04-03 16:35:32 -0600163 return ("%p", "(void*)(&%s)" % name)
Jon Ashburn52f79b52014-12-12 16:10:45 -0700164 return ("%p", "(void*)(%s)" % name)
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600165
Courtney Goeltzenleuchterf579fa62015-06-10 17:39:03 -0600166 def _gen_create_msg_callback(self):
Tobin Ehlise8185062014-12-17 08:01:59 -0700167 r_body = []
Tobin Ehlis08fafd02015-06-12 12:49:01 -0600168 r_body.append('%s' % self.lineinfo.get())
Courtney Goeltzenleuchterf579fa62015-06-10 17:39:03 -0600169 r_body.append('VK_LAYER_EXPORT VkResult VKAPI vkDbgCreateMsgCallback(VkInstance instance, VkFlags msgFlags, const PFN_vkDbgMsgCallback pfnMsgCallback, void* pUserData, VkDbgMsgCallback* pMsgCallback)')
Tobin Ehlise8185062014-12-17 08:01:59 -0700170 r_body.append('{')
Mark Lobodzinskifae78852015-06-23 11:35:12 -0600171 # Switch to this code section for the new per-instance storage and debug callbacks
172 if self.layer_name == 'ObjectTracker':
173 r_body.append(' VkLayerInstanceDispatchTable *pInstanceTable = get_dispatch_table(%s_instance_table_map, instance);' % self.layer_name )
174 r_body.append(' VkResult result = pInstanceTable->DbgCreateMsgCallback(instance, msgFlags, pfnMsgCallback, pUserData, pMsgCallback);')
175 r_body.append(' if (VK_SUCCESS == result) {')
176 r_body.append(' layer_data *my_data = get_my_data_ptr(get_dispatch_key(instance), layer_data_map);')
177 r_body.append(' result = layer_create_msg_callback(my_data->report_data, msgFlags, pfnMsgCallback, pUserData, pMsgCallback);')
178 r_body.append(' }')
179 r_body.append(' return result;')
180 else:
181 # Old version of callbacks for compatibility
182 r_body.append(' return layer_create_msg_callback(instance, instance_dispatch_table(instance), msgFlags, pfnMsgCallback, pUserData, pMsgCallback);')
Tobin Ehlise8185062014-12-17 08:01:59 -0700183 r_body.append('}')
184 return "\n".join(r_body)
185
Courtney Goeltzenleuchterf579fa62015-06-10 17:39:03 -0600186 def _gen_destroy_msg_callback(self):
187 r_body = []
Tobin Ehlis08fafd02015-06-12 12:49:01 -0600188 r_body.append('%s' % self.lineinfo.get())
Courtney Goeltzenleuchterf579fa62015-06-10 17:39:03 -0600189 r_body.append('VK_LAYER_EXPORT VkResult VKAPI vkDbgDestroyMsgCallback(VkInstance instance, VkDbgMsgCallback msgCallback)')
190 r_body.append('{')
Mark Lobodzinskifae78852015-06-23 11:35:12 -0600191 # Switch to this code section for the new per-instance storage and debug callbacks
192 if self.layer_name == 'ObjectTracker':
193 r_body.append(' VkLayerInstanceDispatchTable *pInstanceTable = get_dispatch_table(%s_instance_table_map, instance);' % self.layer_name )
194 r_body.append(' VkResult result = pInstanceTable->DbgDestroyMsgCallback(instance, msgCallback);')
195 r_body.append(' layer_data *my_data = get_my_data_ptr(get_dispatch_key(instance), layer_data_map);')
196 r_body.append(' layer_destroy_msg_callback(my_data->report_data, msgCallback);')
197 r_body.append(' return result;')
198 else:
199 # Old version of callbacks for compatibility
200 r_body.append(' return layer_destroy_msg_callback(instance, instance_dispatch_table(instance), msgCallback);')
Courtney Goeltzenleuchterf579fa62015-06-10 17:39:03 -0600201 r_body.append('}')
202 return "\n".join(r_body)
Tobin Ehlise8185062014-12-17 08:01:59 -0700203
Tony Barbour59a47322015-06-24 16:06:58 -0600204 def _gen_layer_get_global_extension_props(self, layer="Generic"):
205 ggep_body = []
Courtney Goeltzenleuchter110fdf92015-06-29 15:39:26 -0600206 # generated layers do not provide any global extensions
207 ggep_body.append('%s' % self.lineinfo.get())
208
209 ggep_body.append('')
210 ggep_body.append('VK_LAYER_EXPORT VkResult VKAPI vkGetGlobalExtensionProperties(const char *pLayerName, uint32_t *pCount, VkExtensionProperties* pProperties)')
211 ggep_body.append('{')
212 ggep_body.append(' return util_GetExtensionProperties(0, NULL, pCount, pProperties);')
213 ggep_body.append('}')
214 return "\n".join(ggep_body)
215
216 def _gen_layer_get_global_layer_props(self, layer="Generic"):
217 ggep_body = []
218 if layer == 'Generic':
219 # Do nothing, extension definition part of generic.h
Tony Barbour59a47322015-06-24 16:06:58 -0600220 ggep_body.append('%s' % self.lineinfo.get())
Courtney Goeltzenleuchterd971b612015-06-17 20:51:59 -0600221 else:
Tony Barbour59a47322015-06-24 16:06:58 -0600222 ggep_body.append('%s' % self.lineinfo.get())
Courtney Goeltzenleuchter79a5a962015-07-07 17:51:45 -0600223 ggep_body.append('static const VkLayerProperties globalLayerProps[] = {')
Tony Barbour59a47322015-06-24 16:06:58 -0600224 ggep_body.append(' {')
Tony Barbour59a47322015-06-24 16:06:58 -0600225 ggep_body.append(' "%s",' % layer)
Courtney Goeltzenleuchter110fdf92015-06-29 15:39:26 -0600226 ggep_body.append(' VK_API_VERSION, // specVersion')
227 ggep_body.append(' VK_MAKE_VERSION(0, 1, 0), // implVersion')
Tony Barbour59a47322015-06-24 16:06:58 -0600228 ggep_body.append(' "layer: %s",' % layer)
229 ggep_body.append(' }')
230 ggep_body.append('};')
231 ggep_body.append('')
232 ggep_body.append('%s' % self.lineinfo.get())
Tony Barbour59a47322015-06-24 16:06:58 -0600233 ggep_body.append('')
Courtney Goeltzenleuchter110fdf92015-06-29 15:39:26 -0600234 ggep_body.append('VK_LAYER_EXPORT VkResult VKAPI vkGetGlobalLayerProperties(uint32_t *pCount, VkLayerProperties* pProperties)')
Tony Barbour59a47322015-06-24 16:06:58 -0600235 ggep_body.append('{')
Courtney Goeltzenleuchter79a5a962015-07-07 17:51:45 -0600236 ggep_body.append(' return util_GetLayerProperties(ARRAY_SIZE(globalLayerProps), globalLayerProps, pCount, pProperties);')
Tony Barbour59a47322015-06-24 16:06:58 -0600237 ggep_body.append('}')
238 return "\n".join(ggep_body)
239
Courtney Goeltzenleuchter3f9f7c42015-07-06 09:11:12 -0600240 def _gen_layer_get_physical_device_layer_props(self, layer="Generic"):
241 gpdlp_body = []
242 if layer == 'Generic':
243 # Do nothing, extension definition part of generic.h
244 gpdlp_body.append('%s' % self.lineinfo.get())
245 else:
246 gpdlp_body.append('%s' % self.lineinfo.get())
Courtney Goeltzenleuchter79a5a962015-07-07 17:51:45 -0600247 gpdlp_body.append('static const VkLayerProperties deviceLayerProps[] = {')
Courtney Goeltzenleuchter3f9f7c42015-07-06 09:11:12 -0600248 gpdlp_body.append(' {')
249 gpdlp_body.append(' "%s",' % layer)
250 gpdlp_body.append(' VK_API_VERSION,')
251 gpdlp_body.append(' VK_MAKE_VERSION(0, 1, 0),')
252 gpdlp_body.append(' "layer: %s",' % layer)
253 gpdlp_body.append(' }')
254 gpdlp_body.append('};')
255 gpdlp_body.append('VK_LAYER_EXPORT VkResult VKAPI vkGetPhysicalDeviceLayerProperties(VkPhysicalDevice physicalDevice, uint32_t *pCount, VkLayerProperties* pProperties)')
256 gpdlp_body.append('{')
Courtney Goeltzenleuchter79a5a962015-07-07 17:51:45 -0600257 gpdlp_body.append(' return util_GetLayerProperties(ARRAY_SIZE(deviceLayerProps), deviceLayerProps, pCount, pProperties);')
Courtney Goeltzenleuchter3f9f7c42015-07-06 09:11:12 -0600258 gpdlp_body.append('}')
259 gpdlp_body.append('')
260 return "\n".join(gpdlp_body)
261
Mike Stroyanbf237d72015-04-03 17:45:53 -0600262 def _generate_dispatch_entrypoints(self, qual=""):
Mike Stroyan938c2532015-04-03 13:58:35 -0600263 if qual:
264 qual += " "
265
Mike Stroyan938c2532015-04-03 13:58:35 -0600266 funcs = []
267 intercepted = []
268 for proto in self.protos:
Jon Ashburn8d1b0b52015-05-18 13:20:15 -0600269 if proto.name == "GetDeviceProcAddr" or proto.name == "GetInstanceProcAddr":
Jon Ashburn8fd08252015-05-28 16:25:02 -0600270 continue
Mike Stroyan70c05e82015-04-08 10:27:43 -0600271 else:
Mike Stroyan3e3a1eb2015-04-03 17:13:23 -0600272 intercept = self.generate_intercept(proto, qual)
Mike Stroyan938c2532015-04-03 13:58:35 -0600273 if intercept is None:
274 # fill in default intercept for certain entrypoints
Courtney Goeltzenleuchterf579fa62015-06-10 17:39:03 -0600275 if 'DbgCreateMsgCallback' == proto.name:
276 intercept = self._gen_layer_dbg_create_msg_callback()
277 elif 'DbgDestroyMsgCallback' == proto.name:
278 intercept = self._gen_layer_dbg_destroy_msg_callback()
279 elif 'CreateDevice' == proto.name:
280 funcs.append('/* CreateDevice HERE */')
Tony Barbour59a47322015-06-24 16:06:58 -0600281 elif 'GetGlobalExtensionProperties' == proto.name:
282 intercept = self._gen_layer_get_global_extension_props(self.layer_name)
Courtney Goeltzenleuchter110fdf92015-06-29 15:39:26 -0600283 elif 'GetGlobalLayerProperties' == proto.name:
284 intercept = self._gen_layer_get_global_layer_props(self.layer_name)
285 elif 'GetPhysicalDeviceLayerProperties' == proto.name:
286 intercept = self._gen_layer_get_physical_device_layer_props(self.layer_name)
Tony Barbour59a47322015-06-24 16:06:58 -0600287
Mike Stroyan938c2532015-04-03 13:58:35 -0600288 if intercept is not None:
289 funcs.append(intercept)
Jon Ashburn747f2b62015-06-18 15:02:58 -0600290 if not "WSI" in proto.name:
291 intercepted.append(proto)
Mike Stroyan938c2532015-04-03 13:58:35 -0600292
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -0600293 prefix="vk"
Mike Stroyan938c2532015-04-03 13:58:35 -0600294 lookups = []
295 for proto in intercepted:
Mike Stroyan938c2532015-04-03 13:58:35 -0600296 lookups.append("if (!strcmp(name, \"%s\"))" % proto.name)
297 lookups.append(" return (void*) %s%s;" %
298 (prefix, proto.name))
Mike Stroyan938c2532015-04-03 13:58:35 -0600299
300 # add customized layer_intercept_proc
301 body = []
Tobin Ehlis08fafd02015-06-12 12:49:01 -0600302 body.append('%s' % self.lineinfo.get())
Mike Stroyan938c2532015-04-03 13:58:35 -0600303 body.append("static inline void* layer_intercept_proc(const char *name)")
304 body.append("{")
305 body.append(generate_get_proc_addr_check("name"))
306 body.append("")
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -0600307 body.append(" name += 2;")
Mike Stroyan938c2532015-04-03 13:58:35 -0600308 body.append(" %s" % "\n ".join(lookups))
309 body.append("")
310 body.append(" return NULL;")
311 body.append("}")
Jon Ashburnf6b33db2015-05-05 14:22:52 -0600312 # add layer_intercept_instance_proc
313 lookups = []
314 for proto in self.protos:
Jon Ashburn95a77ba2015-05-15 15:09:35 -0600315 if not proto_is_global(proto):
Jon Ashburnf6b33db2015-05-05 14:22:52 -0600316 continue
317
318 if not proto in intercepted:
319 continue
Courtney Goeltzenleuchterca173b82015-06-25 18:01:43 -0600320 if proto.name == "CreateDevice":
321 continue
Jon Ashburnf6b33db2015-05-05 14:22:52 -0600322 lookups.append("if (!strcmp(name, \"%s\"))" % proto.name)
323 lookups.append(" return (void*) %s%s;" % (prefix, proto.name))
324
325 body.append("static inline void* layer_intercept_instance_proc(const char *name)")
326 body.append("{")
327 body.append(generate_get_proc_addr_check("name"))
328 body.append("")
329 body.append(" name += 2;")
330 body.append(" %s" % "\n ".join(lookups))
331 body.append("")
332 body.append(" return NULL;")
333 body.append("}")
334
Mike Stroyan938c2532015-04-03 13:58:35 -0600335 funcs.append("\n".join(body))
Mike Stroyan938c2532015-04-03 13:58:35 -0600336 return "\n\n".join(funcs)
337
Tobin Ehlisca915872014-11-18 11:28:33 -0700338 def _generate_extensions(self):
339 exts = []
Mark Lobodzinskifae78852015-06-23 11:35:12 -0600340 exts.append('%s' % self.lineinfo.get())
Courtney Goeltzenleuchterf579fa62015-06-10 17:39:03 -0600341 exts.append(self._gen_create_msg_callback())
342 exts.append(self._gen_destroy_msg_callback())
Tobin Ehlisb870cbb2015-04-15 07:46:12 -0600343 return "\n".join(exts)
344
Jon Ashburnf6b33db2015-05-05 14:22:52 -0600345 def _generate_layer_gpa_function(self, extensions=[], instance_extensions=[]):
Jon Ashburnbacb0f52015-04-06 10:58:22 -0600346 func_body = []
Mark Lobodzinskifae78852015-06-23 11:35:12 -0600347#
Courtney Goeltzenleuchter3f9f7c42015-07-06 09:11:12 -0600348# New style of GPA Functions for the new layer_data/layer_logging changes
Mark Lobodzinskifae78852015-06-23 11:35:12 -0600349#
350 if self.layer_name == 'ObjectTracker':
351 func_body.append("VK_LAYER_EXPORT void* VKAPI vkGetDeviceProcAddr(VkDevice device, const char* funcName)\n"
352 "{\n"
353 " void* addr;\n"
354 " if (device == VK_NULL_HANDLE) {\n"
355 " return NULL;\n"
356 " }\n"
357 " /* loader uses this to force layer initialization; device object is wrapped */\n"
358 " if (!strcmp(\"vkGetDeviceProcAddr\", funcName)) {\n"
359 " initDeviceTable(%s_device_table_map, (const VkBaseLayerObject *) device);\n"
360 " return (void *) vkGetDeviceProcAddr;\n"
361 " }\n\n"
362 " addr = layer_intercept_proc(funcName);\n"
363 " if (addr)\n"
364 " return addr;" % self.layer_name)
365 if 0 != len(extensions):
Courtney Goeltzenleuchter3f9f7c42015-07-06 09:11:12 -0600366 func_body.append('%s' % self.lineinfo.get())
367 func_body.append(' layer_data *my_device_data = get_my_data_ptr(get_dispatch_key(device), layer_data_map);')
Mark Lobodzinskifae78852015-06-23 11:35:12 -0600368 for (ext_enable, ext_list) in extensions:
369 extra_space = ""
370 if 0 != len(ext_enable):
Courtney Goeltzenleuchter3f9f7c42015-07-06 09:11:12 -0600371 func_body.append(' if (my_device_data->%s) {' % ext_enable)
Mark Lobodzinskifae78852015-06-23 11:35:12 -0600372 extra_space = " "
373 for ext_name in ext_list:
374 func_body.append(' %sif (!strcmp("%s", funcName))\n'
375 ' %sreturn reinterpret_cast<void*>(%s);' % (extra_space, ext_name, extra_space, ext_name))
376 if 0 != len(ext_enable):
377 func_body.append(' }\n')
378 func_body.append("\n if (get_dispatch_table(%s_device_table_map, device)->GetDeviceProcAddr == NULL)\n"
379 " return NULL;\n"
380 " return get_dispatch_table(%s_device_table_map, device)->GetDeviceProcAddr(device, funcName);\n"
381 "}\n" % (self.layer_name, self.layer_name))
382 func_body.append("VK_LAYER_EXPORT void* VKAPI vkGetInstanceProcAddr(VkInstance instance, const char* funcName)\n"
383 "{\n"
384 " void* addr;\n"
385 " if (instance == VK_NULL_HANDLE) {\n"
386 " return NULL;\n"
387 " }\n"
388 " /* loader uses this to force layer initialization; instance object is wrapped */\n"
389 " if (!strcmp(\"vkGetInstanceProcAddr\", funcName)) {\n"
390 " initInstanceTable(%s_instance_table_map, (const VkBaseLayerObject *) instance);\n"
391 " return (void *) vkGetInstanceProcAddr;\n"
392 " }\n\n"
393 " addr = layer_intercept_instance_proc(funcName);\n"
394 " if (addr) {\n"
395 " return addr;"
396 " }\n" % self.layer_name)
Jon Ashburnbacb0f52015-04-06 10:58:22 -0600397
Mark Lobodzinskifae78852015-06-23 11:35:12 -0600398 if 0 != len(instance_extensions):
399 for ext_name in instance_extensions:
400 func_body.append(" layer_data *my_data = get_my_data_ptr(get_dispatch_key(instance), layer_data_map);\n"
401 " addr = debug_report_get_instance_proc_addr(my_data->report_data, funcName);\n"
402 " if (addr) {\n"
403 " return addr;\n"
404 " }\n"
405 " if (get_dispatch_table(%s_instance_table_map, instance)->GetInstanceProcAddr == NULL) {\n"
406 " return NULL;\n"
407 " }\n"
408 " return get_dispatch_table(%s_instance_table_map, instance)->GetInstanceProcAddr(instance, funcName);\n"
409 "}\n" % (self.layer_name, self.layer_name))
410 return "\n".join(func_body)
411 else:
412#
413# TODO:: Old-style GPA Functions -- no local storage, no new logging mechanism. Left for compatibility.
414#
415 func_body.append('%s' % self.lineinfo.get())
416 func_body.append("VK_LAYER_EXPORT void* VKAPI vkGetDeviceProcAddr(VkDevice device, const char* funcName)\n"
417 "{\n"
418 " void* addr;\n"
419 " if (device == VK_NULL_HANDLE) {\n"
420 " return NULL;\n"
421 " }\n"
422 " loader_platform_thread_once(&initOnce, init%s);\n\n"
423 " /* loader uses this to force layer initialization; device object is wrapped */\n"
424 " if (!strcmp(\"vkGetDeviceProcAddr\", funcName)) {\n"
425 " initDeviceTable((const VkBaseLayerObject *) device);\n"
426 " return (void *) vkGetDeviceProcAddr;\n"
427 " }\n\n"
428 " addr = layer_intercept_proc(funcName);\n"
429 " if (addr)\n"
430 " return addr;" % self.layer_name)
431 func_body.append('')
432 func_body.append(' VkLayerDispatchTable *pDisp = device_dispatch_table(device);')
433 if 0 != len(extensions):
434 extra_space = ""
435 for (ext_enable, ext_list) in extensions:
436 if 0 != len(ext_enable):
437 func_body.append(' if (deviceExtMap.size() == 0 || deviceExtMap[pDisp].%s)' % ext_enable)
438 func_body.append(' {')
439 extra_space = " "
440 for ext_name in ext_list:
441 func_body.append(' %sif (!strcmp("%s", funcName))\n'
442 ' return reinterpret_cast<void*>(%s);' % (extra_space, ext_name, ext_name))
443 if 0 != len(ext_enable):
444 func_body.append(' }')
445 func_body.append('%s' % self.lineinfo.get())
446 func_body.append(" {\n"
447 " if (pDisp->GetDeviceProcAddr == NULL)\n"
448 " return NULL;\n"
449 " return pDisp->GetDeviceProcAddr(device, funcName);\n"
450 " }\n"
451 "}\n")
452 func_body.append("VK_LAYER_EXPORT void* VKAPI vkGetInstanceProcAddr(VkInstance instance, const char* funcName)\n"
453 "{\n"
454 " void* addr;\n"
455 " if (instance == VK_NULL_HANDLE) {\n"
456 " return NULL;\n"
457 " }\n"
458 " loader_platform_thread_once(&initOnce, init%s);\n\n"
459 " /* loader uses this to force layer initialization; instance object is wrapped */\n"
460 " if (!strcmp(\"vkGetInstanceProcAddr\", funcName)) {\n"
461 " initInstanceTable((const VkBaseLayerObject *) instance);\n"
462 " return (void *) vkGetInstanceProcAddr;\n"
463 " }\n\n"
464 " addr = layer_intercept_instance_proc(funcName);\n"
465 " if (addr)\n"
466 " return addr;" % self.layer_name)
467 if 0 != len(instance_extensions):
468 for ext_name in instance_extensions:
469 func_body.append(" {\n"
470 " void *fptr;\n"
471 " fptr = %s(funcName);\n"
472 " if (fptr) return fptr;\n"
473 " }\n" % ext_name)
474 func_body.append(" VkLayerInstanceDispatchTable* pTable = instance_dispatch_table(instance);\n"
475 " if (pTable->GetInstanceProcAddr == NULL)\n"
476 " return NULL;\n"
477 " return pTable->GetInstanceProcAddr(instance, funcName);\n"
478 "}\n")
479 return "\n".join(func_body)
Jon Ashburnf6b33db2015-05-05 14:22:52 -0600480
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600481
Mike Stroyaned238bb2015-05-15 08:50:57 -0600482 def _generate_layer_initialization(self, init_opts=False, prefix='vk', lockname=None, condname=None):
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -0600483 func_body = ["#include \"vk_dispatch_table_helper.h\""]
Tobin Ehlis08fafd02015-06-12 12:49:01 -0600484 func_body.append('%s' % self.lineinfo.get())
Jon Ashburnd6badbc2015-02-16 08:26:50 -0700485 func_body.append('static void init%s(void)\n'
Mike Stroyan3e3a1eb2015-04-03 17:13:23 -0600486 '{\n' % self.layer_name)
Jon Ashburnd6badbc2015-02-16 08:26:50 -0700487 if init_opts:
Tobin Ehlis08fafd02015-06-12 12:49:01 -0600488 func_body.append('%s' % self.lineinfo.get())
Jon Ashburnd6badbc2015-02-16 08:26:50 -0700489 func_body.append(' const char *strOpt;')
Mike Stroyan3e3a1eb2015-04-03 17:13:23 -0600490 func_body.append(' // initialize %s options' % self.layer_name)
Courtney Goeltzenleuchterf579fa62015-06-10 17:39:03 -0600491 func_body.append(' getLayerOptionEnum("%sReportLevel", (uint32_t *) &g_reportFlags);' % self.layer_name)
Mike Stroyan3e3a1eb2015-04-03 17:13:23 -0600492 func_body.append(' g_actionIsDefault = getLayerOptionEnum("%sDebugAction", (uint32_t *) &g_debugAction);' % self.layer_name)
Jon Ashburnd6badbc2015-02-16 08:26:50 -0700493 func_body.append('')
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -0600494 func_body.append(' if (g_debugAction & VK_DBG_LAYER_ACTION_LOG_MSG)')
Jon Ashburnd6badbc2015-02-16 08:26:50 -0700495 func_body.append(' {')
Mike Stroyan3e3a1eb2015-04-03 17:13:23 -0600496 func_body.append(' strOpt = getLayerOption("%sLogFilename");' % self.layer_name)
Jon Ashburnd6badbc2015-02-16 08:26:50 -0700497 func_body.append(' if (strOpt)')
498 func_body.append(' {')
499 func_body.append(' g_logFile = fopen(strOpt, "w");')
500 func_body.append(' }')
501 func_body.append(' if (g_logFile == NULL)')
502 func_body.append(' g_logFile = stdout;')
503 func_body.append(' }')
504 func_body.append('')
Tobin Ehlis36f1b462015-02-23 14:09:16 -0700505 if lockname is not None:
Tobin Ehlis08fafd02015-06-12 12:49:01 -0600506 func_body.append('%s' % self.lineinfo.get())
Tobin Ehlis36f1b462015-02-23 14:09:16 -0700507 func_body.append(" if (!%sLockInitialized)" % lockname)
508 func_body.append(" {")
509 func_body.append(" // TODO/TBD: Need to delete this mutex sometime. How???")
510 func_body.append(" loader_platform_thread_create_mutex(&%sLock);" % lockname)
Mike Stroyaned238bb2015-05-15 08:50:57 -0600511 if condname is not None:
512 func_body.append(" loader_platform_thread_init_cond(&%sCond);" % condname)
Tobin Ehlis36f1b462015-02-23 14:09:16 -0700513 func_body.append(" %sLockInitialized = 1;" % lockname)
514 func_body.append(" }")
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600515 func_body.append("}\n")
Jon Ashburn8c5cbcf2015-05-07 10:27:37 -0600516 func_body.append('')
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600517 return "\n".join(func_body)
518
519class LayerFuncsSubcommand(Subcommand):
520 def generate_header(self):
Tobin Ehlis0c6f9ee2015-07-03 09:42:57 -0600521 return '#include <vk_layer.h>\n#include "loader.h"'
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600522
523 def generate_body(self):
Mike Stroyan3e3a1eb2015-04-03 17:13:23 -0600524 return self._generate_dispatch_entrypoints("static")
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600525
526class LayerDispatchSubcommand(Subcommand):
527 def generate_header(self):
528 return '#include "layer_wrappers.h"'
529
530 def generate_body(self):
Jon Ashburnd6badbc2015-02-16 08:26:50 -0700531 return self._generate_layer_initialization()
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600532
533class GenericLayerSubcommand(Subcommand):
534 def generate_header(self):
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500535 gen_header = []
Courtney Goeltzenleuchter79a5a962015-07-07 17:51:45 -0600536 gen_header.append('%s' % self.lineinfo.get())
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500537 gen_header.append('#include <stdio.h>')
538 gen_header.append('#include <stdlib.h>')
539 gen_header.append('#include <string.h>')
540 gen_header.append('#include <unordered_map>')
Tobin Ehlisb835d1b2015-07-03 10:34:49 -0600541 gen_header.append('#include "vk_loader_platform.h"')
Tobin Ehlis0c6f9ee2015-07-03 09:42:57 -0600542 gen_header.append('#include "vk_layer.h"')
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500543 gen_header.append('//The following is #included again to catch certain OS-specific functions being used:')
544 gen_header.append('')
Tobin Ehlisb835d1b2015-07-03 10:34:49 -0600545 gen_header.append('#include "vk_loader_platform.h"')
Tobin Ehlisa0cb02e2015-07-03 10:15:26 -0600546 gen_header.append('#include "vk_layer_config.h"')
547 gen_header.append('#include "vk_layer_msg.h"')
548 gen_header.append('#include "vk_layer_table.h"')
Courtney Goeltzenleuchter3f9f7c42015-07-06 09:11:12 -0600549 gen_header.append('#include "vk_layer_extension_utils.h"')
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500550 gen_header.append('')
Courtney Goeltzenleuchter79a5a962015-07-07 17:51:45 -0600551 gen_header.append('#include "generic.h"')
Courtney Goeltzenleuchter110fdf92015-06-29 15:39:26 -0600552 gen_header.append('')
Tobin Ehlisa30e7e52015-07-06 14:02:36 -0600553 gen_header.append('%s' % self.lineinfo.get())
554 gen_header.append('#define LAYER_EXT_ARRAY_SIZE 1')
555 gen_header.append('#define LAYER_DEV_EXT_ARRAY_SIZE 1')
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500556 gen_header.append('static LOADER_PLATFORM_THREAD_ONCE_DECLARATION(initOnce);')
Jon Ashburn747f2b62015-06-18 15:02:58 -0600557 gen_header.append('struct devExts {')
558 gen_header.append(' bool wsi_lunarg_enabled;')
559 gen_header.append('};')
560 gen_header.append('static std::unordered_map<void *, struct devExts> deviceExtMap;')
561 gen_header.append('')
562 gen_header.append('static void createDeviceRegisterExtensions(const VkDeviceCreateInfo* pCreateInfo, VkDevice device)')
563 gen_header.append('{')
564 gen_header.append(' uint32_t i, ext_idx;')
565 gen_header.append(' VkLayerDispatchTable *pDisp = device_dispatch_table(device);')
566 gen_header.append(' deviceExtMap[pDisp].wsi_lunarg_enabled = false;')
567 gen_header.append(' for (i = 0; i < pCreateInfo->extensionCount; i++) {')
Courtney Goeltzenleuchter110fdf92015-06-29 15:39:26 -0600568 gen_header.append(' if (strcmp(pCreateInfo->ppEnabledExtensionNames[i], VK_WSI_LUNARG_EXTENSION_NAME) == 0)')
Jon Ashburn747f2b62015-06-18 15:02:58 -0600569 gen_header.append(' deviceExtMap[pDisp].wsi_lunarg_enabled = true;')
570 gen_header.append('')
571 gen_header.append(' }')
572 gen_header.append('}')
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500573 gen_header.append('')
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500574 return "\n".join(gen_header)
Tobin Ehlis7a314ea2015-07-07 11:02:44 -0600575
Mike Stroyan3e3a1eb2015-04-03 17:13:23 -0600576 def generate_intercept(self, proto, qual):
Courtney Goeltzenleuchter110fdf92015-06-29 15:39:26 -0600577 if proto.name in [ 'GetGlobalLayerProperties', 'GetGlobalExtensionProperties', 'GetPhysicalDeviceLayerProperties', 'GetPhysicalDeviceExtensionProperties' ]:
Mike Stroyan00087e62015-04-03 14:39:16 -0600578 # use default version
579 return None
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -0600580 decl = proto.c_func(prefix="vk", attr="VKAPI")
Mike Stroyan938c2532015-04-03 13:58:35 -0600581 ret_val = ''
582 stmt = ''
583 funcs = []
Jon Ashburn71836d92015-05-12 17:23:55 -0600584 table = ''
Jon Ashburn95a77ba2015-05-15 15:09:35 -0600585 if proto_is_global(proto):
Jon Ashburn71836d92015-05-12 17:23:55 -0600586 table = 'Instance'
Mike Stroyan938c2532015-04-03 13:58:35 -0600587 if proto.ret != "void":
Courtney Goeltzenleuchter0de9e212015-06-26 15:07:53 -0600588 funcs.append('%s' % self.lineinfo.get())
Jon Ashburnb0fbe912015-05-06 10:15:07 -0600589 ret_val = "%s result = " % proto.ret
Mike Stroyan938c2532015-04-03 13:58:35 -0600590 stmt = " return result;\n"
Jon Ashburn8ced4212015-05-22 12:01:50 -0600591 if proto.name == "CreateDevice":
Courtney Goeltzenleuchter0de9e212015-06-26 15:07:53 -0600592 funcs.append('%s' % self.lineinfo.get())
Mike Stroyan938c2532015-04-03 13:58:35 -0600593 funcs.append('%s%s\n'
594 '{\n'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500595 ' char str[1024];\n'
596 ' sprintf(str, "At start of layered %s\\n");\n'
Tony Barboura05dbaa2015-07-09 17:31:46 -0600597 ' layerCbMsg(VK_DBG_REPORT_INFO_BIT,VK_OBJECT_TYPE_PHYSICAL_DEVICE, (uint64_t)physicalDevice, 0, 0, (char *) "GENERIC", (char *) str);\n'
Courtney Goeltzenleuchterca173b82015-06-25 18:01:43 -0600598 ' %sdevice_dispatch_table(*pDevice)->%s;\n'
Courtney Goeltzenleuchterf579fa62015-06-10 17:39:03 -0600599 ' if (result == VK_SUCCESS) {\n'
Courtney Goeltzenleuchter110fdf92015-06-29 15:39:26 -0600600 ' enable_debug_report(pCreateInfo->extensionCount, pCreateInfo->ppEnabledExtensionNames);\n'
Jon Ashburn747f2b62015-06-18 15:02:58 -0600601 ' createDeviceRegisterExtensions(pCreateInfo, *pDevice);\n'
Mike Stroyan938c2532015-04-03 13:58:35 -0600602 ' }\n'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500603 ' sprintf(str, "Completed layered %s\\n");\n'
Tony Barboura05dbaa2015-07-09 17:31:46 -0600604 ' layerCbMsg(VK_DBG_REPORT_INFO_BIT, VK_OBJECT_TYPE_PHYSICAL_DEVICE, (uint64_t)physicalDevice, 0, 0, (char *) "GENERIC", (char *) str);\n'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500605 ' fflush(stdout);\n'
606 ' %s'
607 '}' % (qual, decl, proto.name, ret_val, proto.c_call(), proto.name, stmt))
608 elif proto.name == "DestroyDevice":
Courtney Goeltzenleuchter0de9e212015-06-26 15:07:53 -0600609 funcs.append('%s' % self.lineinfo.get())
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500610 funcs.append('%s%s\n'
611 '{\n'
Courtney Goeltzenleuchter0daf2282015-06-13 21:22:12 -0600612 ' dispatch_key key = get_dispatch_key(device);\n'
Jon Ashburn747f2b62015-06-18 15:02:58 -0600613 ' VkLayerDispatchTable *pDisp = device_dispatch_table(device);\n'
614 ' VkResult res = pDisp->DestroyDevice(device);\n'
615 ' deviceExtMap.erase(pDisp);\n'
Courtney Goeltzenleuchter0daf2282015-06-13 21:22:12 -0600616 ' destroy_device_dispatch_table(key);\n'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500617 ' return res;\n'
618 '}\n' % (qual, decl))
619 elif proto.name == "DestroyInstance":
Courtney Goeltzenleuchter0de9e212015-06-26 15:07:53 -0600620 funcs.append('%s' % self.lineinfo.get())
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500621 funcs.append('%s%s\n'
622 '{\n'
Courtney Goeltzenleuchter0daf2282015-06-13 21:22:12 -0600623 ' dispatch_key key = get_dispatch_key(instance);\n'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500624 ' VkResult res = instance_dispatch_table(instance)->DestroyInstance(instance);\n'
Courtney Goeltzenleuchter0daf2282015-06-13 21:22:12 -0600625 ' destroy_instance_dispatch_table(key);\n'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500626 ' return res;\n'
627 '}\n' % (qual, decl))
Jon Ashburn4d9f4652015-04-08 21:33:34 -0600628 else:
Courtney Goeltzenleuchter0de9e212015-06-26 15:07:53 -0600629 funcs.append('%s' % self.lineinfo.get())
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500630 # CreateInstance needs to use the second parm instead of the first to set the correct dispatch table
631 dispatch_param = proto.params[0].name
632 # Must use 'instance' table for these APIs, 'device' table otherwise
633 table_type = ""
634 if proto_is_global(proto):
635 table_type = "instance"
636 else:
637 table_type = "device"
638 if 'CreateInstance' in proto.name:
639 dispatch_param = '*' + proto.params[1].name
Mike Stroyan938c2532015-04-03 13:58:35 -0600640 funcs.append('%s%s\n'
641 '{\n'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500642 ' %s%s_dispatch_table(%s)->%s;\n'
Mike Stroyan938c2532015-04-03 13:58:35 -0600643 '%s'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -0500644 '}' % (qual, decl, ret_val, table_type, dispatch_param, proto.c_call(), stmt))
Mike Stroyan938c2532015-04-03 13:58:35 -0600645 return "\n\n".join(funcs)
646
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600647 def generate_body(self):
Mike Stroyan3e3a1eb2015-04-03 17:13:23 -0600648 self.layer_name = "Generic"
Jon Ashburn747f2b62015-06-18 15:02:58 -0600649 extensions=[('wsi_lunarg_enabled',
650 ['vkCreateSwapChainWSI', 'vkDestroySwapChainWSI',
651 'vkGetSwapChainInfoWSI', 'vkQueuePresentWSI'])]
Mike Stroyan3e3a1eb2015-04-03 17:13:23 -0600652 body = [self._generate_layer_initialization(True),
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -0600653 self._generate_dispatch_entrypoints("VK_LAYER_EXPORT"),
Courtney Goeltzenleuchterf579fa62015-06-10 17:39:03 -0600654 self._gen_create_msg_callback(),
655 self._gen_destroy_msg_callback(),
Jon Ashburn747f2b62015-06-18 15:02:58 -0600656 self._generate_layer_gpa_function(extensions)]
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600657
658 return "\n\n".join(body)
659
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600660class APIDumpSubcommand(Subcommand):
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600661 def generate_header(self):
Tobin Ehlisd009bae2014-11-24 15:46:55 -0700662 header_txt = []
Tobin Ehlis08fafd02015-06-12 12:49:01 -0600663 header_txt.append('%s' % self.lineinfo.get())
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600664 header_txt.append('#include <fstream>')
665 header_txt.append('#include <iostream>')
666 header_txt.append('#include <string>')
Jon Ashburn747f2b62015-06-18 15:02:58 -0600667 header_txt.append('#include <string.h>')
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600668 header_txt.append('')
669 header_txt.append('static std::ofstream fileStream;')
670 header_txt.append('static std::string fileName = "vk_apidump.txt";')
671 header_txt.append('std::ostream* outputStream = NULL;')
672 header_txt.append('void ConfigureOutputStream(bool writeToFile, bool flushAfterWrite)')
673 header_txt.append('{')
674 header_txt.append(' if(writeToFile)')
675 header_txt.append(' {')
676 header_txt.append(' fileStream.open(fileName);')
677 header_txt.append(' outputStream = &fileStream;')
678 header_txt.append(' }')
679 header_txt.append(' else')
680 header_txt.append(' {')
681 header_txt.append(' outputStream = &std::cout;')
682 header_txt.append(' }')
683 header_txt.append('')
684 header_txt.append(' if(flushAfterWrite)')
685 header_txt.append(' {')
686 header_txt.append(' outputStream->sync_with_stdio(true);')
687 header_txt.append(' }')
688 header_txt.append(' else')
689 header_txt.append(' {')
690 header_txt.append(' outputStream->sync_with_stdio(false);')
691 header_txt.append(' }')
692 header_txt.append('}')
693 header_txt.append('')
Tobin Ehlis08fafd02015-06-12 12:49:01 -0600694 header_txt.append('%s' % self.lineinfo.get())
Tobin Ehlisb835d1b2015-07-03 10:34:49 -0600695 header_txt.append('#include "vk_loader_platform.h"')
Tobin Ehlis0c6f9ee2015-07-03 09:42:57 -0600696 header_txt.append('#include "vk_layer.h"')
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600697 header_txt.append('#include "vk_struct_string_helper_cpp.h"')
Tobin Ehlisa0cb02e2015-07-03 10:15:26 -0600698 header_txt.append('#include "vk_layer_table.h"')
Courtney Goeltzenleuchter110fdf92015-06-29 15:39:26 -0600699 header_txt.append('#include "vk_layer_extension_utils.h"')
Mark Lobodzinski4c7ac0a2015-05-26 09:29:09 -0500700 header_txt.append('#include <unordered_map>')
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600701 header_txt.append('')
Ian Elliott655cad72015-02-12 17:08:34 -0700702 header_txt.append('// The following is #included again to catch certain OS-specific functions being used:')
Tobin Ehlisb835d1b2015-07-03 10:34:49 -0600703 header_txt.append('#include "vk_loader_platform.h"')
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600704 header_txt.append('')
Jon Ashburnbacb0f52015-04-06 10:58:22 -0600705 header_txt.append('static VkBaseLayerObject *pCurObj;')
Tobin Ehlisa2b199c2015-04-27 17:30:04 -0600706 header_txt.append('static bool g_APIDumpDetailed = true;')
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600707 header_txt.append('')
Jon Ashburn8c5cbcf2015-05-07 10:27:37 -0600708 header_txt.append('static LOADER_PLATFORM_THREAD_ONCE_DECLARATION(initOnce);')
Mark Lobodzinski4c7ac0a2015-05-26 09:29:09 -0500709 header_txt.append('')
Ian Elliott2d4ab1e2015-01-13 17:52:38 -0700710 header_txt.append('static int printLockInitialized = 0;')
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600711 header_txt.append('static loader_platform_thread_mutex printLock;')
712 header_txt.append('')
Tobin Ehlis08fafd02015-06-12 12:49:01 -0600713 header_txt.append('%s' % self.lineinfo.get())
Tobin Ehlisa30e7e52015-07-06 14:02:36 -0600714 header_txt.append('#define LAYER_EXT_ARRAY_SIZE 1')
715 header_txt.append('#define LAYER_DEV_EXT_ARRAY_SIZE 1')
Tobin Ehlisd009bae2014-11-24 15:46:55 -0700716 header_txt.append('#define MAX_TID 513')
Ian Elliott2d4ab1e2015-01-13 17:52:38 -0700717 header_txt.append('static loader_platform_thread_id tidMapping[MAX_TID] = {0};')
Tobin Ehlisd009bae2014-11-24 15:46:55 -0700718 header_txt.append('static uint32_t maxTID = 0;')
719 header_txt.append('// Map actual TID to an index value and return that index')
720 header_txt.append('// This keeps TIDs in range from 0-MAX_TID and simplifies compares between runs')
721 header_txt.append('static uint32_t getTIDIndex() {')
Ian Elliott2d4ab1e2015-01-13 17:52:38 -0700722 header_txt.append(' loader_platform_thread_id tid = loader_platform_get_thread_id();')
Tobin Ehlisd009bae2014-11-24 15:46:55 -0700723 header_txt.append(' for (uint32_t i = 0; i < maxTID; i++) {')
724 header_txt.append(' if (tid == tidMapping[i])')
725 header_txt.append(' return i;')
726 header_txt.append(' }')
727 header_txt.append(" // Don't yet have mapping, set it and return newly set index")
Ian Elliott2d4ab1e2015-01-13 17:52:38 -0700728 header_txt.append(' uint32_t retVal = (uint32_t) maxTID;')
Tobin Ehlisd009bae2014-11-24 15:46:55 -0700729 header_txt.append(' tidMapping[maxTID++] = tid;')
730 header_txt.append(' assert(maxTID < MAX_TID);')
731 header_txt.append(' return retVal;')
732 header_txt.append('}')
Jon Ashburn747f2b62015-06-18 15:02:58 -0600733 header_txt.append('struct devExts {')
734 header_txt.append(' bool wsi_lunarg_enabled;')
735 header_txt.append('};')
736 header_txt.append('')
737 header_txt.append('static std::unordered_map<void *, struct devExts> deviceExtMap;')
738 header_txt.append('')
739 header_txt.append('static void createDeviceRegisterExtensions(const VkDeviceCreateInfo* pCreateInfo, VkDevice device)')
740 header_txt.append('{')
741 header_txt.append(' uint32_t i, ext_idx;')
742 header_txt.append(' VkLayerDispatchTable *pDisp = device_dispatch_table(device);')
743 header_txt.append(' deviceExtMap[pDisp].wsi_lunarg_enabled = false;')
744 header_txt.append(' for (i = 0; i < pCreateInfo->extensionCount; i++) {')
Courtney Goeltzenleuchter110fdf92015-06-29 15:39:26 -0600745 header_txt.append(' if (strcmp(pCreateInfo->ppEnabledExtensionNames[i], VK_WSI_LUNARG_EXTENSION_NAME) == 0)')
Jon Ashburn747f2b62015-06-18 15:02:58 -0600746 header_txt.append(' deviceExtMap[pDisp].wsi_lunarg_enabled = true;')
747 header_txt.append('')
748 header_txt.append(' }')
749 header_txt.append('}')
Mark Lobodzinski4c7ac0a2015-05-26 09:29:09 -0500750 header_txt.append('')
Tobin Ehlisd009bae2014-11-24 15:46:55 -0700751 return "\n".join(header_txt)
Tobin Ehlis12076fc2014-10-22 09:06:33 -0600752
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600753 def generate_init(self):
754 func_body = []
Tobin Ehlis08fafd02015-06-12 12:49:01 -0600755 func_body.append('%s' % self.lineinfo.get())
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600756 func_body.append('#include "vk_dispatch_table_helper.h"')
Tobin Ehlisa0cb02e2015-07-03 10:15:26 -0600757 func_body.append('#include "vk_layer_config.h"')
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600758 func_body.append('')
759 func_body.append('static void init%s(void)' % self.layer_name)
760 func_body.append('{')
761 func_body.append(' using namespace StreamControl;')
762 func_body.append('')
Tobin Ehlisa2b199c2015-04-27 17:30:04 -0600763 func_body.append(' char const*const detailedStr = getLayerOption("APIDumpDetailed");')
764 func_body.append(' if(detailedStr != NULL)')
765 func_body.append(' {')
766 func_body.append(' if(strcmp(detailedStr, "TRUE") == 0)')
767 func_body.append(' {')
768 func_body.append(' g_APIDumpDetailed = true;')
769 func_body.append(' }')
770 func_body.append(' else if(strcmp(detailedStr, "FALSE") == 0)')
771 func_body.append(' {')
772 func_body.append(' g_APIDumpDetailed = false;')
773 func_body.append(' }')
774 func_body.append(' }')
775 func_body.append('')
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600776 func_body.append(' char const*const writeToFileStr = getLayerOption("APIDumpFile");')
777 func_body.append(' bool writeToFile = false;')
778 func_body.append(' if(writeToFileStr != NULL)')
779 func_body.append(' {')
780 func_body.append(' if(strcmp(writeToFileStr, "TRUE") == 0)')
781 func_body.append(' {')
782 func_body.append(' writeToFile = true;')
783 func_body.append(' }')
784 func_body.append(' else if(strcmp(writeToFileStr, "FALSE") == 0)')
785 func_body.append(' {')
786 func_body.append(' writeToFile = false;')
787 func_body.append(' }')
788 func_body.append(' }')
789 func_body.append('')
Tobin Ehlis08fafd02015-06-12 12:49:01 -0600790 func_body.append('%s' % self.lineinfo.get())
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600791 func_body.append(' char const*const noAddrStr = getLayerOption("APIDumpNoAddr");')
792 func_body.append(' if(noAddrStr != NULL)')
793 func_body.append(' {')
794 func_body.append(' if(strcmp(noAddrStr, "FALSE") == 0)')
795 func_body.append(' {')
796 func_body.append(' StreamControl::writeAddress = true;')
797 func_body.append(' }')
798 func_body.append(' else if(strcmp(noAddrStr, "TRUE") == 0)')
799 func_body.append(' {')
800 func_body.append(' StreamControl::writeAddress = false;')
801 func_body.append(' }')
802 func_body.append(' }')
803 func_body.append('')
804 func_body.append(' char const*const flushAfterWriteStr = getLayerOption("APIDumpFlush");')
805 func_body.append(' bool flushAfterWrite = false;')
806 func_body.append(' if(flushAfterWriteStr != NULL)')
807 func_body.append(' {')
808 func_body.append(' if(strcmp(flushAfterWriteStr, "TRUE") == 0)')
809 func_body.append(' {')
810 func_body.append(' flushAfterWrite = true;')
811 func_body.append(' }')
812 func_body.append(' else if(strcmp(flushAfterWriteStr, "FALSE") == 0)')
813 func_body.append(' {')
814 func_body.append(' flushAfterWrite = false;')
815 func_body.append(' }')
816 func_body.append(' }')
817 func_body.append('')
Tobin Ehlis08fafd02015-06-12 12:49:01 -0600818 func_body.append('%s' % self.lineinfo.get())
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600819 func_body.append(' ConfigureOutputStream(writeToFile, flushAfterWrite);')
820 func_body.append('')
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600821 func_body.append(' if (!printLockInitialized)')
822 func_body.append(' {')
823 func_body.append(' // TODO/TBD: Need to delete this mutex sometime. How???')
824 func_body.append(' loader_platform_thread_create_mutex(&printLock);')
825 func_body.append(' printLockInitialized = 1;')
826 func_body.append(' }')
827 func_body.append('}')
828 func_body.append('')
829 return "\n".join(func_body)
Tobin Ehlis99f88672015-01-10 12:42:41 -0700830
Mike Stroyanbf237d72015-04-03 17:45:53 -0600831 def generate_intercept(self, proto, qual):
Courtney Goeltzenleuchter110fdf92015-06-29 15:39:26 -0600832 if proto.name in [ 'GetGlobalLayerProperties','GetGlobalExtensionProperties','GetPhysicalDeviceLayerProperties','GetPhysicalDeviceExtensionProperties']:
Jon Ashburn9fd4cc42015-04-10 14:33:07 -0600833 return None
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -0600834 decl = proto.c_func(prefix="vk", attr="VKAPI")
Mike Stroyanbf237d72015-04-03 17:45:53 -0600835 ret_val = ''
836 stmt = ''
837 funcs = []
838 sp_param_dict = {} # Store 'index' for struct param to print, or an name of binding "Count" param for array to print
839 create_params = 0 # Num of params at end of function that are created and returned as output values
Chia-I Wuf8693382015-04-16 22:02:10 +0800840 if 'AllocDescriptorSets' in proto.name:
Mike Stroyanbf237d72015-04-03 17:45:53 -0600841 create_params = -2
842 elif 'Create' in proto.name or 'Alloc' in proto.name or 'MapMemory' in proto.name:
843 create_params = -1
844 if proto.ret != "void":
Jon Ashburnb0fbe912015-05-06 10:15:07 -0600845 ret_val = "%s result = " % proto.ret
Mike Stroyanbf237d72015-04-03 17:45:53 -0600846 stmt = " return result;\n"
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600847 f_open = 'loader_platform_thread_lock_mutex(&printLock);\n '
Tobin Ehlis08fafd02015-06-12 12:49:01 -0600848 log_func = '%s\n' % self.lineinfo.get()
849 log_func += ' if (StreamControl::writeAddress == true) {'
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600850 log_func += '\n (*outputStream) << "t{" << getTIDIndex() << "} vk%s(' % proto.name
851 log_func_no_addr = '\n (*outputStream) << "t{" << getTIDIndex() << "} vk%s(' % proto.name
852 f_close = '\n loader_platform_thread_unlock_mutex(&printLock);'
Mike Stroyanbf237d72015-04-03 17:45:53 -0600853 pindex = 0
854 prev_count_name = ''
855 for p in proto.params:
856 cp = False
857 if 0 != create_params:
858 # If this is any of the N last params of the func, treat as output
859 for y in range(-1, create_params-1, -1):
860 if p.name == proto.params[y].name:
861 cp = True
862 (pft, pfi) = self._get_printf_params(p.ty, p.name, cp, cpp=True)
Mike Stroyanbf237d72015-04-03 17:45:53 -0600863 log_func += '%s = " << %s << ", ' % (p.name, pfi)
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600864 if "%p" == pft:
865 log_func_no_addr += '%s = address, ' % (p.name)
866 else:
867 log_func_no_addr += '%s = " << %s << ", ' % (p.name, pfi)
Tobin Ehlis08541742015-04-16 15:56:11 -0600868 if prev_count_name != '' and (prev_count_name.replace('Count', '')[1:] in p.name):
Mike Stroyanbf237d72015-04-03 17:45:53 -0600869 sp_param_dict[pindex] = prev_count_name
Tobin Ehlis08541742015-04-16 15:56:11 -0600870 prev_count_name = ''
Mike Stroyanbf237d72015-04-03 17:45:53 -0600871 elif 'pDescriptorSets' == p.name and proto.params[-1].name == 'pCount':
872 sp_param_dict[pindex] = '*pCount'
Chia-I Wuf8693382015-04-16 22:02:10 +0800873 elif vk_helper.is_type(p.ty.strip('*').replace('const ', ''), 'struct'):
Mike Stroyanbf237d72015-04-03 17:45:53 -0600874 sp_param_dict[pindex] = 'index'
Mike Stroyanbf237d72015-04-03 17:45:53 -0600875 if p.name.endswith('Count'):
876 if '*' in p.ty:
877 prev_count_name = "*%s" % p.name
878 else:
879 prev_count_name = p.name
Jon Ashburn9fd4cc42015-04-10 14:33:07 -0600880 pindex += 1
Mike Stroyanbf237d72015-04-03 17:45:53 -0600881 log_func = log_func.strip(', ')
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600882 log_func_no_addr = log_func_no_addr.strip(', ')
Jon Ashburnb0fbe912015-05-06 10:15:07 -0600883 if proto.ret == "VkResult":
Courtney Goeltzenleuchterfb4efc62015-04-10 08:34:15 -0600884 log_func += ') = " << string_VkResult((VkResult)result) << endl'
885 log_func_no_addr += ') = " << string_VkResult((VkResult)result) << endl'
Jon Ashburnb0fbe912015-05-06 10:15:07 -0600886 elif proto.ret == "void*":
887 log_func += ') = " << result << endl'
888 log_func_no_addr += ') = " << result << endl'
Mike Stroyanbf237d72015-04-03 17:45:53 -0600889 else:
890 log_func += ')\\n"'
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600891 log_func_no_addr += ')\\n"'
Mike Stroyanbf237d72015-04-03 17:45:53 -0600892 log_func += ';'
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600893 log_func_no_addr += ';'
Tobin Ehlisa30e7e52015-07-06 14:02:36 -0600894 log_func += '\n }\n else {%s\n }' % log_func_no_addr;
Tobin Ehlis08fafd02015-06-12 12:49:01 -0600895 log_func += '\n%s' % self.lineinfo.get()
Tobin Ehlisb870cbb2015-04-15 07:46:12 -0600896 #print("Proto %s has param_dict: %s" % (proto.name, sp_param_dict))
Mike Stroyanbf237d72015-04-03 17:45:53 -0600897 if len(sp_param_dict) > 0:
Tobin Ehlisa2b199c2015-04-27 17:30:04 -0600898 indent = ' '
899 log_func += '\n%sif (g_APIDumpDetailed) {' % indent
900 indent += ' '
Mike Stroyanbf237d72015-04-03 17:45:53 -0600901 i_decl = False
Tobin Ehlis08fafd02015-06-12 12:49:01 -0600902 log_func += '\n%s' % self.lineinfo.get()
Tobin Ehlisa2b199c2015-04-27 17:30:04 -0600903 log_func += '\n%sstring tmp_str;' % indent
Mike Stroyanbf237d72015-04-03 17:45:53 -0600904 for sp_index in sp_param_dict:
Tobin Ehlisb870cbb2015-04-15 07:46:12 -0600905 #print("sp_index: %s" % str(sp_index))
Mike Stroyanbf237d72015-04-03 17:45:53 -0600906 if 'index' == sp_param_dict[sp_index]:
Courtney Goeltzenleuchter95487bc2015-04-14 18:48:46 -0600907 cis_print_func = 'vk_print_%s' % (proto.params[sp_index].ty.replace('const ', '').strip('*').lower())
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600908 local_name = proto.params[sp_index].name
909 if '*' not in proto.params[sp_index].ty:
910 local_name = '&%s' % proto.params[sp_index].name
Tobin Ehlis08fafd02015-06-12 12:49:01 -0600911 log_func += '\n%s' % self.lineinfo.get()
Tobin Ehlisa2b199c2015-04-27 17:30:04 -0600912 log_func += '\n%sif (%s) {' % (indent, local_name)
913 indent += ' '
914 log_func += '\n%stmp_str = %s(%s, " ");' % (indent, cis_print_func, local_name)
915 log_func += '\n%s(*outputStream) << " %s (" << %s << ")" << endl << tmp_str << endl;' % (indent, local_name, local_name)
916 indent = indent[4:]
917 log_func += '\n%s}' % (indent)
Mike Stroyanbf237d72015-04-03 17:45:53 -0600918 else: # We have a count value stored to iterate over an array
919 print_cast = ''
920 print_func = ''
Courtney Goeltzenleuchter95487bc2015-04-14 18:48:46 -0600921 if vk_helper.is_type(proto.params[sp_index].ty.strip('*').replace('const ', ''), 'struct'):
Mike Stroyanbf237d72015-04-03 17:45:53 -0600922 print_cast = '&'
Courtney Goeltzenleuchter95487bc2015-04-14 18:48:46 -0600923 print_func = 'vk_print_%s' % proto.params[sp_index].ty.replace('const ', '').strip('*').lower()
Mike Stroyanbf237d72015-04-03 17:45:53 -0600924 else:
Tobin Ehlise78dbd82015-04-09 09:19:36 -0600925 print_cast = ''
Mike Stroyanbf237d72015-04-03 17:45:53 -0600926 print_func = 'string_convert_helper'
927 #cis_print_func = 'tmp_str = string_convert_helper((void*)%s[i], " ");' % proto.params[sp_index].name
Tobin Ehlisa30e7e52015-07-06 14:02:36 -0600928 if proto.params[sp_index].ty.strip('*').replace('const ', '') in vulkan.object_non_dispatch_list:
929 cis_print_func = 'tmp_str = %s(%s%s[i].handle, " ");' % (print_func, print_cast, proto.params[sp_index].name)
930 else:
931 cis_print_func = 'tmp_str = %s(%s%s[i], " ");' % (print_func, print_cast, proto.params[sp_index].name)
Mike Stroyanbf237d72015-04-03 17:45:53 -0600932 if not i_decl:
Tobin Ehlisa2b199c2015-04-27 17:30:04 -0600933 log_func += '\n%suint32_t i;' % (indent)
Mike Stroyanbf237d72015-04-03 17:45:53 -0600934 i_decl = True
Courtney Goeltzenleuchterf579fa62015-06-10 17:39:03 -0600935 log_func += '\n%sif (%s) {' % (indent, proto.params[sp_index].name)
936 indent += ' '
Tobin Ehlis08fafd02015-06-12 12:49:01 -0600937 log_func += '\n%s' % self.lineinfo.get()
Tobin Ehlisa2b199c2015-04-27 17:30:04 -0600938 log_func += '\n%sfor (i = 0; i < %s; i++) {' % (indent, sp_param_dict[sp_index])
939 indent += ' '
940 log_func += '\n%s%s' % (indent, cis_print_func)
Tobin Ehlis45a23a82015-04-28 10:58:20 -0600941 log_func += '\n%sif (StreamControl::writeAddress == true) {' % (indent)
942 indent += ' '
Tobin Ehlisa2b199c2015-04-27 17:30:04 -0600943 log_func += '\n%s(*outputStream) << " %s[" << i << "] (" << %s%s[i] << ")" << endl << tmp_str << endl;' % (indent, proto.params[sp_index].name, '&', proto.params[sp_index].name)
944 indent = indent[4:]
Tobin Ehlis45a23a82015-04-28 10:58:20 -0600945 log_func += '\n%s} else {' % (indent)
946 indent += ' '
947 log_func += '\n%s(*outputStream) << " %s[" << i << "] (address)" << endl << " address" << endl;' % (indent, proto.params[sp_index].name)
948 indent = indent[4:]
949 log_func += '\n%s}' % (indent)
950 indent = indent[4:]
Tobin Ehlisa2b199c2015-04-27 17:30:04 -0600951 log_func += '\n%s}' % (indent)
Courtney Goeltzenleuchterf579fa62015-06-10 17:39:03 -0600952 indent = indent[4:]
953 log_func += '\n%s}' % (indent)
Tobin Ehlisa2b199c2015-04-27 17:30:04 -0600954 indent = indent[4:]
955 log_func += '\n%s}' % (indent)
Mark Lobodzinski4c7ac0a2015-05-26 09:29:09 -0500956 table_type = ''
Jon Ashburn95a77ba2015-05-15 15:09:35 -0600957 if proto_is_global(proto):
Mark Lobodzinski4c7ac0a2015-05-26 09:29:09 -0500958 table_type = 'instance'
959 else:
960 table_type = 'device'
Jon Ashburn71836d92015-05-12 17:23:55 -0600961
Mark Lobodzinski4c7ac0a2015-05-26 09:29:09 -0500962 if proto.name == "CreateInstance":
963 dispatch_param = '*' + proto.params[1].name
964 else:
965 dispatch_param = proto.params[0].name
966
Jon Ashburn747f2b62015-06-18 15:02:58 -0600967 if proto.name == "CreateDevice":
Courtney Goeltzenleuchter0de9e212015-06-26 15:07:53 -0600968 funcs.append('%s\n' % self.lineinfo.get())
Jon Ashburn747f2b62015-06-18 15:02:58 -0600969 funcs.append('%s%s\n'
970 '{\n'
971 ' using namespace StreamControl;\n'
Courtney Goeltzenleuchterca173b82015-06-25 18:01:43 -0600972 ' %sdevice_dispatch_table(*pDevice)->%s;\n'
Jon Ashburn747f2b62015-06-18 15:02:58 -0600973 ' if (result == VK_SUCCESS)\n'
974 ' createDeviceRegisterExtensions(pCreateInfo, *pDevice);\n'
975 ' %s%s%s\n'
976 '%s'
Courtney Goeltzenleuchterca173b82015-06-25 18:01:43 -0600977 '}' % (qual, decl, ret_val, proto.c_call(), f_open, log_func, f_close, stmt))
Jon Ashburn747f2b62015-06-18 15:02:58 -0600978 elif proto.name == "DestroyDevice":
Mark Lobodzinski4c7ac0a2015-05-26 09:29:09 -0500979 funcs.append('%s%s\n'
980 '{\n'
981 ' using namespace StreamControl;\n'
Courtney Goeltzenleuchter0daf2282015-06-13 21:22:12 -0600982 ' dispatch_key key = get_dispatch_key(device);\n'
Jon Ashburn747f2b62015-06-18 15:02:58 -0600983 ' VkLayerDispatchTable *pDisp = %s_dispatch_table(%s);\n'
984 ' %spDisp->%s;\n'
985 ' deviceExtMap.erase(pDisp);\n'
Courtney Goeltzenleuchter0daf2282015-06-13 21:22:12 -0600986 ' destroy_device_dispatch_table(key);\n'
Mark Lobodzinski4c7ac0a2015-05-26 09:29:09 -0500987 ' %s%s%s\n'
988 '%s'
Jon Ashburn747f2b62015-06-18 15:02:58 -0600989 '}' % (qual, decl, table_type, dispatch_param, ret_val, proto.c_call(), f_open, log_func, f_close, stmt))
Mark Lobodzinski4c7ac0a2015-05-26 09:29:09 -0500990 elif proto.name == "DestroyInstance":
991 funcs.append('%s%s\n'
992 '{\n'
993 ' using namespace StreamControl;\n'
Courtney Goeltzenleuchter0daf2282015-06-13 21:22:12 -0600994 ' dispatch_key key = get_dispatch_key(instance);\n'
Mark Lobodzinski4c7ac0a2015-05-26 09:29:09 -0500995 ' %s%s_dispatch_table(%s)->%s;\n'
Courtney Goeltzenleuchter0daf2282015-06-13 21:22:12 -0600996 ' destroy_instance_dispatch_table(key);\n'
Mark Lobodzinski4c7ac0a2015-05-26 09:29:09 -0500997 ' %s%s%s\n'
998 '%s'
999 '}' % (qual, decl, ret_val, table_type, dispatch_param, proto.c_call(), f_open, log_func, f_close, stmt))
1000 else:
1001 funcs.append('%s%s\n'
1002 '{\n'
1003 ' using namespace StreamControl;\n'
1004 ' %s%s_dispatch_table(%s)->%s;\n'
Jon Ashburnbacb0f52015-04-06 10:58:22 -06001005 ' %s%s%s\n'
1006 '%s'
Mark Lobodzinski4c7ac0a2015-05-26 09:29:09 -05001007 '}' % (qual, decl, ret_val, table_type, dispatch_param, proto.c_call(), f_open, log_func, f_close, stmt))
Mike Stroyanbf237d72015-04-03 17:45:53 -06001008 return "\n\n".join(funcs)
1009
Tobin Ehlis99f88672015-01-10 12:42:41 -07001010 def generate_body(self):
Tobin Ehlise78dbd82015-04-09 09:19:36 -06001011 self.layer_name = "APIDump"
Jon Ashburn747f2b62015-06-18 15:02:58 -06001012 extensions=[('wsi_lunarg_enabled',
1013 ['vkCreateSwapChainWSI', 'vkDestroySwapChainWSI',
1014 'vkGetSwapChainInfoWSI', 'vkQueuePresentWSI'])]
Tobin Ehlise78dbd82015-04-09 09:19:36 -06001015 body = [self.generate_init(),
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -06001016 self._generate_dispatch_entrypoints("VK_LAYER_EXPORT"),
Jon Ashburn747f2b62015-06-18 15:02:58 -06001017 self._generate_layer_gpa_function(extensions)]
Tobin Ehlis99f88672015-01-10 12:42:41 -07001018 return "\n\n".join(body)
1019
Tobin Ehlis12076fc2014-10-22 09:06:33 -06001020class ObjectTrackerSubcommand(Subcommand):
1021 def generate_header(self):
1022 header_txt = []
Tobin Ehlis08fafd02015-06-12 12:49:01 -06001023 header_txt.append('%s' % self.lineinfo.get())
Mark Lobodzinskifae78852015-06-23 11:35:12 -06001024 header_txt.append('#include <stdio.h>')
1025 header_txt.append('#include <stdlib.h>')
1026 header_txt.append('#include <string.h>')
1027 header_txt.append('#include <inttypes.h>')
1028 header_txt.append('')
1029 header_txt.append('#include "vulkan.h"')
Tobin Ehlisb835d1b2015-07-03 10:34:49 -06001030 header_txt.append('#include "vk_loader_platform.h"')
Mark Lobodzinskifae78852015-06-23 11:35:12 -06001031 header_txt.append('')
Tobin Ehlis803cc492015-06-08 17:36:28 -06001032 header_txt.append('#include <unordered_map>')
1033 header_txt.append('using namespace std;')
Ian Elliott655cad72015-02-12 17:08:34 -07001034 header_txt.append('// The following is #included again to catch certain OS-specific functions being used:')
Tobin Ehlisb835d1b2015-07-03 10:34:49 -06001035 header_txt.append('#include "vk_loader_platform.h"')
Tobin Ehlisa0cb02e2015-07-03 10:15:26 -06001036 header_txt.append('#include "vk_layer_config.h"')
1037 header_txt.append('#include "vk_layer_msg.h"')
Courtney Goeltzenleuchterf579fa62015-06-10 17:39:03 -06001038 header_txt.append('#include "vk_debug_report_lunarg.h"')
Tobin Ehlisa0cb02e2015-07-03 10:15:26 -06001039 header_txt.append('#include "vk_layer_table.h"')
1040 header_txt.append('#include "vk_layer_data.h"')
1041 header_txt.append('#include "vk_layer_logging.h"')
Mark Lobodzinskifae78852015-06-23 11:35:12 -06001042 header_txt.append('')
1043# NOTE: The non-autoGenerated code is in the object_track.h header file
1044 header_txt.append('#include "object_track.h"')
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001045 header_txt.append('')
Jon Ashburn8c5cbcf2015-05-07 10:27:37 -06001046 header_txt.append('static LOADER_PLATFORM_THREAD_ONCE_DECLARATION(initOnce);')
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001047 header_txt.append('')
Tobin Ehlis12076fc2014-10-22 09:06:33 -06001048 return "\n".join(header_txt)
1049
Tony Barboura05dbaa2015-07-09 17:31:46 -06001050 def generate_maps(self):
1051 maps_txt = []
1052 for o in vulkan.core.objects:
1053 maps_txt.append('unordered_map<const void*, OBJTRACK_NODE*> %sMap;' % (o))
1054 maps_txt.append('unordered_map<const void*, OBJTRACK_NODE*> VkSwapChainWSIMap;')
1055 return "\n".join(maps_txt)
1056
1057 def generate_procs(self):
1058 procs_txt = []
1059 for o in vulkan.core.objects:
1060 procs_txt.append('%s' % self.lineinfo.get())
1061 if o in [ 'VkInstance', 'VkPhysicalDevice', 'VkDevice', 'VkQueue', 'VkCmdBuffer']:
1062 procs_txt.append('static void create_obj(%s dispatchable_object, %s vkObj, VkDbgObjectType objType)' % (o, o))
1063 procs_txt.append('{')
1064 procs_txt.append(' log_msg(mdd(dispatchable_object), VK_DBG_REPORT_INFO_BIT, objType, reinterpret_cast<VkUintPtrLeast64>(vkObj), 0, OBJTRACK_NONE, "OBJTRACK",')
1065 procs_txt.append(' "OBJ[%llu] : CREATE %s object 0x%" PRIxLEAST64 , object_track_index++, string_VkDbgObjectType(objType),')
1066 procs_txt.append(' reinterpret_cast<VkUintPtrLeast64>(vkObj));')
1067 else:
1068 procs_txt.append('static void create_obj(VkDevice dispatchable_object, %s vkObj, VkDbgObjectType objType)' % (o))
1069 procs_txt.append('{')
1070 procs_txt.append(' log_msg(mdd(dispatchable_object), VK_DBG_REPORT_INFO_BIT, objType, vkObj.handle, 0, OBJTRACK_NONE, "OBJTRACK",')
1071 procs_txt.append(' "OBJ[%llu] : CREATE %s object 0x%" PRIxLEAST64 , object_track_index++, string_VkDbgObjectType(objType),')
1072 procs_txt.append(' reinterpret_cast<VkUintPtrLeast64>(vkObj.handle));')
1073 procs_txt.append('')
1074 procs_txt.append(' OBJTRACK_NODE* pNewObjNode = new OBJTRACK_NODE;')
1075 procs_txt.append(' pNewObjNode->objType = objType;')
1076 procs_txt.append(' pNewObjNode->status = OBJSTATUS_NONE;')
1077 if o in [ 'VkInstance', 'VkPhysicalDevice', 'VkDevice', 'VkQueue', 'VkCmdBuffer']:
1078 procs_txt.append(' pNewObjNode->vkObj = reinterpret_cast<VkUintPtrLeast64>(vkObj);')
1079 procs_txt.append(' %sMap[vkObj] = pNewObjNode;' % (o))
1080 else:
1081 procs_txt.append(' pNewObjNode->vkObj = reinterpret_cast<VkUintPtrLeast64>(vkObj.handle);')
1082 procs_txt.append(' %sMap[(void*)vkObj.handle] = pNewObjNode;' % (o))
1083 procs_txt.append(' uint32_t objIndex = objTypeToIndex(objType);')
1084 procs_txt.append(' numObjs[objIndex]++;')
1085 procs_txt.append(' numTotalObjs++;')
1086 procs_txt.append('}')
1087 procs_txt.append('')
1088 procs_txt.append('%s' % self.lineinfo.get())
1089 if o in [ 'VkInstance', 'VkPhysicalDevice', 'VkDevice', 'VkQueue', 'VkCmdBuffer']:
1090 procs_txt.append('static void validate_object(%s dispatchable_object, %s object)' % (o, o))
1091 else:
1092 procs_txt.append('static void validate_object(VkDevice dispatchable_object, %s object)' % (o))
1093 procs_txt.append('{')
1094 if o in [ 'VkInstance', 'VkPhysicalDevice', 'VkDevice', 'VkQueue', 'VkCmdBuffer']:
1095 procs_txt.append(' if (%sMap.find(object) == %sMap.end()) {' % (o, o))
1096 procs_txt.append(' log_msg(mdd(dispatchable_object), VK_DBG_REPORT_ERROR_BIT, (VkDbgObjectType) 0, reinterpret_cast<VkUintPtrLeast64>(object), 0, OBJTRACK_INVALID_OBJECT, "OBJTRACK",')
Tobin Ehlis2717d132015-07-10 18:25:07 -06001097 procs_txt.append(' "Invalid %s Object %%p",reinterpret_cast<VkUintPtrLeast64>(object));' % o)
Tony Barboura05dbaa2015-07-09 17:31:46 -06001098 else:
1099 procs_txt.append(' if (%sMap.find((void*)object.handle) == %sMap.end()) {' % (o, o))
1100 procs_txt.append(' log_msg(mdd(dispatchable_object), VK_DBG_REPORT_ERROR_BIT, (VkDbgObjectType) 0, object.handle, 0, OBJTRACK_INVALID_OBJECT, "OBJTRACK",')
Tobin Ehlis2717d132015-07-10 18:25:07 -06001101 procs_txt.append(' "Invalid %s Object %%p",reinterpret_cast<VkUintPtrLeast64>(object.handle));' % o)
Tony Barboura05dbaa2015-07-09 17:31:46 -06001102 procs_txt.append(' }')
1103 procs_txt.append('}')
1104 procs_txt.append('')
1105 procs_txt.append('')
1106 procs_txt.append('%s' % self.lineinfo.get())
1107 if o in [ 'VkInstance', 'VkPhysicalDevice', 'VkDevice', 'VkQueue', 'VkCmdBuffer']:
1108 procs_txt.append('static void destroy_obj(%s dispatchable_object, %s object)' % (o, o))
1109 else:
1110 procs_txt.append('static void destroy_obj(VkDevice dispatchable_object, %s object)' % (o))
1111 procs_txt.append('{')
1112 if o in [ 'VkInstance', 'VkPhysicalDevice', 'VkDevice', 'VkQueue', 'VkCmdBuffer']:
1113 procs_txt.append(' if (%sMap.find(object) != %sMap.end()) {' % (o, o))
1114 procs_txt.append(' OBJTRACK_NODE* pNode = %sMap[object];' % (o))
1115 else:
1116 procs_txt.append(' if (%sMap.find((void*)object.handle) != %sMap.end()) {' % (o, o))
1117 procs_txt.append(' OBJTRACK_NODE* pNode = %sMap[(void*)object.handle];' % (o))
1118 procs_txt.append(' uint32_t objIndex = objTypeToIndex(pNode->objType);')
1119 procs_txt.append(' assert(numTotalObjs > 0);')
1120 procs_txt.append(' numTotalObjs--;')
1121 procs_txt.append(' assert(numObjs[objIndex] > 0);')
1122 procs_txt.append(' numObjs[objIndex]--;')
1123 if o in [ 'VkInstance', 'VkPhysicalDevice', 'VkDevice', 'VkQueue', 'VkCmdBuffer']:
1124 procs_txt.append(' log_msg(mdd(dispatchable_object), VK_DBG_REPORT_INFO_BIT, pNode->objType, reinterpret_cast<VkUintPtrLeast64>(object), 0, OBJTRACK_NONE, "OBJTRACK",')
1125 procs_txt.append(' "OBJ_STAT Destroy %s obj 0x%" PRIxLEAST64 " (%lu total objs remain & %lu %s objs).",')
1126 procs_txt.append(' string_VkDbgObjectType(pNode->objType), reinterpret_cast<VkUintPtrLeast64>(object), numTotalObjs, numObjs[objIndex],')
1127 else:
1128 procs_txt.append(' log_msg(mdd(dispatchable_object), VK_DBG_REPORT_INFO_BIT, pNode->objType, object.handle, 0, OBJTRACK_NONE, "OBJTRACK",')
1129 procs_txt.append(' "OBJ_STAT Destroy %s obj 0x%" PRIxLEAST64 " (%lu total objs remain & %lu %s objs).",')
1130 procs_txt.append(' string_VkDbgObjectType(pNode->objType), reinterpret_cast<VkUintPtrLeast64>(object.handle), numTotalObjs, numObjs[objIndex],')
1131 procs_txt.append(' string_VkDbgObjectType(pNode->objType));')
1132 procs_txt.append(' delete pNode;')
1133 if o in [ 'VkInstance', 'VkPhysicalDevice', 'VkDevice', 'VkQueue', 'VkCmdBuffer']:
1134 procs_txt.append(' %sMap.erase(object);' % (o))
1135 procs_txt.append(' } else {')
1136 procs_txt.append(' log_msg(mdd(dispatchable_object), VK_DBG_REPORT_ERROR_BIT, (VkDbgObjectType) 0, reinterpret_cast<VkUintPtrLeast64>(object), 0, OBJTRACK_NONE, "OBJTRACK",')
1137 procs_txt.append(' "Unable to remove obj 0x%" PRIxLEAST64 ". Was it created? Has it already been destroyed?",')
1138 procs_txt.append(' reinterpret_cast<VkUintPtrLeast64>(object));')
1139 procs_txt.append(' }')
1140 else:
1141 procs_txt.append(' %sMap.erase((void*)object.handle);' % (o))
1142 procs_txt.append(' } else {')
1143 procs_txt.append(' log_msg(mdd(dispatchable_object), VK_DBG_REPORT_ERROR_BIT, (VkDbgObjectType) 0, object.handle, 0, OBJTRACK_NONE, "OBJTRACK",')
1144 procs_txt.append(' "Unable to remove obj 0x%" PRIxLEAST64 ". Was it created? Has it already been destroyed?",')
1145 procs_txt.append(' reinterpret_cast<VkUintPtrLeast64>(object.handle));')
1146 procs_txt.append(' }')
1147 procs_txt.append('}')
1148 procs_txt.append('')
1149 procs_txt.append('%s' % self.lineinfo.get())
1150 if o in [ 'VkInstance', 'VkPhysicalDevice', 'VkDevice', 'VkQueue', 'VkCmdBuffer']:
1151 procs_txt.append('static void set_status(%s dispatchable_object, %s object, VkDbgObjectType objType, ObjectStatusFlags status_flag)' % (o, o))
1152 procs_txt.append('{')
1153 procs_txt.append(' if (object != VK_NULL_HANDLE) {')
1154 procs_txt.append(' if (%sMap.find(object) != %sMap.end()) {' % (o, o))
1155 procs_txt.append(' OBJTRACK_NODE* pNode = %sMap[object];' % (o))
1156 else:
1157 procs_txt.append('static void set_status(VkDevice dispatchable_object, %s object, VkDbgObjectType objType, ObjectStatusFlags status_flag)' % (o))
1158 procs_txt.append('{')
1159 procs_txt.append(' if (object.handle != VK_NULL_HANDLE) {')
1160 procs_txt.append(' if (%sMap.find((void*)object.handle) != %sMap.end()) {' % (o, o))
1161 procs_txt.append(' OBJTRACK_NODE* pNode = %sMap[(void*)object.handle];' % (o))
1162 procs_txt.append(' pNode->status |= status_flag;')
1163 procs_txt.append(' return;')
1164 procs_txt.append(' }')
1165 procs_txt.append(' else {')
1166 procs_txt.append(' // If we do not find it print an error')
1167 if o in [ 'VkInstance', 'VkPhysicalDevice', 'VkDevice', 'VkQueue', 'VkCmdBuffer']:
1168 procs_txt.append(' log_msg(mdd(dispatchable_object), VK_DBG_REPORT_ERROR_BIT, (VkDbgObjectType) 0, reinterpret_cast<VkUintPtrLeast64>(object), 0, OBJTRACK_NONE, "OBJTRACK",')
1169 procs_txt.append(' "Unable to set status for non-existent object 0x%" PRIxLEAST64 " of %s type",')
1170 procs_txt.append(' reinterpret_cast<VkUintPtrLeast64>(object), string_VkDbgObjectType(objType));')
1171 else:
1172 procs_txt.append(' log_msg(mdd(dispatchable_object), VK_DBG_REPORT_ERROR_BIT, (VkDbgObjectType) 0, object.handle, 0, OBJTRACK_NONE, "OBJTRACK",')
1173 procs_txt.append(' "Unable to set status for non-existent object 0x%" PRIxLEAST64 " of %s type",')
1174 procs_txt.append(' reinterpret_cast<VkUintPtrLeast64>(object.handle), string_VkDbgObjectType(objType));')
1175 procs_txt.append(' }')
1176 procs_txt.append(' }')
1177 procs_txt.append('}')
1178 procs_txt.append('')
1179 procs_txt.append('%s' % self.lineinfo.get())
1180 procs_txt.append('static VkBool32 validate_status(')
1181 if o in [ 'VkInstance', 'VkPhysicalDevice', 'VkDevice', 'VkQueue', 'VkCmdBuffer']:
1182 procs_txt.append('%s dispatchable_object, %s object,' % (o, o))
1183 else:
1184 procs_txt.append('VkDevice dispatchable_object, %s object,' % (o))
1185 procs_txt.append(' VkDbgObjectType objType,')
1186 procs_txt.append(' ObjectStatusFlags status_mask,')
1187 procs_txt.append(' ObjectStatusFlags status_flag,')
1188 procs_txt.append(' VkFlags msg_flags,')
1189 procs_txt.append(' OBJECT_TRACK_ERROR error_code,')
1190 procs_txt.append(' const char *fail_msg)')
1191 procs_txt.append('{')
1192 if o in [ 'VkInstance', 'VkPhysicalDevice', 'VkDevice', 'VkQueue', 'VkCmdBuffer']:
1193 procs_txt.append(' if (%sMap.find(object) != %sMap.end()) {' % (o, o))
1194 procs_txt.append(' OBJTRACK_NODE* pNode = %sMap[object];' % (o))
1195 else:
1196 procs_txt.append(' if (%sMap.find((void*)object.handle) != %sMap.end()) {' % (o, o))
1197 procs_txt.append(' OBJTRACK_NODE* pNode = %sMap[(void*)object.handle];' % (o))
1198 procs_txt.append(' if ((pNode->status & status_mask) != status_flag) {')
1199 procs_txt.append(' char str[1024];')
1200 if o in [ 'VkInstance', 'VkPhysicalDevice', 'VkDevice', 'VkQueue', 'VkCmdBuffer']:
1201 procs_txt.append(' log_msg(mdd(dispatchable_object), msg_flags, pNode->objType, reinterpret_cast<VkUintPtrLeast64>(object), 0, OBJTRACK_UNKNOWN_OBJECT, "OBJTRACK",')
1202 procs_txt.append(' "OBJECT VALIDATION WARNING: %s object 0x%" PRIxLEAST64 ": %s", string_VkDbgObjectType(objType),')
1203 procs_txt.append(' reinterpret_cast<VkUintPtrLeast64>(object), fail_msg);')
1204 else:
1205 procs_txt.append(' log_msg(mdd(dispatchable_object), msg_flags, pNode->objType, object.handle, 0, OBJTRACK_UNKNOWN_OBJECT, "OBJTRACK",')
1206 procs_txt.append(' "OBJECT VALIDATION WARNING: %s object 0x%" PRIxLEAST64 ": %s", string_VkDbgObjectType(objType),')
1207 procs_txt.append(' reinterpret_cast<VkUintPtrLeast64>(object.handle), fail_msg);')
1208 procs_txt.append(' return VK_FALSE;')
1209 procs_txt.append(' }')
1210 procs_txt.append(' return VK_TRUE;')
1211 procs_txt.append(' }')
1212 procs_txt.append(' else {')
1213 procs_txt.append(' // If we do not find it print an error')
1214 if o in [ 'VkInstance', 'VkPhysicalDevice', 'VkDevice', 'VkQueue', 'VkCmdBuffer']:
1215 procs_txt.append(' log_msg(mdd(dispatchable_object), msg_flags, (VkDbgObjectType) 0, reinterpret_cast<VkUintPtrLeast64>(object), 0, OBJTRACK_UNKNOWN_OBJECT, "OBJTRACK",')
1216 procs_txt.append(' "Unable to obtain status for non-existent object 0x%" PRIxLEAST64 " of %s type",')
1217 procs_txt.append(' reinterpret_cast<VkUintPtrLeast64>(object), string_VkDbgObjectType(objType));')
1218 else:
1219 procs_txt.append(' log_msg(mdd(dispatchable_object), msg_flags, (VkDbgObjectType) 0, object.handle, 0, OBJTRACK_UNKNOWN_OBJECT, "OBJTRACK",')
1220 procs_txt.append(' "Unable to obtain status for non-existent object 0x%" PRIxLEAST64 " of %s type",')
1221 procs_txt.append(' reinterpret_cast<VkUintPtrLeast64>(object.handle), string_VkDbgObjectType(objType));')
1222 procs_txt.append(' return VK_FALSE;')
1223 procs_txt.append(' }')
1224 procs_txt.append('}')
1225 procs_txt.append('')
1226 procs_txt.append('%s' % self.lineinfo.get())
1227 if o in [ 'VkInstance', 'VkPhysicalDevice', 'VkDevice', 'VkQueue', 'VkCmdBuffer']:
1228 procs_txt.append('static void reset_status(%s dispatchable_object, %s object, VkDbgObjectType objType, ObjectStatusFlags status_flag)' % (o, o))
1229 procs_txt.append('{')
1230 procs_txt.append(' if (%sMap.find(object) != %sMap.end()) {' % (o, o))
1231 procs_txt.append(' OBJTRACK_NODE* pNode = %sMap[object];' % (o))
1232 else:
1233 procs_txt.append('static void reset_status(VkDevice dispatchable_object, %s object, VkDbgObjectType objType, ObjectStatusFlags status_flag)' % (o))
1234 procs_txt.append('{')
1235 procs_txt.append(' if (%sMap.find((void*)object.handle) != %sMap.end()) {' % (o, o))
1236 procs_txt.append(' OBJTRACK_NODE* pNode = %sMap[(void*)object.handle];' % (o))
1237 procs_txt.append(' pNode->status &= ~status_flag;')
1238 procs_txt.append(' return;')
1239 procs_txt.append(' }')
1240 procs_txt.append(' else {')
1241 procs_txt.append(' // If we do not find it print an error')
1242 if o in [ 'VkInstance', 'VkPhysicalDevice', 'VkDevice', 'VkQueue', 'VkCmdBuffer']:
1243 procs_txt.append(' log_msg(mdd(dispatchable_object), VK_DBG_REPORT_ERROR_BIT, objType, reinterpret_cast<VkUintPtrLeast64>(object), 0, OBJTRACK_UNKNOWN_OBJECT, "OBJTRACK",')
1244 procs_txt.append(' "Unable to reset status for non-existent object 0x%" PRIxLEAST64 " of %s type",')
1245 procs_txt.append(' reinterpret_cast<VkUintPtrLeast64>(object), string_VkDbgObjectType(objType));')
1246 else:
1247 procs_txt.append(' log_msg(mdd(dispatchable_object), VK_DBG_REPORT_ERROR_BIT, objType, object.handle, 0, OBJTRACK_UNKNOWN_OBJECT, "OBJTRACK",')
1248 procs_txt.append(' "Unable to reset status for non-existent object 0x%" PRIxLEAST64 " of %s type",')
1249 procs_txt.append(' reinterpret_cast<VkUintPtrLeast64>(object.handle), string_VkDbgObjectType(objType));')
1250 procs_txt.append(' }')
1251 procs_txt.append('}')
1252 procs_txt.append('')
1253 return "\n".join(procs_txt)
1254
1255 def generate_explicit_destroy_instance(self):
1256 gedi_txt = []
1257 gedi_txt.append('%s' % self.lineinfo.get())
1258 gedi_txt.append('VkResult explicit_DestroyInstance(')
1259 gedi_txt.append('VkInstance instance)')
1260 gedi_txt.append('{')
1261 gedi_txt.append(' loader_platform_thread_lock_mutex(&objLock);')
1262 gedi_txt.append(' validate_object(instance, instance);')
1263 gedi_txt.append('')
1264 gedi_txt.append(' destroy_obj(instance, instance);')
1265 gedi_txt.append(' // Report any remaining objects in LL')
1266 for o in vulkan.core.objects:
1267 if o in ['VkPhysicalDevice', 'VkQueue']:
1268 continue
1269 gedi_txt.append(' for (auto it = %sMap.begin(); it != %sMap.end(); ++it) {' % (o, o))
1270 gedi_txt.append(' OBJTRACK_NODE* pNode = it->second;')
1271 gedi_txt.append(' log_msg(mid(instance), VK_DBG_REPORT_ERROR_BIT, pNode->objType, pNode->vkObj, 0, OBJTRACK_OBJECT_LEAK, "OBJTRACK",')
1272 gedi_txt.append(' "OBJ ERROR : %s object 0x%" PRIxLEAST64 " has not been destroyed.", string_VkDbgObjectType(pNode->objType),')
1273 gedi_txt.append(' reinterpret_cast<VkUintPtrLeast64>(pNode->vkObj));')
1274 gedi_txt.append(' }')
1275 gedi_txt.append('')
1276 gedi_txt.append(' dispatch_key key = get_dispatch_key(instance);')
1277 gedi_txt.append(' VkLayerInstanceDispatchTable *pInstanceTable = get_dispatch_table(ObjectTracker_instance_table_map, instance);')
1278 gedi_txt.append(' VkResult result = pInstanceTable->DestroyInstance(instance);')
1279 gedi_txt.append('')
1280 gedi_txt.append(' // Clean up logging callback, if any')
1281 gedi_txt.append(' layer_data *my_data = get_my_data_ptr(key, layer_data_map);')
1282 gedi_txt.append(' if (my_data->logging_callback) {')
1283 gedi_txt.append(' layer_destroy_msg_callback(my_data->report_data, my_data->logging_callback);')
1284 gedi_txt.append(' }')
1285 gedi_txt.append('')
1286 gedi_txt.append(' layer_debug_report_destroy_instance(mid(instance));')
1287 gedi_txt.append(' layer_data_map.erase(pInstanceTable);')
1288 gedi_txt.append('')
1289 gedi_txt.append(' ObjectTracker_instance_table_map.erase(key);')
1290 gedi_txt.append(' assert(ObjectTracker_instance_table_map.size() == 0 && "Should not have any instance mappings hanging around");')
1291 gedi_txt.append('')
1292 gedi_txt.append(' loader_platform_thread_unlock_mutex(&objLock);')
1293 gedi_txt.append(' return result;')
1294 gedi_txt.append('}')
1295 gedi_txt.append('')
1296 return "\n".join(gedi_txt)
1297
1298 def generate_explicit_destroy_device(self):
1299 gedd_txt = []
1300 gedd_txt.append('%s' % self.lineinfo.get())
1301 gedd_txt.append('VkResult explicit_DestroyDevice(')
1302 gedd_txt.append('VkDevice device)')
1303 gedd_txt.append('{')
1304 gedd_txt.append(' loader_platform_thread_lock_mutex(&objLock);')
1305 gedd_txt.append(' validate_object(device, device);')
1306 gedd_txt.append('')
1307 gedd_txt.append(' destroy_obj(device, device);')
1308 gedd_txt.append(' // Report any remaining objects in LL')
1309 for o in vulkan.core.objects:
1310 if o in ['VkPhysicalDevice', 'VkQueue']:
1311 continue
1312 gedd_txt.append(' for (auto it = %sMap.begin(); it != %sMap.end(); ++it) {' % (o, o))
1313 gedd_txt.append(' OBJTRACK_NODE* pNode = it->second;')
1314 gedd_txt.append(' log_msg(mdd(device), VK_DBG_REPORT_ERROR_BIT, pNode->objType, pNode->vkObj, 0, OBJTRACK_OBJECT_LEAK, "OBJTRACK",')
1315 gedd_txt.append(' "OBJ ERROR : %s object 0x%" PRIxLEAST64 " has not been destroyed.", string_VkDbgObjectType(pNode->objType),')
1316 gedd_txt.append(' reinterpret_cast<VkUintPtrLeast64>(pNode->vkObj));')
1317 gedd_txt.append(' }')
1318 gedd_txt.append('')
1319 gedd_txt.append(" // Clean up Queue's MemRef Linked Lists")
1320 gedd_txt.append(' destroyQueueMemRefLists();')
1321 gedd_txt.append('')
1322 gedd_txt.append(' loader_platform_thread_unlock_mutex(&objLock);')
1323 gedd_txt.append('')
1324 gedd_txt.append(' dispatch_key key = get_dispatch_key(device);')
1325 gedd_txt.append(' VkLayerDispatchTable *pDisp = get_dispatch_table(ObjectTracker_device_table_map, device);')
1326 gedd_txt.append(' VkResult result = pDisp->DestroyDevice(device);')
1327 gedd_txt.append(' ObjectTracker_device_table_map.erase(key);')
1328 gedd_txt.append(' assert(ObjectTracker_device_table_map.size() == 0 && "Should not have any instance mappings hanging around");')
1329 gedd_txt.append('')
1330 gedd_txt.append(' return result;')
1331 gedd_txt.append('}')
1332 gedd_txt.append('')
1333 return "\n".join(gedd_txt)
1334
1335 def generate_command_buffer_validates(self):
1336 cbv_txt = []
1337 cbv_txt.append('%s' % self.lineinfo.get())
1338 for o in ['VkPipeline', 'VkDynamicViewportState', 'VkDynamicRasterState', 'VkDynamicColorBlendState', 'VkDynamicDepthStencilState',
1339 'VkPipelineLayout', 'VkBuffer', 'VkEvent', 'VkQueryPool', 'VkRenderPass', 'VkFramebuffer']:
1340 cbv_txt.append('static void validate_object(VkCmdBuffer dispatchable_object, %s object)' % (o))
1341 cbv_txt.append('{')
1342 cbv_txt.append(' if (%sMap.find((void*)object.handle) == %sMap.end()) {' % (o, o))
1343 cbv_txt.append(' log_msg(mdd(dispatchable_object), VK_DBG_REPORT_ERROR_BIT, (VkDbgObjectType) 0, object.handle, 0, OBJTRACK_INVALID_OBJECT, "OBJTRACK",')
Tobin Ehlis2717d132015-07-10 18:25:07 -06001344 cbv_txt.append(' "Invalid %s Object %%p",reinterpret_cast<VkUintPtrLeast64>(object.handle));' % (o))
Tony Barboura05dbaa2015-07-09 17:31:46 -06001345 cbv_txt.append(' }')
1346 cbv_txt.append('}')
1347 cbv_txt.append('')
1348 return "\n".join(cbv_txt)
1349
Mike Stroyan3e3a1eb2015-04-03 17:13:23 -06001350 def generate_intercept(self, proto, qual):
Courtney Goeltzenleuchter110fdf92015-06-29 15:39:26 -06001351 if proto.name in [ 'DbgCreateMsgCallback', 'GetGlobalLayerProperties', 'GetGlobalExtensionProperties','GetPhysicalDeviceLayerProperties', 'GetPhysicalDeviceExtensionProperties' ]:
Mike Stroyan00087e62015-04-03 14:39:16 -06001352 # use default version
1353 return None
Mark Lobodzinski7c75b852015-05-05 15:01:37 -05001354
Tony Barboura05dbaa2015-07-09 17:31:46 -06001355 # Create map of object names to object type enums of the form VkName : VkObjectTypeName
1356 obj_type_mapping = {base_t : base_t.replace("Vk", "VkObjectType") for base_t in vulkan.object_type_list}
Mark Lobodzinski7c75b852015-05-05 15:01:37 -05001357 # Convert object type enum names from UpperCamelCase to UPPER_CASE_WITH_UNDERSCORES
1358 for objectName, objectTypeEnum in obj_type_mapping.items():
1359 obj_type_mapping[objectName] = ucc_to_U_C_C(objectTypeEnum);
1360 # Command Buffer Object doesn't follow the rule.
1361 obj_type_mapping['VkCmdBuffer'] = "VK_OBJECT_TYPE_COMMAND_BUFFER"
Courtney Goeltzenleuchteree4027d2015-06-28 13:01:17 -06001362 obj_type_mapping['VkShaderModule'] = "VK_OBJECT_TYPE_SHADER_MODULE"
Mike Stroyan00087e62015-04-03 14:39:16 -06001363
Mark Lobodzinskifae78852015-06-23 11:35:12 -06001364 explicit_object_tracker_functions = [
1365 "CreateInstance",
1366 "DestroyInstance",
Tony Barbour59a47322015-06-24 16:06:58 -06001367 "GetPhysicalDeviceQueueProperties",
Mark Lobodzinskifae78852015-06-23 11:35:12 -06001368 "CreateDevice",
1369 "DestroyDevice",
1370 "GetDeviceQueue",
1371 "QueueSubmit",
Mark Lobodzinskifae78852015-06-23 11:35:12 -06001372 "QueueBindSparseImageMemory",
1373 "QueueBindSparseBufferMemory",
1374 "GetFenceStatus",
1375 "WaitForFences",
1376 "AllocDescriptorSets",
1377 "MapMemory",
1378 "UnmapMemory",
1379 "FreeMemory",
1380 "DestroySwapChainWSI"
1381 ]
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -06001382 decl = proto.c_func(prefix="vk", attr="VKAPI")
Mike Stroyan00087e62015-04-03 14:39:16 -06001383 param0_name = proto.params[0].name
Mark Lobodzinski48bd16d2015-05-08 09:12:28 -05001384 using_line = ''
Mike Stroyan00087e62015-04-03 14:39:16 -06001385 create_line = ''
Tobin Ehlis9d675942015-06-30 14:32:16 -06001386 object_params = {} # dict of parameters that are VkObject types mapping to the size of array types or '0' if not array
Tobin Ehlis2717d132015-07-10 18:25:07 -06001387 # TODO : For now skipping objs that can be NULL. Really should check these and have special case that allows them to be NULL
1388 valid_null_object_names = ['basePipelineHandle', 'renderPass', 'framebuffer']
Tobin Ehlis803cc492015-06-08 17:36:28 -06001389 # TODO : A few of the skipped types are just "hard" cases that need some more work to support
1390 # Need to handle NULL fences on queue submit, binding null memory, and WSI Image objects
1391 for p in proto.params:
1392 if p.ty in vulkan.core.objects and p.ty not in ['VkPhysicalDevice', 'VkQueue', 'VkFence', 'VkImage', 'VkDeviceMemory']:
Tobin Ehlis46d53622015-07-10 11:10:21 -06001393 if p.name not in valid_null_object_names:
1394 object_params[p.name] = 0
Tobin Ehlis9d675942015-06-30 14:32:16 -06001395 elif vk_helper.is_type(p.ty.replace('const ', '').strip('*'), 'struct'):
1396 struct_type = p.ty.replace('const ', '').strip('*')
1397 if vk_helper.typedef_rev_dict[struct_type] in vk_helper.struct_dict:
1398 struct_type = vk_helper.typedef_rev_dict[struct_type]
1399 for m in sorted(vk_helper.struct_dict[struct_type]):
1400 if vk_helper.struct_dict[struct_type][m]['type'] in vulkan.core.objects and vk_helper.struct_dict[struct_type][m]['type'] not in ['VkPhysicalDevice', 'VkQueue', 'VkFence', 'VkImage', 'VkDeviceMemory']:
Tobin Ehlis46d53622015-07-10 11:10:21 -06001401 if vk_helper.struct_dict[struct_type][m]['name'] not in valid_null_object_names:
1402 param_name = '%s->%s' % (p.name, vk_helper.struct_dict[struct_type][m]['name'])
1403 object_params[param_name] = {}
1404 if vk_helper.struct_dict[struct_type][m]['dyn_array']:
1405 object_params[param_name] = '%s->%s' % (p.name, vk_helper.struct_dict[struct_type][m]['array_size'])
1406 else:
1407 object_params[param_name] = 0
Mike Stroyan00087e62015-04-03 14:39:16 -06001408 funcs = []
Tobin Ehlis803cc492015-06-08 17:36:28 -06001409 mutex_unlock = False
Mark Lobodzinskifae78852015-06-23 11:35:12 -06001410 if proto.name in explicit_object_tracker_functions:
Jon Ashburn4d9f4652015-04-08 21:33:34 -06001411 funcs.append('%s%s\n'
1412 '{\n'
Mark Lobodzinskifae78852015-06-23 11:35:12 -06001413 ' return explicit_%s;\n'
1414 '}' % (qual, decl, proto.c_call()))
1415 return "".join(funcs)
Jon Ashburn4d9f4652015-04-08 21:33:34 -06001416 else:
Mark Lobodzinskifae78852015-06-23 11:35:12 -06001417 if 'Create' in proto.name or 'Alloc' in proto.name:
1418 create_line = ' loader_platform_thread_lock_mutex(&objLock);\n'
1419 create_line += ' if (result == VK_SUCCESS) {\n'
1420 create_line += ' create_obj(%s, *%s, %s);\n' % (param0_name, proto.params[-1].name, obj_type_mapping[proto.params[-1].ty.strip('*').replace('const ', '')])
1421 create_line += ' }\n'
1422 create_line += ' loader_platform_thread_unlock_mutex(&objLock);\n'
1423 if len(object_params) > 0:
1424 if not mutex_unlock:
1425 using_line += ' loader_platform_thread_lock_mutex(&objLock);\n'
1426 mutex_unlock = True
1427 for opn in object_params:
Tobin Ehlis9d675942015-06-30 14:32:16 -06001428 if 0 != object_params[opn]:
1429 using_line += ' for (uint32_t i=0; i<%s; i++)\n' % object_params[opn]
1430 using_line += ' validate_object(%s, %s[i]);\n' % (param0_name, opn)
1431 else:
1432 if '->' in opn:
1433 using_line += ' if (%s)\n' % (opn.split('-')[0])
1434 using_line += ' validate_object(%s, %s);\n' % (param0_name, opn)
1435 else:
1436 using_line += ' validate_object(%s, %s);\n' % (param0_name, opn)
Mark Lobodzinskifae78852015-06-23 11:35:12 -06001437 if mutex_unlock:
1438 using_line += ' loader_platform_thread_unlock_mutex(&objLock);\n'
1439 ret_val = ''
1440 stmt = ''
Mark Lobodzinskifae78852015-06-23 11:35:12 -06001441 if proto.ret != "void":
1442 ret_val = "%s result = " % proto.ret
1443 stmt = " return result;\n"
1444
1445 dispatch_param = proto.params[0].name
1446 if 'CreateInstance' in proto.name:
1447 dispatch_param = '*' + proto.params[1].name
1448
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001449 # Must use 'instance' table for these APIs, 'device' table otherwise
1450 table_type = ""
1451 if proto_is_global(proto):
1452 table_type = "instance"
1453 else:
1454 table_type = "device"
Mike Stroyan00087e62015-04-03 14:39:16 -06001455 funcs.append('%s%s\n'
1456 '{\n'
1457 '%s'
Mark Lobodzinskifae78852015-06-23 11:35:12 -06001458 ' %sget_dispatch_table(ObjectTracker_%s_table_map, %s)->%s;\n'
Mike Stroyan00087e62015-04-03 14:39:16 -06001459 '%s'
Mark Lobodzinskifae78852015-06-23 11:35:12 -06001460 '%s'
1461 '}' % (qual, decl, using_line, ret_val, table_type, dispatch_param, proto.c_call(), create_line, stmt))
Mike Stroyan00087e62015-04-03 14:39:16 -06001462 return "\n\n".join(funcs)
1463
Tobin Ehlis12076fc2014-10-22 09:06:33 -06001464 def generate_body(self):
Mike Stroyan3e3a1eb2015-04-03 17:13:23 -06001465 self.layer_name = "ObjectTracker"
Jon Ashburn747f2b62015-06-18 15:02:58 -06001466 extensions=[('wsi_lunarg_enabled',
1467 ['vkCreateSwapChainWSI', 'vkDestroySwapChainWSI',
Courtney Goeltzenleuchter14e8e502015-07-06 22:30:28 -06001468 'vkGetSwapChainInfoWSI', 'vkQueuePresentWSI'])]
Tony Barboura05dbaa2015-07-09 17:31:46 -06001469 body = [self.generate_maps(),
1470 self.generate_procs(),
1471 self.generate_explicit_destroy_instance(),
1472 self.generate_explicit_destroy_device(),
1473 self.generate_command_buffer_validates(),
1474 self._generate_dispatch_entrypoints("VK_LAYER_EXPORT"),
Tobin Ehlisca915872014-11-18 11:28:33 -07001475 self._generate_extensions(),
Jon Ashburn747f2b62015-06-18 15:02:58 -06001476 self._generate_layer_gpa_function(extensions,
Courtney Goeltzenleuchter17274172015-06-01 14:52:57 -06001477 instance_extensions=['msg_callback_get_proc_addr'])]
Tobin Ehlis12076fc2014-10-22 09:06:33 -06001478 return "\n\n".join(body)
Courtney Goeltzenleuchterb412d212014-11-18 10:40:29 -07001479
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001480class ThreadingSubcommand(Subcommand):
Mike Stroyanbe2b5d82015-07-09 11:01:07 -06001481 thread_check_dispatchable_objects = [
1482 "VkQueue",
1483 "VkCmdBuffer",
1484 ]
1485 thread_check_nondispatchable_objects = [
1486 "VkDeviceMemory",
1487 "VkBuffer",
1488 "VkImage",
1489 "VkDescriptorSet",
1490 "VkDescriptorPool",
1491 "VkSemaphore"
1492 ]
1493 def generate_useObject(self, ty, key):
1494 header_txt = []
1495 header_txt.append('%s' % self.lineinfo.get())
1496 header_txt.append('static void useObject(%s object, const char* type)' % ty)
1497 header_txt.append('{')
1498 header_txt.append(' loader_platform_thread_id tid = loader_platform_get_thread_id();')
1499 header_txt.append(' loader_platform_thread_lock_mutex(&threadingLock);')
1500 header_txt.append(' if (%sObjectsInUse.find(%s) == %sObjectsInUse.end()) {' % (ty, key, ty))
1501 header_txt.append(' %sObjectsInUse[%s] = tid;' % (ty, key))
1502 header_txt.append(' } else {')
1503 header_txt.append(' if (%sObjectsInUse[%s] != tid) {' % (ty, key))
1504 header_txt.append(' char str[1024];')
1505 header_txt.append(' sprintf(str, "THREADING ERROR : object of type %%s is simultaneously used in thread %%ld and thread %%ld", type, %sObjectsInUse[%s], tid);' % (ty, key))
1506 header_txt.append(' layerCbMsg(VK_DBG_REPORT_ERROR_BIT, (VkDbgObjectType) 0, 0, 0, THREADING_CHECKER_MULTIPLE_THREADS, "THREADING", str);')
1507 header_txt.append(' // Wait for thread-safe access to object')
1508 header_txt.append(' while (%sObjectsInUse.find(%s) != %sObjectsInUse.end()) {' % (ty, key, ty))
1509 header_txt.append(' loader_platform_thread_cond_wait(&threadingCond, &threadingLock);')
1510 header_txt.append(' }')
1511 header_txt.append(' %sObjectsInUse[%s] = tid;' % (ty, key))
1512 header_txt.append(' } else {')
1513 header_txt.append(' char str[1024];')
1514 header_txt.append(' sprintf(str, "THREADING ERROR : object of type %s is recursively used in thread %ld", type, tid);')
1515 header_txt.append(' layerCbMsg(VK_DBG_REPORT_ERROR_BIT, (VkDbgObjectType) 0, 0, 0, THREADING_CHECKER_SINGLE_THREAD_REUSE, "THREADING", str);')
1516 header_txt.append(' }')
1517 header_txt.append(' }')
1518 header_txt.append(' loader_platform_thread_unlock_mutex(&threadingLock);')
1519 header_txt.append('}')
1520 return "\n".join(header_txt)
1521 def generate_finishUsingObject(self, ty, key):
1522 header_txt = []
1523 header_txt.append('%s' % self.lineinfo.get())
1524 header_txt.append('static void finishUsingObject(%s object)' % ty)
1525 header_txt.append('{')
1526 header_txt.append(' // Object is no longer in use')
1527 header_txt.append(' loader_platform_thread_lock_mutex(&threadingLock);')
1528 header_txt.append(' %sObjectsInUse.erase(%s);' % (ty, key))
1529 header_txt.append(' loader_platform_thread_cond_broadcast(&threadingCond);')
1530 header_txt.append(' loader_platform_thread_unlock_mutex(&threadingLock);')
1531 header_txt.append('}')
1532 return "\n".join(header_txt)
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001533 def generate_header(self):
1534 header_txt = []
Tobin Ehlis08fafd02015-06-12 12:49:01 -06001535 header_txt.append('%s' % self.lineinfo.get())
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001536 header_txt.append('#include <stdio.h>')
1537 header_txt.append('#include <stdlib.h>')
1538 header_txt.append('#include <string.h>')
1539 header_txt.append('#include <unordered_map>')
Tobin Ehlisb835d1b2015-07-03 10:34:49 -06001540 header_txt.append('#include "vk_loader_platform.h"')
Tobin Ehlis0c6f9ee2015-07-03 09:42:57 -06001541 header_txt.append('#include "vk_layer.h"')
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001542 header_txt.append('#include "threading.h"')
Tobin Ehlisa0cb02e2015-07-03 10:15:26 -06001543 header_txt.append('#include "vk_layer_config.h"')
Courtney Goeltzenleuchter3f9f7c42015-07-06 09:11:12 -06001544 header_txt.append('#include "vk_layer_extension_utils.h"')
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001545 header_txt.append('#include "vk_enum_validate_helper.h"')
1546 header_txt.append('#include "vk_struct_validate_helper.h"')
1547 header_txt.append('//The following is #included again to catch certain OS-specific functions being used:')
Tobin Ehlisb835d1b2015-07-03 10:34:49 -06001548 header_txt.append('#include "vk_loader_platform.h"\n')
Tobin Ehlisa0cb02e2015-07-03 10:15:26 -06001549 header_txt.append('#include "vk_layer_msg.h"\n')
1550 header_txt.append('#include "vk_layer_table.h"\n')
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001551 header_txt.append('')
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001552 header_txt.append('')
1553 header_txt.append('static LOADER_PLATFORM_THREAD_ONCE_DECLARATION(initOnce);')
1554 header_txt.append('')
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001555 header_txt.append('using namespace std;')
Mike Stroyanbe2b5d82015-07-09 11:01:07 -06001556 for ty in self.thread_check_dispatchable_objects:
1557 header_txt.append('static unordered_map<%s, loader_platform_thread_id> %sObjectsInUse;' % (ty, ty))
1558 for ty in self.thread_check_nondispatchable_objects:
1559 header_txt.append('static unordered_map<uint64_t, loader_platform_thread_id> %sObjectsInUse;' % ty)
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001560 header_txt.append('static int threadingLockInitialized = 0;')
1561 header_txt.append('static loader_platform_thread_mutex threadingLock;')
Mike Stroyaned238bb2015-05-15 08:50:57 -06001562 header_txt.append('static loader_platform_thread_cond threadingCond;')
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001563 header_txt.append('static int printLockInitialized = 0;')
1564 header_txt.append('static loader_platform_thread_mutex printLock;\n')
Tobin Ehlis08fafd02015-06-12 12:49:01 -06001565 header_txt.append('%s' % self.lineinfo.get())
Mike Stroyanbe2b5d82015-07-09 11:01:07 -06001566 for ty in self.thread_check_dispatchable_objects:
1567 header_txt.append(self.generate_useObject(ty, "object"))
1568 header_txt.append(self.generate_finishUsingObject(ty, "object"))
Tobin Ehlis08fafd02015-06-12 12:49:01 -06001569 header_txt.append('%s' % self.lineinfo.get())
Mike Stroyanbe2b5d82015-07-09 11:01:07 -06001570 for ty in self.thread_check_nondispatchable_objects:
1571 header_txt.append(self.generate_useObject(ty, "object.handle"))
1572 header_txt.append(self.generate_finishUsingObject(ty, "object.handle"))
1573 header_txt.append('%s' % self.lineinfo.get())
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001574 return "\n".join(header_txt)
1575
1576 def generate_intercept(self, proto, qual):
Courtney Goeltzenleuchterf579fa62015-06-10 17:39:03 -06001577 if proto.name in [ 'DbgCreateMsgCallback' ]:
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001578 # use default version
1579 return None
1580 decl = proto.c_func(prefix="vk", attr="VKAPI")
1581 ret_val = ''
1582 stmt = ''
1583 funcs = []
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001584 table = 'device'
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001585 if proto.ret != "void":
Jon Ashburnb0fbe912015-05-06 10:15:07 -06001586 ret_val = "%s result = " % proto.ret
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001587 stmt = " return result;\n"
Jon Ashburn95a77ba2015-05-15 15:09:35 -06001588 if proto_is_global(proto):
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001589 table = 'instance'
Jon Ashburn71836d92015-05-12 17:23:55 -06001590
Mike Stroyan6ea73e42015-05-11 17:18:14 -06001591 # Memory range calls are special in needed thread checking within structs
1592 if proto.name in ["FlushMappedMemoryRanges","InvalidateMappedMemoryRanges"]:
1593 funcs.append('%s%s\n'
1594 '{\n'
1595 ' for (int i=0; i<memRangeCount; i++) {\n'
Mike Stroyanbe2b5d82015-07-09 11:01:07 -06001596 ' useObject(pMemRanges[i].mem, "VkDeviceMemory");\n'
Mike Stroyan6ea73e42015-05-11 17:18:14 -06001597 ' }\n'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001598 ' %s%s_dispatch_table(%s)->%s;\n'
Mike Stroyan6ea73e42015-05-11 17:18:14 -06001599 ' for (int i=0; i<memRangeCount; i++) {\n'
Mike Stroyanbe2b5d82015-07-09 11:01:07 -06001600 ' finishUsingObject(pMemRanges[i].mem);\n'
Mike Stroyan6ea73e42015-05-11 17:18:14 -06001601 ' }\n'
1602 '%s'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001603 '}' % (qual, decl, ret_val, table, proto.params[0].name, proto.c_call(), stmt))
Mike Stroyan6ea73e42015-05-11 17:18:14 -06001604 return "\n".join(funcs)
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001605 # All functions that do a Get are thread safe
Mike Stroyan6ea73e42015-05-11 17:18:14 -06001606 if 'Get' in proto.name:
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001607 return None
Mike Stroyan6ea73e42015-05-11 17:18:14 -06001608 # All WSI functions are thread safe
1609 if 'WSI' in proto.name:
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001610 return None
Mike Stroyan6ea73e42015-05-11 17:18:14 -06001611 # Initialize in early calls
Courtney Goeltzenleuchterca173b82015-06-25 18:01:43 -06001612 if proto.name == "CreateDevice":
1613 funcs.append('%s' % self.lineinfo.get())
1614 funcs.append('%s%s\n'
1615 '{\n'
1616 ' %sdevice_dispatch_table(*pDevice)->%s;\n'
1617 '%s'
1618 '}' % (qual, decl, ret_val, proto.c_call(), stmt))
1619 return "\n".join(funcs)
1620 elif proto.params[0].ty == "VkPhysicalDevice":
Tobin Ehlis08fafd02015-06-12 12:49:01 -06001621 funcs.append('%s' % self.lineinfo.get())
Mike Stroyan6ea73e42015-05-11 17:18:14 -06001622 funcs.append('%s%s\n'
1623 '{\n'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001624 ' %s%s_dispatch_table(%s)->%s;\n'
Mike Stroyan6ea73e42015-05-11 17:18:14 -06001625 '%s'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001626 '}' % (qual, decl, ret_val, table, proto.params[0].name, proto.c_call(), stmt))
Mike Stroyan6ea73e42015-05-11 17:18:14 -06001627 return "\n".join(funcs)
1628 # Functions changing command buffers need thread safe use of first parameter
1629 if proto.params[0].ty == "VkCmdBuffer":
Tobin Ehlis08fafd02015-06-12 12:49:01 -06001630 funcs.append('%s' % self.lineinfo.get())
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001631 funcs.append('%s%s\n'
1632 '{\n'
Mike Stroyanbe2b5d82015-07-09 11:01:07 -06001633 ' useObject(%s, "%s");\n'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001634 ' %s%s_dispatch_table(%s)->%s;\n'
Mike Stroyanbe2b5d82015-07-09 11:01:07 -06001635 ' finishUsingObject(%s);\n'
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001636 '%s'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001637 '}' % (qual, decl, proto.params[0].name, proto.params[0].ty, ret_val, table, proto.params[0].name, proto.c_call(), proto.params[0].name, stmt))
Mike Stroyan6ea73e42015-05-11 17:18:14 -06001638 return "\n".join(funcs)
1639 # Non-Cmd functions that do a Wait are thread safe
1640 if 'Wait' in proto.name:
1641 return None
1642 # Watch use of certain types of objects passed as any parameter
1643 checked_params = []
1644 for param in proto.params:
Mike Stroyanbe2b5d82015-07-09 11:01:07 -06001645 if param.ty in self.thread_check_dispatchable_objects or param.ty in self.thread_check_nondispatchable_objects:
Mike Stroyan6ea73e42015-05-11 17:18:14 -06001646 checked_params.append(param)
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001647 if proto.name == "DestroyDevice":
1648 funcs.append('%s%s\n'
1649 '{\n'
Courtney Goeltzenleuchter0daf2282015-06-13 21:22:12 -06001650 ' dispatch_key key = get_dispatch_key(device);\n'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001651 ' %s%s_dispatch_table(%s)->%s;\n'
Courtney Goeltzenleuchter0daf2282015-06-13 21:22:12 -06001652 ' destroy_device_dispatch_table(key);\n'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001653 ' return result;\n'
1654 '}\n' % (qual, decl, ret_val, table, proto.params[0].name, proto.c_call()))
1655 return "\n".join(funcs);
1656 elif proto.name == "DestroyInstance":
1657 funcs.append('%s%s\n'
1658 '{\n'
Courtney Goeltzenleuchter0daf2282015-06-13 21:22:12 -06001659 ' dispatch_key key = get_dispatch_key(instance);\n'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001660 ' %s%s_dispatch_table(%s)->%s;\n'
Courtney Goeltzenleuchter0daf2282015-06-13 21:22:12 -06001661 ' destroy_instance_dispatch_table(key);\n'
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001662 ' return result;\n'
1663 '}\n' % (qual, decl, ret_val, table, proto.params[0].name, proto.c_call()))
1664 return "\n".join(funcs);
Courtney Goeltzenleuchterd02a9642015-06-08 14:58:39 -06001665 elif proto.name == "CreateInstance":
1666 funcs.append('%s%s\n'
1667 '{\n'
1668 ' loader_platform_thread_once(&initOnce, initThreading);\n'
1669 '\n'
1670 ' %s %s_dispatch_table(*pInstance)->CreateInstance(pCreateInfo, pInstance);\n'
1671 '\n'
1672 ' if (result == VK_SUCCESS) {\n'
Courtney Goeltzenleuchter110fdf92015-06-29 15:39:26 -06001673 ' enable_debug_report(pCreateInfo->extensionCount, pCreateInfo->ppEnabledExtensionNames);\n'
Courtney Goeltzenleuchterd02a9642015-06-08 14:58:39 -06001674 ' VkLayerInstanceDispatchTable *pTable = instance_dispatch_table(*pInstance);\n'
1675 ' debug_report_init_instance_extension_dispatch_table(\n'
1676 ' pTable,\n'
1677 ' pTable->GetInstanceProcAddr,\n'
1678 ' *pInstance);\n'
1679 ' }\n'
1680 ' return result;\n'
1681 '}\n' % (qual, decl, ret_val, table))
1682 return "\n".join(funcs);
Mike Stroyan6ea73e42015-05-11 17:18:14 -06001683 if len(checked_params) == 0:
1684 return None
1685 # Surround call with useObject and finishUsingObject for each checked_param
Tobin Ehlis08fafd02015-06-12 12:49:01 -06001686 funcs.append('%s' % self.lineinfo.get())
Mike Stroyan6ea73e42015-05-11 17:18:14 -06001687 funcs.append('%s%s' % (qual, decl))
1688 funcs.append('{')
1689 for param in checked_params:
Mike Stroyanbe2b5d82015-07-09 11:01:07 -06001690 funcs.append(' useObject(%s, "%s");' % (param.name, param.ty))
Mark Lobodzinskifb5437a2015-05-22 14:15:36 -05001691 funcs.append(' %s%s_dispatch_table(%s)->%s;' % (ret_val, table, proto.params[0].name, proto.c_call()))
Mike Stroyan6ea73e42015-05-11 17:18:14 -06001692 for param in checked_params:
Mike Stroyanbe2b5d82015-07-09 11:01:07 -06001693 funcs.append(' finishUsingObject(%s);' % param.name)
Mike Stroyan6ea73e42015-05-11 17:18:14 -06001694 funcs.append('%s'
1695 '}' % stmt)
1696 return "\n".join(funcs)
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001697
1698 def generate_body(self):
1699 self.layer_name = "Threading"
Mike Stroyaned238bb2015-05-15 08:50:57 -06001700 body = [self._generate_layer_initialization(True, lockname='threading', condname='threading'),
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001701 self._generate_dispatch_entrypoints("VK_LAYER_EXPORT"),
Courtney Goeltzenleuchter17274172015-06-01 14:52:57 -06001702 self._generate_layer_gpa_function(extensions=[],
1703 instance_extensions=['msg_callback_get_proc_addr']),
1704 self._gen_create_msg_callback(),
1705 self._gen_destroy_msg_callback()]
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001706 return "\n\n".join(body)
1707
Tobin Ehlis12076fc2014-10-22 09:06:33 -06001708def main():
1709 subcommands = {
1710 "layer-funcs" : LayerFuncsSubcommand,
1711 "layer-dispatch" : LayerDispatchSubcommand,
Tobin Ehlisa363cfa2014-11-25 16:59:27 -07001712 "Generic" : GenericLayerSubcommand,
Tobin Ehlise78dbd82015-04-09 09:19:36 -06001713 "APIDump" : APIDumpSubcommand,
Tobin Ehlisa363cfa2014-11-25 16:59:27 -07001714 "ObjectTracker" : ObjectTrackerSubcommand,
Mike Stroyan3712d5c2015-04-02 11:59:05 -06001715 "Threading" : ThreadingSubcommand,
Tobin Ehlis12076fc2014-10-22 09:06:33 -06001716 }
1717
Tobin Ehlis14ff0852014-12-17 17:44:50 -07001718 if len(sys.argv) < 3 or sys.argv[1] not in subcommands or not os.path.exists(sys.argv[2]):
1719 print("Usage: %s <subcommand> <input_header> [options]" % sys.argv[0])
Tobin Ehlis12076fc2014-10-22 09:06:33 -06001720 print
Tobin Ehlis7e65d752015-01-15 17:51:52 -07001721 print("Available subcommands are: %s" % " ".join(subcommands))
Tobin Ehlis12076fc2014-10-22 09:06:33 -06001722 exit(1)
1723
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -06001724 hfp = vk_helper.HeaderFileParser(sys.argv[2])
Tobin Ehlis14ff0852014-12-17 17:44:50 -07001725 hfp.parse()
Courtney Goeltzenleuchterd8e229c2015-04-08 15:36:08 -06001726 vk_helper.enum_val_dict = hfp.get_enum_val_dict()
1727 vk_helper.enum_type_dict = hfp.get_enum_type_dict()
1728 vk_helper.struct_dict = hfp.get_struct_dict()
1729 vk_helper.typedef_fwd_dict = hfp.get_typedef_fwd_dict()
1730 vk_helper.typedef_rev_dict = hfp.get_typedef_rev_dict()
1731 vk_helper.types_dict = hfp.get_types_dict()
Tobin Ehlis14ff0852014-12-17 17:44:50 -07001732
Tobin Ehlis12076fc2014-10-22 09:06:33 -06001733 subcmd = subcommands[sys.argv[1]](sys.argv[2:])
1734 subcmd.run()
1735
1736if __name__ == "__main__":
1737 main()