tests: brxalloc: Add BRU/BRS dynamic allocation test

Signed-off-by: Laurent Pinchart <laurent.pinchart@ideasonboard.com>

1 files changed, 132 insertions, 0 deletions

diff --git a/tests/kms-test-brxalloc.py b/tests/kms-test-brxalloc.py
new file mode 100755
index 0000000..2e3f6cb
--- /dev/null
+++ b/tests/kms-test-brxalloc.py
@@ -0,0 +1,132 @@
+#!/usr/bin/python3
+
+import kmstest
+import pykms
+import time
+
+class Output(object):
+    def __init__(self):
+        self.crtc = None
+        self.connector = None
+        self.fb = None
+
+
+class BRxAllocTest(kmstest.KMSTest):
+    """Test dynamic BRU and BRS allocation in the VSP."""
+
+    def main(self):
+        # This test requires usage of two CRTCs connected to the same VSPDL
+        # instance to test dynamic assignment of the BRU and BRS to pipelines.
+        # This is only occurs on H3 ES2.0 (and M3N which we don't support yet).
+        # Check the SoC model through sysfs as we can't detected it through the
+        # DRM/KMS API.
+        soc = open("/sys/devices/soc0/soc_id", "rb").read().strip().decode()
+        if soc != "r8a7795":
+            self.skip("VSPDL (BRU+BRS) not available")
+            return
+
+        outputs = [Output(), Output()]
+
+        # Get the first and last CRTCs
+        for crtc in self.card.crtcs:
+            if outputs[0].crtc == None or crtc.id < outputs[0].crtc.id:
+                outputs[0].crtc = crtc
+            if outputs[1].crtc == None or crtc.id > outputs[1].crtc.id:
+                outputs[1].crtc = crtc
+
+        # Verify that the two CRTCs share the same planes
+        planes = outputs[0].crtc.possible_planes
+        if planes != outputs[1].crtc.possible_planes:
+            self.skip("Planes differ for CRTCs %u and %u" % \
+                      (outputs[0].crtc.id, outputs[1].crtc.id))
+            return
+
+        self.logger.log("Found %u planes for CRTCs %u and %u" % \
+                        (len(planes), outputs[0].crtc.id, outputs[1].crtc.id))
+
+        # Get one connector for each CRTC
+        for connector in self.card.connectors:
+            # Skip disconnected connectors
+            if not connector.connected():
+                continue
+
+            for crtc in connector.get_possible_crtcs():
+                if crtc == outputs[0].crtc:
+                    outputs[0].connector = connector
+                if crtc == outputs[1].crtc:
+                    outputs[1].connector = connector
+
+        if not outputs[0].connector or not outputs[1].connector:
+            self.skip("No connected connectors for CRTCs %u and %u " % \
+                      (outputs[0].crtc.id, outputs[1].crtc.id))
+            return
+
+        self.start("Moving %u planes from %s to %s" % \
+                   (len(planes), outputs[0].connector.fullname, outputs[1].connector.fullname))
+
+        # Set the initial mode for both outputs and wait 5s for the monitors to
+        # wake up.
+        for output in outputs:
+            # Get the default mode and create a framebuffer
+            mode = output.connector.get_default_mode()
+            output.fb = pykms.DumbFramebuffer(self.card, mode.hdisplay, mode.vdisplay, "XR24")
+            pykms.draw_test_pattern(output.fb)
+
+            # Set the mode with no plane
+            ret = self.atomic_crtc_mode_set(output.crtc, output.connector, mode, sync=True)
+            if ret < 0:
+                self.fail("atomic mode set on %s failed with %d" % \
+                          (output.connector.fullname, ret))
+                return
+
+        self.logger.log("Initial atomic mode set completed")
+        time.sleep(5)
+
+        # Add all planes
+        offset = 0
+        output = outputs[0]
+
+        for plane in planes:
+            self.logger.log("Adding plane %u to %s" % (plane.id, output.connector.fullname))
+
+            source = kmstest.Rect(0, 0, output.fb.width, output.fb.height)
+            destination = kmstest.Rect(offset, offset, output.fb.width, output.fb.height)
+            ret = self.atomic_plane_set(plane, output.crtc, source, destination, output.fb, sync=True)
+            if ret < 0:
+                self.fail("atomic plane set failed with %d" % ret)
+                return
+
+            offset += 50
+
+        time.sleep(2)
+
+        # Move all planes one by one
+        offset = 0
+        output = outputs[1]
+
+        for plane in planes:
+            self.logger.log("Moving plane %u to %s" % (plane.id, output.connector.fullname))
+
+            # Switching CRTC directly is not supported by DRM, start by
+            # disabling the plane.
+            ret = self.atomic_plane_disable(plane)
+            if ret < 0:
+                self.fail("atomic plane disable failed with %d" % ret)
+                return
+
+            source = kmstest.Rect(0, 0, output.fb.width, output.fb.height)
+            destination = kmstest.Rect(offset, offset, output.fb.width, output.fb.height)
+            ret = self.atomic_plane_set(plane, output.crtc, source, destination, output.fb)
+            if ret < 0:
+                self.fail("atomic plane set failed with %d" % ret)
+                return
+
+            self.run(1)
+            offset += 50
+
+        self.success()
+
+        self.atomic_crtc_disable(outputs[0].crtc)
+        self.atomic_crtc_disable(outputs[1].crtc)
+
+BRxAllocTest().execute()