mirror of
https://github.com/baldurk/renderdoc.git
synced 2026-05-06 10:00:40 +00:00
baldurk
145 lines
7.0 KiB
Python
145 lines
7.0 KiB
Python
import renderdoc as rd
|
|
import rdtest
|
|
import struct
|
|
|
|
|
|
class VK_Spec_Constants(rdtest.TestCase):
|
|
demos_test_name = 'VK_Spec_Constants'
|
|
|
|
def check_capture(self):
|
|
# find the first action
|
|
action = self.find_action("Draw")
|
|
|
|
# We should have 4 actions, with spec constant values 0, 1, 2, 3
|
|
for num_colors in range(4):
|
|
self.check(action is not None)
|
|
|
|
self.controller.SetFrameEvent(action.eventId, False)
|
|
|
|
pipe: rd.PipeState = self.controller.GetPipelineState()
|
|
vkpipe = self.controller.GetVulkanPipelineState()
|
|
|
|
shader: rd.ShaderReflection = pipe.GetShaderReflection(rd.ShaderStage.Pixel)
|
|
|
|
# uniform buffer and spec constants
|
|
self.check(len(shader.constantBlocks) == 2)
|
|
self.check(shader.constantBlocks[0].bufferBacked)
|
|
self.check(not shader.constantBlocks[1].bufferBacked)
|
|
self.check(len(shader.constantBlocks[1].variables) == 3)
|
|
|
|
# should be an array of num_colors+1 elements
|
|
array_len = shader.constantBlocks[0].variables[0].type.elements
|
|
if not rdtest.value_compare(array_len, num_colors+1):
|
|
raise rdtest.TestFailureException("CBuffer variable is array of {}, not {}".format(array_len, num_colors+1))
|
|
|
|
if num_colors > 0:
|
|
cbuf = pipe.GetConstantBlock(rd.ShaderStage.Pixel, 0, 0).descriptor
|
|
|
|
cb_vars = self.controller.GetCBufferVariableContents(pipe.GetGraphicsPipelineObject(),
|
|
pipe.GetShader(rd.ShaderStage.Pixel), rd.ShaderStage.Pixel,
|
|
pipe.GetShaderEntryPoint(rd.ShaderStage.Pixel), 0,
|
|
cbuf.resource, cbuf.byteOffset, cbuf.byteSize)
|
|
|
|
self.check(len(cb_vars) == 1)
|
|
|
|
if not rdtest.value_compare(len(cb_vars[0].members), num_colors+1):
|
|
raise rdtest.TestFailureException("CBuffer variable is array of {}, not {}".format(len(cb_vars[0].members), num_colors+1))
|
|
|
|
for col in range(num_colors):
|
|
expected = [0.0, 0.0, 0.0, 0.0]
|
|
expected[col] = 1.0
|
|
|
|
val = [i for i in cb_vars[0].members[col].value.f32v[0:4]]
|
|
|
|
if not rdtest.value_compare(val, expected):
|
|
raise rdtest.TestFailureException("Cbuffer[{}] value {} doesn't match expectation {}".format(col, val, expected))
|
|
|
|
rdtest.log.success("Draw with {} colors uniform buffer is as expected".format(num_colors))
|
|
|
|
cbuf = pipe.GetConstantBlock(rd.ShaderStage.Pixel, 1, 0).descriptor
|
|
|
|
cb_vars = self.controller.GetCBufferVariableContents(pipe.GetGraphicsPipelineObject(),
|
|
pipe.GetShader(rd.ShaderStage.Pixel), rd.ShaderStage.Pixel,
|
|
pipe.GetShaderEntryPoint(rd.ShaderStage.Pixel), 1,
|
|
cbuf.resource, cbuf.byteOffset, cbuf.byteSize)
|
|
|
|
self.check(len(cb_vars) == 3)
|
|
|
|
refl = pipe.GetShaderReflection(rd.ShaderStage.Pixel)
|
|
|
|
for cb_var in cb_vars:
|
|
refl_var = None
|
|
|
|
for cb in refl.constantBlocks:
|
|
if not cb.bufferBacked:
|
|
for cbv in cb.variables:
|
|
if cbv.name == cb_var.name:
|
|
refl_var = cbv
|
|
break
|
|
break
|
|
|
|
if cb_var.name == "numcols":
|
|
if not rdtest.value_compare(cb_var.value.s32v[0], num_colors):
|
|
raise rdtest.TestFailureException(
|
|
"{} spec constant is {}, not {}".format(cb_var.name, cb_var.value.s32v[0], num_colors))
|
|
|
|
read = struct.unpack_from("L", vkpipe.fragmentShader.specializationData, refl_var.byteOffset)[0]
|
|
|
|
if not rdtest.value_compare(read, num_colors):
|
|
raise rdtest.TestFailureException(
|
|
"{} spec constant read manually is {}, not {}".format(cb_var.name, read, num_colors))
|
|
|
|
rdtest.log.success("Draw with {} colors constant {} is as expected".format(num_colors, cb_var.name))
|
|
elif cb_var.name == "NOT_numcols":
|
|
expected = 999
|
|
if num_colors == 2:
|
|
expected = 9999
|
|
|
|
if not rdtest.value_compare(cb_var.value.s32v[0], expected):
|
|
raise rdtest.TestFailureException(
|
|
"{} spec constant is {}, not {}".format(cb_var.name, cb_var.value.s32v[0], expected))
|
|
|
|
read = struct.unpack_from("L", vkpipe.fragmentShader.specializationData, refl_var.byteOffset)[0]
|
|
|
|
if not rdtest.value_compare(read, expected):
|
|
raise rdtest.TestFailureException(
|
|
"{} spec constant read manually is {}, not {}".format(cb_var.name, read, expected))
|
|
|
|
rdtest.log.success("Draw with {} colors constant {} is as expected".format(num_colors, cb_var.name))
|
|
elif cb_var.name == "some_float":
|
|
expected = 1.5
|
|
if num_colors == 1:
|
|
expected = 2.5
|
|
if num_colors == 2:
|
|
expected = 16.5
|
|
|
|
if not rdtest.value_compare(cb_var.value.f32v[0], expected):
|
|
raise rdtest.TestFailureException(
|
|
"{} spec constant is {}, not {}".format(cb_var.name, cb_var.value.f32v[0], expected))
|
|
|
|
read = struct.unpack_from("f", vkpipe.fragmentShader.specializationData, refl_var.byteOffset)[0]
|
|
|
|
if not rdtest.value_compare(read, expected):
|
|
raise rdtest.TestFailureException(
|
|
"{} spec constant read manually is {}, not {}".format(cb_var.name, read, expected))
|
|
|
|
rdtest.log.success("Draw with {} colors constant {} is as expected".format(num_colors, cb_var.name))
|
|
else:
|
|
raise rdtest.TestFailureException("Spec constant {} is unexpected".format(cb_var.name))
|
|
|
|
rdtest.log.success("Draw with {} colors specialisation constant is as expected".format(num_colors))
|
|
|
|
view = pipe.GetViewport(0)
|
|
|
|
# the first num_colors components should be 0.6, the rest should be 0.1 (alpha is always 1.0)
|
|
expected = [0.0, 0.0, 0.0, 1.0]
|
|
for col in range(num_colors):
|
|
expected[col] += 1.0
|
|
|
|
# Sample the centre of the viewport
|
|
self.check_pixel_value(pipe.GetOutputTargets()[0].resource, int(view.x) + int(view.width / 2), int(view.height / 2), expected)
|
|
|
|
rdtest.log.success("Draw with {} colors picked value is as expected".format(num_colors))
|
|
|
|
action = action.next
|