blob: 1e59d0a4584a567aa69ef1632b6482870e6b4c4c [file] [log] [blame]
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -07001#!/usr/bin/python3 -i
2#
Mark Lobodzinski733f7f42017-01-10 11:42:22 -07003# Copyright (c) 2015-2017 The Khronos Group Inc.
4# Copyright (c) 2015-2017 Valve Corporation
5# Copyright (c) 2015-2017 LunarG, Inc.
6# Copyright (c) 2015-2017 Google Inc.
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -07007#
8# Licensed under the Apache License, Version 2.0 (the "License");
9# you may not use this file except in compliance with the License.
10# You may obtain a copy of the License at
11#
12# http://www.apache.org/licenses/LICENSE-2.0
13#
14# Unless required by applicable law or agreed to in writing, software
15# distributed under the License is distributed on an "AS IS" BASIS,
16# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17# See the License for the specific language governing permissions and
18# limitations under the License.
19#
20# Author: Mark Lobodzinski <mark@lunarg.com>
Mark Lobodzinskic67efd02017-01-04 09:16:00 -070021# Author: Tobin Ehlis <tobine@google.com>
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070022
23import os,re,sys
24import xml.etree.ElementTree as etree
25from generator import *
26from collections import namedtuple
27
28#
29# HelperFileOutputGeneratorOptions - subclass of GeneratorOptions.
30class HelperFileOutputGeneratorOptions(GeneratorOptions):
31 def __init__(self,
32 filename = None,
33 directory = '.',
34 apiname = None,
35 profile = None,
36 versions = '.*',
37 emitversions = '.*',
38 defaultExtensions = None,
39 addExtensions = None,
40 removeExtensions = None,
41 sortProcedure = regSortFeatures,
42 prefixText = "",
43 genFuncPointers = True,
44 protectFile = True,
45 protectFeature = True,
46 protectProto = None,
47 protectProtoStr = None,
48 apicall = '',
49 apientry = '',
50 apientryp = '',
51 alignFuncParam = 0,
52 library_name = '',
53 helper_file_type = ''):
54 GeneratorOptions.__init__(self, filename, directory, apiname, profile,
55 versions, emitversions, defaultExtensions,
56 addExtensions, removeExtensions, sortProcedure)
57 self.prefixText = prefixText
58 self.genFuncPointers = genFuncPointers
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070059 self.protectFile = protectFile
60 self.protectFeature = protectFeature
61 self.protectProto = protectProto
62 self.protectProtoStr = protectProtoStr
63 self.apicall = apicall
64 self.apientry = apientry
65 self.apientryp = apientryp
66 self.alignFuncParam = alignFuncParam
67 self.library_name = library_name
68 self.helper_file_type = helper_file_type
69#
Mark Lobodzinski46d388f2016-12-28 10:46:26 -070070# HelperFileOutputGenerator - subclass of OutputGenerator. Outputs Vulkan helper files
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070071class HelperFileOutputGenerator(OutputGenerator):
Mark Lobodzinskia3cc3612017-01-03 13:25:10 -070072 """Generate helper file based on XML element attributes"""
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070073 def __init__(self,
74 errFile = sys.stderr,
75 warnFile = sys.stderr,
76 diagFile = sys.stdout):
77 OutputGenerator.__init__(self, errFile, warnFile, diagFile)
78 # Internal state - accumulators for different inner block text
Mark Lobodzinski5380d132016-12-28 14:45:34 -070079 self.enum_output = '' # string built up of enum string routines
80 self.struct_size_h_output = '' # string built up of struct size header output
81 self.struct_size_c_output = '' # string built up of struct size source output
Mark Lobodzinski46d388f2016-12-28 10:46:26 -070082 # Internal state - accumulators for different inner block text
Mark Lobodzinski46d388f2016-12-28 10:46:26 -070083 self.structNames = [] # List of Vulkan struct typenames
84 self.structTypes = dict() # Map of Vulkan struct typename to required VkStructureType
Mark Lobodzinski46d388f2016-12-28 10:46:26 -070085 self.structMembers = [] # List of StructMemberData records for all Vulkan structs
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -060086 self.object_types = [] # List of all handle types
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -060087 self.debug_report_object_types = [] # Handy copy of debug_report_object_type enum data
Mark Young1ded24b2017-05-30 14:53:50 -060088 self.core_object_types = [] # Handy copy of core_object_type enum data
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -060089 self.device_extension_info = dict() # Dict of device extension name defines and ifdef values
90 self.instance_extension_info = dict() # Dict of instance extension name defines and ifdef values
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -060091
Mark Lobodzinski46d388f2016-12-28 10:46:26 -070092 # Named tuples to store struct and command data
93 self.StructType = namedtuple('StructType', ['name', 'value'])
Mark Lobodzinskic67efd02017-01-04 09:16:00 -070094 self.CommandParam = namedtuple('CommandParam', ['type', 'name', 'ispointer', 'isstaticarray', 'isconst', 'iscount', 'len', 'extstructs', 'cdecl'])
Mark Lobodzinski5380d132016-12-28 14:45:34 -070095 self.StructMemberData = namedtuple('StructMemberData', ['name', 'members', 'ifdef_protect'])
Petr Krause91f7a12017-12-14 20:57:36 +010096
97 self.custom_construct_params = {
98 # safe_VkGraphicsPipelineCreateInfo needs to know if subpass has color and\or depth\stencil attachments to use its pointers
99 'VkGraphicsPipelineCreateInfo' :
100 ', const bool uses_color_attachment, const bool uses_depthstencil_attachment',
101 # safe_VkPipelineViewportStateCreateInfo needs to know if viewport and scissor is dynamic to use its pointers
102 'VkPipelineViewportStateCreateInfo' :
103 ', const bool is_dynamic_viewports, const bool is_dynamic_scissors',
104 }
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700105 #
106 # Called once at the beginning of each run
107 def beginFile(self, genOpts):
108 OutputGenerator.beginFile(self, genOpts)
109 # User-supplied prefix text, if any (list of strings)
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700110 self.helper_file_type = genOpts.helper_file_type
111 self.library_name = genOpts.library_name
112 # File Comment
113 file_comment = '// *** THIS FILE IS GENERATED - DO NOT EDIT ***\n'
114 file_comment += '// See helper_file_generator.py for modifications\n'
115 write(file_comment, file=self.outFile)
116 # Copyright Notice
117 copyright = ''
118 copyright += '\n'
119 copyright += '/***************************************************************************\n'
120 copyright += ' *\n'
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700121 copyright += ' * Copyright (c) 2015-2017 The Khronos Group Inc.\n'
122 copyright += ' * Copyright (c) 2015-2017 Valve Corporation\n'
123 copyright += ' * Copyright (c) 2015-2017 LunarG, Inc.\n'
124 copyright += ' * Copyright (c) 2015-2017 Google Inc.\n'
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700125 copyright += ' *\n'
126 copyright += ' * Licensed under the Apache License, Version 2.0 (the "License");\n'
127 copyright += ' * you may not use this file except in compliance with the License.\n'
128 copyright += ' * You may obtain a copy of the License at\n'
129 copyright += ' *\n'
130 copyright += ' * http://www.apache.org/licenses/LICENSE-2.0\n'
131 copyright += ' *\n'
132 copyright += ' * Unless required by applicable law or agreed to in writing, software\n'
133 copyright += ' * distributed under the License is distributed on an "AS IS" BASIS,\n'
134 copyright += ' * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n'
135 copyright += ' * See the License for the specific language governing permissions and\n'
136 copyright += ' * limitations under the License.\n'
137 copyright += ' *\n'
138 copyright += ' * Author: Mark Lobodzinski <mark@lunarg.com>\n'
Mark Lobodzinskia9c963d2016-12-28 07:45:35 -0700139 copyright += ' * Author: Courtney Goeltzenleuchter <courtneygo@google.com>\n'
140 copyright += ' * Author: Tobin Ehlis <tobine@google.com>\n'
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600141 copyright += ' * Author: Chris Forbes <chrisforbes@google.com>\n'
John Zulaufde972ac2017-10-26 12:07:05 -0600142 copyright += ' * Author: John Zulauf<jzulauf@lunarg.com>\n'
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700143 copyright += ' *\n'
144 copyright += ' ****************************************************************************/\n'
145 write(copyright, file=self.outFile)
146 #
Mark Lobodzinskia3cc3612017-01-03 13:25:10 -0700147 # Write generated file content to output file
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700148 def endFile(self):
149 dest_file = ''
150 dest_file += self.OutputDestFile()
Mark Lobodzinskiafe10542017-01-03 13:22:44 -0700151 # Remove blank lines at EOF
152 if dest_file.endswith('\n'):
153 dest_file = dest_file[:-1]
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700154 write(dest_file, file=self.outFile);
155 # Finish processing in superclass
156 OutputGenerator.endFile(self)
157 #
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600158 # Override parent class to be notified of the beginning of an extension
159 def beginFeature(self, interface, emit):
160 # Start processing in superclass
161 OutputGenerator.beginFeature(self, interface, emit)
162 if self.featureName == 'VK_VERSION_1_0':
163 return
164 nameElem = interface[0][1]
165 name = nameElem.get('name')
166 if 'EXTENSION_NAME' not in name:
167 print("Error in vk.xml file -- extension name is not available")
168 if interface.get('type') == 'instance':
169 self.instance_extension_info[name] = self.featureExtraProtect
170 else:
171 self.device_extension_info[name] = self.featureExtraProtect
172 #
173 # Override parent class to be notified of the end of an extension
174 def endFeature(self):
175 # Finish processing in superclass
176 OutputGenerator.endFeature(self)
177 #
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700178 # Grab group (e.g. C "enum" type) info to output for enum-string conversion helper
179 def genGroup(self, groupinfo, groupName):
180 OutputGenerator.genGroup(self, groupinfo, groupName)
181 groupElem = groupinfo.elem
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700182 # For enum_string_header
183 if self.helper_file_type == 'enum_string_header':
Mark Lobodzinski824ce642016-12-28 08:49:46 -0700184 value_list = []
185 for elem in groupElem.findall('enum'):
186 if elem.get('supported') != 'disabled':
187 item_name = elem.get('name')
188 value_list.append(item_name)
189 if value_list is not None:
190 self.enum_output += self.GenerateEnumStringConversion(groupName, value_list)
Mark Young1ded24b2017-05-30 14:53:50 -0600191 elif self.helper_file_type == 'object_types_header':
192 if groupName == 'VkDebugReportObjectTypeEXT':
193 for elem in groupElem.findall('enum'):
194 if elem.get('supported') != 'disabled':
195 item_name = elem.get('name')
196 self.debug_report_object_types.append(item_name)
197 elif groupName == 'VkObjectType':
198 for elem in groupElem.findall('enum'):
199 if elem.get('supported') != 'disabled':
200 item_name = elem.get('name')
201 self.core_object_types.append(item_name)
202
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700203 #
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700204 # Called for each type -- if the type is a struct/union, grab the metadata
205 def genType(self, typeinfo, name):
206 OutputGenerator.genType(self, typeinfo, name)
207 typeElem = typeinfo.elem
208 # If the type is a struct type, traverse the imbedded <member> tags generating a structure.
209 # Otherwise, emit the tag text.
210 category = typeElem.get('category')
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600211 if category == 'handle':
212 self.object_types.append(name)
213 elif (category == 'struct' or category == 'union'):
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700214 self.structNames.append(name)
215 self.genStruct(typeinfo, name)
216 #
217 # Generate a VkStructureType based on a structure typename
218 def genVkStructureType(self, typename):
219 # Add underscore between lowercase then uppercase
220 value = re.sub('([a-z0-9])([A-Z])', r'\1_\2', typename)
221 # Change to uppercase
222 value = value.upper()
223 # Add STRUCTURE_TYPE_
224 return re.sub('VK_', 'VK_STRUCTURE_TYPE_', value)
225 #
226 # Check if the parameter passed in is a pointer
227 def paramIsPointer(self, param):
228 ispointer = False
229 for elem in param:
230 if ((elem.tag is not 'type') and (elem.tail is not None)) and '*' in elem.tail:
231 ispointer = True
232 return ispointer
233 #
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700234 # Check if the parameter passed in is a static array
235 def paramIsStaticArray(self, param):
236 isstaticarray = 0
237 paramname = param.find('name')
238 if (paramname.tail is not None) and ('[' in paramname.tail):
239 isstaticarray = paramname.tail.count('[')
240 return isstaticarray
241 #
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700242 # Retrieve the type and name for a parameter
243 def getTypeNameTuple(self, param):
244 type = ''
245 name = ''
246 for elem in param:
247 if elem.tag == 'type':
248 type = noneStr(elem.text)
249 elif elem.tag == 'name':
250 name = noneStr(elem.text)
251 return (type, name)
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700252 # Extract length values from latexmath. Currently an inflexible solution that looks for specific
253 # patterns that are found in vk.xml. Will need to be updated when new patterns are introduced.
254 def parseLateXMath(self, source):
255 name = 'ERROR'
256 decoratedName = 'ERROR'
257 if 'mathit' in source:
Mark Lobodzinski36c33862017-02-13 10:15:53 -0700258 # Matches expressions similar to 'latexmath:[\lceil{\mathit{rasterizationSamples} \over 32}\rceil]'
259 match = re.match(r'latexmath\s*\:\s*\[\s*\\l(\w+)\s*\{\s*\\mathit\s*\{\s*(\w+)\s*\}\s*\\over\s*(\d+)\s*\}\s*\\r(\w+)\s*\]', source)
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700260 if not match or match.group(1) != match.group(4):
261 raise 'Unrecognized latexmath expression'
262 name = match.group(2)
mizhenc27f6c72017-03-31 09:08:16 -0600263 # Need to add 1 for ceiling function; otherwise, the allocated packet
264 # size will be less than needed during capture for some title which use
265 # this in VkPipelineMultisampleStateCreateInfo. based on ceiling function
266 # definition,it is '{0}%{1}?{0}/{1} + 1:{0}/{1}'.format(*match.group(2, 3)),
267 # its value <= '{}/{} + 1'.
268 if match.group(1) == 'ceil':
269 decoratedName = '{}/{} + 1'.format(*match.group(2, 3))
270 else:
271 decoratedName = '{}/{}'.format(*match.group(2, 3))
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700272 else:
Mark Lobodzinski36c33862017-02-13 10:15:53 -0700273 # Matches expressions similar to 'latexmath : [dataSize \over 4]'
Mark Young0f183a82017-02-28 09:58:04 -0700274 match = re.match(r'latexmath\s*\:\s*\[\s*(\w+)\s*\\over\s*(\d+)\s*\]', source)
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700275 name = match.group(1)
276 decoratedName = '{}/{}'.format(*match.group(1, 2))
277 return name, decoratedName
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700278 #
279 # Retrieve the value of the len tag
280 def getLen(self, param):
281 result = None
282 len = param.attrib.get('len')
283 if len and len != 'null-terminated':
284 # For string arrays, 'len' can look like 'count,null-terminated', indicating that we
285 # have a null terminated array of strings. We strip the null-terminated from the
286 # 'len' field and only return the parameter specifying the string count
287 if 'null-terminated' in len:
288 result = len.split(',')[0]
289 else:
290 result = len
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700291 if 'latexmath' in len:
292 param_type, param_name = self.getTypeNameTuple(param)
293 len_name, result = self.parseLateXMath(len)
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700294 # Spec has now notation for len attributes, using :: instead of platform specific pointer symbol
295 result = str(result).replace('::', '->')
296 return result
297 #
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700298 # Check if a structure is or contains a dispatchable (dispatchable = True) or
299 # non-dispatchable (dispatchable = False) handle
300 def TypeContainsObjectHandle(self, handle_type, dispatchable):
301 if dispatchable:
302 type_key = 'VK_DEFINE_HANDLE'
303 else:
304 type_key = 'VK_DEFINE_NON_DISPATCHABLE_HANDLE'
305 handle = self.registry.tree.find("types/type/[name='" + handle_type + "'][@category='handle']")
306 if handle is not None and handle.find('type').text == type_key:
307 return True
308 # if handle_type is a struct, search its members
309 if handle_type in self.structNames:
310 member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == handle_type), None)
311 if member_index is not None:
312 for item in self.structMembers[member_index].members:
313 handle = self.registry.tree.find("types/type/[name='" + item.type + "'][@category='handle']")
314 if handle is not None and handle.find('type').text == type_key:
315 return True
316 return False
317 #
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700318 # Generate local ready-access data describing Vulkan structures and unions from the XML metadata
319 def genStruct(self, typeinfo, typeName):
320 OutputGenerator.genStruct(self, typeinfo, typeName)
321 members = typeinfo.elem.findall('.//member')
322 # Iterate over members once to get length parameters for arrays
323 lens = set()
324 for member in members:
325 len = self.getLen(member)
326 if len:
327 lens.add(len)
328 # Generate member info
329 membersInfo = []
330 for member in members:
331 # Get the member's type and name
332 info = self.getTypeNameTuple(member)
333 type = info[0]
334 name = info[1]
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700335 cdecl = self.makeCParamDecl(member, 1)
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700336 # Process VkStructureType
337 if type == 'VkStructureType':
338 # Extract the required struct type value from the comments
339 # embedded in the original text defining the 'typeinfo' element
340 rawXml = etree.tostring(typeinfo.elem).decode('ascii')
341 result = re.search(r'VK_STRUCTURE_TYPE_\w+', rawXml)
342 if result:
343 value = result.group(0)
344 else:
345 value = self.genVkStructureType(typeName)
346 # Store the required type value
347 self.structTypes[typeName] = self.StructType(name=name, value=value)
348 # Store pointer/array/string info
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700349 isstaticarray = self.paramIsStaticArray(member)
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700350 membersInfo.append(self.CommandParam(type=type,
351 name=name,
352 ispointer=self.paramIsPointer(member),
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700353 isstaticarray=isstaticarray,
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700354 isconst=True if 'const' in cdecl else False,
355 iscount=True if name in lens else False,
356 len=self.getLen(member),
Mike Schuchardta40d0b02017-07-23 12:47:47 -0600357 extstructs=self.registry.validextensionstructs[typeName] if name == 'pNext' else None,
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700358 cdecl=cdecl))
Mark Lobodzinski5380d132016-12-28 14:45:34 -0700359 self.structMembers.append(self.StructMemberData(name=typeName, members=membersInfo, ifdef_protect=self.featureExtraProtect))
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700360 #
361 # Enum_string_header: Create a routine to convert an enumerated value into a string
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700362 def GenerateEnumStringConversion(self, groupName, value_list):
363 outstring = '\n'
364 outstring += 'static inline const char* string_%s(%s input_value)\n' % (groupName, groupName)
365 outstring += '{\n'
366 outstring += ' switch ((%s)input_value)\n' % groupName
367 outstring += ' {\n'
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700368 for item in value_list:
369 outstring += ' case %s:\n' % item
370 outstring += ' return "%s";\n' % item
371 outstring += ' default:\n'
372 outstring += ' return "Unhandled %s";\n' % groupName
373 outstring += ' }\n'
374 outstring += '}\n'
375 return outstring
376 #
Mark Lobodzinski64b432f2017-07-24 16:14:16 -0600377 # Tack on a helper which, given an index into a VkPhysicalDeviceFeatures structure, will print the corresponding feature name
378 def DeIndexPhysDevFeatures(self):
379 pdev_members = None
380 for name, members, ifdef in self.structMembers:
381 if name == 'VkPhysicalDeviceFeatures':
382 pdev_members = members
383 break
384 deindex = '\n'
385 deindex += 'static const char * GetPhysDevFeatureString(uint32_t index) {\n'
386 deindex += ' const char * IndexToPhysDevFeatureString[] = {\n'
387 for feature in pdev_members:
388 deindex += ' "%s",\n' % feature.name
389 deindex += ' };\n\n'
390 deindex += ' return IndexToPhysDevFeatureString[index];\n'
391 deindex += '}\n'
392 return deindex
393 #
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700394 # Combine enum string helper header file preamble with body text and return
395 def GenerateEnumStringHelperHeader(self):
396 enum_string_helper_header = '\n'
397 enum_string_helper_header += '#pragma once\n'
398 enum_string_helper_header += '#ifdef _WIN32\n'
399 enum_string_helper_header += '#pragma warning( disable : 4065 )\n'
400 enum_string_helper_header += '#endif\n'
401 enum_string_helper_header += '\n'
402 enum_string_helper_header += '#include <vulkan/vulkan.h>\n'
403 enum_string_helper_header += '\n'
404 enum_string_helper_header += self.enum_output
Mark Lobodzinski64b432f2017-07-24 16:14:16 -0600405 enum_string_helper_header += self.DeIndexPhysDevFeatures()
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700406 return enum_string_helper_header
407 #
Mark Lobodzinski5380d132016-12-28 14:45:34 -0700408 # struct_size_header: build function prototypes for header file
409 def GenerateStructSizeHeader(self):
410 outstring = ''
411 outstring += 'size_t get_struct_chain_size(const void* struct_ptr);\n'
David Pinedob95caa02017-10-05 10:30:02 -0600412 outstring += 'size_t get_struct_size(const void* struct_ptr);\n'
Mark Lobodzinski5380d132016-12-28 14:45:34 -0700413 for item in self.structMembers:
414 lower_case_name = item.name.lower()
415 if item.ifdef_protect != None:
416 outstring += '#ifdef %s\n' % item.ifdef_protect
417 outstring += 'size_t vk_size_%s(const %s* struct_ptr);\n' % (item.name.lower(), item.name)
418 if item.ifdef_protect != None:
419 outstring += '#endif // %s\n' % item.ifdef_protect
420 outstring += '#ifdef __cplusplus\n'
421 outstring += '}\n'
422 outstring += '#endif'
423 return outstring
424 #
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700425 # Combine struct size helper header file preamble with body text and return
426 def GenerateStructSizeHelperHeader(self):
427 struct_size_helper_header = '\n'
428 struct_size_helper_header += '#ifdef __cplusplus\n'
429 struct_size_helper_header += 'extern "C" {\n'
430 struct_size_helper_header += '#endif\n'
431 struct_size_helper_header += '\n'
432 struct_size_helper_header += '#include <stdio.h>\n'
433 struct_size_helper_header += '#include <stdlib.h>\n'
434 struct_size_helper_header += '#include <vulkan/vulkan.h>\n'
435 struct_size_helper_header += '\n'
436 struct_size_helper_header += '// Function Prototypes\n'
437 struct_size_helper_header += self.GenerateStructSizeHeader()
438 return struct_size_helper_header
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700439 #
440 # Helper function for declaring a counter variable only once
441 def DeclareCounter(self, string_var, declare_flag):
442 if declare_flag == False:
443 string_var += ' uint32_t i = 0;\n'
444 declare_flag = True
445 return string_var, declare_flag
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700446 #
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700447 # Build the header of the get_struct_chain_size function
448 def GenerateChainSizePreamble(self):
David Pinedob95caa02017-10-05 10:30:02 -0600449 preamble = '\nsize_t get_struct_chain_size(const void* struct_ptr) {\n'
450 preamble += ' // Use VkApplicationInfo as struct until actual type is resolved\n'
451 preamble += ' VkApplicationInfo* pNext = (VkApplicationInfo*)struct_ptr;\n'
452 preamble += ' size_t struct_size = 0;\n'
453 preamble += ' while (pNext) {\n'
454 preamble += ' switch (pNext->sType) {\n'
455 return preamble
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700456 #
457 # Build the footer of the get_struct_chain_size function
458 def GenerateChainSizePostamble(self):
459 postamble = ' default:\n'
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700460 postamble += ' struct_size += 0;\n'
Joey Bzdekb6875332017-10-19 14:24:05 -0600461 postamble += ' break;'
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700462 postamble += ' }\n'
463 postamble += ' pNext = (VkApplicationInfo*)pNext->pNext;\n'
464 postamble += ' }\n'
465 postamble += ' return struct_size;\n'
David Pinedob95caa02017-10-05 10:30:02 -0600466 postamble += '}\n'
467 return postamble
468 #
469 # Build the header of the get_struct_size function
470 def GenerateStructSizePreamble(self):
471 preamble = '\nsize_t get_struct_size(const void* struct_ptr) {\n'
472 preamble += ' switch (((VkApplicationInfo*)struct_ptr)->sType) {\n'
473 return preamble
474 #
475 # Build the footer of the get_struct_size function
476 def GenerateStructSizePostamble(self):
477 postamble = ' default:\n'
David Pinedob95caa02017-10-05 10:30:02 -0600478 postamble += ' return(0);\n'
479 postamble += ' }\n'
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700480 postamble += '}'
481 return postamble
482 #
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700483 # struct_size_helper source -- create bodies of struct size helper functions
484 def GenerateStructSizeSource(self):
David Pinedob95caa02017-10-05 10:30:02 -0600485 # Construct the bodies of the struct size functions, get_struct_chain_size(),
486 # and get_struct_size() simultaneously
487 struct_size_funcs = ''
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700488 chain_size = self.GenerateChainSizePreamble()
David Pinedob95caa02017-10-05 10:30:02 -0600489 struct_size = self.GenerateStructSizePreamble()
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700490 for item in self.structMembers:
David Pinedob95caa02017-10-05 10:30:02 -0600491 struct_size_funcs += '\n'
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700492 lower_case_name = item.name.lower()
493 if item.ifdef_protect != None:
David Pinedob95caa02017-10-05 10:30:02 -0600494 struct_size_funcs += '#ifdef %s\n' % item.ifdef_protect
495 struct_size += '#ifdef %s\n' % item.ifdef_protect
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700496 chain_size += '#ifdef %s\n' % item.ifdef_protect
497 if item.name in self.structTypes:
498 chain_size += ' case %s: {\n' % self.structTypes[item.name].value
499 chain_size += ' struct_size += vk_size_%s((%s*)pNext);\n' % (item.name.lower(), item.name)
500 chain_size += ' break;\n'
501 chain_size += ' }\n'
David Pinedob95caa02017-10-05 10:30:02 -0600502 struct_size += ' case %s: \n' % self.structTypes[item.name].value
503 struct_size += ' return vk_size_%s((%s*)struct_ptr);\n' % (item.name.lower(), item.name)
504 struct_size_funcs += 'size_t vk_size_%s(const %s* struct_ptr) { \n' % (item.name.lower(), item.name)
505 struct_size_funcs += ' size_t struct_size = 0;\n'
506 struct_size_funcs += ' if (struct_ptr) {\n'
507 struct_size_funcs += ' struct_size = sizeof(%s);\n' % item.name
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700508 counter_declared = False
509 for member in item.members:
510 vulkan_type = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None)
511 if member.ispointer == True:
512 if vulkan_type is not None:
513 # If this is another Vulkan structure call generated size function
514 if member.len is not None:
David Pinedob95caa02017-10-05 10:30:02 -0600515 struct_size_funcs, counter_declared = self.DeclareCounter(struct_size_funcs, counter_declared)
516 struct_size_funcs += ' for (i = 0; i < struct_ptr->%s; i++) {\n' % member.len
517 struct_size_funcs += ' struct_size += vk_size_%s(&struct_ptr->%s[i]);\n' % (member.type.lower(), member.name)
518 struct_size_funcs += ' }\n'
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700519 else:
David Pinedob95caa02017-10-05 10:30:02 -0600520 struct_size_funcs += ' struct_size += vk_size_%s(struct_ptr->%s);\n' % (member.type.lower(), member.name)
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700521 else:
522 if member.type == 'char':
523 # Deal with sizes of character strings
524 if member.len is not None:
David Pinedob95caa02017-10-05 10:30:02 -0600525 struct_size_funcs, counter_declared = self.DeclareCounter(struct_size_funcs, counter_declared)
526 struct_size_funcs += ' for (i = 0; i < struct_ptr->%s; i++) {\n' % member.len
527 struct_size_funcs += ' struct_size += (sizeof(char*) + (sizeof(char) * (1 + strlen(struct_ptr->%s[i]))));\n' % (member.name)
528 struct_size_funcs += ' }\n'
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700529 else:
David Pinedob95caa02017-10-05 10:30:02 -0600530 struct_size_funcs += ' struct_size += (struct_ptr->%s != NULL) ? sizeof(char)*(1+strlen(struct_ptr->%s)) : 0;\n' % (member.name, member.name)
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700531 else:
532 if member.len is not None:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700533 # Avoid using 'sizeof(void)', which generates compile-time warnings/errors
534 checked_type = member.type
535 if checked_type == 'void':
536 checked_type = 'void*'
David Pinedob95caa02017-10-05 10:30:02 -0600537 struct_size_funcs += ' struct_size += (struct_ptr->%s ) * sizeof(%s);\n' % (member.len, checked_type)
538 struct_size_funcs += ' }\n'
539 struct_size_funcs += ' return struct_size;\n'
540 struct_size_funcs += '}\n'
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700541 if item.ifdef_protect != None:
David Pinedob95caa02017-10-05 10:30:02 -0600542 struct_size_funcs += '#endif // %s\n' % item.ifdef_protect
543 struct_size += '#endif // %s\n' % item.ifdef_protect
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700544 chain_size += '#endif // %s\n' % item.ifdef_protect
545 chain_size += self.GenerateChainSizePostamble()
David Pinedob95caa02017-10-05 10:30:02 -0600546 struct_size += self.GenerateStructSizePostamble()
547 return_value = struct_size_funcs + chain_size + struct_size;
548 return return_value
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700549 #
550 # Combine struct size helper source file preamble with body text and return
551 def GenerateStructSizeHelperSource(self):
552 struct_size_helper_source = '\n'
553 struct_size_helper_source += '#include "vk_struct_size_helper.h"\n'
554 struct_size_helper_source += '#include <string.h>\n'
555 struct_size_helper_source += '#include <assert.h>\n'
556 struct_size_helper_source += '\n'
557 struct_size_helper_source += '// Function Definitions\n'
558 struct_size_helper_source += self.GenerateStructSizeSource()
559 return struct_size_helper_source
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700560 #
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700561 # Combine safe struct helper header file preamble with body text and return
562 def GenerateSafeStructHelperHeader(self):
563 safe_struct_helper_header = '\n'
564 safe_struct_helper_header += '#pragma once\n'
565 safe_struct_helper_header += '#include <vulkan/vulkan.h>\n'
566 safe_struct_helper_header += '\n'
567 safe_struct_helper_header += self.GenerateSafeStructHeader()
568 return safe_struct_helper_header
569 #
570 # safe_struct header: build function prototypes for header file
571 def GenerateSafeStructHeader(self):
572 safe_struct_header = ''
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700573 for item in self.structMembers:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700574 if self.NeedSafeStruct(item) == True:
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700575 safe_struct_header += '\n'
576 if item.ifdef_protect != None:
577 safe_struct_header += '#ifdef %s\n' % item.ifdef_protect
578 safe_struct_header += 'struct safe_%s {\n' % (item.name)
579 for member in item.members:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700580 if member.type in self.structNames:
581 member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None)
582 if member_index is not None and self.NeedSafeStruct(self.structMembers[member_index]) == True:
583 if member.ispointer:
584 safe_struct_header += ' safe_%s* %s;\n' % (member.type, member.name)
585 else:
586 safe_struct_header += ' safe_%s %s;\n' % (member.type, member.name)
587 continue
588 if member.len is not None and (self.TypeContainsObjectHandle(member.type, True) or self.TypeContainsObjectHandle(member.type, False)):
589 safe_struct_header += ' %s* %s;\n' % (member.type, member.name)
590 else:
591 safe_struct_header += '%s;\n' % member.cdecl
Petr Krause91f7a12017-12-14 20:57:36 +0100592 safe_struct_header += ' safe_%s(const %s* in_struct%s);\n' % (item.name, item.name, self.custom_construct_params.get(item.name, ''))
Mark Lobodzinski5cd08512017-09-12 09:50:25 -0600593 safe_struct_header += ' safe_%s(const safe_%s& src);\n' % (item.name, item.name)
Chris Forbesfb633832017-10-03 18:11:54 -0700594 safe_struct_header += ' safe_%s& operator=(const safe_%s& src);\n' % (item.name, item.name)
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700595 safe_struct_header += ' safe_%s();\n' % item.name
596 safe_struct_header += ' ~safe_%s();\n' % item.name
Petr Krause91f7a12017-12-14 20:57:36 +0100597 safe_struct_header += ' void initialize(const %s* in_struct%s);\n' % (item.name, self.custom_construct_params.get(item.name, ''))
598 safe_struct_header += ' void initialize(const safe_%s* src);\n' % (item.name)
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700599 safe_struct_header += ' %s *ptr() { return reinterpret_cast<%s *>(this); }\n' % (item.name, item.name)
600 safe_struct_header += ' %s const *ptr() const { return reinterpret_cast<%s const *>(this); }\n' % (item.name, item.name)
601 safe_struct_header += '};\n'
602 if item.ifdef_protect != None:
603 safe_struct_header += '#endif // %s\n' % item.ifdef_protect
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700604 return safe_struct_header
605 #
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600606 # Generate extension helper header file
607 def GenerateExtensionHelperHeader(self):
608 extension_helper_header = '\n'
609 extension_helper_header += '#ifndef VK_EXTENSION_HELPER_H_\n'
610 extension_helper_header += '#define VK_EXTENSION_HELPER_H_\n'
611 struct = '\n'
Tobin Ehlis84154d32017-06-09 15:46:14 -0600612 extension_helper_header += '#include <vulkan/vulkan.h>\n'
Tobin Ehlisd922d4c2017-06-14 09:43:04 -0600613 extension_helper_header += '#include <string.h>\n'
Tobin Ehlis84154d32017-06-09 15:46:14 -0600614 extension_helper_header += '#include <utility>\n'
615 extension_helper_header += '\n'
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600616 extension_dict = dict()
617 for type in ['Instance', 'Device']:
618 if type == 'Instance':
619 extension_dict = self.instance_extension_info
620 struct += 'struct InstanceExtensions { \n'
621 else:
622 extension_dict = self.device_extension_info
623 struct += 'struct DeviceExtensions : public InstanceExtensions { \n'
624 for ext_name, ifdef in extension_dict.items():
625 bool_name = ext_name.lower()
626 bool_name = re.sub('_extension_name', '', bool_name)
627 struct += ' bool %s{false};\n' % bool_name
628 struct += '\n'
629 if type == 'Instance':
630 struct += ' void InitFromInstanceCreateInfo(const VkInstanceCreateInfo *pCreateInfo) {\n'
631 else:
632 struct += ' void InitFromDeviceCreateInfo(const InstanceExtensions *instance_extensions, const VkDeviceCreateInfo *pCreateInfo) {\n'
633 struct += '\n'
634 struct += ' static const std::pair<char const *, bool %sExtensions::*> known_extensions[]{\n' % type
635 for ext_name, ifdef in extension_dict.items():
636 if ifdef is not None:
637 struct += '#ifdef %s\n' % ifdef
638 bool_name = ext_name.lower()
639 bool_name = re.sub('_extension_name', '', bool_name)
640 struct += ' {%s, &%sExtensions::%s},\n' % (ext_name, type, bool_name)
641 if ifdef is not None:
642 struct += '#endif\n'
643 struct += ' };\n'
644 struct += '\n'
645 struct += ' // Initialize struct data\n'
646 for ext_name, ifdef in self.instance_extension_info.items():
647 bool_name = ext_name.lower()
648 bool_name = re.sub('_extension_name', '', bool_name)
649 if type == 'Device':
650 struct += ' %s = instance_extensions->%s;\n' % (bool_name, bool_name)
651 struct += '\n'
652 struct += ' for (uint32_t i = 0; i < pCreateInfo->enabledExtensionCount; i++) {\n'
653 struct += ' for (auto ext : known_extensions) {\n'
654 struct += ' if (!strcmp(ext.first, pCreateInfo->ppEnabledExtensionNames[i])) {\n'
655 struct += ' this->*(ext.second) = true;\n'
656 struct += ' break;\n'
657 struct += ' }\n'
658 struct += ' }\n'
659 struct += ' }\n'
660 struct += ' }\n'
661 struct += '};\n'
662 struct += '\n'
663 extension_helper_header += struct
664 extension_helper_header += '\n'
665 extension_helper_header += '#endif // VK_EXTENSION_HELPER_H_\n'
666 return extension_helper_header
667 #
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600668 # Combine object types helper header file preamble with body text and return
669 def GenerateObjectTypesHelperHeader(self):
670 object_types_helper_header = '\n'
671 object_types_helper_header += '#pragma once\n'
672 object_types_helper_header += '\n'
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600673 object_types_helper_header += '#include <vulkan/vulkan.h>\n\n'
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600674 object_types_helper_header += self.GenerateObjectTypesHeader()
675 return object_types_helper_header
676 #
677 # Object types header: create object enum type header file
678 def GenerateObjectTypesHeader(self):
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600679 object_types_header = '// Object Type enum for validation layer internal object handling\n'
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600680 object_types_header += 'typedef enum VulkanObjectType {\n'
681 object_types_header += ' kVulkanObjectTypeUnknown = 0,\n'
682 enum_num = 1
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600683 type_list = [];
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600684
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600685 # Output enum definition as each handle is processed, saving the names to use for the conversion routine
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600686 for item in self.object_types:
687 fixup_name = item[2:]
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600688 enum_entry = 'kVulkanObjectType%s' % fixup_name
689 object_types_header += ' ' + enum_entry
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600690 object_types_header += ' = %d,\n' % enum_num
691 enum_num += 1
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600692 type_list.append(enum_entry)
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600693 object_types_header += ' kVulkanObjectTypeMax = %d,\n' % enum_num
Mark Lobodzinskiff92ff82017-04-11 15:31:51 -0600694 object_types_header += '} VulkanObjectType;\n\n'
695
696 # Output name string helper
697 object_types_header += '// Array of object name strings for OBJECT_TYPE enum conversion\n'
Mark Lobodzinski8eb37422017-04-18 14:22:10 -0600698 object_types_header += 'static const char * const object_string[kVulkanObjectTypeMax] = {\n'
Mark Lobodzinskiff92ff82017-04-11 15:31:51 -0600699 object_types_header += ' "Unknown",\n'
700 for item in self.object_types:
701 fixup_name = item[2:]
702 object_types_header += ' "%s",\n' % fixup_name
703 object_types_header += '};\n'
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600704
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600705 # Output a conversion routine from the layer object definitions to the debug report definitions
Mark Lobodzinskiff92ff82017-04-11 15:31:51 -0600706 object_types_header += '\n'
Mark Young1ded24b2017-05-30 14:53:50 -0600707 object_types_header += '// Helper array to get Vulkan VK_EXT_debug_report object type enum from the internal layers version\n'
Mark Lobodzinskic51dbb72017-04-13 14:25:39 -0600708 object_types_header += 'const VkDebugReportObjectTypeEXT get_debug_report_enum[] = {\n'
Mark Lobodzinski34ad5f62017-11-14 10:19:40 -0700709 object_types_header += ' VK_DEBUG_REPORT_OBJECT_TYPE_UNKNOWN_EXT, // No Match\n'
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600710 for object_type in type_list:
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600711 search_type = object_type.replace("kVulkanObjectType", "").lower()
712 for vk_object_type in self.debug_report_object_types:
713 target_type = vk_object_type.replace("VK_DEBUG_REPORT_OBJECT_TYPE_", "").lower()
714 target_type = target_type[:-4]
715 target_type = target_type.replace("_", "")
716 if search_type == target_type:
Mark Lobodzinskiecf0ae12017-04-13 08:36:18 -0600717 object_types_header += ' %s, // %s\n' % (vk_object_type, object_type)
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600718 break
Mark Lobodzinskiecf0ae12017-04-13 08:36:18 -0600719 object_types_header += '};\n'
Mark Young1ded24b2017-05-30 14:53:50 -0600720
721 # Output a conversion routine from the layer object definitions to the core object type definitions
722 object_types_header += '\n'
723 object_types_header += '// Helper array to get Official Vulkan VkObjectType enum from the internal layers version\n'
724 object_types_header += 'const VkObjectType get_object_type_enum[] = {\n'
Mark Lobodzinski34ad5f62017-11-14 10:19:40 -0700725 object_types_header += ' VK_OBJECT_TYPE_UNKNOWN, // No Match\n'
Mark Young1ded24b2017-05-30 14:53:50 -0600726 for object_type in type_list:
Mark Young1ded24b2017-05-30 14:53:50 -0600727 search_type = object_type.replace("kVulkanObjectType", "").lower()
728 for vk_object_type in self.core_object_types:
729 target_type = vk_object_type.replace("VK_OBJECT_TYPE_", "").lower()
730 target_type = target_type.replace("_", "")
731 if search_type == target_type:
732 object_types_header += ' %s, // %s\n' % (vk_object_type, object_type)
Mark Young1ded24b2017-05-30 14:53:50 -0600733 break
Mark Young1ded24b2017-05-30 14:53:50 -0600734 object_types_header += '};\n'
735
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600736 return object_types_header
737 #
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700738 # Determine if a structure needs a safe_struct helper function
739 # That is, it has an sType or one of its members is a pointer
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700740 def NeedSafeStruct(self, structure):
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700741 if 'sType' == structure.name:
742 return True
743 for member in structure.members:
744 if member.ispointer == True:
745 return True
746 return False
747 #
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700748 # Combine safe struct helper source file preamble with body text and return
749 def GenerateSafeStructHelperSource(self):
750 safe_struct_helper_source = '\n'
751 safe_struct_helper_source += '#include "vk_safe_struct.h"\n'
752 safe_struct_helper_source += '#include <string.h>\n'
753 safe_struct_helper_source += '\n'
754 safe_struct_helper_source += self.GenerateSafeStructSource()
755 return safe_struct_helper_source
756 #
757 # safe_struct source -- create bodies of safe struct helper functions
758 def GenerateSafeStructSource(self):
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700759 safe_struct_body = []
Mark Lobodzinski560729b2017-03-06 08:59:14 -0700760 wsi_structs = ['VkXlibSurfaceCreateInfoKHR',
761 'VkXcbSurfaceCreateInfoKHR',
762 'VkWaylandSurfaceCreateInfoKHR',
763 'VkMirSurfaceCreateInfoKHR',
764 'VkAndroidSurfaceCreateInfoKHR',
765 'VkWin32SurfaceCreateInfoKHR'
766 ]
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700767 for item in self.structMembers:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700768 if self.NeedSafeStruct(item) == False:
769 continue
Mark Lobodzinski560729b2017-03-06 08:59:14 -0700770 if item.name in wsi_structs:
771 continue
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700772 if item.ifdef_protect != None:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700773 safe_struct_body.append("#ifdef %s\n" % item.ifdef_protect)
774 ss_name = "safe_%s" % item.name
775 init_list = '' # list of members in struct constructor initializer
776 default_init_list = '' # Default constructor just inits ptrs to nullptr in initializer
777 init_func_txt = '' # Txt for initialize() function that takes struct ptr and inits members
778 construct_txt = '' # Body of constuctor as well as body of initialize() func following init_func_txt
779 destruct_txt = ''
Petr Krause91f7a12017-12-14 20:57:36 +0100780
781 custom_construct_txt = {
782 # VkWriteDescriptorSet is special case because pointers may be non-null but ignored
783 'VkWriteDescriptorSet' :
784 ' switch (descriptorType) {\n'
785 ' case VK_DESCRIPTOR_TYPE_SAMPLER:\n'
786 ' case VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER:\n'
787 ' case VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE:\n'
788 ' case VK_DESCRIPTOR_TYPE_STORAGE_IMAGE:\n'
789 ' case VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT:\n'
790 ' if (descriptorCount && in_struct->pImageInfo) {\n'
791 ' pImageInfo = new VkDescriptorImageInfo[descriptorCount];\n'
792 ' for (uint32_t i=0; i<descriptorCount; ++i) {\n'
793 ' pImageInfo[i] = in_struct->pImageInfo[i];\n'
794 ' }\n'
795 ' }\n'
796 ' break;\n'
797 ' case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER:\n'
798 ' case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER:\n'
799 ' case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC:\n'
800 ' case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC:\n'
801 ' if (descriptorCount && in_struct->pBufferInfo) {\n'
802 ' pBufferInfo = new VkDescriptorBufferInfo[descriptorCount];\n'
803 ' for (uint32_t i=0; i<descriptorCount; ++i) {\n'
804 ' pBufferInfo[i] = in_struct->pBufferInfo[i];\n'
805 ' }\n'
806 ' }\n'
807 ' break;\n'
808 ' case VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER:\n'
809 ' case VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER:\n'
810 ' if (descriptorCount && in_struct->pTexelBufferView) {\n'
811 ' pTexelBufferView = new VkBufferView[descriptorCount];\n'
812 ' for (uint32_t i=0; i<descriptorCount; ++i) {\n'
813 ' pTexelBufferView[i] = in_struct->pTexelBufferView[i];\n'
814 ' }\n'
815 ' }\n'
816 ' break;\n'
817 ' default:\n'
818 ' break;\n'
819 ' }\n',
820 'VkShaderModuleCreateInfo' :
821 ' if (in_struct->pCode) {\n'
822 ' pCode = reinterpret_cast<uint32_t *>(new uint8_t[codeSize]);\n'
823 ' memcpy((void *)pCode, (void *)in_struct->pCode, codeSize);\n'
824 ' }\n',
825 # VkGraphicsPipelineCreateInfo is special case because its pointers may be non-null but ignored
826 'VkGraphicsPipelineCreateInfo' :
827 ' if (stageCount && in_struct->pStages) {\n'
828 ' pStages = new safe_VkPipelineShaderStageCreateInfo[stageCount];\n'
829 ' for (uint32_t i=0; i<stageCount; ++i) {\n'
830 ' pStages[i].initialize(&in_struct->pStages[i]);\n'
831 ' }\n'
832 ' }\n'
833 ' if (in_struct->pVertexInputState)\n'
834 ' pVertexInputState = new safe_VkPipelineVertexInputStateCreateInfo(in_struct->pVertexInputState);\n'
835 ' else\n'
836 ' pVertexInputState = NULL;\n'
837 ' if (in_struct->pInputAssemblyState)\n'
838 ' pInputAssemblyState = new safe_VkPipelineInputAssemblyStateCreateInfo(in_struct->pInputAssemblyState);\n'
839 ' else\n'
840 ' pInputAssemblyState = NULL;\n'
841 ' bool has_tessellation_stage = false;\n'
842 ' if (stageCount && pStages)\n'
843 ' for (uint32_t i=0; i<stageCount && !has_tessellation_stage; ++i)\n'
844 ' if (pStages[i].stage == VK_SHADER_STAGE_TESSELLATION_CONTROL_BIT || pStages[i].stage == VK_SHADER_STAGE_TESSELLATION_EVALUATION_BIT)\n'
845 ' has_tessellation_stage = true;\n'
846 ' if (in_struct->pTessellationState && has_tessellation_stage)\n'
847 ' pTessellationState = new safe_VkPipelineTessellationStateCreateInfo(in_struct->pTessellationState);\n'
848 ' else\n'
849 ' pTessellationState = NULL; // original pTessellationState pointer ignored\n'
850 ' bool has_rasterization = in_struct->pRasterizationState ? !in_struct->pRasterizationState->rasterizerDiscardEnable : false;\n'
851 ' if (in_struct->pViewportState && has_rasterization) {\n'
852 ' bool is_dynamic_viewports = false;\n'
853 ' bool is_dynamic_scissors = false;\n'
854 ' if (in_struct->pDynamicState && in_struct->pDynamicState->pDynamicStates) {\n'
855 ' for (uint32_t i = 0; i < in_struct->pDynamicState->dynamicStateCount && !is_dynamic_viewports; ++i)\n'
856 ' if (in_struct->pDynamicState->pDynamicStates[i] == VK_DYNAMIC_STATE_VIEWPORT)\n'
857 ' is_dynamic_viewports = true;\n'
858 ' for (uint32_t i = 0; i < in_struct->pDynamicState->dynamicStateCount && !is_dynamic_scissors; ++i)\n'
859 ' if (in_struct->pDynamicState->pDynamicStates[i] == VK_DYNAMIC_STATE_SCISSOR)\n'
860 ' is_dynamic_scissors = true;\n'
861 ' }\n'
862 ' pViewportState = new safe_VkPipelineViewportStateCreateInfo(in_struct->pViewportState, is_dynamic_viewports, is_dynamic_scissors);\n'
863 ' } else\n'
864 ' pViewportState = NULL; // original pViewportState pointer ignored\n'
865 ' if (in_struct->pRasterizationState)\n'
866 ' pRasterizationState = new safe_VkPipelineRasterizationStateCreateInfo(in_struct->pRasterizationState);\n'
867 ' else\n'
868 ' pRasterizationState = NULL;\n'
869 ' if (in_struct->pMultisampleState && has_rasterization)\n'
870 ' pMultisampleState = new safe_VkPipelineMultisampleStateCreateInfo(in_struct->pMultisampleState);\n'
871 ' else\n'
872 ' pMultisampleState = NULL; // original pMultisampleState pointer ignored\n'
873 ' // needs a tracked subpass state uses_depthstencil_attachment\n'
874 ' if (in_struct->pDepthStencilState && has_rasterization && uses_depthstencil_attachment)\n'
875 ' pDepthStencilState = new safe_VkPipelineDepthStencilStateCreateInfo(in_struct->pDepthStencilState);\n'
876 ' else\n'
877 ' pDepthStencilState = NULL; // original pDepthStencilState pointer ignored\n'
878 ' // needs a tracked subpass state usesColorAttachment\n'
879 ' if (in_struct->pColorBlendState && has_rasterization && uses_color_attachment)\n'
880 ' pColorBlendState = new safe_VkPipelineColorBlendStateCreateInfo(in_struct->pColorBlendState);\n'
881 ' else\n'
882 ' pColorBlendState = NULL; // original pColorBlendState pointer ignored\n'
883 ' if (in_struct->pDynamicState)\n'
884 ' pDynamicState = new safe_VkPipelineDynamicStateCreateInfo(in_struct->pDynamicState);\n'
885 ' else\n'
886 ' pDynamicState = NULL;\n',
887 # VkPipelineViewportStateCreateInfo is special case because its pointers may be non-null but ignored
888 'VkPipelineViewportStateCreateInfo' :
889 ' if (in_struct->pViewports && !is_dynamic_viewports) {\n'
890 ' pViewports = new VkViewport[in_struct->viewportCount];\n'
891 ' memcpy ((void *)pViewports, (void *)in_struct->pViewports, sizeof(VkViewport)*in_struct->viewportCount);\n'
892 ' }\n'
893 ' else\n'
894 ' pViewports = NULL;\n'
895 ' if (in_struct->pScissors && !is_dynamic_scissors) {\n'
896 ' pScissors = new VkRect2D[in_struct->scissorCount];\n'
897 ' memcpy ((void *)pScissors, (void *)in_struct->pScissors, sizeof(VkRect2D)*in_struct->scissorCount);\n'
898 ' }\n'
899 ' else\n'
900 ' pScissors = NULL;\n',
901 }
902
903 custom_copy_txt = {
904 # VkGraphicsPipelineCreateInfo is special case because it has custom construct parameters
905 'VkGraphicsPipelineCreateInfo' :
906 ' if (stageCount && src.pStages) {\n'
907 ' pStages = new safe_VkPipelineShaderStageCreateInfo[stageCount];\n'
908 ' for (uint32_t i=0; i<stageCount; ++i) {\n'
909 ' pStages[i].initialize(&src.pStages[i]);\n'
910 ' }\n'
911 ' }\n'
912 ' if (src.pVertexInputState)\n'
913 ' pVertexInputState = new safe_VkPipelineVertexInputStateCreateInfo(*src.pVertexInputState);\n'
914 ' else\n'
915 ' pVertexInputState = NULL;\n'
916 ' if (src.pInputAssemblyState)\n'
917 ' pInputAssemblyState = new safe_VkPipelineInputAssemblyStateCreateInfo(*src.pInputAssemblyState);\n'
918 ' else\n'
919 ' pInputAssemblyState = NULL;\n'
920 ' bool has_tessellation_stage = false;\n'
921 ' if (stageCount && pStages)\n'
922 ' for (uint32_t i=0; i<stageCount && !has_tessellation_stage; ++i)\n'
923 ' if (pStages[i].stage == VK_SHADER_STAGE_TESSELLATION_CONTROL_BIT || pStages[i].stage == VK_SHADER_STAGE_TESSELLATION_EVALUATION_BIT)\n'
924 ' has_tessellation_stage = true;\n'
925 ' if (src.pTessellationState && has_tessellation_stage)\n'
926 ' pTessellationState = new safe_VkPipelineTessellationStateCreateInfo(*src.pTessellationState);\n'
927 ' else\n'
928 ' pTessellationState = NULL; // original pTessellationState pointer ignored\n'
929 ' bool has_rasterization = src.pRasterizationState ? !src.pRasterizationState->rasterizerDiscardEnable : false;\n'
930 ' if (src.pViewportState && has_rasterization) {\n'
931 ' pViewportState = new safe_VkPipelineViewportStateCreateInfo(*src.pViewportState);\n'
932 ' } else\n'
933 ' pViewportState = NULL; // original pViewportState pointer ignored\n'
934 ' if (src.pRasterizationState)\n'
935 ' pRasterizationState = new safe_VkPipelineRasterizationStateCreateInfo(*src.pRasterizationState);\n'
936 ' else\n'
937 ' pRasterizationState = NULL;\n'
938 ' if (src.pMultisampleState && has_rasterization)\n'
939 ' pMultisampleState = new safe_VkPipelineMultisampleStateCreateInfo(*src.pMultisampleState);\n'
940 ' else\n'
941 ' pMultisampleState = NULL; // original pMultisampleState pointer ignored\n'
942 ' if (src.pDepthStencilState && has_rasterization)\n'
943 ' pDepthStencilState = new safe_VkPipelineDepthStencilStateCreateInfo(*src.pDepthStencilState);\n'
944 ' else\n'
945 ' pDepthStencilState = NULL; // original pDepthStencilState pointer ignored\n'
946 ' if (src.pColorBlendState && has_rasterization)\n'
947 ' pColorBlendState = new safe_VkPipelineColorBlendStateCreateInfo(*src.pColorBlendState);\n'
948 ' else\n'
949 ' pColorBlendState = NULL; // original pColorBlendState pointer ignored\n'
950 ' if (src.pDynamicState)\n'
951 ' pDynamicState = new safe_VkPipelineDynamicStateCreateInfo(*src.pDynamicState);\n'
952 ' else\n'
953 ' pDynamicState = NULL;\n',
954 # VkPipelineViewportStateCreateInfo is special case because it has custom construct parameters
955 'VkPipelineViewportStateCreateInfo' :
956 ' if (src.pViewports) {\n'
957 ' pViewports = new VkViewport[src.viewportCount];\n'
958 ' memcpy ((void *)pViewports, (void *)src.pViewports, sizeof(VkViewport)*src.viewportCount);\n'
959 ' }\n'
960 ' else\n'
961 ' pViewports = NULL;\n'
962 ' if (src.pScissors) {\n'
963 ' pScissors = new VkRect2D[src.scissorCount];\n'
964 ' memcpy ((void *)pScissors, (void *)src.pScissors, sizeof(VkRect2D)*src.scissorCount);\n'
965 ' }\n'
966 ' else\n'
967 ' pScissors = NULL;\n',
968 }
969
Mike Schuchardt81485762017-09-04 11:38:42 -0600970 custom_destruct_txt = {'VkShaderModuleCreateInfo' :
971 ' if (pCode)\n'
972 ' delete[] reinterpret_cast<const uint8_t *>(pCode);\n' }
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700973
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700974 for member in item.members:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700975 m_type = member.type
976 if member.type in self.structNames:
977 member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None)
978 if member_index is not None and self.NeedSafeStruct(self.structMembers[member_index]) == True:
979 m_type = 'safe_%s' % member.type
980 if member.ispointer and 'safe_' not in m_type and self.TypeContainsObjectHandle(member.type, False) == False:
981 # Ptr types w/o a safe_struct, for non-null case need to allocate new ptr and copy data in
Mark Lobodzinski560729b2017-03-06 08:59:14 -0700982 if m_type in ['void', 'char']:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700983 # For these exceptions just copy initial value over for now
984 init_list += '\n %s(in_struct->%s),' % (member.name, member.name)
985 init_func_txt += ' %s = in_struct->%s;\n' % (member.name, member.name)
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700986 else:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700987 default_init_list += '\n %s(nullptr),' % (member.name)
988 init_list += '\n %s(nullptr),' % (member.name)
989 init_func_txt += ' %s = nullptr;\n' % (member.name)
990 if 'pNext' != member.name and 'void' not in m_type:
Mark Lobodzinski51160a12017-01-18 11:05:48 -0700991 if not member.isstaticarray and (member.len is None or '/' in member.len):
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700992 construct_txt += ' if (in_struct->%s) {\n' % member.name
993 construct_txt += ' %s = new %s(*in_struct->%s);\n' % (member.name, m_type, member.name)
994 construct_txt += ' }\n'
995 destruct_txt += ' if (%s)\n' % member.name
996 destruct_txt += ' delete %s;\n' % member.name
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700997 else:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700998 construct_txt += ' if (in_struct->%s) {\n' % member.name
999 construct_txt += ' %s = new %s[in_struct->%s];\n' % (member.name, m_type, member.len)
1000 construct_txt += ' memcpy ((void *)%s, (void *)in_struct->%s, sizeof(%s)*in_struct->%s);\n' % (member.name, member.name, m_type, member.len)
1001 construct_txt += ' }\n'
1002 destruct_txt += ' if (%s)\n' % member.name
1003 destruct_txt += ' delete[] %s;\n' % member.name
1004 elif member.isstaticarray or member.len is not None:
1005 if member.len is None:
1006 # Extract length of static array by grabbing val between []
1007 static_array_size = re.match(r"[^[]*\[([^]]*)\]", member.cdecl)
1008 construct_txt += ' for (uint32_t i=0; i<%s; ++i) {\n' % static_array_size.group(1)
1009 construct_txt += ' %s[i] = in_struct->%s[i];\n' % (member.name, member.name)
1010 construct_txt += ' }\n'
1011 else:
1012 # Init array ptr to NULL
1013 default_init_list += '\n %s(nullptr),' % member.name
1014 init_list += '\n %s(nullptr),' % member.name
1015 init_func_txt += ' %s = nullptr;\n' % member.name
1016 array_element = 'in_struct->%s[i]' % member.name
1017 if member.type in self.structNames:
1018 member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None)
1019 if member_index is not None and self.NeedSafeStruct(self.structMembers[member_index]) == True:
1020 array_element = '%s(&in_struct->safe_%s[i])' % (member.type, member.name)
1021 construct_txt += ' if (%s && in_struct->%s) {\n' % (member.len, member.name)
1022 construct_txt += ' %s = new %s[%s];\n' % (member.name, m_type, member.len)
1023 destruct_txt += ' if (%s)\n' % member.name
1024 destruct_txt += ' delete[] %s;\n' % member.name
1025 construct_txt += ' for (uint32_t i=0; i<%s; ++i) {\n' % (member.len)
1026 if 'safe_' in m_type:
1027 construct_txt += ' %s[i].initialize(&in_struct->%s[i]);\n' % (member.name, member.name)
Mark Lobodzinskie20e4562017-01-03 11:14:26 -07001028 else:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -07001029 construct_txt += ' %s[i] = %s;\n' % (member.name, array_element)
1030 construct_txt += ' }\n'
1031 construct_txt += ' }\n'
1032 elif member.ispointer == True:
1033 construct_txt += ' if (in_struct->%s)\n' % member.name
1034 construct_txt += ' %s = new %s(in_struct->%s);\n' % (member.name, m_type, member.name)
1035 construct_txt += ' else\n'
1036 construct_txt += ' %s = NULL;\n' % member.name
1037 destruct_txt += ' if (%s)\n' % member.name
1038 destruct_txt += ' delete %s;\n' % member.name
1039 elif 'safe_' in m_type:
1040 init_list += '\n %s(&in_struct->%s),' % (member.name, member.name)
1041 init_func_txt += ' %s.initialize(&in_struct->%s);\n' % (member.name, member.name)
1042 else:
1043 init_list += '\n %s(in_struct->%s),' % (member.name, member.name)
1044 init_func_txt += ' %s = in_struct->%s;\n' % (member.name, member.name)
1045 if '' != init_list:
1046 init_list = init_list[:-1] # hack off final comma
1047 if item.name in custom_construct_txt:
1048 construct_txt = custom_construct_txt[item.name]
Mike Schuchardt81485762017-09-04 11:38:42 -06001049 if item.name in custom_destruct_txt:
1050 destruct_txt = custom_destruct_txt[item.name]
Petr Krause91f7a12017-12-14 20:57:36 +01001051 safe_struct_body.append("\n%s::%s(const %s* in_struct%s) :%s\n{\n%s}" % (ss_name, ss_name, item.name, self.custom_construct_params.get(item.name, ''), init_list, construct_txt))
Mark Lobodzinskic67efd02017-01-04 09:16:00 -07001052 if '' != default_init_list:
1053 default_init_list = " :%s" % (default_init_list[:-1])
1054 safe_struct_body.append("\n%s::%s()%s\n{}" % (ss_name, ss_name, default_init_list))
1055 # Create slight variation of init and construct txt for copy constructor that takes a src object reference vs. struct ptr
1056 copy_construct_init = init_func_txt.replace('in_struct->', 'src.')
1057 copy_construct_txt = construct_txt.replace(' (in_struct->', ' (src.') # Exclude 'if' blocks from next line
1058 copy_construct_txt = copy_construct_txt.replace('(in_struct->', '(*src.') # Pass object to copy constructors
1059 copy_construct_txt = copy_construct_txt.replace('in_struct->', 'src.') # Modify remaining struct refs for src object
Petr Krause91f7a12017-12-14 20:57:36 +01001060 if item.name in custom_copy_txt:
1061 copy_construct_txt = custom_copy_txt[item.name]
Chris Forbesfb633832017-10-03 18:11:54 -07001062 copy_assign_txt = ' if (&src == this) return *this;\n\n' + destruct_txt + '\n' + copy_construct_init + copy_construct_txt + '\n return *this;'
Mark Lobodzinski5cd08512017-09-12 09:50:25 -06001063 safe_struct_body.append("\n%s::%s(const %s& src)\n{\n%s%s}" % (ss_name, ss_name, ss_name, copy_construct_init, copy_construct_txt)) # Copy constructor
Chris Forbesfb633832017-10-03 18:11:54 -07001064 safe_struct_body.append("\n%s& %s::operator=(const %s& src)\n{\n%s\n}" % (ss_name, ss_name, ss_name, copy_assign_txt)) # Copy assignment operator
Mark Lobodzinskic67efd02017-01-04 09:16:00 -07001065 safe_struct_body.append("\n%s::~%s()\n{\n%s}" % (ss_name, ss_name, destruct_txt))
Petr Krause91f7a12017-12-14 20:57:36 +01001066 safe_struct_body.append("\nvoid %s::initialize(const %s* in_struct%s)\n{\n%s%s}" % (ss_name, item.name, self.custom_construct_params.get(item.name, ''), init_func_txt, construct_txt))
Mark Lobodzinskic67efd02017-01-04 09:16:00 -07001067 # Copy initializer uses same txt as copy constructor but has a ptr and not a reference
1068 init_copy = copy_construct_init.replace('src.', 'src->')
1069 init_construct = copy_construct_txt.replace('src.', 'src->')
Mark Lobodzinski5cd08512017-09-12 09:50:25 -06001070 safe_struct_body.append("\nvoid %s::initialize(const %s* src)\n{\n%s%s}" % (ss_name, ss_name, init_copy, init_construct))
Mark Lobodzinskie20e4562017-01-03 11:14:26 -07001071 if item.ifdef_protect != None:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -07001072 safe_struct_body.append("#endif // %s\n" % item.ifdef_protect)
1073 return "\n".join(safe_struct_body)
Mark Lobodzinskie20e4562017-01-03 11:14:26 -07001074 #
John Zulaufde972ac2017-10-26 12:07:05 -06001075 # Generate the type map
1076 def GenerateTypeMapHelperHeader(self):
1077 prefix = 'Lvl'
1078 fprefix = 'lvl_'
1079 typemap = prefix + 'TypeMap'
1080 idmap = prefix + 'STypeMap'
John Zulaufde972ac2017-10-26 12:07:05 -06001081 type_member = 'Type'
1082 id_member = 'kSType'
Mike Schuchardt97662b02017-12-06 13:31:29 -07001083 id_decl = 'static const VkStructureType '
John Zulaufde972ac2017-10-26 12:07:05 -06001084 generic_header = prefix + 'GenericHeader'
1085 typename_func = fprefix + 'typename'
1086 idname_func = fprefix + 'stype_name'
1087 find_func = fprefix + 'find_in_chain'
1088
1089 explanatory_comment = '\n'.join((
1090 '// These empty generic templates are specialized for each type with sType',
1091 '// members and for each sType -- providing a two way map between structure',
Mike Schuchardt97662b02017-12-06 13:31:29 -07001092 '// types and sTypes'))
John Zulaufde972ac2017-10-26 12:07:05 -06001093
1094 empty_typemap = 'template <typename T> struct ' + typemap + ' {};'
1095 typemap_format = 'template <> struct {template}<{typename}> {{\n'
John Zulaufde972ac2017-10-26 12:07:05 -06001096 typemap_format += ' {id_decl}{id_member} = {id_value};\n'
1097 typemap_format += '}};\n'
1098
1099 empty_idmap = 'template <VkStructureType id> struct ' + idmap + ' {};'
1100 idmap_format = ''.join((
1101 'template <> struct {template}<{id_value}> {{\n',
1102 ' typedef {typename} {typedef};\n',
John Zulaufde972ac2017-10-26 12:07:05 -06001103 '}};\n'))
1104
1105 # Define the utilities (here so any renaming stays consistent), if this grows large, refactor to a fixed .h file
1106 utilities_format = '\n'.join((
1107 '// Header "base class" for pNext chain traversal',
1108 'struct {header} {{',
1109 ' VkStructureType sType;',
1110 ' const {header} *pNext;',
1111 '}};',
1112 '',
1113 '// Find an entry of the given type in the pNext chain',
1114 'template <typename T> const T *{find_func}(const void *next) {{',
1115 ' const {header} *current = reinterpret_cast<const {header} *>(next);',
1116 ' const T *found = nullptr;',
1117 ' while (current) {{',
1118 ' if ({type_map}<T>::{id_member} == current->sType) {{',
1119 ' found = reinterpret_cast<const T*>(current);',
1120 ' current = nullptr;',
1121 ' }} else {{',
1122 ' current = current->pNext;',
1123 ' }}',
1124 ' }}',
1125 ' return found;',
1126 '}}',
Mike Schuchardt97662b02017-12-06 13:31:29 -07001127 ''))
John Zulaufde972ac2017-10-26 12:07:05 -06001128
1129 code = []
1130 code.append('\n'.join((
1131 '#pragma once',
1132 '#include <vulkan/vulkan.h>\n',
1133 explanatory_comment, '',
1134 empty_idmap,
1135 empty_typemap, '',
Mike Schuchardt97662b02017-12-06 13:31:29 -07001136 utilities_format.format(id_member=id_member, id_map=idmap, type_map=typemap,
John Zulaufde972ac2017-10-26 12:07:05 -06001137 type_member=type_member, header=generic_header, typename_func=typename_func, idname_func=idname_func,
1138 find_func=find_func), ''
1139 )))
1140
1141 # Generate the specializations for each type and stype
1142
1143 for item in self.structMembers:
1144 typename = item.name
1145 info = self.structTypes.get(typename)
1146 if not info:
1147 continue
1148
1149 if item.ifdef_protect != None:
1150 code.append('#ifdef %s' % item.ifdef_protect)
1151
1152 code.append('// Map type {} to id {}'.format(typename, info.value))
1153 code.append(typemap_format.format(template=typemap, typename=typename, id_value=info.value,
Mike Schuchardt97662b02017-12-06 13:31:29 -07001154 id_decl=id_decl, id_member=id_member))
1155 code.append(idmap_format.format(template=idmap, typename=typename, id_value=info.value, typedef=type_member))
John Zulaufde972ac2017-10-26 12:07:05 -06001156
1157 if item.ifdef_protect != None:
1158 code.append('#endif // %s' % item.ifdef_protect)
1159
John Zulaufde972ac2017-10-26 12:07:05 -06001160 return "\n".join(code)
1161
1162 #
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -07001163 # Create a helper file and return it as a string
1164 def OutputDestFile(self):
Mark Lobodzinski46d388f2016-12-28 10:46:26 -07001165 if self.helper_file_type == 'enum_string_header':
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -07001166 return self.GenerateEnumStringHelperHeader()
Mark Lobodzinski46d388f2016-12-28 10:46:26 -07001167 elif self.helper_file_type == 'struct_size_header':
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -07001168 return self.GenerateStructSizeHelperHeader()
Mark Lobodzinski46d388f2016-12-28 10:46:26 -07001169 elif self.helper_file_type == 'struct_size_source':
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -07001170 return self.GenerateStructSizeHelperSource()
Mark Lobodzinskie20e4562017-01-03 11:14:26 -07001171 elif self.helper_file_type == 'safe_struct_header':
1172 return self.GenerateSafeStructHelperHeader()
1173 elif self.helper_file_type == 'safe_struct_source':
1174 return self.GenerateSafeStructHelperSource()
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -06001175 elif self.helper_file_type == 'object_types_header':
1176 return self.GenerateObjectTypesHelperHeader()
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -06001177 elif self.helper_file_type == 'extension_helper_header':
1178 return self.GenerateExtensionHelperHeader()
John Zulaufde972ac2017-10-26 12:07:05 -06001179 elif self.helper_file_type == 'typemap_helper_header':
1180 return self.GenerateTypeMapHelperHeader()
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -07001181 else:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -07001182 return 'Bad Helper File Generator Option %s' % self.helper_file_type
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -07001183