#include <ncurses.h>
#include <string.h>		// strlen()

/*
ペンをキー入力でコントロールする．
pen：ペンの状態（参照渡し）
x, y：ペンの位置（参照渡し）
w, h：画面サイズ
key：入力キー
*/
void Ctrl(int *pen, int *x, int *y, int w, int h, int key)
{
	switch (key) {
	case '-': *pen = 0; break;		// on
	case '+': *pen = 1; break;		// off
	case ' ': *pen = (*pen + 1)%2; break;	// on/off切替

	case 'h': (*x)--; if (*x < 0) *x = 0; break;		// ←
	case 'j': (*y)++; if (*y >= h) *y = h - 1; break;	// ↓
	case 'k': (*y)--; if (*y < 0) *y = 0; break;		// ↑
	case 'l': (*x)++; if (*x >= w) *x = w - 1; break;	// →
	}
}

/*
メッセージを画面中央に表示し，キー入力を待つ．
msg：メッセージ文字列
w, h：画面サイズ
return：入力キー
*/
int Info(char *msg, int w, int h)
{
	mvaddstr(h/2, (w - strlen(msg))/2, msg);
	refresh();
	return (getch());
}

/*
ゲーム本体
*/
void Game(void)
{
	int	pen;	// ペンの状態（0:off，1:on）が入るよ
	int	x, y;	// ペンの位置が入るよ
	int	w, h;	// 画面のサイズが入るよ
	int	key;	// 入力キーの文字が入るよ

	// 初期設定
	erase();			// 画面をクリア
	getmaxyx(stdscr, h, w);		// 画面サイズを取得
	pen = 0; x = w/2; y = h/2;	// ペンの状態・位置を初期化

	while (1) {
		// 描画
//		erase();		// 画面を消去しなければ軌跡が残るよ
		if (pen) mvaddch(y, x, '#');	// ペンのキャラクタを描画
		refresh();		// 画面を表示

		// 入力
		key = getch();		// 端末からキーを入力

		// ペンの操作
		Ctrl(&pen, &x, &y, w, h, key);

		if (key == 'q') break;	// [Q]で終了
	}
	Info("Done. Hit any key!", w, h);	// 終了確認
}

/*
タイトル画面を表示し，キー入力を待つ．
return：入力キー
*/
int Title(void)
{
	erase();
	mvaddstr( 1, 3, "PenDraw");
	mvaddstr( 3, 3, "Key:");
	mvaddstr( 4, 3, "  [ ][+][-] - pen on/off");
	mvaddstr( 5, 3, "  [H][J][K][L] - move");
	mvaddstr( 6, 3, "  [Q] - quit");
	refresh();
	return (getch());
}

int main(void)
{
	int	key;	// 入力キーの文字が入るよ

	// 初期設定
	initscr();	// 端末制御を開始
	noecho();	// 入力キーは表示するなよ
	cbreak();	// 入力バッファは使うなよ
	curs_set(0);	// カーソルは表示するなよ
	timeout(-1);	// キー入力は来るまで待てよ

	// メインループ
	while (1) {
		key = Title();		// タイトル
		if (key == 'q') break;
		Game();			// ゲーム本体
	}

	// 終了
	endwin();	// 端末制御を終了
	return (0);
}