blob: 1f7dda67b495c3fa8df0cfa3ee4bb6a0f21498a1 [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>
John Zulaufd7435c62018-03-16 11:52:57 -060022# Author: John Zulauf <jzulauf@lunarg.com>
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070023
24import os,re,sys
25import xml.etree.ElementTree as etree
26from generator import *
27from collections import namedtuple
Mark Lobodzinski62f71562017-10-24 13:41:18 -060028from common_codegen import *
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070029
30#
31# HelperFileOutputGeneratorOptions - subclass of GeneratorOptions.
32class HelperFileOutputGeneratorOptions(GeneratorOptions):
33 def __init__(self,
34 filename = None,
35 directory = '.',
36 apiname = None,
37 profile = None,
38 versions = '.*',
39 emitversions = '.*',
40 defaultExtensions = None,
41 addExtensions = None,
42 removeExtensions = None,
Mark Lobodzinski62f71562017-10-24 13:41:18 -060043 emitExtensions = None,
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070044 sortProcedure = regSortFeatures,
45 prefixText = "",
46 genFuncPointers = True,
47 protectFile = True,
48 protectFeature = True,
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070049 apicall = '',
50 apientry = '',
51 apientryp = '',
52 alignFuncParam = 0,
53 library_name = '',
Mark Lobodzinski62f71562017-10-24 13:41:18 -060054 expandEnumerants = True,
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070055 helper_file_type = ''):
56 GeneratorOptions.__init__(self, filename, directory, apiname, profile,
57 versions, emitversions, defaultExtensions,
Mark Lobodzinski62f71562017-10-24 13:41:18 -060058 addExtensions, removeExtensions, emitExtensions, sortProcedure)
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070059 self.prefixText = prefixText
60 self.genFuncPointers = genFuncPointers
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070061 self.protectFile = protectFile
62 self.protectFeature = protectFeature
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070063 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
Mark Lobodzinski46d388f2016-12-28 10:46:26 -070080 # Internal state - accumulators for different inner block text
Mark Lobodzinski46d388f2016-12-28 10:46:26 -070081 self.structNames = [] # List of Vulkan struct typenames
82 self.structTypes = dict() # Map of Vulkan struct typename to required VkStructureType
Mark Lobodzinski46d388f2016-12-28 10:46:26 -070083 self.structMembers = [] # List of StructMemberData records for all Vulkan structs
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -060084 self.object_types = [] # List of all handle types
John Zulaufd7435c62018-03-16 11:52:57 -060085 self.object_type_aliases = [] # Aliases to handles types (for handles that were extensions)
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -060086 self.debug_report_object_types = [] # Handy copy of debug_report_object_type enum data
Mark Young1ded24b2017-05-30 14:53:50 -060087 self.core_object_types = [] # Handy copy of core_object_type enum data
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -060088 self.device_extension_info = dict() # Dict of device extension name defines and ifdef values
89 self.instance_extension_info = dict() # Dict of instance extension name defines and ifdef values
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -060090
Mark Lobodzinski46d388f2016-12-28 10:46:26 -070091 # Named tuples to store struct and command data
92 self.StructType = namedtuple('StructType', ['name', 'value'])
Mark Lobodzinskic67efd02017-01-04 09:16:00 -070093 self.CommandParam = namedtuple('CommandParam', ['type', 'name', 'ispointer', 'isstaticarray', 'isconst', 'iscount', 'len', 'extstructs', 'cdecl'])
Mark Lobodzinski5380d132016-12-28 14:45:34 -070094 self.StructMemberData = namedtuple('StructMemberData', ['name', 'members', 'ifdef_protect'])
Petr Krause91f7a12017-12-14 20:57:36 +010095
96 self.custom_construct_params = {
97 # safe_VkGraphicsPipelineCreateInfo needs to know if subpass has color and\or depth\stencil attachments to use its pointers
98 'VkGraphicsPipelineCreateInfo' :
99 ', const bool uses_color_attachment, const bool uses_depthstencil_attachment',
100 # safe_VkPipelineViewportStateCreateInfo needs to know if viewport and scissor is dynamic to use its pointers
101 'VkPipelineViewportStateCreateInfo' :
102 ', const bool is_dynamic_viewports, const bool is_dynamic_scissors',
103 }
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700104 #
105 # Called once at the beginning of each run
106 def beginFile(self, genOpts):
107 OutputGenerator.beginFile(self, genOpts)
108 # User-supplied prefix text, if any (list of strings)
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700109 self.helper_file_type = genOpts.helper_file_type
110 self.library_name = genOpts.library_name
111 # File Comment
112 file_comment = '// *** THIS FILE IS GENERATED - DO NOT EDIT ***\n'
113 file_comment += '// See helper_file_generator.py for modifications\n'
114 write(file_comment, file=self.outFile)
115 # Copyright Notice
116 copyright = ''
117 copyright += '\n'
118 copyright += '/***************************************************************************\n'
119 copyright += ' *\n'
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700120 copyright += ' * Copyright (c) 2015-2017 The Khronos Group Inc.\n'
121 copyright += ' * Copyright (c) 2015-2017 Valve Corporation\n'
122 copyright += ' * Copyright (c) 2015-2017 LunarG, Inc.\n'
123 copyright += ' * Copyright (c) 2015-2017 Google Inc.\n'
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700124 copyright += ' *\n'
125 copyright += ' * Licensed under the Apache License, Version 2.0 (the "License");\n'
126 copyright += ' * you may not use this file except in compliance with the License.\n'
127 copyright += ' * You may obtain a copy of the License at\n'
128 copyright += ' *\n'
129 copyright += ' * http://www.apache.org/licenses/LICENSE-2.0\n'
130 copyright += ' *\n'
131 copyright += ' * Unless required by applicable law or agreed to in writing, software\n'
132 copyright += ' * distributed under the License is distributed on an "AS IS" BASIS,\n'
133 copyright += ' * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n'
134 copyright += ' * See the License for the specific language governing permissions and\n'
135 copyright += ' * limitations under the License.\n'
136 copyright += ' *\n'
137 copyright += ' * Author: Mark Lobodzinski <mark@lunarg.com>\n'
Mark Lobodzinskia9c963d2016-12-28 07:45:35 -0700138 copyright += ' * Author: Courtney Goeltzenleuchter <courtneygo@google.com>\n'
139 copyright += ' * Author: Tobin Ehlis <tobine@google.com>\n'
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600140 copyright += ' * Author: Chris Forbes <chrisforbes@google.com>\n'
John Zulaufde972ac2017-10-26 12:07:05 -0600141 copyright += ' * Author: John Zulauf<jzulauf@lunarg.com>\n'
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700142 copyright += ' *\n'
143 copyright += ' ****************************************************************************/\n'
144 write(copyright, file=self.outFile)
145 #
Mark Lobodzinskia3cc3612017-01-03 13:25:10 -0700146 # Write generated file content to output file
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700147 def endFile(self):
148 dest_file = ''
149 dest_file += self.OutputDestFile()
Mark Lobodzinskiafe10542017-01-03 13:22:44 -0700150 # Remove blank lines at EOF
151 if dest_file.endswith('\n'):
152 dest_file = dest_file[:-1]
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700153 write(dest_file, file=self.outFile);
154 # Finish processing in superclass
155 OutputGenerator.endFile(self)
156 #
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600157 # Override parent class to be notified of the beginning of an extension
158 def beginFeature(self, interface, emit):
159 # Start processing in superclass
160 OutputGenerator.beginFeature(self, interface, emit)
Mark Lobodzinski62f71562017-10-24 13:41:18 -0600161 self.featureExtraProtect = GetFeatureProtect(interface)
162
Mark Lobodzinski31964ca2017-09-18 14:15:09 -0600163 if self.featureName == 'VK_VERSION_1_0' or self.featureName == 'VK_VERSION_1_1':
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600164 return
John Zulauff6feb2a2018-04-12 14:24:57 -0600165 name = self.featureName
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600166 nameElem = interface[0][1]
John Zulauff6feb2a2018-04-12 14:24:57 -0600167 name_define = nameElem.get('name')
168 if 'EXTENSION_NAME' not in name_define:
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600169 print("Error in vk.xml file -- extension name is not available")
John Zulauff6feb2a2018-04-12 14:24:57 -0600170 info = {'define': name_define, 'ifdef': self.featureExtraProtect}
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600171 if interface.get('type') == 'instance':
John Zulauff6feb2a2018-04-12 14:24:57 -0600172 self.instance_extension_info[name] = info
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600173 else:
John Zulauff6feb2a2018-04-12 14:24:57 -0600174 self.device_extension_info[name] = info
175
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600176 #
177 # Override parent class to be notified of the end of an extension
178 def endFeature(self):
179 # Finish processing in superclass
180 OutputGenerator.endFeature(self)
181 #
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700182 # Grab group (e.g. C "enum" type) info to output for enum-string conversion helper
Mike Schuchardtf375c7c2017-12-28 11:23:48 -0700183 def genGroup(self, groupinfo, groupName, alias):
184 OutputGenerator.genGroup(self, groupinfo, groupName, alias)
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700185 groupElem = groupinfo.elem
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700186 # For enum_string_header
187 if self.helper_file_type == 'enum_string_header':
Mike Schuchardtdf1e8dd2018-03-09 09:02:56 -0700188 value_set = set()
Mark Lobodzinski824ce642016-12-28 08:49:46 -0700189 for elem in groupElem.findall('enum'):
Mike Schuchardtf375c7c2017-12-28 11:23:48 -0700190 if elem.get('supported') != 'disabled' and elem.get('alias') == None:
Mike Schuchardtdf1e8dd2018-03-09 09:02:56 -0700191 value_set.add(elem.get('name'))
192 self.enum_output += self.GenerateEnumStringConversion(groupName, value_set)
Mark Young1ded24b2017-05-30 14:53:50 -0600193 elif self.helper_file_type == 'object_types_header':
194 if groupName == 'VkDebugReportObjectTypeEXT':
195 for elem in groupElem.findall('enum'):
196 if elem.get('supported') != 'disabled':
197 item_name = elem.get('name')
198 self.debug_report_object_types.append(item_name)
199 elif groupName == 'VkObjectType':
200 for elem in groupElem.findall('enum'):
201 if elem.get('supported') != 'disabled':
202 item_name = elem.get('name')
203 self.core_object_types.append(item_name)
204
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700205 #
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700206 # Called for each type -- if the type is a struct/union, grab the metadata
Mike Schuchardtf375c7c2017-12-28 11:23:48 -0700207 def genType(self, typeinfo, name, alias):
208 OutputGenerator.genType(self, typeinfo, name, alias)
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700209 typeElem = typeinfo.elem
210 # If the type is a struct type, traverse the imbedded <member> tags generating a structure.
211 # Otherwise, emit the tag text.
212 category = typeElem.get('category')
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600213 if category == 'handle':
John Zulaufd7435c62018-03-16 11:52:57 -0600214 if alias:
215 self.object_type_aliases.append((name,alias))
216 else:
217 self.object_types.append(name)
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600218 elif (category == 'struct' or category == 'union'):
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700219 self.structNames.append(name)
Mike Schuchardtf375c7c2017-12-28 11:23:48 -0700220 self.genStruct(typeinfo, name, alias)
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700221 #
222 # Generate a VkStructureType based on a structure typename
223 def genVkStructureType(self, typename):
224 # Add underscore between lowercase then uppercase
225 value = re.sub('([a-z0-9])([A-Z])', r'\1_\2', typename)
226 # Change to uppercase
227 value = value.upper()
228 # Add STRUCTURE_TYPE_
229 return re.sub('VK_', 'VK_STRUCTURE_TYPE_', value)
230 #
231 # Check if the parameter passed in is a pointer
232 def paramIsPointer(self, param):
233 ispointer = False
234 for elem in param:
235 if ((elem.tag is not 'type') and (elem.tail is not None)) and '*' in elem.tail:
236 ispointer = True
237 return ispointer
238 #
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700239 # Check if the parameter passed in is a static array
240 def paramIsStaticArray(self, param):
241 isstaticarray = 0
242 paramname = param.find('name')
243 if (paramname.tail is not None) and ('[' in paramname.tail):
244 isstaticarray = paramname.tail.count('[')
245 return isstaticarray
246 #
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700247 # Retrieve the type and name for a parameter
248 def getTypeNameTuple(self, param):
249 type = ''
250 name = ''
251 for elem in param:
252 if elem.tag == 'type':
253 type = noneStr(elem.text)
254 elif elem.tag == 'name':
255 name = noneStr(elem.text)
256 return (type, name)
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700257 # Extract length values from latexmath. Currently an inflexible solution that looks for specific
258 # patterns that are found in vk.xml. Will need to be updated when new patterns are introduced.
259 def parseLateXMath(self, source):
260 name = 'ERROR'
261 decoratedName = 'ERROR'
262 if 'mathit' in source:
Mark Lobodzinski36c33862017-02-13 10:15:53 -0700263 # Matches expressions similar to 'latexmath:[\lceil{\mathit{rasterizationSamples} \over 32}\rceil]'
264 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 -0700265 if not match or match.group(1) != match.group(4):
266 raise 'Unrecognized latexmath expression'
267 name = match.group(2)
mizhenc27f6c72017-03-31 09:08:16 -0600268 # Need to add 1 for ceiling function; otherwise, the allocated packet
269 # size will be less than needed during capture for some title which use
270 # this in VkPipelineMultisampleStateCreateInfo. based on ceiling function
271 # definition,it is '{0}%{1}?{0}/{1} + 1:{0}/{1}'.format(*match.group(2, 3)),
272 # its value <= '{}/{} + 1'.
273 if match.group(1) == 'ceil':
274 decoratedName = '{}/{} + 1'.format(*match.group(2, 3))
275 else:
276 decoratedName = '{}/{}'.format(*match.group(2, 3))
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700277 else:
Mark Lobodzinski36c33862017-02-13 10:15:53 -0700278 # Matches expressions similar to 'latexmath : [dataSize \over 4]'
Mark Young0f183a82017-02-28 09:58:04 -0700279 match = re.match(r'latexmath\s*\:\s*\[\s*(\w+)\s*\\over\s*(\d+)\s*\]', source)
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700280 name = match.group(1)
281 decoratedName = '{}/{}'.format(*match.group(1, 2))
282 return name, decoratedName
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700283 #
284 # Retrieve the value of the len tag
285 def getLen(self, param):
286 result = None
287 len = param.attrib.get('len')
288 if len and len != 'null-terminated':
289 # For string arrays, 'len' can look like 'count,null-terminated', indicating that we
290 # have a null terminated array of strings. We strip the null-terminated from the
291 # 'len' field and only return the parameter specifying the string count
292 if 'null-terminated' in len:
293 result = len.split(',')[0]
294 else:
295 result = len
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700296 if 'latexmath' in len:
297 param_type, param_name = self.getTypeNameTuple(param)
298 len_name, result = self.parseLateXMath(len)
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700299 # Spec has now notation for len attributes, using :: instead of platform specific pointer symbol
300 result = str(result).replace('::', '->')
301 return result
302 #
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700303 # Check if a structure is or contains a dispatchable (dispatchable = True) or
304 # non-dispatchable (dispatchable = False) handle
305 def TypeContainsObjectHandle(self, handle_type, dispatchable):
306 if dispatchable:
307 type_key = 'VK_DEFINE_HANDLE'
308 else:
309 type_key = 'VK_DEFINE_NON_DISPATCHABLE_HANDLE'
310 handle = self.registry.tree.find("types/type/[name='" + handle_type + "'][@category='handle']")
311 if handle is not None and handle.find('type').text == type_key:
312 return True
313 # if handle_type is a struct, search its members
314 if handle_type in self.structNames:
315 member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == handle_type), None)
316 if member_index is not None:
317 for item in self.structMembers[member_index].members:
318 handle = self.registry.tree.find("types/type/[name='" + item.type + "'][@category='handle']")
319 if handle is not None and handle.find('type').text == type_key:
320 return True
321 return False
322 #
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700323 # Generate local ready-access data describing Vulkan structures and unions from the XML metadata
Mike Schuchardtf375c7c2017-12-28 11:23:48 -0700324 def genStruct(self, typeinfo, typeName, alias):
325 OutputGenerator.genStruct(self, typeinfo, typeName, alias)
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700326 members = typeinfo.elem.findall('.//member')
327 # Iterate over members once to get length parameters for arrays
328 lens = set()
329 for member in members:
330 len = self.getLen(member)
331 if len:
332 lens.add(len)
333 # Generate member info
334 membersInfo = []
335 for member in members:
336 # Get the member's type and name
337 info = self.getTypeNameTuple(member)
338 type = info[0]
339 name = info[1]
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700340 cdecl = self.makeCParamDecl(member, 1)
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700341 # Process VkStructureType
342 if type == 'VkStructureType':
343 # Extract the required struct type value from the comments
344 # embedded in the original text defining the 'typeinfo' element
345 rawXml = etree.tostring(typeinfo.elem).decode('ascii')
346 result = re.search(r'VK_STRUCTURE_TYPE_\w+', rawXml)
347 if result:
348 value = result.group(0)
349 else:
350 value = self.genVkStructureType(typeName)
351 # Store the required type value
352 self.structTypes[typeName] = self.StructType(name=name, value=value)
353 # Store pointer/array/string info
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700354 isstaticarray = self.paramIsStaticArray(member)
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700355 membersInfo.append(self.CommandParam(type=type,
356 name=name,
357 ispointer=self.paramIsPointer(member),
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700358 isstaticarray=isstaticarray,
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700359 isconst=True if 'const' in cdecl else False,
360 iscount=True if name in lens else False,
361 len=self.getLen(member),
Mike Schuchardta40d0b02017-07-23 12:47:47 -0600362 extstructs=self.registry.validextensionstructs[typeName] if name == 'pNext' else None,
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700363 cdecl=cdecl))
Mark Lobodzinski5380d132016-12-28 14:45:34 -0700364 self.structMembers.append(self.StructMemberData(name=typeName, members=membersInfo, ifdef_protect=self.featureExtraProtect))
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700365 #
366 # Enum_string_header: Create a routine to convert an enumerated value into a string
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700367 def GenerateEnumStringConversion(self, groupName, value_list):
368 outstring = '\n'
369 outstring += 'static inline const char* string_%s(%s input_value)\n' % (groupName, groupName)
370 outstring += '{\n'
371 outstring += ' switch ((%s)input_value)\n' % groupName
372 outstring += ' {\n'
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700373 for item in value_list:
374 outstring += ' case %s:\n' % item
375 outstring += ' return "%s";\n' % item
376 outstring += ' default:\n'
377 outstring += ' return "Unhandled %s";\n' % groupName
378 outstring += ' }\n'
379 outstring += '}\n'
380 return outstring
381 #
Mark Lobodzinski64b432f2017-07-24 16:14:16 -0600382 # Tack on a helper which, given an index into a VkPhysicalDeviceFeatures structure, will print the corresponding feature name
383 def DeIndexPhysDevFeatures(self):
384 pdev_members = None
385 for name, members, ifdef in self.structMembers:
386 if name == 'VkPhysicalDeviceFeatures':
387 pdev_members = members
388 break
389 deindex = '\n'
Mark Young2ee6aea2018-02-21 15:30:27 -0700390 deindex += 'static inline const char * GetPhysDevFeatureString(uint32_t index) {\n'
Mark Lobodzinski64b432f2017-07-24 16:14:16 -0600391 deindex += ' const char * IndexToPhysDevFeatureString[] = {\n'
392 for feature in pdev_members:
393 deindex += ' "%s",\n' % feature.name
394 deindex += ' };\n\n'
395 deindex += ' return IndexToPhysDevFeatureString[index];\n'
396 deindex += '}\n'
397 return deindex
398 #
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700399 # Combine enum string helper header file preamble with body text and return
400 def GenerateEnumStringHelperHeader(self):
401 enum_string_helper_header = '\n'
402 enum_string_helper_header += '#pragma once\n'
403 enum_string_helper_header += '#ifdef _WIN32\n'
404 enum_string_helper_header += '#pragma warning( disable : 4065 )\n'
405 enum_string_helper_header += '#endif\n'
406 enum_string_helper_header += '\n'
407 enum_string_helper_header += '#include <vulkan/vulkan.h>\n'
408 enum_string_helper_header += '\n'
409 enum_string_helper_header += self.enum_output
Mark Lobodzinski64b432f2017-07-24 16:14:16 -0600410 enum_string_helper_header += self.DeIndexPhysDevFeatures()
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700411 return enum_string_helper_header
412 #
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700413 # Helper function for declaring a counter variable only once
414 def DeclareCounter(self, string_var, declare_flag):
415 if declare_flag == False:
416 string_var += ' uint32_t i = 0;\n'
417 declare_flag = True
418 return string_var, declare_flag
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700419 #
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700420 # Combine safe struct helper header file preamble with body text and return
421 def GenerateSafeStructHelperHeader(self):
422 safe_struct_helper_header = '\n'
423 safe_struct_helper_header += '#pragma once\n'
424 safe_struct_helper_header += '#include <vulkan/vulkan.h>\n'
425 safe_struct_helper_header += '\n'
426 safe_struct_helper_header += self.GenerateSafeStructHeader()
427 return safe_struct_helper_header
428 #
429 # safe_struct header: build function prototypes for header file
430 def GenerateSafeStructHeader(self):
431 safe_struct_header = ''
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700432 for item in self.structMembers:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700433 if self.NeedSafeStruct(item) == True:
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700434 safe_struct_header += '\n'
435 if item.ifdef_protect != None:
436 safe_struct_header += '#ifdef %s\n' % item.ifdef_protect
437 safe_struct_header += 'struct safe_%s {\n' % (item.name)
438 for member in item.members:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700439 if member.type in self.structNames:
440 member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None)
441 if member_index is not None and self.NeedSafeStruct(self.structMembers[member_index]) == True:
442 if member.ispointer:
443 safe_struct_header += ' safe_%s* %s;\n' % (member.type, member.name)
444 else:
445 safe_struct_header += ' safe_%s %s;\n' % (member.type, member.name)
446 continue
447 if member.len is not None and (self.TypeContainsObjectHandle(member.type, True) or self.TypeContainsObjectHandle(member.type, False)):
448 safe_struct_header += ' %s* %s;\n' % (member.type, member.name)
449 else:
450 safe_struct_header += '%s;\n' % member.cdecl
Petr Krause91f7a12017-12-14 20:57:36 +0100451 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 -0600452 safe_struct_header += ' safe_%s(const safe_%s& src);\n' % (item.name, item.name)
Chris Forbesfb633832017-10-03 18:11:54 -0700453 safe_struct_header += ' safe_%s& operator=(const safe_%s& src);\n' % (item.name, item.name)
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700454 safe_struct_header += ' safe_%s();\n' % item.name
455 safe_struct_header += ' ~safe_%s();\n' % item.name
Petr Krause91f7a12017-12-14 20:57:36 +0100456 safe_struct_header += ' void initialize(const %s* in_struct%s);\n' % (item.name, self.custom_construct_params.get(item.name, ''))
457 safe_struct_header += ' void initialize(const safe_%s* src);\n' % (item.name)
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700458 safe_struct_header += ' %s *ptr() { return reinterpret_cast<%s *>(this); }\n' % (item.name, item.name)
459 safe_struct_header += ' %s const *ptr() const { return reinterpret_cast<%s const *>(this); }\n' % (item.name, item.name)
460 safe_struct_header += '};\n'
461 if item.ifdef_protect != None:
462 safe_struct_header += '#endif // %s\n' % item.ifdef_protect
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700463 return safe_struct_header
464 #
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600465 # Generate extension helper header file
466 def GenerateExtensionHelperHeader(self):
Mark Lobodzinskibfb7ab92017-10-27 13:22:23 -0600467
468 V_1_0_instance_extensions_promoted_to_core = [
469 'vk_khr_device_group_creation',
470 'vk_khr_external_memory_capabilities',
471 'vk_khr_external_fence_capabilities',
472 'vk_khr_external_semaphore_capabilities',
473 'vk_khr_get_physical_device_properties_2',
474 ]
475
476 V_1_0_device_extensions_promoted_to_core = [
477 'vk_khr_bind_memory_2',
478 'vk_khr_device_group',
479 'vk_khr_descriptor_update_template',
480 'vk_khr_sampler_ycbcr_conversion',
481 'vk_khr_get_memory_requirements_2',
482 'vk_khr_maintenance3',
483 'vk_khr_maintenance1',
484 'vk_khr_multiview',
485 'vk_khr_external_memory',
486 'vk_khr_external_semaphore',
487 'vk_khr_16bit_storage',
488 'vk_khr_external_fence',
489 'vk_khr_maintenance2',
490 'vk_khr_variable_pointers',
491 'vk_khr_dedicated_allocation',
492 ]
John Zulauff6feb2a2018-04-12 14:24:57 -0600493 output = [
494 '',
495 '#ifndef VK_EXTENSION_HELPER_H_',
496 '#define VK_EXTENSION_HELPER_H_',
497 '#include <vulkan/vulkan.h>',
498 '#include <string.h>',
499 '#include <utility>',
500 '',
501 '']
Mark Lobodzinskibfb7ab92017-10-27 13:22:23 -0600502
John Zulauff6feb2a2018-04-12 14:24:57 -0600503 def guarded(ifdef, value):
504 if ifdef is not None:
505 return '\n'.join([ '#ifdef %s' % ifdef, value, '#endif' ])
506 else:
507 return value
John Zulauf380bd942018-04-10 13:12:34 -0600508
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600509 for type in ['Instance', 'Device']:
John Zulauff6feb2a2018-04-12 14:24:57 -0600510 struct_type = '%sExtensions' % type
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600511 if type == 'Instance':
John Zulauff6feb2a2018-04-12 14:24:57 -0600512 extension_dict = self.instance_extension_info
Mark Lobodzinskibfb7ab92017-10-27 13:22:23 -0600513 promoted_ext_list = V_1_0_instance_extensions_promoted_to_core
John Zulauff6feb2a2018-04-12 14:24:57 -0600514 struct_decl = 'struct %s {' % struct_type
515 instance_struct_type = struct_type
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600516 else:
John Zulauff6feb2a2018-04-12 14:24:57 -0600517 extension_dict = self.device_extension_info
Mark Lobodzinskibfb7ab92017-10-27 13:22:23 -0600518 promoted_ext_list = V_1_0_device_extensions_promoted_to_core
John Zulauff6feb2a2018-04-12 14:24:57 -0600519 struct_decl = 'struct %s : public %s {' % (struct_type, instance_struct_type)
520
521 extension_items = sorted(extension_dict.items())
522
523 field_name = { ext_name: re.sub('_extension_name', '', info['define'].lower()) for ext_name, info in extension_items }
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600524 if type == 'Instance':
John Zulauff6feb2a2018-04-12 14:24:57 -0600525 instance_field_name = field_name
Mark Lobodzinskibfb7ab92017-10-27 13:22:23 -0600526
John Zulauff6feb2a2018-04-12 14:24:57 -0600527 struct = [struct_decl]
528 struct.extend([ ' bool %s{false};' % field_name[ext_name] for ext_name, info in extension_items])
529 struct.append('')
530 if type == 'Instance':
531 struct.extend([
532 ' uint32_t NormalizeApiVersion(uint32_t specified_version) {',
533 ' uint32_t api_version = (specified_version < VK_API_VERSION_1_1) ? VK_API_VERSION_1_0 : VK_API_VERSION_1_1;',
534 ' return api_version;',
535 ' }',
536 '',
537 ' uint32_t InitFromInstanceCreateInfo(uint32_t requested_api_version, const VkInstanceCreateInfo *pCreateInfo) {'])
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600538 else:
John Zulauff6feb2a2018-04-12 14:24:57 -0600539 struct.extend([
540 ' uint32_t InitFromDeviceCreateInfo(const InstanceExtensions *instance_extensions, uint32_t requested_api_version,',
541 ' const VkDeviceCreateInfo *pCreateInfo) {'])
Mark Lobodzinskibfb7ab92017-10-27 13:22:23 -0600542
John Zulauff6feb2a2018-04-12 14:24:57 -0600543 struct.extend([
544 '',
545 ' static const std::vector<const char *> V_1_0_promoted_%s_extensions = {' % type.lower() ])
546 struct.extend([' %s_EXTENSION_NAME,' % ext_name.upper() for ext_name in promoted_ext_list])
547 struct.extend([
548 ' };',
549 '',
550 ' static const std::pair<char const *, bool %sExtensions::*> known_extensions[]{' % type])
551 init_format = ' {%s, &' + type + 'Extensions::%s},'
552 struct.extend([guarded(info['ifdef'], init_format % (info['define'], field_name[ext_name])) for ext_name, info in extension_items])
553 struct.extend([
554 ' };',
555 '',
556 ' // Initialize struct data'])
Mark Lobodzinskibfb7ab92017-10-27 13:22:23 -0600557
John Zulauff6feb2a2018-04-12 14:24:57 -0600558 if type == 'Device':
559 struct.extend([ ' %s = instance_extensions->%s;' % (name, name) for name in sorted(instance_field_name.values()) ])
560
561 struct.extend([
562 '',
563 ' for (uint32_t i = 0; i < pCreateInfo->enabledExtensionCount; i++) {',
564 ' for (auto ext : known_extensions) {',
565 ' if (!strcmp(ext.first, pCreateInfo->ppEnabledExtensionNames[i])) {',
566 ' this->*(ext.second) = true;',
567 ' break;',
568 ' }',
569 ' }',
570 ' }',
571 ' uint32_t api_version = NormalizeApiVersion(requested_api_version);',
572 ' if (api_version >= VK_API_VERSION_1_1) {',
573 ' for (auto promoted_ext : V_1_0_promoted_%s_extensions) {' % type.lower(),
574 ' for (auto ext : known_extensions) {',
575 ' if (!strcmp(ext.first, promoted_ext)) {',
576 ' this->*(ext.second) = true;',
577 ' break;',
578 ' }',
579 ' }',
580 ' }',
581 ' }',
582 ' return api_version;',
583 ' }',
584 '};',
585 ''])
586
Mark Lobodzinskifc9451f2018-01-03 11:18:31 -0700587 # Output reference lists of instance/device extension names
John Zulauff6feb2a2018-04-12 14:24:57 -0600588 struct.append('static const char * const k%sExtensionNames = ' % type)
589 struct.extend([guarded(info['ifdef'], ' %s' % info['define']) for ext_name, info in extension_items])
590 struct.extend([';', ''])
591 output.extend(struct)
592
593 output.extend(['', '#endif // VK_EXTENSION_HELPER_H_'])
594 return '\n'.join(output)
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600595 #
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600596 # Combine object types helper header file preamble with body text and return
597 def GenerateObjectTypesHelperHeader(self):
598 object_types_helper_header = '\n'
599 object_types_helper_header += '#pragma once\n'
600 object_types_helper_header += '\n'
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600601 object_types_helper_header += '#include <vulkan/vulkan.h>\n\n'
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600602 object_types_helper_header += self.GenerateObjectTypesHeader()
603 return object_types_helper_header
604 #
605 # Object types header: create object enum type header file
606 def GenerateObjectTypesHeader(self):
Mark Young6ba8abe2017-11-09 10:37:04 -0700607 object_types_header = ''
608 object_types_header += '// Object Type enum for validation layer internal object handling\n'
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600609 object_types_header += 'typedef enum VulkanObjectType {\n'
610 object_types_header += ' kVulkanObjectTypeUnknown = 0,\n'
611 enum_num = 1
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600612 type_list = [];
John Zulaufd7435c62018-03-16 11:52:57 -0600613 enum_entry_map = {}
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600614
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600615 # Output enum definition as each handle is processed, saving the names to use for the conversion routine
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600616 for item in self.object_types:
617 fixup_name = item[2:]
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600618 enum_entry = 'kVulkanObjectType%s' % fixup_name
John Zulaufd7435c62018-03-16 11:52:57 -0600619 enum_entry_map[item] = enum_entry
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600620 object_types_header += ' ' + enum_entry
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600621 object_types_header += ' = %d,\n' % enum_num
622 enum_num += 1
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600623 type_list.append(enum_entry)
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600624 object_types_header += ' kVulkanObjectTypeMax = %d,\n' % enum_num
John Zulaufd7435c62018-03-16 11:52:57 -0600625 object_types_header += ' // Aliases for backwards compatibilty of "promoted" types\n'
626 for (name, alias) in self.object_type_aliases:
627 fixup_name = name[2:]
628 object_types_header += ' kVulkanObjectType{} = {},\n'.format(fixup_name, enum_entry_map[alias])
Mark Lobodzinskiff92ff82017-04-11 15:31:51 -0600629 object_types_header += '} VulkanObjectType;\n\n'
630
631 # Output name string helper
632 object_types_header += '// Array of object name strings for OBJECT_TYPE enum conversion\n'
Mark Lobodzinski8eb37422017-04-18 14:22:10 -0600633 object_types_header += 'static const char * const object_string[kVulkanObjectTypeMax] = {\n'
Mark Lobodzinskiff92ff82017-04-11 15:31:51 -0600634 object_types_header += ' "Unknown",\n'
635 for item in self.object_types:
636 fixup_name = item[2:]
637 object_types_header += ' "%s",\n' % fixup_name
638 object_types_header += '};\n'
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600639
John Zulauf311a4892018-03-12 15:48:06 -0600640 # Key creation helper for map comprehensions that convert between k<Name> and VK<Name> symbols
641 def to_key(regex, raw_key): return re.search(regex, raw_key).group(1).lower().replace("_","")
642
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600643 # Output a conversion routine from the layer object definitions to the debug report definitions
John Zulauf311a4892018-03-12 15:48:06 -0600644 # As the VK_DEBUG_REPORT types are not being updated, specify UNKNOWN for unmatched types
Mark Lobodzinskiff92ff82017-04-11 15:31:51 -0600645 object_types_header += '\n'
Mark Young1ded24b2017-05-30 14:53:50 -0600646 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 -0600647 object_types_header += 'const VkDebugReportObjectTypeEXT get_debug_report_enum[] = {\n'
Mark Young6ba8abe2017-11-09 10:37:04 -0700648 object_types_header += ' VK_DEBUG_REPORT_OBJECT_TYPE_UNKNOWN_EXT, // kVulkanObjectTypeUnknown\n'
Gabríel Arthúr Pétursson1a271d02018-03-18 17:34:01 +0000649
John Zulauf311a4892018-03-12 15:48:06 -0600650 dbg_re = '^VK_DEBUG_REPORT_OBJECT_TYPE_(.*)_EXT$'
651 dbg_map = {to_key(dbg_re, dbg) : dbg for dbg in self.debug_report_object_types}
652 dbg_default = 'VK_DEBUG_REPORT_OBJECT_TYPE_UNKNOWN_EXT'
653 for object_type in type_list:
654 vk_object_type = dbg_map.get(object_type.replace("kVulkanObjectType", "").lower(), dbg_default)
655 object_types_header += ' %s, // %s\n' % (vk_object_type, object_type)
Mark Lobodzinskiecf0ae12017-04-13 08:36:18 -0600656 object_types_header += '};\n'
Mark Young1ded24b2017-05-30 14:53:50 -0600657
658 # Output a conversion routine from the layer object definitions to the core object type definitions
John Zulauf311a4892018-03-12 15:48:06 -0600659 # This will intentionally *fail* for unmatched types as the VK_OBJECT_TYPE list should match the kVulkanObjectType list
Mark Young1ded24b2017-05-30 14:53:50 -0600660 object_types_header += '\n'
661 object_types_header += '// Helper array to get Official Vulkan VkObjectType enum from the internal layers version\n'
662 object_types_header += 'const VkObjectType get_object_type_enum[] = {\n'
Mark Young6ba8abe2017-11-09 10:37:04 -0700663 object_types_header += ' VK_OBJECT_TYPE_UNKNOWN, // kVulkanObjectTypeUnknown\n'
John Zulauf311a4892018-03-12 15:48:06 -0600664
665 vko_re = '^VK_OBJECT_TYPE_(.*)'
666 vko_map = {to_key(vko_re, vko) : vko for vko in self.core_object_types}
Mark Young1ded24b2017-05-30 14:53:50 -0600667 for object_type in type_list:
John Zulauf311a4892018-03-12 15:48:06 -0600668 vk_object_type = vko_map[object_type.replace("kVulkanObjectType", "").lower()]
669 object_types_header += ' %s, // %s\n' % (vk_object_type, object_type)
Mark Young1ded24b2017-05-30 14:53:50 -0600670 object_types_header += '};\n'
671
Mark Young6ba8abe2017-11-09 10:37:04 -0700672 # Create a function to convert from VkDebugReportObjectTypeEXT to VkObjectType
673 object_types_header += '\n'
674 object_types_header += '// Helper function to convert from VkDebugReportObjectTypeEXT to VkObjectType\n'
Mark Young8504ba62018-03-21 13:35:34 -0600675 object_types_header += 'static inline VkObjectType convertDebugReportObjectToCoreObject(VkDebugReportObjectTypeEXT debug_report_obj){\n'
Mark Young6ba8abe2017-11-09 10:37:04 -0700676 object_types_header += ' if (debug_report_obj == VK_DEBUG_REPORT_OBJECT_TYPE_UNKNOWN_EXT) {\n'
677 object_types_header += ' return VK_OBJECT_TYPE_UNKNOWN;\n'
678 for core_object_type in self.core_object_types:
679 core_target_type = core_object_type.replace("VK_OBJECT_TYPE_", "").lower()
680 core_target_type = core_target_type.replace("_", "")
681 for dr_object_type in self.debug_report_object_types:
682 dr_target_type = dr_object_type.replace("VK_DEBUG_REPORT_OBJECT_TYPE_", "").lower()
683 dr_target_type = dr_target_type[:-4]
684 dr_target_type = dr_target_type.replace("_", "")
685 if core_target_type == dr_target_type:
686 object_types_header += ' } else if (debug_report_obj == %s) {\n' % dr_object_type
687 object_types_header += ' return %s;\n' % core_object_type
688 break
689 object_types_header += ' }\n'
690 object_types_header += ' return VK_OBJECT_TYPE_UNKNOWN;\n'
691 object_types_header += '}\n'
692
693 # Create a function to convert from VkObjectType to VkDebugReportObjectTypeEXT
694 object_types_header += '\n'
695 object_types_header += '// Helper function to convert from VkDebugReportObjectTypeEXT to VkObjectType\n'
Mark Young8504ba62018-03-21 13:35:34 -0600696 object_types_header += 'static inline VkDebugReportObjectTypeEXT convertCoreObjectToDebugReportObject(VkObjectType core_report_obj){\n'
Mark Young6ba8abe2017-11-09 10:37:04 -0700697 object_types_header += ' if (core_report_obj == VK_OBJECT_TYPE_UNKNOWN) {\n'
698 object_types_header += ' return VK_DEBUG_REPORT_OBJECT_TYPE_UNKNOWN_EXT;\n'
699 for core_object_type in self.core_object_types:
700 core_target_type = core_object_type.replace("VK_OBJECT_TYPE_", "").lower()
701 core_target_type = core_target_type.replace("_", "")
702 for dr_object_type in self.debug_report_object_types:
703 dr_target_type = dr_object_type.replace("VK_DEBUG_REPORT_OBJECT_TYPE_", "").lower()
704 dr_target_type = dr_target_type[:-4]
705 dr_target_type = dr_target_type.replace("_", "")
706 if core_target_type == dr_target_type:
707 object_types_header += ' } else if (core_report_obj == %s) {\n' % core_object_type
708 object_types_header += ' return %s;\n' % dr_object_type
709 break
710 object_types_header += ' }\n'
711 object_types_header += ' return VK_DEBUG_REPORT_OBJECT_TYPE_UNKNOWN_EXT;\n'
712 object_types_header += '}\n'
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600713 return object_types_header
714 #
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700715 # Determine if a structure needs a safe_struct helper function
716 # That is, it has an sType or one of its members is a pointer
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700717 def NeedSafeStruct(self, structure):
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700718 if 'sType' == structure.name:
719 return True
720 for member in structure.members:
721 if member.ispointer == True:
722 return True
723 return False
724 #
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700725 # Combine safe struct helper source file preamble with body text and return
726 def GenerateSafeStructHelperSource(self):
727 safe_struct_helper_source = '\n'
728 safe_struct_helper_source += '#include "vk_safe_struct.h"\n'
729 safe_struct_helper_source += '#include <string.h>\n'
Mark Lobodzinskibffb7fd2018-03-24 13:24:18 -0600730 safe_struct_helper_source += '#ifdef VK_USE_PLATFORM_ANDROID_KHR\n'
731 safe_struct_helper_source += '#if __ANDROID_API__ < __ANDROID_API_O__\n'
732 safe_struct_helper_source += 'struct AHardwareBuffer {};\n'
733 safe_struct_helper_source += '#endif\n'
734 safe_struct_helper_source += '#endif\n'
735
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700736 safe_struct_helper_source += '\n'
737 safe_struct_helper_source += self.GenerateSafeStructSource()
738 return safe_struct_helper_source
739 #
740 # safe_struct source -- create bodies of safe struct helper functions
741 def GenerateSafeStructSource(self):
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700742 safe_struct_body = []
Mark Lobodzinski560729b2017-03-06 08:59:14 -0700743 wsi_structs = ['VkXlibSurfaceCreateInfoKHR',
744 'VkXcbSurfaceCreateInfoKHR',
745 'VkWaylandSurfaceCreateInfoKHR',
746 'VkMirSurfaceCreateInfoKHR',
747 'VkAndroidSurfaceCreateInfoKHR',
748 'VkWin32SurfaceCreateInfoKHR'
749 ]
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700750 for item in self.structMembers:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700751 if self.NeedSafeStruct(item) == False:
752 continue
Mark Lobodzinski560729b2017-03-06 08:59:14 -0700753 if item.name in wsi_structs:
754 continue
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700755 if item.ifdef_protect != None:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700756 safe_struct_body.append("#ifdef %s\n" % item.ifdef_protect)
757 ss_name = "safe_%s" % item.name
758 init_list = '' # list of members in struct constructor initializer
759 default_init_list = '' # Default constructor just inits ptrs to nullptr in initializer
760 init_func_txt = '' # Txt for initialize() function that takes struct ptr and inits members
761 construct_txt = '' # Body of constuctor as well as body of initialize() func following init_func_txt
762 destruct_txt = ''
Petr Krause91f7a12017-12-14 20:57:36 +0100763
764 custom_construct_txt = {
765 # VkWriteDescriptorSet is special case because pointers may be non-null but ignored
766 'VkWriteDescriptorSet' :
767 ' switch (descriptorType) {\n'
768 ' case VK_DESCRIPTOR_TYPE_SAMPLER:\n'
769 ' case VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER:\n'
770 ' case VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE:\n'
771 ' case VK_DESCRIPTOR_TYPE_STORAGE_IMAGE:\n'
772 ' case VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT:\n'
773 ' if (descriptorCount && in_struct->pImageInfo) {\n'
774 ' pImageInfo = new VkDescriptorImageInfo[descriptorCount];\n'
775 ' for (uint32_t i=0; i<descriptorCount; ++i) {\n'
776 ' pImageInfo[i] = in_struct->pImageInfo[i];\n'
777 ' }\n'
778 ' }\n'
779 ' break;\n'
780 ' case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER:\n'
781 ' case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER:\n'
782 ' case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC:\n'
783 ' case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC:\n'
784 ' if (descriptorCount && in_struct->pBufferInfo) {\n'
785 ' pBufferInfo = new VkDescriptorBufferInfo[descriptorCount];\n'
786 ' for (uint32_t i=0; i<descriptorCount; ++i) {\n'
787 ' pBufferInfo[i] = in_struct->pBufferInfo[i];\n'
788 ' }\n'
789 ' }\n'
790 ' break;\n'
791 ' case VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER:\n'
792 ' case VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER:\n'
793 ' if (descriptorCount && in_struct->pTexelBufferView) {\n'
794 ' pTexelBufferView = new VkBufferView[descriptorCount];\n'
795 ' for (uint32_t i=0; i<descriptorCount; ++i) {\n'
796 ' pTexelBufferView[i] = in_struct->pTexelBufferView[i];\n'
797 ' }\n'
798 ' }\n'
799 ' break;\n'
800 ' default:\n'
801 ' break;\n'
802 ' }\n',
Petr Kraus42f6f8d2017-12-17 17:37:33 +0100803 'VkShaderModuleCreateInfo' :
Petr Krause91f7a12017-12-14 20:57:36 +0100804 ' if (in_struct->pCode) {\n'
805 ' pCode = reinterpret_cast<uint32_t *>(new uint8_t[codeSize]);\n'
806 ' memcpy((void *)pCode, (void *)in_struct->pCode, codeSize);\n'
807 ' }\n',
808 # VkGraphicsPipelineCreateInfo is special case because its pointers may be non-null but ignored
809 'VkGraphicsPipelineCreateInfo' :
810 ' if (stageCount && in_struct->pStages) {\n'
811 ' pStages = new safe_VkPipelineShaderStageCreateInfo[stageCount];\n'
812 ' for (uint32_t i=0; i<stageCount; ++i) {\n'
813 ' pStages[i].initialize(&in_struct->pStages[i]);\n'
814 ' }\n'
815 ' }\n'
816 ' if (in_struct->pVertexInputState)\n'
817 ' pVertexInputState = new safe_VkPipelineVertexInputStateCreateInfo(in_struct->pVertexInputState);\n'
818 ' else\n'
819 ' pVertexInputState = NULL;\n'
820 ' if (in_struct->pInputAssemblyState)\n'
821 ' pInputAssemblyState = new safe_VkPipelineInputAssemblyStateCreateInfo(in_struct->pInputAssemblyState);\n'
822 ' else\n'
823 ' pInputAssemblyState = NULL;\n'
824 ' bool has_tessellation_stage = false;\n'
825 ' if (stageCount && pStages)\n'
826 ' for (uint32_t i=0; i<stageCount && !has_tessellation_stage; ++i)\n'
827 ' if (pStages[i].stage == VK_SHADER_STAGE_TESSELLATION_CONTROL_BIT || pStages[i].stage == VK_SHADER_STAGE_TESSELLATION_EVALUATION_BIT)\n'
828 ' has_tessellation_stage = true;\n'
829 ' if (in_struct->pTessellationState && has_tessellation_stage)\n'
830 ' pTessellationState = new safe_VkPipelineTessellationStateCreateInfo(in_struct->pTessellationState);\n'
831 ' else\n'
832 ' pTessellationState = NULL; // original pTessellationState pointer ignored\n'
833 ' bool has_rasterization = in_struct->pRasterizationState ? !in_struct->pRasterizationState->rasterizerDiscardEnable : false;\n'
834 ' if (in_struct->pViewportState && has_rasterization) {\n'
835 ' bool is_dynamic_viewports = false;\n'
836 ' bool is_dynamic_scissors = false;\n'
837 ' if (in_struct->pDynamicState && in_struct->pDynamicState->pDynamicStates) {\n'
838 ' for (uint32_t i = 0; i < in_struct->pDynamicState->dynamicStateCount && !is_dynamic_viewports; ++i)\n'
839 ' if (in_struct->pDynamicState->pDynamicStates[i] == VK_DYNAMIC_STATE_VIEWPORT)\n'
840 ' is_dynamic_viewports = true;\n'
841 ' for (uint32_t i = 0; i < in_struct->pDynamicState->dynamicStateCount && !is_dynamic_scissors; ++i)\n'
842 ' if (in_struct->pDynamicState->pDynamicStates[i] == VK_DYNAMIC_STATE_SCISSOR)\n'
843 ' is_dynamic_scissors = true;\n'
844 ' }\n'
845 ' pViewportState = new safe_VkPipelineViewportStateCreateInfo(in_struct->pViewportState, is_dynamic_viewports, is_dynamic_scissors);\n'
846 ' } else\n'
847 ' pViewportState = NULL; // original pViewportState pointer ignored\n'
848 ' if (in_struct->pRasterizationState)\n'
849 ' pRasterizationState = new safe_VkPipelineRasterizationStateCreateInfo(in_struct->pRasterizationState);\n'
850 ' else\n'
851 ' pRasterizationState = NULL;\n'
852 ' if (in_struct->pMultisampleState && has_rasterization)\n'
853 ' pMultisampleState = new safe_VkPipelineMultisampleStateCreateInfo(in_struct->pMultisampleState);\n'
854 ' else\n'
855 ' pMultisampleState = NULL; // original pMultisampleState pointer ignored\n'
856 ' // needs a tracked subpass state uses_depthstencil_attachment\n'
857 ' if (in_struct->pDepthStencilState && has_rasterization && uses_depthstencil_attachment)\n'
858 ' pDepthStencilState = new safe_VkPipelineDepthStencilStateCreateInfo(in_struct->pDepthStencilState);\n'
859 ' else\n'
860 ' pDepthStencilState = NULL; // original pDepthStencilState pointer ignored\n'
861 ' // needs a tracked subpass state usesColorAttachment\n'
862 ' if (in_struct->pColorBlendState && has_rasterization && uses_color_attachment)\n'
863 ' pColorBlendState = new safe_VkPipelineColorBlendStateCreateInfo(in_struct->pColorBlendState);\n'
864 ' else\n'
865 ' pColorBlendState = NULL; // original pColorBlendState pointer ignored\n'
866 ' if (in_struct->pDynamicState)\n'
867 ' pDynamicState = new safe_VkPipelineDynamicStateCreateInfo(in_struct->pDynamicState);\n'
868 ' else\n'
869 ' pDynamicState = NULL;\n',
870 # VkPipelineViewportStateCreateInfo is special case because its pointers may be non-null but ignored
871 'VkPipelineViewportStateCreateInfo' :
872 ' if (in_struct->pViewports && !is_dynamic_viewports) {\n'
873 ' pViewports = new VkViewport[in_struct->viewportCount];\n'
874 ' memcpy ((void *)pViewports, (void *)in_struct->pViewports, sizeof(VkViewport)*in_struct->viewportCount);\n'
875 ' }\n'
876 ' else\n'
877 ' pViewports = NULL;\n'
878 ' if (in_struct->pScissors && !is_dynamic_scissors) {\n'
879 ' pScissors = new VkRect2D[in_struct->scissorCount];\n'
880 ' memcpy ((void *)pScissors, (void *)in_struct->pScissors, sizeof(VkRect2D)*in_struct->scissorCount);\n'
881 ' }\n'
882 ' else\n'
883 ' pScissors = NULL;\n',
Petr Kraus42f6f8d2017-12-17 17:37:33 +0100884 # VkDescriptorSetLayoutBinding is special case because its pImmutableSamplers pointer may be non-null but ignored
885 'VkDescriptorSetLayoutBinding' :
886 ' const bool sampler_type = in_struct->descriptorType == VK_DESCRIPTOR_TYPE_SAMPLER || in_struct->descriptorType == VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER;\n'
887 ' if (descriptorCount && in_struct->pImmutableSamplers && sampler_type) {\n'
888 ' pImmutableSamplers = new VkSampler[descriptorCount];\n'
889 ' for (uint32_t i=0; i<descriptorCount; ++i) {\n'
890 ' pImmutableSamplers[i] = in_struct->pImmutableSamplers[i];\n'
891 ' }\n'
892 ' }\n',
Petr Krause91f7a12017-12-14 20:57:36 +0100893 }
894
895 custom_copy_txt = {
896 # VkGraphicsPipelineCreateInfo is special case because it has custom construct parameters
897 'VkGraphicsPipelineCreateInfo' :
898 ' if (stageCount && src.pStages) {\n'
899 ' pStages = new safe_VkPipelineShaderStageCreateInfo[stageCount];\n'
900 ' for (uint32_t i=0; i<stageCount; ++i) {\n'
901 ' pStages[i].initialize(&src.pStages[i]);\n'
902 ' }\n'
903 ' }\n'
904 ' if (src.pVertexInputState)\n'
905 ' pVertexInputState = new safe_VkPipelineVertexInputStateCreateInfo(*src.pVertexInputState);\n'
906 ' else\n'
907 ' pVertexInputState = NULL;\n'
908 ' if (src.pInputAssemblyState)\n'
909 ' pInputAssemblyState = new safe_VkPipelineInputAssemblyStateCreateInfo(*src.pInputAssemblyState);\n'
910 ' else\n'
911 ' pInputAssemblyState = NULL;\n'
912 ' bool has_tessellation_stage = false;\n'
913 ' if (stageCount && pStages)\n'
914 ' for (uint32_t i=0; i<stageCount && !has_tessellation_stage; ++i)\n'
915 ' if (pStages[i].stage == VK_SHADER_STAGE_TESSELLATION_CONTROL_BIT || pStages[i].stage == VK_SHADER_STAGE_TESSELLATION_EVALUATION_BIT)\n'
916 ' has_tessellation_stage = true;\n'
917 ' if (src.pTessellationState && has_tessellation_stage)\n'
918 ' pTessellationState = new safe_VkPipelineTessellationStateCreateInfo(*src.pTessellationState);\n'
919 ' else\n'
920 ' pTessellationState = NULL; // original pTessellationState pointer ignored\n'
921 ' bool has_rasterization = src.pRasterizationState ? !src.pRasterizationState->rasterizerDiscardEnable : false;\n'
922 ' if (src.pViewportState && has_rasterization) {\n'
923 ' pViewportState = new safe_VkPipelineViewportStateCreateInfo(*src.pViewportState);\n'
924 ' } else\n'
925 ' pViewportState = NULL; // original pViewportState pointer ignored\n'
926 ' if (src.pRasterizationState)\n'
927 ' pRasterizationState = new safe_VkPipelineRasterizationStateCreateInfo(*src.pRasterizationState);\n'
928 ' else\n'
929 ' pRasterizationState = NULL;\n'
930 ' if (src.pMultisampleState && has_rasterization)\n'
931 ' pMultisampleState = new safe_VkPipelineMultisampleStateCreateInfo(*src.pMultisampleState);\n'
932 ' else\n'
933 ' pMultisampleState = NULL; // original pMultisampleState pointer ignored\n'
934 ' if (src.pDepthStencilState && has_rasterization)\n'
935 ' pDepthStencilState = new safe_VkPipelineDepthStencilStateCreateInfo(*src.pDepthStencilState);\n'
936 ' else\n'
937 ' pDepthStencilState = NULL; // original pDepthStencilState pointer ignored\n'
938 ' if (src.pColorBlendState && has_rasterization)\n'
939 ' pColorBlendState = new safe_VkPipelineColorBlendStateCreateInfo(*src.pColorBlendState);\n'
940 ' else\n'
941 ' pColorBlendState = NULL; // original pColorBlendState pointer ignored\n'
942 ' if (src.pDynamicState)\n'
943 ' pDynamicState = new safe_VkPipelineDynamicStateCreateInfo(*src.pDynamicState);\n'
944 ' else\n'
945 ' pDynamicState = NULL;\n',
946 # VkPipelineViewportStateCreateInfo is special case because it has custom construct parameters
947 'VkPipelineViewportStateCreateInfo' :
948 ' if (src.pViewports) {\n'
949 ' pViewports = new VkViewport[src.viewportCount];\n'
950 ' memcpy ((void *)pViewports, (void *)src.pViewports, sizeof(VkViewport)*src.viewportCount);\n'
951 ' }\n'
952 ' else\n'
953 ' pViewports = NULL;\n'
954 ' if (src.pScissors) {\n'
955 ' pScissors = new VkRect2D[src.scissorCount];\n'
956 ' memcpy ((void *)pScissors, (void *)src.pScissors, sizeof(VkRect2D)*src.scissorCount);\n'
957 ' }\n'
958 ' else\n'
959 ' pScissors = NULL;\n',
960 }
961
Mike Schuchardt81485762017-09-04 11:38:42 -0600962 custom_destruct_txt = {'VkShaderModuleCreateInfo' :
963 ' if (pCode)\n'
964 ' delete[] reinterpret_cast<const uint8_t *>(pCode);\n' }
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700965
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700966 for member in item.members:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700967 m_type = member.type
968 if member.type in self.structNames:
969 member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None)
970 if member_index is not None and self.NeedSafeStruct(self.structMembers[member_index]) == True:
971 m_type = 'safe_%s' % member.type
972 if member.ispointer and 'safe_' not in m_type and self.TypeContainsObjectHandle(member.type, False) == False:
973 # 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 -0700974 if m_type in ['void', 'char']:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700975 # For these exceptions just copy initial value over for now
976 init_list += '\n %s(in_struct->%s),' % (member.name, member.name)
977 init_func_txt += ' %s = in_struct->%s;\n' % (member.name, member.name)
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700978 else:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700979 default_init_list += '\n %s(nullptr),' % (member.name)
980 init_list += '\n %s(nullptr),' % (member.name)
981 init_func_txt += ' %s = nullptr;\n' % (member.name)
982 if 'pNext' != member.name and 'void' not in m_type:
Mark Lobodzinski51160a12017-01-18 11:05:48 -0700983 if not member.isstaticarray and (member.len is None or '/' in member.len):
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700984 construct_txt += ' if (in_struct->%s) {\n' % member.name
985 construct_txt += ' %s = new %s(*in_struct->%s);\n' % (member.name, m_type, member.name)
986 construct_txt += ' }\n'
987 destruct_txt += ' if (%s)\n' % member.name
988 destruct_txt += ' delete %s;\n' % member.name
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700989 else:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700990 construct_txt += ' if (in_struct->%s) {\n' % member.name
991 construct_txt += ' %s = new %s[in_struct->%s];\n' % (member.name, m_type, member.len)
992 construct_txt += ' memcpy ((void *)%s, (void *)in_struct->%s, sizeof(%s)*in_struct->%s);\n' % (member.name, member.name, m_type, member.len)
993 construct_txt += ' }\n'
994 destruct_txt += ' if (%s)\n' % member.name
995 destruct_txt += ' delete[] %s;\n' % member.name
996 elif member.isstaticarray or member.len is not None:
997 if member.len is None:
998 # Extract length of static array by grabbing val between []
999 static_array_size = re.match(r"[^[]*\[([^]]*)\]", member.cdecl)
1000 construct_txt += ' for (uint32_t i=0; i<%s; ++i) {\n' % static_array_size.group(1)
1001 construct_txt += ' %s[i] = in_struct->%s[i];\n' % (member.name, member.name)
1002 construct_txt += ' }\n'
1003 else:
1004 # Init array ptr to NULL
1005 default_init_list += '\n %s(nullptr),' % member.name
1006 init_list += '\n %s(nullptr),' % member.name
1007 init_func_txt += ' %s = nullptr;\n' % member.name
1008 array_element = 'in_struct->%s[i]' % member.name
1009 if member.type in self.structNames:
1010 member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None)
1011 if member_index is not None and self.NeedSafeStruct(self.structMembers[member_index]) == True:
1012 array_element = '%s(&in_struct->safe_%s[i])' % (member.type, member.name)
1013 construct_txt += ' if (%s && in_struct->%s) {\n' % (member.len, member.name)
1014 construct_txt += ' %s = new %s[%s];\n' % (member.name, m_type, member.len)
1015 destruct_txt += ' if (%s)\n' % member.name
1016 destruct_txt += ' delete[] %s;\n' % member.name
1017 construct_txt += ' for (uint32_t i=0; i<%s; ++i) {\n' % (member.len)
1018 if 'safe_' in m_type:
1019 construct_txt += ' %s[i].initialize(&in_struct->%s[i]);\n' % (member.name, member.name)
Mark Lobodzinskie20e4562017-01-03 11:14:26 -07001020 else:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -07001021 construct_txt += ' %s[i] = %s;\n' % (member.name, array_element)
1022 construct_txt += ' }\n'
1023 construct_txt += ' }\n'
1024 elif member.ispointer == True:
1025 construct_txt += ' if (in_struct->%s)\n' % member.name
1026 construct_txt += ' %s = new %s(in_struct->%s);\n' % (member.name, m_type, member.name)
1027 construct_txt += ' else\n'
1028 construct_txt += ' %s = NULL;\n' % member.name
1029 destruct_txt += ' if (%s)\n' % member.name
1030 destruct_txt += ' delete %s;\n' % member.name
1031 elif 'safe_' in m_type:
1032 init_list += '\n %s(&in_struct->%s),' % (member.name, member.name)
1033 init_func_txt += ' %s.initialize(&in_struct->%s);\n' % (member.name, member.name)
1034 else:
1035 init_list += '\n %s(in_struct->%s),' % (member.name, member.name)
1036 init_func_txt += ' %s = in_struct->%s;\n' % (member.name, member.name)
1037 if '' != init_list:
1038 init_list = init_list[:-1] # hack off final comma
1039 if item.name in custom_construct_txt:
1040 construct_txt = custom_construct_txt[item.name]
Mike Schuchardt81485762017-09-04 11:38:42 -06001041 if item.name in custom_destruct_txt:
1042 destruct_txt = custom_destruct_txt[item.name]
Petr Krause91f7a12017-12-14 20:57:36 +01001043 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 -07001044 if '' != default_init_list:
1045 default_init_list = " :%s" % (default_init_list[:-1])
1046 safe_struct_body.append("\n%s::%s()%s\n{}" % (ss_name, ss_name, default_init_list))
1047 # Create slight variation of init and construct txt for copy constructor that takes a src object reference vs. struct ptr
1048 copy_construct_init = init_func_txt.replace('in_struct->', 'src.')
1049 copy_construct_txt = construct_txt.replace(' (in_struct->', ' (src.') # Exclude 'if' blocks from next line
1050 copy_construct_txt = copy_construct_txt.replace('(in_struct->', '(*src.') # Pass object to copy constructors
1051 copy_construct_txt = copy_construct_txt.replace('in_struct->', 'src.') # Modify remaining struct refs for src object
Petr Krause91f7a12017-12-14 20:57:36 +01001052 if item.name in custom_copy_txt:
1053 copy_construct_txt = custom_copy_txt[item.name]
Chris Forbesfb633832017-10-03 18:11:54 -07001054 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 -06001055 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 -07001056 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 -07001057 safe_struct_body.append("\n%s::~%s()\n{\n%s}" % (ss_name, ss_name, destruct_txt))
Petr Krause91f7a12017-12-14 20:57:36 +01001058 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 -07001059 # Copy initializer uses same txt as copy constructor but has a ptr and not a reference
1060 init_copy = copy_construct_init.replace('src.', 'src->')
1061 init_construct = copy_construct_txt.replace('src.', 'src->')
Mark Lobodzinski5cd08512017-09-12 09:50:25 -06001062 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 -07001063 if item.ifdef_protect != None:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -07001064 safe_struct_body.append("#endif // %s\n" % item.ifdef_protect)
1065 return "\n".join(safe_struct_body)
Mark Lobodzinskie20e4562017-01-03 11:14:26 -07001066 #
John Zulaufde972ac2017-10-26 12:07:05 -06001067 # Generate the type map
1068 def GenerateTypeMapHelperHeader(self):
1069 prefix = 'Lvl'
1070 fprefix = 'lvl_'
1071 typemap = prefix + 'TypeMap'
1072 idmap = prefix + 'STypeMap'
John Zulaufde972ac2017-10-26 12:07:05 -06001073 type_member = 'Type'
1074 id_member = 'kSType'
Mike Schuchardt97662b02017-12-06 13:31:29 -07001075 id_decl = 'static const VkStructureType '
John Zulaufde972ac2017-10-26 12:07:05 -06001076 generic_header = prefix + 'GenericHeader'
1077 typename_func = fprefix + 'typename'
1078 idname_func = fprefix + 'stype_name'
1079 find_func = fprefix + 'find_in_chain'
John Zulauf65ac9d52018-01-23 11:20:50 -07001080 init_func = fprefix + 'init_struct'
John Zulaufde972ac2017-10-26 12:07:05 -06001081
1082 explanatory_comment = '\n'.join((
1083 '// These empty generic templates are specialized for each type with sType',
1084 '// members and for each sType -- providing a two way map between structure',
Mike Schuchardt97662b02017-12-06 13:31:29 -07001085 '// types and sTypes'))
John Zulaufde972ac2017-10-26 12:07:05 -06001086
1087 empty_typemap = 'template <typename T> struct ' + typemap + ' {};'
1088 typemap_format = 'template <> struct {template}<{typename}> {{\n'
John Zulaufde972ac2017-10-26 12:07:05 -06001089 typemap_format += ' {id_decl}{id_member} = {id_value};\n'
1090 typemap_format += '}};\n'
1091
1092 empty_idmap = 'template <VkStructureType id> struct ' + idmap + ' {};'
1093 idmap_format = ''.join((
1094 'template <> struct {template}<{id_value}> {{\n',
1095 ' typedef {typename} {typedef};\n',
John Zulaufde972ac2017-10-26 12:07:05 -06001096 '}};\n'))
1097
1098 # Define the utilities (here so any renaming stays consistent), if this grows large, refactor to a fixed .h file
1099 utilities_format = '\n'.join((
1100 '// Header "base class" for pNext chain traversal',
1101 'struct {header} {{',
1102 ' VkStructureType sType;',
1103 ' const {header} *pNext;',
1104 '}};',
1105 '',
1106 '// Find an entry of the given type in the pNext chain',
1107 'template <typename T> const T *{find_func}(const void *next) {{',
1108 ' const {header} *current = reinterpret_cast<const {header} *>(next);',
1109 ' const T *found = nullptr;',
1110 ' while (current) {{',
1111 ' if ({type_map}<T>::{id_member} == current->sType) {{',
1112 ' found = reinterpret_cast<const T*>(current);',
1113 ' current = nullptr;',
1114 ' }} else {{',
1115 ' current = current->pNext;',
1116 ' }}',
1117 ' }}',
1118 ' return found;',
1119 '}}',
John Zulauf65ac9d52018-01-23 11:20:50 -07001120 '',
1121 '// Init the header of an sType struct with pNext',
1122 'template <typename T> T {init_func}(void *p_next) {{',
1123 ' T out = {{}};',
1124 ' out.sType = {type_map}<T>::kSType;',
1125 ' out.pNext = p_next;',
1126 ' return out;',
1127 '}}',
1128 '',
1129 '// Init the header of an sType struct',
1130 'template <typename T> T {init_func}() {{',
1131 ' T out = {{}};',
1132 ' out.sType = {type_map}<T>::kSType;',
1133 ' return out;',
1134 '}}',
1135
Mike Schuchardt97662b02017-12-06 13:31:29 -07001136 ''))
John Zulaufde972ac2017-10-26 12:07:05 -06001137
1138 code = []
John Zulauf65ac9d52018-01-23 11:20:50 -07001139
1140 # Generate header
John Zulaufde972ac2017-10-26 12:07:05 -06001141 code.append('\n'.join((
1142 '#pragma once',
1143 '#include <vulkan/vulkan.h>\n',
1144 explanatory_comment, '',
1145 empty_idmap,
John Zulauf65ac9d52018-01-23 11:20:50 -07001146 empty_typemap, '')))
John Zulaufde972ac2017-10-26 12:07:05 -06001147
1148 # Generate the specializations for each type and stype
John Zulaufde972ac2017-10-26 12:07:05 -06001149 for item in self.structMembers:
1150 typename = item.name
1151 info = self.structTypes.get(typename)
1152 if not info:
1153 continue
1154
1155 if item.ifdef_protect != None:
1156 code.append('#ifdef %s' % item.ifdef_protect)
1157
1158 code.append('// Map type {} to id {}'.format(typename, info.value))
1159 code.append(typemap_format.format(template=typemap, typename=typename, id_value=info.value,
Mike Schuchardt97662b02017-12-06 13:31:29 -07001160 id_decl=id_decl, id_member=id_member))
1161 code.append(idmap_format.format(template=idmap, typename=typename, id_value=info.value, typedef=type_member))
John Zulaufde972ac2017-10-26 12:07:05 -06001162
1163 if item.ifdef_protect != None:
1164 code.append('#endif // %s' % item.ifdef_protect)
1165
John Zulauf65ac9d52018-01-23 11:20:50 -07001166 # Generate utilities for all types
1167 code.append('\n'.join((
1168 utilities_format.format(id_member=id_member, id_map=idmap, type_map=typemap,
1169 type_member=type_member, header=generic_header, typename_func=typename_func, idname_func=idname_func,
1170 find_func=find_func, init_func=init_func), ''
1171 )))
1172
John Zulaufde972ac2017-10-26 12:07:05 -06001173 return "\n".join(code)
1174
1175 #
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -07001176 # Create a helper file and return it as a string
1177 def OutputDestFile(self):
Mark Lobodzinski46d388f2016-12-28 10:46:26 -07001178 if self.helper_file_type == 'enum_string_header':
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -07001179 return self.GenerateEnumStringHelperHeader()
Mark Lobodzinskie20e4562017-01-03 11:14:26 -07001180 elif self.helper_file_type == 'safe_struct_header':
1181 return self.GenerateSafeStructHelperHeader()
1182 elif self.helper_file_type == 'safe_struct_source':
1183 return self.GenerateSafeStructHelperSource()
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -06001184 elif self.helper_file_type == 'object_types_header':
1185 return self.GenerateObjectTypesHelperHeader()
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -06001186 elif self.helper_file_type == 'extension_helper_header':
1187 return self.GenerateExtensionHelperHeader()
John Zulaufde972ac2017-10-26 12:07:05 -06001188 elif self.helper_file_type == 'typemap_helper_header':
1189 return self.GenerateTypeMapHelperHeader()
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -07001190 else:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -07001191 return 'Bad Helper File Generator Option %s' % self.helper_file_type
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -07001192