nuclear@1: #include <stdio.h>
nuclear@1: #include <stdlib.h>
nuclear@1: #include <string.h>
nuclear@7: #include <signal.h>
nuclear@1: #include "inttypes.h"
nuclear@1: #include "gfx.h"
nuclear@1: #include "keyb.h"
nuclear@1: #include "mouse.h"
nuclear@1: #include "logger.h"
nuclear@6: #include "scene.h"
nuclear@9: #include "rayzor.h"
nuclear@9: #include "screen.h"
nuclear@9: #include "modeller.h"
nuclear@9: #include "renderer.h"
nuclear@9: #include "scrman.h"
nuclear@10: #include "timer.h"
nuclear@10: 
nuclear@10: #ifdef __DOS__
nuclear@10: #undef USE_ASM_SWAPBUF
nuclear@10: #endif
nuclear@10: 
nuclear@10: #ifdef USE_ASM_SWAPBUF
nuclear@10: // defined in swapbuf.asm
nuclear@10: extern "C" void swap_buffers_asm(void *dest, void *src, int xsz, int ysz, int bpp);
nuclear@10: #endif
nuclear@1: 
nuclear@6: static bool init();
nuclear@6: static void cleanup();
nuclear@1: static void display();
nuclear@1: static void swap_buffers();
nuclear@9: static void draw_cursor(uint32_t *buf, int mx, int my);
nuclear@1: static void handle_keyboard();
nuclear@1: static void handle_mouse();
nuclear@1: static bool parse_args(int argc, char **argv);
nuclear@7: static void sig(int s);
nuclear@1: 
nuclear@9: uint32_t *fb_pixels;
nuclear@9: int fb_width = 640;
nuclear@9: int fb_height = 480;
nuclear@9: int fb_bpp = 32;
nuclear@9: Scene *scene;
nuclear@9: 
nuclear@6: static bool novideo;
nuclear@9: static void *fb;
nuclear@1: static int rbits, gbits, bbits;
nuclear@1: static int rshift, gshift, bshift;
nuclear@1: static unsigned int rmask, gmask, bmask;
nuclear@1: 
nuclear@10: static bool use_asm_swap = true;
nuclear@7: static bool use_mouse;
nuclear@9: static int mouse_x, mouse_y;
nuclear@1: static bool quit;
nuclear@1: 
nuclear@1: int main(int argc, char **argv)
nuclear@1: {
nuclear@10: 	unsigned long start_msec, msec;
nuclear@10: 	unsigned long nframes = 0;
nuclear@10: 
nuclear@1: 	if(!parse_args(argc, argv)) {
nuclear@1: 		return 1;
nuclear@1: 	}
nuclear@6: 	if(!init()) {
nuclear@1: 		return 1;
nuclear@1: 	}
nuclear@1: 
nuclear@10: 	start_msec = get_msec();
nuclear@10: 
nuclear@1: 	// main loop
nuclear@1: 	for(;;) {
nuclear@1: 		handle_keyboard();
nuclear@1: 		handle_mouse();
nuclear@1: 		if(quit) break;
nuclear@1: 
nuclear@1: 		display();
nuclear@10: 		++nframes;
nuclear@6: 
nuclear@6: 		if(novideo) break;
nuclear@1: 	}
nuclear@1: 
nuclear@10: 	msec = get_msec() - start_msec;
nuclear@10: 
nuclear@6: 	cleanup();
nuclear@10: 
nuclear@10: 	printf("Average framerate: %g\n", (float)nframes / ((float)msec / 1000.0f));
nuclear@1: 	printf("Thank you for using Rayzor!\n");
nuclear@1: 	return 0;
nuclear@1: }
nuclear@1: 
nuclear@9: void quit_app()
nuclear@9: {
nuclear@9: 	quit = true;
nuclear@9: }
nuclear@9: 
nuclear@6: static bool init()
nuclear@6: {
nuclear@7: 	signal(SIGINT, sig);
nuclear@7: 	signal(SIGSEGV, sig);
nuclear@7: 	signal(SIGILL, sig);
nuclear@7: 	signal(SIGFPE, sig);
nuclear@7: 
nuclear@10: 	init_timer(128);
nuclear@10: 
nuclear@6: 	if(!novideo) {
nuclear@6: 		if(kb_init(32) == -1) {
nuclear@6: 			fprintf(stderr, "failed to initialize keyboard driver\n");
nuclear@6: 			return false;
nuclear@6: 		}
nuclear@6: 
nuclear@9: 		if(!(fb = set_video_mode(fb_width, fb_height, fb_bpp))) {
nuclear@6: 			set_text_mode();
nuclear@9: 			fprintf(stderr, "failed to set video mode: %dx%d %dbpp\n", fb_width, fb_height, fb_bpp);
nuclear@6: 			return false;
nuclear@6: 		}
nuclear@9: 		fb_bpp = get_color_depth();
nuclear@6: 		get_color_bits(&rbits, &gbits, &bbits);
nuclear@6: 		get_color_shift(&rshift, &gshift, &bshift);
nuclear@6: 		get_color_mask(&rmask, &gmask, &bmask);
nuclear@6: 
nuclear@10: 		printlog("video resolution: %dx%d\n", fb_width, fb_height);
nuclear@9: 		printlog("bpp: %d (%d %d %d)\n", fb_bpp, rbits, gbits, bbits);
nuclear@6: 		printlog("shift: %d %d %d\n", rshift, gshift, bshift);
nuclear@6: 		printlog("mask: %x %x %x\n", rmask, gmask, bmask);
nuclear@7: 
nuclear@7: 		if(have_mouse()) {
nuclear@7: 			use_mouse = true;
nuclear@9: 			set_mouse_limits(0, 0, fb_width - 1, fb_height - 1);
nuclear@7: 		}
nuclear@6: 	} else {
nuclear@6: 		logger_output(stdout);
nuclear@6: 		printlog("novideo (debug) mode\n");
nuclear@10: 		fb_bpp = 24;
nuclear@6: 		rbits = gbits = bbits = 8;
nuclear@6: 	}
nuclear@6: 
nuclear@9: 	fb_pixels = new uint32_t[fb_width * fb_height * 4];
nuclear@9: 	if(!fb_pixels) {
nuclear@6: 		return false;
nuclear@6: 	}
nuclear@6: 
nuclear@9: 	scene = new Scene;
nuclear@6: 
nuclear@6: 	Sphere *sph = new Sphere;
nuclear@9: 	scene->add_object(sph);
nuclear@6: 
nuclear@9: 	Modeller *modeller = new Modeller;
nuclear@9: 	if(!modeller->init()) {
nuclear@9: 		return false;
nuclear@9: 	}
nuclear@9: 	add_screen(modeller);
nuclear@9: 
nuclear@9: 	Renderer *renderer = new Renderer;
nuclear@9: 	if(!renderer->init()) {
nuclear@9: 		return false;
nuclear@9: 	}
nuclear@9: 	add_screen(renderer);
nuclear@9: 
nuclear@9: 	activate_screen(modeller); // start the modeller screen
nuclear@6: 	return true;
nuclear@6: }
nuclear@6: 
nuclear@6: static void cleanup()
nuclear@6: {
nuclear@9: 	delete scene;
nuclear@9: 	delete [] fb_pixels;
nuclear@6: 
nuclear@9: 	destroy_screens();
nuclear@6: 
nuclear@6: 	if(!novideo) {
nuclear@6: 		set_text_mode();
nuclear@6: 		kb_shutdown();
nuclear@6: 	}
nuclear@6: }
nuclear@6: 
nuclear@1: static void display()
nuclear@1: {
nuclear@9: 	Screen *scr = active_screen();
nuclear@9: 	if(scr) {
nuclear@9: 		scr->update();
nuclear@9: 		scr->draw();
nuclear@9: 	}
nuclear@6: 
nuclear@7: 	// draw the mouse cursor
nuclear@7: 	if(use_mouse) {
nuclear@9: 		draw_cursor(fb_pixels, mouse_x, mouse_y);
nuclear@7: 	}
nuclear@7: 
nuclear@6: 	if(!novideo) {
nuclear@10: 		wait_vsync();
nuclear@10: #ifdef USE_ASM_SWAPBUF
nuclear@10: 		swap_buffers_asm(fb, fb_pixels, fb_width, fb_height, fb_bpp);
nuclear@10: #else
nuclear@6: 		swap_buffers();
nuclear@10: #endif
nuclear@1: 	}
nuclear@1: }
nuclear@1: 
nuclear@1: #define PACK_RGB(r, g, b) \
nuclear@1: 	((((r) << rshift) & rmask) | \
nuclear@1: 	 (((g) << gshift) & gmask) | \
nuclear@1: 	 (((b) << bshift) & bmask))
nuclear@1: 
nuclear@11: #define UNPACK_RED(c)	(((c) >> 16) & 0xff)
nuclear@9: #define UNPACK_GREEN(c)	(((c) >> 8) & 0xff)
nuclear@11: #define UNPACK_BLUE(c)	((c) & 0xff)
nuclear@9: 
nuclear@1: static void swap_buffers()
nuclear@1: {
nuclear@9: 	uint32_t *src = fb_pixels;
nuclear@9: 	int num_pixels = fb_width * fb_height;
nuclear@1: 
nuclear@9: 	switch(fb_bpp) {
nuclear@1: 	case 32:
nuclear@9: 		memcpy(fb, fb_pixels, num_pixels * 4);
nuclear@1: 		break;
nuclear@1: 
nuclear@1: 	case 24:
nuclear@9: 		{
nuclear@9: 			unsigned char *dest = (unsigned char*)fb;
nuclear@10: 			for(int i=0; i<num_pixels-1; i++) {
nuclear@10: 				*((uint32_t*)dest) = *src++;
nuclear@10: 				dest += 3;
nuclear@9: 			}
nuclear@10: 			*dest++ = UNPACK_RED(*src);
nuclear@10: 			*dest++ = UNPACK_GREEN(*src);
nuclear@10: 			*dest++ = UNPACK_BLUE(*src);
nuclear@9: 		}
nuclear@1: 		break;
nuclear@1: 
nuclear@1: 	case 16:
nuclear@1: 		{
nuclear@1: 			uint16_t *dest = (uint16_t*)fb;
nuclear@1: 			for(int i=0; i<num_pixels; i++) {
nuclear@9: 				uint32_t c = *src++;
nuclear@9: 				unsigned char r = UNPACK_RED(c);
nuclear@9: 				unsigned char g = UNPACK_GREEN(c);
nuclear@9: 				unsigned char b = UNPACK_BLUE(c);
nuclear@9: 
nuclear@11: 				*dest++ = (((r) << 8) & 0xf800) |
nuclear@11: 					(((g) << 3) & 0x7e0) |
nuclear@11: 					(((b) >> 3) & 0x1f);
nuclear@1: 			}
nuclear@1: 		}
nuclear@1: 		break;
nuclear@1: 
nuclear@1: 	default:
nuclear@1: 		break;
nuclear@1: 	}
nuclear@1: }
nuclear@1: 
nuclear@9: static void draw_cursor(uint32_t *buf, int mx, int my)
nuclear@7: {
nuclear@9: 	uint32_t *cptr = buf + my * fb_width + mx;
nuclear@7: 	int i, cw[2] = {4, 4}, ch[2] = {4, 4};
nuclear@7: 
nuclear@7: 	if(mx < cw[0]) cw[0] = mx;
nuclear@7: 	if(my < ch[0]) ch[0] = my;
nuclear@9: 	if(fb_width - mx < cw[1]) cw[1] = fb_width - mx - 1;
nuclear@9: 	if(fb_height - my < ch[1]) ch[1] = fb_height - my - 1;
nuclear@7: 
nuclear@7: 	for(i=1; i<cw[0]; i++) {
nuclear@9: 		int idx = -i;
nuclear@9: 		cptr[idx] = 0xffffff;
nuclear@7: 	}
nuclear@7: 	for(i=1; i<cw[1]; i++) {
nuclear@9: 		int idx = i;
nuclear@9: 		cptr[idx] = 0xffffff;
nuclear@7: 	}
nuclear@7: 	for(i=1; i<ch[0]; i++) {
nuclear@9: 		int idx = -i * fb_width;
nuclear@9: 		cptr[idx] = 0xffffff;
nuclear@7: 	}
nuclear@7: 	for(i=1; i<ch[1]; i++) {
nuclear@9: 		int idx = i * fb_width;
nuclear@9: 		cptr[idx] = 0xffffff;
nuclear@7: 	}
nuclear@7: }
nuclear@7: 
nuclear@1: static void handle_keyboard()
nuclear@1: {
nuclear@5: 	int key;
nuclear@9: 	Screen *scr = active_screen();
nuclear@1: 
nuclear@6: 	if(novideo) return;
nuclear@6: 
nuclear@5: 	while((key = kb_getkey()) != -1) {
nuclear@10: 		switch(key) {
nuclear@10: 		case '`':
nuclear@10: 			use_asm_swap = !use_asm_swap;
nuclear@10: 			break;
nuclear@10: 
nuclear@10: 		default:
nuclear@10: 			break;
nuclear@10: 		}
nuclear@9: 		scr->handle_keyboard(key, true);	// TODO also generate release events...
nuclear@1: 	}
nuclear@1: }
nuclear@1: 
nuclear@1: static void handle_mouse()
nuclear@1: {
nuclear@9: 	static int prev_mx, prev_my, prev_bnmask, bndiff;
nuclear@9: 	int mx, my, bnmask;
nuclear@9: 	Screen *scr = active_screen();
nuclear@7: 
nuclear@7: 	if(!use_mouse || novideo) return;
nuclear@7: 
nuclear@7: 	bnmask = read_mouse(&mx, &my);
nuclear@9: 	if(scr && (bndiff = bnmask ^ prev_bnmask)) {
nuclear@9: 		for(int i=0; i<8; i++) {
nuclear@9: 			int bit = 1 << i;
nuclear@9: 			if(bndiff & bit) {
nuclear@9: 				scr->handle_mbutton(i, bnmask & bit, mx, my);
nuclear@9: 			}
nuclear@9: 		}
nuclear@7: 	}
nuclear@9: 	prev_bnmask = bnmask;
nuclear@9: 
nuclear@9: 	if(scr && (mx != prev_mx || my != prev_my)) {
nuclear@9: 		scr->handle_mmotion(mx, my);
nuclear@7: 	}
nuclear@9: 	prev_mx = mx;
nuclear@9: 	prev_my = my;
nuclear@10: 
nuclear@10: 	mouse_x = mx;
nuclear@10: 	mouse_y = my;
nuclear@7: }
nuclear@7: 
nuclear@1: 
nuclear@1: static struct {
nuclear@1: 	int opt;
nuclear@1: 	const char *lopt;
nuclear@1: 	const char *desc;
nuclear@1: } options[] = {
nuclear@1: 	{'s', "size", "resolution <xres>x<yres>[:bpp]"},
nuclear@6: 	{'n', "novid", "don't switch video mode (for debugging)"},
nuclear@1: 	{'h', "help", "print usage information and exit"},
nuclear@1: 	{-1, 0, 0}
nuclear@1: };
nuclear@1: 
nuclear@1: static void print_usage(const char *argv0)
nuclear@1: {
nuclear@1: 	printf("%s usage\n", argv0);
nuclear@1: 	for(int i=0; options[i].opt != -1; i++) {
nuclear@1: 		printf("  -%c, -%s: %s\n", options[i].opt, options[i].lopt, options[i].desc);
nuclear@1: 	}
nuclear@1: 	exit(0);
nuclear@1: }
nuclear@1: 
nuclear@1: static bool parse_args(int argc, char **argv)
nuclear@1: {
nuclear@1: 	for(int i=1; i<argc; i++) {
nuclear@1: 		if(argv[i][0] == '-') {
nuclear@1: 			int opt = -1;
nuclear@1: 
nuclear@1: 			for(int j=0; options[j].opt != -1; j++) {
nuclear@1: 				if(argv[i][2] == 0) {
nuclear@1: 					if(argv[i][1] == options[j].opt) {
nuclear@1: 						opt = options[j].opt;
nuclear@1: 						break;
nuclear@1: 					}
nuclear@1: 				} else {
nuclear@1: 					if(strcmp(argv[i] + 1, options[j].lopt) == 0) {
nuclear@1: 						opt = options[j].opt;
nuclear@1: 						break;
nuclear@1: 					}
nuclear@1: 				}
nuclear@1: 			}
nuclear@1: 
nuclear@1: 			switch(opt) {
nuclear@1: 			case 's':
nuclear@9: 				if(sscanf(argv[++i], "%dx%d:%d", &fb_width, &fb_height, &fb_bpp) < 2) {
nuclear@1: 					fprintf(stderr, "%s must be followed by a resolution: WxH\n", argv[i - 1]);
nuclear@1: 					return false;
nuclear@1: 				}
nuclear@1: 				break;
nuclear@1: 
nuclear@6: 			case 'n':
nuclear@6: 				novideo = true;
nuclear@6: 				break;
nuclear@6: 
nuclear@1: 			case 'h':
nuclear@1: 				print_usage(argv[0]);	// doesn't return
nuclear@1: 				break;
nuclear@1: 
nuclear@1: 			default:
nuclear@1: 				fprintf(stderr, "unknown option: %s\n", argv[i]);
nuclear@1: 				return false;
nuclear@1: 			}
nuclear@1: 		} else {
nuclear@1: 			fprintf(stderr, "unexpected argument: %s\n", argv[i]);
nuclear@1: 			return false;
nuclear@1: 		}
nuclear@1: 	}
nuclear@1: 	return true;
nuclear@1: }
nuclear@7: 
nuclear@7: static void sig(int s)
nuclear@7: {
nuclear@7: 	cleanup();
nuclear@7: 	fprintf(stderr, "signal caught: %d\n", s);
nuclear@9: 	exit(1);
nuclear@7: }