mirror of
https://github.com/baldurk/renderdoc.git
synced 2026-09-17 11:15:39 +00:00
baldurk
12636a1607
* These were available for python 3.6 too so was never any reason not to use them except not being aware they existed.
113 lines
5.5 KiB
Python
113 lines
5.5 KiB
Python
import rdtest
|
|
import renderdoc as rd
|
|
|
|
|
|
class VK_Image_Layouts(rdtest.TestCase):
|
|
demos_test_name = 'VK_Image_Layouts'
|
|
|
|
def get_capture_options(self):
|
|
opts = rd.CaptureOptions()
|
|
|
|
# Ref all resources to pull in the image with unbound data
|
|
opts.refAllResources = True
|
|
|
|
return opts
|
|
|
|
def check_capture(self):
|
|
self.controller.SetFrameEvent(0, False)
|
|
|
|
pipe = self.controller.GetVulkanPipelineState()
|
|
|
|
# Check that the layout is reported correctly at the start of the frame
|
|
for img in pipe.images:
|
|
res = self.get_resource(img.resourceId)
|
|
if res.name == "Image:Preinitialised":
|
|
if img.layouts[0].name != "VK_IMAGE_LAYOUT_PREINITIALIZED":
|
|
raise rdtest.TestFailureException(f"Pre-initialised image is in {img.layouts[0].name} layout")
|
|
elif res.name == "Image:Undefined":
|
|
if img.layouts[0].name != "VK_IMAGE_LAYOUT_UNDEFINED":
|
|
raise rdtest.TestFailureException(f"Undefined image is in {img.layouts[0].name} layout")
|
|
elif res.name == "Image:Swapchain":
|
|
if img.layouts[0].name != "VK_IMAGE_LAYOUT_PRESENT_SRC_KHR":
|
|
raise rdtest.TestFailureException(f"Swapchain image is in {img.layouts[0].name} layout")
|
|
|
|
action = self.find_action("Before Transition")
|
|
|
|
assert action is not None
|
|
|
|
self.controller.SetFrameEvent(action.eventId, False)
|
|
|
|
pipe = self.controller.GetVulkanPipelineState()
|
|
|
|
# Check that the layout is reported correctly before transitions still
|
|
for img in pipe.images:
|
|
res = self.get_resource(img.resourceId)
|
|
if res.name == "Image:Preinitialised":
|
|
if img.layouts[0].name != "VK_IMAGE_LAYOUT_PREINITIALIZED":
|
|
raise rdtest.TestFailureException(f"Pre-initialised image is in {img.layouts[0].name} layout")
|
|
# check the preinitialised image has the right data
|
|
self.check_pixel_value(img.resourceId, 0, 0, [0.25, 0.25, 0.25, 0.25], eps=0.01)
|
|
elif res.name == "Image:Undefined":
|
|
if img.layouts[0].name != "VK_IMAGE_LAYOUT_UNDEFINED":
|
|
raise rdtest.TestFailureException(f"Undefined image is in {img.layouts[0].name} layout")
|
|
# don't check undefined image contents
|
|
elif res.name == "Image:Swapchain":
|
|
if img.layouts[0].name != "VK_IMAGE_LAYOUT_PRESENT_SRC_KHR":
|
|
raise rdtest.TestFailureException(f"Swapchain image is in {img.layouts[0].name} layout")
|
|
|
|
action = self.find_action("vkCmdDraw")
|
|
|
|
assert action is not None
|
|
|
|
self.controller.SetFrameEvent(action.eventId, False)
|
|
|
|
pipe = self.controller.GetVulkanPipelineState()
|
|
|
|
# Check that after transitions, the images are in the right state
|
|
for img in pipe.images:
|
|
res = self.get_resource(img.resourceId)
|
|
if res.name == "Image:Preinitialised":
|
|
if img.layouts[0].name != "VK_IMAGE_LAYOUT_TRANSFER_SRC_OPTIMAL":
|
|
raise rdtest.TestFailureException(f"Pre-initialised image is in {img.layouts[0].name} layout")
|
|
# check the preinitialised image has the right data
|
|
self.check_pixel_value(img.resourceId, 0, 0, [0.25, 0.25, 0.25, 0.25], eps=0.01)
|
|
elif res.name == "Image:Undefined":
|
|
if img.layouts[0].name != "VK_IMAGE_LAYOUT_TRANSFER_DST_OPTIMAL":
|
|
raise rdtest.TestFailureException(f"Undefined image is in {img.layouts[0].name} layout")
|
|
# check the preinitialised image data was copied correctly to the previously undefined image
|
|
self.check_pixel_value(img.resourceId, 0, 0, [0.25, 0.25, 0.25, 0.25], eps=0.01)
|
|
elif img.resourceId == pipe.currentPass.framebuffer.attachments[0].resource:
|
|
if img.layouts[0].name != "VK_IMAGE_LAYOUT_GENERAL":
|
|
raise rdtest.TestFailureException(f"Rendered swapchain image is in {img.layouts[0].name} layout")
|
|
|
|
action = self.find_action("Preinit clear")
|
|
|
|
assert action is not None
|
|
|
|
self.controller.SetFrameEvent(action.eventId+1, False)
|
|
|
|
pipe = self.controller.GetVulkanPipelineState()
|
|
|
|
# check that the pre-initialised image has transitioned + been cleared
|
|
for img in pipe.images:
|
|
res = self.get_resource(img.resourceId)
|
|
if res.name == "Image:Preinitialised":
|
|
if img.layouts[0].name != "VK_IMAGE_LAYOUT_GENERAL":
|
|
raise rdtest.TestFailureException(f"Pre-initialised image is in {img.layouts[0].name} layout")
|
|
# check the preinitialised image has the right data
|
|
self.check_pixel_value(img.resourceId, 0, 0, [0.8, 0.8, 0.8, 1.0], eps=0.01)
|
|
|
|
# finally check that it has reset to the correct value back at the start of the frame
|
|
self.controller.SetFrameEvent(1, False)
|
|
|
|
pipe = self.controller.GetVulkanPipelineState()
|
|
|
|
for img in pipe.images:
|
|
res = self.get_resource(img.resourceId)
|
|
if res.name == "Image:Preinitialised":
|
|
if img.layouts[0].name != "VK_IMAGE_LAYOUT_PREINITIALIZED":
|
|
raise rdtest.TestFailureException(f"Pre-initialised image is in {img.layouts[0].name} layout")
|
|
# check the preinitialised image has the right data
|
|
self.check_pixel_value(img.resourceId, 0, 0, [0.25, 0.25, 0.25, 0.25], eps=0.01)
|
|
|