blob: 1c958a68839af5c1d00f74438a8c8e86f947da9f [file] [log] [blame]
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -07001#!/usr/bin/python3 -i
2#
3# Copyright (c) 2015-2016 The Khronos Group Inc.
4# Copyright (c) 2015-2016 Valve Corporation
5# Copyright (c) 2015-2016 LunarG, Inc.
6# Copyright (c) 2015-2016 Google Inc.
7#
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>
21
22import os,re,sys
23import xml.etree.ElementTree as etree
24from generator import *
25from collections import namedtuple
26
27#
28# HelperFileOutputGeneratorOptions - subclass of GeneratorOptions.
29class HelperFileOutputGeneratorOptions(GeneratorOptions):
30 def __init__(self,
31 filename = None,
32 directory = '.',
33 apiname = None,
34 profile = None,
35 versions = '.*',
36 emitversions = '.*',
37 defaultExtensions = None,
38 addExtensions = None,
39 removeExtensions = None,
40 sortProcedure = regSortFeatures,
41 prefixText = "",
42 genFuncPointers = True,
43 protectFile = True,
44 protectFeature = True,
45 protectProto = None,
46 protectProtoStr = None,
47 apicall = '',
48 apientry = '',
49 apientryp = '',
50 alignFuncParam = 0,
51 library_name = '',
52 helper_file_type = ''):
53 GeneratorOptions.__init__(self, filename, directory, apiname, profile,
54 versions, emitversions, defaultExtensions,
55 addExtensions, removeExtensions, sortProcedure)
56 self.prefixText = prefixText
57 self.genFuncPointers = genFuncPointers
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070058 self.protectFile = protectFile
59 self.protectFeature = protectFeature
60 self.protectProto = protectProto
61 self.protectProtoStr = protectProtoStr
62 self.apicall = apicall
63 self.apientry = apientry
64 self.apientryp = apientryp
65 self.alignFuncParam = alignFuncParam
66 self.library_name = library_name
67 self.helper_file_type = helper_file_type
68#
Mark Lobodzinski46d388f2016-12-28 10:46:26 -070069# HelperFileOutputGenerator - subclass of OutputGenerator. Outputs Vulkan helper files
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070070class HelperFileOutputGenerator(OutputGenerator):
Mark Lobodzinskia3cc3612017-01-03 13:25:10 -070071 """Generate helper file based on XML element attributes"""
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070072 def __init__(self,
73 errFile = sys.stderr,
74 warnFile = sys.stderr,
75 diagFile = sys.stdout):
76 OutputGenerator.__init__(self, errFile, warnFile, diagFile)
77 # Internal state - accumulators for different inner block text
Mark Lobodzinski5380d132016-12-28 14:45:34 -070078 self.enum_output = '' # string built up of enum string routines
79 self.struct_size_h_output = '' # string built up of struct size header output
80 self.struct_size_c_output = '' # string built up of struct size source output
Mark Lobodzinski46d388f2016-12-28 10:46:26 -070081 # Internal state - accumulators for different inner block text
Mark Lobodzinski46d388f2016-12-28 10:46:26 -070082 self.structNames = [] # List of Vulkan struct typenames
83 self.structTypes = dict() # Map of Vulkan struct typename to required VkStructureType
84 self.handleTypes = set() # Set of handle type names
85 self.commands = [] # List of CommandData records for all Vulkan commands
86 self.structMembers = [] # List of StructMemberData records for all Vulkan structs
87 self.flags = set() # Map of flags typenames
88 # Named tuples to store struct and command data
89 self.StructType = namedtuple('StructType', ['name', 'value'])
90 self.CommandParam = namedtuple('CommandParam', ['type', 'name', 'ispointer', 'isconst', 'iscount', 'len', 'extstructs', 'cdecl', 'islocal', 'iscreate', 'isdestroy'])
91 self.CommandData = namedtuple('CommandData', ['name', 'return_type', 'params', 'cdecl'])
Mark Lobodzinski5380d132016-12-28 14:45:34 -070092 self.StructMemberData = namedtuple('StructMemberData', ['name', 'members', 'ifdef_protect'])
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070093 #
94 # Called once at the beginning of each run
95 def beginFile(self, genOpts):
96 OutputGenerator.beginFile(self, genOpts)
97 # User-supplied prefix text, if any (list of strings)
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070098 self.helper_file_type = genOpts.helper_file_type
99 self.library_name = genOpts.library_name
100 # File Comment
101 file_comment = '// *** THIS FILE IS GENERATED - DO NOT EDIT ***\n'
102 file_comment += '// See helper_file_generator.py for modifications\n'
103 write(file_comment, file=self.outFile)
104 # Copyright Notice
105 copyright = ''
106 copyright += '\n'
107 copyright += '/***************************************************************************\n'
108 copyright += ' *\n'
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700109 copyright += ' * Copyright (c) 2015-2017 The Khronos Group Inc.\n'
110 copyright += ' * Copyright (c) 2015-2017 Valve Corporation\n'
111 copyright += ' * Copyright (c) 2015-2017 LunarG, Inc.\n'
112 copyright += ' * Copyright (c) 2015-2017 Google Inc.\n'
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700113 copyright += ' *\n'
114 copyright += ' * Licensed under the Apache License, Version 2.0 (the "License");\n'
115 copyright += ' * you may not use this file except in compliance with the License.\n'
116 copyright += ' * You may obtain a copy of the License at\n'
117 copyright += ' *\n'
118 copyright += ' * http://www.apache.org/licenses/LICENSE-2.0\n'
119 copyright += ' *\n'
120 copyright += ' * Unless required by applicable law or agreed to in writing, software\n'
121 copyright += ' * distributed under the License is distributed on an "AS IS" BASIS,\n'
122 copyright += ' * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n'
123 copyright += ' * See the License for the specific language governing permissions and\n'
124 copyright += ' * limitations under the License.\n'
125 copyright += ' *\n'
126 copyright += ' * Author: Mark Lobodzinski <mark@lunarg.com>\n'
Mark Lobodzinskia9c963d2016-12-28 07:45:35 -0700127 copyright += ' * Author: Courtney Goeltzenleuchter <courtneygo@google.com>\n'
128 copyright += ' * Author: Tobin Ehlis <tobine@google.com>\n'
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700129 copyright += ' *\n'
130 copyright += ' ****************************************************************************/\n'
131 write(copyright, file=self.outFile)
132 #
Mark Lobodzinskia3cc3612017-01-03 13:25:10 -0700133 # Write generated file content to output file
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700134 def endFile(self):
135 dest_file = ''
136 dest_file += self.OutputDestFile()
Mark Lobodzinskiafe10542017-01-03 13:22:44 -0700137 # Remove blank lines at EOF
138 if dest_file.endswith('\n'):
139 dest_file = dest_file[:-1]
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700140 write(dest_file, file=self.outFile);
141 # Finish processing in superclass
142 OutputGenerator.endFile(self)
143 #
144 # Grab group (e.g. C "enum" type) info to output for enum-string conversion helper
145 def genGroup(self, groupinfo, groupName):
146 OutputGenerator.genGroup(self, groupinfo, groupName)
147 groupElem = groupinfo.elem
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700148 # For enum_string_header
149 if self.helper_file_type == 'enum_string_header':
Mark Lobodzinski824ce642016-12-28 08:49:46 -0700150 value_list = []
151 for elem in groupElem.findall('enum'):
152 if elem.get('supported') != 'disabled':
153 item_name = elem.get('name')
154 value_list.append(item_name)
155 if value_list is not None:
156 self.enum_output += self.GenerateEnumStringConversion(groupName, value_list)
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700157 #
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700158 # Called for each type -- if the type is a struct/union, grab the metadata
159 def genType(self, typeinfo, name):
160 OutputGenerator.genType(self, typeinfo, name)
161 typeElem = typeinfo.elem
162 # If the type is a struct type, traverse the imbedded <member> tags generating a structure.
163 # Otherwise, emit the tag text.
164 category = typeElem.get('category')
165 if (category == 'struct' or category == 'union'):
166 self.structNames.append(name)
167 self.genStruct(typeinfo, name)
168 #
169 # Generate a VkStructureType based on a structure typename
170 def genVkStructureType(self, typename):
171 # Add underscore between lowercase then uppercase
172 value = re.sub('([a-z0-9])([A-Z])', r'\1_\2', typename)
173 # Change to uppercase
174 value = value.upper()
175 # Add STRUCTURE_TYPE_
176 return re.sub('VK_', 'VK_STRUCTURE_TYPE_', value)
177 #
178 # Check if the parameter passed in is a pointer
179 def paramIsPointer(self, param):
180 ispointer = False
181 for elem in param:
182 if ((elem.tag is not 'type') and (elem.tail is not None)) and '*' in elem.tail:
183 ispointer = True
184 return ispointer
185 #
186 # Retrieve the type and name for a parameter
187 def getTypeNameTuple(self, param):
188 type = ''
189 name = ''
190 for elem in param:
191 if elem.tag == 'type':
192 type = noneStr(elem.text)
193 elif elem.tag == 'name':
194 name = noneStr(elem.text)
195 return (type, name)
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700196 # Extract length values from latexmath. Currently an inflexible solution that looks for specific
197 # patterns that are found in vk.xml. Will need to be updated when new patterns are introduced.
198 def parseLateXMath(self, source):
199 name = 'ERROR'
200 decoratedName = 'ERROR'
201 if 'mathit' in source:
202 # Matches expressions similar to 'latexmath:[$\lceil{\mathit{rasterizationSamples} \over 32}\rceil$]'
203 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)
204 if not match or match.group(1) != match.group(4):
205 raise 'Unrecognized latexmath expression'
206 name = match.group(2)
207 decoratedName = '{}/{}'.format(*match.group(2, 3))
208 else:
209 # Matches expressions similar to 'latexmath : [$dataSize \over 4$]'
210 match = re.match(r'latexmath\s*\:\s*\[\s*\$\s*(\w+)\s*\\over\s*(\d+)\s*\$\s*\]', source)
211 name = match.group(1)
212 decoratedName = '{}/{}'.format(*match.group(1, 2))
213 return name, decoratedName
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700214 #
215 # Retrieve the value of the len tag
216 def getLen(self, param):
217 result = None
218 len = param.attrib.get('len')
219 if len and len != 'null-terminated':
220 # For string arrays, 'len' can look like 'count,null-terminated', indicating that we
221 # have a null terminated array of strings. We strip the null-terminated from the
222 # 'len' field and only return the parameter specifying the string count
223 if 'null-terminated' in len:
224 result = len.split(',')[0]
225 else:
226 result = len
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700227 if 'latexmath' in len:
228 param_type, param_name = self.getTypeNameTuple(param)
229 len_name, result = self.parseLateXMath(len)
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700230 # Spec has now notation for len attributes, using :: instead of platform specific pointer symbol
231 result = str(result).replace('::', '->')
232 return result
233 #
234 # Generate local ready-access data describing Vulkan structures and unions from the XML metadata
235 def genStruct(self, typeinfo, typeName):
236 OutputGenerator.genStruct(self, typeinfo, typeName)
237 members = typeinfo.elem.findall('.//member')
238 # Iterate over members once to get length parameters for arrays
239 lens = set()
240 for member in members:
241 len = self.getLen(member)
242 if len:
243 lens.add(len)
244 # Generate member info
245 membersInfo = []
246 for member in members:
247 # Get the member's type and name
248 info = self.getTypeNameTuple(member)
249 type = info[0]
250 name = info[1]
251 cdecl = self.makeCParamDecl(member, 0)
252 # Process VkStructureType
253 if type == 'VkStructureType':
254 # Extract the required struct type value from the comments
255 # embedded in the original text defining the 'typeinfo' element
256 rawXml = etree.tostring(typeinfo.elem).decode('ascii')
257 result = re.search(r'VK_STRUCTURE_TYPE_\w+', rawXml)
258 if result:
259 value = result.group(0)
260 else:
261 value = self.genVkStructureType(typeName)
262 # Store the required type value
263 self.structTypes[typeName] = self.StructType(name=name, value=value)
264 # Store pointer/array/string info
265 membersInfo.append(self.CommandParam(type=type,
266 name=name,
267 ispointer=self.paramIsPointer(member),
268 isconst=True if 'const' in cdecl else False,
269 iscount=True if name in lens else False,
270 len=self.getLen(member),
271 extstructs=member.attrib.get('validextensionstructs') if name == 'pNext' else None,
272 cdecl=cdecl,
273 islocal=False,
274 iscreate=False,
275 isdestroy=False))
Mark Lobodzinski5380d132016-12-28 14:45:34 -0700276 self.structMembers.append(self.StructMemberData(name=typeName, members=membersInfo, ifdef_protect=self.featureExtraProtect))
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700277 #
278 # Enum_string_header: Create a routine to convert an enumerated value into a string
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700279 def GenerateEnumStringConversion(self, groupName, value_list):
280 outstring = '\n'
281 outstring += 'static inline const char* string_%s(%s input_value)\n' % (groupName, groupName)
282 outstring += '{\n'
283 outstring += ' switch ((%s)input_value)\n' % groupName
284 outstring += ' {\n'
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700285 for item in value_list:
286 outstring += ' case %s:\n' % item
287 outstring += ' return "%s";\n' % item
288 outstring += ' default:\n'
289 outstring += ' return "Unhandled %s";\n' % groupName
290 outstring += ' }\n'
291 outstring += '}\n'
292 return outstring
293 #
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700294 # Combine enum string helper header file preamble with body text and return
295 def GenerateEnumStringHelperHeader(self):
296 enum_string_helper_header = '\n'
297 enum_string_helper_header += '#pragma once\n'
298 enum_string_helper_header += '#ifdef _WIN32\n'
299 enum_string_helper_header += '#pragma warning( disable : 4065 )\n'
300 enum_string_helper_header += '#endif\n'
301 enum_string_helper_header += '\n'
302 enum_string_helper_header += '#include <vulkan/vulkan.h>\n'
303 enum_string_helper_header += '\n'
304 enum_string_helper_header += self.enum_output
305 return enum_string_helper_header
306 #
Mark Lobodzinski5380d132016-12-28 14:45:34 -0700307 # struct_size_header: build function prototypes for header file
308 def GenerateStructSizeHeader(self):
309 outstring = ''
310 outstring += 'size_t get_struct_chain_size(const void* struct_ptr);\n'
311 for item in self.structMembers:
312 lower_case_name = item.name.lower()
313 if item.ifdef_protect != None:
314 outstring += '#ifdef %s\n' % item.ifdef_protect
315 outstring += 'size_t vk_size_%s(const %s* struct_ptr);\n' % (item.name.lower(), item.name)
316 if item.ifdef_protect != None:
317 outstring += '#endif // %s\n' % item.ifdef_protect
318 outstring += '#ifdef __cplusplus\n'
319 outstring += '}\n'
320 outstring += '#endif'
321 return outstring
322 #
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700323 # Combine struct size helper header file preamble with body text and return
324 def GenerateStructSizeHelperHeader(self):
325 struct_size_helper_header = '\n'
326 struct_size_helper_header += '#ifdef __cplusplus\n'
327 struct_size_helper_header += 'extern "C" {\n'
328 struct_size_helper_header += '#endif\n'
329 struct_size_helper_header += '\n'
330 struct_size_helper_header += '#include <stdio.h>\n'
331 struct_size_helper_header += '#include <stdlib.h>\n'
332 struct_size_helper_header += '#include <vulkan/vulkan.h>\n'
333 struct_size_helper_header += '\n'
334 struct_size_helper_header += '// Function Prototypes\n'
335 struct_size_helper_header += self.GenerateStructSizeHeader()
336 return struct_size_helper_header
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700337 #
338 # Helper function for declaring a counter variable only once
339 def DeclareCounter(self, string_var, declare_flag):
340 if declare_flag == False:
341 string_var += ' uint32_t i = 0;\n'
342 declare_flag = True
343 return string_var, declare_flag
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700344 #
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700345 # Build the header of the get_struct_chain_size function
346 def GenerateChainSizePreamble(self):
347 preable = '\nsize_t get_struct_chain_size(const void* struct_ptr) {\n'
348 preable += ' // Use VkApplicationInfo as struct until actual type is resolved\n'
349 preable += ' VkApplicationInfo* pNext = (VkApplicationInfo*)struct_ptr;\n'
350 preable += ' size_t struct_size = 0;\n'
351 preable += ' while (pNext) {\n'
352 preable += ' switch (pNext->sType) {\n'
353 return preable
354 #
355 # Build the footer of the get_struct_chain_size function
356 def GenerateChainSizePostamble(self):
357 postamble = ' default:\n'
358 postamble += ' assert(0);\n'
359 postamble += ' struct_size += 0;\n'
360 postamble += ' }\n'
361 postamble += ' pNext = (VkApplicationInfo*)pNext->pNext;\n'
362 postamble += ' }\n'
363 postamble += ' return struct_size;\n'
364 postamble += '}'
365 return postamble
366 #
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700367 # struct_size_helper source -- create bodies of struct size helper functions
368 def GenerateStructSizeSource(self):
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700369 # Construct the body of the routine and get_struct_chain_size() simultaneously
370 struct_size_body = ''
371 chain_size = self.GenerateChainSizePreamble()
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700372 for item in self.structMembers:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700373 struct_size_body += '\n'
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700374 lower_case_name = item.name.lower()
375 if item.ifdef_protect != None:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700376 struct_size_body += '#ifdef %s\n' % item.ifdef_protect
377 chain_size += '#ifdef %s\n' % item.ifdef_protect
378 if item.name in self.structTypes:
379 chain_size += ' case %s: {\n' % self.structTypes[item.name].value
380 chain_size += ' struct_size += vk_size_%s((%s*)pNext);\n' % (item.name.lower(), item.name)
381 chain_size += ' break;\n'
382 chain_size += ' }\n'
383 struct_size_body += 'size_t vk_size_%s(const %s* struct_ptr) {\n' % (item.name.lower(), item.name)
384 struct_size_body += ' size_t struct_size = 0;\n'
385 struct_size_body += ' if (struct_ptr) {\n'
386 struct_size_body += ' struct_size = sizeof(%s);\n' % item.name
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700387 counter_declared = False
388 for member in item.members:
389 vulkan_type = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None)
390 if member.ispointer == True:
391 if vulkan_type is not None:
392 # If this is another Vulkan structure call generated size function
393 if member.len is not None:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700394 struct_size_body, counter_declared = self.DeclareCounter(struct_size_body, counter_declared)
395 struct_size_body += ' for (i = 0; i < struct_ptr->%s; i++) {\n' % member.len
396 struct_size_body += ' struct_size += vk_size_%s(&struct_ptr->%s[i]);\n' % (member.type.lower(), member.name)
397 struct_size_body += ' }\n'
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700398 else:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700399 struct_size_body += ' struct_size += vk_size_%s(struct_ptr->%s);\n' % (member.type.lower(), member.name)
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700400 else:
401 if member.type == 'char':
402 # Deal with sizes of character strings
403 if member.len is not None:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700404 struct_size_body, counter_declared = self.DeclareCounter(struct_size_body, counter_declared)
405 struct_size_body += ' for (i = 0; i < struct_ptr->%s; i++) {\n' % member.len
406 struct_size_body += ' struct_size += (sizeof(char*) + (sizeof(char) * (1 + strlen(struct_ptr->%s[i]))));\n' % (member.name)
407 struct_size_body += ' }\n'
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700408 else:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700409 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 -0700410 else:
411 if member.len is not None:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700412 # Avoid using 'sizeof(void)', which generates compile-time warnings/errors
413 checked_type = member.type
414 if checked_type == 'void':
415 checked_type = 'void*'
416 struct_size_body += ' struct_size += struct_ptr->%s * sizeof(%s);\n' % (member.len, checked_type)
417 struct_size_body += ' }\n'
418 struct_size_body += ' return struct_size;\n'
419 struct_size_body += '}\n'
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700420 if item.ifdef_protect != None:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700421 struct_size_body += '#endif // %s\n' % item.ifdef_protect
422 chain_size += '#endif // %s\n' % item.ifdef_protect
423 chain_size += self.GenerateChainSizePostamble()
424 struct_size_body += chain_size;
425 return struct_size_body
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700426 #
427 # Combine struct size helper source file preamble with body text and return
428 def GenerateStructSizeHelperSource(self):
429 struct_size_helper_source = '\n'
430 struct_size_helper_source += '#include "vk_struct_size_helper.h"\n'
431 struct_size_helper_source += '#include <string.h>\n'
432 struct_size_helper_source += '#include <assert.h>\n'
433 struct_size_helper_source += '\n'
434 struct_size_helper_source += '// Function Definitions\n'
435 struct_size_helper_source += self.GenerateStructSizeSource()
436 return struct_size_helper_source
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700437 #
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700438 # Combine safe struct helper header file preamble with body text and return
439 def GenerateSafeStructHelperHeader(self):
440 safe_struct_helper_header = '\n'
441 safe_struct_helper_header += '#pragma once\n'
442 safe_struct_helper_header += '#include <vulkan/vulkan.h>\n'
443 safe_struct_helper_header += '\n'
444 safe_struct_helper_header += self.GenerateSafeStructHeader()
445 return safe_struct_helper_header
446 #
447 # safe_struct header: build function prototypes for header file
448 def GenerateSafeStructHeader(self):
449 safe_struct_header = ''
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700450 for item in self.structMembers:
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700451 if self.GenSafeStruct(item) == True:
452 safe_struct_header += '\n'
453 if item.ifdef_protect != None:
454 safe_struct_header += '#ifdef %s\n' % item.ifdef_protect
455 safe_struct_header += 'struct safe_%s {\n' % (item.name)
456 for member in item.members:
457 safe_struct_header += '%s;\n' % member.cdecl
458 # Boilerplate
459 safe_struct_header += ' safe_%s(const %s* in_struct);\n' % (item.name, item.name)
460 safe_struct_header += ' safe_%s(const safe_%s& src);\n' % (item.name, item.name)
461 safe_struct_header += ' safe_%s();\n' % item.name
462 safe_struct_header += ' ~safe_%s();\n' % item.name
463 safe_struct_header += ' void initialize(const %s* in_struct);\n' % item.name
464 safe_struct_header += ' void initialize(const safe_%s* src);\n' % item.name
465 safe_struct_header += ' %s *ptr() { return reinterpret_cast<%s *>(this); }\n' % (item.name, item.name)
466 safe_struct_header += ' %s const *ptr() const { return reinterpret_cast<%s const *>(this); }\n' % (item.name, item.name)
467 safe_struct_header += '};\n'
468 if item.ifdef_protect != None:
469 safe_struct_header += '#endif // %s\n' % item.ifdef_protect
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700470 return safe_struct_header
471 #
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700472 # Determine if a structure needs a safe_struct helper function
473 # That is, it has an sType or one of its members is a pointer
474 def GenSafeStruct(self, structure):
475 if 'sType' == structure.name:
476 return True
477 for member in structure.members:
478 if member.ispointer == True:
479 return True
480 return False
481 #
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700482 # Combine safe struct helper source file preamble with body text and return
483 def GenerateSafeStructHelperSource(self):
484 safe_struct_helper_source = '\n'
485 safe_struct_helper_source += '#include "vk_safe_struct.h"\n'
486 safe_struct_helper_source += '#include <string.h>\n'
487 safe_struct_helper_source += '\n'
488 safe_struct_helper_source += self.GenerateSafeStructSource()
489 return safe_struct_helper_source
490 #
491 # safe_struct source -- create bodies of safe struct helper functions
492 def GenerateSafeStructSource(self):
493 safe_struct_body = ''
494 for item in self.structMembers:
495 safe_struct_body += '\n'
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700496 if item.ifdef_protect != None:
497 safe_struct_body += '#ifdef %s\n' % item.ifdef_protect
498 safe_struct_body += 'size_t vk_size_%s(const %s* struct_ptr) {\n' % (item.name.lower(), item.name)
499 safe_struct_body += ' size_t struct_size = 0;\n'
500 safe_struct_body += ' if (struct_ptr) {\n'
501 safe_struct_body += ' struct_size = sizeof(%s);\n' % item.name
502 counter_declared = False
503 for member in item.members:
504 vulkan_type = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None)
505 if member.ispointer == True:
506 if vulkan_type is not None:
507 # If this is another Vulkan structure call generated size function
508 if member.len is not None:
509 safe_struct_body, counter_declared = self.DeclareCounter(safe_struct_body, counter_declared)
510 safe_struct_body += ' for (i = 0; i < struct_ptr->%s; i++) {\n' % member.len
511 safe_struct_body += ' struct_size += vk_size_%s(&struct_ptr->%s[i]);\n' % (member.type.lower(), member.name)
512 safe_struct_body += ' }\n'
513 else:
514 safe_struct_body += ' struct_size += vk_size_%s(struct_ptr->%s);\n' % (member.type.lower(), member.name)
515 else:
516 if member.type == 'char':
517 # Deal with sizes of character strings
518 if member.len is not None:
519 safe_struct_body, counter_declared = self.DeclareCounter(safe_struct_body, counter_declared)
520 safe_struct_body += ' for (i = 0; i < struct_ptr->%s; i++) {\n' % member.len
521 safe_struct_body += ' struct_size += (sizeof(char*) + (sizeof(char) * (1 + strlen(struct_ptr->%s[i]))));\n' % (member.name)
522 safe_struct_body += ' }\n'
523 else:
524 safe_struct_body += ' struct_size += (struct_ptr->%s != NULL) ? sizeof(char)*(1+strlen(struct_ptr->%s)) : 0;\n' % (member.name, member.name)
525 else:
526 if member.len is not None:
527 safe_struct_body += ' struct_size += struct_ptr->%s * sizeof(%s);\n' % (member.len, member.name)
528 safe_struct_body += ' }\n'
529 safe_struct_body += ' return struct_size\n'
530 safe_struct_body += '}\n'
531 if item.ifdef_protect != None:
532 safe_struct_body += '#endif // %s\n' % item.ifdef_protect
533 return safe_struct_body
534 #
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700535 # Create a helper file and return it as a string
536 def OutputDestFile(self):
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700537 if self.helper_file_type == 'enum_string_header':
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700538 return self.GenerateEnumStringHelperHeader()
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700539 elif self.helper_file_type == 'struct_size_header':
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700540 return self.GenerateStructSizeHelperHeader()
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700541 elif self.helper_file_type == 'struct_size_source':
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700542 return self.GenerateStructSizeHelperSource()
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700543 elif self.helper_file_type == 'safe_struct_header':
544 return self.GenerateSafeStructHelperHeader()
545 elif self.helper_file_type == 'safe_struct_source':
546 return self.GenerateSafeStructHelperSource()
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700547 else:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700548 return 'Bad Helper Generator Option %s' % self.helper_file_type
549