/* bootpackのメイン */

#include "bootpack.h"
#include <stdio.h>

void make_window8(unsigned char *buf, int xsize, int ysize, char *title, char act);
void putfouts8_asc_sht(struct SHEET *sht, int x, int y, int c, int b, char *s, int l);
void make_textbox8(struct SHEET *sht, int x0, int y0, int sx, int sy, int c);
void make_wtitle8(unsigned char *buf, int xsize, char *title, char act);
void console_task(struct SHEET *sheet);

#define KEYCMD_LED		0xed

void HariMain(void)
{
	struct BOOTINFO *binfo = (struct BOOTINFO *) ADR_BOOTINFO;
	struct SHTCTL *shtctl;
	char s[40];
	struct FIFO32 fifo, keycmd;
	int fifobuf[128], keycmd_buf[32];
	int mx, my, i, cursor_x, cursor_c;
	unsigned int memtotal;
	struct MOUSE_DEC mdec;
	struct MEMMAN *memman = (struct MEMMAN *) MEMMAN_ADDR;
	unsigned char *buf_back, buf_mouse[256], *buf_win, *buf_cons;
	struct SHEET *sht_back, *sht_mouse, *sht_win, *sht_cons;
	struct TASK *task_a, *task_cons;
	struct TIMER *timer;
	static char keytable0[0x80] = {
		0,   0,   '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '-', '^', 0,   0,
		'Q', 'W', 'E', 'R', 'T', 'Y', 'U', 'I', 'O', 'P', '@', '[', 0,   0,   'A', 'S',
		'D', 'F', 'G', 'H', 'J', 'K', 'L', ';', ':', 0,   0,   ']', 'Z', 'X', 'C', 'V',
		'B', 'N', 'M', ',', '.', '/', 0,   '*', 0,   ' ', 0,   0,   0,   0,   0,   0,
		0,   0,   0,   0,   0,   0,   0,   '7', '8', '9', '-', '4', '5', '6', '+', '1',
		'2', '3', '0', '.', 0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
		0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
		0,   0,   0,   0x5c, 0,  0,   0,   0,   0,   0,   0,   0,   0,   0x5c, 0,  0
	};
	static char keytable1[0x80] = {
		0,   0,   '!', 0x22, '#', '$', '%', '&', 0x27, '(', ')', '~', '=', '~', 0,   0,
		'Q', 'W', 'E', 'R', 'T', 'Y', 'U', 'I', 'O', 'P', '`', '{', 0,   0,   'A', 'S',
		'D', 'F', 'G', 'H', 'J', 'K', 'L', '+', '*', 0,   0,   '}', 'Z', 'X', 'C', 'V',
		'B', 'N', 'M', '<', '>', '?', 0,   '*', 0,   ' ', 0,   0,   0,   0,   0,   0,
		0,   0,   0,   0,   0,   0,   0,   '7', '8', '9', '-', '4', '5', '6', '+', '1',
		'2', '3', '0', '.', 0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
		0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
		0,   0,   0,   '_', 0,   0,   0,   0,   0,   0,   0,   0,   0,   '|', 0,   0
	};
	int key_to = 0, key_shift = 0, key_leds = (binfo->leds >> 4) & 7, keycmd_wait = -1;

	init_gdtidt();
	init_pic();
	io_sti(); /* IDT/PICの初期化が終わったのでCPUの割り込み禁止を解除 */
	fifo32_init(&fifo, 128, fifobuf, 0);
	init_pit();
	init_keyboard(&fifo, 256);
	enable_mouse(&fifo, 512, &mdec);
	io_out8(PIC0_IMR, 0xf8); /* PITとPIC1とキーボードを許可(11111000) */
	io_out8(PIC1_IMR, 0xef); /* マウスを許可(11101111) */
	fifo32_init(&keycmd, 32, keycmd_buf, 0);

	memtotal = memtest(0x00400000, 0xbfffffff);
	memman_init(memman);
	memman_free(memman, 0x00001000, 0x0009e000); /* 0x00001000 - 0x0009efff */
	memman_free(memman, 0x00400000, memtotal - 0x00400000);

	init_palette();
	shtctl = shtctl_init(memman, binfo->vram, binfo->scrnx, binfo->scrny);
	task_a = task_init(memman);
	fifo.task = task_a;
	task_run(task_a, 1, 2);

	/* sht_back */
	sht_back  = sheet_alloc(shtctl);
	buf_back  = (unsigned char *) memman_alloc_4k(memman, binfo->scrnx * binfo->scrny);
	sheet_setbuf(sht_back, buf_back, binfo->scrnx, binfo->scrny, -1); /* 透明色なし */
	init_screen8(buf_back, binfo->scrnx, binfo->scrny);

	/* sht_cons */
	sht_cons = sheet_alloc(shtctl);
	buf_cons = (unsigned char *) memman_alloc_4k(memman, 256 * 165);
	sheet_setbuf(sht_cons, buf_cons, 256, 165, -1); /* 透明色なし */
	make_window8(buf_cons, 256, 165, "console", 0);
	make_textbox8(sht_cons, 8, 28, 240, 128, COL8_000000);
	task_cons = task_alloc();
	task_cons->tss.esp = memman_alloc_4k(memman, 64 * 1024) + 64 * 1024 - 8;
	task_cons->tss.eip = (int) &console_task;
	task_cons->tss.es = 1 * 8;
	task_cons->tss.cs = 2 * 8;
	task_cons->tss.ss = 1 * 8;
	task_cons->tss.ds = 1 * 8;
	task_cons->tss.fs = 1 * 8;
	task_cons->tss.gs = 1 * 8;
	*((int *) (task_cons->tss.esp + 4)) = (int) sht_cons;
	task_run(task_cons, 2, 2); /* level=2, priority=2 */

	/* sht_win */
	sht_win   = sheet_alloc(shtctl);
	buf_win   = (unsigned char *) memman_alloc_4k(memman, 160 * 52);
	sheet_setbuf(sht_win, buf_win, 144, 52, -1); /* 透明色なし */
	make_window8(buf_win, 144, 52, "task_a", 1);
	make_textbox8(sht_win, 8, 28, 128, 16, COL8_FFFFFF);
	cursor_x = 8;
	cursor_c = COL8_FFFFFF;
	timer = timer_alloc();
	timer_init(timer, &fifo, 1);
	timer_settime(timer, 50);

	/* sht_mouse */
	sht_mouse = sheet_alloc(shtctl);
	sheet_setbuf(sht_mouse, buf_mouse, 16, 16, 99);
	init_mouse_cursor8(buf_mouse, 99);
	mx = (binfo->scrnx - 16) / 2; /* 画面中央になるように座標計算 */
	my = (binfo->scrny - 28 - 16) / 2;

	sheet_slide(sht_back,  0,  0);
	sheet_slide(sht_cons, 32,  4);
	sheet_slide(sht_win,  64, 56);
	sheet_slide(sht_mouse, mx, my);
	sheet_updown(sht_back,  0);
	sheet_updown(sht_cons,  1);
	sheet_updown(sht_win,   2);
	sheet_updown(sht_mouse, 3);
	sprintf(s, "(%3d, %3d)", mx, my);
	putfouts8_asc_sht(sht_back, 0, 0, COL8_FFFFFF, COL8_008484, s, 10);
	sprintf(s, "memory %dMB   free : %dKB",
			memtotal / (1024 * 1024), memman_total(memman) / 1024);
	putfouts8_asc_sht(sht_back, 0, 32, COL8_FFFFFF, COL8_008484, s, 40);

	/* 最初にキーボード状態との食い違いがないように、設定しておくことにする */
	fifo32_put(&keycmd, KEYCMD_LED);
	fifo32_put(&keycmd, key_leds);

	while (1) {

		if (fifo32_status(&keycmd) > 0 && keycmd_wait < 0) {
			/* キーボードコントローラに送るデータがあれば、送る */
			keycmd_wait = fifo32_get(&keycmd);
			wait_KBC_sendready();
			io_out8(PORT_KEYDAT, keycmd_wait);
		}

		io_cli();

		// FIFOバッファにデータが入っていなければタスクを休ませる
		if (fifo32_status(&fifo) == 0) {
			task_sleep(task_a);
			io_sti();
			continue;
		}

		// FIFOバッファにあるデータを変数 i へ
		i = fifo32_get(&fifo);

		io_sti();

		// -----------------------------------------
		// キーボード入力のための割り込み
		// -----------------------------------------
		if (256 <= i && i <= 511) {
			sprintf(s, "%02X", i - 256);
			putfouts8_asc_sht(sht_back, 0, 16, COL8_FFFFFF, COL8_008484, s, 2);
			if (i < 0x80 + 256) { // キーコードを文字コードに変換
				if (key_shift == 0)
					s[0] = keytable0[i - 256];
				else
					s[0] = keytable1[i - 256];
			} else {
				s[0] = 0;
			}

			// 入力された文字はアルファベット
			if ('A' <= s[0] && s[0] <= 'Z') {
				// 大文字を小文字に変換
				if (((key_leds & 4) == 0 && key_shift == 0) ||
						((key_leds & 4) != 0 && key_shift != 0)) {
					s[0] += 0x20;
				}
			}

			if (s[0] != 0) { /* 通常文字 */
				if (key_to == 0) {	/* タスクAへ */
					if (cursor_x < 128) {
						/* 一文字表示してから、カーソルを1つ進める */
						s[1] = 0;
						putfouts8_asc_sht(sht_win, cursor_x, 28, COL8_000000, COL8_FFFFFF, s, 1);
						cursor_x += 8;
					}
				} else {	/* コンソールへ */
					fifo32_put(&task_cons->fifo, s[0] + 256);
				}
			}

			switch (i)
			{
			// バックスペース
			case (256 + 0x0e):
				// タスクＡで処理する
				if (key_to == 0)
					if (cursor_x > 8) {
						// カーソルをスペースで消してから、カーソルを1つ戻す
						putfouts8_asc_sht(sht_win, cursor_x, 
							28, COL8_000000, COL8_FFFFFF, " ", 1);
						cursor_x -= 8;
					}
				// コンソールへ渡す
				else 
					fifo32_put(&task_cons->fifo, 8 + 256);
				break;

			// タブ（TAB）キー
			case (256 + 0x0f):
				// アクティブなウィンドウを変更
				make_wtitle8(buf_win,  sht_win->bxsize,  "task_a",  key_to == 0 ? 0 : 1);
				make_wtitle8(buf_cons, sht_cons->bxsize, "console", key_to == 0 ? 1 : 0);
				key_to = (key_to == 0 ? 1 : 0);
				sheet_refresh(sht_win,  0, 0, sht_win->bxsize,  21);
				sheet_refresh(sht_cons, 0, 0, sht_cons->bxsize, 21);
				break;

			// 左シフトON
			case (256 + 0x2a):
				key_shift |= 1;
				break;

			// 右シフトON
			case (256 + 0x36):
				key_shift |= 2;
				break;

			// 左シフトOFF
			case (256 + 0xaa):
				key_shift &= ~1;
				break;

			// 右シフトOFF
			case (256 + 0xb6):
				key_shift &= ~2;
				break;

			// CapsLock
			case (256 + 0x3a):
				key_leds ^= 4;
				fifo32_put(&keycmd, KEYCMD_LED);
				fifo32_put(&keycmd, key_leds);
				break;

			// NumLock
			case (256 + 0x45):
				key_leds ^= 2;
				fifo32_put(&keycmd, KEYCMD_LED);
				fifo32_put(&keycmd, key_leds);
				break;

			// ScrollLock
			case (256 + 0x46):
				key_leds ^= 1;
				fifo32_put(&keycmd, KEYCMD_LED);
				fifo32_put(&keycmd, key_leds);
				break;

			// キーボードがデータを無事に受け取った
			case (256 + 0xfa):
				keycmd_wait = -1;
				break;

			// キーボードがデータを無事に受け取れなかった
			case (256 + 0xfe):
				wait_KBC_sendready();
				io_out8(PORT_KEYDAT, keycmd_wait);
				break;
			}

			/* カーソルの再表示 */
			boxfill8(sht_win->buf, sht_win->bxsize, cursor_c, cursor_x, 28, cursor_x + 7, 43);
			sheet_refresh(sht_win, cursor_x, 28, cursor_x + 8, 44);

		} else if (512 <= i && i <= 767) { /* マウスデータ */
			if (mouse_decode(&mdec, i - 512) != 0) {
				/* データが3バイト揃ったので表示 */
				sprintf(s, "[lcr %4d %4d]", mdec.x, mdec.y);
				if ((mdec.btn & 0x01) != 0) {
					s[1] = 'L';
				}
				if ((mdec.btn & 0x02) != 0) {
					s[3] = 'R';
				}
				if ((mdec.btn & 0x04) != 0) {
					s[2] = 'C';
				}
				putfouts8_asc_sht(sht_back, 32, 16, COL8_FFFFFF, COL8_008484, s, 15);
				/* マウスカーソルの移動 */
				mx += mdec.x;
				my += mdec.y;
				if (mx < 0) {
					mx = 0;
				}
				if (my < 0) {
					my = 0;
				}
				if (mx > binfo->scrnx - 1) {
					mx = binfo->scrnx - 1;
				}
				if (my > binfo->scrny - 1) {
					my = binfo->scrny - 1;
				}
				sprintf(s, "(%3d, %3d)", mx, my);
				putfouts8_asc_sht(sht_back, 0, 0, COL8_FFFFFF, COL8_008484, s, 10);
				sheet_slide(sht_mouse, mx, my);
				if ((mdec.btn & 0x01) != 0) {
					/* 左ボタンを押していたら、sht_winを動かす */
					sheet_slide(sht_win, mx - 80, my - 8);
				}
			}
		} else if (i <= 1) { /* カーソル用タイマ */
			if (i != 0) {
				timer_init(timer, &fifo, 0); /* 次は0を */
				cursor_c = COL8_000000;
			} else {
				timer_init(timer, &fifo, 1); /* 次は1を */
				cursor_c = COL8_FFFFFF;
			}
			timer_settime(timer, 50);
			boxfill8(sht_win->buf, sht_win->bxsize, cursor_c, cursor_x, 28, cursor_x + 7, 43);
			sheet_refresh(sht_win, cursor_x, 28, cursor_x + 8, 44);
		}
	}
}

/* --------------------------------------------------------------------
 * ウィンドウを生成する関数
--------------------------------------------------------------------- */
void make_window8(unsigned char *buf, int xsize, int ysize, char *title, char act)
{
	boxfill8(buf, xsize, COL8_C6C6C6, 0,         0,         xsize - 1, 0        );
	boxfill8(buf, xsize, COL8_FFFFFF, 1,         1,         xsize - 2, 1        );
	boxfill8(buf, xsize, COL8_C6C6C6, 0,         0,         0,         ysize - 1);
	boxfill8(buf, xsize, COL8_FFFFFF, 1,         1,         1,         ysize - 2);
	boxfill8(buf, xsize, COL8_848484, xsize - 2, 1,         xsize - 2, ysize - 2);
	boxfill8(buf, xsize, COL8_000000, xsize - 1, 0,         xsize - 1, ysize - 1);
	boxfill8(buf, xsize, COL8_C6C6C6, 2,         2,         xsize - 3, ysize - 3);
	boxfill8(buf, xsize, COL8_848484, 1,         ysize - 2, xsize - 2, ysize - 2);
	boxfill8(buf, xsize, COL8_000000, 0,         ysize - 1, xsize - 1, ysize - 1);
	make_wtitle8(buf, xsize, title, act);
	return;
}

/* --------------------------------------------------------------------
 * ウィンドウタイトルを生成する関数
--------------------------------------------------------------------- */
void make_wtitle8(unsigned char *buf, int xsize, char *title, char act)
{
	// ×ボタンの色関係情報
	static char closebtn[14][16] = {
		"OOOOOOOOOOOOOOO@",
		"OQQQQQQQQQQQQQ$@",
		"OQQQQQQQQQQQQQ$@",
		"OQQQ@@QQQQ@@QQ$@",
		"OQQQQ@@QQ@@QQQ$@",
		"OQQQQQ@@@@QQQQ$@",
		"OQQQQQQ@@QQQQQ$@",
		"OQQQQQ@@@@QQQQ$@",
		"OQQQQ@@QQ@@QQQ$@",
		"OQQQ@@QQQQ@@QQ$@",
		"OQQQQQQQQQQQQQ$@",
		"OQQQQQQQQQQQQQ$@",
		"O$$$$$$$$$$$$$$@",
		"@@@@@@@@@@@@@@@@"
	};
	int x, y;
	char c, tc, tbc;

	if (act != 0) {        // ウィンドウがアクティブ
		tc = COL8_FFFFFF;  // タイトル文字の色
		tbc = COL8_000084; // タイトルバーの色

	} else {               // ウィンドウがアクティブじゃない
		tc = COL8_C6C6C6;  // タイトル文字の色
		tbc = COL8_848484; // タイトルバーの色
	}

	// タイトルを書く
	boxfill8(buf, xsize, tbc, 3, 3, xsize - 4, 20);
	putfonts8_asc(buf, xsize, 24, 4, tc, title);

	// ×ボタンを色情報に変換
	for (y = 0; y < 14; y++) {
		for (x = 0; x < 16; x++) {
			c = closebtn[y][x];
			switch (c)
			{
			case '@':
				c = COL8_000000;
				break;
			case '$':
				c = COL8_848484;
				break;
			case 'Q':
				c = COL8_C6C6C6;
				break;
			default:
				c = COL8_FFFFFF;
			}

			buf[(5 + y) * xsize + (xsize - 21 + x)] = c;
		}
	}
	return;
}

void putfouts8_asc_sht(struct SHEET *sht, int x, int y, int c, int b, char *s, int l)
{
	boxfill8(sht->buf, sht->bxsize, b, x, y, x + l * 8 - 1, y + 15);
	putfonts8_asc(sht->buf, sht->bxsize, x, y, c, s);
	sheet_refresh(sht, x, y, x + l * 8, y + 16);
	return;
}

void make_textbox8(struct SHEET *sht, int x0, int y0, int sx, int sy, int c)
{
	int x1 = x0 + sx, y1 = y0 + sy;
	boxfill8(sht->buf, sht->bxsize, COL8_848484, x0 - 2, y0 - 3, x1 + 1, y0 - 3);
	boxfill8(sht->buf, sht->bxsize, COL8_848484, x0 - 3, y0 - 3, x0 - 3, y1 + 1);
	boxfill8(sht->buf, sht->bxsize, COL8_FFFFFF, x0 - 3, y1 + 2, x1 + 1, y1 + 2);
	boxfill8(sht->buf, sht->bxsize, COL8_FFFFFF, x1 + 2, y0 - 3, x1 + 2, y1 + 2);
	boxfill8(sht->buf, sht->bxsize, COL8_000000, x0 - 1, y0 - 2, x1 + 0, y0 - 2);
	boxfill8(sht->buf, sht->bxsize, COL8_000000, x0 - 2, y0 - 2, x0 - 2, y1 + 0);
	boxfill8(sht->buf, sht->bxsize, COL8_C6C6C6, x0 - 2, y1 + 1, x1 + 0, y1 + 1);
	boxfill8(sht->buf, sht->bxsize, COL8_C6C6C6, x1 + 1, y0 - 2, x1 + 1, y1 + 1);
	boxfill8(sht->buf, sht->bxsize, c,           x0 - 1, y0 - 1, x1 + 0, y1 + 0);
	return;
}

void console_task(struct SHEET *sheet)
{
	struct TIMER *timer;
	struct TASK *task = task_now();
	int i, fifobuf[128], cursor_x = 16, cursor_c = COL8_000000;
	char s[2];

	// FIFOバッファを生成し、カーソル用タイマをセット
	fifo32_init(&task->fifo, 128, fifobuf, task);
	timer = timer_alloc();
	timer_init(timer, &task->fifo, 1);
	timer_settime(timer, 50);

	/* プロンプト表示 */
	putfouts8_asc_sht(sheet, 8, 28, COL8_FFFFFF, COL8_000000, ">", 1);

	while (1) {

		io_cli();

		// FIFOバッファにデータが入っていなければタスクを休ませる
		if (fifo32_status(&task->fifo) == 0) {
			task_sleep(task);
			io_sti();
			continue;
		}
		
		// FIFOバッファにあるデータを変数 i へ
		i = fifo32_get(&task->fifo);

		io_sti();

		// -----------------------------------------
		// カーソルの点滅のためのタイマ処理割り込み
		// -----------------------------------------
		if (i <= 1) {
			if (i != 0) {
				timer_init(timer, &task->fifo, 0); /* 次は0を */
				cursor_c = COL8_FFFFFF;
			} else {
				timer_init(timer, &task->fifo, 1); /* 次は1を */
				cursor_c = COL8_000000;
			}
			timer_settime(timer, 50);
		}

		// -----------------------------------------
		// キーボード入力のための割り込み
		// -----------------------------------------
		if (256 <= i && i <= 511) {
			if (i == 8 + 256) {
				/* バックスペース */
				if (cursor_x > 16) {
					/* カーソルをスペースで消してから、カーソルを1つ戻す */
					putfouts8_asc_sht(sheet, cursor_x, 28, COL8_FFFFFF, COL8_000000, " ", 1);
					cursor_x -= 8;
				}
			} else {
				/* 一般文字 */
				if (cursor_x < 240) {
					/* 一文字表示してから、カーソルを1つ進める */
					s[0] = i - 256;
					s[1] = 0;
					putfouts8_asc_sht(sheet, cursor_x, 28, COL8_FFFFFF, COL8_000000, s, 1);
					cursor_x += 8;
				}
			}
		}
		/* カーソル再表示 */
		boxfill8(sheet->buf, sheet->bxsize, cursor_c, cursor_x, 28, cursor_x + 7, 43);
		sheet_refresh(sheet, cursor_x, 28, cursor_x + 8, 44);
	}
}