1 files changed, 11 insertions, 6 deletions
diff --git a/examples/game.c b/examples/game.c
index 79beb7c..0be4461 100644
--- a/examples/game.c
+++ b/examples/game.c
@@ -29,15 +29,20 @@ int main(int argc, char **argv)
     }
     yasl_gl_swap_interval(s, 1);
     int result = 0, frames = 0;
+    bool online = true;
     for (bool running = true; running;) {
+        /* Drain events before each frame; while the display is gone, block until it returns. */
         yasl_event e;
-        int n;
-        while ((n = yasl_wait_event(&e, 0)) > 0) {
-            if (e.type == YASL_EVENT_CLOSE || e.type == YASL_EVENT_QUIT ||
-                (e.type == YASL_EVENT_KEY && e.key.key == YASL_KEY_ESCAPE)) running = false;
-            if (e.type == YASL_EVENT_DISPLAY_LOST) { result = 1; running = false; }
-        }
+        int n = yasl_wait_event(&e, online ? 0 : -1);
         if (n < 0) { result = 1; break; }
+        if (e.type == YASL_EVENT_CLOSE || e.type == YASL_EVENT_QUIT ||
+            (e.type == YASL_EVENT_KEY && e.key.key == YASL_KEY_ESCAPE)) running = false;
+        if (e.type == YASL_EVENT_DISPLAY_LOST) online = false;
+        if (e.type == YASL_EVENT_DISPLAY_RESTORED) {
+            /* The context is new: bind it again and restore GL state here. */
+            online = yasl_gl_make_current(s) && yasl_gl_swap_interval(s, 1);
+        }
+        if (n > 0 || !online || !running) continue;
         int w, h;
         yasl_surface_get_framebuffer_size(s, &w, &h);
         viewport(0, 0, w, h);