change parts with number keys

[dosdemo] / src / demo.c

diff --git a/src/demo.c b/src/demo.c
index 98f8001..7d22dc4 100644 (file)
--- a/src/demo.c
+++ b/src/demo.c
@@ -3,6 +3,7 @@
 #include <string.h>
 #include <math.h>
 #include <errno.h>
+#include <limits.h>
 #include "demo.h"
 #include "screen.h"
 
@@ -15,10 +16,12 @@ int mouse_x, mouse_y;
 unsigned int mouse_bmask;
 
 static unsigned long nframes;
-static const char *start_scr_name = "tunnel";
+static const char *start_scr_name;
 
 int demo_init(int argc, char **argv)
 {
+       struct screen *scr;
+
        if(argv[1]) {
                start_scr_name = argv[1];
        }
@@ -26,11 +29,19 @@ int demo_init(int argc, char **argv)
        if(scr_init() == -1) {
                return -1;
        }
-       if(scr_change(scr_lookup(start_scr_name), 4000) == -1) {
-               fprintf(stderr, "screen %s not found\n", start_scr_name);
+       if(start_scr_name) {
+               scr = scr_lookup(start_scr_name);
+       } else {
+               scr = scr_screen(0);
+       }
+
+       if(!scr || scr_change(scr, 4000) == -1) {
+               fprintf(stderr, "screen %s not found\n", start_scr_name ? start_scr_name : "0");
                return -1;
        }
 
+       /* clear the framebuffer at least once */
+       memset(fb_pixels, 0, fb_width * fb_height * fb_bpp / CHAR_BIT);
        return 0;
 }
 
@@ -61,6 +72,11 @@ void demo_keyboard(int key, int state)
                        break;
 
                default:
+                       if(key >= '1' && key <= '1' + scr_num_screens()) {
+                               int idx = key - '1';
+                               printf("change screen %d\n", idx);
+                               scr_change(scr_screen(idx), 4000);
+                       }
                        break;
                }
        }