blob: b1525bfc4551db4c05b5d17152882e591b179c69 [file] [log] [blame]
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001#!/usr/bin/env python3
2#
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06003# Vulkan
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07004#
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#
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -070025
Peter Lohrmannd221ea12015-03-25 21:35:32 -070026import os, sys
27
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -060028# add main repo directory so vulkan.py can be imported. This needs to be a complete path.
Peter Lohrmannd221ea12015-03-25 21:35:32 -070029glv_scripts_path = os.path.dirname(os.path.abspath(__file__))
30main_path = os.path.abspath(glv_scripts_path + "/../../../")
31sys.path.append(main_path)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -070032
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -060033import vulkan
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -070034
35class Subcommand(object):
36 def __init__(self, argv):
37 self.argv = argv
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -060038 self.headers = vulkan.headers
39 self.protos = vulkan.protos
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -070040
41 def run(self):
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -070042 print(self.generate())
43
44 def generate(self):
45 copyright = self.generate_copyright()
46 header = self.generate_header()
47 body = self.generate_body()
48 footer = self.generate_footer()
49
50 contents = []
51 if copyright:
52 contents.append(copyright)
53 if header:
54 contents.append(header)
55 if body:
56 contents.append(body)
57 if footer:
58 contents.append(footer)
59
60 return "\n\n".join(contents)
61
62 def generate_copyright(self):
63 return """/* THIS FILE IS GENERATED. DO NOT EDIT. */
64
65/*
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -060066 * Vulkan
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -070067 *
68 * Copyright (C) 2014 LunarG, Inc.
69 *
70 * Permission is hereby granted, free of charge, to any person obtaining a
71 * copy of this software and associated documentation files (the "Software"),
72 * to deal in the Software without restriction, including without limitation
73 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
74 * and/or sell copies of the Software, and to permit persons to whom the
75 * Software is furnished to do so, subject to the following conditions:
76 *
77 * The above copyright notice and this permission notice shall be included
78 * in all copies or substantial portions of the Software.
79 *
80 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
81 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
82 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
83 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
84 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
85 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
86 * DEALINGS IN THE SOFTWARE.
87 */"""
88
89 def generate_header(self):
90 return "\n".join(["#include <" + h + ">" for h in self.headers])
91
92 def generate_body(self):
93 pass
94
95 def generate_footer(self):
96 pass
97
Jon Ashburn8f445632015-02-12 10:38:36 -070098 # Return set of printf '%' qualifier, input to that qualifier, and any dereference
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -060099 def _get_printf_params(self, vk_type, name, output_param):
Jon Ashburn8f445632015-02-12 10:38:36 -0700100 deref = ""
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700101 # TODO : Need ENUM and STRUCT checks here
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600102 if "VkImage_LAYOUT" in vk_type:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600103 return ("%s", "string_%s(%s)" % (vk_type.replace('const ', '').strip('*'), name), deref)
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600104 if "VkClearColor" in vk_type:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600105 return ("%p", "(void*)&%s" % name, deref)
106 if "_type" in vk_type.lower(): # TODO : This should be generic ENUM check
107 return ("%s", "string_%s(%s)" % (vk_type.replace('const ', '').strip('*'), name), deref)
108 if "char*" == vk_type:
Jon Ashburn8f445632015-02-12 10:38:36 -0700109 return ("%s", name, "*")
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600110 if "uint64_t" in vk_type:
111 if '*' in vk_type:
Jon Ashburn8f445632015-02-12 10:38:36 -0700112 return ("%lu", "(%s == NULL) ? 0 : *(%s)" % (name, name), "*")
113 return ("%lu", name, deref)
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600114 if "size_t" in vk_type:
115 if '*' in vk_type:
Jon Ashburn8f445632015-02-12 10:38:36 -0700116 return ("%zu", "(%s == NULL) ? 0 : *(%s)" % (name, name), "*")
117 return ("%zu", name, deref)
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600118 if "float" in vk_type:
119 if '[' in vk_type: # handle array, current hard-coded to 4 (TODO: Make this dynamic)
Jon Ashburn8f445632015-02-12 10:38:36 -0700120 return ("[%f, %f, %f, %f]", "%s[0], %s[1], %s[2], %s[3]" % (name, name, name, name), deref)
121 return ("%f", name, deref)
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600122 if "bool" in vk_type or 'xcb_randr_crtc_t' in vk_type:
Jon Ashburn8f445632015-02-12 10:38:36 -0700123 return ("%u", name, deref)
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600124 if True in [t in vk_type.lower() for t in ["int", "flags", "mask", "xcb_window_t"]]:
125 if '[' in vk_type: # handle array, current hard-coded to 4 (TODO: Make this dynamic)
Jon Ashburn8f445632015-02-12 10:38:36 -0700126 return ("[%i, %i, %i, %i]", "%s[0], %s[1], %s[2], %s[3]" % (name, name, name, name), deref)
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600127 if '*' in vk_type:
Jon Ashburn8f445632015-02-12 10:38:36 -0700128 return ("%i", "(%s == NULL) ? 0 : *(%s)" % (name, name), "*")
129 return ("%i", name, deref)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700130 if output_param:
Jon Ashburn8f445632015-02-12 10:38:36 -0700131 return ("%p", "(void*)%s" % name, deref)
132 return ("%p", "(void*)(%s)" % name, deref)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700133
134 def _generate_trace_func_ptrs(self):
135 func_ptrs = []
136 func_ptrs.append('// Pointers to real functions and declarations of hooked functions')
137 func_ptrs.append('#ifdef WIN32')
138 func_ptrs.append('extern INIT_ONCE gInitOnce;')
139 for proto in self.protos:
140 if True not in [skip_str in proto.name for skip_str in ['Dbg', 'Wsi']]: #Dbg' not in proto.name and 'Wsi' not in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600141 func_ptrs.append('#define __HOOKED_vk%s hooked_vk%s' % (proto.name, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700142
143 func_ptrs.append('\n#elif defined(PLATFORM_LINUX)')
144 func_ptrs.append('extern pthread_once_t gInitOnce;')
145 for proto in self.protos:
146 if True not in [skip_str in proto.name for skip_str in ['Dbg', 'Wsi']]:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600147 func_ptrs.append('#define __HOOKED_vk%s vk%s' % (proto.name, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700148
149 func_ptrs.append('#endif\n')
150 return "\n".join(func_ptrs)
151
152 def _generate_trace_func_ptrs_ext(self, func_class='Wsi'):
153 func_ptrs = []
154 func_ptrs.append('#ifdef WIN32')
155 for proto in self.protos:
156 if func_class in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600157 func_ptrs.append('#define __HOOKED_vk%s hooked_vk%s' % (proto.name, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700158
159 func_ptrs.append('#elif defined(__linux__)')
160 for proto in self.protos:
161 if func_class in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600162 func_ptrs.append('#define __HOOKED_vk%s vk%s' % (proto.name, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700163
164 func_ptrs.append('#endif\n')
165 return "\n".join(func_ptrs)
166
167 def _generate_trace_func_protos(self):
168 func_protos = []
169 func_protos.append('// Hooked function prototypes\n')
170 for proto in self.protos:
171 if 'Dbg' not in proto.name and 'Wsi' not in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600172 func_protos.append('GLVTRACER_EXPORT %s;' % proto.c_func(prefix="__HOOKED_vk", attr="VKAPI"))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700173
174 return "\n".join(func_protos)
175
176 def _generate_trace_func_protos_ext(self, func_class='Wsi'):
177 func_protos = []
178 func_protos.append('// Hooked function prototypes\n')
179 for proto in self.protos:
180 if func_class in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600181 func_protos.append('GLVTRACER_EXPORT %s;' % proto.c_func(prefix="__HOOKED_vk", attr="VKAPI"))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700182
183 return "\n".join(func_protos)
184
185
Peter Lohrmann358d0092015-04-03 12:03:44 -0700186 def _generate_trace_real_func_ptr_protos(self):
187 func_ptr_assign = []
188 func_ptr_assign.append('')
189 for proto in self.protos:
190 if 'Dbg' not in proto.name and 'Wsi' not in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600191 func_ptr_assign.append('extern %s( VKAPI * real_vk%s)(' % (proto.ret, proto.name))
Peter Lohrmann358d0092015-04-03 12:03:44 -0700192 for p in proto.params:
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600193 func_ptr_assign.append(' %s %s,' % (p.ty, p.name))
Peter Lohrmann358d0092015-04-03 12:03:44 -0700194 func_ptr_assign[-1] = func_ptr_assign[-1].replace(',', ');\n')
195 return "\n".join(func_ptr_assign)
196
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700197 def _generate_func_ptr_assignments(self):
198 func_ptr_assign = []
199 for proto in self.protos:
200 if 'Dbg' not in proto.name and 'Wsi' not in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600201 func_ptr_assign.append('%s( VKAPI * real_vk%s)(' % (proto.ret, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700202 for p in proto.params:
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600203 func_ptr_assign.append(' %s %s,' % (p.ty, p.name))
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600204 func_ptr_assign[-1] = func_ptr_assign[-1].replace(',', ') = vk%s;\n' % (proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700205 return "\n".join(func_ptr_assign)
206
Peter Lohrmann358d0092015-04-03 12:03:44 -0700207
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700208 def _generate_func_ptr_assignments_ext(self, func_class='Wsi'):
209 func_ptr_assign = []
210 for proto in self.protos:
211 if func_class in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600212 func_ptr_assign.append('static %s( VKAPI * real_vk%s)(' % (proto.ret, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700213 for p in proto.params:
214 func_ptr_assign.append(' %s %s,' % (p.ty, p.name))
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600215 func_ptr_assign[-1] = func_ptr_assign[-1].replace(',', ') = vk%s;\n' % (proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700216 return "\n".join(func_ptr_assign)
217
218 def _generate_attach_hooks(self):
219 hooks_txt = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600220 hooks_txt.append('// declared as extern in glvtrace_vk_helpers.h')
Peter Lohrmann358d0092015-04-03 12:03:44 -0700221 hooks_txt.append('BOOL isHooked = FALSE;\n')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700222 hooks_txt.append('void AttachHooks()\n{\n BOOL hookSuccess = TRUE;\n#if defined(WIN32)')
223 hooks_txt.append(' Mhook_BeginMultiOperation(FALSE);')
Tobin Ehlis2012fce2015-01-15 17:53:54 -0700224 # TODO : Verify if CreateInstance is appropriate to key off of here
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600225 hooks_txt.append(' if (real_vkCreateInstance != NULL)')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700226 hooks_txt.append(' {\n isHooked = TRUE;')
227 hook_operator = '='
228 for proto in self.protos:
229 if 'Dbg' not in proto.name and 'Wsi' not in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600230 hooks_txt.append(' hookSuccess %s Mhook_SetHook((PVOID*)&real_vk%s, hooked_vk%s);' % (hook_operator, proto.name, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700231 hook_operator = '&='
232 hooks_txt.append(' }\n')
233 hooks_txt.append(' if (!hookSuccess)\n {')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600234 hooks_txt.append(' glv_LogError("Failed to hook Vulkan.");\n }\n')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700235 hooks_txt.append(' Mhook_EndMultiOperation();\n')
236 hooks_txt.append('#elif defined(__linux__)')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600237 hooks_txt.append(' if (real_vkCreateInstance == vkCreateInstance)')
238 hooks_txt.append(' hookSuccess = glv_platform_get_next_lib_sym((PVOID*)&real_vkCreateInstance,"vkCreateInstance");')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700239 hooks_txt.append(' isHooked = TRUE;')
240 for proto in self.protos:
Tobin Ehlis2012fce2015-01-15 17:53:54 -0700241 if 'Dbg' not in proto.name and 'Wsi' not in proto.name and 'CreateInstance' not in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600242 hooks_txt.append(' hookSuccess %s glv_platform_get_next_lib_sym((PVOID*)&real_vk%s, "vk%s");' % (hook_operator, proto.name, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700243 hooks_txt.append(' if (!hookSuccess)\n {')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600244 hooks_txt.append(' glv_LogError("Failed to hook Vulkan.");\n }\n')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700245 hooks_txt.append('#endif\n}\n')
246 return "\n".join(hooks_txt)
247
248 def _generate_attach_hooks_ext(self, func_class='Wsi'):
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600249 func_ext_dict = {'Wsi': '_vkwsix11ext', 'Dbg': '_vkdbg'}
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700250 first_proto_dict = {'Wsi': 'WsiX11AssociateConnection', 'Dbg': 'DbgSetValidationLevel'}
251 hooks_txt = []
252 hooks_txt.append('void AttachHooks%s()\n{\n BOOL hookSuccess = TRUE;\n#if defined(WIN32)' % func_ext_dict[func_class])
253 hooks_txt.append(' Mhook_BeginMultiOperation(FALSE);')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600254 hooks_txt.append(' if (real_vk%s != NULL)' % first_proto_dict[func_class])
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700255 hooks_txt.append(' {')
256 hook_operator = '='
257 for proto in self.protos:
258 if func_class in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600259 hooks_txt.append(' hookSuccess %s Mhook_SetHook((PVOID*)&real_vk%s, hooked_vk%s);' % (hook_operator, proto.name, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700260 hook_operator = '&='
261 hooks_txt.append(' }\n')
262 hooks_txt.append(' if (!hookSuccess)\n {')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600263 hooks_txt.append(' glv_LogError("Failed to hook Vulkan ext %s.");\n }\n' % func_class)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700264 hooks_txt.append(' Mhook_EndMultiOperation();\n')
265 hooks_txt.append('#elif defined(__linux__)')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600266 hooks_txt.append(' hookSuccess = glv_platform_get_next_lib_sym((PVOID*)&real_vk%s, "vk%s");' % (first_proto_dict[func_class], first_proto_dict[func_class]))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700267 for proto in self.protos:
268 if func_class in proto.name and first_proto_dict[func_class] not in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600269 hooks_txt.append(' hookSuccess %s glv_platform_get_next_lib_sym((PVOID*)&real_vk%s, "vk%s");' % (hook_operator, proto.name, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700270 hooks_txt.append(' if (!hookSuccess)\n {')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600271 hooks_txt.append(' glv_LogError("Failed to hook Vulkan ext %s.");\n }\n' % func_class)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700272 hooks_txt.append('#endif\n}\n')
273 return "\n".join(hooks_txt)
274
275 def _generate_detach_hooks(self):
276 hooks_txt = []
277 hooks_txt.append('void DetachHooks()\n{\n#ifdef __linux__\n return;\n#elif defined(WIN32)')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600278 hooks_txt.append(' BOOL unhookSuccess = TRUE;\n if (real_vkGetGpuInfo != NULL)\n {')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700279 hook_operator = '='
280 for proto in self.protos:
281 if 'Dbg' not in proto.name and 'Wsi' not in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600282 hooks_txt.append(' unhookSuccess %s Mhook_Unhook((PVOID*)&real_vk%s);' % (hook_operator, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700283 hook_operator = '&='
284 hooks_txt.append(' }\n isHooked = FALSE;')
285 hooks_txt.append(' if (!unhookSuccess)\n {')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600286 hooks_txt.append(' glv_LogError("Failed to unhook Vulkan.");\n }')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700287 hooks_txt.append('#endif\n}')
288 hooks_txt.append('#ifdef WIN32\nINIT_ONCE gInitOnce = INIT_ONCE_STATIC_INIT;\n#elif defined(PLATFORM_LINUX)\npthread_once_t gInitOnce = PTHREAD_ONCE_INIT;\n#endif\n')
289 return "\n".join(hooks_txt)
290
291 def _generate_detach_hooks_ext(self, func_class='Wsi'):
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600292 func_ext_dict = {'Wsi': '_vkwsix11ext', 'Dbg': '_vkdbg'}
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700293 first_proto_dict = {'Wsi': 'WsiX11AssociateConnection', 'Dbg': 'DbgSetValidationLevel'}
294 hooks_txt = []
295 hooks_txt.append('void DetachHooks%s()\n{\n#ifdef WIN32' % func_ext_dict[func_class])
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600296 hooks_txt.append(' BOOL unhookSuccess = TRUE;\n if (real_vk%s != NULL)\n {' % first_proto_dict[func_class])
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700297 hook_operator = '='
298 for proto in self.protos:
299 if func_class in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600300 hooks_txt.append(' unhookSuccess %s Mhook_Unhook((PVOID*)&real_vk%s);' % (hook_operator, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700301 hook_operator = '&='
302 hooks_txt.append(' }')
303 hooks_txt.append(' if (!unhookSuccess)\n {')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600304 hooks_txt.append(' glv_LogError("Failed to unhook Vulkan ext %s.");\n }' % func_class)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700305 hooks_txt.append('#elif defined(__linux__)\n return;\n#endif\n}\n')
306 return "\n".join(hooks_txt)
307
Jon Ashburne2248392014-12-16 18:37:04 -0700308 def _generate_init_funcs(self):
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700309 init_tracer = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600310 init_tracer.append('void send_vk_api_version_packet()\n{')
311 init_tracer.append(' struct_vkApiVersion* pPacket;')
Jon Ashburne2248392014-12-16 18:37:04 -0700312 init_tracer.append(' glv_trace_packet_header* pHeader;')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600313 init_tracer.append(' pHeader = glv_create_trace_packet(GLV_TID_VULKAN, GLV_TPI_VK_vkApiVersion, sizeof(struct_vkApiVersion), 0);')
314 init_tracer.append(' pPacket = interpret_body_as_vkApiVersion(pHeader, FALSE);')
315 init_tracer.append(' pPacket->version = VK_API_VERSION;')
Jon Ashburne2248392014-12-16 18:37:04 -0700316 init_tracer.append(' FINISH_TRACE_PACKET();\n}\n')
317
Peter Lohrmann358d0092015-04-03 12:03:44 -0700318 init_tracer.append('extern GLV_CRITICAL_SECTION g_memInfoLock;')
Ian Elliottbc9ca5f2015-02-27 11:10:59 -0700319 init_tracer.append('void InitTracer(void)\n{')
Jon Ashburn1ba771d2015-02-19 17:04:06 -0700320 init_tracer.append(' char *ipAddr = glv_get_global_var("GLVLIB_TRACE_IPADDR");')
321 init_tracer.append(' if (ipAddr == NULL)')
322 init_tracer.append(' ipAddr = "127.0.0.1";')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600323 init_tracer.append(' gMessageStream = glv_MessageStream_create(FALSE, ipAddr, GLV_BASE_PORT + GLV_TID_VULKAN);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700324 init_tracer.append(' glv_trace_set_trace_file(glv_FileLike_create_msg(gMessageStream));')
325 init_tracer.append('// glv_tracelog_set_log_file(glv_FileLike_create_file(fopen("glv_log_traceside.txt","w")));')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600326 init_tracer.append(' glv_tracelog_set_tracer_id(GLV_TID_VULKAN);')
Jon Ashburn1ba771d2015-02-19 17:04:06 -0700327 init_tracer.append(' glv_create_critical_section(&g_memInfoLock);')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600328 init_tracer.append(' send_vk_api_version_packet();\n}\n')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700329 return "\n".join(init_tracer)
330
Tobin Ehlis1d3dd2b2015-03-11 17:19:54 -0600331 # Take a list of params and return a list of dicts w/ ptr param details
332 def _get_packet_ptr_param_list(self, params):
333 ptr_param_list = []
334 # TODO : This is a slightly nicer way to handle custom cases than initial code, however
335 # this can still be further generalized to eliminate more custom code
336 # big case to handle is when ptrs to structs have embedded data that needs to be accounted for in packet
Courtney Goeltzenleuchterddcb6192015-04-14 18:48:46 -0600337 custom_ptr_dict = {'VkDeviceCreateInfo': {'add_txt': 'add_VkDeviceCreateInfo_to_packet(pHeader, (VkDeviceCreateInfo**) &(pPacket->pCreateInfo), pCreateInfo)',
Tobin Ehlis1d3dd2b2015-03-11 17:19:54 -0600338 'finalize_txt': ''},
Courtney Goeltzenleuchter382489d2015-04-10 08:34:15 -0600339 'VkApplicationInfo': {'add_txt': 'add_VkApplicationInfo_to_packet(pHeader, (VkApplicationInfo**)&(pPacket->pAppInfo), pAppInfo)',
Tobin Ehlis1d3dd2b2015-03-11 17:19:54 -0600340 'finalize_txt': ''},
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600341 'VK_PHYSICAL_GPU': {'add_txt': 'glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pGpus), *pGpuCount*sizeof(VK_PHYSICAL_GPU), pGpus)',
Tobin Ehlis1d3dd2b2015-03-11 17:19:54 -0600342 'finalize_txt': 'default'},
343 'pDataSize': {'add_txt': 'glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pDataSize), sizeof(size_t), &_dataSize)',
344 'finalize_txt': 'glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pDataSize))'},
345 'pData': {'add_txt': 'glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pData), _dataSize, pData)',
346 'finalize_txt': 'default'},
347 'pName': {'add_txt': 'glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pName), ((pName != NULL) ? strlen(pName) + 1 : 0), pName)',
348 'finalize_txt': 'default'},
349 'pExtName': {'add_txt': 'glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pExtName), ((pExtName != NULL) ? strlen(pExtName) + 1 : 0), pExtName)',
350 'finalize_txt': 'default'},
351 'pDescriptorSets': {'add_txt': 'glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pDescriptorSets), customSize, pDescriptorSets)',
352 'finalize_txt': 'default'},
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600353 'pUpdateChain': {'add_txt': 'add_update_descriptors_to_trace_packet(pHeader, (void**)&(pPacket->pUpdateChain), pUpdateChain)',
354 'finalize_txt': 'default'},
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600355 'VkShaderCreateInfo': {'add_txt': 'glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo), sizeof(VkShaderCreateInfo), pCreateInfo);\n glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo->pCode), ((pCreateInfo != NULL) ? pCreateInfo->codeSize : 0), pCreateInfo->pCode)',
Tobin Ehlis1d3dd2b2015-03-11 17:19:54 -0600356 'finalize_txt': 'glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pCreateInfo->pCode));\n glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pCreateInfo))'},
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600357 'VkFramebufferCreateInfo': {'add_txt': 'glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo), sizeof(VkFramebufferCreateInfo), pCreateInfo);\n glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo->pColorAttachments), colorCount * sizeof(VkColorAttachmentBindInfo), pCreateInfo->pColorAttachments);\n glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo->pDepthStencilAttachment), dsSize, pCreateInfo->pDepthStencilAttachment)',
Tobin Ehlis1d3dd2b2015-03-11 17:19:54 -0600358 'finalize_txt': 'glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pCreateInfo->pColorAttachments));\n glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pCreateInfo->pDepthStencilAttachment));\n glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pCreateInfo))'},
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600359 'VkRenderPassCreateInfo': {'add_txt': 'glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo), sizeof(VkRenderPassCreateInfo), pCreateInfo);\n glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo->pColorLoadOps), colorCount * sizeof(VkAttachmentLoadOp), pCreateInfo->pColorLoadOps);\n glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo->pColorStoreOps), colorCount * sizeof(VkAttachmentStoreOp), pCreateInfo->pColorStoreOps);\n glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo->pColorLoadClearValues), colorCount * sizeof(VkClearColor), pCreateInfo->pColorLoadClearValues)',
Tobin Ehlis1d3dd2b2015-03-11 17:19:54 -0600360 'finalize_txt': 'glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pCreateInfo->pColorLoadOps));\n glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pCreateInfo->pColorStoreOps));\n glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pCreateInfo->pColorLoadClearValues));\n glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pCreateInfo))'},
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600361 'VkCmdBufferBeginInfo': {'add_txt': 'glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pBeginInfo), sizeof(VkCmdBufferBeginInfo), pBeginInfo);\n add_begin_cmdbuf_to_trace_packet(pHeader, (void**)&(pPacket->pBeginInfo->pNext), pBeginInfo->pNext)',
Tobin Ehlis1d3dd2b2015-03-11 17:19:54 -0600362 'finalize_txt': 'glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pBeginInfo))'},
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600363 'VkDynamicVpStateCreateInfo': {'add_txt': 'glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo), sizeof(VkDynamicVpStateCreateInfo), pCreateInfo);\n glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo->pViewports), vpsCount * sizeof(VkViewport), pCreateInfo->pViewports);\n glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo->pScissors), vpsCount * sizeof(VkRect), pCreateInfo->pScissors)',
Tobin Ehlis1d3dd2b2015-03-11 17:19:54 -0600364 'finalize_txt': 'glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pCreateInfo->pViewports));\n glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pCreateInfo->pScissors));\n glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pCreateInfo))'},
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600365 'VkMemoryAllocInfo': {'add_txt': 'glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pAllocInfo), sizeof(VkMemoryAllocInfo), pAllocInfo);\n add_alloc_memory_to_trace_packet(pHeader, (void**)&(pPacket->pAllocInfo->pNext), pAllocInfo->pNext)',
Tobin Ehlis1d3dd2b2015-03-11 17:19:54 -0600366 'finalize_txt': 'glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pAllocInfo))'},
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600367 'VkGraphicsPipelineCreateInfo': {'add_txt': 'glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo), sizeof(VkGraphicsPipelineCreateInfo), pCreateInfo);\n add_pipeline_state_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo->pNext), pCreateInfo->pNext)',
Tobin Ehlis1d3dd2b2015-03-11 17:19:54 -0600368 'finalize_txt': 'glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pCreateInfo))'},
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600369 'VkDescriptorPoolCreateInfo': {'add_txt': 'glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo), sizeof(VkDescriptorPoolCreateInfo), pCreateInfo);\n glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo->pTypeCount), pCreateInfo->count * sizeof(VkDescriptorTypeCount), pCreateInfo->pTypeCount)',
Tobin Ehlis1d3dd2b2015-03-11 17:19:54 -0600370 'finalize_txt': 'glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pCreateInfo->pTypeCount));\n glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pCreateInfo))'},
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600371 'VkComputePipelineCreateInfo': {'add_txt': 'glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo), sizeof(VkComputePipelineCreateInfo), pCreateInfo);\n add_pipeline_state_to_trace_packet(pHeader, (void**)&(pPacket->pCreateInfo->pNext), pCreateInfo->pNext);\n add_pipeline_shader_to_trace_packet(pHeader, (VkPipelineShader*)&pPacket->pCreateInfo->cs, &pCreateInfo->cs)',
Tobin Ehlis1d3dd2b2015-03-11 17:19:54 -0600372 'finalize_txt': 'finalize_pipeline_shader_address(pHeader, &pPacket->pCreateInfo->cs);\n glv_finalize_buffer_address(pHeader, (void**)&(pPacket->pCreateInfo))'},
373 }
374 for p in params:
375 pp_dict = {}
376 if '*' in p.ty and p.name not in ['pSysMem', 'pReserved']:
377 if 'const' in p.ty.lower() and 'count' in params[params.index(p)-1].name.lower():
378 pp_dict['add_txt'] = 'glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->%s), %s*sizeof(%s), %s)' % (p.name, params[params.index(p)-1].name, p.ty.strip('*').replace('const ', ''), p.name)
379 elif p.ty.strip('*').replace('const ', '') in custom_ptr_dict:
380 pp_dict['add_txt'] = custom_ptr_dict[p.ty.strip('*').replace('const ', '')]['add_txt']
381 pp_dict['finalize_txt'] = custom_ptr_dict[p.ty.strip('*').replace('const ', '')]['finalize_txt']
382 elif p.name in custom_ptr_dict:
383 pp_dict['add_txt'] = custom_ptr_dict[p.name]['add_txt']
384 pp_dict['finalize_txt'] = custom_ptr_dict[p.name]['finalize_txt']
385 # TODO : This is custom hack to account for 2 pData items with dataSize param for sizing
386 if 'pData' == p.name and 'dataSize' == params[params.index(p)-1].name:
387 pp_dict['add_txt'] = pp_dict['add_txt'].replace('_dataSize', 'dataSize')
388 else:
389 pp_dict['add_txt'] = 'glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->%s), sizeof(%s), %s)' % (p.name, p.ty.strip('*').replace('const ', ''), p.name)
390 if 'finalize_txt' not in pp_dict or 'default' == pp_dict['finalize_txt']:
391 pp_dict['finalize_txt'] = 'glv_finalize_buffer_address(pHeader, (void**)&(pPacket->%s))' % (p.name)
392 pp_dict['index'] = params.index(p)
393 ptr_param_list.append(pp_dict)
394 return ptr_param_list
395
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600396 # Take a list of params and return a list of packet size elements
397 def _get_packet_size(self, params):
Tobin Ehlisf57562c2015-03-13 07:18:05 -0600398 ps = [] # List of elements to be added together to account for packet size for given params
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600399 skip_list = [] # store params that are already accounted for so we don't count them twice
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600400 # Dict of specific params with unique custom sizes
401 custom_size_dict = {'pSetBindPoints': '(VK_SHADER_STAGE_COMPUTE * sizeof(uint32_t))', # Accounting for largest possible array
402 }
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600403 for p in params:
404 #First handle custom cases
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600405 if p.name in ['pCreateInfo', 'pUpdateChain', 'pSetLayoutInfoList', 'pBeginInfo', 'pAllocInfo']:
Tobin Ehlisf57562c2015-03-13 07:18:05 -0600406 ps.append('get_struct_chain_size((void*)%s)' % p.name)
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600407 skip_list.append(p.name)
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600408 elif p.name in custom_size_dict:
409 ps.append(custom_size_dict[p.name])
410 skip_list.append(p.name)
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600411 # Skip any params already handled
412 if p.name in skip_list:
413 continue
414 # Now check to identify dynamic arrays which depend on two params
415 if 'count' in p.name.lower():
416 next_idx = params.index(p)+1
417 # If next element is a const *, then multiply count and array type
418 if next_idx < len(params) and '*' in params[next_idx].ty and 'const' in params[next_idx].ty.lower():
419 if '*' in p.ty:
420 ps.append('*%s*sizeof(%s)' % (p.name, params[next_idx].ty.strip('*').replace('const ', '')))
421 else:
422 ps.append('%s*sizeof(%s)' % (p.name, params[next_idx].ty.strip('*').replace('const ', '')))
423 skip_list.append(params[next_idx].name)
424 elif '*' in p.ty and p.name not in ['pSysMem', 'pReserved']:
425 if 'pData' == p.name:
426 if 'dataSize' == params[params.index(p)-1].name:
427 ps.append('dataSize')
428 elif 'counterCount' == params[params.index(p)-1].name:
429 ps.append('sizeof(%s)' % p.ty.strip('*').replace('const ', ''))
430 else:
431 ps.append('((pDataSize != NULL && pData != NULL) ? *pDataSize : 0)')
432 elif '**' in p.ty and 'void' in p.ty:
433 ps.append('sizeof(void*)')
434 elif 'void' in p.ty:
435 ps.append('sizeof(%s)' % p.name)
436 elif 'char' in p.ty:
437 ps.append('((%s != NULL) ? strlen(%s) + 1 : 0)' % (p.name, p.name))
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600438 elif 'pDataSize' in p.name:
439 ps.append('((pDataSize != NULL) ? sizeof(size_t) : 0)')
440 elif 'IMAGE_SUBRESOURCE' in p.ty and 'pSubresource' == p.name:
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600441 ps.append('((pSubresource != NULL) ? sizeof(VkImage_SUBRESOURCE) : 0)')
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600442 else:
443 ps.append('sizeof(%s)' % (p.ty.strip('*').replace('const ', '')))
444 return ps
445
Tobin Ehlis50990512015-02-05 11:29:45 -0700446 # Generate functions used to trace API calls and store the input and result data into a packet
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600447 # Here's the general flow of code insertion w/ option items flagged w/ "?"
448 # Result decl?
449 # Packet struct decl
450 # ?Special case : setup call to function first and do custom API call time tracking
451 # CREATE_PACKET
452 # Call (w/ result?)
453 # Assign packet values
454 # FINISH packet
455 # return result?
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700456 def _generate_trace_funcs(self):
457 func_body = []
Peter Lohrmann358d0092015-04-03 12:03:44 -0700458 manually_written_hooked_funcs = ['CreateInstance', 'EnumerateLayers', 'EnumerateGpus',
459 'AllocDescriptorSets', 'MapMemory', 'UnmapMemory',
460 'CmdPipelineBarrier', 'CmdWaitEvents']
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700461 for proto in self.protos:
Peter Lohrmann358d0092015-04-03 12:03:44 -0700462 if proto.name in manually_written_hooked_funcs:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600463 func_body.append( '// __HOOKED_vk%s is manually written. Look in glvtrace_vk_trace.c\n' % proto.name)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700464 elif 'Dbg' not in proto.name and 'Wsi' not in proto.name:
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600465 raw_packet_update_list = [] # non-ptr elements placed directly into packet
Tobin Ehlis1d3dd2b2015-03-11 17:19:54 -0600466 ptr_packet_update_list = [] # ptr elements to be updated into packet
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700467 return_txt = ''
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600468 packet_size = []
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700469 in_data_size = False # flag when we need to capture local input size variable for in/out size
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600470 func_body.append('GLVTRACER_EXPORT %s VKAPI __HOOKED_vk%s(' % (proto.ret, proto.name))
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600471 for p in proto.params: # TODO : For all of the ptr types, check them for NULL and return 0 if NULL
472 if '[' in p.ty: # Correctly declare static arrays in function parameters
473 func_body.append(' %s %s[%s],' % (p.ty[:p.ty.find('[')], p.name, p.ty[p.ty.find('[')+1:p.ty.find(']')]))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700474 else:
475 func_body.append(' %s %s,' % (p.ty, p.name))
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600476 if '*' in p.ty and p.name not in ['pSysMem', 'pReserved']:
477 if 'pDataSize' in p.name:
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700478 in_data_size = True;
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700479 else:
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600480 if '[' in p.ty:
Tobin Ehlis8a5a85b2015-02-25 11:30:27 -0700481 array_str = p.ty[p.ty.find('[')+1:p.ty.find(']')]
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600482 raw_packet_update_list.append(' memcpy((void*)pPacket->color, color, %s * sizeof(%s));' % (array_str, p.ty.strip('*').replace('const ', '').replace('[%s]' % array_str, '')))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700483 else:
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600484 raw_packet_update_list.append(' pPacket->%s = %s;' % (p.name, p.name))
485 # Get list of packet size modifiers due to ptr params
486 packet_size = self._get_packet_size(proto.params)
Tobin Ehlis1d3dd2b2015-03-11 17:19:54 -0600487 ptr_packet_update_list = self._get_packet_ptr_param_list(proto.params)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700488 func_body[-1] = func_body[-1].replace(',', ')')
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600489 # End of function declaration portion, begin function body
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700490 func_body.append('{\n glv_trace_packet_header* pHeader;')
Tobin Ehlis2012fce2015-01-15 17:53:54 -0700491 if 'void' not in proto.ret or '*' in proto.ret:
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700492 func_body.append(' %s result;' % proto.ret)
493 return_txt = 'result = '
494 if in_data_size:
Jon Ashburn53f54a32015-02-11 09:32:29 -0700495 func_body.append(' size_t _dataSize;')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600496 func_body.append(' struct_vk%s* pPacket = NULL;' % proto.name)
Jon Ashburn8cb39a32015-02-02 12:39:24 -0700497 # functions that have non-standard sequence of packet creation and calling real function
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600498 # NOTE: Anytime we call the function before CREATE_TRACE_PACKET, need to add custom code for correctly tracking API call time
Peter Lohrmann358d0092015-04-03 12:03:44 -0700499 if proto.name in ['CreateFramebuffer', 'CreateRenderPass', 'CreateDynamicViewportState',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600500 'CreateDescriptorRegion']:
Tobin Ehlisf57562c2015-03-13 07:18:05 -0600501 # these are regular case as far as sequence of tracing but have some custom size element
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600502 if 'CreateFramebuffer' == proto.name:
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600503 func_body.append(' int dsSize = (pCreateInfo != NULL && pCreateInfo->pDepthStencilAttachment != NULL) ? sizeof(VkDepthStencilBindInfo) : 0;')
Tobin Ehlis50990512015-02-05 11:29:45 -0700504 func_body.append(' uint32_t colorCount = (pCreateInfo != NULL && pCreateInfo->pColorAttachments != NULL) ? pCreateInfo->colorAttachmentCount : 0;')
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600505 func_body.append(' CREATE_TRACE_PACKET(vkCreateFramebuffer, get_struct_chain_size((void*)pCreateInfo) + sizeof(VkFramebuffer));')
Jon Ashburna02bc242015-01-02 18:28:26 -0700506 elif 'CreateRenderPass' == proto.name:
Jon Ashburnec7dcb42015-02-20 10:30:32 -0700507 func_body.append(' uint32_t colorCount = (pCreateInfo != NULL && (pCreateInfo->pColorLoadOps != NULL || pCreateInfo->pColorStoreOps != NULL || pCreateInfo->pColorLoadClearValues != NULL)) ? pCreateInfo->colorAttachmentCount : 0;')
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600508 func_body.append(' CREATE_TRACE_PACKET(vkCreateRenderPass, get_struct_chain_size((void*)pCreateInfo) + sizeof(VkRenderPass));')
Jon Ashburn8e7dcd02015-02-04 08:50:35 -0700509 elif 'CreateDynamicViewportState' == proto.name:
Courtney Goeltzenleuchterc6e32f92015-02-11 14:13:34 -0700510 func_body.append(' uint32_t vpsCount = (pCreateInfo != NULL && pCreateInfo->pViewports != NULL) ? pCreateInfo->viewportAndScissorCount : 0;')
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600511 func_body.append(' CREATE_TRACE_PACKET(vkCreateDynamicViewportState, get_struct_chain_size((void*)pCreateInfo) + sizeof(VkDynamicVpState));')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600512 elif 'CreateDescriptorRegion' == proto.name:
Tobin Ehlis50990512015-02-05 11:29:45 -0700513 func_body.append(' uint32_t rgCount = (pCreateInfo != NULL && pCreateInfo->pTypeCount != NULL) ? pCreateInfo->count : 0;')
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600514 func_body.append(' CREATE_TRACE_PACKET(vkCreateDescriptorRegion, get_struct_chain_size((void*)pCreateInfo) + sizeof(VkDescriptorRegion));')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600515 func_body.append(' %sreal_vk%s;' % (return_txt, proto.c_call()))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700516 else:
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600517 if (0 == len(packet_size)):
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600518 func_body.append(' CREATE_TRACE_PACKET(vk%s, 0);' % (proto.name))
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600519 else:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600520 func_body.append(' CREATE_TRACE_PACKET(vk%s, %s);' % (proto.name, ' + '.join(packet_size)))
521 func_body.append(' %sreal_vk%s;' % (return_txt, proto.c_call()))
Jon Ashburn53f54a32015-02-11 09:32:29 -0700522 if in_data_size:
523 func_body.append(' _dataSize = (pDataSize == NULL || pData == NULL) ? 0 : *pDataSize;')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600524 func_body.append(' pPacket = interpret_body_as_vk%s(pHeader);' % proto.name)
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -0600525 func_body.append('\n'.join(raw_packet_update_list))
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600526 for pp_dict in ptr_packet_update_list: #buff_ptr_indices:
527 func_body.append(' %s;' % (pp_dict['add_txt']))
528 if 'void' not in proto.ret or '*' in proto.ret:
529 func_body.append(' pPacket->result = result;')
530 for pp_dict in ptr_packet_update_list:
Courtney Goeltzenleuchterddcb6192015-04-14 18:48:46 -0600531 if ('DeviceCreateInfo' not in proto.params[pp_dict['index']].ty) and ('APPLICATION_INFO' not in proto.params[pp_dict['index']].ty) and ('pUpdateChain' != proto.params[pp_dict['index']].name):
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600532 func_body.append(' %s;' % (pp_dict['finalize_txt']))
533 func_body.append(' FINISH_TRACE_PACKET();')
534 if 'AllocMemory' in proto.name:
535 func_body.append(' add_new_handle_to_mem_info(*pMem, pAllocInfo->allocationSize, NULL);')
536 elif 'FreeMemory' in proto.name:
537 func_body.append(' rm_handle_from_mem_info(mem);')
Tobin Ehlis2012fce2015-01-15 17:53:54 -0700538 if 'void' not in proto.ret or '*' in proto.ret:
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700539 func_body.append(' return result;')
540 func_body.append('}\n')
541 return "\n".join(func_body)
542
543 def _generate_trace_funcs_ext(self, func_class='Wsi'):
544 thread_once_funcs = ['DbgRegisterMsgCallback', 'DbgUnregisterMsgCallback', 'DbgSetGlobalOption']
545 func_body = []
546 for proto in self.protos:
547 if func_class in proto.name:
548 packet_update_txt = ''
549 return_txt = ''
550 packet_size = ''
551 buff_ptr_indices = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600552 func_body.append('GLVTRACER_EXPORT %s VKAPI __HOOKED_vk%s(' % (proto.ret, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700553 for p in proto.params: # TODO : For all of the ptr types, check them for NULL and return 0 is NULL
554 func_body.append(' %s %s,' % (p.ty, p.name))
555 if 'Size' in p.name:
556 packet_size += p.name
557 if '*' in p.ty and 'pSysMem' != p.name:
Tobin Ehlis2012fce2015-01-15 17:53:54 -0700558 if 'char' in p.ty:
Chia-I Wu7461fcf2014-12-27 15:16:07 +0800559 packet_size += '((%s != NULL) ? strlen(%s) + 1 : 0) + ' % (p.name, p.name)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700560 elif 'Size' not in packet_size:
Tobin Ehlis2012fce2015-01-15 17:53:54 -0700561 packet_size += 'sizeof(%s) + ' % p.ty.strip('*').replace('const ', '')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700562 buff_ptr_indices.append(proto.params.index(p))
563 if 'pConnectionInfo' in p.name:
564 packet_size += '((pConnectionInfo->pConnection != NULL) ? sizeof(void *) : 0)'
565 else:
566 packet_update_txt += ' pPacket->%s = %s;\n' % (p.name, p.name)
567 if '' == packet_size:
568 packet_size = '0'
569 else:
570 packet_size = packet_size.strip(' + ')
571 func_body[-1] = func_body[-1].replace(',', ')')
572 func_body.append('{\n glv_trace_packet_header* pHeader;')
Tobin Ehlis2012fce2015-01-15 17:53:54 -0700573 if 'void' not in proto.ret or '*' in proto.ret:
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700574 func_body.append(' %s result;' % proto.ret)
575 return_txt = 'result = '
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600576 func_body.append(' struct_vk%s* pPacket = NULL;' % proto.name)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700577 if proto.name in thread_once_funcs:
578 func_body.append(' glv_platform_thread_once(&gInitOnce, InitTracer);')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600579 func_body.append(' SEND_ENTRYPOINT_ID(vk%s);' % proto.name)
Ian Elliottbc9ca5f2015-02-27 11:10:59 -0700580 if 'DbgRegisterMsgCallback' in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600581 func_body.append(' CREATE_TRACE_PACKET(vk%s, sizeof(char));' % proto.name)
Ian Elliottbc9ca5f2015-02-27 11:10:59 -0700582 else:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600583 func_body.append(' CREATE_TRACE_PACKET(vk%s, %s);' % (proto.name, packet_size))
584 func_body.append(' %sreal_vk%s;' % (return_txt, proto.c_call()))
585 func_body.append(' pPacket = interpret_body_as_vk%s(pHeader);' % proto.name)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700586 func_body.append(packet_update_txt.strip('\n'))
587 for idx in buff_ptr_indices:
Tobin Ehlis2012fce2015-01-15 17:53:54 -0700588 if 'char' in proto.params[idx].ty:
Chia-I Wu7461fcf2014-12-27 15:16:07 +0800589 func_body.append(' glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->%s), ((%s != NULL) ? strlen(%s) + 1 : 0), %s);' % (proto.params[idx].name, proto.params[idx].name, proto.params[idx].name, proto.params[idx].name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700590 elif 'Size' in proto.params[idx-1].name:
591 func_body.append(' glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->%s), %s, %s);' % (proto.params[idx].name, proto.params[idx-1].name, proto.params[idx].name))
Ian Elliottbc9ca5f2015-02-27 11:10:59 -0700592 elif 'DbgRegisterMsgCallback' in proto.name:
593 func_body.append(' glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->%s), sizeof(%s), %s);' % (proto.params[idx].name, 'char', proto.params[idx].name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700594 else:
Tobin Ehlis2012fce2015-01-15 17:53:54 -0700595 func_body.append(' glv_add_buffer_to_trace_packet(pHeader, (void**)&(pPacket->%s), sizeof(%s), %s);' % (proto.params[idx].name, proto.params[idx].ty.strip('*').replace('const ', ''), proto.params[idx].name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700596 if 'WsiX11AssociateConnection' in proto.name:
597 func_body.append(' if (pConnectionInfo->pConnection != NULL) {')
598 func_body.append(' glv_add_buffer_to_trace_packet(pHeader, (void**) &(pPacket->pConnectionInfo->pConnection), sizeof(void *), pConnectionInfo->pConnection);')
599 func_body.append(' glv_finalize_buffer_address(pHeader, (void**) &(pPacket->pConnectionInfo->pConnection));')
600 func_body.append(' }')
Tobin Ehlis2012fce2015-01-15 17:53:54 -0700601 if 'void' not in proto.ret or '*' in proto.ret:
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700602 func_body.append(' pPacket->result = result;')
603 for idx in buff_ptr_indices:
604 func_body.append(' glv_finalize_buffer_address(pHeader, (void**)&(pPacket->%s));' % (proto.params[idx].name))
605 func_body.append(' FINISH_TRACE_PACKET();')
Tobin Ehlis2012fce2015-01-15 17:53:54 -0700606 if 'void' not in proto.ret or '*' in proto.ret:
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700607 func_body.append(' return result;')
608 func_body.append('}\n')
609 return "\n".join(func_body)
610
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700611 def _generate_packet_id_enum(self):
612 pid_enum = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600613 pid_enum.append('enum GLV_TRACE_PACKET_ID_VK')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700614 pid_enum.append('{')
615 first_func = True
616 for proto in self.protos:
617 if first_func:
618 first_func = False
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600619 pid_enum.append(' GLV_TPI_VK_vkApiVersion = GLV_TPI_BEGIN_API_HERE,')
620 pid_enum.append(' GLV_TPI_VK_vk%s,' % proto.name)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700621 else:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600622 pid_enum.append(' GLV_TPI_VK_vk%s,' % proto.name)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700623 pid_enum.append('};\n')
624 return "\n".join(pid_enum)
625
626 def _generate_stringify_func(self):
627 func_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600628 func_body.append('static const char *stringify_vk_packet_id(const enum GLV_TRACE_PACKET_ID_VK id, const glv_trace_packet_header* pHeader)')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700629 func_body.append('{')
630 func_body.append(' static char str[1024];')
631 func_body.append(' switch(id) {')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600632 func_body.append(' case GLV_TPI_VK_vkApiVersion:')
Jon Ashburne2248392014-12-16 18:37:04 -0700633 func_body.append(' {')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600634 func_body.append(' struct_vkApiVersion* pPacket = (struct_vkApiVersion*)(pHeader->pBody);')
635 func_body.append(' snprintf(str, 1024, "vkApiVersion = 0x%x", pPacket->version);')
Jon Ashburne2248392014-12-16 18:37:04 -0700636 func_body.append(' return str;')
637 func_body.append(' }')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700638 for proto in self.protos:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600639 func_body.append(' case GLV_TPI_VK_vk%s:' % proto.name)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700640 func_body.append(' {')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600641 func_str = 'vk%s(' % proto.name
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700642 print_vals = ''
643 create_func = False
644 if 'Create' in proto.name or 'Alloc' in proto.name or 'MapMemory' in proto.name:
645 create_func = True
646 for p in proto.params:
647 last_param = False
648 if (p.name == proto.params[-1].name):
649 last_param = True
650 if last_param and create_func: # last param of create func
Jon Ashburn8f445632015-02-12 10:38:36 -0700651 (pft, pfi, ptr) = self._get_printf_params(p.ty,'pPacket->%s' % p.name, True)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700652 else:
Jon Ashburn8f445632015-02-12 10:38:36 -0700653 (pft, pfi, ptr) = self._get_printf_params(p.ty, 'pPacket->%s' % p.name, False)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700654 if last_param == True:
Jon Ashburn8f445632015-02-12 10:38:36 -0700655 func_str += '%s%s = %s)' % (ptr, p.name, pft)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700656 print_vals += ', %s' % (pfi)
657 else:
Jon Ashburn8f445632015-02-12 10:38:36 -0700658 func_str += '%s%s = %s, ' % (ptr, p.name, pft)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700659 print_vals += ', %s' % (pfi)
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600660 func_body.append(' struct_vk%s* pPacket = (struct_vk%s*)(pHeader->pBody);' % (proto.name, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700661 func_body.append(' snprintf(str, 1024, "%s"%s);' % (func_str, print_vals))
662 func_body.append(' return str;')
663 func_body.append(' }')
664 func_body.append(' default:')
665 func_body.append(' return NULL;')
666 func_body.append(' }')
667 func_body.append('};\n')
668 return "\n".join(func_body)
669
670 def _generate_interp_func(self):
671 interp_func_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600672 interp_func_body.append('static glv_trace_packet_header* interpret_trace_packet_vk(glv_trace_packet_header* pHeader)')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700673 interp_func_body.append('{')
674 interp_func_body.append(' if (pHeader == NULL)')
675 interp_func_body.append(' {')
676 interp_func_body.append(' return NULL;')
677 interp_func_body.append(' }')
678 interp_func_body.append(' switch (pHeader->packet_id)')
679 interp_func_body.append(' {')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600680 interp_func_body.append(' case GLV_TPI_VK_vkApiVersion:\n {')
681 interp_func_body.append(' return interpret_body_as_vkApiVersion(pHeader, TRUE)->header;\n }')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700682 for proto in self.protos:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600683 interp_func_body.append(' case GLV_TPI_VK_vk%s:\n {' % proto.name)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700684 header_prefix = 'h'
685 if 'Wsi' in proto.name or 'Dbg' in proto.name:
686 header_prefix = 'pH'
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600687 interp_func_body.append(' return interpret_body_as_vk%s(pHeader)->%seader;\n }' % (proto.name, header_prefix))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700688 interp_func_body.append(' default:')
689 interp_func_body.append(' return NULL;')
690 interp_func_body.append(' }')
691 interp_func_body.append(' return NULL;')
692 interp_func_body.append('}')
693 return "\n".join(interp_func_body)
694
695 def _generate_struct_util_funcs(self):
696 pid_enum = []
697 pid_enum.append('//=============================================================================')
Courtney Goeltzenleuchter382489d2015-04-10 08:34:15 -0600698 pid_enum.append('static void add_VkApplicationInfo_to_packet(glv_trace_packet_header* pHeader, VkApplicationInfo** ppStruct, const VkApplicationInfo *pInStruct)')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700699 pid_enum.append('{')
Courtney Goeltzenleuchter382489d2015-04-10 08:34:15 -0600700 pid_enum.append(' glv_add_buffer_to_trace_packet(pHeader, (void**)ppStruct, sizeof(VkApplicationInfo), pInStruct);')
Chia-I Wu7461fcf2014-12-27 15:16:07 +0800701 pid_enum.append(' glv_add_buffer_to_trace_packet(pHeader, (void**)&((*ppStruct)->pAppName), strlen(pInStruct->pAppName) + 1, pInStruct->pAppName);')
702 pid_enum.append(' glv_add_buffer_to_trace_packet(pHeader, (void**)&((*ppStruct)->pEngineName), strlen(pInStruct->pEngineName) + 1, pInStruct->pEngineName);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700703 pid_enum.append(' glv_finalize_buffer_address(pHeader, (void**)&((*ppStruct)->pAppName));')
704 pid_enum.append(' glv_finalize_buffer_address(pHeader, (void**)&((*ppStruct)->pEngineName));')
705 pid_enum.append(' glv_finalize_buffer_address(pHeader, (void**)&*ppStruct);')
706 pid_enum.append('};\n')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600707 pid_enum.append('//=============================================================================\n')
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600708 # TODO : Create add_VkInstanceCreateInfo_to_packet() function here - used in glvtrace_vk_trace.c = add_VkInstanceCreateInfo_to_packet(pHeader, (VkInstanceCreateInfo**)&(pPacket->pCreateInfo), pCreateInfo);
Courtney Goeltzenleuchterddcb6192015-04-14 18:48:46 -0600709 pid_enum.append('static void add_VkDeviceCreateInfo_to_packet(glv_trace_packet_header* pHeader, VkDeviceCreateInfo** ppStruct, const VkDeviceCreateInfo *pInStruct)')
Jon Ashburn29669a42015-04-04 14:52:07 -0600710 pid_enum.append('{')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600711 pid_enum.append(' uint32_t i;')
Courtney Goeltzenleuchterddcb6192015-04-14 18:48:46 -0600712 pid_enum.append(' glv_add_buffer_to_trace_packet(pHeader, (void**)ppStruct, sizeof(VkDeviceCreateInfo), pInStruct);')
713 pid_enum.append(' glv_add_buffer_to_trace_packet(pHeader, (void**)&(*ppStruct)->pRequestedQueues, pInStruct->queueRecordCount*sizeof(VkDeviceQueueCreateInfo), pInStruct->pRequestedQueues);')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600714 pid_enum.append(' glv_finalize_buffer_address(pHeader, (void**)&(*ppStruct)->pRequestedQueues);')
Jon Ashburn29669a42015-04-04 14:52:07 -0600715 pid_enum.append(' if (pInStruct->extensionCount > 0) ')
716 pid_enum.append(' {')
717 pid_enum.append(' glv_add_buffer_to_trace_packet(pHeader, (void**)(&(*ppStruct)->ppEnabledExtensionNames), pInStruct->extensionCount * sizeof(char *), pInStruct->ppEnabledExtensionNames);')
718 pid_enum.append(' for (i = 0; i < pInStruct->extensionCount; i++)')
719 pid_enum.append(' {')
720 pid_enum.append(' glv_add_buffer_to_trace_packet(pHeader, (void**)(&((*ppStruct)->ppEnabledExtensionNames[i])), strlen(pInStruct->ppEnabledExtensionNames[i]) + 1, pInStruct->ppEnabledExtensionNames[i]);')
721 pid_enum.append(' glv_finalize_buffer_address(pHeader, (void**)(&((*ppStruct)->ppEnabledExtensionNames[i])));')
722 pid_enum.append(' }')
723 pid_enum.append(' glv_finalize_buffer_address(pHeader, (void **)&(*ppStruct)->ppEnabledExtensionNames);')
724 pid_enum.append(' }')
Courtney Goeltzenleuchterddcb6192015-04-14 18:48:46 -0600725 pid_enum.append(' VkLayerCreateInfo *pNext = ( VkLayerCreateInfo *) pInStruct->pNext;')
Jon Ashburn780112b2015-01-09 17:30:41 -0700726 pid_enum.append(' while (pNext != NULL)')
727 pid_enum.append(' {')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600728 pid_enum.append(' if ((pNext->sType == VK_STRUCTURE_TYPE_LAYER_CREATE_INFO) && pNext->layerCount > 0)')
Jon Ashburn780112b2015-01-09 17:30:41 -0700729 pid_enum.append(' {')
Courtney Goeltzenleuchterddcb6192015-04-14 18:48:46 -0600730 pid_enum.append(' glv_add_buffer_to_trace_packet(pHeader, (void**)(&((*ppStruct)->pNext)), sizeof(VkLayerCreateInfo), pNext);')
Jon Ashburn780112b2015-01-09 17:30:41 -0700731 pid_enum.append(' glv_finalize_buffer_address(pHeader, (void**)(&((*ppStruct)->pNext)));')
Courtney Goeltzenleuchterddcb6192015-04-14 18:48:46 -0600732 pid_enum.append(' VkLayerCreateInfo **ppOutStruct = (VkLayerCreateInfo **) &((*ppStruct)->pNext);')
Mark Lobodzinskie2d07a52015-01-29 08:55:56 -0600733 pid_enum.append(' glv_add_buffer_to_trace_packet(pHeader, (void**)(&(*ppOutStruct)->ppActiveLayerNames), pNext->layerCount * sizeof(char *), pNext->ppActiveLayerNames);')
Jon Ashburn780112b2015-01-09 17:30:41 -0700734 pid_enum.append(' for (i = 0; i < pNext->layerCount; i++)')
735 pid_enum.append(' {')
736 pid_enum.append(' glv_add_buffer_to_trace_packet(pHeader, (void**)(&((*ppOutStruct)->ppActiveLayerNames[i])), strlen(pNext->ppActiveLayerNames[i]) + 1, pNext->ppActiveLayerNames[i]);')
737 pid_enum.append(' glv_finalize_buffer_address(pHeader, (void**)(&((*ppOutStruct)->ppActiveLayerNames[i])));')
738 pid_enum.append(' }')
739 pid_enum.append(' glv_finalize_buffer_address(pHeader, (void **)&(*ppOutStruct)->ppActiveLayerNames);')
740 pid_enum.append(' }')
Courtney Goeltzenleuchterddcb6192015-04-14 18:48:46 -0600741 pid_enum.append(' pNext = ( VkLayerCreateInfo *) pNext->pNext;')
Jon Ashburn780112b2015-01-09 17:30:41 -0700742 pid_enum.append(' }')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700743 pid_enum.append(' glv_finalize_buffer_address(pHeader, (void**)ppStruct);')
744 pid_enum.append('}\n')
Courtney Goeltzenleuchterddcb6192015-04-14 18:48:46 -0600745 pid_enum.append('static VkDeviceCreateInfo* interpret_VkDeviceCreateInfo(glv_trace_packet_header* pHeader, intptr_t ptr_variable)')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700746 pid_enum.append('{')
Courtney Goeltzenleuchterddcb6192015-04-14 18:48:46 -0600747 pid_enum.append(' VkDeviceCreateInfo* pVkDeviceCreateInfo = (VkDeviceCreateInfo*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)ptr_variable);\n')
748 pid_enum.append(' if (pVkDeviceCreateInfo != NULL)')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700749 pid_enum.append(' {')
Jon Ashburn780112b2015-01-09 17:30:41 -0700750 pid_enum.append(' uint32_t i;')
Mark Lobodzinskie2d07a52015-01-29 08:55:56 -0600751 pid_enum.append(' const char** pNames;')
Courtney Goeltzenleuchterddcb6192015-04-14 18:48:46 -0600752 pid_enum.append(' pVkDeviceCreateInfo->pRequestedQueues = (const VkDeviceQueueCreateInfo *)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pVkDeviceCreateInfo->pRequestedQueues);\n')
753 pid_enum.append(' if (pVkDeviceCreateInfo->extensionCount > 0)')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700754 pid_enum.append(' {')
Courtney Goeltzenleuchterddcb6192015-04-14 18:48:46 -0600755 pid_enum.append(' pVkDeviceCreateInfo->ppEnabledExtensionNames = (const char *const*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pVkDeviceCreateInfo->ppEnabledExtensionNames);')
756 pid_enum.append(' pNames = (const char**)pVkDeviceCreateInfo->ppEnabledExtensionNames;')
757 pid_enum.append(' for (i = 0; i < pVkDeviceCreateInfo->extensionCount; i++)')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700758 pid_enum.append(' {')
Courtney Goeltzenleuchterddcb6192015-04-14 18:48:46 -0600759 pid_enum.append(' pNames[i] = (const char*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)(pVkDeviceCreateInfo->ppEnabledExtensionNames[i]));')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700760 pid_enum.append(' }')
761 pid_enum.append(' }')
Courtney Goeltzenleuchterddcb6192015-04-14 18:48:46 -0600762 pid_enum.append(' VkLayerCreateInfo *pNext = ( VkLayerCreateInfo *) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pVkDeviceCreateInfo->pNext);')
Jon Ashburn780112b2015-01-09 17:30:41 -0700763 pid_enum.append(' while (pNext != NULL)')
764 pid_enum.append(' {')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600765 pid_enum.append(' if ((pNext->sType == VK_STRUCTURE_TYPE_LAYER_CREATE_INFO) && pNext->layerCount > 0)')
Jon Ashburn780112b2015-01-09 17:30:41 -0700766 pid_enum.append(' {')
Mark Lobodzinskie2d07a52015-01-29 08:55:56 -0600767 pid_enum.append(' pNext->ppActiveLayerNames = (const char**) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)(pNext->ppActiveLayerNames));')
768 pid_enum.append(' pNames = (const char**)pNext->ppActiveLayerNames;')
Jon Ashburn780112b2015-01-09 17:30:41 -0700769 pid_enum.append(' for (i = 0; i < pNext->layerCount; i++)')
770 pid_enum.append(' {')
Mark Lobodzinskie2d07a52015-01-29 08:55:56 -0600771 pid_enum.append(' pNames[i] = (const char*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)(pNext->ppActiveLayerNames[i]));')
Jon Ashburn780112b2015-01-09 17:30:41 -0700772 pid_enum.append(' }')
773 pid_enum.append(' }')
Courtney Goeltzenleuchterddcb6192015-04-14 18:48:46 -0600774 pid_enum.append(' pNext = ( VkLayerCreateInfo *) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pNext->pNext);')
Jon Ashburn780112b2015-01-09 17:30:41 -0700775 pid_enum.append(' }')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700776 pid_enum.append(' }\n')
Courtney Goeltzenleuchterddcb6192015-04-14 18:48:46 -0600777 pid_enum.append(' return pVkDeviceCreateInfo;')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700778 pid_enum.append('}\n')
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600779 pid_enum.append('static void interpret_pipeline_shader(glv_trace_packet_header* pHeader, VkPipelineShader* pShader)')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700780 pid_enum.append('{')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700781 pid_enum.append(' if (pShader != NULL)')
782 pid_enum.append(' {')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700783 pid_enum.append(' // constant buffers')
784 pid_enum.append(' if (pShader->linkConstBufferCount > 0)')
785 pid_enum.append(' {')
Mark Lobodzinskie2d07a52015-01-29 08:55:56 -0600786 pid_enum.append(' uint32_t i;')
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600787 pid_enum.append(' pShader->pLinkConstBufferInfo = (const VkLinkConstBuffer*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pShader->pLinkConstBufferInfo);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700788 pid_enum.append(' for (i = 0; i < pShader->linkConstBufferCount; i++)')
789 pid_enum.append(' {')
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600790 pid_enum.append(' VkLinkConstBuffer* pBuffer = (VkLinkConstBuffer*)pShader->pLinkConstBufferInfo;')
Mark Lobodzinskie2d07a52015-01-29 08:55:56 -0600791 pid_enum.append(' pBuffer[i].pBufferData = (const void*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pShader->pLinkConstBufferInfo[i].pBufferData);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700792 pid_enum.append(' }')
793 pid_enum.append(' }')
794 pid_enum.append(' }')
795 pid_enum.append('}\n')
796 pid_enum.append('//=============================================================================')
797 return "\n".join(pid_enum)
798
Tobin Ehlis50990512015-02-05 11:29:45 -0700799 # Interpret functions used on replay to read in packets and interpret their contents
Peter Lohrmanncde614c2015-03-27 12:57:10 -0700800 # This code gets generated into glv_vk_vk_structs.h file
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -0700801 def _generate_interp_funcs(self):
802 # Custom txt for given function and parameter. First check if param is NULL, then insert txt if not
Tobin Ehlis50990512015-02-05 11:29:45 -0700803 # TODO : This code is now too large and complex, need to make codegen smarter for pointers embedded in struct params to handle those cases automatically
Courtney Goeltzenleuchter382489d2015-04-10 08:34:15 -0600804 custom_case_dict = { 'CreateInstance' : {'param': 'pAppInfo', 'txt': ['VkApplicationInfo* pInfo = (VkApplicationInfo*)pPacket->pAppInfo;\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600805 'pInfo->pAppName = (const char*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pAppInfo->pAppName);\n',
806 'pInfo->pEngineName = (const char*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pAppInfo->pEngineName);']},
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600807 'CreateShader' : {'param': 'pCreateInfo', 'txt': ['VkShaderCreateInfo* pInfo = (VkShaderCreateInfo*)pPacket->pCreateInfo;\n',
Jon Ashburna02bc242015-01-02 18:28:26 -0700808 'pInfo->pCode = glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pCreateInfo->pCode);']},
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600809 'CreateDynamicViewportState' : {'param': 'pCreateInfo', 'txt': ['VkDynamicVpStateCreateInfo* pInfo = (VkDynamicVpStateCreateInfo*)pPacket->pCreateInfo;\n',
810 'pInfo->pViewports = (VkViewport*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pCreateInfo->pViewports);\n',
811 'pInfo->pScissors = (VkRect*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pCreateInfo->pScissors);']},
812 'CreateFramebuffer' : {'param': 'pCreateInfo', 'txt': ['VkFramebufferCreateInfo* pInfo = (VkFramebufferCreateInfo*)pPacket->pCreateInfo;\n',
813 'pInfo->pColorAttachments = (VkColorAttachmentBindInfo*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pCreateInfo->pColorAttachments);\n',
814 'pInfo->pDepthStencilAttachment = (VkDepthStencilBindInfo*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pCreateInfo->pDepthStencilAttachment);\n']},
815 'CreateRenderPass' : {'param': 'pCreateInfo', 'txt': ['VkRenderPassCreateInfo* pInfo = (VkRenderPassCreateInfo*)pPacket->pCreateInfo;\n',
816 'pInfo->pColorLoadOps = (VkAttachmentLoadOp*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pCreateInfo->pColorLoadOps);\n',
817 'pInfo->pColorStoreOps = (VkAttachmentStoreOp*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pCreateInfo->pColorStoreOps);\n',
818 'pInfo->pColorLoadClearValues = (VkClearColor*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pCreateInfo->pColorLoadClearValues);\n']},
819 'CreateDescriptorPool' : {'param': 'pCreateInfo', 'txt': ['VkDescriptorPoolCreateInfo* pInfo = (VkDescriptorPoolCreateInfo*)pPacket->pCreateInfo;\n',
820 'pInfo->pTypeCount = (VkDescriptorTypeCount*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pCreateInfo->pTypeCount);\n']},
821 'CmdWaitEvents' : {'param': 'pWaitInfo', 'txt': ['VkEventWaitInfo* pInfo = (VkEventWaitInfo*)pPacket->pWaitInfo;\n',
822 'pInfo->pEvents = (VkEvent*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pWaitInfo->pEvents);\n',
Tobin Ehlis50990512015-02-05 11:29:45 -0700823 'pInfo->ppMemBarriers = (const void**) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pWaitInfo->ppMemBarriers);\n',
824 'uint32_t i;\n',
825 'for (i = 0; i < pInfo->memBarrierCount; i++) {\n',
826 ' void** ppLocalMemBarriers = (void**)&pInfo->ppMemBarriers[i];\n',
827 ' *ppLocalMemBarriers = (void*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pInfo->ppMemBarriers[i]);\n',
828 '}']},
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600829 'CmdPipelineBarrier' : {'param': 'pBarrier', 'txt': ['VkPipelineBarrier* pBarrier = (VkPipelineBarrier*)pPacket->pBarrier;\n',
830 'pBarrier->pEvents = (VkPipeEvent*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pBarrier->pEvents);\n',
Tobin Ehlis50990512015-02-05 11:29:45 -0700831 'pBarrier->ppMemBarriers = (const void**) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pBarrier->ppMemBarriers);\n',
832 'uint32_t i;\n',
833 'for (i = 0; i < pBarrier->memBarrierCount; i++) {\n',
834 ' void** ppLocalMemBarriers = (void**)&pBarrier->ppMemBarriers[i];\n',
835 ' *ppLocalMemBarriers = (void*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pBarrier->ppMemBarriers[i]);\n',
836 '}']},
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600837 'CreateDescriptorSetLayout' : {'param': 'pSetLayoutInfoList', 'txt': ['if (pPacket->pSetLayoutInfoList->sType == VK_STRUCTURE_TYPE_DESCRIPTOR_SET_LAYOUT_CREATE_INFO) {\n',
838 ' // need to make a non-const pointer to the pointer so that we can properly change the original pointer to the interpretted one\n',
839 ' void** ppNextVoidPtr = (void**)&(pPacket->pSetLayoutInfoList->pNext);\n',
840 ' *ppNextVoidPtr = (void*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pSetLayoutInfoList->pNext);\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600841 ' VkDescriptorSetLayoutCreateInfo* pNext = (VkDescriptorSetLayoutCreateInfo*)pPacket->pSetLayoutInfoList->pNext;\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600842 ' while (NULL != pNext)\n', ' {\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700843 ' switch(pNext->sType)\n', ' {\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600844 ' case VK_STRUCTURE_TYPE_DESCRIPTOR_SET_LAYOUT_CREATE_INFO:\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700845 ' {\n' ,
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600846 ' void** ppNextVoidPtr = (void**)&pNext->pNext;\n',
847 ' *ppNextVoidPtr = (void*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pNext->pNext);\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700848 ' break;\n',
849 ' }\n',
850 ' default:\n',
851 ' {\n',
852 ' glv_LogError("Encountered an unexpected type in descriptor set layout create list.\\n");\n',
853 ' pPacket->header = NULL;\n',
854 ' pNext->pNext = NULL;\n',
855 ' }\n',
Jon Ashburn7fd7eff2015-02-04 10:55:47 -0700856 ' }\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600857 ' pNext = (VkDescriptorSetLayoutCreateInfo*)pNext->pNext;\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600858 ' }\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700859 '} else {\n',
860 ' // This is unexpected.\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600861 ' glv_LogError("CreateDescriptorSetLayout must have LayoutInfoList stype of VK_STRCTURE_TYPE_DESCRIPTOR_SET_LAYOUT_CREATE_INFO\\n");\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700862 ' pPacket->header = NULL;\n',
Jon Ashburn7fd7eff2015-02-04 10:55:47 -0700863 '}']},
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600864 'BeginCommandBuffer' : {'param': 'pBeginInfo', 'txt': ['if (pPacket->pBeginInfo->sType == VK_STRUCTURE_TYPE_CMD_BUFFER_BEGIN_INFO) {\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700865 ' // need to make a non-const pointer to the pointer so that we can properly change the original pointer to the interpretted one\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600866 ' VkCmdBufferGraphicsBeginInfo** ppNext = (VkCmdBufferGraphicsBeginInfo**)&(pPacket->pBeginInfo->pNext);\n',
867 ' *ppNext = (VkCmdBufferGraphicsBeginInfo*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pBeginInfo->pNext);\n',
868 ' VkCmdBufferGraphicsBeginInfo* pNext = *ppNext;\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700869 ' while (NULL != pNext)\n', ' {\n',
870 ' switch(pNext->sType)\n', ' {\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600871 ' case VK_STRUCTURE_TYPE_CMD_BUFFER_GRAPHICS_BEGIN_INFO:\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700872 ' {\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600873 ' ppNext = (VkCmdBufferGraphicsBeginInfo**) &pNext->pNext;\n',
874 ' *ppNext = (VkCmdBufferGraphicsBeginInfo*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pNext->pNext);\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700875 ' break;\n',
876 ' }\n',
877 ' default:\n',
878 ' {\n',
879 ' glv_LogError("Encountered an unexpected type in begin command buffer list.\\n");\n',
880 ' pPacket->header = NULL;\n',
881 ' pNext->pNext = NULL;\n',
882 ' }\n',
Jon Ashburna02bc242015-01-02 18:28:26 -0700883 ' }\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600884 ' pNext = (VkCmdBufferGraphicsBeginInfo*)pNext->pNext;\n',
Jon Ashburna02bc242015-01-02 18:28:26 -0700885 ' }\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700886 '} else {\n',
887 ' // This is unexpected.\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600888 ' glv_LogError("BeginCommandBuffer must have BeginInfo stype of VK_STRUCTURE_TYPE_CMD_BUFFER_BEGIN_INFO.\\n");\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700889 ' pPacket->header = NULL;\n',
Jon Ashburna02bc242015-01-02 18:28:26 -0700890 '}']},
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600891 'AllocMemory' : {'param': 'pAllocInfo', 'txt': ['if (pPacket->pAllocInfo->sType == VK_STRUCTURE_TYPE_MEMORY_ALLOC_INFO) {\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600892 ' VkMemoryAllocInfo** ppNext = (VkMemoryAllocInfo**) &(pPacket->pAllocInfo->pNext);\n',
893 ' *ppNext = (VkMemoryAllocInfo*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pAllocInfo->pNext);\n',
894 ' VkMemoryAllocInfo* pNext = (VkMemoryAllocInfo*) *ppNext;\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700895 ' while (NULL != pNext)\n', ' {\n',
896 ' switch(pNext->sType)\n', ' {\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600897 ' case VK_STRUCTURE_TYPE_MEMORY_ALLOC_BUFFER_INFO:\n',
898 ' case VK_STRUCTURE_TYPE_MEMORY_ALLOC_IMAGE_INFO:\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700899 ' {\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600900 ' ppNext = (VkMemoryAllocInfo **) &(pNext->pNext);\n',
901 ' *ppNext = (VkMemoryAllocInfo*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pNext->pNext);\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700902 ' break;\n',
903 ' }\n',
904 ' default:\n',
905 ' {\n',
906 ' glv_LogError("Encountered an unexpected type alloc memory list.\\n");\n',
907 ' pPacket->header = NULL;\n',
908 ' pNext->pNext = NULL;\n',
909 ' }\n',
Jon Ashburn3039e9c2015-02-03 07:33:48 -0700910 ' }\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600911 ' pNext = (VkMemoryAllocInfo*)pNext->pNext;\n',
Jon Ashburn3039e9c2015-02-03 07:33:48 -0700912 ' }\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700913 '} else {\n',
914 ' // This is unexpected.\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600915 ' glv_LogError("AllocMemory must have AllocInfo stype of VK_STRUCTURE_TYPE_MEMORY_ALLOC_INFO.\\n");\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700916 ' pPacket->header = NULL;\n',
Jon Ashburn3039e9c2015-02-03 07:33:48 -0700917 '}']},
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600918 'UpdateDescriptors' : {'param': 'pUpdateChain', 'txt': ['VkUpdateSamplers* pNext = (VkUpdateSamplers*)pPacket->pUpdateChain;\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600919 'while ((NULL != pNext) && (VK_NULL_HANDLE != pNext))\n', '{\n',
Tobin Ehlis9570fc42015-02-04 10:53:31 -0700920 ' switch(pNext->sType)\n', ' {\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600921 ' case VK_STRUCTURE_TYPE_UPDATE_AS_COPY:\n',
Tobin Ehlis9570fc42015-02-04 10:53:31 -0700922 ' {\n',
923 ' void** ppNextVoidPtr = (void**)&pNext->pNext;\n',
924 ' *ppNextVoidPtr = (void*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pNext->pNext);\n',
925 ' break;\n',
926 ' }\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600927 ' case VK_STRUCTURE_TYPE_UPDATE_SAMPLERS:\n',
Tobin Ehlis9570fc42015-02-04 10:53:31 -0700928 ' {\n',
929 ' void** ppNextVoidPtr = (void**)&pNext->pNext;\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600930 ' VkUpdateSamplers* pUS = (VkUpdateSamplers*)pNext;\n',
Tobin Ehlis9570fc42015-02-04 10:53:31 -0700931 ' *ppNextVoidPtr = (void*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pNext->pNext);\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600932 ' pUS->pSamplers = (VK_SAMPLER*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pUS->pSamplers);\n',
Tobin Ehlis9570fc42015-02-04 10:53:31 -0700933 ' break;\n',
934 ' }\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600935 ' case VK_STRUCTURE_TYPE_UPDATE_SAMPLER_TEXTURES:\n',
Tobin Ehlis9570fc42015-02-04 10:53:31 -0700936 ' {\n',
937 ' void** ppNextVoidPtr = (void**)&pNext->pNext;\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600938 ' VkUpdateSamplerTextures* pUST = (VkUpdateSamplerTextures*)pNext;\n',
Tobin Ehlis9570fc42015-02-04 10:53:31 -0700939 ' *ppNextVoidPtr = (void*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pNext->pNext);\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600940 ' pUST->pSamplerImageViews = (VK_SAMPLER_IMAGE_VIEW_INFO*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pUST->pSamplerImageViews);\n',
Tobin Ehlis9570fc42015-02-04 10:53:31 -0700941 ' uint32_t i;\n',
942 ' for (i = 0; i < pUST->count; i++) {\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600943 ' VkImageViewAttachInfo** ppLocalImageView = (VkImageViewAttachInfo**)&pUST->pSamplerImageViews[i].pImageView;\n',
944 ' *ppLocalImageView = (VkImageViewAttachInfo*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pUST->pSamplerImageViews[i].pImageView);\n',
Tobin Ehlis9570fc42015-02-04 10:53:31 -0700945 ' }\n',
946 ' break;\n',
947 ' }\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600948 ' case VK_STRUCTURE_TYPE_UPDATE_IMAGES:\n',
Tobin Ehlis9570fc42015-02-04 10:53:31 -0700949 ' {\n',
950 ' void** ppNextVoidPtr = (void**)&pNext->pNext;\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600951 ' VkUpdateImages* pUI = (VkUpdateImages*)pNext;\n',
Tobin Ehlis9570fc42015-02-04 10:53:31 -0700952 ' *ppNextVoidPtr = (void*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pNext->pNext);\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600953 ' VkImageViewAttachInfo** ppLocalImageView = (VkImageViewAttachInfo**)&pUI->pImageViews;\n',
954 ' *ppLocalImageView = (VkImageViewAttachInfo*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pUI->pImageViews);\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600955 ' uint32_t i;\n',
956 ' for (i = 0; i < pUI->count; i++) {\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600957 ' VkImageViewAttachInfo** ppLocalImageViews = (VkImageViewAttachInfo**)&pUI->pImageViews[i];\n',
958 ' *ppLocalImageViews = (VkImageViewAttachInfo*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pUI->pImageViews[i]);\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600959 ' }\n',
Tobin Ehlis9570fc42015-02-04 10:53:31 -0700960 ' break;\n',
961 ' }\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600962 ' case VK_STRUCTURE_TYPE_UPDATE_BUFFERS:\n',
Tobin Ehlis9570fc42015-02-04 10:53:31 -0700963 ' {\n',
964 ' void** ppNextVoidPtr = (void**)&pNext->pNext;\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600965 ' VkUpdateBuffers* pUB = (VkUpdateBuffers*)pNext;\n',
Tobin Ehlis9570fc42015-02-04 10:53:31 -0700966 ' *ppNextVoidPtr = (void*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pNext->pNext);\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600967 ' VkBufferViewAttachInfo** ppLocalBufferView = (VkBufferViewAttachInfo**)&pUB->pBufferViews;\n',
968 ' *ppLocalBufferView = (VkBufferViewAttachInfo*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pUB->pBufferViews);\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600969 ' uint32_t i;\n',
970 ' for (i = 0; i < pUB->count; i++) {\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600971 ' VkBufferViewAttachInfo** ppLocalBufferViews = (VkBufferViewAttachInfo**)&pUB->pBufferViews[i];\n',
972 ' *ppLocalBufferViews = (VkBufferViewAttachInfo*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pUB->pBufferViews[i]);\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600973 ' }\n',
Tobin Ehlis9570fc42015-02-04 10:53:31 -0700974 ' break;\n',
975 ' }\n',
976 ' default:\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700977 ' {\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600978 ' glv_LogError("Encountered an unexpected type in update descriptors pUpdateChain.\\n");\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700979 ' pPacket->header = NULL;\n',
980 ' pNext->pNext = NULL;\n',
981 ' }\n',
Tobin Ehlis9570fc42015-02-04 10:53:31 -0700982 ' }\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600983 ' pNext = (VkUpdateSamplers*)pNext->pNext;\n',
Tobin Ehlis9570fc42015-02-04 10:53:31 -0700984 '}']},
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600985 'CreateGraphicsPipeline' : {'param': 'pCreateInfo', 'txt': ['if (pPacket->pCreateInfo->sType == VK_STRUCTURE_TYPE_GRAPHICS_PIPELINE_CREATE_INFO) {\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700986 ' // need to make a non-const pointer to the pointer so that we can properly change the original pointer to the interpretted one\n',
987 ' void** ppNextVoidPtr = (void**)&pPacket->pCreateInfo->pNext;\n',
988 ' *ppNextVoidPtr = (void*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->pCreateInfo->pNext);\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -0600989 ' VkPipelineShaderStageCreateInfo* pNext = (VkPipelineShaderStageCreateInfo*)pPacket->pCreateInfo->pNext;\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600990 ' while ((NULL != pNext) && (VK_NULL_HANDLE != pNext))\n', '{\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700991 ' switch(pNext->sType)\n', ' {\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -0600992 ' case VK_STRUCTURE_TYPE_PIPELINE_IA_STATE_CREATE_INFO:\n',
993 ' case VK_STRUCTURE_TYPE_PIPELINE_TESS_STATE_CREATE_INFO:\n',
994 ' case VK_STRUCTURE_TYPE_PIPELINE_RS_STATE_CREATE_INFO:\n',
995 ' case VK_STRUCTURE_TYPE_PIPELINE_VP_STATE_CREATE_INFO:\n',
996 ' case VK_STRUCTURE_TYPE_PIPELINE_MS_STATE_CREATE_INFO:\n',
997 ' case VK_STRUCTURE_TYPE_PIPELINE_DS_STATE_CREATE_INFO:\n',
Peter Lohrmann78109572015-03-10 15:30:36 -0700998 ' {\n',
999 ' void** ppNextVoidPtr = (void**)&pNext->pNext;\n',
1000 ' *ppNextVoidPtr = (void*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pNext->pNext);\n',
1001 ' break;\n',
1002 ' }\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001003 ' case VK_STRUCTURE_TYPE_PIPELINE_CB_STATE_CREATE_INFO:\n',
Peter Lohrmann78109572015-03-10 15:30:36 -07001004 ' {\n',
1005 ' void** ppNextVoidPtr = (void**)&pNext->pNext;\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001006 ' VkPipelineCbStateCreateInfo *pCb = (VkPipelineCbStateCreateInfo *) pNext;\n',
Peter Lohrmann78109572015-03-10 15:30:36 -07001007 ' *ppNextVoidPtr = (void*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pNext->pNext);\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001008 ' pCb->pAttachments = (VkPipelineCbAttachmentState*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pCb->pAttachments);\n',
Peter Lohrmann78109572015-03-10 15:30:36 -07001009 ' break;\n',
1010 ' }\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001011 ' case VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO:\n',
Peter Lohrmann78109572015-03-10 15:30:36 -07001012 ' {\n',
1013 ' void** ppNextVoidPtr = (void**)&pNext->pNext;\n',
1014 ' *ppNextVoidPtr = (void*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pNext->pNext);\n',
1015 ' interpret_pipeline_shader(pHeader, &pNext->shader);\n',
1016 ' break;\n',
1017 ' }\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001018 ' case VK_STRUCTURE_TYPE_PIPELINE_VERTEX_INPUT_CREATE_INFO:\n',
Peter Lohrmann78109572015-03-10 15:30:36 -07001019 ' {\n',
1020 ' void** ppNextVoidPtr = (void**)&pNext->pNext;\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001021 ' VkPipelineVertexInputCreateInfo *pVi = (VkPipelineVertexInputCreateInfo *) pNext;\n',
Peter Lohrmann78109572015-03-10 15:30:36 -07001022 ' *ppNextVoidPtr = (void*)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pNext->pNext);\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001023 ' pVi->pVertexBindingDescriptions = (VkVertexInputBindingDescription*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pVi->pVertexBindingDescriptions);\n',
1024 ' pVi->pVertexAttributeDescriptions = (VkVertexInputAttributeDescription*) glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pVi->pVertexAttributeDescriptions);\n',
Peter Lohrmann78109572015-03-10 15:30:36 -07001025 ' break;\n',
1026 ' }\n',
1027 ' default:\n',
1028 ' {\n',
1029 ' glv_LogError("Encountered an unexpected type in pipeline state list.\\n");\n',
1030 ' pPacket->header = NULL;\n',
1031 ' pNext->pNext = NULL;\n',
1032 ' }\n',
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001033 ' }\n',
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001034 ' pNext = (VkPipelineShaderStageCreateInfo*)pNext->pNext;\n',
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001035 ' }\n',
Peter Lohrmann78109572015-03-10 15:30:36 -07001036 '} else {\n',
1037 ' // This is unexpected.\n',
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001038 ' glv_LogError("CreateGraphicsPipeline must have CreateInfo stype of VK_STRUCTURE_TYPE_GRAPHICS_PIPELINE_CREATE_INFO.\\n");\n',
Peter Lohrmann78109572015-03-10 15:30:36 -07001039 ' pPacket->header = NULL;\n',
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001040 '}']},
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001041 'CreateComputePipeline' : {'param': 'pCreateInfo', 'txt': ['interpret_pipeline_shader(pHeader, (VkPipelineShader*)(&pPacket->pCreateInfo->cs));']}}
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001042 if_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001043 if_body.append('typedef struct struct_vkApiVersion {')
Jon Ashburne2248392014-12-16 18:37:04 -07001044 if_body.append(' glv_trace_packet_header* header;')
1045 if_body.append(' uint32_t version;')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001046 if_body.append('} struct_vkApiVersion;\n')
1047 if_body.append('static struct_vkApiVersion* interpret_body_as_vkApiVersion(glv_trace_packet_header* pHeader, BOOL check_version)')
Jon Ashburne2248392014-12-16 18:37:04 -07001048 if_body.append('{')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001049 if_body.append(' struct_vkApiVersion* pPacket = (struct_vkApiVersion*)pHeader->pBody;')
Jon Ashburne2248392014-12-16 18:37:04 -07001050 if_body.append(' pPacket->header = pHeader;')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001051 if_body.append(' if (check_version && pPacket->version != VK_API_VERSION)')
1052 if_body.append(' glv_LogError("Trace file from older VK version 0x%x, vk replayer built from version 0x%x, replayer may fail\\n", pPacket->version, VK_API_VERSION);')
Jon Ashburne2248392014-12-16 18:37:04 -07001053 if_body.append(' return pPacket;')
1054 if_body.append('}\n')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001055 for proto in self.protos:
1056 if 'Wsi' not in proto.name and 'Dbg' not in proto.name:
1057 if 'UnmapMemory' == proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001058 proto.params.append(vulkan.Param("void*", "pData"))
1059 if_body.append('typedef struct struct_vk%s {' % proto.name)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001060 if_body.append(' glv_trace_packet_header* header;')
1061 for p in proto.params:
1062 if '[4]' in p.ty:
1063 if_body.append(' %s %s[4];' % (p.ty.strip('[4]'), p.name))
1064 else:
1065 if_body.append(' %s %s;' % (p.ty, p.name))
Mark Lobodzinskie2d07a52015-01-29 08:55:56 -06001066 if 'void' != proto.ret:
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001067 if_body.append(' %s result;' % proto.ret)
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001068 if_body.append('} struct_vk%s;\n' % proto.name)
1069 if_body.append('static struct_vk%s* interpret_body_as_vk%s(glv_trace_packet_header* pHeader)' % (proto.name, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001070 if_body.append('{')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001071 if_body.append(' struct_vk%s* pPacket = (struct_vk%s*)pHeader->pBody;' % (proto.name, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001072 if_body.append(' pPacket->header = pHeader;')
1073 for p in proto.params:
1074 if '*' in p.ty:
Courtney Goeltzenleuchterddcb6192015-04-14 18:48:46 -06001075 if 'DeviceCreateInfo' in p.ty:
1076 if_body.append(' pPacket->%s = interpret_VkDeviceCreateInfo(pHeader, (intptr_t)pPacket->%s);' % (p.name, p.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001077 else:
1078 if_body.append(' pPacket->%s = (%s)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->%s);' % (p.name, p.ty, p.name))
Tobin Ehlis81b1b3d2015-03-10 11:04:17 -06001079 # TODO : Generalize this custom code to kill dict data struct above.
1080 # Really the point of this block is to catch params w/ embedded ptrs to structs and chains of structs
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001081 if proto.name in custom_case_dict and p.name == custom_case_dict[proto.name]['param']:
1082 if_body.append(' if (pPacket->%s != NULL)' % custom_case_dict[proto.name]['param'])
1083 if_body.append(' {')
1084 if_body.append(' %s' % " ".join(custom_case_dict[proto.name]['txt']))
1085 if_body.append(' }')
1086 if_body.append(' return pPacket;')
1087 if_body.append('}\n')
1088 return "\n".join(if_body)
1089
1090 def _generate_interp_funcs_ext(self, func_class='Wsi'):
1091 if_body = []
1092 for proto in self.protos:
1093 if func_class in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001094 if_body.append('typedef struct struct_vk%s {' % proto.name)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001095 if_body.append(' glv_trace_packet_header* pHeader;')
1096 for p in proto.params:
1097 if_body.append(' %s %s;' % (p.ty, p.name))
Mark Lobodzinskie2d07a52015-01-29 08:55:56 -06001098 if 'void' != proto.ret:
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001099 if_body.append(' %s result;' % proto.ret)
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001100 if_body.append('} struct_vk%s;\n' % proto.name)
1101 if_body.append('static struct_vk%s* interpret_body_as_vk%s(glv_trace_packet_header* pHeader)' % (proto.name, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001102 if_body.append('{')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001103 if_body.append(' struct_vk%s* pPacket = (struct_vk%s*)pHeader->pBody;' % (proto.name, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001104 if_body.append(' pPacket->pHeader = pHeader;')
1105 for p in proto.params:
1106 if '*' in p.ty:
1107 if_body.append(' pPacket->%s = (%s)glv_trace_packet_interpret_buffer_pointer(pHeader, (intptr_t)pPacket->%s);' % (p.name, p.ty, p.name))
1108 if_body.append(' return pPacket;')
1109 if_body.append('}\n')
1110 return "\n".join(if_body)
1111
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001112 def _generate_replay_func_ptrs(self):
1113 xf_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001114 xf_body.append('struct vkFuncs {')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001115 xf_body.append(' void init_funcs(void * libHandle);')
1116 xf_body.append(' void *m_libHandle;\n')
1117 for proto in self.protos:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001118 xf_body.append(' typedef %s( VKAPI * type_vk%s)(' % (proto.ret, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001119 for p in proto.params:
1120 if '[4]' in p.ty:
1121 xf_body.append(' %s %s[4],' % (p.ty.strip('[4]'), p.name))
1122 else:
1123 xf_body.append(' %s %s,' % (p.ty, p.name))
1124 xf_body[-1] = xf_body[-1].replace(',', ');')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001125 xf_body.append(' type_vk%s real_vk%s;' % (proto.name, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001126 xf_body.append('};')
1127 return "\n".join(xf_body)
1128
1129 def _map_decl(self, type1, type2, name):
1130 return ' std::map<%s, %s> %s;' % (type1, type2, name)
1131
1132 def _add_to_map_decl(self, type1, type2, name):
1133 txt = ' void add_to_map(%s* pTraceVal, %s* pReplayVal)\n {\n' % (type1, type2)
1134 txt += ' assert(pTraceVal != NULL);\n'
1135 txt += ' assert(pReplayVal != NULL);\n'
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001136 txt += ' %s[*pTraceVal] = *pReplayVal;\n }\n' % name
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001137 return txt
1138
1139 def _rm_from_map_decl(self, ty, name):
1140 txt = ' void rm_from_map(const %s& key)\n {\n' % (ty)
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001141 txt += ' %s.erase(key);\n }\n' % name
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001142 return txt
1143
1144 def _remap_decl(self, ty, name):
1145 txt = ' %s remap(const %s& value)\n {\n' % (ty, ty)
1146 txt += ' std::map<%s, %s>::const_iterator q = %s.find(value);\n' % (ty, ty, name)
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001147 txt += ' return (q == %s.end()) ? VK_NULL_HANDLE : q->second;\n }\n' % name
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001148 return txt
1149
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001150 def _generate_replay_objMemory_funcs(self):
1151 rof_body = []
1152 # Custom code for memory mapping functions for app writes into mapped memory
1153 rof_body.append('// memory mapping functions for app writes into mapped memory')
1154 rof_body.append(' bool isPendingAlloc()')
1155 rof_body.append(' {')
1156 rof_body.append(' return m_pendingAlloc;')
1157 rof_body.append(' }')
1158 rof_body.append('')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001159 rof_body.append(' void setAllocInfo(const VkMemoryAllocInfo *info, const bool pending)')
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001160 rof_body.append(' {')
1161 rof_body.append(' m_pendingAlloc = pending;')
1162 rof_body.append(' m_allocInfo = *info;')
1163 rof_body.append(' }')
1164 rof_body.append('')
1165 rof_body.append(' void setMemoryDataAddr(void *pBuf)')
1166 rof_body.append(' {')
1167 rof_body.append(' if (m_mapRange.empty())')
1168 rof_body.append(' {')
1169 rof_body.append(' glv_LogError("gpuMemory::setMemoryDataAddr() m_mapRange is empty\\n");')
1170 rof_body.append(' return;')
1171 rof_body.append(' }')
1172 rof_body.append(' MapRange mr = m_mapRange.back();')
1173 rof_body.append(' if (mr.pData != NULL)')
1174 rof_body.append(' glv_LogWarn("gpuMemory::setMemoryDataAddr() data already mapped overwrite old mapping\\n");')
1175 rof_body.append(' else if (pBuf == NULL)')
1176 rof_body.append(' glv_LogWarn("gpuMemory::setMemoryDataAddr() adding NULL pointer\\n");')
1177 rof_body.append(' mr.pData = pBuf;')
1178 rof_body.append(' }')
1179 rof_body.append('')
1180 rof_body.append(' void setMemoryMapRange(void *pBuf, const size_t size, const size_t offset, const bool pending)')
1181 rof_body.append(' {')
1182 rof_body.append(' MapRange mr;')
1183 rof_body.append(' mr.pData = pBuf;')
1184 rof_body.append(' mr.size = size;')
1185 rof_body.append(' mr.offset = offset;')
1186 rof_body.append(' mr.pending = pending;')
1187 rof_body.append(' m_mapRange.push_back(mr);')
1188 rof_body.append(' }')
1189 rof_body.append('')
1190 rof_body.append(' void copyMappingData(const void* pSrcData)')
1191 rof_body.append(' {')
1192 rof_body.append(' if (m_mapRange.empty())')
1193 rof_body.append(' {')
1194 rof_body.append(' glv_LogError("gpuMemory::copyMappingData() m_mapRange is empty\\n");')
1195 rof_body.append(' return;')
1196 rof_body.append(' }')
1197 rof_body.append(' MapRange mr = m_mapRange.back();')
1198 rof_body.append(' if (!pSrcData || !mr.pData)')
1199 rof_body.append(' {')
1200 rof_body.append(' if (!pSrcData)')
1201 rof_body.append(' glv_LogError("gpuMemory::copyMappingData() null src pointer\\n");')
1202 rof_body.append(' else')
1203 rof_body.append(' glv_LogError("gpuMemory::copyMappingData() null dest pointer size=%u\\n", m_allocInfo.allocationSize);')
1204 rof_body.append(' m_mapRange.pop_back();')
1205 rof_body.append(' return;')
1206 rof_body.append(' }')
1207 rof_body.append(' memcpy(mr.pData, pSrcData, m_allocInfo.allocationSize);')
1208 rof_body.append(' if (!mr.pending)')
1209 rof_body.append(' m_mapRange.pop_back();')
1210 rof_body.append(' }')
1211 rof_body.append('')
1212 rof_body.append(' size_t getMemoryMapSize()')
1213 rof_body.append(' {')
1214 rof_body.append(' return (!m_mapRange.empty()) ? m_mapRange.back().size : 0;')
1215 rof_body.append(' }\n')
1216 return "\n".join(rof_body)
1217
Peter Lohrmann75728222015-04-02 11:45:31 -07001218 def _generate_replay_objmapper_class(self):
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001219 # Create dict mapping member var names to VK type (i.e. 'm_imageViews' : 'VkImage_VIEW')
Tobin Ehlis2012fce2015-01-15 17:53:54 -07001220 obj_map_dict = {}
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001221 for ty in vulkan.object_type_list:
1222 if ty in vulkan.object_parent_list:
Tobin Ehlis2012fce2015-01-15 17:53:54 -07001223 continue
Peter Lohrmanne70d1392015-04-15 12:45:51 -07001224 if (ty.startswith('Vk')):
1225 mem_var = ty.replace('Vk', '').lower()
Tobin Ehlis2012fce2015-01-15 17:53:54 -07001226 mem_var_list = mem_var.split('_')
Tobin Ehlis2012fce2015-01-15 17:53:54 -07001227 mem_var = 'm_%s%ss' % (mem_var_list[0], "".join([m.title() for m in mem_var_list[1:]]))
1228 obj_map_dict[mem_var] = ty
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001229 rc_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001230 rc_body.append('typedef struct _VKAllocInfo {')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001231 rc_body.append(' VkGpuSize size;')
Peter Lohrmann3f0d6972015-04-01 18:12:34 -07001232 rc_body.append(' void *pData;')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001233 rc_body.append('} VKAllocInfo;')
Peter Lohrmann75728222015-04-02 11:45:31 -07001234 rc_body.append('')
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001235 rc_body.append('class objMemory {')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001236 rc_body.append('public:')
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001237 rc_body.append(' objMemory() : m_numAllocations(0), m_pMemReqs(NULL) {}')
1238 rc_body.append(' ~objMemory() { free(m_pMemReqs);}')
1239 rc_body.append(' void setCount(const uint32_t num)')
1240 rc_body.append(' {')
1241 rc_body.append(' m_numAllocations = num;')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001242 rc_body.append(' }\n')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001243 rc_body.append(' void setReqs(const VkMemoryRequirements *pReqs, const uint32_t num)')
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001244 rc_body.append(' {')
1245 rc_body.append(' if (m_numAllocations != num && m_numAllocations != 0)')
1246 rc_body.append(' glv_LogError("objMemory::setReqs, internal mismatch on number of allocations");')
1247 rc_body.append(' if (m_pMemReqs == NULL && pReqs != NULL)')
1248 rc_body.append(' {')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001249 rc_body.append(' m_pMemReqs = (VkMemoryRequirements *) glv_malloc(num * sizeof(VkMemoryRequirements));')
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001250 rc_body.append(' if (m_pMemReqs == NULL)')
1251 rc_body.append(' {')
1252 rc_body.append(' glv_LogError("objMemory::setReqs out of memory");')
1253 rc_body.append(' return;')
1254 rc_body.append(' }')
1255 rc_body.append(' memcpy(m_pMemReqs, pReqs, num);')
1256 rc_body.append(' }')
1257 rc_body.append(' }\n')
1258 rc_body.append('private:')
1259 rc_body.append(' uint32_t m_numAllocations;')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001260 rc_body.append(' VkMemoryRequirements *m_pMemReqs;')
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001261 rc_body.append('};')
1262 rc_body.append('')
1263 rc_body.append('class gpuMemory {')
1264 rc_body.append('public:')
1265 rc_body.append(' gpuMemory() : m_pendingAlloc(false) {m_allocInfo.allocationSize = 0;}')
1266 rc_body.append(' ~gpuMemory() {}')
1267 rc_body.append(self._generate_replay_objMemory_funcs())
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001268 rc_body.append('private:')
1269 rc_body.append(' bool m_pendingAlloc;')
1270 rc_body.append(' struct MapRange {')
1271 rc_body.append(' bool pending;')
1272 rc_body.append(' size_t size;')
1273 rc_body.append(' size_t offset;')
1274 rc_body.append(' void* pData;')
1275 rc_body.append(' };')
1276 rc_body.append(' std::vector<MapRange> m_mapRange;')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001277 rc_body.append(' VkMemoryAllocInfo m_allocInfo;')
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001278 rc_body.append('};')
1279 rc_body.append('')
1280 rc_body.append('typedef struct _imageObj {')
1281 rc_body.append(' objMemory imageMem;')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001282 rc_body.append(' VkImage replayImage;')
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001283 rc_body.append(' } imageObj;')
1284 rc_body.append('')
1285 rc_body.append('typedef struct _bufferObj {')
1286 rc_body.append(' objMemory bufferMem;')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001287 rc_body.append(' VkBuffer replayBuffer;')
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001288 rc_body.append(' } bufferObj;')
1289 rc_body.append('')
1290 rc_body.append('typedef struct _gpuMemObj {')
1291 rc_body.append(' gpuMemory *pGpuMem;')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001292 rc_body.append(' VkGpuMemory replayGpuMem;')
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001293 rc_body.append(' } gpuMemObj;')
1294 rc_body.append('')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001295 rc_body.append('class vkReplayObjMapper {')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001296 rc_body.append('public:')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001297 rc_body.append(' vkReplayObjMapper() {}')
1298 rc_body.append(' ~vkReplayObjMapper() {}')
Peter Lohrmann75728222015-04-02 11:45:31 -07001299 rc_body.append('')
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001300 rc_body.append(' bool m_adjustForGPU; // true if replay adjusts behavior based on GPU')
1301 # Code for memory objects for handling replay GPU != trace GPU object memory requirements
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001302 rc_body.append(' void init_objMemCount(const VkBaseObject& object, const uint32_t &num)\n {')
1303 rc_body.append(' VkImage img = static_cast <VkImage> (object);')
1304 rc_body.append(' std::map<VkImage, imageObj>::const_iterator it = m_images.find(img);')
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001305 rc_body.append(' if (it != m_images.end())')
1306 rc_body.append(' {')
1307 rc_body.append(' objMemory obj = it->second.imageMem;')
1308 rc_body.append(' obj.setCount(num);')
1309 rc_body.append(' return;')
1310 rc_body.append(' }')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001311 rc_body.append(' VkBuffer buf = static_cast <VkBuffer> (object);')
1312 rc_body.append(' std::map<VkBuffer, bufferObj>::const_iterator itb = m_buffers.find(buf);')
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001313 rc_body.append(' if (itb != m_buffers.end())')
1314 rc_body.append(' {')
1315 rc_body.append(' objMemory obj = itb->second.bufferMem;')
1316 rc_body.append(' obj.setCount(num);')
1317 rc_body.append(' return;')
1318 rc_body.append(' }')
1319 rc_body.append(' return;')
1320 rc_body.append(' }\n')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001321 rc_body.append(' void init_objMemReqs(const VkBaseObject& object, const VkMemoryRequirements *pMemReqs, const unsigned int num)\n {')
1322 rc_body.append(' VkImage img = static_cast <VkImage> (object);')
1323 rc_body.append(' std::map<VkImage, imageObj>::const_iterator it = m_images.find(img);')
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001324 rc_body.append(' if (it != m_images.end())')
1325 rc_body.append(' {')
1326 rc_body.append(' objMemory obj = it->second.imageMem;')
1327 rc_body.append(' obj.setReqs(pMemReqs, num);')
1328 rc_body.append(' return;')
1329 rc_body.append(' }')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001330 rc_body.append(' VkBuffer buf = static_cast <VkBuffer> (object);')
1331 rc_body.append(' std::map<VkBuffer, bufferObj>::const_iterator itb = m_buffers.find(buf);')
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001332 rc_body.append(' if (itb != m_buffers.end())')
1333 rc_body.append(' {')
1334 rc_body.append(' objMemory obj = itb->second.bufferMem;')
1335 rc_body.append(' obj.setReqs(pMemReqs, num);')
1336 rc_body.append(' return;')
1337 rc_body.append(' }')
1338 rc_body.append(' return;')
Peter Lohrmann75728222015-04-02 11:45:31 -07001339 rc_body.append(' }')
1340 rc_body.append('')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001341 rc_body.append(' void clear_all_map_handles()\n {')
1342 for var in sorted(obj_map_dict):
1343 rc_body.append(' %s.clear();' % var)
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001344 rc_body.append(' }\n')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001345 for var in sorted(obj_map_dict):
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001346 if obj_map_dict[var] == 'VkImage':
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001347 rc_body.append(self._map_decl(obj_map_dict[var], 'imageObj', var))
1348 rc_body.append(self._add_to_map_decl(obj_map_dict[var], 'imageObj', var))
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001349 rc_body.append(self._rm_from_map_decl(obj_map_dict[var], var))
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001350 rc_body.append(' VkImage remap(const VkImage& value)')
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001351 rc_body.append(' {')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001352 rc_body.append(' std::map<VkImage, imageObj>::const_iterator q = m_images.find(value);')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001353 rc_body.append(' return (q == m_images.end()) ? VK_NULL_HANDLE : q->second.replayImage;')
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001354 rc_body.append(' }\n')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001355 elif obj_map_dict[var] == 'VkBuffer':
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001356 rc_body.append(self._map_decl(obj_map_dict[var], 'bufferObj', var))
1357 rc_body.append(self._add_to_map_decl(obj_map_dict[var], 'bufferObj', var))
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001358 rc_body.append(self._rm_from_map_decl(obj_map_dict[var], var))
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001359 rc_body.append(' VkBuffer remap(const VkBuffer& value)')
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001360 rc_body.append(' {')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001361 rc_body.append(' std::map<VkBuffer, bufferObj>::const_iterator q = m_buffers.find(value);')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001362 rc_body.append(' return (q == m_buffers.end()) ? VK_NULL_HANDLE : q->second.replayBuffer;')
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001363 rc_body.append(' }\n')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001364 elif obj_map_dict[var] == 'VkGpuMemory':
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001365 rc_body.append(self._map_decl(obj_map_dict[var], 'gpuMemObj', var))
1366 rc_body.append(self._add_to_map_decl(obj_map_dict[var], 'gpuMemObj', var))
Jon Ashburn16239cd2015-03-24 11:05:02 -06001367 rc_body.append(self._rm_from_map_decl(obj_map_dict[var], var))
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001368 rc_body.append(' VkGpuMemory remap(const VkGpuMemory& value)')
Jon Ashburn16239cd2015-03-24 11:05:02 -06001369 rc_body.append(' {')
Peter Lohrmanne70d1392015-04-15 12:45:51 -07001370 rc_body.append(' std::map<VkGpuMemory, gpuMemObj>::const_iterator q = m_gpumemorys.find(value);')
1371 rc_body.append(' return (q == m_gpumemorys.end()) ? VK_NULL_HANDLE : q->second.replayGpuMem;')
Jon Ashburn16239cd2015-03-24 11:05:02 -06001372 rc_body.append(' }\n')
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001373 else:
1374 rc_body.append(self._map_decl(obj_map_dict[var], obj_map_dict[var], var))
1375 rc_body.append(self._add_to_map_decl(obj_map_dict[var], obj_map_dict[var], var))
1376 rc_body.append(self._rm_from_map_decl(obj_map_dict[var], var))
1377 rc_body.append(self._remap_decl(obj_map_dict[var], var))
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001378 # VkDynamicStateObject code
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001379 state_obj_remap_types = vulkan.object_dynamic_state_list
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001380 rc_body.append(' VkDynamicStateObject remap(const VkDynamicStateObject& state)\n {')
1381 rc_body.append(' VkDynamicStateObject obj;')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001382 for t in state_obj_remap_types:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001383 rc_body.append(' if ((obj = remap(static_cast <%s> (state))) != VK_NULL_HANDLE)' % t)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001384 rc_body.append(' return obj;')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001385 rc_body.append(' return VK_NULL_HANDLE;\n }')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001386 rc_body.append(' void rm_from_map(const VkDynamicStateObject& state)\n {')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001387 for t in state_obj_remap_types:
1388 rc_body.append(' rm_from_map(static_cast <%s> (state));' % t)
1389 rc_body.append(' }')
Peter Lohrmann75728222015-04-02 11:45:31 -07001390 rc_body.append('')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001391 # OBJECT code
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001392 rc_body.append(' VkObject remap(const VkObject& object)\n {')
1393 rc_body.append(' VkObject obj;')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001394 obj_remap_types = vulkan.object_list
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001395 for var in obj_remap_types:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001396 rc_body.append(' if ((obj = remap(static_cast <%s> (object))) != VK_NULL_HANDLE)' % (var))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001397 rc_body.append(' return obj;')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001398 rc_body.append(' return VK_NULL_HANDLE;\n }')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001399 rc_body.append(' void rm_from_map(const VkObject & objKey)\n {')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001400 for var in obj_remap_types:
1401 rc_body.append(' rm_from_map(static_cast <%s> (objKey));' % (var))
1402 rc_body.append(' }')
Tobin Ehlis3ef02702015-04-15 11:51:51 -06001403 rc_body.append(' VkBaseObject remap(const VkBaseObject& object)\n {')
1404 rc_body.append(' VkBaseObject obj;')
Peter Lohrmanne70d1392015-04-15 12:45:51 -07001405 base_obj_remap_types = ['VkDevice', 'VkQueue', 'VkGpuMemory', 'VkObject']
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001406 for t in base_obj_remap_types:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001407 rc_body.append(' if ((obj = remap(static_cast <%s> (object))) != VK_NULL_HANDLE)' % t)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001408 rc_body.append(' return obj;')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001409 rc_body.append(' return VK_NULL_HANDLE;')
Tony Barbourb30dcd42015-02-02 13:21:18 -07001410 rc_body.append(' }')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001411 rc_body.append('};')
1412 return "\n".join(rc_body)
1413
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001414 def _generate_replay_init_funcs(self):
1415 rif_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001416 rif_body.append('void vkFuncs::init_funcs(void * handle)\n{\n m_libHandle = handle;')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001417 for proto in self.protos:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001418 rif_body.append(' real_vk%s = (type_vk%s)(glv_platform_get_library_entrypoint(handle, "vk%s"));' % (proto.name, proto.name, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001419 rif_body.append('}')
1420 return "\n".join(rif_body)
1421
1422 def _get_packet_param(self, t, n):
1423 # list of types that require remapping
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001424 remap_list = vulkan.object_type_list
Tobin Ehlis2012fce2015-01-15 17:53:54 -07001425 param_exclude_list = ['p1', 'p2', 'pGpus', 'pDescriptorSets']
1426 if t.strip('*').replace('const ', '') in remap_list and n not in param_exclude_list:
1427 if '*' in t:
1428 if 'const ' not in t:
Peter Lohrmann75728222015-04-02 11:45:31 -07001429 return 'm_objMapper.remap(*pPacket->%s)' % (n)
Tobin Ehlis2012fce2015-01-15 17:53:54 -07001430 else: # TODO : Don't remap array ptrs?
1431 return 'pPacket->%s' % (n)
Peter Lohrmann75728222015-04-02 11:45:31 -07001432 return 'm_objMapper.remap(pPacket->%s)' % (n)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001433 return 'pPacket->%s' % (n)
1434
Tobin Ehlis45bc7f82015-01-16 15:13:34 -07001435 def _gen_replay_enum_gpus(self):
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001436 ieg_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001437 ieg_body.append(' returnValue = manually_handle_vkEnumerateGpus(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001438 return "\n".join(ieg_body)
1439
1440 def _gen_replay_get_gpu_info(self):
1441 ggi_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001442 ggi_body.append(' returnValue = manually_handle_vkGetGpuInfo(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001443 return "\n".join(ggi_body)
1444
1445 def _gen_replay_create_device(self):
1446 cd_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001447 cd_body.append(' returnValue = manually_handle_vkCreateDevice(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001448 return "\n".join(cd_body)
1449
1450 def _gen_replay_get_extension_support(self):
1451 ges_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001452 ges_body.append(' returnValue = manually_handle_vkGetExtensionSupport(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001453 return "\n".join(ges_body)
1454
1455 def _gen_replay_queue_submit(self):
1456 qs_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001457 qs_body.append(' returnValue = manually_handle_vkQueueSubmit(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001458 return "\n".join(qs_body)
1459
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001460 def _gen_replay_get_object_info(self):
1461 goi_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001462 goi_body.append(' returnValue = manually_handle_vkGetObjectInfo(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001463 return "\n".join(goi_body)
1464
1465 def _gen_replay_get_format_info(self):
1466 gfi_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001467 gfi_body.append(' returnValue = manually_handle_vkGetFormatInfo(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001468 return "\n".join(gfi_body)
1469
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001470 def _gen_replay_create_image(self):
1471 ci_body = []
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001472 ci_body.append(' imageObj local_imageObj;')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001473 ci_body.append(' replayResult = m_vkFuncs.real_vkCreateImage(m_objMapper.remap(pPacket->device), pPacket->pCreateInfo, &local_imageObj.replayImage);')
1474 ci_body.append(' if (replayResult == VK_SUCCESS)')
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001475 ci_body.append(' {')
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001476 ci_body.append(' m_objMapper.add_to_map(pPacket->pImage, &local_imageObj);')
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001477 ci_body.append(' }')
1478 return "\n".join(ci_body)
1479
1480 def _gen_replay_create_buffer(self):
1481 cb_body = []
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001482 cb_body.append(' bufferObj local_bufferObj;')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001483 cb_body.append(' replayResult = m_vkFuncs.real_vkCreateBuffer(m_objMapper.remap(pPacket->device), pPacket->pCreateInfo, &local_bufferObj.replayBuffer);')
1484 cb_body.append(' if (replayResult == VK_SUCCESS)')
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001485 cb_body.append(' {')
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001486 cb_body.append(' m_objMapper.add_to_map(pPacket->pBuffer, &local_bufferObj);')
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001487 cb_body.append(' }')
1488 return "\n".join(cb_body)
1489
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001490 def _gen_replay_get_image_subresource_info(self):
1491 isi_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001492 isi_body.append(' returnValue = manually_handle_vkGetImageSubresourceInfo(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001493 return "\n".join(isi_body)
1494
Tobin Ehlisfc04b892015-01-22 12:29:31 -07001495 def _gen_replay_update_descriptors(self):
1496 ud_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001497 ud_body.append(' returnValue = manually_handle_vkUpdateDescriptors(pPacket);')
Tobin Ehlis8361b562015-02-03 14:41:26 -07001498 return "\n".join(ud_body)
Tobin Ehlisfc04b892015-01-22 12:29:31 -07001499
1500 def _gen_replay_create_descriptor_set_layout(self):
1501 cdsl_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001502 cdsl_body.append(' returnValue = manually_handle_vkCreateDescriptorSetLayout(pPacket);')
Tobin Ehlis8361b562015-02-03 14:41:26 -07001503 return "\n".join(cdsl_body)
Tobin Ehlisfc04b892015-01-22 12:29:31 -07001504
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001505 def _gen_replay_create_graphics_pipeline(self):
1506 cgp_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001507 cgp_body.append(' returnValue = manually_handle_vkCreateGraphicsPipeline(pPacket);')
Courtney Goeltzenleuchter32876a12015-03-25 15:37:49 -06001508 return "\n".join(cgp_body)
1509
Tobin Ehlis50990512015-02-05 11:29:45 -07001510 def _gen_replay_cmd_wait_events(self):
1511 cwe_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001512 cwe_body.append(' returnValue = manually_handle_vkCmdWaitEvents(pPacket);')
Tobin Ehlis50990512015-02-05 11:29:45 -07001513 return "\n".join(cwe_body)
1514
Jon Ashburnc46fc502015-02-10 10:36:22 -07001515 def _gen_replay_cmd_pipeline_barrier(self):
1516 cpb_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001517 cpb_body.append(' returnValue = manually_handle_vkCmdPipelineBarrier(pPacket);')
Jon Ashburnc46fc502015-02-10 10:36:22 -07001518 return "\n".join(cpb_body)
1519
Jon Ashburna02bc242015-01-02 18:28:26 -07001520 def _gen_replay_create_framebuffer(self):
1521 cf_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001522 cf_body.append(' returnValue = manually_handle_vkCreateFramebuffer(pPacket);')
Jon Ashburna02bc242015-01-02 18:28:26 -07001523 return "\n".join(cf_body)
1524
1525 def _gen_replay_create_renderpass(self):
1526 cr_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001527 cr_body.append(' returnValue = manually_handle_vkCreateRenderPass(pPacket);')
Jon Ashburna02bc242015-01-02 18:28:26 -07001528 return "\n".join(cr_body)
1529
1530 def _gen_replay_begin_command_buffer(self):
1531 bcb_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001532 bcb_body.append(' returnValue = manually_handle_vkBeginCommandBuffer(pPacket);')
Jon Ashburna02bc242015-01-02 18:28:26 -07001533 return "\n".join(bcb_body)
1534
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001535 def _gen_replay_store_pipeline(self):
1536 sp_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001537 sp_body.append(' returnValue = manually_handle_vkStorePipeline(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001538 return "\n".join(sp_body)
1539
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001540 def _gen_replay_get_multi_gpu_compatibility(self):
1541 gmgc_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001542 gmgc_body.append(' returnValue = manually_handle_vkGetMultiGpuCompatibility(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001543 return "\n".join(gmgc_body)
1544
1545 def _gen_replay_destroy_object(self):
1546 do_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001547 do_body.append(' returnValue = manually_handle_vkDestroyObject(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001548 return "\n".join(do_body)
1549
1550 def _gen_replay_wait_for_fences(self):
1551 wf_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001552 wf_body.append(' returnValue = manually_handle_vkWaitForFences(pPacket);')
Mark Lobodzinskiebe814d2015-04-07 16:07:57 -05001553 return "\n".join(wf_body)
1554
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001555 def _gen_replay_wsi_associate_connection(self):
1556 wac_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001557 wac_body.append(' returnValue = manually_handle_vkWsiX11AssociateConnection(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001558 return "\n".join(wac_body)
1559
1560 def _gen_replay_wsi_get_msc(self):
1561 wgm_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001562 wgm_body.append(' returnValue = manually_handle_vkWsiX11GetMSC(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001563 return "\n".join(wgm_body)
1564
1565 def _gen_replay_wsi_create_presentable_image(self):
1566 cpi_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001567 cpi_body.append(' returnValue = manually_handle_vkWsiX11CreatePresentableImage(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001568 return "\n".join(cpi_body)
1569
1570 def _gen_replay_wsi_queue_present(self):
1571 wqp_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001572 wqp_body.append(' returnValue = manually_handle_vkWsiX11QueuePresent(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001573 return "\n".join(wqp_body)
1574
Jon Ashburn16239cd2015-03-24 11:05:02 -06001575 def _gen_replay_alloc_memory(self):
1576 am_body = []
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001577 am_body.append(' gpuMemObj local_mem;')
1578 am_body.append(' if (!m_objMapper.m_adjustForGPU)')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001579 am_body.append(' replayResult = m_vkFuncs.real_vkAllocMemory(m_objMapper.remap(pPacket->device), pPacket->pAllocInfo, &local_mem.replayGpuMem);')
1580 am_body.append(' if (replayResult == VK_SUCCESS || m_objMapper.m_adjustForGPU)')
Jon Ashburn16239cd2015-03-24 11:05:02 -06001581 am_body.append(' {')
Jon Ashburncce1cb52015-03-26 16:15:18 -06001582 am_body.append(' local_mem.pGpuMem = new (gpuMemory);')
1583 am_body.append(' if (local_mem.pGpuMem)')
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001584 am_body.append(' local_mem.pGpuMem->setAllocInfo(pPacket->pAllocInfo, m_objMapper.m_adjustForGPU);')
1585 am_body.append(' m_objMapper.add_to_map(pPacket->pMem, &local_mem);')
Jon Ashburn16239cd2015-03-24 11:05:02 -06001586 am_body.append(' }')
1587 return "\n".join(am_body)
1588
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001589 def _gen_replay_free_memory(self):
1590 fm_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001591 fm_body.append(' returnValue = manually_handle_vkFreeMemory(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001592 return "\n".join(fm_body)
1593
1594 def _gen_replay_map_memory(self):
1595 mm_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001596 mm_body.append(' returnValue = manually_handle_vkMapMemory(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001597 return "\n".join(mm_body)
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001598
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001599 def _gen_replay_unmap_memory(self):
1600 um_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001601 um_body.append(' returnValue = manually_handle_vkUnmapMemory(pPacket);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001602 return "\n".join(um_body)
1603
Jon Ashburn16239cd2015-03-24 11:05:02 -06001604 def _gen_replay_pin_system_memory(self):
1605 psm_body = []
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001606 psm_body.append(' gpuMemObj local_mem;')
Jon Ashburnd4daf592015-03-27 16:23:47 -06001607 psm_body.append(' /* TODO do we need to skip (make pending) this call for m_adjustForGPU */')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001608 psm_body.append(' replayResult = m_vkFuncs.real_vkPinSystemMemory(m_objMapper.remap(pPacket->device), pPacket->pSysMem, pPacket->memSize, &local_mem.replayGpuMem);')
1609 psm_body.append(' if (replayResult == VK_SUCCESS)')
Peter Lohrmanne59ba282015-04-06 14:14:44 -07001610 psm_body.append(' m_objMapper.add_to_map(pPacket->pMem, &local_mem);')
Jon Ashburn16239cd2015-03-24 11:05:02 -06001611 return "\n".join(psm_body)
1612
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001613 # I don't think this function is being generated anymore (ie, it may have been removed from VK)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001614 def _gen_replay_bind_dynamic_memory_view(self):
1615 bdmv_body = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001616 bdmv_body.append(' VK_MEMORY_VIEW_ATTACH_INFO memView;')
1617 bdmv_body.append(' memcpy(&memView, pPacket->pMemView, sizeof(VK_MEMORY_VIEW_ATTACH_INFO));')
Peter Lohrmann75728222015-04-02 11:45:31 -07001618 bdmv_body.append(' memView.mem = m_objMapper.remap(pPacket->pMemView->mem);')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001619 bdmv_body.append(' m_vkFuncs.real_vkCmdBindDynamicMemoryView(m_objMapper.remap(pPacket->cmdBuffer), pPacket->pipelineBindPoint, &memView);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001620 return "\n".join(bdmv_body)
1621
Tobin Ehlis50990512015-02-05 11:29:45 -07001622 # Generate main replay case statements where actual replay API call is dispatched based on input packet data
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001623 def _generate_replay(self):
1624 # map protos to custom functions if body is fully custom
Tobin Ehlis45bc7f82015-01-16 15:13:34 -07001625 custom_body_dict = {'EnumerateGpus': self._gen_replay_enum_gpus,
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001626 'GetGpuInfo': self._gen_replay_get_gpu_info,
1627 'CreateDevice': self._gen_replay_create_device,
1628 'GetExtensionSupport': self._gen_replay_get_extension_support,
1629 'QueueSubmit': self._gen_replay_queue_submit,
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001630 'GetObjectInfo': self._gen_replay_get_object_info,
1631 'GetFormatInfo': self._gen_replay_get_format_info,
Jon Ashburn7f8acc72015-03-23 09:27:33 -06001632 'CreateImage': self._gen_replay_create_image,
1633 'CreateBuffer': self._gen_replay_create_buffer,
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001634 'GetImageSubresourceInfo': self._gen_replay_get_image_subresource_info,
1635 'CreateGraphicsPipeline': self._gen_replay_create_graphics_pipeline,
Jon Ashburna02bc242015-01-02 18:28:26 -07001636 'CreateFramebuffer': self._gen_replay_create_framebuffer,
1637 'CreateRenderPass': self._gen_replay_create_renderpass,
1638 'BeginCommandBuffer': self._gen_replay_begin_command_buffer,
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001639 'StorePipeline': self._gen_replay_store_pipeline,
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001640 'GetMultiGpuCompatibility': self._gen_replay_get_multi_gpu_compatibility,
1641 'DestroyObject': self._gen_replay_destroy_object,
1642 'WaitForFences': self._gen_replay_wait_for_fences,
1643 'WsiX11AssociateConnection': self._gen_replay_wsi_associate_connection,
1644 'WsiX11GetMSC': self._gen_replay_wsi_get_msc,
1645 'WsiX11CreatePresentableImage': self._gen_replay_wsi_create_presentable_image,
1646 'WsiX11QueuePresent': self._gen_replay_wsi_queue_present,
Jon Ashburn16239cd2015-03-24 11:05:02 -06001647 'AllocMemory': self._gen_replay_alloc_memory,
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001648 'FreeMemory': self._gen_replay_free_memory,
1649 'MapMemory': self._gen_replay_map_memory,
1650 'UnmapMemory': self._gen_replay_unmap_memory,
Jon Ashburn16239cd2015-03-24 11:05:02 -06001651 'PinSystemMemory': self._gen_replay_pin_system_memory,
Tobin Ehlis8361b562015-02-03 14:41:26 -07001652 'CmdBindDynamicMemoryView': self._gen_replay_bind_dynamic_memory_view,
1653 'UpdateDescriptors': self._gen_replay_update_descriptors,
Tobin Ehlis50990512015-02-05 11:29:45 -07001654 'CreateDescriptorSetLayout': self._gen_replay_create_descriptor_set_layout,
Jon Ashburnc46fc502015-02-10 10:36:22 -07001655 'CmdWaitEvents': self._gen_replay_cmd_wait_events,
1656 'CmdPipelineBarrier': self._gen_replay_cmd_pipeline_barrier}
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001657 # TODO : Need to guard CreateInstance with "if (!m_display->m_initedVK)" check
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001658 # Despite returning a value, don't check these funcs b/c custom code includes check already
Peter Lohrmann3f0d6972015-04-01 18:12:34 -07001659 custom_check_ret_val = ['EnumerateGpus', 'GetGpuInfo', 'CreateDevice', 'GetExtensionSupport', 'QueueSubmit', 'GetObjectInfo',
1660 'GetFormatInfo', 'GetImageSubresourceInfo', 'CreateDescriptorSetLayout', 'CreateGraphicsPipeline',
1661 'CreateFramebuffer', 'CreateRenderPass', 'BeginCommandBuffer', 'StorePipeline', 'GetMultiGpuCompatibility',
Peter Lohrmann95c369a2015-04-02 10:06:19 -07001662 'DestroyObject', 'WaitForFences', 'FreeMemory', 'MapMemory', 'UnmapMemory',
1663 'WsiX11AssociateConnection', 'WsiX11GetMSC', 'WsiX11CreatePresentableImage', 'WsiX11QueuePresent']
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001664 # multi-gpu Open funcs w/ list of local params to create
1665 custom_open_params = {'OpenSharedMemory': (-1,),
Peter Lohrmanne70d1392015-04-15 12:45:51 -07001666 'OpenSharedSemaphore': (-1,),
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001667 'OpenPeerMemory': (-1,),
1668 'OpenPeerImage': (-1, -2,)}
1669 # Functions that create views are unique from other create functions
Jon Ashburnb1b63ed2015-02-03 11:24:08 -07001670 create_view_list = ['CreateBufferView', 'CreateImageView', 'CreateColorAttachmentView', 'CreateDepthStencilView', 'CreateComputePipeline']
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001671 # Functions to treat as "Create' that don't have 'Create' in the name
Peter Lohrmanne70d1392015-04-15 12:45:51 -07001672 special_create_list = ['LoadPipeline', 'LoadPipelineDerivative', 'AllocMemory', 'GetDeviceQueue', 'PinSystemMemory', 'AllocDescriptorSets']
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001673 # A couple funcs use do while loops
Peter Lohrmanne70d1392015-04-15 12:45:51 -07001674 do_while_dict = {'GetFenceStatus': 'replayResult != pPacket->result && pPacket->result == VK_SUCCESS', 'GetEventStatus': '(pPacket->result == VK_EVENT_SET || pPacket->result == VK_EVENT_RESET) && replayResult != pPacket->result'}
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001675 rbody = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001676 rbody.append('glv_replay::GLV_REPLAY_RESULT vkReplay::replay(glv_trace_packet_header *packet)')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001677 rbody.append('{')
1678 rbody.append(' glv_replay::GLV_REPLAY_RESULT returnValue = glv_replay::GLV_REPLAY_SUCCESS;')
Courtney Goeltzenleuchter382489d2015-04-10 08:34:15 -06001679 rbody.append(' VkResult replayResult = VK_ERROR_UNKNOWN;')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001680 rbody.append(' switch (packet->packet_id)')
1681 rbody.append(' {')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001682 rbody.append(' case GLV_TPI_VK_vkApiVersion:')
Jon Ashburn6f4b3032015-02-03 08:57:28 -07001683 rbody.append(' break; // nothing to replay on the version packet')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001684 for proto in self.protos:
1685 ret_value = False
1686 create_view = False
1687 create_func = False
Tobin Ehlis2012fce2015-01-15 17:53:54 -07001688 # TODO : How to handle void* return of GetProcAddr?
1689 if ('void' not in proto.ret) and (proto.name not in custom_check_ret_val):
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001690 ret_value = True
1691 if proto.name in create_view_list:
1692 create_view = True
1693 elif 'Create' in proto.name or proto.name in special_create_list:
1694 create_func = True
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001695 rbody.append(' case GLV_TPI_VK_vk%s:' % proto.name)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001696 rbody.append(' {')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001697 rbody.append(' struct_vk%s* pPacket = (struct_vk%s*)(packet->pBody);' % (proto.name, proto.name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001698 if proto.name in custom_body_dict:
1699 rbody.append(custom_body_dict[proto.name]())
1700 else:
1701 if proto.name in custom_open_params:
Peter Lohrmanne70d1392015-04-15 12:45:51 -07001702 rbody.append(' VkDevice handle;')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001703 for pidx in custom_open_params[proto.name]:
Tobin Ehlis2012fce2015-01-15 17:53:54 -07001704 rbody.append(' %s local_%s;' % (proto.params[pidx].ty.replace('const ', '').strip('*'), proto.params[pidx].name))
Peter Lohrmann75728222015-04-02 11:45:31 -07001705 rbody.append(' handle = m_objMapper.remap(pPacket->device);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001706 elif create_view:
Tobin Ehlis2012fce2015-01-15 17:53:54 -07001707 rbody.append(' %s createInfo;' % (proto.params[1].ty.strip('*').replace('const ', '')))
1708 rbody.append(' memcpy(&createInfo, pPacket->pCreateInfo, sizeof(%s));' % (proto.params[1].ty.strip('*').replace('const ', '')))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001709 if 'CreateComputePipeline' == proto.name:
Peter Lohrmann75728222015-04-02 11:45:31 -07001710 rbody.append(' createInfo.cs.shader = m_objMapper.remap(pPacket->pCreateInfo->cs.shader);')
Jon Ashburnb1b63ed2015-02-03 11:24:08 -07001711 elif 'CreateBufferView' == proto.name:
Peter Lohrmann75728222015-04-02 11:45:31 -07001712 rbody.append(' createInfo.buffer = m_objMapper.remap(pPacket->pCreateInfo->buffer);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001713 else:
Peter Lohrmann75728222015-04-02 11:45:31 -07001714 rbody.append(' createInfo.image = m_objMapper.remap(pPacket->pCreateInfo->image);')
Tobin Ehlis2012fce2015-01-15 17:53:54 -07001715 rbody.append(' %s local_%s;' % (proto.params[-1].ty.strip('*').replace('const ', ''), proto.params[-1].name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001716 elif create_func: # Declare local var to store created handle into
Tobin Ehlis2012fce2015-01-15 17:53:54 -07001717 rbody.append(' %s local_%s;' % (proto.params[-1].ty.strip('*').replace('const ', ''), proto.params[-1].name))
Tobin Ehlis377b4622015-01-20 13:50:59 -07001718 if 'AllocDescriptorSets' == proto.name:
Jon Ashburn200ccb52015-02-04 12:57:25 -07001719 rbody.append(' %s local_%s[100];' % (proto.params[-2].ty.strip('*').replace('const ', ''), proto.params[-2].name))
Peter Lohrmanne70d1392015-04-15 12:45:51 -07001720 rbody.append(' VkDescriptorSetLayout localDescSets[100];')
Jon Ashburn200ccb52015-02-04 12:57:25 -07001721 rbody.append(' assert(pPacket->count <= 100);')
1722 rbody.append(' for (uint32_t i = 0; i < pPacket->count; i++)')
1723 rbody.append(' {')
Peter Lohrmann75728222015-04-02 11:45:31 -07001724 rbody.append(' localDescSets[i] = m_objMapper.remap(pPacket->%s[i]);' % (proto.params[-3].name))
Jon Ashburn200ccb52015-02-04 12:57:25 -07001725 rbody.append(' }')
1726 elif proto.name == 'ClearDescriptorSets':
Peter Lohrmanne70d1392015-04-15 12:45:51 -07001727 rbody.append(' VkDescriptorSet localDescSets[100];')
Jon Ashburn200ccb52015-02-04 12:57:25 -07001728 rbody.append(' assert(pPacket->count <= 100);')
1729 rbody.append(' for (uint32_t i = 0; i < pPacket->count; i++)')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001730 rbody.append(' {')
Peter Lohrmann75728222015-04-02 11:45:31 -07001731 rbody.append(' localDescSets[i] = m_objMapper.remap(pPacket->%s[i]);' % (proto.params[-1].name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001732 rbody.append(' }')
Peter Lohrmanne70d1392015-04-15 12:45:51 -07001733 elif proto.name == 'ResetFences':
1734 rbody.append(' VkFence* fences = GLV_NEW_ARRAY(VkFence, pPacket->fenceCount);')
1735 rbody.append(' for (uint32_t i = 0; i < pPacket->fenceCount; i++)')
1736 rbody.append(' {')
1737 rbody.append(' fences[i] = m_objMapper.remap(pPacket->%s[i]);' % (proto.params[-1].name))
1738 rbody.append(' }')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001739 elif proto.name in do_while_dict:
1740 rbody.append(' do {')
Jon Ashburnffd5f142015-02-03 13:39:05 -07001741 elif proto.name == 'EnumerateLayers':
1742 rbody.append(' char **bufptr = GLV_NEW_ARRAY(char *, pPacket->maxLayerCount);')
1743 rbody.append(' char **ptrLayers = (pPacket->pOutLayers == NULL) ? bufptr : (char **) pPacket->pOutLayers;')
1744 rbody.append(' for (unsigned int i = 0; i < pPacket->maxLayerCount; i++)')
1745 rbody.append(' bufptr[i] = GLV_NEW_ARRAY(char, pPacket->maxStringSize);')
Jon Ashburnd698ca22015-02-12 12:37:46 -07001746 elif proto.name == 'DestroyInstance':
Peter Lohrmanne70d1392015-04-15 12:45:51 -07001747 rbody.append(' vkDbgUnregisterMsgCallback(m_objMapper.remap(pPacket->instance), g_fpDbgMsgCallback);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001748 rr_string = ' '
1749 if ret_value:
1750 rr_string = ' replayResult = '
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001751 rr_string += 'm_vkFuncs.real_vk%s(' % proto.name
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001752 for p in proto.params:
1753 # For last param of Create funcs, pass address of param
Tobin Ehlis377b4622015-01-20 13:50:59 -07001754 if create_func:
1755 if p.name == proto.params[-1].name:
1756 rr_string += '&local_%s, ' % p.name
1757 elif proto.name == 'AllocDescriptorSets' and p.name == proto.params[-2].name:
1758 rr_string += 'local_%s, ' % p.name
1759 else:
1760 rr_string += '%s, ' % self._get_packet_param(p.ty, p.name)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001761 else:
1762 rr_string += '%s, ' % self._get_packet_param(p.ty, p.name)
1763 rr_string = '%s);' % rr_string[:-2]
Jon Ashburn200ccb52015-02-04 12:57:25 -07001764 if proto.name in custom_open_params:
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001765 rr_list = rr_string.split(', ')
Peter Lohrmann75728222015-04-02 11:45:31 -07001766 rr_list[0] = rr_list[0].replace('m_objMapper.remap(pPacket->device)', 'handle')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001767 for pidx in custom_open_params[proto.name]:
1768 rr_list[pidx] = '&local_%s' % proto.params[pidx].name
1769 rr_string = ', '.join(rr_list)
1770 rr_string += ');'
1771 elif create_view:
1772 rr_list = rr_string.split(', ')
1773 rr_list[-2] = '&createInfo'
1774 rr_list[-1] = '&local_%s);' % proto.params[-1].name
1775 rr_string = ', '.join(rr_list)
1776 # this is a sneaky shortcut to use generic create code below to add_to_map
1777 create_func = True
Jon Ashburnffd5f142015-02-03 13:39:05 -07001778 elif proto.name == 'EnumerateLayers':
1779 rr_string = rr_string.replace('pPacket->pOutLayers', 'ptrLayers')
Jon Ashburn200ccb52015-02-04 12:57:25 -07001780 elif proto.name == 'ClearDescriptorSets':
1781 rr_string = rr_string.replace('pPacket->pDescriptorSets', 'localDescSets')
1782 elif proto.name == 'AllocDescriptorSets':
1783 rr_string = rr_string.replace('pPacket->pSetLayouts', 'localDescSets')
Peter Lohrmanne70d1392015-04-15 12:45:51 -07001784 elif proto.name == 'ResetFences':
1785 rr_string = rr_string.replace('m_objMapper.remap(*pPacket->pFences)', 'fences')
1786
1787 # insert the real_*(..) call
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001788 rbody.append(rr_string)
Peter Lohrmanne70d1392015-04-15 12:45:51 -07001789
1790 # handle return values or anything that needs to happen after the real_*(..) call
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001791 if 'DestroyDevice' in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001792 rbody.append(' if (replayResult == VK_SUCCESS)')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001793 rbody.append(' {')
Peter Lohrmann53e89242015-02-27 15:35:15 -08001794 rbody.append(' m_pCBDump = NULL;')
1795 rbody.append(' m_pDSDump = NULL;')
Peter Lohrmanncaf39d52015-03-24 17:19:24 -07001796 rbody.append(' m_pGlvSnapshotPrint = NULL;')
Peter Lohrmann75728222015-04-02 11:45:31 -07001797 rbody.append(' m_objMapper.rm_from_map(pPacket->device);')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001798 rbody.append(' m_display->m_initedVK = false;')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001799 rbody.append(' }')
Jon Ashburn6f4b3032015-02-03 08:57:28 -07001800 if 'DestroyInstance' in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001801 rbody.append(' if (replayResult == VK_SUCCESS)')
Jon Ashburn6f4b3032015-02-03 08:57:28 -07001802 rbody.append(' {')
Jon Ashburn6f58a162015-02-03 09:17:12 -07001803 rbody.append(' // TODO need to handle multiple instances and only clearing maps within an instance.')
1804 rbody.append(' // TODO this only works with a single instance used at any given time.')
Peter Lohrmann75728222015-04-02 11:45:31 -07001805 rbody.append(' m_objMapper.clear_all_map_handles();')
Jon Ashburn6f4b3032015-02-03 08:57:28 -07001806 rbody.append(' }')
Tobin Ehlis377b4622015-01-20 13:50:59 -07001807 elif 'AllocDescriptorSets' in proto.name:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001808 rbody.append(' if (replayResult == VK_SUCCESS)')
Tobin Ehlis377b4622015-01-20 13:50:59 -07001809 rbody.append(' {')
1810 rbody.append(' for (uint32_t i = 0; i < local_pCount; i++) {')
Peter Lohrmann75728222015-04-02 11:45:31 -07001811 rbody.append(' m_objMapper.add_to_map(&pPacket->%s[i], &local_%s[i]);' % (proto.params[-2].name, proto.params[-2].name))
Tobin Ehlis377b4622015-01-20 13:50:59 -07001812 rbody.append(' }')
1813 rbody.append(' }')
Peter Lohrmanne70d1392015-04-15 12:45:51 -07001814 elif proto.name == 'ResetFences':
1815 rbody.append(' GLV_DELETE(fences);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001816 elif create_func: # save handle mapping if create successful
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001817 rbody.append(' if (replayResult == VK_SUCCESS)')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001818 rbody.append(' {')
Peter Lohrmann75728222015-04-02 11:45:31 -07001819 rbody.append(' m_objMapper.add_to_map(pPacket->%s, &local_%s);' % (proto.params[-1].name, proto.params[-1].name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001820 if 'AllocMemory' == proto.name:
Peter Lohrmann75728222015-04-02 11:45:31 -07001821 rbody.append(' m_objMapper.add_entry_to_mapData(local_%s, pPacket->pAllocInfo->allocationSize);' % (proto.params[-1].name))
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001822 rbody.append(' }')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001823 elif proto.name in do_while_dict:
1824 rbody[-1] = ' %s' % rbody[-1]
1825 rbody.append(' } while (%s);' % do_while_dict[proto.name])
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001826 rbody.append(' if (pPacket->result != VK_NOT_READY || replayResult != VK_SUCCESS)')
Jon Ashburnffd5f142015-02-03 13:39:05 -07001827 elif proto.name == 'EnumerateLayers':
1828 rbody.append(' for (unsigned int i = 0; i < pPacket->maxLayerCount; i++)')
1829 rbody.append(' GLV_DELETE(bufptr[i]);')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001830 if ret_value:
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001831 rbody.append(' CHECK_RETURN_VALUE(vk%s);' % proto.name)
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001832 if 'MsgCallback' in proto.name:
1833 rbody.pop()
1834 rbody.pop()
1835 rbody.pop()
1836 rbody.append(' // Just eating these calls as no way to restore dbg func ptr.')
1837 rbody.append(' break;')
1838 rbody.append(' }')
1839 rbody.append(' default:')
1840 rbody.append(' glv_LogWarn("Unrecognized packet_id %u, skipping\\n", packet->packet_id);')
1841 rbody.append(' returnValue = glv_replay::GLV_REPLAY_INVALID_ID;')
1842 rbody.append(' break;')
1843 rbody.append(' }')
1844 rbody.append(' return returnValue;')
1845 rbody.append('}')
1846 return "\n".join(rbody)
1847
1848class GlaveTraceHeader(Subcommand):
1849 def generate_header(self):
1850 header_txt = []
Peter Lohrmanncde614c2015-03-27 12:57:10 -07001851 header_txt.append('#include "glv_vk_vk_structs.h"')
1852 header_txt.append('#include "glv_vk_packet_id.h"\n')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001853 header_txt.append('void AttachHooks();')
1854 header_txt.append('void DetachHooks();')
Ian Elliottbc9ca5f2015-02-27 11:10:59 -07001855 header_txt.append('void InitTracer(void);\n')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001856 return "\n".join(header_txt)
1857
1858 def generate_body(self):
1859 body = [self._generate_trace_func_ptrs(),
Peter Lohrmann358d0092015-04-03 12:03:44 -07001860 self._generate_trace_func_protos(),
1861 self._generate_trace_real_func_ptr_protos()]
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001862
1863 return "\n".join(body)
1864
1865class GlaveTraceC(Subcommand):
1866 def generate_header(self):
1867 header_txt = []
1868 header_txt.append('#include "glv_platform.h"')
1869 header_txt.append('#include "glv_common.h"')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001870 header_txt.append('#include "glvtrace_vk_helpers.h"')
1871 header_txt.append('#include "glvtrace_vk_vk.h"')
1872 header_txt.append('#include "glvtrace_vk_vkdbg.h"')
1873 header_txt.append('#include "glvtrace_vk_vkwsix11ext.h"')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001874 header_txt.append('#include "glv_interconnect.h"')
1875 header_txt.append('#include "glv_filelike.h"')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001876 header_txt.append('#include "vk_struct_size_helper.h"')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001877 header_txt.append('#ifdef WIN32')
1878 header_txt.append('#include "mhook/mhook-lib/mhook.h"')
1879 header_txt.append('#endif')
1880 header_txt.append('#include "glv_trace_packet_utils.h"')
1881 header_txt.append('#include <stdio.h>\n')
1882 return "\n".join(header_txt)
1883
1884 def generate_body(self):
1885 body = [self._generate_func_ptr_assignments(),
1886 self._generate_attach_hooks(),
1887 self._generate_detach_hooks(),
Jon Ashburne2248392014-12-16 18:37:04 -07001888 self._generate_init_funcs(),
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001889 self._generate_trace_funcs()]
1890
1891 return "\n".join(body)
1892
1893class GlavePacketID(Subcommand):
1894 def generate_header(self):
1895 header_txt = []
1896 header_txt.append('#pragma once\n')
1897 header_txt.append('#include "glv_trace_packet_utils.h"')
1898 header_txt.append('#include "glv_trace_packet_identifiers.h"')
1899 header_txt.append('#include "glv_interconnect.h"')
Peter Lohrmanncde614c2015-03-27 12:57:10 -07001900 header_txt.append('#include "glv_vk_vk_structs.h"')
1901 header_txt.append('#include "glv_vk_vkdbg_structs.h"')
1902 header_txt.append('#include "glv_vk_vkwsix11ext_structs.h"')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001903 header_txt.append('#include "vk_enum_string_helper.h"')
Piers Danielle2bca482015-02-24 13:58:47 -07001904 header_txt.append('#if defined(WIN32)')
1905 header_txt.append('#define snprintf _snprintf')
1906 header_txt.append('#endif')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001907 header_txt.append('#define SEND_ENTRYPOINT_ID(entrypoint) ;')
1908 header_txt.append('//#define SEND_ENTRYPOINT_ID(entrypoint) glv_TraceInfo(#entrypoint "\\n");\n')
1909 header_txt.append('#define SEND_ENTRYPOINT_PARAMS(entrypoint, ...) ;')
1910 header_txt.append('//#define SEND_ENTRYPOINT_PARAMS(entrypoint, ...) glv_TraceInfo(entrypoint, __VA_ARGS__);\n')
1911 header_txt.append('#define CREATE_TRACE_PACKET(entrypoint, buffer_bytes_needed) \\')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001912 header_txt.append(' pHeader = glv_create_trace_packet(GLV_TID_VULKAN, GLV_TPI_VK_##entrypoint, sizeof(struct_##entrypoint), buffer_bytes_needed);\n')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001913 header_txt.append('#define FINISH_TRACE_PACKET() \\')
1914 header_txt.append(' glv_finalize_trace_packet(pHeader); \\')
1915 header_txt.append(' glv_write_trace_packet(pHeader, glv_trace_get_trace_file()); \\')
1916 header_txt.append(' glv_delete_trace_packet(&pHeader);')
1917 return "\n".join(header_txt)
1918
1919 def generate_body(self):
1920 body = [self._generate_packet_id_enum(),
1921 self._generate_stringify_func(),
1922 self._generate_interp_func()]
1923
1924 return "\n".join(body)
1925
1926class GlaveCoreStructs(Subcommand):
1927 def generate_header(self):
1928 header_txt = []
1929 header_txt.append('#pragma once\n')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001930 header_txt.append('#include "vulkan.h"')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001931 header_txt.append('#include "glv_trace_packet_utils.h"\n')
1932 return "\n".join(header_txt)
1933
1934 def generate_body(self):
1935 body = [self._generate_struct_util_funcs(),
1936 self._generate_interp_funcs()]
1937
1938 return "\n".join(body)
1939
1940class GlaveWsiHeader(Subcommand):
1941 def generate_header(self):
1942 header_txt = []
1943 header_txt.append('#pragma once\n')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001944 header_txt.append('#include "vulkan.h"')
Piers Danielle2bca482015-02-24 13:58:47 -07001945 header_txt.append('#if defined(PLATFORM_LINUX) || defined(XCB_NVIDIA)')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001946 header_txt.append('#include "vkWsiX11Ext.h"\n')
David Pinedo8e9cb3b2015-02-10 15:02:08 -07001947 header_txt.append('#else')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001948 header_txt.append('#include "vkWsiWinExt.h"')
David Pinedo8e9cb3b2015-02-10 15:02:08 -07001949 header_txt.append('#endif')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001950 header_txt.append('void AttachHooks_vkwsix11ext();')
1951 header_txt.append('void DetachHooks_vkwsix11ext();')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001952 return "\n".join(header_txt)
1953
1954 def generate_body(self):
1955 body = [self._generate_trace_func_ptrs_ext(),
1956 self._generate_trace_func_protos_ext()]
1957
1958 return "\n".join(body)
1959
1960class GlaveWsiC(Subcommand):
1961 def generate_header(self):
1962 header_txt = []
1963 header_txt.append('#include "glv_platform.h"')
1964 header_txt.append('#include "glv_common.h"')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001965 header_txt.append('#include "glvtrace_vk_vkwsix11ext.h"')
Peter Lohrmanncde614c2015-03-27 12:57:10 -07001966 header_txt.append('#include "glv_vk_vkwsix11ext_structs.h"')
1967 header_txt.append('#include "glv_vk_packet_id.h"')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001968 header_txt.append('#ifdef WIN32')
1969 header_txt.append('#include "mhook/mhook-lib/mhook.h"')
1970 header_txt.append('#endif')
1971 return "\n".join(header_txt)
1972
1973 def generate_body(self):
1974 body = [self._generate_func_ptr_assignments_ext(),
1975 self._generate_attach_hooks_ext(),
1976 self._generate_detach_hooks_ext(),
1977 self._generate_trace_funcs_ext()]
1978
1979 return "\n".join(body)
1980
1981class GlaveWsiStructs(Subcommand):
1982 def generate_header(self):
1983 header_txt = []
1984 header_txt.append('#pragma once\n')
Piers Danielle2bca482015-02-24 13:58:47 -07001985 header_txt.append('#if defined(PLATFORM_LINUX) || defined(XCB_NVIDIA)')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001986 header_txt.append('#include "vkWsiX11Ext.h"')
David Pinedo8e9cb3b2015-02-10 15:02:08 -07001987 header_txt.append('#else')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06001988 header_txt.append('#include "vkWsiWinExt.h"')
David Pinedo8e9cb3b2015-02-10 15:02:08 -07001989 header_txt.append('#endif')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07001990 header_txt.append('#include "glv_trace_packet_utils.h"\n')
1991 return "\n".join(header_txt)
1992
1993 def generate_body(self):
1994 body = [self._generate_interp_funcs_ext()]
1995
1996 return "\n".join(body)
1997
1998class GlaveDbgHeader(Subcommand):
1999 def generate_header(self):
2000 header_txt = []
2001 header_txt.append('#pragma once\n')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06002002 header_txt.append('#include "vulkan.h"')
2003 header_txt.append('#include "vkDbg.h"\n')
2004 header_txt.append('void AttachHooks_vkdbg();')
2005 header_txt.append('void DetachHooks_vkdbg();')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07002006 return "\n".join(header_txt)
2007
2008 def generate_body(self):
2009 body = [self._generate_trace_func_ptrs_ext('Dbg'),
2010 self._generate_trace_func_protos_ext('Dbg')]
2011
2012 return "\n".join(body)
2013
2014class GlaveDbgC(Subcommand):
2015 def generate_header(self):
2016 header_txt = []
2017 header_txt.append('#include "glv_platform.h"')
2018 header_txt.append('#include "glv_common.h"')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06002019 header_txt.append('#include "glvtrace_vk_vk.h"')
2020 header_txt.append('#include "glvtrace_vk_vkdbg.h"')
Peter Lohrmanncde614c2015-03-27 12:57:10 -07002021 header_txt.append('#include "glv_vk_vkdbg_structs.h"')
2022 header_txt.append('#include "glv_vk_packet_id.h"')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07002023 header_txt.append('#ifdef WIN32')
2024 header_txt.append('#include "mhook/mhook-lib/mhook.h"')
2025 header_txt.append('#endif')
2026 return "\n".join(header_txt)
2027
2028 def generate_body(self):
2029 body = [self._generate_func_ptr_assignments_ext('Dbg'),
2030 self._generate_attach_hooks_ext('Dbg'),
2031 self._generate_detach_hooks_ext('Dbg'),
2032 self._generate_trace_funcs_ext('Dbg')]
2033
2034 return "\n".join(body)
2035
2036class GlaveDbgStructs(Subcommand):
2037 def generate_header(self):
2038 header_txt = []
2039 header_txt.append('#pragma once\n')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06002040 header_txt.append('#include "vkDbg.h"')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07002041 header_txt.append('#include "glv_trace_packet_utils.h"\n')
2042 return "\n".join(header_txt)
2043
2044 def generate_body(self):
2045 body = [self._generate_interp_funcs_ext('Dbg')]
2046
2047 return "\n".join(body)
2048
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06002049class GlaveReplayVkFuncPtrs(Subcommand):
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07002050 def generate_header(self):
2051 header_txt = []
2052 header_txt.append('#pragma once\n')
Piers Danielle2bca482015-02-24 13:58:47 -07002053 header_txt.append('#if defined(PLATFORM_LINUX) || defined(XCB_NVIDIA)')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07002054 header_txt.append('#include <xcb/xcb.h>\n')
David Pinedo8e9cb3b2015-02-10 15:02:08 -07002055 header_txt.append('#endif')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06002056 header_txt.append('#include "vulkan.h"')
2057 header_txt.append('#include "vkDbg.h"')
Piers Danielle2bca482015-02-24 13:58:47 -07002058 header_txt.append('#if defined(PLATFORM_LINUX) || defined(XCB_NVIDIA)')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06002059 header_txt.append('#include "vkWsiX11Ext.h"')
David Pinedo8e9cb3b2015-02-10 15:02:08 -07002060 header_txt.append('#else')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06002061 header_txt.append('#include "vkWsiWinExt.h"')
David Pinedo8e9cb3b2015-02-10 15:02:08 -07002062 header_txt.append('#endif')
Peter Lohrmannaf44b452015-03-30 18:29:22 -07002063
2064 def generate_body(self):
2065 body = [self._generate_replay_func_ptrs()]
2066 return "\n".join(body)
2067
Peter Lohrmann75728222015-04-02 11:45:31 -07002068class GlaveReplayObjMapperHeader(Subcommand):
Jon Ashburn013aa1c2015-02-13 11:25:53 -07002069 def generate_header(self):
Ian Elliott91e681e2015-02-18 15:35:00 -07002070 header_txt = []
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07002071 header_txt.append('#pragma once\n')
2072 header_txt.append('#include <set>')
2073 header_txt.append('#include <map>')
2074 header_txt.append('#include <vector>')
2075 header_txt.append('#include <string>')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06002076 header_txt.append('#include "vulkan.h"')
2077 header_txt.append('#include "vkDbg.h"')
Jon Ashburn15908772015-02-17 13:28:11 -07002078 header_txt.append('#if defined(PLATFORM_LINUX) || defined(XCB_NVIDIA)')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06002079 header_txt.append('#include "vkWsiX11Ext.h"')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07002080 header_txt.append('#else')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06002081 header_txt.append('#include "vkWsiWinExt.h"')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07002082 header_txt.append('#endif')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07002083 return "\n".join(header_txt)
2084
2085 def generate_body(self):
Peter Lohrmann75728222015-04-02 11:45:31 -07002086 body = [self._generate_replay_objmapper_class()]
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07002087 return "\n".join(body)
2088
2089class GlaveReplayC(Subcommand):
2090 def generate_header(self):
2091 header_txt = []
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06002092 header_txt.append('#include "glvreplay_vk_vkreplay.h"\n')
2093 header_txt.append('#include "glvreplay_vk.h"\n')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07002094 header_txt.append('#include "glvreplay_main.h"\n')
2095 header_txt.append('#include <algorithm>')
2096 header_txt.append('#include <queue>')
Peter Lohrmann3f0d6972015-04-01 18:12:34 -07002097 header_txt.append('\n')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07002098 header_txt.append('extern "C" {')
Peter Lohrmanncde614c2015-03-27 12:57:10 -07002099 header_txt.append('#include "glv_vk_vk_structs.h"')
2100 header_txt.append('#include "glv_vk_vkdbg_structs.h"')
2101 header_txt.append('#include "glv_vk_vkwsix11ext_structs.h"')
2102 header_txt.append('#include "glv_vk_packet_id.h"')
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06002103 header_txt.append('#include "vk_enum_string_helper.h"\n}\n')
2104 header_txt.append('#define APP_NAME "glvreplay_vk"')
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07002105 header_txt.append('#define IDI_ICON 101\n')
Peter Lohrmann3f0d6972015-04-01 18:12:34 -07002106
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07002107 return "\n".join(header_txt)
2108
2109 def generate_body(self):
Peter Lohrmann3f0d6972015-04-01 18:12:34 -07002110 body = [self._generate_replay_init_funcs(),
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07002111 self._generate_replay()]
2112
2113 return "\n".join(body)
2114
2115def main():
2116 subcommands = {
2117 "glave-trace-h" : GlaveTraceHeader,
2118 "glave-trace-c" : GlaveTraceC,
2119 "glave-packet-id" : GlavePacketID,
2120 "glave-core-structs" : GlaveCoreStructs,
2121 "glave-wsi-trace-h" : GlaveWsiHeader,
2122 "glave-wsi-trace-c" : GlaveWsiC,
2123 "glave-wsi-trace-structs" : GlaveWsiStructs,
2124 "glave-dbg-trace-h" : GlaveDbgHeader,
2125 "glave-dbg-trace-c" : GlaveDbgC,
2126 "glave-dbg-trace-structs" : GlaveDbgStructs,
Courtney Goeltzenleuchter4bffc072015-04-14 16:33:28 -06002127 "glave-replay-vk-funcs" : GlaveReplayVkFuncPtrs,
Peter Lohrmann75728222015-04-02 11:45:31 -07002128 "glave-replay-obj-mapper-h" : GlaveReplayObjMapperHeader,
Tobin Ehlisf5e1fc52014-12-15 18:14:12 -07002129 "glave-replay-c" : GlaveReplayC,
2130 }
2131
2132 if len(sys.argv) < 2 or sys.argv[1] not in subcommands:
2133 print("Usage: %s <subcommand> [options]" % sys.argv[0])
2134 print
2135 print("Available sucommands are: %s" % " ".join(subcommands))
2136 exit(1)
2137
2138 subcmd = subcommands[sys.argv[1]](sys.argv[2:])
2139 subcmd.run()
2140
2141if __name__ == "__main__":
2142 main()