summaryrefslogtreecommitdiff
path: root/py
diff options
context:
space:
mode:
Diffstat (limited to 'py')
-rwxr-xr-xpy/alpha-test.py67
-rw-r--r--py/helpers.py40
-rwxr-xr-xpy/trans-test.py332
3 files changed, 439 insertions, 0 deletions
diff --git a/py/alpha-test.py b/py/alpha-test.py
new file mode 100755
index 0000000..15ba045
--- /dev/null
+++ b/py/alpha-test.py
@@ -0,0 +1,67 @@
+#!/usr/bin/python3.4
+
+import pykms
+from helpers import *
+import time
+
+# This hack makes drm initialize the fbcon, setting up the default connector
+card = pykms.Card()
+card = 0
+
+card = pykms.Card()
+
+conn = card.get_first_connected_connector()
+mode = conn.get_default_mode()
+crtc = conn.get_current_crtc()
+
+planes = []
+for p in card.get_planes():
+ if p.supports_crtc(crtc) == False:
+ continue
+ planes.append(p)
+
+if len(planes) != 3:
+ print("Need 3 planes!")
+ exit(1)
+
+disable_planes(card)
+
+w = mode.hdisplay
+h = mode.vdisplay
+
+fbs=[]
+
+for i in range(len(planes)):
+ fbs.append(pykms.DumbFramebuffer(card, w, h, "AR24"))
+
+pykms.draw_rect(fbs[0], 50, 50, 200, 200, pykms.RGB(128, 255, 0, 0))
+pykms.draw_rect(fbs[1], 150, 50, 200, 200, pykms.RGB(128, 0, 255, 0))
+pykms.draw_rect(fbs[2], 50, 150, 200, 200, pykms.RGB(128, 0, 0, 255))
+
+
+set_props(crtc, {
+ "trans-key-mode": 0,
+ "trans-key": 0,
+ "background": 0,
+ "alpha_blender": 1,
+})
+
+for i in range(len(planes)):
+ plane = planes[i]
+ fb = fbs[i]
+
+ print("set crtc {}, plane {}, fb {}".format(crtc.id(), p.id(), fbs[i].id()))
+
+ set_props(plane, {
+ "FB_ID": fb.id(),
+ "CRTC_ID": crtc.id(),
+ "SRC_W": fb.width() << 16,
+ "SRC_H": fb.height() << 16,
+ "CRTC_W": fb.width(),
+ "CRTC_H": fb.height(),
+ "zorder": i,
+ })
+
+ time.sleep(1)
+
+input("press enter to exit\n")
diff --git a/py/helpers.py b/py/helpers.py
new file mode 100644
index 0000000..15f6cfa
--- /dev/null
+++ b/py/helpers.py
@@ -0,0 +1,40 @@
+import pykms
+
+def add_props(areq, ob, map):
+ for key, value in map.items():
+ areq.add(ob, key, value)
+
+
+def props(o):
+ o.refresh_props()
+ map = o.get_prop_map()
+ for propid in map:
+ prop = o.card().get_prop(propid)
+ print("%-15s %d (%#x)" % (prop.name(), map[propid], map[propid]))
+
+def set_props(ob, map):
+ areq = pykms.AtomicReq(ob.card())
+
+ for key, value in map.items():
+ areq.add(ob, key, value)
+
+ if areq.commit() != 0:
+ print("commit failed")
+
+red = pykms.RGB(255, 0, 0)
+green = pykms.RGB(0, 255, 0)
+blue = pykms.RGB(0, 0, 255)
+yellow = pykms.RGB(255, 255, 0)
+purple = pykms.RGB(255, 0, 255)
+white = pykms.RGB(255, 255, 255)
+cyan = pykms.RGB(0, 255, 255)
+
+def disable_planes(card):
+ areq = pykms.AtomicReq(card)
+
+ for p in card.get_planes():
+ areq.add(p, "FB_ID", 0)
+ areq.add(p, "CRTC_ID", 0)
+
+ if areq.commit() != 0:
+ print("disabling planes failed")
diff --git a/py/trans-test.py b/py/trans-test.py
new file mode 100755
index 0000000..3d1d5ff
--- /dev/null
+++ b/py/trans-test.py
@@ -0,0 +1,332 @@
+#!/usr/bin/python3.4
+
+import pykms
+from helpers import *
+import time
+
+# This hack makes drm initialize the fbcon, setting up the default connector
+card = pykms.Card()
+card = 0
+
+card = pykms.Card()
+
+conn = card.get_first_connected_connector()
+mode = conn.get_default_mode()
+crtc = conn.get_current_crtc()
+
+planes = []
+for p in card.get_planes():
+ if p.supports_crtc(crtc) == False:
+ continue
+ planes.append(p)
+
+disable_planes(card)
+
+w = mode.hdisplay
+h = mode.vdisplay
+
+fbs=[]
+
+def test_am5_trans_dest():
+ fbs.append(pykms.DumbFramebuffer(card, w, h, "XR24"))
+ fbs.append(pykms.DumbFramebuffer(card, w, h, "XR24"))
+
+ fb = fbs[0]
+ pykms.draw_rect(fb, 0, 0, fb.width(), fb.height(), purple)
+ pykms.draw_rect(fb, 100, 100, 100, 200, green)
+ pykms.draw_rect(fb, 300, 100, 100, 200, red)
+ pykms.draw_rect(fb, 500, 100, 100, 200, white)
+
+ fb = fbs[1]
+ pykms.draw_rect(fb, 0, 0, fb.width(), fb.height(), cyan)
+ pykms.draw_rect(fb, 250, 100, 200, 200, yellow)
+
+ set_props(crtc, {
+ "trans-key-mode": 1,
+ "trans-key": purple.rgb888(),
+ "background": 0,
+ "alpha_blender": 0,
+ })
+
+ plane = 0
+
+ for i in range(0,2):
+ print("set crtc {}, plane {}, fb {}".format(crtc.id(), planes[i].id(), fbs[i].id()))
+
+ plane = planes[i]
+ fb = fbs[i]
+ set_props(plane, {
+ "FB_ID": fb.id(),
+ "CRTC_ID": crtc.id(),
+ "SRC_W": fb.width() << 16,
+ "SRC_H": fb.height() << 16,
+ "CRTC_W": fb.width(),
+ "CRTC_H": fb.height(),
+ "zorder": i,
+ })
+
+ time.sleep(1)
+
+def test_am5_trans_src():
+ fbs.append(pykms.DumbFramebuffer(card, w, h, "XR24"))
+ fbs.append(pykms.DumbFramebuffer(card, w, h, "XR24"))
+
+ fb = fbs[0]
+ pykms.draw_rect(fb, 0, 0, fb.width(), fb.height(), white)
+ pykms.draw_rect(fb, 200, 200, 100, 100, red)
+ pykms.draw_rect(fb, fb.width() - 300, 200, 100, 100, green)
+
+ fb = fbs[1]
+ pykms.draw_rect(fb, 0, 0, fb.width(), fb.height(), cyan)
+ pykms.draw_rect(fb, 100, 100, 500, 500, purple)
+
+ set_props(crtc, {
+ "trans-key-mode": 2,
+ "trans-key": purple.rgb888(),
+ "background": 0,
+ "alpha_blender": 0,
+ })
+
+ plane = 0
+
+ for i in range(0,2):
+ print("set crtc {}, plane {}, fb {}".format(crtc.id(), planes[i].id(), fbs[i].id()))
+
+ plane = planes[i]
+ fb = fbs[i]
+ set_props(plane, {
+ "FB_ID": fb.id(),
+ "CRTC_ID": crtc.id(),
+ "SRC_W": fb.width() << 16,
+ "SRC_H": fb.height() << 16,
+ "CRTC_W": fb.width(),
+ "CRTC_H": fb.height(),
+ "zorder": 3 if i == 1 else 0,
+ })
+
+ time.sleep(1)
+
+def test_am4_normal_trans_dst():
+ fbs.append(pykms.DumbFramebuffer(card, w, h, "XR24"))
+ fbs.append(pykms.DumbFramebuffer(card, w * 2 // 3, h, "XR24"))
+ fbs.append(pykms.DumbFramebuffer(card, w * 2 // 3, h, "XR24"))
+
+ fb = fbs[0]
+ pykms.draw_rect(fb, 0, 0, w, h, purple)
+ pykms.draw_rect(fb, 100, 50, 50, 200, green)
+ pykms.draw_rect(fb, 200, 50, 50, 200, red)
+ pykms.draw_rect(fb, 300, 50, 50, 200, white)
+
+ fb = fbs[1]
+ pykms.draw_rect(fb, 0, 0, fb.width(), fb.height(), blue)
+
+ fb = fbs[2]
+ pykms.draw_rect(fb, 0, 0, fb.width(), fb.height(), cyan)
+
+ set_props(crtc, {
+ "trans-key-mode": 1,
+ "trans-key": purple.rgb888(),
+ "background": 0,
+ "alpha_blender": 0,
+ })
+
+ time.sleep(1)
+
+ plane = planes[0]
+ fb = fbs[0]
+ set_props(plane, {
+ "FB_ID": fb.id(),
+ "CRTC_ID": crtc.id(),
+ "SRC_W": fb.width() << 16,
+ "SRC_H": fb.height() << 16,
+ "CRTC_W": w,
+ "CRTC_H": h,
+ })
+
+ time.sleep(1)
+
+ plane = planes[1]
+ fb = fbs[1]
+ set_props(plane, {
+ "FB_ID": fb.id(),
+ "CRTC_ID": crtc.id(),
+ "SRC_X": 0 << 16,
+ "SRC_Y": 0 << 16,
+ "SRC_W": fb.width() << 16,
+ "SRC_H": fb.height() << 16,
+ "CRTC_X": 0,
+ "CRTC_Y": 0,
+ "CRTC_W": fb.width(),
+ "CRTC_H": fb.height(),
+ })
+
+ time.sleep(1)
+
+ plane = planes[2]
+ fb = fbs[2]
+ set_props(plane, {
+ "FB_ID": fb.id(),
+ "CRTC_ID": crtc.id(),
+ "SRC_X": 0 << 16,
+ "SRC_Y": 0 << 16,
+ "SRC_W": fb.width() << 16,
+ "SRC_H": fb.height() << 16,
+ "CRTC_X": w // 3,
+ "CRTC_Y": 0,
+ "CRTC_W": fb.width(),
+ "CRTC_H": fb.height(),
+ })
+
+def test_am4_normal_trans_src():
+ fbs.append(pykms.DumbFramebuffer(card, w, h, "XR24"))
+ fbs.append(pykms.DumbFramebuffer(card, w // 2, h, "XR24"))
+ fbs.append(pykms.DumbFramebuffer(card, w // 2, h, "XR24"))
+
+ fb = fbs[0]
+ pykms.draw_rect(fb, 0, 0, w, h, pykms.RGB(128, 255, 255))
+ pykms.draw_rect(fb, 200, 100, 50, 200, red)
+ pykms.draw_rect(fb, w - 200 - 50, 100, 50, 200, green)
+
+ fb = fbs[1]
+ pykms.draw_rect(fb, 0, 0, fb.width(), fb.height(), blue)
+ pykms.draw_rect(fb, 100, 100, fb.width() - 200, fb.height() - 200, purple)
+
+ fb = fbs[2]
+ pykms.draw_rect(fb, 0, 0, fb.width(), fb.height(), cyan)
+ pykms.draw_rect(fb, 100, 100, fb.width() - 200, fb.height() - 200, purple)
+
+ set_props(crtc, {
+ "trans-key-mode": 2,
+ "trans-key": purple.rgb888(),
+ "background": 0,
+ "alpha_blender": 0,
+ })
+
+ time.sleep(1)
+
+ plane = planes[0]
+ fb = fbs[0]
+ set_props(plane, {
+ "FB_ID": fb.id(),
+ "CRTC_ID": crtc.id(),
+ "SRC_W": fb.width() << 16,
+ "SRC_H": fb.height() << 16,
+ "CRTC_W": w,
+ "CRTC_H": h,
+ })
+
+ time.sleep(1)
+
+ plane = planes[1]
+ fb = fbs[1]
+ set_props(plane, {
+ "FB_ID": fb.id(),
+ "CRTC_ID": crtc.id(),
+ "SRC_X": 0 << 16,
+ "SRC_Y": 0 << 16,
+ "SRC_W": fb.width() << 16,
+ "SRC_H": fb.height() << 16,
+ "CRTC_X": 0,
+ "CRTC_Y": 0,
+ "CRTC_W": fb.width(),
+ "CRTC_H": fb.height(),
+ })
+
+ time.sleep(1)
+
+ plane = planes[2]
+ fb = fbs[2]
+ set_props(plane, {
+ "FB_ID": fb.id(),
+ "CRTC_ID": crtc.id(),
+ "SRC_X": 0 << 16,
+ "SRC_Y": 0 << 16,
+ "SRC_W": fb.width() << 16,
+ "SRC_H": fb.height() << 16,
+ "CRTC_X": w - fb.width(),
+ "CRTC_Y": 0,
+ "CRTC_W": fb.width(),
+ "CRTC_H": fb.height(),
+ })
+
+def test_am4_alpha_trans_src():
+ fbs.append(pykms.DumbFramebuffer(card, w, h, "XR24"))
+ fbs.append(pykms.DumbFramebuffer(card, w // 2, h, "XR24"))
+ fbs.append(pykms.DumbFramebuffer(card, w // 2, h, "XR24"))
+
+ fb = fbs[0]
+ pykms.draw_rect(fb, 0, 0, w, h, purple)
+ pykms.draw_rect(fb, 200, 100, 50, 200, red)
+ pykms.draw_rect(fb, w - 200 - 50, 100, 50, 200, green)
+
+ fb = fbs[1]
+ pykms.draw_rect(fb, 0, 0, fb.width(), fb.height(), blue)
+ pykms.draw_rect(fb, 100, 100, fb.width() - 200, fb.height() - 200, purple)
+
+ fb = fbs[2]
+ pykms.draw_rect(fb, 0, 0, fb.width(), fb.height(), cyan)
+ pykms.draw_rect(fb, 100, 100, fb.width() - 200, fb.height() - 200, purple)
+
+ set_props(crtc, {
+ "trans-key-mode": 1,
+ "trans-key": purple.rgb888(),
+ "background": 0,
+ "alpha_blender": 1,
+ })
+
+ time.sleep(1)
+
+ plane = planes[0]
+ fb = fbs[0]
+ set_props(plane, {
+ "FB_ID": fb.id(),
+ "CRTC_ID": crtc.id(),
+ "SRC_W": fb.width() << 16,
+ "SRC_H": fb.height() << 16,
+ "CRTC_W": w,
+ "CRTC_H": h,
+ })
+
+ time.sleep(1)
+
+ plane = planes[1]
+ fb = fbs[1]
+ set_props(plane, {
+ "FB_ID": fb.id(),
+ "CRTC_ID": crtc.id(),
+ "SRC_X": 0 << 16,
+ "SRC_Y": 0 << 16,
+ "SRC_W": fb.width() << 16,
+ "SRC_H": fb.height() << 16,
+ "CRTC_X": 0,
+ "CRTC_Y": 0,
+ "CRTC_W": fb.width(),
+ "CRTC_H": fb.height(),
+ })
+
+ time.sleep(1)
+
+ plane = planes[2]
+ fb = fbs[2]
+ set_props(plane, {
+ "FB_ID": fb.id(),
+ "CRTC_ID": crtc.id(),
+ "SRC_X": 0 << 16,
+ "SRC_Y": 0 << 16,
+ "SRC_W": fb.width() << 16,
+ "SRC_H": fb.height() << 16,
+ "CRTC_X": w - fb.width(),
+ "CRTC_Y": 0,
+ "CRTC_W": fb.width(),
+ "CRTC_H": fb.height(),
+ })
+
+
+
+#test_am5_trans_dest()
+test_am5_trans_src()
+#test_am4_normal_trans_dst()
+#test_am4_normal_trans_src()
+#test_am4_alpha_trans_src()
+
+input("press enter to exit\n")
generated by cgit v1.2.3 (git 2.25.1) at 2025-11-04 03:35:41 +0000