Add plane_move.py test for testing plane move between crtcs

Signed-off-by: Peter Ujfalusi <peter.ujfalusi@ti.com>

1 files changed, 128 insertions, 0 deletions

diff --git a/py/tests/plane_move.py b/py/tests/plane_move.py
new file mode 100755
index 0000000..2f9dee5
--- /dev/null
+++ b/py/tests/plane_move.py
@@ -0,0 +1,128 @@
+#!/usr/bin/python3
+
+import pykms
+import random
+import time
+import sys
+import select
+import selectors
+
+if len(sys.argv) != 3:
+    print('Usage: plane_move.py <connector0> <connector1>')
+    sys.exit()
+
+card = pykms.Card()
+
+if not card.has_atomic:
+    print('Atomic modesetting is not supported')
+    sys.exit(-1)
+
+res = pykms.ResourceManager(card)
+
+conn_list = []
+crtc_list = []
+mode_list = []
+rootplane_list = []
+fb_list = []
+colors = []
+
+src_w = 300
+src_h = 300
+
+for i in range(2):
+    conn = res.reserve_connector(sys.argv[i + 1])
+    if conn is None:
+        print('Invalid connector: {}'.format(sys.argv[i + 1]))
+        sys.exit(-1)
+
+    if conn.connected() == True:
+        conn_list.append(conn)
+    else:
+        print('connector: {} is not connected'.format(sys.argv[i + 1]))
+        sys.exit(-1)
+
+    crtc = res.reserve_crtc(conn)
+    crtc_list.append(crtc)
+
+    mode = conn.get_default_mode()
+    mode_list.append(mode)
+
+    fb_tmp = pykms.DumbFramebuffer(card, src_w, src_h, 'XR24');
+    fb_list.append(fb_tmp)
+
+    rootplane = res.reserve_primary_plane(crtc, pykms.PixelFormat.XRGB8888)
+    rootplane_list.append(rootplane)
+
+card.disable_planes()
+
+print('Using the following connectors:')
+for i in range(2):
+    print(' {}: {} ({}x{})'.format(conn_list[i].idx, conn_list[i].fullname,
+        mode_list[i].hdisplay, mode_list[i].vdisplay))
+
+colors.append(pykms.red)
+colors.append(pykms.green)
+
+for i in range(2):
+    pykms.draw_rect(fb_list[i], 0, 0, src_w, src_h, colors[i])
+
+for i in range(2):
+    req = pykms.AtomicReq(card)
+    modeb = mode_list[i].to_blob(card)
+    req.add(conn_list[i], 'CRTC_ID', crtc_list[i].id)
+    req.add(crtc_list[i], {'ACTIVE': 1,
+                    'MODE_ID': modeb.id})
+    req.add(rootplane_list[i], {'FB_ID': fb_list[i].id,
+                'CRTC_ID': crtc_list[i].id,
+                'SRC_W': src_w << 16,
+                'SRC_H': src_h << 16,
+                'CRTC_W': src_w,
+                'CRTC_H': src_h})
+
+    req.commit_sync(allow_modeset = True)
+
+print('\nRed box on {}, Green box on {}.'.format(conn_list[0].fullname,
+                                                 conn_list[1].fullname))
+input('ENTER to continue\n')
+
+# FIXME: it should be possible to move plane without disabling it, but the
+# omapdrm driver does not supports it at the moment.
+req = pykms.AtomicReq(card)
+req.add(rootplane_list[0], {"FB_ID": 0,
+                "CRTC_ID": 0})
+req.commit_sync(allow_modeset = True)
+
+req = pykms.AtomicReq(card)
+req.add(rootplane_list[0], {'FB_ID': fb_list[0].id,
+            'CRTC_ID': crtc_list[1].id,
+            'SRC_W': src_w << 16,
+            'SRC_H': src_h << 16,
+            'CRTC_X': 150,
+            'CRTC_Y': 150,
+            'CRTC_W': src_w,
+            'CRTC_H': src_h})
+req.commit_sync(allow_modeset = True)
+
+print('The red box from {} is moved underneath the green box on {}.'.format(
+                                conn_list[0].fullname, conn_list[1].fullname))
+input('ENTER to continue\n')
+
+# FIXME: it should be possible to move plane without disabling it, but the
+# omapdrm driver does not supports it at the moment.
+req = pykms.AtomicReq(card)
+req.add(rootplane_list[1], {"FB_ID": 0,
+                "CRTC_ID": 0})
+req.commit_sync(allow_modeset = True)
+
+req = pykms.AtomicReq(card)
+req.add(rootplane_list[1], {'FB_ID': fb_list[1].id,
+            'CRTC_ID': crtc_list[0].id,
+            'SRC_W': src_w << 16,
+            'SRC_H': src_h << 16,
+            'CRTC_W': src_w,
+            'CRTC_H': src_h})
+req.commit_sync(allow_modeset = True)
+
+print('Green box on {}, Red box on {}.'.format(conn_list[0].fullname,
+                                               conn_list[1].fullname))
+input('ENTER to exit\n')