From: Tomi Valkeinen Date: Thu, 17 Mar 2016 16:47:47 +0000 (+0200) Subject: py: add transparency key and alpha tests X-Git-Url: https://git.ti.com/gitweb?p=android%2Fexternal-libkmsxx.git;a=commitdiff_plain;h=2d658a3d17e512f4ea71dc144d38f3c5774a3792 py: add transparency key and alpha tests --- 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")