import renderdoc as rd
import rdtest
import os


class VK_RGP_Capture(rdtest.TestCase):
    demos_test_name = 'VK_Simple_Triangle'

    # Need to enable RGP mode before opening the capture
    def run(self):
        obj: rd.SDObject = rd.SetConfigSetting("AMD.RGP.Enable")
        if obj is not None:
            obj.data.basic.b = True
        super().run()

    def check_capture(self):
        apiprops: rd.APIProperties = self.controller.GetAPIProperties()

        if not apiprops.rgpCapture:
            rdtest.log.print("RGP capture not tested")
            return

        path = self.controller.CreateRGPProfile(rd.CreateHeadlessWindowingData(100, 100))

        if os.path.exists(path) and os.path.getsize(path) > 100:
            rdtest.log.success("RGP capture created successfully")
        else:
            raise rdtest.TestFailureException("RGP capture failed")