From e71c232660a7480c2b56f6e76e83fad4c7e9da8a Mon Sep 17 00:00:00 2001 From: Jyri Sarha Date: Fri, 24 Mar 2017 22:47:53 +0200 Subject: Add ctm_test.py test for testing CRTC's CTM color matrix property. --- py/tests/ctm_test.py | 86 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 86 insertions(+) create mode 100755 py/tests/ctm_test.py (limited to 'py/tests') diff --git a/py/tests/ctm_test.py b/py/tests/ctm_test.py new file mode 100755 index 0000000..7ceed6f --- /dev/null +++ b/py/tests/ctm_test.py @@ -0,0 +1,86 @@ +#!/usr/bin/python3 + +import sys +import pykms + +def ctm_to_blob(ctm, card): + len=9 + arr = bytearray(len*8) + view = memoryview(arr).cast("I") + + for x in range(len): + i, d = divmod(ctm[x], 1) + if i < 0: + i = -i + sign = 1 << 31 + else: + sign = 0 + view[x * 2 + 0] = int(d * ((2 ** 32) - 1)) + view[x * 2 + 1] = int(i) | sign + #print("%f = %08x.%08x" % (ctm[x], view[x * 2 + 1], view[x * 2 + 0])) + + return pykms.Blob(card, arr); + + +if len(sys.argv) > 1: + conn_name = sys.argv[1] +else: + conn_name = "" + +card = pykms.Card() +res = pykms.ResourceManager(card) +conn = res.reserve_connector(conn_name) +crtc = res.reserve_crtc(conn) +mode = conn.get_default_mode() + +fb = pykms.DumbFramebuffer(card, mode.hdisplay, mode.vdisplay, "XR24"); +pykms.draw_test_pattern(fb); + +crtc.set_mode(conn, fb, mode) + +input("press enter to set normal ctm\n") + +ctm = [ 1.0, 0.0, 0.0, + 0.0, 1.0, 0.0, + 0.0, 0.0, 1.0 ] + +ctmb = ctm_to_blob(ctm, card) + +crtc.set_prop("CTM", ctmb.id) + +input("press enter to set new ctm\n") + +ctm = [ 0.0, 1.0, 0.0, + 0.0, 0.0, 1.0, + 1.0, 0.0, 0.0 ] + +ctmb = ctm_to_blob(ctm, card) + +crtc.set_prop("CTM", ctmb.id) + +print("r->b g->r b->g ctm active\n") + +input("press enter to set new ctm\n") + +ctm = [ 0.0, 0.0, 1.0, + 1.0, 0.0, 0.0, + 0.0, 1.0, 0.0 ] + +ctmb = ctm_to_blob(ctm, card) + +crtc.set_prop("CTM", ctmb.id) +input("r->g g->b b->r ctm active\n") + +input("press enter to turn off the crtc\n") + +crtc.disable_mode() + +input("press enter to enable crtc again\n") + +crtc.set_mode(conn, fb, mode) + +input("press enter to remove ctm\n") + +crtc.set_prop("CTM", 0) + +input("press enter to exit\n") -- cgit v1.2.3 From 7450c364a57ec7d9e2abd61ac6e025c53e9e7a42 Mon Sep 17 00:00:00 2001 From: Jyri Sarha Date: Fri, 12 May 2017 10:44:21 +0300 Subject: plane_csc.py: Test for different YCbCr encodings --- py/tests/plane_csc.py | 66 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 66 insertions(+) create mode 100755 py/tests/plane_csc.py (limited to 'py/tests') diff --git a/py/tests/plane_csc.py b/py/tests/plane_csc.py new file mode 100755 index 0000000..be92c00 --- /dev/null +++ b/py/tests/plane_csc.py @@ -0,0 +1,66 @@ +#!/usr/bin/python3 + +import pykms + +card = pykms.Card() +res = pykms.ResourceManager(card) +conn = res.reserve_connector("HDMI") +crtc = res.reserve_crtc(conn) +mode = conn.get_default_mode() +modeb = mode.to_blob(card) +plane = res.reserve_generic_plane(crtc, pykms.PixelFormat.UYVY) +#plane = res.reserve_generic_plane(crtc, pykms.PixelFormat.Undefined) + +print("Got plane %d %d" % (plane.idx, plane.id)) + +fb = pykms.DumbFramebuffer(card, mode.hdisplay, mode.vdisplay, "UYVY"); +pykms.draw_test_pattern(fb); + +req = pykms.AtomicReq(card) +req.add(conn, "CRTC_ID", crtc.id) +req.add(crtc, {"ACTIVE": 1, + "MODE_ID": modeb.id}) + +input("Press enter to enable crtc idx %d at %s" % (crtc.idx, conn.fullname)) +r = req.commit_sync(allow_modeset = True) + +input("Press enter to enable plane idx %d at %s" % (plane.idx, conn.fullname)) + +req = pykms.AtomicReq(card) +req.add(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, + "zorder": 0}) +r = req.commit_sync() +print("Plane enable request returned %d\n" % r) + +yuv_types = [pykms.YUVType.BT601_Lim, + pykms.YUVType.BT601_Full, + pykms.YUVType.BT709_Lim, + pykms.YUVType.BT709_Full] + +encoding_enums = plane.get_prop("COLOR_ENCODING").enums +range_enums = plane.get_prop("COLOR_RANGE").enums + +for i in range(0, 2): + for j in range(0, 2): + input("press enter to for encoding: \"%s\" range: \"%s\"\n" % + (encoding_enums[i], range_enums[j])) + + req = pykms.AtomicReq(card) + req.add(plane, {"COLOR_ENCODING": i, + "COLOR_RANGE": j}) + req.commit_sync() + + for t in yuv_types: + input("press enter to redraw with yuv_type %s\n" % t) + pykms.draw_test_pattern(fb, t); + +input("press enter to exit\n") -- cgit v1.2.3