Mark Lobodzinski | 7ada59c | 2016-12-27 11:11:54 -0700 | [diff] [blame] | 1 | #!/usr/bin/python3 -i |
| 2 | # |
Mark Lobodzinski | 733f7f4 | 2017-01-10 11:42:22 -0700 | [diff] [blame] | 3 | # 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 Lobodzinski | 7ada59c | 2016-12-27 11:11:54 -0700 | [diff] [blame] | 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> |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 21 | # Author: Tobin Ehlis <tobine@google.com> |
Mark Lobodzinski | 7ada59c | 2016-12-27 11:11:54 -0700 | [diff] [blame] | 22 | |
| 23 | import os,re,sys |
| 24 | import xml.etree.ElementTree as etree |
| 25 | from generator import * |
| 26 | from collections import namedtuple |
| 27 | |
| 28 | # |
| 29 | # HelperFileOutputGeneratorOptions - subclass of GeneratorOptions. |
| 30 | class 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 Lobodzinski | 7ada59c | 2016-12-27 11:11:54 -0700 | [diff] [blame] | 59 | 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 Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 70 | # HelperFileOutputGenerator - subclass of OutputGenerator. Outputs Vulkan helper files |
Mark Lobodzinski | 7ada59c | 2016-12-27 11:11:54 -0700 | [diff] [blame] | 71 | class HelperFileOutputGenerator(OutputGenerator): |
Mark Lobodzinski | a3cc361 | 2017-01-03 13:25:10 -0700 | [diff] [blame] | 72 | """Generate helper file based on XML element attributes""" |
Mark Lobodzinski | 7ada59c | 2016-12-27 11:11:54 -0700 | [diff] [blame] | 73 | 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 Lobodzinski | 5380d13 | 2016-12-28 14:45:34 -0700 | [diff] [blame] | 79 | 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 Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 82 | # Internal state - accumulators for different inner block text |
Mark Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 83 | self.structNames = [] # List of Vulkan struct typenames |
| 84 | self.structTypes = dict() # Map of Vulkan struct typename to required VkStructureType |
Mark Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 85 | self.structMembers = [] # List of StructMemberData records for all Vulkan structs |
Mark Lobodzinski | 7cb7da3 | 2017-04-03 16:58:04 -0600 | [diff] [blame] | 86 | self.object_types = [] # List of all handle types |
Mark Lobodzinski | 4c51cd0 | 2017-04-04 12:07:38 -0600 | [diff] [blame] | 87 | self.debug_report_object_types = [] # Handy copy of debug_report_object_type enum data |
Mark Young | 1ded24b | 2017-05-30 14:53:50 -0600 | [diff] [blame] | 88 | self.core_object_types = [] # Handy copy of core_object_type enum data |
Mark Lobodzinski | af86c38 | 2017-06-01 07:42:13 -0600 | [diff] [blame] | 89 | 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 Lobodzinski | 4c51cd0 | 2017-04-04 12:07:38 -0600 | [diff] [blame] | 91 | |
Mark Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 92 | # Named tuples to store struct and command data |
| 93 | self.StructType = namedtuple('StructType', ['name', 'value']) |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 94 | self.CommandParam = namedtuple('CommandParam', ['type', 'name', 'ispointer', 'isstaticarray', 'isconst', 'iscount', 'len', 'extstructs', 'cdecl']) |
Mark Lobodzinski | 5380d13 | 2016-12-28 14:45:34 -0700 | [diff] [blame] | 95 | self.StructMemberData = namedtuple('StructMemberData', ['name', 'members', 'ifdef_protect']) |
Mark Lobodzinski | 7ada59c | 2016-12-27 11:11:54 -0700 | [diff] [blame] | 96 | # |
| 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 Lobodzinski | 7ada59c | 2016-12-27 11:11:54 -0700 | [diff] [blame] | 101 | 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 Lobodzinski | e20e456 | 2017-01-03 11:14:26 -0700 | [diff] [blame] | 112 | 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 Lobodzinski | 7ada59c | 2016-12-27 11:11:54 -0700 | [diff] [blame] | 116 | 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 Lobodzinski | a9c963d | 2016-12-28 07:45:35 -0700 | [diff] [blame] | 130 | copyright += ' * Author: Courtney Goeltzenleuchter <courtneygo@google.com>\n' |
| 131 | copyright += ' * Author: Tobin Ehlis <tobine@google.com>\n' |
Mark Lobodzinski | af86c38 | 2017-06-01 07:42:13 -0600 | [diff] [blame] | 132 | copyright += ' * Author: Chris Forbes <chrisforbes@google.com>\n' |
Mark Lobodzinski | 7ada59c | 2016-12-27 11:11:54 -0700 | [diff] [blame] | 133 | copyright += ' *\n' |
| 134 | copyright += ' ****************************************************************************/\n' |
| 135 | write(copyright, file=self.outFile) |
| 136 | # |
Mark Lobodzinski | a3cc361 | 2017-01-03 13:25:10 -0700 | [diff] [blame] | 137 | # Write generated file content to output file |
Mark Lobodzinski | 7ada59c | 2016-12-27 11:11:54 -0700 | [diff] [blame] | 138 | def endFile(self): |
| 139 | dest_file = '' |
| 140 | dest_file += self.OutputDestFile() |
Mark Lobodzinski | afe1054 | 2017-01-03 13:22:44 -0700 | [diff] [blame] | 141 | # Remove blank lines at EOF |
| 142 | if dest_file.endswith('\n'): |
| 143 | dest_file = dest_file[:-1] |
Mark Lobodzinski | 7ada59c | 2016-12-27 11:11:54 -0700 | [diff] [blame] | 144 | write(dest_file, file=self.outFile); |
| 145 | # Finish processing in superclass |
| 146 | OutputGenerator.endFile(self) |
| 147 | # |
Mark Lobodzinski | af86c38 | 2017-06-01 07:42:13 -0600 | [diff] [blame] | 148 | # 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 Lobodzinski | 7ada59c | 2016-12-27 11:11:54 -0700 | [diff] [blame] | 168 | # 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 Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 172 | # For enum_string_header |
| 173 | if self.helper_file_type == 'enum_string_header': |
Mark Lobodzinski | 824ce64 | 2016-12-28 08:49:46 -0700 | [diff] [blame] | 174 | 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 Young | 1ded24b | 2017-05-30 14:53:50 -0600 | [diff] [blame] | 181 | 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 Lobodzinski | 7ada59c | 2016-12-27 11:11:54 -0700 | [diff] [blame] | 193 | # |
Mark Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 194 | # 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 Lobodzinski | 7cb7da3 | 2017-04-03 16:58:04 -0600 | [diff] [blame] | 201 | if category == 'handle': |
| 202 | self.object_types.append(name) |
| 203 | elif (category == 'struct' or category == 'union'): |
Mark Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 204 | 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 Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 224 | # 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 Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 232 | # 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 Lobodzinski | f8f44fa | 2017-01-06 08:47:48 -0700 | [diff] [blame] | 242 | # 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 Lobodzinski | 36c3386 | 2017-02-13 10:15:53 -0700 | [diff] [blame] | 248 | # 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 Lobodzinski | f8f44fa | 2017-01-06 08:47:48 -0700 | [diff] [blame] | 250 | if not match or match.group(1) != match.group(4): |
| 251 | raise 'Unrecognized latexmath expression' |
| 252 | name = match.group(2) |
mizhen | c27f6c7 | 2017-03-31 09:08:16 -0600 | [diff] [blame] | 253 | # 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 Lobodzinski | f8f44fa | 2017-01-06 08:47:48 -0700 | [diff] [blame] | 262 | else: |
Mark Lobodzinski | 36c3386 | 2017-02-13 10:15:53 -0700 | [diff] [blame] | 263 | # Matches expressions similar to 'latexmath : [dataSize \over 4]' |
Mark Young | 0f183a8 | 2017-02-28 09:58:04 -0700 | [diff] [blame] | 264 | match = re.match(r'latexmath\s*\:\s*\[\s*(\w+)\s*\\over\s*(\d+)\s*\]', source) |
Mark Lobodzinski | f8f44fa | 2017-01-06 08:47:48 -0700 | [diff] [blame] | 265 | name = match.group(1) |
| 266 | decoratedName = '{}/{}'.format(*match.group(1, 2)) |
| 267 | return name, decoratedName |
Mark Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 268 | # |
| 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 Lobodzinski | f8f44fa | 2017-01-06 08:47:48 -0700 | [diff] [blame] | 281 | if 'latexmath' in len: |
| 282 | param_type, param_name = self.getTypeNameTuple(param) |
| 283 | len_name, result = self.parseLateXMath(len) |
Mark Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 284 | # Spec has now notation for len attributes, using :: instead of platform specific pointer symbol |
| 285 | result = str(result).replace('::', '->') |
| 286 | return result |
| 287 | # |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 288 | # 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 Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 308 | # 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 Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 325 | cdecl = self.makeCParamDecl(member, 1) |
Mark Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 326 | # 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 Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 339 | isstaticarray = self.paramIsStaticArray(member) |
Mark Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 340 | membersInfo.append(self.CommandParam(type=type, |
| 341 | name=name, |
| 342 | ispointer=self.paramIsPointer(member), |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 343 | isstaticarray=isstaticarray, |
Mark Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 344 | isconst=True if 'const' in cdecl else False, |
| 345 | iscount=True if name in lens else False, |
| 346 | len=self.getLen(member), |
Mike Schuchardt | a40d0b0 | 2017-07-23 12:47:47 -0600 | [diff] [blame] | 347 | extstructs=self.registry.validextensionstructs[typeName] if name == 'pNext' else None, |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 348 | cdecl=cdecl)) |
Mark Lobodzinski | 5380d13 | 2016-12-28 14:45:34 -0700 | [diff] [blame] | 349 | self.structMembers.append(self.StructMemberData(name=typeName, members=membersInfo, ifdef_protect=self.featureExtraProtect)) |
Mark Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 350 | # |
| 351 | # Enum_string_header: Create a routine to convert an enumerated value into a string |
Mark Lobodzinski | 7ada59c | 2016-12-27 11:11:54 -0700 | [diff] [blame] | 352 | 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 Lobodzinski | 7ada59c | 2016-12-27 11:11:54 -0700 | [diff] [blame] | 358 | 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 Lobodzinski | 64b432f | 2017-07-24 16:14:16 -0600 | [diff] [blame] | 367 | # Tack on a helper which, given an index into a VkPhysicalDeviceFeatures structure, will print the corresponding feature name |
| 368 | def DeIndexPhysDevFeatures(self): |
| 369 | pdev_members = None |
| 370 | for name, members, ifdef in self.structMembers: |
| 371 | if name == 'VkPhysicalDeviceFeatures': |
| 372 | pdev_members = members |
| 373 | break |
| 374 | deindex = '\n' |
| 375 | deindex += 'static const char * GetPhysDevFeatureString(uint32_t index) {\n' |
| 376 | deindex += ' const char * IndexToPhysDevFeatureString[] = {\n' |
| 377 | for feature in pdev_members: |
| 378 | deindex += ' "%s",\n' % feature.name |
| 379 | deindex += ' };\n\n' |
| 380 | deindex += ' return IndexToPhysDevFeatureString[index];\n' |
| 381 | deindex += '}\n' |
| 382 | return deindex |
| 383 | # |
Mark Lobodzinski | 1a2f1a3 | 2016-12-28 15:41:15 -0700 | [diff] [blame] | 384 | # Combine enum string helper header file preamble with body text and return |
| 385 | def GenerateEnumStringHelperHeader(self): |
| 386 | enum_string_helper_header = '\n' |
| 387 | enum_string_helper_header += '#pragma once\n' |
| 388 | enum_string_helper_header += '#ifdef _WIN32\n' |
| 389 | enum_string_helper_header += '#pragma warning( disable : 4065 )\n' |
| 390 | enum_string_helper_header += '#endif\n' |
| 391 | enum_string_helper_header += '\n' |
| 392 | enum_string_helper_header += '#include <vulkan/vulkan.h>\n' |
| 393 | enum_string_helper_header += '\n' |
| 394 | enum_string_helper_header += self.enum_output |
Mark Lobodzinski | 64b432f | 2017-07-24 16:14:16 -0600 | [diff] [blame] | 395 | enum_string_helper_header += self.DeIndexPhysDevFeatures() |
Mark Lobodzinski | 1a2f1a3 | 2016-12-28 15:41:15 -0700 | [diff] [blame] | 396 | return enum_string_helper_header |
| 397 | # |
Mark Lobodzinski | 5380d13 | 2016-12-28 14:45:34 -0700 | [diff] [blame] | 398 | # struct_size_header: build function prototypes for header file |
| 399 | def GenerateStructSizeHeader(self): |
| 400 | outstring = '' |
| 401 | outstring += 'size_t get_struct_chain_size(const void* struct_ptr);\n' |
| 402 | for item in self.structMembers: |
| 403 | lower_case_name = item.name.lower() |
| 404 | if item.ifdef_protect != None: |
| 405 | outstring += '#ifdef %s\n' % item.ifdef_protect |
| 406 | outstring += 'size_t vk_size_%s(const %s* struct_ptr);\n' % (item.name.lower(), item.name) |
| 407 | if item.ifdef_protect != None: |
| 408 | outstring += '#endif // %s\n' % item.ifdef_protect |
| 409 | outstring += '#ifdef __cplusplus\n' |
| 410 | outstring += '}\n' |
| 411 | outstring += '#endif' |
| 412 | return outstring |
| 413 | # |
Mark Lobodzinski | 1a2f1a3 | 2016-12-28 15:41:15 -0700 | [diff] [blame] | 414 | # Combine struct size helper header file preamble with body text and return |
| 415 | def GenerateStructSizeHelperHeader(self): |
| 416 | struct_size_helper_header = '\n' |
| 417 | struct_size_helper_header += '#ifdef __cplusplus\n' |
| 418 | struct_size_helper_header += 'extern "C" {\n' |
| 419 | struct_size_helper_header += '#endif\n' |
| 420 | struct_size_helper_header += '\n' |
| 421 | struct_size_helper_header += '#include <stdio.h>\n' |
| 422 | struct_size_helper_header += '#include <stdlib.h>\n' |
| 423 | struct_size_helper_header += '#include <vulkan/vulkan.h>\n' |
| 424 | struct_size_helper_header += '\n' |
| 425 | struct_size_helper_header += '// Function Prototypes\n' |
| 426 | struct_size_helper_header += self.GenerateStructSizeHeader() |
| 427 | return struct_size_helper_header |
Mark Lobodzinski | 71f1ea1 | 2016-12-29 10:23:47 -0700 | [diff] [blame] | 428 | # |
| 429 | # Helper function for declaring a counter variable only once |
| 430 | def DeclareCounter(self, string_var, declare_flag): |
| 431 | if declare_flag == False: |
| 432 | string_var += ' uint32_t i = 0;\n' |
| 433 | declare_flag = True |
| 434 | return string_var, declare_flag |
Mark Lobodzinski | 1a2f1a3 | 2016-12-28 15:41:15 -0700 | [diff] [blame] | 435 | # |
Mark Lobodzinski | f36e58b | 2016-12-29 14:04:15 -0700 | [diff] [blame] | 436 | # Build the header of the get_struct_chain_size function |
| 437 | def GenerateChainSizePreamble(self): |
| 438 | preable = '\nsize_t get_struct_chain_size(const void* struct_ptr) {\n' |
| 439 | preable += ' // Use VkApplicationInfo as struct until actual type is resolved\n' |
| 440 | preable += ' VkApplicationInfo* pNext = (VkApplicationInfo*)struct_ptr;\n' |
| 441 | preable += ' size_t struct_size = 0;\n' |
| 442 | preable += ' while (pNext) {\n' |
| 443 | preable += ' switch (pNext->sType) {\n' |
| 444 | return preable |
| 445 | # |
| 446 | # Build the footer of the get_struct_chain_size function |
| 447 | def GenerateChainSizePostamble(self): |
| 448 | postamble = ' default:\n' |
| 449 | postamble += ' assert(0);\n' |
| 450 | postamble += ' struct_size += 0;\n' |
| 451 | postamble += ' }\n' |
| 452 | postamble += ' pNext = (VkApplicationInfo*)pNext->pNext;\n' |
| 453 | postamble += ' }\n' |
| 454 | postamble += ' return struct_size;\n' |
| 455 | postamble += '}' |
| 456 | return postamble |
| 457 | # |
Mark Lobodzinski | 1a2f1a3 | 2016-12-28 15:41:15 -0700 | [diff] [blame] | 458 | # struct_size_helper source -- create bodies of struct size helper functions |
| 459 | def GenerateStructSizeSource(self): |
Mark Lobodzinski | f36e58b | 2016-12-29 14:04:15 -0700 | [diff] [blame] | 460 | # Construct the body of the routine and get_struct_chain_size() simultaneously |
| 461 | struct_size_body = '' |
| 462 | chain_size = self.GenerateChainSizePreamble() |
Mark Lobodzinski | 1a2f1a3 | 2016-12-28 15:41:15 -0700 | [diff] [blame] | 463 | for item in self.structMembers: |
Mark Lobodzinski | f36e58b | 2016-12-29 14:04:15 -0700 | [diff] [blame] | 464 | struct_size_body += '\n' |
Mark Lobodzinski | 1a2f1a3 | 2016-12-28 15:41:15 -0700 | [diff] [blame] | 465 | lower_case_name = item.name.lower() |
| 466 | if item.ifdef_protect != None: |
Mark Lobodzinski | f36e58b | 2016-12-29 14:04:15 -0700 | [diff] [blame] | 467 | struct_size_body += '#ifdef %s\n' % item.ifdef_protect |
| 468 | chain_size += '#ifdef %s\n' % item.ifdef_protect |
| 469 | if item.name in self.structTypes: |
| 470 | chain_size += ' case %s: {\n' % self.structTypes[item.name].value |
| 471 | chain_size += ' struct_size += vk_size_%s((%s*)pNext);\n' % (item.name.lower(), item.name) |
| 472 | chain_size += ' break;\n' |
| 473 | chain_size += ' }\n' |
| 474 | struct_size_body += 'size_t vk_size_%s(const %s* struct_ptr) {\n' % (item.name.lower(), item.name) |
| 475 | struct_size_body += ' size_t struct_size = 0;\n' |
| 476 | struct_size_body += ' if (struct_ptr) {\n' |
| 477 | struct_size_body += ' struct_size = sizeof(%s);\n' % item.name |
Mark Lobodzinski | 71f1ea1 | 2016-12-29 10:23:47 -0700 | [diff] [blame] | 478 | counter_declared = False |
| 479 | for member in item.members: |
| 480 | vulkan_type = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None) |
| 481 | if member.ispointer == True: |
| 482 | if vulkan_type is not None: |
| 483 | # If this is another Vulkan structure call generated size function |
| 484 | if member.len is not None: |
Mark Lobodzinski | f36e58b | 2016-12-29 14:04:15 -0700 | [diff] [blame] | 485 | struct_size_body, counter_declared = self.DeclareCounter(struct_size_body, counter_declared) |
| 486 | struct_size_body += ' for (i = 0; i < struct_ptr->%s; i++) {\n' % member.len |
| 487 | struct_size_body += ' struct_size += vk_size_%s(&struct_ptr->%s[i]);\n' % (member.type.lower(), member.name) |
| 488 | struct_size_body += ' }\n' |
Mark Lobodzinski | 71f1ea1 | 2016-12-29 10:23:47 -0700 | [diff] [blame] | 489 | else: |
Mark Lobodzinski | f36e58b | 2016-12-29 14:04:15 -0700 | [diff] [blame] | 490 | struct_size_body += ' struct_size += vk_size_%s(struct_ptr->%s);\n' % (member.type.lower(), member.name) |
Mark Lobodzinski | 71f1ea1 | 2016-12-29 10:23:47 -0700 | [diff] [blame] | 491 | else: |
| 492 | if member.type == 'char': |
| 493 | # Deal with sizes of character strings |
| 494 | if member.len is not None: |
Mark Lobodzinski | f36e58b | 2016-12-29 14:04:15 -0700 | [diff] [blame] | 495 | struct_size_body, counter_declared = self.DeclareCounter(struct_size_body, counter_declared) |
| 496 | struct_size_body += ' for (i = 0; i < struct_ptr->%s; i++) {\n' % member.len |
| 497 | struct_size_body += ' struct_size += (sizeof(char*) + (sizeof(char) * (1 + strlen(struct_ptr->%s[i]))));\n' % (member.name) |
| 498 | struct_size_body += ' }\n' |
Mark Lobodzinski | 71f1ea1 | 2016-12-29 10:23:47 -0700 | [diff] [blame] | 499 | else: |
Mark Lobodzinski | f36e58b | 2016-12-29 14:04:15 -0700 | [diff] [blame] | 500 | struct_size_body += ' struct_size += (struct_ptr->%s != NULL) ? sizeof(char)*(1+strlen(struct_ptr->%s)) : 0;\n' % (member.name, member.name) |
Mark Lobodzinski | 71f1ea1 | 2016-12-29 10:23:47 -0700 | [diff] [blame] | 501 | else: |
| 502 | if member.len is not None: |
Mark Lobodzinski | f36e58b | 2016-12-29 14:04:15 -0700 | [diff] [blame] | 503 | # Avoid using 'sizeof(void)', which generates compile-time warnings/errors |
| 504 | checked_type = member.type |
| 505 | if checked_type == 'void': |
| 506 | checked_type = 'void*' |
mizhen | c27f6c7 | 2017-03-31 09:08:16 -0600 | [diff] [blame] | 507 | struct_size_body += ' struct_size += (struct_ptr->%s ) * sizeof(%s);\n' % (member.len, checked_type) |
Mark Lobodzinski | f36e58b | 2016-12-29 14:04:15 -0700 | [diff] [blame] | 508 | struct_size_body += ' }\n' |
| 509 | struct_size_body += ' return struct_size;\n' |
| 510 | struct_size_body += '}\n' |
Mark Lobodzinski | 1a2f1a3 | 2016-12-28 15:41:15 -0700 | [diff] [blame] | 511 | if item.ifdef_protect != None: |
Mark Lobodzinski | f36e58b | 2016-12-29 14:04:15 -0700 | [diff] [blame] | 512 | struct_size_body += '#endif // %s\n' % item.ifdef_protect |
| 513 | chain_size += '#endif // %s\n' % item.ifdef_protect |
| 514 | chain_size += self.GenerateChainSizePostamble() |
| 515 | struct_size_body += chain_size; |
| 516 | return struct_size_body |
Mark Lobodzinski | 1a2f1a3 | 2016-12-28 15:41:15 -0700 | [diff] [blame] | 517 | # |
| 518 | # Combine struct size helper source file preamble with body text and return |
| 519 | def GenerateStructSizeHelperSource(self): |
| 520 | struct_size_helper_source = '\n' |
| 521 | struct_size_helper_source += '#include "vk_struct_size_helper.h"\n' |
| 522 | struct_size_helper_source += '#include <string.h>\n' |
| 523 | struct_size_helper_source += '#include <assert.h>\n' |
| 524 | struct_size_helper_source += '\n' |
| 525 | struct_size_helper_source += '// Function Definitions\n' |
| 526 | struct_size_helper_source += self.GenerateStructSizeSource() |
| 527 | return struct_size_helper_source |
Mark Lobodzinski | 1a2f1a3 | 2016-12-28 15:41:15 -0700 | [diff] [blame] | 528 | # |
Mark Lobodzinski | e20e456 | 2017-01-03 11:14:26 -0700 | [diff] [blame] | 529 | # Combine safe struct helper header file preamble with body text and return |
| 530 | def GenerateSafeStructHelperHeader(self): |
| 531 | safe_struct_helper_header = '\n' |
| 532 | safe_struct_helper_header += '#pragma once\n' |
| 533 | safe_struct_helper_header += '#include <vulkan/vulkan.h>\n' |
| 534 | safe_struct_helper_header += '\n' |
| 535 | safe_struct_helper_header += self.GenerateSafeStructHeader() |
| 536 | return safe_struct_helper_header |
| 537 | # |
| 538 | # safe_struct header: build function prototypes for header file |
| 539 | def GenerateSafeStructHeader(self): |
| 540 | safe_struct_header = '' |
Mark Lobodzinski | e20e456 | 2017-01-03 11:14:26 -0700 | [diff] [blame] | 541 | for item in self.structMembers: |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 542 | if self.NeedSafeStruct(item) == True: |
Mark Lobodzinski | 5c87384 | 2017-01-03 13:22:10 -0700 | [diff] [blame] | 543 | safe_struct_header += '\n' |
| 544 | if item.ifdef_protect != None: |
| 545 | safe_struct_header += '#ifdef %s\n' % item.ifdef_protect |
| 546 | safe_struct_header += 'struct safe_%s {\n' % (item.name) |
| 547 | for member in item.members: |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 548 | if member.type in self.structNames: |
| 549 | member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None) |
| 550 | if member_index is not None and self.NeedSafeStruct(self.structMembers[member_index]) == True: |
| 551 | if member.ispointer: |
| 552 | safe_struct_header += ' safe_%s* %s;\n' % (member.type, member.name) |
| 553 | else: |
| 554 | safe_struct_header += ' safe_%s %s;\n' % (member.type, member.name) |
| 555 | continue |
| 556 | if member.len is not None and (self.TypeContainsObjectHandle(member.type, True) or self.TypeContainsObjectHandle(member.type, False)): |
| 557 | safe_struct_header += ' %s* %s;\n' % (member.type, member.name) |
| 558 | else: |
| 559 | safe_struct_header += '%s;\n' % member.cdecl |
Mark Lobodzinski | 5cd0851 | 2017-09-12 09:50:25 -0600 | [diff] [blame] | 560 | safe_struct_header += ' safe_%s(const %s* in_struct);\n' % (item.name, item.name) |
| 561 | safe_struct_header += ' safe_%s(const safe_%s& src);\n' % (item.name, item.name) |
Chris Forbes | fb63383 | 2017-10-03 18:11:54 -0700 | [diff] [blame^] | 562 | safe_struct_header += ' safe_%s& operator=(const safe_%s& src);\n' % (item.name, item.name) |
Mark Lobodzinski | 5c87384 | 2017-01-03 13:22:10 -0700 | [diff] [blame] | 563 | safe_struct_header += ' safe_%s();\n' % item.name |
| 564 | safe_struct_header += ' ~safe_%s();\n' % item.name |
Mark Lobodzinski | 5cd0851 | 2017-09-12 09:50:25 -0600 | [diff] [blame] | 565 | safe_struct_header += ' void initialize(const %s* in_struct);\n' % item.name |
| 566 | safe_struct_header += ' void initialize(const safe_%s* src);\n' % item.name |
Mark Lobodzinski | 5c87384 | 2017-01-03 13:22:10 -0700 | [diff] [blame] | 567 | safe_struct_header += ' %s *ptr() { return reinterpret_cast<%s *>(this); }\n' % (item.name, item.name) |
| 568 | safe_struct_header += ' %s const *ptr() const { return reinterpret_cast<%s const *>(this); }\n' % (item.name, item.name) |
| 569 | safe_struct_header += '};\n' |
| 570 | if item.ifdef_protect != None: |
| 571 | safe_struct_header += '#endif // %s\n' % item.ifdef_protect |
Mark Lobodzinski | e20e456 | 2017-01-03 11:14:26 -0700 | [diff] [blame] | 572 | return safe_struct_header |
| 573 | # |
Mark Lobodzinski | af86c38 | 2017-06-01 07:42:13 -0600 | [diff] [blame] | 574 | # Generate extension helper header file |
| 575 | def GenerateExtensionHelperHeader(self): |
| 576 | extension_helper_header = '\n' |
| 577 | extension_helper_header += '#ifndef VK_EXTENSION_HELPER_H_\n' |
| 578 | extension_helper_header += '#define VK_EXTENSION_HELPER_H_\n' |
| 579 | struct = '\n' |
Tobin Ehlis | 84154d3 | 2017-06-09 15:46:14 -0600 | [diff] [blame] | 580 | extension_helper_header += '#include <vulkan/vulkan.h>\n' |
Tobin Ehlis | d922d4c | 2017-06-14 09:43:04 -0600 | [diff] [blame] | 581 | extension_helper_header += '#include <string.h>\n' |
Tobin Ehlis | 84154d3 | 2017-06-09 15:46:14 -0600 | [diff] [blame] | 582 | extension_helper_header += '#include <utility>\n' |
| 583 | extension_helper_header += '\n' |
Mark Lobodzinski | af86c38 | 2017-06-01 07:42:13 -0600 | [diff] [blame] | 584 | extension_dict = dict() |
| 585 | for type in ['Instance', 'Device']: |
| 586 | if type == 'Instance': |
| 587 | extension_dict = self.instance_extension_info |
| 588 | struct += 'struct InstanceExtensions { \n' |
| 589 | else: |
| 590 | extension_dict = self.device_extension_info |
| 591 | struct += 'struct DeviceExtensions : public InstanceExtensions { \n' |
| 592 | for ext_name, ifdef in extension_dict.items(): |
| 593 | bool_name = ext_name.lower() |
| 594 | bool_name = re.sub('_extension_name', '', bool_name) |
| 595 | struct += ' bool %s{false};\n' % bool_name |
| 596 | struct += '\n' |
| 597 | if type == 'Instance': |
| 598 | struct += ' void InitFromInstanceCreateInfo(const VkInstanceCreateInfo *pCreateInfo) {\n' |
| 599 | else: |
| 600 | struct += ' void InitFromDeviceCreateInfo(const InstanceExtensions *instance_extensions, const VkDeviceCreateInfo *pCreateInfo) {\n' |
| 601 | struct += '\n' |
| 602 | struct += ' static const std::pair<char const *, bool %sExtensions::*> known_extensions[]{\n' % type |
| 603 | for ext_name, ifdef in extension_dict.items(): |
| 604 | if ifdef is not None: |
| 605 | struct += '#ifdef %s\n' % ifdef |
| 606 | bool_name = ext_name.lower() |
| 607 | bool_name = re.sub('_extension_name', '', bool_name) |
| 608 | struct += ' {%s, &%sExtensions::%s},\n' % (ext_name, type, bool_name) |
| 609 | if ifdef is not None: |
| 610 | struct += '#endif\n' |
| 611 | struct += ' };\n' |
| 612 | struct += '\n' |
| 613 | struct += ' // Initialize struct data\n' |
| 614 | for ext_name, ifdef in self.instance_extension_info.items(): |
| 615 | bool_name = ext_name.lower() |
| 616 | bool_name = re.sub('_extension_name', '', bool_name) |
| 617 | if type == 'Device': |
| 618 | struct += ' %s = instance_extensions->%s;\n' % (bool_name, bool_name) |
| 619 | struct += '\n' |
| 620 | struct += ' for (uint32_t i = 0; i < pCreateInfo->enabledExtensionCount; i++) {\n' |
| 621 | struct += ' for (auto ext : known_extensions) {\n' |
| 622 | struct += ' if (!strcmp(ext.first, pCreateInfo->ppEnabledExtensionNames[i])) {\n' |
| 623 | struct += ' this->*(ext.second) = true;\n' |
| 624 | struct += ' break;\n' |
| 625 | struct += ' }\n' |
| 626 | struct += ' }\n' |
| 627 | struct += ' }\n' |
| 628 | struct += ' }\n' |
| 629 | struct += '};\n' |
| 630 | struct += '\n' |
| 631 | extension_helper_header += struct |
| 632 | extension_helper_header += '\n' |
| 633 | extension_helper_header += '#endif // VK_EXTENSION_HELPER_H_\n' |
| 634 | return extension_helper_header |
| 635 | # |
Mark Lobodzinski | 7cb7da3 | 2017-04-03 16:58:04 -0600 | [diff] [blame] | 636 | # Combine object types helper header file preamble with body text and return |
| 637 | def GenerateObjectTypesHelperHeader(self): |
| 638 | object_types_helper_header = '\n' |
| 639 | object_types_helper_header += '#pragma once\n' |
| 640 | object_types_helper_header += '\n' |
Mark Lobodzinski | 4c51cd0 | 2017-04-04 12:07:38 -0600 | [diff] [blame] | 641 | object_types_helper_header += '#include <vulkan/vulkan.h>\n\n' |
Mark Lobodzinski | 7cb7da3 | 2017-04-03 16:58:04 -0600 | [diff] [blame] | 642 | object_types_helper_header += self.GenerateObjectTypesHeader() |
| 643 | return object_types_helper_header |
| 644 | # |
| 645 | # Object types header: create object enum type header file |
| 646 | def GenerateObjectTypesHeader(self): |
Mark Lobodzinski | 4c51cd0 | 2017-04-04 12:07:38 -0600 | [diff] [blame] | 647 | object_types_header = '// Object Type enum for validation layer internal object handling\n' |
Mark Lobodzinski | 7cb7da3 | 2017-04-03 16:58:04 -0600 | [diff] [blame] | 648 | object_types_header += 'typedef enum VulkanObjectType {\n' |
| 649 | object_types_header += ' kVulkanObjectTypeUnknown = 0,\n' |
| 650 | enum_num = 1 |
Mark Lobodzinski | 4c51cd0 | 2017-04-04 12:07:38 -0600 | [diff] [blame] | 651 | type_list = []; |
Mark Lobodzinski | 7cb7da3 | 2017-04-03 16:58:04 -0600 | [diff] [blame] | 652 | |
Mark Lobodzinski | 4c51cd0 | 2017-04-04 12:07:38 -0600 | [diff] [blame] | 653 | # Output enum definition as each handle is processed, saving the names to use for the conversion routine |
Mark Lobodzinski | 7cb7da3 | 2017-04-03 16:58:04 -0600 | [diff] [blame] | 654 | for item in self.object_types: |
| 655 | fixup_name = item[2:] |
Mark Lobodzinski | 4c51cd0 | 2017-04-04 12:07:38 -0600 | [diff] [blame] | 656 | enum_entry = 'kVulkanObjectType%s' % fixup_name |
| 657 | object_types_header += ' ' + enum_entry |
Mark Lobodzinski | 7cb7da3 | 2017-04-03 16:58:04 -0600 | [diff] [blame] | 658 | object_types_header += ' = %d,\n' % enum_num |
| 659 | enum_num += 1 |
Mark Lobodzinski | 4c51cd0 | 2017-04-04 12:07:38 -0600 | [diff] [blame] | 660 | type_list.append(enum_entry) |
Mark Lobodzinski | 7cb7da3 | 2017-04-03 16:58:04 -0600 | [diff] [blame] | 661 | object_types_header += ' kVulkanObjectTypeMax = %d,\n' % enum_num |
Mark Lobodzinski | ff92ff8 | 2017-04-11 15:31:51 -0600 | [diff] [blame] | 662 | object_types_header += '} VulkanObjectType;\n\n' |
| 663 | |
| 664 | # Output name string helper |
| 665 | object_types_header += '// Array of object name strings for OBJECT_TYPE enum conversion\n' |
Mark Lobodzinski | 8eb3742 | 2017-04-18 14:22:10 -0600 | [diff] [blame] | 666 | object_types_header += 'static const char * const object_string[kVulkanObjectTypeMax] = {\n' |
Mark Lobodzinski | ff92ff8 | 2017-04-11 15:31:51 -0600 | [diff] [blame] | 667 | object_types_header += ' "Unknown",\n' |
| 668 | for item in self.object_types: |
| 669 | fixup_name = item[2:] |
| 670 | object_types_header += ' "%s",\n' % fixup_name |
| 671 | object_types_header += '};\n' |
Mark Lobodzinski | 7cb7da3 | 2017-04-03 16:58:04 -0600 | [diff] [blame] | 672 | |
Mark Lobodzinski | 4c51cd0 | 2017-04-04 12:07:38 -0600 | [diff] [blame] | 673 | # Output a conversion routine from the layer object definitions to the debug report definitions |
Mark Lobodzinski | ff92ff8 | 2017-04-11 15:31:51 -0600 | [diff] [blame] | 674 | object_types_header += '\n' |
Mark Young | 1ded24b | 2017-05-30 14:53:50 -0600 | [diff] [blame] | 675 | object_types_header += '// Helper array to get Vulkan VK_EXT_debug_report object type enum from the internal layers version\n' |
Mark Lobodzinski | c51dbb7 | 2017-04-13 14:25:39 -0600 | [diff] [blame] | 676 | object_types_header += 'const VkDebugReportObjectTypeEXT get_debug_report_enum[] = {\n' |
Mark Lobodzinski | 4c51cd0 | 2017-04-04 12:07:38 -0600 | [diff] [blame] | 677 | for object_type in type_list: |
| 678 | done = False |
Mark Lobodzinski | 4c51cd0 | 2017-04-04 12:07:38 -0600 | [diff] [blame] | 679 | search_type = object_type.replace("kVulkanObjectType", "").lower() |
| 680 | for vk_object_type in self.debug_report_object_types: |
| 681 | target_type = vk_object_type.replace("VK_DEBUG_REPORT_OBJECT_TYPE_", "").lower() |
| 682 | target_type = target_type[:-4] |
| 683 | target_type = target_type.replace("_", "") |
| 684 | if search_type == target_type: |
Mark Lobodzinski | ecf0ae1 | 2017-04-13 08:36:18 -0600 | [diff] [blame] | 685 | object_types_header += ' %s, // %s\n' % (vk_object_type, object_type) |
Mark Lobodzinski | 4c51cd0 | 2017-04-04 12:07:38 -0600 | [diff] [blame] | 686 | done = True |
| 687 | break |
| 688 | if done == False: |
Mark Young | abc2d6e | 2017-07-07 07:59:56 -0600 | [diff] [blame] | 689 | object_types_header += ' VK_DEBUG_REPORT_OBJECT_TYPE_UNKNOWN_EXT, // No Match\n' |
Mark Lobodzinski | ecf0ae1 | 2017-04-13 08:36:18 -0600 | [diff] [blame] | 690 | object_types_header += '};\n' |
Mark Young | 1ded24b | 2017-05-30 14:53:50 -0600 | [diff] [blame] | 691 | |
| 692 | # Output a conversion routine from the layer object definitions to the core object type definitions |
| 693 | object_types_header += '\n' |
| 694 | object_types_header += '// Helper array to get Official Vulkan VkObjectType enum from the internal layers version\n' |
| 695 | object_types_header += 'const VkObjectType get_object_type_enum[] = {\n' |
| 696 | for object_type in type_list: |
| 697 | done = False |
| 698 | search_type = object_type.replace("kVulkanObjectType", "").lower() |
| 699 | for vk_object_type in self.core_object_types: |
| 700 | target_type = vk_object_type.replace("VK_OBJECT_TYPE_", "").lower() |
| 701 | target_type = target_type.replace("_", "") |
| 702 | if search_type == target_type: |
| 703 | object_types_header += ' %s, // %s\n' % (vk_object_type, object_type) |
| 704 | done = True |
| 705 | break |
| 706 | if done == False: |
Mark Young | abc2d6e | 2017-07-07 07:59:56 -0600 | [diff] [blame] | 707 | object_types_header += ' VK_OBJECT_TYPE_UNKNOWN, // No Match\n' |
Mark Young | 1ded24b | 2017-05-30 14:53:50 -0600 | [diff] [blame] | 708 | object_types_header += '};\n' |
| 709 | |
Mark Lobodzinski | 7cb7da3 | 2017-04-03 16:58:04 -0600 | [diff] [blame] | 710 | return object_types_header |
| 711 | # |
Mark Lobodzinski | 5c87384 | 2017-01-03 13:22:10 -0700 | [diff] [blame] | 712 | # Determine if a structure needs a safe_struct helper function |
| 713 | # That is, it has an sType or one of its members is a pointer |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 714 | def NeedSafeStruct(self, structure): |
Mark Lobodzinski | 5c87384 | 2017-01-03 13:22:10 -0700 | [diff] [blame] | 715 | if 'sType' == structure.name: |
| 716 | return True |
| 717 | for member in structure.members: |
| 718 | if member.ispointer == True: |
| 719 | return True |
| 720 | return False |
| 721 | # |
Mark Lobodzinski | e20e456 | 2017-01-03 11:14:26 -0700 | [diff] [blame] | 722 | # Combine safe struct helper source file preamble with body text and return |
| 723 | def GenerateSafeStructHelperSource(self): |
| 724 | safe_struct_helper_source = '\n' |
| 725 | safe_struct_helper_source += '#include "vk_safe_struct.h"\n' |
| 726 | safe_struct_helper_source += '#include <string.h>\n' |
| 727 | safe_struct_helper_source += '\n' |
| 728 | safe_struct_helper_source += self.GenerateSafeStructSource() |
| 729 | return safe_struct_helper_source |
| 730 | # |
| 731 | # safe_struct source -- create bodies of safe struct helper functions |
| 732 | def GenerateSafeStructSource(self): |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 733 | safe_struct_body = [] |
Mark Lobodzinski | 560729b | 2017-03-06 08:59:14 -0700 | [diff] [blame] | 734 | wsi_structs = ['VkXlibSurfaceCreateInfoKHR', |
| 735 | 'VkXcbSurfaceCreateInfoKHR', |
| 736 | 'VkWaylandSurfaceCreateInfoKHR', |
| 737 | 'VkMirSurfaceCreateInfoKHR', |
| 738 | 'VkAndroidSurfaceCreateInfoKHR', |
| 739 | 'VkWin32SurfaceCreateInfoKHR' |
| 740 | ] |
Mark Lobodzinski | e20e456 | 2017-01-03 11:14:26 -0700 | [diff] [blame] | 741 | for item in self.structMembers: |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 742 | if self.NeedSafeStruct(item) == False: |
| 743 | continue |
Mark Lobodzinski | 560729b | 2017-03-06 08:59:14 -0700 | [diff] [blame] | 744 | if item.name in wsi_structs: |
| 745 | continue |
Mark Lobodzinski | e20e456 | 2017-01-03 11:14:26 -0700 | [diff] [blame] | 746 | if item.ifdef_protect != None: |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 747 | safe_struct_body.append("#ifdef %s\n" % item.ifdef_protect) |
| 748 | ss_name = "safe_%s" % item.name |
| 749 | init_list = '' # list of members in struct constructor initializer |
| 750 | default_init_list = '' # Default constructor just inits ptrs to nullptr in initializer |
| 751 | init_func_txt = '' # Txt for initialize() function that takes struct ptr and inits members |
| 752 | construct_txt = '' # Body of constuctor as well as body of initialize() func following init_func_txt |
| 753 | destruct_txt = '' |
Mark Lobodzinski | 5cd0851 | 2017-09-12 09:50:25 -0600 | [diff] [blame] | 754 | # VkWriteDescriptorSet is special case because pointers may be non-null but ignored |
| 755 | custom_construct_txt = {'VkWriteDescriptorSet' : |
| 756 | ' switch (descriptorType) {\n' |
| 757 | ' case VK_DESCRIPTOR_TYPE_SAMPLER:\n' |
| 758 | ' case VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER:\n' |
| 759 | ' case VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE:\n' |
| 760 | ' case VK_DESCRIPTOR_TYPE_STORAGE_IMAGE:\n' |
| 761 | ' case VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT:\n' |
| 762 | ' if (descriptorCount && in_struct->pImageInfo) {\n' |
| 763 | ' pImageInfo = new VkDescriptorImageInfo[descriptorCount];\n' |
| 764 | ' for (uint32_t i=0; i<descriptorCount; ++i) {\n' |
| 765 | ' pImageInfo[i] = in_struct->pImageInfo[i];\n' |
| 766 | ' }\n' |
| 767 | ' }\n' |
| 768 | ' break;\n' |
| 769 | ' case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER:\n' |
| 770 | ' case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER:\n' |
| 771 | ' case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC:\n' |
| 772 | ' case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC:\n' |
| 773 | ' if (descriptorCount && in_struct->pBufferInfo) {\n' |
| 774 | ' pBufferInfo = new VkDescriptorBufferInfo[descriptorCount];\n' |
| 775 | ' for (uint32_t i=0; i<descriptorCount; ++i) {\n' |
| 776 | ' pBufferInfo[i] = in_struct->pBufferInfo[i];\n' |
| 777 | ' }\n' |
| 778 | ' }\n' |
| 779 | ' break;\n' |
| 780 | ' case VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER:\n' |
| 781 | ' case VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER:\n' |
| 782 | ' if (descriptorCount && in_struct->pTexelBufferView) {\n' |
| 783 | ' pTexelBufferView = new VkBufferView[descriptorCount];\n' |
| 784 | ' for (uint32_t i=0; i<descriptorCount; ++i) {\n' |
| 785 | ' pTexelBufferView[i] = in_struct->pTexelBufferView[i];\n' |
| 786 | ' }\n' |
| 787 | ' }\n' |
| 788 | ' break;\n' |
| 789 | ' default:\n' |
| 790 | ' break;\n' |
| 791 | ' }\n', |
| 792 | 'VkShaderModuleCreateInfo' : |
| 793 | ' if (in_struct->pCode) {\n' |
| 794 | ' pCode = reinterpret_cast<uint32_t *>(new uint8_t[codeSize]);\n' |
| 795 | ' memcpy((void *)pCode, (void *)in_struct->pCode, codeSize);\n' |
| 796 | ' }\n'} |
Mike Schuchardt | 8148576 | 2017-09-04 11:38:42 -0600 | [diff] [blame] | 797 | custom_destruct_txt = {'VkShaderModuleCreateInfo' : |
| 798 | ' if (pCode)\n' |
| 799 | ' delete[] reinterpret_cast<const uint8_t *>(pCode);\n' } |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 800 | |
Mark Lobodzinski | e20e456 | 2017-01-03 11:14:26 -0700 | [diff] [blame] | 801 | for member in item.members: |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 802 | m_type = member.type |
| 803 | if member.type in self.structNames: |
| 804 | member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None) |
| 805 | if member_index is not None and self.NeedSafeStruct(self.structMembers[member_index]) == True: |
| 806 | m_type = 'safe_%s' % member.type |
| 807 | if member.ispointer and 'safe_' not in m_type and self.TypeContainsObjectHandle(member.type, False) == False: |
| 808 | # Ptr types w/o a safe_struct, for non-null case need to allocate new ptr and copy data in |
Mark Lobodzinski | 560729b | 2017-03-06 08:59:14 -0700 | [diff] [blame] | 809 | if m_type in ['void', 'char']: |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 810 | # For these exceptions just copy initial value over for now |
| 811 | init_list += '\n %s(in_struct->%s),' % (member.name, member.name) |
| 812 | init_func_txt += ' %s = in_struct->%s;\n' % (member.name, member.name) |
Mark Lobodzinski | e20e456 | 2017-01-03 11:14:26 -0700 | [diff] [blame] | 813 | else: |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 814 | default_init_list += '\n %s(nullptr),' % (member.name) |
| 815 | init_list += '\n %s(nullptr),' % (member.name) |
| 816 | init_func_txt += ' %s = nullptr;\n' % (member.name) |
| 817 | if 'pNext' != member.name and 'void' not in m_type: |
Mark Lobodzinski | 51160a1 | 2017-01-18 11:05:48 -0700 | [diff] [blame] | 818 | if not member.isstaticarray and (member.len is None or '/' in member.len): |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 819 | construct_txt += ' if (in_struct->%s) {\n' % member.name |
| 820 | construct_txt += ' %s = new %s(*in_struct->%s);\n' % (member.name, m_type, member.name) |
| 821 | construct_txt += ' }\n' |
| 822 | destruct_txt += ' if (%s)\n' % member.name |
| 823 | destruct_txt += ' delete %s;\n' % member.name |
Mark Lobodzinski | e20e456 | 2017-01-03 11:14:26 -0700 | [diff] [blame] | 824 | else: |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 825 | construct_txt += ' if (in_struct->%s) {\n' % member.name |
| 826 | construct_txt += ' %s = new %s[in_struct->%s];\n' % (member.name, m_type, member.len) |
| 827 | construct_txt += ' memcpy ((void *)%s, (void *)in_struct->%s, sizeof(%s)*in_struct->%s);\n' % (member.name, member.name, m_type, member.len) |
| 828 | construct_txt += ' }\n' |
| 829 | destruct_txt += ' if (%s)\n' % member.name |
| 830 | destruct_txt += ' delete[] %s;\n' % member.name |
| 831 | elif member.isstaticarray or member.len is not None: |
| 832 | if member.len is None: |
| 833 | # Extract length of static array by grabbing val between [] |
| 834 | static_array_size = re.match(r"[^[]*\[([^]]*)\]", member.cdecl) |
| 835 | construct_txt += ' for (uint32_t i=0; i<%s; ++i) {\n' % static_array_size.group(1) |
| 836 | construct_txt += ' %s[i] = in_struct->%s[i];\n' % (member.name, member.name) |
| 837 | construct_txt += ' }\n' |
| 838 | else: |
| 839 | # Init array ptr to NULL |
| 840 | default_init_list += '\n %s(nullptr),' % member.name |
| 841 | init_list += '\n %s(nullptr),' % member.name |
| 842 | init_func_txt += ' %s = nullptr;\n' % member.name |
| 843 | array_element = 'in_struct->%s[i]' % member.name |
| 844 | if member.type in self.structNames: |
| 845 | member_index = next((i for i, v in enumerate(self.structMembers) if v[0] == member.type), None) |
| 846 | if member_index is not None and self.NeedSafeStruct(self.structMembers[member_index]) == True: |
| 847 | array_element = '%s(&in_struct->safe_%s[i])' % (member.type, member.name) |
| 848 | construct_txt += ' if (%s && in_struct->%s) {\n' % (member.len, member.name) |
| 849 | construct_txt += ' %s = new %s[%s];\n' % (member.name, m_type, member.len) |
| 850 | destruct_txt += ' if (%s)\n' % member.name |
| 851 | destruct_txt += ' delete[] %s;\n' % member.name |
| 852 | construct_txt += ' for (uint32_t i=0; i<%s; ++i) {\n' % (member.len) |
| 853 | if 'safe_' in m_type: |
| 854 | construct_txt += ' %s[i].initialize(&in_struct->%s[i]);\n' % (member.name, member.name) |
Mark Lobodzinski | e20e456 | 2017-01-03 11:14:26 -0700 | [diff] [blame] | 855 | else: |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 856 | construct_txt += ' %s[i] = %s;\n' % (member.name, array_element) |
| 857 | construct_txt += ' }\n' |
| 858 | construct_txt += ' }\n' |
| 859 | elif member.ispointer == True: |
| 860 | construct_txt += ' if (in_struct->%s)\n' % member.name |
| 861 | construct_txt += ' %s = new %s(in_struct->%s);\n' % (member.name, m_type, member.name) |
| 862 | construct_txt += ' else\n' |
| 863 | construct_txt += ' %s = NULL;\n' % member.name |
| 864 | destruct_txt += ' if (%s)\n' % member.name |
| 865 | destruct_txt += ' delete %s;\n' % member.name |
| 866 | elif 'safe_' in m_type: |
| 867 | init_list += '\n %s(&in_struct->%s),' % (member.name, member.name) |
| 868 | init_func_txt += ' %s.initialize(&in_struct->%s);\n' % (member.name, member.name) |
| 869 | else: |
| 870 | init_list += '\n %s(in_struct->%s),' % (member.name, member.name) |
| 871 | init_func_txt += ' %s = in_struct->%s;\n' % (member.name, member.name) |
| 872 | if '' != init_list: |
| 873 | init_list = init_list[:-1] # hack off final comma |
| 874 | if item.name in custom_construct_txt: |
| 875 | construct_txt = custom_construct_txt[item.name] |
Mike Schuchardt | 8148576 | 2017-09-04 11:38:42 -0600 | [diff] [blame] | 876 | if item.name in custom_destruct_txt: |
| 877 | destruct_txt = custom_destruct_txt[item.name] |
Mark Lobodzinski | 5cd0851 | 2017-09-12 09:50:25 -0600 | [diff] [blame] | 878 | 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)) |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 879 | if '' != default_init_list: |
| 880 | default_init_list = " :%s" % (default_init_list[:-1]) |
| 881 | safe_struct_body.append("\n%s::%s()%s\n{}" % (ss_name, ss_name, default_init_list)) |
| 882 | # Create slight variation of init and construct txt for copy constructor that takes a src object reference vs. struct ptr |
| 883 | copy_construct_init = init_func_txt.replace('in_struct->', 'src.') |
| 884 | copy_construct_txt = construct_txt.replace(' (in_struct->', ' (src.') # Exclude 'if' blocks from next line |
| 885 | copy_construct_txt = copy_construct_txt.replace('(in_struct->', '(*src.') # Pass object to copy constructors |
| 886 | copy_construct_txt = copy_construct_txt.replace('in_struct->', 'src.') # Modify remaining struct refs for src object |
Chris Forbes | fb63383 | 2017-10-03 18:11:54 -0700 | [diff] [blame^] | 887 | copy_assign_txt = ' if (&src == this) return *this;\n\n' + destruct_txt + '\n' + copy_construct_init + copy_construct_txt + '\n return *this;' |
Mark Lobodzinski | 5cd0851 | 2017-09-12 09:50:25 -0600 | [diff] [blame] | 888 | safe_struct_body.append("\n%s::%s(const %s& src)\n{\n%s%s}" % (ss_name, ss_name, ss_name, copy_construct_init, copy_construct_txt)) # Copy constructor |
Chris Forbes | fb63383 | 2017-10-03 18:11:54 -0700 | [diff] [blame^] | 889 | safe_struct_body.append("\n%s& %s::operator=(const %s& src)\n{\n%s\n}" % (ss_name, ss_name, ss_name, copy_assign_txt)) # Copy assignment operator |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 890 | safe_struct_body.append("\n%s::~%s()\n{\n%s}" % (ss_name, ss_name, destruct_txt)) |
Mark Lobodzinski | 5cd0851 | 2017-09-12 09:50:25 -0600 | [diff] [blame] | 891 | safe_struct_body.append("\nvoid %s::initialize(const %s* in_struct)\n{\n%s%s}" % (ss_name, item.name, init_func_txt, construct_txt)) |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 892 | # Copy initializer uses same txt as copy constructor but has a ptr and not a reference |
| 893 | init_copy = copy_construct_init.replace('src.', 'src->') |
| 894 | init_construct = copy_construct_txt.replace('src.', 'src->') |
Mark Lobodzinski | 5cd0851 | 2017-09-12 09:50:25 -0600 | [diff] [blame] | 895 | safe_struct_body.append("\nvoid %s::initialize(const %s* src)\n{\n%s%s}" % (ss_name, ss_name, init_copy, init_construct)) |
Mark Lobodzinski | e20e456 | 2017-01-03 11:14:26 -0700 | [diff] [blame] | 896 | if item.ifdef_protect != None: |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 897 | safe_struct_body.append("#endif // %s\n" % item.ifdef_protect) |
| 898 | return "\n".join(safe_struct_body) |
Mark Lobodzinski | e20e456 | 2017-01-03 11:14:26 -0700 | [diff] [blame] | 899 | # |
Mark Lobodzinski | 7ada59c | 2016-12-27 11:11:54 -0700 | [diff] [blame] | 900 | # Create a helper file and return it as a string |
| 901 | def OutputDestFile(self): |
Mark Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 902 | if self.helper_file_type == 'enum_string_header': |
Mark Lobodzinski | 1a2f1a3 | 2016-12-28 15:41:15 -0700 | [diff] [blame] | 903 | return self.GenerateEnumStringHelperHeader() |
Mark Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 904 | elif self.helper_file_type == 'struct_size_header': |
Mark Lobodzinski | 1a2f1a3 | 2016-12-28 15:41:15 -0700 | [diff] [blame] | 905 | return self.GenerateStructSizeHelperHeader() |
Mark Lobodzinski | 46d388f | 2016-12-28 10:46:26 -0700 | [diff] [blame] | 906 | elif self.helper_file_type == 'struct_size_source': |
Mark Lobodzinski | 1a2f1a3 | 2016-12-28 15:41:15 -0700 | [diff] [blame] | 907 | return self.GenerateStructSizeHelperSource() |
Mark Lobodzinski | e20e456 | 2017-01-03 11:14:26 -0700 | [diff] [blame] | 908 | elif self.helper_file_type == 'safe_struct_header': |
| 909 | return self.GenerateSafeStructHelperHeader() |
| 910 | elif self.helper_file_type == 'safe_struct_source': |
| 911 | return self.GenerateSafeStructHelperSource() |
Mark Lobodzinski | 7cb7da3 | 2017-04-03 16:58:04 -0600 | [diff] [blame] | 912 | elif self.helper_file_type == 'object_types_header': |
| 913 | return self.GenerateObjectTypesHelperHeader() |
Mark Lobodzinski | af86c38 | 2017-06-01 07:42:13 -0600 | [diff] [blame] | 914 | elif self.helper_file_type == 'extension_helper_header': |
| 915 | return self.GenerateExtensionHelperHeader() |
Mark Lobodzinski | 71f1ea1 | 2016-12-29 10:23:47 -0700 | [diff] [blame] | 916 | else: |
Mark Lobodzinski | c67efd0 | 2017-01-04 09:16:00 -0700 | [diff] [blame] | 917 | return 'Bad Helper File Generator Option %s' % self.helper_file_type |
Mark Lobodzinski | f36e58b | 2016-12-29 14:04:15 -0700 | [diff] [blame] | 918 | |