summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xtests/kms-test-routing.py148
1 files changed, 148 insertions, 0 deletions
diff --git a/tests/kms-test-routing.py b/tests/kms-test-routing.py
new file mode 100755
index 0000000..2cf02dd
--- /dev/null
+++ b/tests/kms-test-routing.py
@@ -0,0 +1,148 @@
+#!/usr/bin/python3
+
+import kmstest
+import pykms
+import time
+
+class Pipeline(object):
+ def __init__(self, crtc):
+ self.crtc = crtc
+ self.connector = None
+ self.plane = None
+ self.mode_blob = None
+
+
+class RoutingTest(kmstest.KMSTest):
+ """Test output routing."""
+
+ def main(self):
+
+ # Create the reverse map from CRTC to possible connectors and calculate
+ # the largest resolution.
+ self.crtc_to_connectors = {}
+ max_hdisplay = 0
+ max_vdisplay = 0
+
+ for connector in self.card.connectors:
+ if connector.fullname.startswith('writeback-'):
+ continue
+
+ mode = connector.get_default_mode()
+ max_hdisplay = max(mode.hdisplay, max_hdisplay)
+ max_vdisplay = max(mode.vdisplay, max_vdisplay)
+
+ for crtc in connector.get_possible_crtcs():
+ if not crtc in self.crtc_to_connectors:
+ self.crtc_to_connectors[crtc] = []
+ self.crtc_to_connectors[crtc].append(connector)
+
+ # Find a connector that can be routed to at least two CRTCs that have
+ # at least two output routes each.
+ shared_connector = None
+ for connector in self.card.connectors:
+ if connector.fullname.startswith('writeback-'):
+ continue
+
+ pipes = []
+ for crtc in connector.get_possible_crtcs():
+ if len(self.crtc_to_connectors[crtc]) >= 2:
+ pipes.append(Pipeline(crtc))
+
+ if len(pipes) >= 2:
+ shared_connector = connector
+ break
+
+ if not shared_connector:
+ self.skip("No suitable connector")
+ return
+
+ # Allocate planes for each CRTC.
+ pool = [(pipe, list(pipe.crtc.possible_planes)) for pipe in pipes]
+ while len(pool):
+ pool.sort(key=lambda elem: len(elem[1]), reverse=True)
+ pipe, planes = pool[-1]
+ pipe.plane = planes[0]
+ pool = [(elem[0], [p for p in elem[1] if p != pipe.plane]) for elem in pool[:-1]]
+
+ # Create a framebuffer big enough for all connectors.
+ fb = pykms.DumbFramebuffer(self.card, max_hdisplay, max_vdisplay, "XR24")
+ pykms.draw_test_pattern(fb)
+
+ self.start("Moving connector %s between CRTCs %s" % \
+ (shared_connector.fullname, [pipe.crtc.id for pipe in pipes]))
+
+ self.logger.log("Highest display resolution: %ux%u" % (max_hdisplay, max_vdisplay))
+
+ for master_pipe in pipes:
+ req = kmstest.AtomicRequest(self)
+ connectors = self.allocate_connectors(pipes, master_pipe, shared_connector)
+ route = []
+
+ for pipe in pipes:
+ if pipe.connector and not pipe.connector in connectors.values():
+ req.add(pipe.connector, 'CRTC_ID', 0)
+
+ pipe.connector = connectors[pipe.crtc]
+ mode = pipe.connector.get_default_mode()
+ pipe.mode_blob = mode.to_blob(self.card)
+
+ req.add(pipe.connector, 'CRTC_ID', pipe.crtc.id)
+ req.add(pipe.crtc, {'ACTIVE': 1, 'MODE_ID': pipe.mode_blob.id})
+ req.add(pipe.plane, {
+ 'FB_ID': fb.id,
+ 'CRTC_ID': pipe.crtc.id,
+ 'SRC_X': 0,
+ 'SRC_Y': 0,
+ 'SRC_W': int(mode.hdisplay * 65536),
+ 'SRC_H': int(mode.vdisplay * 65536),
+ 'CRTC_X': 0,
+ 'CRTC_Y': 0,
+ 'CRTC_W': mode.hdisplay,
+ 'CRTC_H': mode.vdisplay,
+ })
+
+ route.append("CRTC %u to connector %s" % (pipe.crtc.id, pipe.connector.fullname))
+
+ self.logger.log("Routing " + ", ".join(route))
+
+ ret = req.commit_sync(True)
+ if ret < 0:
+ self.fail("atomic commit failed with %d" % ret)
+ return
+
+ time.sleep(5)
+
+ self.success()
+
+ for pipe in pipes:
+ self.atomic_crtc_disable(pipe.crtc)
+
+
+ def allocate_connectors(self, pipes, master_pipe, shared_connector):
+ # Allocate one connector for each CRTC. Create a pool of available
+ # connectors for each CRTC, sorted by the number of connectors, and
+ # allocate started with the CRTC that has the least number of options.
+ # The master CRTC is always given the shared connector.
+ pool = []
+ for pipe in pipes:
+ if pipe == master_pipe:
+ pool.append((pipe.crtc, [shared_connector]))
+ continue
+
+ pool.append((pipe.crtc, list(self.crtc_to_connectors[pipe.crtc])))
+
+ allocated = {}
+ while len(pool):
+ pool.sort(key=lambda elem: len(elem[1]), reverse=True)
+ crtc, connectors = pool[-1]
+
+ connector = connectors[0]
+ allocated[crtc] = connector
+
+ # Remove the selected connector from all elements in the pool
+ pool = [(elem[0], [c for c in elem[1] if c != connector]) for elem in pool[:-1]]
+
+ return allocated
+
+
+RoutingTest().execute()