import renderdoc as rd import rdtest class VK_Parameter_Zoo(rdtest.TestCase): demos_test_name = 'VK_Parameter_Zoo' def check_capture(self): draw = self.find_draw("Color Draw") self.check(draw is not None) draw = draw.next self.controller.SetFrameEvent(draw.eventId, False) pipe: rd.PipeState = self.controller.GetPipelineState() self.check_pixel_value(pipe.GetOutputTargets()[0].resourceId, 0.5, 0.5, [0.0, 1.0, 0.0, 1.0]) # Readback reported extension support descriptor_update_template = self.find_draw("KHR_descriptor_update_template") is not None push_descriptor = self.find_draw("KHR_push_descriptor") is not None # Find the draw that contains resource references draw = self.find_draw("References") self.check(draw is not None) draw = draw.next self.controller.SetFrameEvent(draw.eventId, False) vkpipe: rd.VKState = self.controller.GetVulkanPipelineState() res_names = {} for res in self.controller.GetResources(): res: rd.ResourceDescription res_names[res.resourceId] = res.name expected_binds = [ {'sampname': 'refsamp'}, {'resname': 'refcombinedimg', 'sampname': 'refcombinedsamp'}, {'resname': 'refsampled'}, {'resname': 'refstorage'}, {'resname': 'refunitexel'}, {'resname': 'refstoretexel'}, {'resname': 'refunibuf'}, {'resname': 'refstorebuf'}, {'resname': 'refunibufdyn'}, {'resname': 'refstorebufdyn'}, ] setidx = 0 for descset in vkpipe.graphics.descriptorSets: descset: rd.VKDescriptorSet if setidx == 2 and not descset.pushDescriptor: raise rdtest.TestFailureException("Expected set {} to be a push set", setidx) if setidx != 2 and descset.pushDescriptor: raise rdtest.TestFailureException("Expected set {} to be a non-push set", setidx) for bindidx in range(len(descset.bindings)): sampname = res_names.get(descset.bindings[bindidx].binds[0].samplerResourceId, '!' + str(setidx)) resname = res_names.get(descset.bindings[bindidx].binds[0].resourceResourceId, '!' + str(setidx)) expected_samp = expected_binds[bindidx].get('sampname', '!') + str(setidx) expected_res = expected_binds[bindidx].get('resname', '!') + str(setidx) if not sampname == expected_samp: raise rdtest.TestFailureException( "Expected binding {} in set {} to have sampler {} but got {}".format( bindidx, setidx, expected_samp, sampname)) if not resname == expected_res: raise rdtest.TestFailureException( "Expected binding {} in set {} to have resource {} but got {}".format( bindidx, setidx, expected_res, resname)) rdtest.log.success("Resources in set {} were found as expected".format(setidx)) setidx = setidx + 1 # Since we can only have one push descriptor set we have a second draw for push AND template updates if descriptor_update_template and push_descriptor: draw = self.find_draw("PushTemplReferences") self.check(draw is not None) draw = draw.next self.controller.SetFrameEvent(draw.eventId, False) vkpipe: rd.VKState = self.controller.GetVulkanPipelineState() descset: rd.VKDescriptorSet = vkpipe.graphics.descriptorSets[2] if not descset.pushDescriptor: raise rdtest.TestFailureException("Expected set 2 to be a push set") for bindidx in range(len(descset.bindings)): sampname = res_names.get(descset.bindings[bindidx].binds[0].samplerResourceId, '!' + str(setidx)) resname = res_names.get(descset.bindings[bindidx].binds[0].resourceResourceId, '!' + str(setidx)) expected_samp = expected_binds[bindidx].get('sampname', '!') + str(setidx) expected_res = expected_binds[bindidx].get('resname', '!') + str(setidx) if not sampname == expected_samp: raise rdtest.TestFailureException( "Expected binding {} in set {} to have sampler {} but got {}".format( bindidx, setidx, expected_samp, sampname)) if not resname == expected_res: raise rdtest.TestFailureException( "Expected binding {} in set {} to have resource {} but got {}".format( bindidx, setidx, expected_res, resname)) rdtest.log.success("Resources in push template set were found as expected") rdtest.log.success("All resources were found as expected") draw = self.find_draw("Tools available") self.check(len(draw.children) > 1) self.check(any([d.name == 'RenderDoc' for d in draw.children])) rdtest.log.success("RenderDoc tool was listed as available") draw = self.find_draw("ASM Draw") self.check(draw is not None) draw = draw.next self.controller.SetFrameEvent(draw.eventId, False) vkpipe: rd.VKState = self.controller.GetVulkanPipelineState() desc_set: rd.VKDescriptorSet = vkpipe.graphics.descriptorSets[0] self.check(len(desc_set.bindings) == 11) binding = desc_set.bindings[10] self.check(binding.dynamicallyUsedCount == 1) self.check(len(binding.binds) == 4) self.check(not binding.binds[0].dynamicallyUsed) self.check(binding.binds[1].dynamicallyUsed) self.check(not binding.binds[2].dynamicallyUsed) self.check(not binding.binds[3].dynamicallyUsed) self.check(len(vkpipe.viewportScissor.viewportScissors) == 0) postvs_data = self.get_postvs(draw, rd.MeshDataStage.VSOut, 0, draw.numIndices) postvs_ref = { 0: { 'vtx': 0, 'idx': 0, '_Position': [-1.0, 1.0, 0.0, 1.0], }, 1: { 'vtx': 1, 'idx': 1, '_Position': [ 1.0, 1.0, 0.0, 1.0], }, 2: { 'vtx': 2, 'idx': 2, '_Position': [-1.0, -1.0, 0.0, 1.0], }, 3: { 'vtx': 3, 'idx': 3, '_Position': [ 1.0, -1.0, 0.0, 1.0], }, } self.check_mesh_data(postvs_ref, postvs_data) rdtest.log.success("ASM Draw is as expected") draw = self.find_draw("Immutable Draw") self.check(draw is not None) draw = draw.next self.controller.SetFrameEvent(draw.eventId, False) vkpipe: rd.VKState = self.controller.GetVulkanPipelineState() desc_set: rd.VKDescriptorSet = vkpipe.graphics.descriptorSets[0] if desc_set.bindings[0].binds[0].filter.minify != rd.FilterMode.Linear: raise rdtest.TestFailureException( "Expected linear sampler at binding slot 0 in immutable draw") if self.get_resource(desc_set.bindings[0].binds[0].samplerResourceId).name != "validSampler": raise rdtest.TestFailureException( "Expected validSampler to be at binding slot 0 in immutable draw") # Check for resource leaks if len(self.controller.GetStructuredFile().chunks) > 500: raise rdtest.TestFailureException( "Too many chunks found: {}".format(len(self.controller.GetStructuredFile().chunks)))