blob: 0d35b8370ba9a7013d9dd59a2d1b041f2d6c53a9 [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 Lobodzinski46d388f2016-12-28 10:46:26 -070086 # Named tuples to store struct and command data
87 self.StructType = namedtuple('StructType', ['name', 'value'])
Mark Lobodzinskic67efd02017-01-04 09:16:00 -070088 self.CommandParam = namedtuple('CommandParam', ['type', 'name', 'ispointer', 'isstaticarray', 'isconst', 'iscount', 'len', 'extstructs', 'cdecl'])
Mark Lobodzinski5380d132016-12-28 14:45:34 -070089 self.StructMemberData = namedtuple('StructMemberData', ['name', 'members', 'ifdef_protect'])
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070090 #
91 # Called once at the beginning of each run
92 def beginFile(self, genOpts):
93 OutputGenerator.beginFile(self, genOpts)
94 # User-supplied prefix text, if any (list of strings)
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070095 self.helper_file_type = genOpts.helper_file_type
96 self.library_name = genOpts.library_name
97 # File Comment
98 file_comment = '// *** THIS FILE IS GENERATED - DO NOT EDIT ***\n'
99 file_comment += '// See helper_file_generator.py for modifications\n'
100 write(file_comment, file=self.outFile)
101 # Copyright Notice
102 copyright = ''
103 copyright += '\n'
104 copyright += '/***************************************************************************\n'
105 copyright += ' *\n'
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700106 copyright += ' * Copyright (c) 2015-2017 The Khronos Group Inc.\n'
107 copyright += ' * Copyright (c) 2015-2017 Valve Corporation\n'
108 copyright += ' * Copyright (c) 2015-2017 LunarG, Inc.\n'
109 copyright += ' * Copyright (c) 2015-2017 Google Inc.\n'
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700110 copyright += ' *\n'
111 copyright += ' * Licensed under the Apache License, Version 2.0 (the "License");\n'
112 copyright += ' * you may not use this file except in compliance with the License.\n'
113 copyright += ' * You may obtain a copy of the License at\n'
114 copyright += ' *\n'
115 copyright += ' * http://www.apache.org/licenses/LICENSE-2.0\n'
116 copyright += ' *\n'
117 copyright += ' * Unless required by applicable law or agreed to in writing, software\n'
118 copyright += ' * distributed under the License is distributed on an "AS IS" BASIS,\n'
119 copyright += ' * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n'
120 copyright += ' * See the License for the specific language governing permissions and\n'
121 copyright += ' * limitations under the License.\n'
122 copyright += ' *\n'
123 copyright += ' * Author: Mark Lobodzinski <mark@lunarg.com>\n'
Mark Lobodzinskia9c963d2016-12-28 07:45:35 -0700124 copyright += ' * Author: Courtney Goeltzenleuchter <courtneygo@google.com>\n'
125 copyright += ' * Author: Tobin Ehlis <tobine@google.com>\n'
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700126 copyright += ' *\n'
127 copyright += ' ****************************************************************************/\n'
128 write(copyright, file=self.outFile)
129 #
Mark Lobodzinskia3cc3612017-01-03 13:25:10 -0700130 # Write generated file content to output file
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700131 def endFile(self):
132 dest_file = ''
133 dest_file += self.OutputDestFile()
Mark Lobodzinskiafe10542017-01-03 13:22:44 -0700134 # Remove blank lines at EOF
135 if dest_file.endswith('\n'):
136 dest_file = dest_file[:-1]
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700137 write(dest_file, file=self.outFile);
138 # Finish processing in superclass
139 OutputGenerator.endFile(self)
140 #
141 # Grab group (e.g. C "enum" type) info to output for enum-string conversion helper
142 def genGroup(self, groupinfo, groupName):
143 OutputGenerator.genGroup(self, groupinfo, groupName)
144 groupElem = groupinfo.elem
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700145 # For enum_string_header
146 if self.helper_file_type == 'enum_string_header':
Mark Lobodzinski824ce642016-12-28 08:49:46 -0700147 value_list = []
148 for elem in groupElem.findall('enum'):
149 if elem.get('supported') != 'disabled':
150 item_name = elem.get('name')
151 value_list.append(item_name)
152 if value_list is not None:
153 self.enum_output += self.GenerateEnumStringConversion(groupName, value_list)
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700154 #
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700155 # Called for each type -- if the type is a struct/union, grab the metadata
156 def genType(self, typeinfo, name):
157 OutputGenerator.genType(self, typeinfo, name)
158 typeElem = typeinfo.elem
159 # If the type is a struct type, traverse the imbedded <member> tags generating a structure.
160 # Otherwise, emit the tag text.
161 category = typeElem.get('category')
162 if (category == 'struct' or category == 'union'):
163 self.structNames.append(name)
164 self.genStruct(typeinfo, name)
165 #
166 # Generate a VkStructureType based on a structure typename
167 def genVkStructureType(self, typename):
168 # Add underscore between lowercase then uppercase
169 value = re.sub('([a-z0-9])([A-Z])', r'\1_\2', typename)
170 # Change to uppercase
171 value = value.upper()
172 # Add STRUCTURE_TYPE_
173 return re.sub('VK_', 'VK_STRUCTURE_TYPE_', value)
174 #
175 # Check if the parameter passed in is a pointer
176 def paramIsPointer(self, param):
177 ispointer = False
178 for elem in param:
179 if ((elem.tag is not 'type') and (elem.tail is not None)) and '*' in elem.tail:
180 ispointer = True
181 return ispointer
182 #
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700183 # Check if the parameter passed in is a static array
184 def paramIsStaticArray(self, param):
185 isstaticarray = 0
186 paramname = param.find('name')
187 if (paramname.tail is not None) and ('[' in paramname.tail):
188 isstaticarray = paramname.tail.count('[')
189 return isstaticarray
190 #
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700191 # Retrieve the type and name for a parameter
192 def getTypeNameTuple(self, param):
193 type = ''
194 name = ''
195 for elem in param:
196 if elem.tag == 'type':
197 type = noneStr(elem.text)
198 elif elem.tag == 'name':
199 name = noneStr(elem.text)
200 return (type, name)
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700201 # Extract length values from latexmath. Currently an inflexible solution that looks for specific
202 # patterns that are found in vk.xml. Will need to be updated when new patterns are introduced.
203 def parseLateXMath(self, source):
204 name = 'ERROR'
205 decoratedName = 'ERROR'
206 if 'mathit' in source:
Mark Lobodzinski36c33862017-02-13 10:15:53 -0700207 # Matches expressions similar to 'latexmath:[\lceil{\mathit{rasterizationSamples} \over 32}\rceil]'
208 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 -0700209 if not match or match.group(1) != match.group(4):
210 raise 'Unrecognized latexmath expression'
211 name = match.group(2)
mizhenc27f6c72017-03-31 09:08:16 -0600212 # Need to add 1 for ceiling function; otherwise, the allocated packet
213 # size will be less than needed during capture for some title which use
214 # this in VkPipelineMultisampleStateCreateInfo. based on ceiling function
215 # definition,it is '{0}%{1}?{0}/{1} + 1:{0}/{1}'.format(*match.group(2, 3)),
216 # its value <= '{}/{} + 1'.
217 if match.group(1) == 'ceil':
218 decoratedName = '{}/{} + 1'.format(*match.group(2, 3))
219 else:
220 decoratedName = '{}/{}'.format(*match.group(2, 3))
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700221 else:
Mark Lobodzinski36c33862017-02-13 10:15:53 -0700222 # Matches expressions similar to 'latexmath : [dataSize \over 4]'
Mark Young0f183a82017-02-28 09:58:04 -0700223 match = re.match(r'latexmath\s*\:\s*\[\s*(\w+)\s*\\over\s*(\d+)\s*\]', source)
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700224 name = match.group(1)
225 decoratedName = '{}/{}'.format(*match.group(1, 2))
226 return name, decoratedName
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700227 #
228 # Retrieve the value of the len tag
229 def getLen(self, param):
230 result = None
231 len = param.attrib.get('len')
232 if len and len != 'null-terminated':
233 # For string arrays, 'len' can look like 'count,null-terminated', indicating that we
234 # have a null terminated array of strings. We strip the null-terminated from the
235 # 'len' field and only return the parameter specifying the string count
236 if 'null-terminated' in len:
237 result = len.split(',')[0]
238 else:
239 result = len
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700240 if 'latexmath' in len:
241 param_type, param_name = self.getTypeNameTuple(param)
242 len_name, result = self.parseLateXMath(len)
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700243 # Spec has now notation for len attributes, using :: instead of platform specific pointer symbol
244 result = str(result).replace('::', '->')
245 return result
246 #
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700247 # Check if a structure is or contains a dispatchable (dispatchable = True) or
248 # non-dispatchable (dispatchable = False) handle
249 def TypeContainsObjectHandle(self, handle_type, dispatchable):
250 if dispatchable:
251 type_key = 'VK_DEFINE_HANDLE'
252 else:
253 type_key = 'VK_DEFINE_NON_DISPATCHABLE_HANDLE'
254 handle = self.registry.tree.find("types/type/[name='" + handle_type + "'][@category='handle']")
255 if handle is not None and handle.find('type').text == type_key:
256 return True
257 # if handle_type is a struct, search its members
258 if handle_type in self.structNames:
259 member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == handle_type), None)
260 if member_index is not None:
261 for item in self.structMembers[member_index].members:
262 handle = self.registry.tree.find("types/type/[name='" + item.type + "'][@category='handle']")
263 if handle is not None and handle.find('type').text == type_key:
264 return True
265 return False
266 #
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700267 # Generate local ready-access data describing Vulkan structures and unions from the XML metadata
268 def genStruct(self, typeinfo, typeName):
269 OutputGenerator.genStruct(self, typeinfo, typeName)
270 members = typeinfo.elem.findall('.//member')
271 # Iterate over members once to get length parameters for arrays
272 lens = set()
273 for member in members:
274 len = self.getLen(member)
275 if len:
276 lens.add(len)
277 # Generate member info
278 membersInfo = []
279 for member in members:
280 # Get the member's type and name
281 info = self.getTypeNameTuple(member)
282 type = info[0]
283 name = info[1]
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700284 cdecl = self.makeCParamDecl(member, 1)
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700285 # Process VkStructureType
286 if type == 'VkStructureType':
287 # Extract the required struct type value from the comments
288 # embedded in the original text defining the 'typeinfo' element
289 rawXml = etree.tostring(typeinfo.elem).decode('ascii')
290 result = re.search(r'VK_STRUCTURE_TYPE_\w+', rawXml)
291 if result:
292 value = result.group(0)
293 else:
294 value = self.genVkStructureType(typeName)
295 # Store the required type value
296 self.structTypes[typeName] = self.StructType(name=name, value=value)
297 # Store pointer/array/string info
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700298 isstaticarray = self.paramIsStaticArray(member)
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700299 membersInfo.append(self.CommandParam(type=type,
300 name=name,
301 ispointer=self.paramIsPointer(member),
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700302 isstaticarray=isstaticarray,
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700303 isconst=True if 'const' in cdecl else False,
304 iscount=True if name in lens else False,
305 len=self.getLen(member),
306 extstructs=member.attrib.get('validextensionstructs') if name == 'pNext' else None,
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700307 cdecl=cdecl))
Mark Lobodzinski5380d132016-12-28 14:45:34 -0700308 self.structMembers.append(self.StructMemberData(name=typeName, members=membersInfo, ifdef_protect=self.featureExtraProtect))
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700309 #
310 # Enum_string_header: Create a routine to convert an enumerated value into a string
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700311 def GenerateEnumStringConversion(self, groupName, value_list):
312 outstring = '\n'
313 outstring += 'static inline const char* string_%s(%s input_value)\n' % (groupName, groupName)
314 outstring += '{\n'
315 outstring += ' switch ((%s)input_value)\n' % groupName
316 outstring += ' {\n'
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700317 for item in value_list:
318 outstring += ' case %s:\n' % item
319 outstring += ' return "%s";\n' % item
320 outstring += ' default:\n'
321 outstring += ' return "Unhandled %s";\n' % groupName
322 outstring += ' }\n'
323 outstring += '}\n'
324 return outstring
325 #
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700326 # Combine enum string helper header file preamble with body text and return
327 def GenerateEnumStringHelperHeader(self):
328 enum_string_helper_header = '\n'
329 enum_string_helper_header += '#pragma once\n'
330 enum_string_helper_header += '#ifdef _WIN32\n'
331 enum_string_helper_header += '#pragma warning( disable : 4065 )\n'
332 enum_string_helper_header += '#endif\n'
333 enum_string_helper_header += '\n'
334 enum_string_helper_header += '#include <vulkan/vulkan.h>\n'
335 enum_string_helper_header += '\n'
336 enum_string_helper_header += self.enum_output
337 return enum_string_helper_header
338 #
Mark Lobodzinski5380d132016-12-28 14:45:34 -0700339 # struct_size_header: build function prototypes for header file
340 def GenerateStructSizeHeader(self):
341 outstring = ''
342 outstring += 'size_t get_struct_chain_size(const void* struct_ptr);\n'
343 for item in self.structMembers:
344 lower_case_name = item.name.lower()
345 if item.ifdef_protect != None:
346 outstring += '#ifdef %s\n' % item.ifdef_protect
347 outstring += 'size_t vk_size_%s(const %s* struct_ptr);\n' % (item.name.lower(), item.name)
348 if item.ifdef_protect != None:
349 outstring += '#endif // %s\n' % item.ifdef_protect
350 outstring += '#ifdef __cplusplus\n'
351 outstring += '}\n'
352 outstring += '#endif'
353 return outstring
354 #
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700355 # Combine struct size helper header file preamble with body text and return
356 def GenerateStructSizeHelperHeader(self):
357 struct_size_helper_header = '\n'
358 struct_size_helper_header += '#ifdef __cplusplus\n'
359 struct_size_helper_header += 'extern "C" {\n'
360 struct_size_helper_header += '#endif\n'
361 struct_size_helper_header += '\n'
362 struct_size_helper_header += '#include <stdio.h>\n'
363 struct_size_helper_header += '#include <stdlib.h>\n'
364 struct_size_helper_header += '#include <vulkan/vulkan.h>\n'
365 struct_size_helper_header += '\n'
366 struct_size_helper_header += '// Function Prototypes\n'
367 struct_size_helper_header += self.GenerateStructSizeHeader()
368 return struct_size_helper_header
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700369 #
370 # Helper function for declaring a counter variable only once
371 def DeclareCounter(self, string_var, declare_flag):
372 if declare_flag == False:
373 string_var += ' uint32_t i = 0;\n'
374 declare_flag = True
375 return string_var, declare_flag
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700376 #
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700377 # Build the header of the get_struct_chain_size function
378 def GenerateChainSizePreamble(self):
379 preable = '\nsize_t get_struct_chain_size(const void* struct_ptr) {\n'
380 preable += ' // Use VkApplicationInfo as struct until actual type is resolved\n'
381 preable += ' VkApplicationInfo* pNext = (VkApplicationInfo*)struct_ptr;\n'
382 preable += ' size_t struct_size = 0;\n'
383 preable += ' while (pNext) {\n'
384 preable += ' switch (pNext->sType) {\n'
385 return preable
386 #
387 # Build the footer of the get_struct_chain_size function
388 def GenerateChainSizePostamble(self):
389 postamble = ' default:\n'
390 postamble += ' assert(0);\n'
391 postamble += ' struct_size += 0;\n'
392 postamble += ' }\n'
393 postamble += ' pNext = (VkApplicationInfo*)pNext->pNext;\n'
394 postamble += ' }\n'
395 postamble += ' return struct_size;\n'
396 postamble += '}'
397 return postamble
398 #
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700399 # struct_size_helper source -- create bodies of struct size helper functions
400 def GenerateStructSizeSource(self):
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700401 # Construct the body of the routine and get_struct_chain_size() simultaneously
402 struct_size_body = ''
403 chain_size = self.GenerateChainSizePreamble()
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700404 for item in self.structMembers:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700405 struct_size_body += '\n'
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700406 lower_case_name = item.name.lower()
407 if item.ifdef_protect != None:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700408 struct_size_body += '#ifdef %s\n' % item.ifdef_protect
409 chain_size += '#ifdef %s\n' % item.ifdef_protect
410 if item.name in self.structTypes:
411 chain_size += ' case %s: {\n' % self.structTypes[item.name].value
412 chain_size += ' struct_size += vk_size_%s((%s*)pNext);\n' % (item.name.lower(), item.name)
413 chain_size += ' break;\n'
414 chain_size += ' }\n'
415 struct_size_body += 'size_t vk_size_%s(const %s* struct_ptr) {\n' % (item.name.lower(), item.name)
416 struct_size_body += ' size_t struct_size = 0;\n'
417 struct_size_body += ' if (struct_ptr) {\n'
418 struct_size_body += ' struct_size = sizeof(%s);\n' % item.name
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700419 counter_declared = False
420 for member in item.members:
421 vulkan_type = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None)
422 if member.ispointer == True:
423 if vulkan_type is not None:
424 # If this is another Vulkan structure call generated size function
425 if member.len is not None:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700426 struct_size_body, counter_declared = self.DeclareCounter(struct_size_body, counter_declared)
427 struct_size_body += ' for (i = 0; i < struct_ptr->%s; i++) {\n' % member.len
428 struct_size_body += ' struct_size += vk_size_%s(&struct_ptr->%s[i]);\n' % (member.type.lower(), member.name)
429 struct_size_body += ' }\n'
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700430 else:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700431 struct_size_body += ' struct_size += vk_size_%s(struct_ptr->%s);\n' % (member.type.lower(), member.name)
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700432 else:
433 if member.type == 'char':
434 # Deal with sizes of character strings
435 if member.len is not None:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700436 struct_size_body, counter_declared = self.DeclareCounter(struct_size_body, counter_declared)
437 struct_size_body += ' for (i = 0; i < struct_ptr->%s; i++) {\n' % member.len
438 struct_size_body += ' struct_size += (sizeof(char*) + (sizeof(char) * (1 + strlen(struct_ptr->%s[i]))));\n' % (member.name)
439 struct_size_body += ' }\n'
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700440 else:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700441 struct_size_body += ' 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 -0700442 else:
443 if member.len is not None:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700444 # Avoid using 'sizeof(void)', which generates compile-time warnings/errors
445 checked_type = member.type
446 if checked_type == 'void':
447 checked_type = 'void*'
mizhenc27f6c72017-03-31 09:08:16 -0600448 struct_size_body += ' struct_size += (struct_ptr->%s ) * sizeof(%s);\n' % (member.len, checked_type)
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700449 struct_size_body += ' }\n'
450 struct_size_body += ' return struct_size;\n'
451 struct_size_body += '}\n'
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700452 if item.ifdef_protect != None:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700453 struct_size_body += '#endif // %s\n' % item.ifdef_protect
454 chain_size += '#endif // %s\n' % item.ifdef_protect
455 chain_size += self.GenerateChainSizePostamble()
456 struct_size_body += chain_size;
457 return struct_size_body
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700458 #
459 # Combine struct size helper source file preamble with body text and return
460 def GenerateStructSizeHelperSource(self):
461 struct_size_helper_source = '\n'
462 struct_size_helper_source += '#include "vk_struct_size_helper.h"\n'
463 struct_size_helper_source += '#include <string.h>\n'
464 struct_size_helper_source += '#include <assert.h>\n'
465 struct_size_helper_source += '\n'
466 struct_size_helper_source += '// Function Definitions\n'
467 struct_size_helper_source += self.GenerateStructSizeSource()
468 return struct_size_helper_source
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700469 #
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700470 # Combine safe struct helper header file preamble with body text and return
471 def GenerateSafeStructHelperHeader(self):
472 safe_struct_helper_header = '\n'
473 safe_struct_helper_header += '#pragma once\n'
474 safe_struct_helper_header += '#include <vulkan/vulkan.h>\n'
475 safe_struct_helper_header += '\n'
476 safe_struct_helper_header += self.GenerateSafeStructHeader()
477 return safe_struct_helper_header
478 #
479 # safe_struct header: build function prototypes for header file
480 def GenerateSafeStructHeader(self):
481 safe_struct_header = ''
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700482 for item in self.structMembers:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700483 if self.NeedSafeStruct(item) == True:
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700484 safe_struct_header += '\n'
485 if item.ifdef_protect != None:
486 safe_struct_header += '#ifdef %s\n' % item.ifdef_protect
487 safe_struct_header += 'struct safe_%s {\n' % (item.name)
488 for member in item.members:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700489 if member.type in self.structNames:
490 member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None)
491 if member_index is not None and self.NeedSafeStruct(self.structMembers[member_index]) == True:
492 if member.ispointer:
493 safe_struct_header += ' safe_%s* %s;\n' % (member.type, member.name)
494 else:
495 safe_struct_header += ' safe_%s %s;\n' % (member.type, member.name)
496 continue
497 if member.len is not None and (self.TypeContainsObjectHandle(member.type, True) or self.TypeContainsObjectHandle(member.type, False)):
498 safe_struct_header += ' %s* %s;\n' % (member.type, member.name)
499 else:
500 safe_struct_header += '%s;\n' % member.cdecl
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700501 safe_struct_header += ' safe_%s(const %s* in_struct);\n' % (item.name, item.name)
502 safe_struct_header += ' safe_%s(const safe_%s& src);\n' % (item.name, item.name)
503 safe_struct_header += ' safe_%s();\n' % item.name
504 safe_struct_header += ' ~safe_%s();\n' % item.name
505 safe_struct_header += ' void initialize(const %s* in_struct);\n' % item.name
506 safe_struct_header += ' void initialize(const safe_%s* src);\n' % item.name
507 safe_struct_header += ' %s *ptr() { return reinterpret_cast<%s *>(this); }\n' % (item.name, item.name)
508 safe_struct_header += ' %s const *ptr() const { return reinterpret_cast<%s const *>(this); }\n' % (item.name, item.name)
509 safe_struct_header += '};\n'
510 if item.ifdef_protect != None:
511 safe_struct_header += '#endif // %s\n' % item.ifdef_protect
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700512 return safe_struct_header
513 #
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700514 # Determine if a structure needs a safe_struct helper function
515 # That is, it has an sType or one of its members is a pointer
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700516 def NeedSafeStruct(self, structure):
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700517 if 'sType' == structure.name:
518 return True
519 for member in structure.members:
520 if member.ispointer == True:
521 return True
522 return False
523 #
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700524 # Combine safe struct helper source file preamble with body text and return
525 def GenerateSafeStructHelperSource(self):
526 safe_struct_helper_source = '\n'
527 safe_struct_helper_source += '#include "vk_safe_struct.h"\n'
528 safe_struct_helper_source += '#include <string.h>\n'
529 safe_struct_helper_source += '\n'
530 safe_struct_helper_source += self.GenerateSafeStructSource()
531 return safe_struct_helper_source
532 #
533 # safe_struct source -- create bodies of safe struct helper functions
534 def GenerateSafeStructSource(self):
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700535 safe_struct_body = []
Mark Lobodzinski560729b2017-03-06 08:59:14 -0700536 wsi_structs = ['VkXlibSurfaceCreateInfoKHR',
537 'VkXcbSurfaceCreateInfoKHR',
538 'VkWaylandSurfaceCreateInfoKHR',
539 'VkMirSurfaceCreateInfoKHR',
540 'VkAndroidSurfaceCreateInfoKHR',
541 'VkWin32SurfaceCreateInfoKHR'
542 ]
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700543 for item in self.structMembers:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700544 if self.NeedSafeStruct(item) == False:
545 continue
Mark Lobodzinski560729b2017-03-06 08:59:14 -0700546 if item.name in wsi_structs:
547 continue
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700548 if item.ifdef_protect != None:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700549 safe_struct_body.append("#ifdef %s\n" % item.ifdef_protect)
550 ss_name = "safe_%s" % item.name
551 init_list = '' # list of members in struct constructor initializer
552 default_init_list = '' # Default constructor just inits ptrs to nullptr in initializer
553 init_func_txt = '' # Txt for initialize() function that takes struct ptr and inits members
554 construct_txt = '' # Body of constuctor as well as body of initialize() func following init_func_txt
555 destruct_txt = ''
556 # VkWriteDescriptorSet is special case because pointers may be non-null but ignored
557 custom_construct_txt = {'VkWriteDescriptorSet' :
558 ' switch (descriptorType) {\n'
559 ' case VK_DESCRIPTOR_TYPE_SAMPLER:\n'
560 ' case VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER:\n'
561 ' case VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE:\n'
562 ' case VK_DESCRIPTOR_TYPE_STORAGE_IMAGE:\n'
563 ' case VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT:\n'
564 ' if (descriptorCount && in_struct->pImageInfo) {\n'
565 ' pImageInfo = new VkDescriptorImageInfo[descriptorCount];\n'
566 ' for (uint32_t i=0; i<descriptorCount; ++i) {\n'
567 ' pImageInfo[i] = in_struct->pImageInfo[i];\n'
568 ' }\n'
569 ' }\n'
570 ' break;\n'
571 ' case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER:\n'
572 ' case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER:\n'
573 ' case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC:\n'
574 ' case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC:\n'
575 ' if (descriptorCount && in_struct->pBufferInfo) {\n'
576 ' pBufferInfo = new VkDescriptorBufferInfo[descriptorCount];\n'
577 ' for (uint32_t i=0; i<descriptorCount; ++i) {\n'
578 ' pBufferInfo[i] = in_struct->pBufferInfo[i];\n'
579 ' }\n'
580 ' }\n'
581 ' break;\n'
582 ' case VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER:\n'
583 ' case VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER:\n'
584 ' if (descriptorCount && in_struct->pTexelBufferView) {\n'
585 ' pTexelBufferView = new VkBufferView[descriptorCount];\n'
586 ' for (uint32_t i=0; i<descriptorCount; ++i) {\n'
587 ' pTexelBufferView[i] = in_struct->pTexelBufferView[i];\n'
588 ' }\n'
589 ' }\n'
590 ' break;\n'
591 ' default:\n'
592 ' break;\n'
593 ' }\n'}
594
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700595 for member in item.members:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700596 m_type = member.type
597 if member.type in self.structNames:
598 member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None)
599 if member_index is not None and self.NeedSafeStruct(self.structMembers[member_index]) == True:
600 m_type = 'safe_%s' % member.type
601 if member.ispointer and 'safe_' not in m_type and self.TypeContainsObjectHandle(member.type, False) == False:
602 # 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 -0700603 if m_type in ['void', 'char']:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700604 # For these exceptions just copy initial value over for now
605 init_list += '\n %s(in_struct->%s),' % (member.name, member.name)
606 init_func_txt += ' %s = in_struct->%s;\n' % (member.name, member.name)
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700607 else:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700608 default_init_list += '\n %s(nullptr),' % (member.name)
609 init_list += '\n %s(nullptr),' % (member.name)
610 init_func_txt += ' %s = nullptr;\n' % (member.name)
611 if 'pNext' != member.name and 'void' not in m_type:
Mark Lobodzinski51160a12017-01-18 11:05:48 -0700612 if not member.isstaticarray and (member.len is None or '/' in member.len):
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700613 construct_txt += ' if (in_struct->%s) {\n' % member.name
614 construct_txt += ' %s = new %s(*in_struct->%s);\n' % (member.name, m_type, member.name)
615 construct_txt += ' }\n'
616 destruct_txt += ' if (%s)\n' % member.name
617 destruct_txt += ' delete %s;\n' % member.name
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700618 else:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700619 construct_txt += ' if (in_struct->%s) {\n' % member.name
620 construct_txt += ' %s = new %s[in_struct->%s];\n' % (member.name, m_type, member.len)
621 construct_txt += ' memcpy ((void *)%s, (void *)in_struct->%s, sizeof(%s)*in_struct->%s);\n' % (member.name, member.name, m_type, member.len)
622 construct_txt += ' }\n'
623 destruct_txt += ' if (%s)\n' % member.name
624 destruct_txt += ' delete[] %s;\n' % member.name
625 elif member.isstaticarray or member.len is not None:
626 if member.len is None:
627 # Extract length of static array by grabbing val between []
628 static_array_size = re.match(r"[^[]*\[([^]]*)\]", member.cdecl)
629 construct_txt += ' for (uint32_t i=0; i<%s; ++i) {\n' % static_array_size.group(1)
630 construct_txt += ' %s[i] = in_struct->%s[i];\n' % (member.name, member.name)
631 construct_txt += ' }\n'
632 else:
633 # Init array ptr to NULL
634 default_init_list += '\n %s(nullptr),' % member.name
635 init_list += '\n %s(nullptr),' % member.name
636 init_func_txt += ' %s = nullptr;\n' % member.name
637 array_element = 'in_struct->%s[i]' % member.name
638 if member.type in self.structNames:
639 member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None)
640 if member_index is not None and self.NeedSafeStruct(self.structMembers[member_index]) == True:
641 array_element = '%s(&in_struct->safe_%s[i])' % (member.type, member.name)
642 construct_txt += ' if (%s && in_struct->%s) {\n' % (member.len, member.name)
643 construct_txt += ' %s = new %s[%s];\n' % (member.name, m_type, member.len)
644 destruct_txt += ' if (%s)\n' % member.name
645 destruct_txt += ' delete[] %s;\n' % member.name
646 construct_txt += ' for (uint32_t i=0; i<%s; ++i) {\n' % (member.len)
647 if 'safe_' in m_type:
648 construct_txt += ' %s[i].initialize(&in_struct->%s[i]);\n' % (member.name, member.name)
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700649 else:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700650 construct_txt += ' %s[i] = %s;\n' % (member.name, array_element)
651 construct_txt += ' }\n'
652 construct_txt += ' }\n'
653 elif member.ispointer == True:
654 construct_txt += ' if (in_struct->%s)\n' % member.name
655 construct_txt += ' %s = new %s(in_struct->%s);\n' % (member.name, m_type, member.name)
656 construct_txt += ' else\n'
657 construct_txt += ' %s = NULL;\n' % member.name
658 destruct_txt += ' if (%s)\n' % member.name
659 destruct_txt += ' delete %s;\n' % member.name
660 elif 'safe_' in m_type:
661 init_list += '\n %s(&in_struct->%s),' % (member.name, member.name)
662 init_func_txt += ' %s.initialize(&in_struct->%s);\n' % (member.name, member.name)
663 else:
664 init_list += '\n %s(in_struct->%s),' % (member.name, member.name)
665 init_func_txt += ' %s = in_struct->%s;\n' % (member.name, member.name)
666 if '' != init_list:
667 init_list = init_list[:-1] # hack off final comma
668 if item.name in custom_construct_txt:
669 construct_txt = custom_construct_txt[item.name]
670 safe_struct_body.append("\n%s::%s(const %s* in_struct) :%s\n{\n%s}" % (ss_name, ss_name, item.name, init_list, construct_txt))
671 if '' != default_init_list:
672 default_init_list = " :%s" % (default_init_list[:-1])
673 safe_struct_body.append("\n%s::%s()%s\n{}" % (ss_name, ss_name, default_init_list))
674 # Create slight variation of init and construct txt for copy constructor that takes a src object reference vs. struct ptr
675 copy_construct_init = init_func_txt.replace('in_struct->', 'src.')
676 copy_construct_txt = construct_txt.replace(' (in_struct->', ' (src.') # Exclude 'if' blocks from next line
677 copy_construct_txt = copy_construct_txt.replace('(in_struct->', '(*src.') # Pass object to copy constructors
678 copy_construct_txt = copy_construct_txt.replace('in_struct->', 'src.') # Modify remaining struct refs for src object
679 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
680 safe_struct_body.append("\n%s::~%s()\n{\n%s}" % (ss_name, ss_name, destruct_txt))
681 safe_struct_body.append("\nvoid %s::initialize(const %s* in_struct)\n{\n%s%s}" % (ss_name, item.name, init_func_txt, construct_txt))
682 # Copy initializer uses same txt as copy constructor but has a ptr and not a reference
683 init_copy = copy_construct_init.replace('src.', 'src->')
684 init_construct = copy_construct_txt.replace('src.', 'src->')
685 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 -0700686 if item.ifdef_protect != None:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700687 safe_struct_body.append("#endif // %s\n" % item.ifdef_protect)
688 return "\n".join(safe_struct_body)
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700689 #
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700690 # Create a helper file and return it as a string
691 def OutputDestFile(self):
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700692 if self.helper_file_type == 'enum_string_header':
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700693 return self.GenerateEnumStringHelperHeader()
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700694 elif self.helper_file_type == 'struct_size_header':
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700695 return self.GenerateStructSizeHelperHeader()
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700696 elif self.helper_file_type == 'struct_size_source':
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700697 return self.GenerateStructSizeHelperSource()
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700698 elif self.helper_file_type == 'safe_struct_header':
699 return self.GenerateSafeStructHelperHeader()
700 elif self.helper_file_type == 'safe_struct_source':
701 return self.GenerateSafeStructHelperSource()
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700702 else:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700703 return 'Bad Helper File Generator Option %s' % self.helper_file_type
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700704