test.py: add dmabuf & omap options

diff --git a/py/tests/test.py b/py/tests/test.py
index de00a4371ed63da10d675fb3f519bb063e4c8237..b11add7e976949f591788aa098d0c6723795374d 100755 (executable)
--- a/py/tests/test.py
+++ b/py/tests/test.py
@@ -2,14 +2,33 @@
 
 import pykms
 
-card = pykms.Card()
+# draw test pattern via dmabuf?
+dmabuf = False
+
+# Use omap?
+omap = False
+
+if omap:
+       card = pykms.OmapCard()
+else:
+       card = pykms.Card()
+
 res = pykms.ResourceManager(card)
 conn = res.reserve_connector()
 crtc = res.reserve_crtc(conn)
-
 mode = conn.get_default_mode()
 
-fb = pykms.DumbFramebuffer(card, mode.hdisplay, mode.vdisplay, "XR24");
+if omap:
+       origfb = pykms.OmapFramebuffer(card, mode.hdisplay, mode.vdisplay, "XR24");
+else:
+       origfb = pykms.DumbFramebuffer(card, mode.hdisplay, mode.vdisplay, "XR24");
+
+if dmabuf:
+       fb = pykms.ExtFramebuffer(card, origfb.width, origfb.height, origfb.format,
+               [origfb.fd(0)], [origfb.stride(0)], [origfb.offset(0)])
+else:
+       fb = origfb
+
 pykms.draw_test_pattern(fb);
 
 crtc.set_mode(conn, fb, mode)