nuclear@1: #include <stdio.h>
nuclear@1: #include <stdlib.h>
nuclear@1: #include <string.h>
nuclear@1: #include <math.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@1: 
nuclear@1: static void display();
nuclear@1: static void swap_buffers();
nuclear@1: static void handle_keyboard();
nuclear@1: static void handle_mouse();
nuclear@1: static bool parse_args(int argc, char **argv);
nuclear@1: 
nuclear@1: static int xsz = 800;
nuclear@1: static int ysz = 600;
nuclear@1: static int bpp = 16;
nuclear@1: static int bytespp;
nuclear@1: static unsigned char *fb;
nuclear@1: static unsigned char *backbuf;
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@1: static bool quit;
nuclear@1: 
nuclear@1: int main(int argc, char **argv)
nuclear@1: {
nuclear@1: 	if(!parse_args(argc, argv)) {
nuclear@1: 		return 1;
nuclear@1: 	}
nuclear@1: 	if(kb_init(32) == -1) {
nuclear@1: 		fprintf(stderr, "failed to initialize keyboard driver\n");
nuclear@1: 		return 1;
nuclear@1: 	}
nuclear@1: 	if(!(fb = (unsigned char*)set_video_mode(xsz, ysz, bpp))) {
nuclear@1: 		set_text_mode();
nuclear@1: 		fprintf(stderr, "failed to set video mode: %dx%d %dbpp\n", xsz, ysz, bpp);
nuclear@1: 		return 1;
nuclear@1: 	}
nuclear@1: 	bpp = get_color_depth();
nuclear@1: 	get_color_bits(&rbits, &gbits, &bbits);
nuclear@1: 	get_color_shift(&rshift, &gshift, &bshift);
nuclear@1: 	get_color_mask(&rmask, &gmask, &bmask);
nuclear@1: 	bytespp = (int)ceil(bpp / 8.0);
nuclear@1: 
nuclear@1: 	printlog("bpp: %d (%d %d %d)\n", bpp, rbits, gbits, bbits);
nuclear@1: 	printlog("shift: %d %d %d\n", rshift, gshift, bshift);
nuclear@1: 	printlog("mask: %x %x %x\n", rmask, gmask, bmask);
nuclear@1: 
nuclear@1: 	backbuf = new unsigned char[xsz * ysz * 3];
nuclear@1: 
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@1: 	}
nuclear@1: 
nuclear@1: 	delete [] backbuf;
nuclear@1: 
nuclear@1: 	set_text_mode();
nuclear@1: 	kb_shutdown();
nuclear@1: 
nuclear@1: 	printf("Thank you for using Rayzor!\n");
nuclear@1: 	return 0;
nuclear@1: }
nuclear@1: 
nuclear@1: static void display()
nuclear@1: {
nuclear@1: 	unsigned char *fbptr = backbuf;
nuclear@1: 
nuclear@1: 	for(int i=0; i<ysz; i++) {
nuclear@1: 		for(int j=0; j<xsz; j++) {
nuclear@1: 			bool chess = ((i / 16) & 1) == ((j / 16) & 1);
nuclear@1: 			fbptr[chess ? 0 : 2] = 255;
nuclear@1: 			fbptr[1] = 128;
nuclear@1: 			fbptr[chess ? 2 : 0] = 32;
nuclear@1: 			fbptr += 3;
nuclear@1: 		}
nuclear@1: 	}
nuclear@1: 
nuclear@1: 	swap_buffers();
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@1: static void swap_buffers()
nuclear@1: {
nuclear@1: 	unsigned char *src = backbuf;
nuclear@1: 	int num_pixels = xsz * ysz;
nuclear@1: 
nuclear@1: 	switch(bpp) {
nuclear@1: 	case 32:
nuclear@1: 		{
nuclear@1: 			uint32_t *dest = (uint32_t*)fb;
nuclear@1: 			for(int i=0; i<num_pixels; i++) {
nuclear@1: 				*dest++ = PACK_RGB(src[0], src[1], src[2]);
nuclear@1: 				src += 3;
nuclear@1: 			}
nuclear@1: 		}
nuclear@1: 		break;
nuclear@1: 
nuclear@1: 	case 24:
nuclear@1: 		memcpy(fb, backbuf, num_pixels * 3);
nuclear@1: 		break;
nuclear@1: 
nuclear@1: 	case 16:
nuclear@1: 	case 15:
nuclear@1: 		{
nuclear@1: 			int srs = 8 - rbits;
nuclear@1: 			int sgs = 8 - gbits;
nuclear@1: 			int sbs = 8 - bbits;
nuclear@1: 			uint16_t *dest = (uint16_t*)fb;
nuclear@1: 			for(int i=0; i<num_pixels; i++) {
nuclear@1: 				*dest++ = PACK_RGB(src[0] >> srs, src[1] >> sgs, src[2] >> sbs);
nuclear@1: 				src += 3;
nuclear@1: 			}
nuclear@1: 		}
nuclear@1: 		break;
nuclear@1: 
nuclear@1: 	default:
nuclear@1: 		break;
nuclear@1: 	}
nuclear@1: }
nuclear@1: 
nuclear@1: static void handle_keyboard()
nuclear@1: {
nuclear@1: 	if(!kb_isdown(KB_ANY))
nuclear@1: 		return;
nuclear@1: 
nuclear@1: 	int c = kb_getkey();
nuclear@1: 	switch(c) {
nuclear@1: 	case 27:
nuclear@1: 		quit = true;
nuclear@1: 		return;
nuclear@1: 	}
nuclear@1: }
nuclear@1: 
nuclear@1: static void handle_mouse()
nuclear@1: {
nuclear@1: }
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@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@1: 				if(sscanf(argv[++i], "%dx%d:%d", &xsz, &ysz, &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@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: }