blob: 530a833c0d1a3c097e8f0b7aadaccd2a4796b5c0 [file] [log] [blame]
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -07001#!/usr/bin/python3 -i
2#
Mark Lobodzinski733f7f42017-01-10 11:42:22 -07003# Copyright (c) 2015-2017 The Khronos Group Inc.
4# Copyright (c) 2015-2017 Valve Corporation
5# Copyright (c) 2015-2017 LunarG, Inc.
6# Copyright (c) 2015-2017 Google Inc.
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -07007#
8# Licensed under the Apache License, Version 2.0 (the "License");
9# you may not use this file except in compliance with the License.
10# You may obtain a copy of the License at
11#
12# http://www.apache.org/licenses/LICENSE-2.0
13#
14# Unless required by applicable law or agreed to in writing, software
15# distributed under the License is distributed on an "AS IS" BASIS,
16# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17# See the License for the specific language governing permissions and
18# limitations under the License.
19#
20# Author: Mark Lobodzinski <mark@lunarg.com>
Mark Lobodzinskic67efd02017-01-04 09:16:00 -070021# Author: Tobin Ehlis <tobine@google.com>
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070022
23import os,re,sys
24import xml.etree.ElementTree as etree
25from generator import *
26from collections import namedtuple
27
28#
29# HelperFileOutputGeneratorOptions - subclass of GeneratorOptions.
30class HelperFileOutputGeneratorOptions(GeneratorOptions):
31 def __init__(self,
32 filename = None,
33 directory = '.',
34 apiname = None,
35 profile = None,
36 versions = '.*',
37 emitversions = '.*',
38 defaultExtensions = None,
39 addExtensions = None,
40 removeExtensions = None,
41 sortProcedure = regSortFeatures,
42 prefixText = "",
43 genFuncPointers = True,
44 protectFile = True,
45 protectFeature = True,
46 protectProto = None,
47 protectProtoStr = None,
48 apicall = '',
49 apientry = '',
50 apientryp = '',
51 alignFuncParam = 0,
52 library_name = '',
53 helper_file_type = ''):
54 GeneratorOptions.__init__(self, filename, directory, apiname, profile,
55 versions, emitversions, defaultExtensions,
56 addExtensions, removeExtensions, sortProcedure)
57 self.prefixText = prefixText
58 self.genFuncPointers = genFuncPointers
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070059 self.protectFile = protectFile
60 self.protectFeature = protectFeature
61 self.protectProto = protectProto
62 self.protectProtoStr = protectProtoStr
63 self.apicall = apicall
64 self.apientry = apientry
65 self.apientryp = apientryp
66 self.alignFuncParam = alignFuncParam
67 self.library_name = library_name
68 self.helper_file_type = helper_file_type
69#
Mark Lobodzinski46d388f2016-12-28 10:46:26 -070070# HelperFileOutputGenerator - subclass of OutputGenerator. Outputs Vulkan helper files
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070071class HelperFileOutputGenerator(OutputGenerator):
Mark Lobodzinskia3cc3612017-01-03 13:25:10 -070072 """Generate helper file based on XML element attributes"""
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070073 def __init__(self,
74 errFile = sys.stderr,
75 warnFile = sys.stderr,
76 diagFile = sys.stdout):
77 OutputGenerator.__init__(self, errFile, warnFile, diagFile)
78 # Internal state - accumulators for different inner block text
Mark Lobodzinski5380d132016-12-28 14:45:34 -070079 self.enum_output = '' # string built up of enum string routines
80 self.struct_size_h_output = '' # string built up of struct size header output
81 self.struct_size_c_output = '' # string built up of struct size source output
Mark Lobodzinski46d388f2016-12-28 10:46:26 -070082 # Internal state - accumulators for different inner block text
Mark Lobodzinski46d388f2016-12-28 10:46:26 -070083 self.structNames = [] # List of Vulkan struct typenames
84 self.structTypes = dict() # Map of Vulkan struct typename to required VkStructureType
Mark Lobodzinski46d388f2016-12-28 10:46:26 -070085 self.structMembers = [] # List of StructMemberData records for all Vulkan structs
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -060086 self.object_types = [] # List of all handle types
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -060087 self.debug_report_object_types = [] # Handy copy of debug_report_object_type enum data
Mark Young1ded24b2017-05-30 14:53:50 -060088 self.core_object_types = [] # Handy copy of core_object_type enum data
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -060089 self.device_extension_info = dict() # Dict of device extension name defines and ifdef values
90 self.instance_extension_info = dict() # Dict of instance extension name defines and ifdef values
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -060091
Mark Lobodzinski46d388f2016-12-28 10:46:26 -070092 # Named tuples to store struct and command data
93 self.StructType = namedtuple('StructType', ['name', 'value'])
Mark Lobodzinskic67efd02017-01-04 09:16:00 -070094 self.CommandParam = namedtuple('CommandParam', ['type', 'name', 'ispointer', 'isstaticarray', 'isconst', 'iscount', 'len', 'extstructs', 'cdecl'])
Mark Lobodzinski5380d132016-12-28 14:45:34 -070095 self.StructMemberData = namedtuple('StructMemberData', ['name', 'members', 'ifdef_protect'])
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -070096 #
97 # Called once at the beginning of each run
98 def beginFile(self, genOpts):
99 OutputGenerator.beginFile(self, genOpts)
100 # User-supplied prefix text, if any (list of strings)
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700101 self.helper_file_type = genOpts.helper_file_type
102 self.library_name = genOpts.library_name
103 # File Comment
104 file_comment = '// *** THIS FILE IS GENERATED - DO NOT EDIT ***\n'
105 file_comment += '// See helper_file_generator.py for modifications\n'
106 write(file_comment, file=self.outFile)
107 # Copyright Notice
108 copyright = ''
109 copyright += '\n'
110 copyright += '/***************************************************************************\n'
111 copyright += ' *\n'
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700112 copyright += ' * Copyright (c) 2015-2017 The Khronos Group Inc.\n'
113 copyright += ' * Copyright (c) 2015-2017 Valve Corporation\n'
114 copyright += ' * Copyright (c) 2015-2017 LunarG, Inc.\n'
115 copyright += ' * Copyright (c) 2015-2017 Google Inc.\n'
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700116 copyright += ' *\n'
117 copyright += ' * Licensed under the Apache License, Version 2.0 (the "License");\n'
118 copyright += ' * you may not use this file except in compliance with the License.\n'
119 copyright += ' * You may obtain a copy of the License at\n'
120 copyright += ' *\n'
121 copyright += ' * http://www.apache.org/licenses/LICENSE-2.0\n'
122 copyright += ' *\n'
123 copyright += ' * Unless required by applicable law or agreed to in writing, software\n'
124 copyright += ' * distributed under the License is distributed on an "AS IS" BASIS,\n'
125 copyright += ' * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n'
126 copyright += ' * See the License for the specific language governing permissions and\n'
127 copyright += ' * limitations under the License.\n'
128 copyright += ' *\n'
129 copyright += ' * Author: Mark Lobodzinski <mark@lunarg.com>\n'
Mark Lobodzinskia9c963d2016-12-28 07:45:35 -0700130 copyright += ' * Author: Courtney Goeltzenleuchter <courtneygo@google.com>\n'
131 copyright += ' * Author: Tobin Ehlis <tobine@google.com>\n'
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600132 copyright += ' * Author: Chris Forbes <chrisforbes@google.com>\n'
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700133 copyright += ' *\n'
134 copyright += ' ****************************************************************************/\n'
135 write(copyright, file=self.outFile)
136 #
Mark Lobodzinskia3cc3612017-01-03 13:25:10 -0700137 # Write generated file content to output file
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700138 def endFile(self):
139 dest_file = ''
140 dest_file += self.OutputDestFile()
Mark Lobodzinskiafe10542017-01-03 13:22:44 -0700141 # Remove blank lines at EOF
142 if dest_file.endswith('\n'):
143 dest_file = dest_file[:-1]
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700144 write(dest_file, file=self.outFile);
145 # Finish processing in superclass
146 OutputGenerator.endFile(self)
147 #
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600148 # Override parent class to be notified of the beginning of an extension
149 def beginFeature(self, interface, emit):
150 # Start processing in superclass
151 OutputGenerator.beginFeature(self, interface, emit)
152 if self.featureName == 'VK_VERSION_1_0':
153 return
154 nameElem = interface[0][1]
155 name = nameElem.get('name')
156 if 'EXTENSION_NAME' not in name:
157 print("Error in vk.xml file -- extension name is not available")
158 if interface.get('type') == 'instance':
159 self.instance_extension_info[name] = self.featureExtraProtect
160 else:
161 self.device_extension_info[name] = self.featureExtraProtect
162 #
163 # Override parent class to be notified of the end of an extension
164 def endFeature(self):
165 # Finish processing in superclass
166 OutputGenerator.endFeature(self)
167 #
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700168 # Grab group (e.g. C "enum" type) info to output for enum-string conversion helper
169 def genGroup(self, groupinfo, groupName):
170 OutputGenerator.genGroup(self, groupinfo, groupName)
171 groupElem = groupinfo.elem
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700172 # For enum_string_header
173 if self.helper_file_type == 'enum_string_header':
Mark Lobodzinski824ce642016-12-28 08:49:46 -0700174 value_list = []
175 for elem in groupElem.findall('enum'):
176 if elem.get('supported') != 'disabled':
177 item_name = elem.get('name')
178 value_list.append(item_name)
179 if value_list is not None:
180 self.enum_output += self.GenerateEnumStringConversion(groupName, value_list)
Mark Young1ded24b2017-05-30 14:53:50 -0600181 elif self.helper_file_type == 'object_types_header':
182 if groupName == 'VkDebugReportObjectTypeEXT':
183 for elem in groupElem.findall('enum'):
184 if elem.get('supported') != 'disabled':
185 item_name = elem.get('name')
186 self.debug_report_object_types.append(item_name)
187 elif groupName == 'VkObjectType':
188 for elem in groupElem.findall('enum'):
189 if elem.get('supported') != 'disabled':
190 item_name = elem.get('name')
191 self.core_object_types.append(item_name)
192
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700193 #
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700194 # Called for each type -- if the type is a struct/union, grab the metadata
195 def genType(self, typeinfo, name):
196 OutputGenerator.genType(self, typeinfo, name)
197 typeElem = typeinfo.elem
198 # If the type is a struct type, traverse the imbedded <member> tags generating a structure.
199 # Otherwise, emit the tag text.
200 category = typeElem.get('category')
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600201 if category == 'handle':
202 self.object_types.append(name)
203 elif (category == 'struct' or category == 'union'):
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700204 self.structNames.append(name)
205 self.genStruct(typeinfo, name)
206 #
207 # Generate a VkStructureType based on a structure typename
208 def genVkStructureType(self, typename):
209 # Add underscore between lowercase then uppercase
210 value = re.sub('([a-z0-9])([A-Z])', r'\1_\2', typename)
211 # Change to uppercase
212 value = value.upper()
213 # Add STRUCTURE_TYPE_
214 return re.sub('VK_', 'VK_STRUCTURE_TYPE_', value)
215 #
216 # Check if the parameter passed in is a pointer
217 def paramIsPointer(self, param):
218 ispointer = False
219 for elem in param:
220 if ((elem.tag is not 'type') and (elem.tail is not None)) and '*' in elem.tail:
221 ispointer = True
222 return ispointer
223 #
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700224 # Check if the parameter passed in is a static array
225 def paramIsStaticArray(self, param):
226 isstaticarray = 0
227 paramname = param.find('name')
228 if (paramname.tail is not None) and ('[' in paramname.tail):
229 isstaticarray = paramname.tail.count('[')
230 return isstaticarray
231 #
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700232 # Retrieve the type and name for a parameter
233 def getTypeNameTuple(self, param):
234 type = ''
235 name = ''
236 for elem in param:
237 if elem.tag == 'type':
238 type = noneStr(elem.text)
239 elif elem.tag == 'name':
240 name = noneStr(elem.text)
241 return (type, name)
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700242 # Extract length values from latexmath. Currently an inflexible solution that looks for specific
243 # patterns that are found in vk.xml. Will need to be updated when new patterns are introduced.
244 def parseLateXMath(self, source):
245 name = 'ERROR'
246 decoratedName = 'ERROR'
247 if 'mathit' in source:
Mark Lobodzinski36c33862017-02-13 10:15:53 -0700248 # Matches expressions similar to 'latexmath:[\lceil{\mathit{rasterizationSamples} \over 32}\rceil]'
249 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 -0700250 if not match or match.group(1) != match.group(4):
251 raise 'Unrecognized latexmath expression'
252 name = match.group(2)
mizhenc27f6c72017-03-31 09:08:16 -0600253 # Need to add 1 for ceiling function; otherwise, the allocated packet
254 # size will be less than needed during capture for some title which use
255 # this in VkPipelineMultisampleStateCreateInfo. based on ceiling function
256 # definition,it is '{0}%{1}?{0}/{1} + 1:{0}/{1}'.format(*match.group(2, 3)),
257 # its value <= '{}/{} + 1'.
258 if match.group(1) == 'ceil':
259 decoratedName = '{}/{} + 1'.format(*match.group(2, 3))
260 else:
261 decoratedName = '{}/{}'.format(*match.group(2, 3))
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700262 else:
Mark Lobodzinski36c33862017-02-13 10:15:53 -0700263 # Matches expressions similar to 'latexmath : [dataSize \over 4]'
Mark Young0f183a82017-02-28 09:58:04 -0700264 match = re.match(r'latexmath\s*\:\s*\[\s*(\w+)\s*\\over\s*(\d+)\s*\]', source)
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700265 name = match.group(1)
266 decoratedName = '{}/{}'.format(*match.group(1, 2))
267 return name, decoratedName
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700268 #
269 # Retrieve the value of the len tag
270 def getLen(self, param):
271 result = None
272 len = param.attrib.get('len')
273 if len and len != 'null-terminated':
274 # For string arrays, 'len' can look like 'count,null-terminated', indicating that we
275 # have a null terminated array of strings. We strip the null-terminated from the
276 # 'len' field and only return the parameter specifying the string count
277 if 'null-terminated' in len:
278 result = len.split(',')[0]
279 else:
280 result = len
Mark Lobodzinskif8f44fa2017-01-06 08:47:48 -0700281 if 'latexmath' in len:
282 param_type, param_name = self.getTypeNameTuple(param)
283 len_name, result = self.parseLateXMath(len)
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700284 # Spec has now notation for len attributes, using :: instead of platform specific pointer symbol
285 result = str(result).replace('::', '->')
286 return result
287 #
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700288 # Check if a structure is or contains a dispatchable (dispatchable = True) or
289 # non-dispatchable (dispatchable = False) handle
290 def TypeContainsObjectHandle(self, handle_type, dispatchable):
291 if dispatchable:
292 type_key = 'VK_DEFINE_HANDLE'
293 else:
294 type_key = 'VK_DEFINE_NON_DISPATCHABLE_HANDLE'
295 handle = self.registry.tree.find("types/type/[name='" + handle_type + "'][@category='handle']")
296 if handle is not None and handle.find('type').text == type_key:
297 return True
298 # if handle_type is a struct, search its members
299 if handle_type in self.structNames:
300 member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == handle_type), None)
301 if member_index is not None:
302 for item in self.structMembers[member_index].members:
303 handle = self.registry.tree.find("types/type/[name='" + item.type + "'][@category='handle']")
304 if handle is not None and handle.find('type').text == type_key:
305 return True
306 return False
307 #
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700308 # Generate local ready-access data describing Vulkan structures and unions from the XML metadata
309 def genStruct(self, typeinfo, typeName):
310 OutputGenerator.genStruct(self, typeinfo, typeName)
311 members = typeinfo.elem.findall('.//member')
312 # Iterate over members once to get length parameters for arrays
313 lens = set()
314 for member in members:
315 len = self.getLen(member)
316 if len:
317 lens.add(len)
318 # Generate member info
319 membersInfo = []
320 for member in members:
321 # Get the member's type and name
322 info = self.getTypeNameTuple(member)
323 type = info[0]
324 name = info[1]
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700325 cdecl = self.makeCParamDecl(member, 1)
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700326 # Process VkStructureType
327 if type == 'VkStructureType':
328 # Extract the required struct type value from the comments
329 # embedded in the original text defining the 'typeinfo' element
330 rawXml = etree.tostring(typeinfo.elem).decode('ascii')
331 result = re.search(r'VK_STRUCTURE_TYPE_\w+', rawXml)
332 if result:
333 value = result.group(0)
334 else:
335 value = self.genVkStructureType(typeName)
336 # Store the required type value
337 self.structTypes[typeName] = self.StructType(name=name, value=value)
338 # Store pointer/array/string info
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700339 isstaticarray = self.paramIsStaticArray(member)
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700340 membersInfo.append(self.CommandParam(type=type,
341 name=name,
342 ispointer=self.paramIsPointer(member),
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700343 isstaticarray=isstaticarray,
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700344 isconst=True if 'const' in cdecl else False,
345 iscount=True if name in lens else False,
346 len=self.getLen(member),
347 extstructs=member.attrib.get('validextensionstructs') if name == 'pNext' else None,
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700348 cdecl=cdecl))
Mark Lobodzinski5380d132016-12-28 14:45:34 -0700349 self.structMembers.append(self.StructMemberData(name=typeName, members=membersInfo, ifdef_protect=self.featureExtraProtect))
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700350 #
351 # Enum_string_header: Create a routine to convert an enumerated value into a string
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700352 def GenerateEnumStringConversion(self, groupName, value_list):
353 outstring = '\n'
354 outstring += 'static inline const char* string_%s(%s input_value)\n' % (groupName, groupName)
355 outstring += '{\n'
356 outstring += ' switch ((%s)input_value)\n' % groupName
357 outstring += ' {\n'
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700358 for item in value_list:
359 outstring += ' case %s:\n' % item
360 outstring += ' return "%s";\n' % item
361 outstring += ' default:\n'
362 outstring += ' return "Unhandled %s";\n' % groupName
363 outstring += ' }\n'
364 outstring += '}\n'
365 return outstring
366 #
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700367 # Combine enum string helper header file preamble with body text and return
368 def GenerateEnumStringHelperHeader(self):
369 enum_string_helper_header = '\n'
370 enum_string_helper_header += '#pragma once\n'
371 enum_string_helper_header += '#ifdef _WIN32\n'
372 enum_string_helper_header += '#pragma warning( disable : 4065 )\n'
373 enum_string_helper_header += '#endif\n'
374 enum_string_helper_header += '\n'
375 enum_string_helper_header += '#include <vulkan/vulkan.h>\n'
376 enum_string_helper_header += '\n'
377 enum_string_helper_header += self.enum_output
378 return enum_string_helper_header
379 #
Mark Lobodzinski5380d132016-12-28 14:45:34 -0700380 # struct_size_header: build function prototypes for header file
381 def GenerateStructSizeHeader(self):
382 outstring = ''
383 outstring += 'size_t get_struct_chain_size(const void* struct_ptr);\n'
384 for item in self.structMembers:
385 lower_case_name = item.name.lower()
386 if item.ifdef_protect != None:
387 outstring += '#ifdef %s\n' % item.ifdef_protect
388 outstring += 'size_t vk_size_%s(const %s* struct_ptr);\n' % (item.name.lower(), item.name)
389 if item.ifdef_protect != None:
390 outstring += '#endif // %s\n' % item.ifdef_protect
391 outstring += '#ifdef __cplusplus\n'
392 outstring += '}\n'
393 outstring += '#endif'
394 return outstring
395 #
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700396 # Combine struct size helper header file preamble with body text and return
397 def GenerateStructSizeHelperHeader(self):
398 struct_size_helper_header = '\n'
399 struct_size_helper_header += '#ifdef __cplusplus\n'
400 struct_size_helper_header += 'extern "C" {\n'
401 struct_size_helper_header += '#endif\n'
402 struct_size_helper_header += '\n'
403 struct_size_helper_header += '#include <stdio.h>\n'
404 struct_size_helper_header += '#include <stdlib.h>\n'
405 struct_size_helper_header += '#include <vulkan/vulkan.h>\n'
406 struct_size_helper_header += '\n'
407 struct_size_helper_header += '// Function Prototypes\n'
408 struct_size_helper_header += self.GenerateStructSizeHeader()
409 return struct_size_helper_header
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700410 #
411 # Helper function for declaring a counter variable only once
412 def DeclareCounter(self, string_var, declare_flag):
413 if declare_flag == False:
414 string_var += ' uint32_t i = 0;\n'
415 declare_flag = True
416 return string_var, declare_flag
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700417 #
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700418 # Build the header of the get_struct_chain_size function
419 def GenerateChainSizePreamble(self):
420 preable = '\nsize_t get_struct_chain_size(const void* struct_ptr) {\n'
421 preable += ' // Use VkApplicationInfo as struct until actual type is resolved\n'
422 preable += ' VkApplicationInfo* pNext = (VkApplicationInfo*)struct_ptr;\n'
423 preable += ' size_t struct_size = 0;\n'
424 preable += ' while (pNext) {\n'
425 preable += ' switch (pNext->sType) {\n'
426 return preable
427 #
428 # Build the footer of the get_struct_chain_size function
429 def GenerateChainSizePostamble(self):
430 postamble = ' default:\n'
431 postamble += ' assert(0);\n'
432 postamble += ' struct_size += 0;\n'
433 postamble += ' }\n'
434 postamble += ' pNext = (VkApplicationInfo*)pNext->pNext;\n'
435 postamble += ' }\n'
436 postamble += ' return struct_size;\n'
437 postamble += '}'
438 return postamble
439 #
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700440 # struct_size_helper source -- create bodies of struct size helper functions
441 def GenerateStructSizeSource(self):
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700442 # Construct the body of the routine and get_struct_chain_size() simultaneously
443 struct_size_body = ''
444 chain_size = self.GenerateChainSizePreamble()
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700445 for item in self.structMembers:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700446 struct_size_body += '\n'
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700447 lower_case_name = item.name.lower()
448 if item.ifdef_protect != None:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700449 struct_size_body += '#ifdef %s\n' % item.ifdef_protect
450 chain_size += '#ifdef %s\n' % item.ifdef_protect
451 if item.name in self.structTypes:
452 chain_size += ' case %s: {\n' % self.structTypes[item.name].value
453 chain_size += ' struct_size += vk_size_%s((%s*)pNext);\n' % (item.name.lower(), item.name)
454 chain_size += ' break;\n'
455 chain_size += ' }\n'
456 struct_size_body += 'size_t vk_size_%s(const %s* struct_ptr) {\n' % (item.name.lower(), item.name)
457 struct_size_body += ' size_t struct_size = 0;\n'
458 struct_size_body += ' if (struct_ptr) {\n'
459 struct_size_body += ' struct_size = sizeof(%s);\n' % item.name
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700460 counter_declared = False
461 for member in item.members:
462 vulkan_type = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None)
463 if member.ispointer == True:
464 if vulkan_type is not None:
465 # If this is another Vulkan structure call generated size function
466 if member.len is not None:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700467 struct_size_body, counter_declared = self.DeclareCounter(struct_size_body, counter_declared)
468 struct_size_body += ' for (i = 0; i < struct_ptr->%s; i++) {\n' % member.len
469 struct_size_body += ' struct_size += vk_size_%s(&struct_ptr->%s[i]);\n' % (member.type.lower(), member.name)
470 struct_size_body += ' }\n'
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700471 else:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700472 struct_size_body += ' struct_size += vk_size_%s(struct_ptr->%s);\n' % (member.type.lower(), member.name)
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700473 else:
474 if member.type == 'char':
475 # Deal with sizes of character strings
476 if member.len is not None:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700477 struct_size_body, counter_declared = self.DeclareCounter(struct_size_body, counter_declared)
478 struct_size_body += ' for (i = 0; i < struct_ptr->%s; i++) {\n' % member.len
479 struct_size_body += ' struct_size += (sizeof(char*) + (sizeof(char) * (1 + strlen(struct_ptr->%s[i]))));\n' % (member.name)
480 struct_size_body += ' }\n'
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700481 else:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700482 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 -0700483 else:
484 if member.len is not None:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700485 # Avoid using 'sizeof(void)', which generates compile-time warnings/errors
486 checked_type = member.type
487 if checked_type == 'void':
488 checked_type = 'void*'
mizhenc27f6c72017-03-31 09:08:16 -0600489 struct_size_body += ' struct_size += (struct_ptr->%s ) * sizeof(%s);\n' % (member.len, checked_type)
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700490 struct_size_body += ' }\n'
491 struct_size_body += ' return struct_size;\n'
492 struct_size_body += '}\n'
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700493 if item.ifdef_protect != None:
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700494 struct_size_body += '#endif // %s\n' % item.ifdef_protect
495 chain_size += '#endif // %s\n' % item.ifdef_protect
496 chain_size += self.GenerateChainSizePostamble()
497 struct_size_body += chain_size;
498 return struct_size_body
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700499 #
500 # Combine struct size helper source file preamble with body text and return
501 def GenerateStructSizeHelperSource(self):
502 struct_size_helper_source = '\n'
503 struct_size_helper_source += '#include "vk_struct_size_helper.h"\n'
504 struct_size_helper_source += '#include <string.h>\n'
505 struct_size_helper_source += '#include <assert.h>\n'
506 struct_size_helper_source += '\n'
507 struct_size_helper_source += '// Function Definitions\n'
508 struct_size_helper_source += self.GenerateStructSizeSource()
509 return struct_size_helper_source
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700510 #
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700511 # Combine safe struct helper header file preamble with body text and return
512 def GenerateSafeStructHelperHeader(self):
513 safe_struct_helper_header = '\n'
514 safe_struct_helper_header += '#pragma once\n'
515 safe_struct_helper_header += '#include <vulkan/vulkan.h>\n'
516 safe_struct_helper_header += '\n'
517 safe_struct_helper_header += self.GenerateSafeStructHeader()
518 return safe_struct_helper_header
519 #
520 # safe_struct header: build function prototypes for header file
521 def GenerateSafeStructHeader(self):
522 safe_struct_header = ''
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700523 for item in self.structMembers:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700524 if self.NeedSafeStruct(item) == True:
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700525 safe_struct_header += '\n'
526 if item.ifdef_protect != None:
527 safe_struct_header += '#ifdef %s\n' % item.ifdef_protect
528 safe_struct_header += 'struct safe_%s {\n' % (item.name)
529 for member in item.members:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700530 if member.type in self.structNames:
531 member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None)
532 if member_index is not None and self.NeedSafeStruct(self.structMembers[member_index]) == True:
533 if member.ispointer:
534 safe_struct_header += ' safe_%s* %s;\n' % (member.type, member.name)
535 else:
536 safe_struct_header += ' safe_%s %s;\n' % (member.type, member.name)
537 continue
538 if member.len is not None and (self.TypeContainsObjectHandle(member.type, True) or self.TypeContainsObjectHandle(member.type, False)):
539 safe_struct_header += ' %s* %s;\n' % (member.type, member.name)
540 else:
541 safe_struct_header += '%s;\n' % member.cdecl
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700542 safe_struct_header += ' safe_%s(const %s* in_struct);\n' % (item.name, item.name)
543 safe_struct_header += ' safe_%s(const safe_%s& src);\n' % (item.name, item.name)
544 safe_struct_header += ' safe_%s();\n' % item.name
545 safe_struct_header += ' ~safe_%s();\n' % item.name
546 safe_struct_header += ' void initialize(const %s* in_struct);\n' % item.name
547 safe_struct_header += ' void initialize(const safe_%s* src);\n' % item.name
548 safe_struct_header += ' %s *ptr() { return reinterpret_cast<%s *>(this); }\n' % (item.name, item.name)
549 safe_struct_header += ' %s const *ptr() const { return reinterpret_cast<%s const *>(this); }\n' % (item.name, item.name)
550 safe_struct_header += '};\n'
551 if item.ifdef_protect != None:
552 safe_struct_header += '#endif // %s\n' % item.ifdef_protect
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700553 return safe_struct_header
554 #
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600555 # Generate extension helper header file
556 def GenerateExtensionHelperHeader(self):
557 extension_helper_header = '\n'
558 extension_helper_header += '#ifndef VK_EXTENSION_HELPER_H_\n'
559 extension_helper_header += '#define VK_EXTENSION_HELPER_H_\n'
560 struct = '\n'
Tobin Ehlis84154d32017-06-09 15:46:14 -0600561 extension_helper_header += '#include <vulkan/vulkan.h>\n'
562 extension_helper_header += '#include <utility>\n'
563 extension_helper_header += '\n'
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600564 extension_dict = dict()
565 for type in ['Instance', 'Device']:
566 if type == 'Instance':
567 extension_dict = self.instance_extension_info
568 struct += 'struct InstanceExtensions { \n'
569 else:
570 extension_dict = self.device_extension_info
571 struct += 'struct DeviceExtensions : public InstanceExtensions { \n'
572 for ext_name, ifdef in extension_dict.items():
573 bool_name = ext_name.lower()
574 bool_name = re.sub('_extension_name', '', bool_name)
575 struct += ' bool %s{false};\n' % bool_name
576 struct += '\n'
577 if type == 'Instance':
578 struct += ' void InitFromInstanceCreateInfo(const VkInstanceCreateInfo *pCreateInfo) {\n'
579 else:
580 struct += ' void InitFromDeviceCreateInfo(const InstanceExtensions *instance_extensions, const VkDeviceCreateInfo *pCreateInfo) {\n'
581 struct += '\n'
582 struct += ' static const std::pair<char const *, bool %sExtensions::*> known_extensions[]{\n' % type
583 for ext_name, ifdef in extension_dict.items():
584 if ifdef is not None:
585 struct += '#ifdef %s\n' % ifdef
586 bool_name = ext_name.lower()
587 bool_name = re.sub('_extension_name', '', bool_name)
588 struct += ' {%s, &%sExtensions::%s},\n' % (ext_name, type, bool_name)
589 if ifdef is not None:
590 struct += '#endif\n'
591 struct += ' };\n'
592 struct += '\n'
593 struct += ' // Initialize struct data\n'
594 for ext_name, ifdef in self.instance_extension_info.items():
595 bool_name = ext_name.lower()
596 bool_name = re.sub('_extension_name', '', bool_name)
597 if type == 'Device':
598 struct += ' %s = instance_extensions->%s;\n' % (bool_name, bool_name)
599 struct += '\n'
600 struct += ' for (uint32_t i = 0; i < pCreateInfo->enabledExtensionCount; i++) {\n'
601 struct += ' for (auto ext : known_extensions) {\n'
602 struct += ' if (!strcmp(ext.first, pCreateInfo->ppEnabledExtensionNames[i])) {\n'
603 struct += ' this->*(ext.second) = true;\n'
604 struct += ' break;\n'
605 struct += ' }\n'
606 struct += ' }\n'
607 struct += ' }\n'
608 struct += ' }\n'
609 struct += '};\n'
610 struct += '\n'
611 extension_helper_header += struct
612 extension_helper_header += '\n'
613 extension_helper_header += '#endif // VK_EXTENSION_HELPER_H_\n'
614 return extension_helper_header
615 #
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600616 # Combine object types helper header file preamble with body text and return
617 def GenerateObjectTypesHelperHeader(self):
618 object_types_helper_header = '\n'
619 object_types_helper_header += '#pragma once\n'
620 object_types_helper_header += '\n'
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600621 object_types_helper_header += '#include <vulkan/vulkan.h>\n\n'
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600622 object_types_helper_header += self.GenerateObjectTypesHeader()
623 return object_types_helper_header
624 #
625 # Object types header: create object enum type header file
626 def GenerateObjectTypesHeader(self):
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600627 object_types_header = '// Object Type enum for validation layer internal object handling\n'
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600628 object_types_header += 'typedef enum VulkanObjectType {\n'
629 object_types_header += ' kVulkanObjectTypeUnknown = 0,\n'
630 enum_num = 1
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600631 type_list = [];
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600632
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600633 # Output enum definition as each handle is processed, saving the names to use for the conversion routine
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600634 for item in self.object_types:
635 fixup_name = item[2:]
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600636 enum_entry = 'kVulkanObjectType%s' % fixup_name
637 object_types_header += ' ' + enum_entry
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600638 object_types_header += ' = %d,\n' % enum_num
639 enum_num += 1
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600640 type_list.append(enum_entry)
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600641 object_types_header += ' kVulkanObjectTypeMax = %d,\n' % enum_num
Mark Lobodzinskiff92ff82017-04-11 15:31:51 -0600642 object_types_header += '} VulkanObjectType;\n\n'
643
644 # Output name string helper
645 object_types_header += '// Array of object name strings for OBJECT_TYPE enum conversion\n'
Mark Lobodzinski8eb37422017-04-18 14:22:10 -0600646 object_types_header += 'static const char * const object_string[kVulkanObjectTypeMax] = {\n'
Mark Lobodzinskiff92ff82017-04-11 15:31:51 -0600647 object_types_header += ' "Unknown",\n'
648 for item in self.object_types:
649 fixup_name = item[2:]
650 object_types_header += ' "%s",\n' % fixup_name
651 object_types_header += '};\n'
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600652
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600653 # Output a conversion routine from the layer object definitions to the debug report definitions
Mark Lobodzinskiff92ff82017-04-11 15:31:51 -0600654 object_types_header += '\n'
Mark Young1ded24b2017-05-30 14:53:50 -0600655 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 -0600656 object_types_header += 'const VkDebugReportObjectTypeEXT get_debug_report_enum[] = {\n'
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600657 for object_type in type_list:
658 done = False
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600659 search_type = object_type.replace("kVulkanObjectType", "").lower()
660 for vk_object_type in self.debug_report_object_types:
661 target_type = vk_object_type.replace("VK_DEBUG_REPORT_OBJECT_TYPE_", "").lower()
662 target_type = target_type[:-4]
663 target_type = target_type.replace("_", "")
664 if search_type == target_type:
Mark Lobodzinskiecf0ae12017-04-13 08:36:18 -0600665 object_types_header += ' %s, // %s\n' % (vk_object_type, object_type)
Mark Lobodzinski4c51cd02017-04-04 12:07:38 -0600666 done = True
667 break
668 if done == False:
Mark Lobodzinskiff92ff82017-04-11 15:31:51 -0600669 if object_type == 'kVulkanObjectTypeDebugReportCallbackEXT':
Mark Lobodzinskiecf0ae12017-04-13 08:36:18 -0600670 object_types_header += ' VK_DEBUG_REPORT_OBJECT_TYPE_DEBUG_REPORT_EXT, // kVulkanObjectTypeDebugReportCallbackEXT\n'
Mark Lobodzinskiff92ff82017-04-11 15:31:51 -0600671 else:
Mark Lobodzinskiecf0ae12017-04-13 08:36:18 -0600672 object_types_header += ' VK_DEBUG_REPORT_OBJECT_TYPE_UNKNOWN_EXT; // No Match\n'
673 object_types_header += '};\n'
Mark Young1ded24b2017-05-30 14:53:50 -0600674
675 # Output a conversion routine from the layer object definitions to the core object type definitions
676 object_types_header += '\n'
677 object_types_header += '// Helper array to get Official Vulkan VkObjectType enum from the internal layers version\n'
678 object_types_header += 'const VkObjectType get_object_type_enum[] = {\n'
679 for object_type in type_list:
680 done = False
681 search_type = object_type.replace("kVulkanObjectType", "").lower()
682 for vk_object_type in self.core_object_types:
683 target_type = vk_object_type.replace("VK_OBJECT_TYPE_", "").lower()
684 target_type = target_type.replace("_", "")
685 if search_type == target_type:
686 object_types_header += ' %s, // %s\n' % (vk_object_type, object_type)
687 done = True
688 break
689 if done == False:
690 object_types_header += ' VK_OBJECT_TYPE_UNKNOWN; // No Match\n'
691 object_types_header += '};\n'
692
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600693 return object_types_header
694 #
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700695 # Determine if a structure needs a safe_struct helper function
696 # That is, it has an sType or one of its members is a pointer
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700697 def NeedSafeStruct(self, structure):
Mark Lobodzinski5c873842017-01-03 13:22:10 -0700698 if 'sType' == structure.name:
699 return True
700 for member in structure.members:
701 if member.ispointer == True:
702 return True
703 return False
704 #
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700705 # Combine safe struct helper source file preamble with body text and return
706 def GenerateSafeStructHelperSource(self):
707 safe_struct_helper_source = '\n'
708 safe_struct_helper_source += '#include "vk_safe_struct.h"\n'
709 safe_struct_helper_source += '#include <string.h>\n'
710 safe_struct_helper_source += '\n'
711 safe_struct_helper_source += self.GenerateSafeStructSource()
712 return safe_struct_helper_source
713 #
714 # safe_struct source -- create bodies of safe struct helper functions
715 def GenerateSafeStructSource(self):
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700716 safe_struct_body = []
Mark Lobodzinski560729b2017-03-06 08:59:14 -0700717 wsi_structs = ['VkXlibSurfaceCreateInfoKHR',
718 'VkXcbSurfaceCreateInfoKHR',
719 'VkWaylandSurfaceCreateInfoKHR',
720 'VkMirSurfaceCreateInfoKHR',
721 'VkAndroidSurfaceCreateInfoKHR',
722 'VkWin32SurfaceCreateInfoKHR'
723 ]
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700724 for item in self.structMembers:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700725 if self.NeedSafeStruct(item) == False:
726 continue
Mark Lobodzinski560729b2017-03-06 08:59:14 -0700727 if item.name in wsi_structs:
728 continue
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700729 if item.ifdef_protect != None:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700730 safe_struct_body.append("#ifdef %s\n" % item.ifdef_protect)
731 ss_name = "safe_%s" % item.name
732 init_list = '' # list of members in struct constructor initializer
733 default_init_list = '' # Default constructor just inits ptrs to nullptr in initializer
734 init_func_txt = '' # Txt for initialize() function that takes struct ptr and inits members
735 construct_txt = '' # Body of constuctor as well as body of initialize() func following init_func_txt
736 destruct_txt = ''
737 # VkWriteDescriptorSet is special case because pointers may be non-null but ignored
738 custom_construct_txt = {'VkWriteDescriptorSet' :
739 ' switch (descriptorType) {\n'
740 ' case VK_DESCRIPTOR_TYPE_SAMPLER:\n'
741 ' case VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER:\n'
742 ' case VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE:\n'
743 ' case VK_DESCRIPTOR_TYPE_STORAGE_IMAGE:\n'
744 ' case VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT:\n'
745 ' if (descriptorCount && in_struct->pImageInfo) {\n'
746 ' pImageInfo = new VkDescriptorImageInfo[descriptorCount];\n'
747 ' for (uint32_t i=0; i<descriptorCount; ++i) {\n'
748 ' pImageInfo[i] = in_struct->pImageInfo[i];\n'
749 ' }\n'
750 ' }\n'
751 ' break;\n'
752 ' case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER:\n'
753 ' case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER:\n'
754 ' case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC:\n'
755 ' case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC:\n'
756 ' if (descriptorCount && in_struct->pBufferInfo) {\n'
757 ' pBufferInfo = new VkDescriptorBufferInfo[descriptorCount];\n'
758 ' for (uint32_t i=0; i<descriptorCount; ++i) {\n'
759 ' pBufferInfo[i] = in_struct->pBufferInfo[i];\n'
760 ' }\n'
761 ' }\n'
762 ' break;\n'
763 ' case VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER:\n'
764 ' case VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER:\n'
765 ' if (descriptorCount && in_struct->pTexelBufferView) {\n'
766 ' pTexelBufferView = new VkBufferView[descriptorCount];\n'
767 ' for (uint32_t i=0; i<descriptorCount; ++i) {\n'
768 ' pTexelBufferView[i] = in_struct->pTexelBufferView[i];\n'
769 ' }\n'
770 ' }\n'
771 ' break;\n'
772 ' default:\n'
773 ' break;\n'
774 ' }\n'}
775
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700776 for member in item.members:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700777 m_type = member.type
778 if member.type in self.structNames:
779 member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None)
780 if member_index is not None and self.NeedSafeStruct(self.structMembers[member_index]) == True:
781 m_type = 'safe_%s' % member.type
782 if member.ispointer and 'safe_' not in m_type and self.TypeContainsObjectHandle(member.type, False) == False:
783 # 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 -0700784 if m_type in ['void', 'char']:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700785 # For these exceptions just copy initial value over for now
786 init_list += '\n %s(in_struct->%s),' % (member.name, member.name)
787 init_func_txt += ' %s = in_struct->%s;\n' % (member.name, member.name)
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700788 else:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700789 default_init_list += '\n %s(nullptr),' % (member.name)
790 init_list += '\n %s(nullptr),' % (member.name)
791 init_func_txt += ' %s = nullptr;\n' % (member.name)
792 if 'pNext' != member.name and 'void' not in m_type:
Mark Lobodzinski51160a12017-01-18 11:05:48 -0700793 if not member.isstaticarray and (member.len is None or '/' in member.len):
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700794 construct_txt += ' if (in_struct->%s) {\n' % member.name
795 construct_txt += ' %s = new %s(*in_struct->%s);\n' % (member.name, m_type, member.name)
796 construct_txt += ' }\n'
797 destruct_txt += ' if (%s)\n' % member.name
798 destruct_txt += ' delete %s;\n' % member.name
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700799 else:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700800 construct_txt += ' if (in_struct->%s) {\n' % member.name
801 construct_txt += ' %s = new %s[in_struct->%s];\n' % (member.name, m_type, member.len)
802 construct_txt += ' memcpy ((void *)%s, (void *)in_struct->%s, sizeof(%s)*in_struct->%s);\n' % (member.name, member.name, m_type, member.len)
803 construct_txt += ' }\n'
804 destruct_txt += ' if (%s)\n' % member.name
805 destruct_txt += ' delete[] %s;\n' % member.name
806 elif member.isstaticarray or member.len is not None:
807 if member.len is None:
808 # Extract length of static array by grabbing val between []
809 static_array_size = re.match(r"[^[]*\[([^]]*)\]", member.cdecl)
810 construct_txt += ' for (uint32_t i=0; i<%s; ++i) {\n' % static_array_size.group(1)
811 construct_txt += ' %s[i] = in_struct->%s[i];\n' % (member.name, member.name)
812 construct_txt += ' }\n'
813 else:
814 # Init array ptr to NULL
815 default_init_list += '\n %s(nullptr),' % member.name
816 init_list += '\n %s(nullptr),' % member.name
817 init_func_txt += ' %s = nullptr;\n' % member.name
818 array_element = 'in_struct->%s[i]' % member.name
819 if member.type in self.structNames:
820 member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None)
821 if member_index is not None and self.NeedSafeStruct(self.structMembers[member_index]) == True:
822 array_element = '%s(&in_struct->safe_%s[i])' % (member.type, member.name)
823 construct_txt += ' if (%s && in_struct->%s) {\n' % (member.len, member.name)
824 construct_txt += ' %s = new %s[%s];\n' % (member.name, m_type, member.len)
825 destruct_txt += ' if (%s)\n' % member.name
826 destruct_txt += ' delete[] %s;\n' % member.name
827 construct_txt += ' for (uint32_t i=0; i<%s; ++i) {\n' % (member.len)
828 if 'safe_' in m_type:
829 construct_txt += ' %s[i].initialize(&in_struct->%s[i]);\n' % (member.name, member.name)
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700830 else:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700831 construct_txt += ' %s[i] = %s;\n' % (member.name, array_element)
832 construct_txt += ' }\n'
833 construct_txt += ' }\n'
834 elif member.ispointer == True:
835 construct_txt += ' if (in_struct->%s)\n' % member.name
836 construct_txt += ' %s = new %s(in_struct->%s);\n' % (member.name, m_type, member.name)
837 construct_txt += ' else\n'
838 construct_txt += ' %s = NULL;\n' % member.name
839 destruct_txt += ' if (%s)\n' % member.name
840 destruct_txt += ' delete %s;\n' % member.name
841 elif 'safe_' in m_type:
842 init_list += '\n %s(&in_struct->%s),' % (member.name, member.name)
843 init_func_txt += ' %s.initialize(&in_struct->%s);\n' % (member.name, member.name)
844 else:
845 init_list += '\n %s(in_struct->%s),' % (member.name, member.name)
846 init_func_txt += ' %s = in_struct->%s;\n' % (member.name, member.name)
847 if '' != init_list:
848 init_list = init_list[:-1] # hack off final comma
849 if item.name in custom_construct_txt:
850 construct_txt = custom_construct_txt[item.name]
851 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))
852 if '' != default_init_list:
853 default_init_list = " :%s" % (default_init_list[:-1])
854 safe_struct_body.append("\n%s::%s()%s\n{}" % (ss_name, ss_name, default_init_list))
855 # Create slight variation of init and construct txt for copy constructor that takes a src object reference vs. struct ptr
856 copy_construct_init = init_func_txt.replace('in_struct->', 'src.')
857 copy_construct_txt = construct_txt.replace(' (in_struct->', ' (src.') # Exclude 'if' blocks from next line
858 copy_construct_txt = copy_construct_txt.replace('(in_struct->', '(*src.') # Pass object to copy constructors
859 copy_construct_txt = copy_construct_txt.replace('in_struct->', 'src.') # Modify remaining struct refs for src object
860 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
861 safe_struct_body.append("\n%s::~%s()\n{\n%s}" % (ss_name, ss_name, destruct_txt))
862 safe_struct_body.append("\nvoid %s::initialize(const %s* in_struct)\n{\n%s%s}" % (ss_name, item.name, init_func_txt, construct_txt))
863 # Copy initializer uses same txt as copy constructor but has a ptr and not a reference
864 init_copy = copy_construct_init.replace('src.', 'src->')
865 init_construct = copy_construct_txt.replace('src.', 'src->')
866 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 -0700867 if item.ifdef_protect != None:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700868 safe_struct_body.append("#endif // %s\n" % item.ifdef_protect)
869 return "\n".join(safe_struct_body)
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700870 #
Mark Lobodzinski7ada59c2016-12-27 11:11:54 -0700871 # Create a helper file and return it as a string
872 def OutputDestFile(self):
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700873 if self.helper_file_type == 'enum_string_header':
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700874 return self.GenerateEnumStringHelperHeader()
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700875 elif self.helper_file_type == 'struct_size_header':
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700876 return self.GenerateStructSizeHelperHeader()
Mark Lobodzinski46d388f2016-12-28 10:46:26 -0700877 elif self.helper_file_type == 'struct_size_source':
Mark Lobodzinski1a2f1a32016-12-28 15:41:15 -0700878 return self.GenerateStructSizeHelperSource()
Mark Lobodzinskie20e4562017-01-03 11:14:26 -0700879 elif self.helper_file_type == 'safe_struct_header':
880 return self.GenerateSafeStructHelperHeader()
881 elif self.helper_file_type == 'safe_struct_source':
882 return self.GenerateSafeStructHelperSource()
Mark Lobodzinski7cb7da32017-04-03 16:58:04 -0600883 elif self.helper_file_type == 'object_types_header':
884 return self.GenerateObjectTypesHelperHeader()
Mark Lobodzinskiaf86c382017-06-01 07:42:13 -0600885 elif self.helper_file_type == 'extension_helper_header':
886 return self.GenerateExtensionHelperHeader()
Mark Lobodzinski71f1ea12016-12-29 10:23:47 -0700887 else:
Mark Lobodzinskic67efd02017-01-04 09:16:00 -0700888 return 'Bad Helper File Generator Option %s' % self.helper_file_type
Mark Lobodzinskif36e58b2016-12-29 14:04:15 -0700889