X-Git-Url: http://git.mutantstargoat.com/user/nuclear/?a=blobdiff_plain;f=src%2Fscreen.c;h=178cf7ddfb35274accbfd154eadea3451226cb38;hb=7340877bf7b27117a0a1720f628ab663df6b4244;hp=a949fc467a127e14ee51764ca8c0931a72be79ca;hpb=757e7697c9a4434df6f3fdf3234de504bb4636bc;p=dosdemo

diff --git a/src/screen.c b/src/screen.c
index a949fc4..178cf7d 100644
--- a/src/screen.c
+++ b/src/screen.c
@@ -1,14 +1,17 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
+#include <assert.h>
 #include "screen.h"
 #include "demo.h"
 
 struct screen *tunnel_screen(void);
 struct screen *fract_screen(void);
+struct screen *mike_screen(void);
 
 #define NUM_SCR	32
 static struct screen *scr[NUM_SCR];
+static int num_screens;
 
 static struct screen *cur, *prev, *next;
 static long trans_start, trans_dur;
@@ -23,9 +26,15 @@ int scr_init(void)
 	if(!(scr[idx++] = fract_screen())) {
 		return -1;
 	}
+	if (!(scr[idx++] = mike_screen())) {
+		return -1;
+	}
+	num_screens = idx;
+
+	assert(num_screens <= NUM_SCR);
 
-	for(i=0; i<NUM_SCR; i++) {
-		if(scr[i] && scr[i]->init() == -1) {
+	for(i=0; i<num_screens; i++) {
+		if(scr[i]->init() == -1) {
 			return -1;
 		}
 	}
@@ -35,8 +44,7 @@ int scr_init(void)
 void scr_shutdown(void)
 {
 	int i;
-	for(i=0; i<NUM_SCR; i++) {
-		if(!scr[i]) break;
+	for(i=0; i<num_screens; i++) {
 		scr[i]->shutdown();
 	}
 }
@@ -64,8 +72,7 @@ void scr_draw(void)
 struct screen *scr_lookup(const char *name)
 {
 	int i;
-	for(i=0; i<NUM_SCR; i++) {
-		if(!scr[i]) break;
+	for(i=0; i<num_screens; i++) {
 		if(strcmp(scr[i]->name, name) == 0) {
 			return scr[i];
 		}
@@ -73,6 +80,16 @@ struct screen *scr_lookup(const char *name)
 	return 0;
 }
 
+struct screen *scr_screen(int idx)
+{
+	return scr[idx];
+}
+
+int scr_num_screens(void)
+{
+	return num_screens;
+}
+
 int scr_change(struct screen *s, long trans_time)
 {
 	if(!s) return -1;