Merge pull request #37 from tomba/work

Formatting changes and testmodes.py

2 files changed, 50 insertions, 0 deletions

diff --git a/py/pykms/pykmsbase.cpp b/py/pykms/pykmsbase.cpp
index 668e6e3..b4dc090 100644
--- a/py/pykms/pykmsbase.cpp
+++ b/py/pykms/pykmsbase.cpp
@@ -229,6 +229,9 @@ void init_pykmsbase(py::module &m)
 
 			.def_property("hsync", &Videomode::hsync, &Videomode::set_hsync)
 			.def_property("vsync", &Videomode::vsync, &Videomode::set_vsync)
+
+			.def("to_string_short", &Videomode::to_string_short)
+			.def("to_string_long", &Videomode::to_string_long)
 			;
 
 
diff --git a/py/tests/testmodes.py b/py/tests/testmodes.py
new file mode 100755
index 0000000..0508f31
--- /dev/null
+++ b/py/tests/testmodes.py
@@ -0,0 +1,47 @@
+#!/usr/bin/python3
+
+import sys
+import pykms
+import argparse
+
+parser = argparse.ArgumentParser()
+parser.add_argument("-c", "--connector", default="")
+args = parser.parse_args()
+
+card = pykms.Card()
+
+res = pykms.ResourceManager(card)
+conn = res.reserve_connector(args.connector)
+crtc = res.reserve_crtc(conn)
+plane = res.reserve_generic_plane(crtc)
+
+card.disable_planes()
+
+modes = conn.get_modes()
+
+def even(i):
+	return i & ~1
+
+for mode in modes:
+	long_str = mode.to_string_long()
+	short_str = mode.to_string_short()
+
+	print(long_str)
+
+	modeb = mode.to_blob(card)
+
+	fb = pykms.DumbFramebuffer(card, mode.hdisplay, mode.vdisplay, "XR24");
+	pykms.draw_test_pattern(fb);
+	pykms.draw_text(fb, even((fb.width // 2) - (8 * len(short_str)) // 2), 4, short_str, pykms.white)
+
+	req = pykms.AtomicReq(card)
+
+	req.add_connector(conn, crtc)
+	req.add_crtc(crtc, modeb)
+	req.add_plane(plane, fb, crtc, dst=(0, 0, mode.hdisplay, mode.vdisplay))
+
+	req.commit_sync(allow_modeset = True)
+
+	input("press enter to show next videomode\n")
+
+print("done")