py: add db test

3 files changed, 76 insertions, 2 deletions

diff --git a/py/CMakeLists.txt b/py/CMakeLists.txt
index 8ad476b..42dae20 100644
--- a/py/CMakeLists.txt
+++ b/py/CMakeLists.txt
@@ -17,7 +17,7 @@ set_source_files_properties(pykms.i PROPERTIES CPLUSPLUS ON)
 swig_add_module(pykms python pykms.i)
 swig_link_libraries(pykms kms++ kmstest ${LIBDRM_LIBRARIES} ${PYTHON_LIBRARIES})
 
-add_custom_target(pyextras SOURCES test.py functest.py)
+add_custom_target(pyextras SOURCES test.py functest.py db.py)
 
 add_test(NAME pytest COMMAND "${CMAKE_CURRENT_SOURCE_DIR}/functest.py")
 set_property(TEST pytest PROPERTY
diff --git a/py/db.py b/py/db.py
new file mode 100755
index 0000000..fbc1998
--- /dev/null
+++ b/py/db.py
@@ -0,0 +1,71 @@
+#!/usr/bin/python3.4
+
+import sys
+import pykms
+import selectors
+
+bar_width = 20
+bar_speed = 8
+
+class FlipHandler(pykms.PageFlipHandlerBase):
+    def __init__(self):
+        super().__init__()
+        self.bar_xpos = 0
+        self.front_buf = 0
+        self.fb1 = pykms.DumbFramebuffer(card, mode.hdisplay, mode.vdisplay, "XR24");
+        self.fb2 = pykms.DumbFramebuffer(card, mode.hdisplay, mode.vdisplay, "XR24");
+
+    def handle_page_flip(self, frame, time):
+        if self.front_buf == 0:
+            fb = self.fb2
+        else:
+            fb = self.fb1
+
+        self.front_buf = self.front_buf ^ 1
+
+        current_xpos = self.bar_xpos;
+        old_xpos = (current_xpos + (fb.width() - bar_width - bar_speed)) % (fb.width() - bar_width);
+        new_xpos = (current_xpos + bar_speed) % (fb.width() - bar_width);
+
+        self.bar_xpos = new_xpos
+
+        pykms.draw_color_bar(fb, old_xpos, new_xpos, bar_width)
+
+        if card.has_atomic():
+            ctx = pykms.AtomicReq(card)
+            ctx.add(crtc, "FB_ID", fb.id())
+            ctx.commit(self)
+        else:
+            crtc.page_flip(fb, self)
+
+
+
+card = pykms.Card()
+conn = card.get_first_connected_connector()
+mode = conn.get_default_mode()
+crtc = conn.get_current_crtc()
+
+fliphandler = FlipHandler()
+
+crtc.set_mode(conn, fliphandler.fb1, mode)
+
+fliphandler.handle_page_flip(0, 0)
+
+def readdrm(conn, mask):
+    #print("EVENT");
+    card.call_page_flip_handlers()
+
+def readkey(conn, mask):
+    #print("KEY EVENT");
+    sys.stdin.readline()
+    exit(0)
+
+sel = selectors.DefaultSelector()
+sel.register(card.fd(), selectors.EVENT_READ, readdrm)
+sel.register(sys.stdin, selectors.EVENT_READ, readkey)
+
+while True:
+    events = sel.select()
+    for key, mask in events:
+        callback = key.data
+        callback(key.fileobj, mask)
diff --git a/py/pykms.i b/py/pykms.i
index 1dcd0ca..6d8de67 100644
--- a/py/pykms.i
+++ b/py/pykms.i
@@ -1,4 +1,4 @@
-%module pykms
+%module(directors="1") pykms
 %{
 #include "kms++.h"
 
@@ -10,6 +10,8 @@ using namespace kms;
 %include "std_string.i"
 %include "stdint.i"
 
+%feature("director") PageFlipHandlerBase;
+
 %include "decls.h"
 %include "drmobject.h"
 %include "atomicreq.h"
@@ -21,5 +23,6 @@ using namespace kms;
 %include "plane.h"
 %include "connector.h"
 %include "encoder.h"
+%include "pagefliphandler.h"
 
 %include "kmstest.h"