265 lines
9.2 KiB
Python
265 lines
9.2 KiB
Python
# Copyright (c) 2018 The Android Open Source Project
|
|
# Copyright (c) 2018 Google Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from .common.codegen import CodeGen
|
|
from .common.vulkantypes import \
|
|
VulkanAPI, makeVulkanTypeSimple, iterateVulkanType, VulkanTypeIterator
|
|
|
|
from .wrapperdefs import VulkanWrapperGenerator
|
|
from .wrapperdefs import STRUCT_EXTENSION_PARAM, STRUCT_EXTENSION_PARAM_FOR_WRITE
|
|
|
|
class HandleMapCodegen(VulkanTypeIterator):
|
|
def __init__(self, cgen, inputVar, handlemapVarName, prefix, isHandleFunc):
|
|
self.cgen = cgen
|
|
self.inputVar = inputVar
|
|
self.prefix = prefix
|
|
self.handlemapVarName = handlemapVarName
|
|
|
|
def makeAccess(varName, asPtr = True):
|
|
return lambda t: self.cgen.generalAccess(t, parentVarName = varName, asPtr = asPtr)
|
|
|
|
def makeLengthAccess(varName):
|
|
return lambda t: self.cgen.generalLengthAccess(t, parentVarName = varName)
|
|
|
|
def makeLengthAccessGuard(varName):
|
|
return lambda t: self.cgen.generalLengthAccessGuard(t, parentVarName=varName)
|
|
|
|
self.exprAccessor = makeAccess(self.inputVar)
|
|
self.exprAccessorValue = makeAccess(self.inputVar, asPtr = False)
|
|
self.lenAccessor = makeLengthAccess(self.inputVar)
|
|
self.lenAccessorGuard = makeLengthAccessGuard(self.inputVar)
|
|
|
|
self.checked = False
|
|
self.isHandleFunc = isHandleFunc
|
|
|
|
def needSkip(self, vulkanType):
|
|
return False
|
|
|
|
def makeCastExpr(self, vulkanType):
|
|
return "(%s)" % (
|
|
self.cgen.makeCTypeDecl(vulkanType, useParamName=False))
|
|
|
|
def asNonConstCast(self, access, vulkanType):
|
|
if vulkanType.staticArrExpr:
|
|
casted = "%s(%s)" % (self.makeCastExpr(vulkanType.getForAddressAccess().getForNonConstAccess()), access)
|
|
elif vulkanType.accessibleAsPointer():
|
|
casted = "%s(%s)" % (self.makeCastExpr(vulkanType.getForNonConstAccess()), access)
|
|
else:
|
|
casted = "%s(%s)" % (self.makeCastExpr(vulkanType.getForAddressAccess().getForNonConstAccess()), access)
|
|
return casted
|
|
|
|
def onCheck(self, vulkanType):
|
|
pass
|
|
|
|
def endCheck(self, vulkanType):
|
|
pass
|
|
|
|
def onCompoundType(self, vulkanType):
|
|
|
|
if self.needSkip(vulkanType):
|
|
self.cgen.line("// TODO: Unsupported : %s" %
|
|
self.cgen.makeCTypeDecl(vulkanType))
|
|
return
|
|
access = self.exprAccessor(vulkanType)
|
|
lenAccess = self.lenAccessor(vulkanType)
|
|
lenAccessGuard = self.lenAccessorGuard(vulkanType)
|
|
|
|
isPtr = vulkanType.pointerIndirectionLevels > 0
|
|
|
|
if lenAccessGuard is not None:
|
|
self.cgen.beginIf(lenAccessGuard)
|
|
|
|
if isPtr:
|
|
self.cgen.beginIf(access)
|
|
|
|
if lenAccess is not None:
|
|
|
|
loopVar = "i"
|
|
access = "%s + %s" % (access, loopVar)
|
|
forInit = "uint32_t %s = 0" % loopVar
|
|
forCond = "%s < (uint32_t)%s" % (loopVar, lenAccess)
|
|
forIncr = "++%s" % loopVar
|
|
|
|
self.cgen.beginFor(forInit, forCond, forIncr)
|
|
|
|
accessCasted = self.asNonConstCast(access, vulkanType)
|
|
self.cgen.funcCall(None, self.prefix + vulkanType.typeName,
|
|
[self.handlemapVarName, accessCasted])
|
|
|
|
if lenAccess is not None:
|
|
self.cgen.endFor()
|
|
|
|
if isPtr:
|
|
self.cgen.endIf()
|
|
|
|
if lenAccessGuard is not None:
|
|
self.cgen.endIf()
|
|
|
|
def onString(self, vulkanType):
|
|
pass
|
|
|
|
def onStringArray(self, vulkanType):
|
|
pass
|
|
|
|
def onStaticArr(self, vulkanType):
|
|
if not self.isHandleFunc(vulkanType):
|
|
return
|
|
|
|
accessLhs = self.exprAccessor(vulkanType)
|
|
lenAccess = self.lenAccessor(vulkanType)
|
|
|
|
self.cgen.stmt("%s->mapHandles_%s(%s%s, %s)" % \
|
|
(self.handlemapVarName, vulkanType.typeName,
|
|
self.makeCastExpr(vulkanType.getForAddressAccess().getForNonConstAccess()),
|
|
accessLhs, lenAccess))
|
|
|
|
def onStructExtension(self, vulkanType):
|
|
access = self.exprAccessor(vulkanType)
|
|
|
|
castedAccessExpr = "(%s)(%s)" % ("void*", access)
|
|
self.cgen.beginIf(access)
|
|
self.cgen.funcCall(None, self.prefix + "extension_struct",
|
|
[self.handlemapVarName, castedAccessExpr])
|
|
self.cgen.endIf()
|
|
|
|
def onPointer(self, vulkanType):
|
|
if self.needSkip(vulkanType):
|
|
return
|
|
|
|
if not self.isHandleFunc(vulkanType):
|
|
return
|
|
|
|
access = self.exprAccessor(vulkanType)
|
|
lenAccess = self.lenAccessor(vulkanType)
|
|
lenAccess = "1" if lenAccess is None else lenAccess
|
|
|
|
self.cgen.beginIf(access)
|
|
|
|
self.cgen.stmt( \
|
|
"%s->mapHandles_%s(%s%s, %s)" % \
|
|
(self.handlemapVarName,
|
|
vulkanType.typeName,
|
|
self.makeCastExpr(vulkanType.getForNonConstAccess()),
|
|
access,
|
|
lenAccess))
|
|
|
|
self.cgen.endIf()
|
|
|
|
def onValue(self, vulkanType):
|
|
if not self.isHandleFunc(vulkanType):
|
|
return
|
|
access = self.exprAccessor(vulkanType)
|
|
self.cgen.stmt(
|
|
"%s->mapHandles_%s(%s%s)" % \
|
|
(self.handlemapVarName, vulkanType.typeName,
|
|
self.makeCastExpr(vulkanType.getForAddressAccess().getForNonConstAccess()),
|
|
access))
|
|
|
|
class VulkanHandleMap(VulkanWrapperGenerator):
|
|
def __init__(self, module, typeInfo):
|
|
VulkanWrapperGenerator.__init__(self, module, typeInfo)
|
|
|
|
self.codegen = CodeGen()
|
|
|
|
self.handlemapPrefix = "handlemap_"
|
|
self.toMapVar = "toMap"
|
|
self.handlemapVarName = "handlemap"
|
|
self.handlemapParam = \
|
|
makeVulkanTypeSimple(False, "VulkanHandleMapping", 1,
|
|
self.handlemapVarName)
|
|
self.voidType = makeVulkanTypeSimple(False, "void", 0)
|
|
|
|
self.handlemapCodegen = \
|
|
HandleMapCodegen(
|
|
None,
|
|
self.toMapVar,
|
|
self.handlemapVarName,
|
|
self.handlemapPrefix,
|
|
lambda vtype : typeInfo.isHandleType(vtype.typeName))
|
|
|
|
self.knownDefs = {}
|
|
|
|
self.extensionHandlemapPrototype = \
|
|
VulkanAPI(self.handlemapPrefix + "extension_struct",
|
|
self.voidType,
|
|
[self.handlemapParam, STRUCT_EXTENSION_PARAM_FOR_WRITE])
|
|
|
|
def onBegin(self,):
|
|
VulkanWrapperGenerator.onBegin(self)
|
|
self.module.appendImpl(self.codegen.makeFuncDecl(
|
|
self.extensionHandlemapPrototype))
|
|
|
|
def onGenType(self, typeXml, name, alias):
|
|
VulkanWrapperGenerator.onGenType(self, typeXml, name, alias)
|
|
|
|
if name in self.knownDefs:
|
|
return
|
|
|
|
category = self.typeInfo.categoryOf(name)
|
|
|
|
if category in ["struct", "union"] and alias:
|
|
self.module.appendHeader(
|
|
self.codegen.makeFuncAlias(self.handlemapPrefix + name,
|
|
self.handlemapPrefix + alias))
|
|
|
|
if category in ["struct", "union"] and not alias:
|
|
|
|
structInfo = self.typeInfo.structs[name]
|
|
|
|
typeFromName = \
|
|
lambda varname: \
|
|
makeVulkanTypeSimple(varname == "from", name, 1, varname)
|
|
|
|
handlemapParams = \
|
|
[self.handlemapParam] + \
|
|
list(map(typeFromName, [self.toMapVar]))
|
|
|
|
handlemapPrototype = \
|
|
VulkanAPI(self.handlemapPrefix + name,
|
|
self.voidType,
|
|
handlemapParams)
|
|
|
|
def funcDefGenerator(cgen):
|
|
self.handlemapCodegen.cgen = cgen
|
|
for p in handlemapParams:
|
|
cgen.stmt("(void)%s" % p.paramName)
|
|
for member in structInfo.members:
|
|
iterateVulkanType(self.typeInfo, member,
|
|
self.handlemapCodegen)
|
|
|
|
self.module.appendHeader(
|
|
self.codegen.makeFuncDecl(handlemapPrototype))
|
|
self.module.appendImpl(
|
|
self.codegen.makeFuncImpl(handlemapPrototype, funcDefGenerator))
|
|
|
|
def onGenCmd(self, cmdinfo, name, alias):
|
|
VulkanWrapperGenerator.onGenCmd(self, cmdinfo, name, alias)
|
|
|
|
def onEnd(self,):
|
|
VulkanWrapperGenerator.onEnd(self)
|
|
|
|
def forEachExtensionHandlemap(ext, castedAccess, cgen):
|
|
cgen.funcCall(None, self.handlemapPrefix + ext.name,
|
|
[self.handlemapVarName, castedAccess])
|
|
|
|
self.module.appendImpl(
|
|
self.codegen.makeFuncImpl(
|
|
self.extensionHandlemapPrototype,
|
|
lambda cgen: self.emitForEachStructExtension(
|
|
cgen,
|
|
self.voidType,
|
|
STRUCT_EXTENSION_PARAM_FOR_WRITE,
|
|
forEachExtensionHandlemap)))
|