freebsd-dev/sys/teken/demo/teken_demo.c

355 lines
7.8 KiB
C

/*-
* Copyright (c) 2008-2009 Ed Schouten <ed@FreeBSD.org>
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*
* $FreeBSD$
*/
#include <sys/ioctl.h>
#include <assert.h>
#include <errno.h>
#include <inttypes.h>
#include <locale.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <ncurses.h>
#if defined(__FreeBSD__)
#include <libutil.h>
#elif defined(__linux__)
#include <pty.h>
#else
#include <util.h>
#endif
#include <teken.h>
static tf_bell_t test_bell;
static tf_cursor_t test_cursor;
static tf_putchar_t test_putchar;
static tf_fill_t test_fill;
static tf_copy_t test_copy;
static tf_param_t test_param;
static tf_respond_t test_respond;
static teken_funcs_t tf = {
.tf_bell = test_bell,
.tf_cursor = test_cursor,
.tf_putchar = test_putchar,
.tf_fill = test_fill,
.tf_copy = test_copy,
.tf_param = test_param,
.tf_respond = test_respond,
};
struct pixel {
teken_char_t c;
teken_attr_t a;
};
#define NCOLS 80
#define NROWS 24
struct pixel buffer[NCOLS][NROWS];
static int ptfd;
static void
printchar(const teken_pos_t *p)
{
int y, x, attr = 0;
struct pixel *px;
char str[5] = { 0 };
assert(p->tp_row < NROWS);
assert(p->tp_col < NCOLS);
getyx(stdscr, y, x);
px = &buffer[p->tp_col][p->tp_row];
/* Convert Unicode to UTF-8. */
if (px->c < 0x80) {
str[0] = px->c;
} else if (px->c < 0x800) {
str[0] = 0xc0 | (px->c >> 6);
str[1] = 0x80 | (px->c & 0x3f);
} else if (px->c < 0x10000) {
str[0] = 0xe0 | (px->c >> 12);
str[1] = 0x80 | ((px->c >> 6) & 0x3f);
str[2] = 0x80 | (px->c & 0x3f);
} else {
str[0] = 0xf0 | (px->c >> 18);
str[1] = 0x80 | ((px->c >> 12) & 0x3f);
str[2] = 0x80 | ((px->c >> 6) & 0x3f);
str[3] = 0x80 | (px->c & 0x3f);
}
if (px->a.ta_format & TF_BOLD)
attr |= A_BOLD;
if (px->a.ta_format & TF_UNDERLINE)
attr |= A_UNDERLINE;
if (px->a.ta_format & TF_BLINK)
attr |= A_BLINK;
if (px->a.ta_format & TF_REVERSE)
attr |= A_REVERSE;
bkgdset(attr | COLOR_PAIR(teken_256to8(px->a.ta_fgcolor) +
8 * teken_256to8(px->a.ta_bgcolor)));
mvaddstr(p->tp_row, p->tp_col, str);
move(y, x);
}
static void
test_bell(void *s __unused)
{
beep();
}
static void
test_cursor(void *s __unused, const teken_pos_t *p)
{
move(p->tp_row, p->tp_col);
}
static void
test_putchar(void *s __unused, const teken_pos_t *p, teken_char_t c,
const teken_attr_t *a)
{
buffer[p->tp_col][p->tp_row].c = c;
buffer[p->tp_col][p->tp_row].a = *a;
printchar(p);
}
static void
test_fill(void *s, const teken_rect_t *r, teken_char_t c,
const teken_attr_t *a)
{
teken_pos_t p;
/* Braindead implementation of fill() - just call putchar(). */
for (p.tp_row = r->tr_begin.tp_row; p.tp_row < r->tr_end.tp_row; p.tp_row++)
for (p.tp_col = r->tr_begin.tp_col; p.tp_col < r->tr_end.tp_col; p.tp_col++)
test_putchar(s, &p, c, a);
}
static void
test_copy(void *s __unused, const teken_rect_t *r, const teken_pos_t *p)
{
int nrow, ncol, x, y; /* Has to be signed - >= 0 comparison */
teken_pos_t d;
/*
* Copying is a little tricky. We must make sure we do it in
* correct order, to make sure we don't overwrite our own data.
*/
nrow = r->tr_end.tp_row - r->tr_begin.tp_row;
ncol = r->tr_end.tp_col - r->tr_begin.tp_col;
if (p->tp_row < r->tr_begin.tp_row) {
/* Copy from top to bottom. */
if (p->tp_col < r->tr_begin.tp_col) {
/* Copy from left to right. */
for (y = 0; y < nrow; y++) {
d.tp_row = p->tp_row + y;
for (x = 0; x < ncol; x++) {
d.tp_col = p->tp_col + x;
buffer[d.tp_col][d.tp_row] =
buffer[r->tr_begin.tp_col + x][r->tr_begin.tp_row + y];
printchar(&d);
}
}
} else {
/* Copy from right to left. */
for (y = 0; y < nrow; y++) {
d.tp_row = p->tp_row + y;
for (x = ncol - 1; x >= 0; x--) {
d.tp_col = p->tp_col + x;
buffer[d.tp_col][d.tp_row] =
buffer[r->tr_begin.tp_col + x][r->tr_begin.tp_row + y];
printchar(&d);
}
}
}
} else {
/* Copy from bottom to top. */
if (p->tp_col < r->tr_begin.tp_col) {
/* Copy from left to right. */
for (y = nrow - 1; y >= 0; y--) {
d.tp_row = p->tp_row + y;
for (x = 0; x < ncol; x++) {
d.tp_col = p->tp_col + x;
buffer[d.tp_col][d.tp_row] =
buffer[r->tr_begin.tp_col + x][r->tr_begin.tp_row + y];
printchar(&d);
}
}
} else {
/* Copy from right to left. */
for (y = nrow - 1; y >= 0; y--) {
d.tp_row = p->tp_row + y;
for (x = ncol - 1; x >= 0; x--) {
d.tp_col = p->tp_col + x;
buffer[d.tp_col][d.tp_row] =
buffer[r->tr_begin.tp_col + x][r->tr_begin.tp_row + y];
printchar(&d);
}
}
}
}
}
static void
test_param(void *s __unused, int cmd, unsigned int value)
{
switch (cmd) {
case TP_SHOWCURSOR:
curs_set(value);
break;
case TP_KEYPADAPP:
keypad(stdscr, value ? TRUE : FALSE);
break;
}
}
static void
test_respond(void *s __unused, const void *buf, size_t len)
{
write(ptfd, buf, len);
}
static void
redraw_border(void)
{
unsigned int i;
for (i = 0; i < NROWS; i++)
mvaddch(i, NCOLS, '|');
for (i = 0; i < NCOLS; i++)
mvaddch(NROWS, i, '-');
mvaddch(NROWS, NCOLS, '+');
}
static void
redraw_all(void)
{
teken_pos_t tp;
for (tp.tp_row = 0; tp.tp_row < NROWS; tp.tp_row++)
for (tp.tp_col = 0; tp.tp_col < NCOLS; tp.tp_col++)
printchar(&tp);
redraw_border();
}
int
main(int argc __unused, char *argv[] __unused)
{
struct winsize ws;
teken_t t;
teken_pos_t tp;
fd_set rfds;
char b[256];
ssize_t bl;
const int ccolors[8] = {
COLOR_BLACK, COLOR_RED, COLOR_GREEN, COLOR_YELLOW,
COLOR_BLUE, COLOR_MAGENTA, COLOR_CYAN, COLOR_WHITE
};
int i, j;
setlocale(LC_CTYPE, "UTF-8");
tp.tp_row = ws.ws_row = NROWS;
tp.tp_col = ws.ws_col = NCOLS;
switch (forkpty(&ptfd, NULL, NULL, &ws)) {
case -1:
perror("forkpty");
exit(1);
case 0:
setenv("TERM", "xterm", 1);
setenv("LC_CTYPE", "UTF-8", 0);
execlp("zsh", "-zsh", NULL);
execlp("bash", "-bash", NULL);
execlp("sh", "-sh", NULL);
_exit(1);
}
teken_init(&t, &tf, NULL);
teken_set_winsize(&t, &tp);
initscr();
raw();
start_color();
for (i = 0; i < 8; i++)
for (j = 0; j < 8; j++)
init_pair(i + 8 * j, ccolors[i], ccolors[j]);
redraw_border();
FD_ZERO(&rfds);
for (;;) {
FD_SET(STDIN_FILENO, &rfds);
FD_SET(ptfd, &rfds);
if (select(ptfd + 1, &rfds, NULL, NULL, NULL) < 0) {
if (errno == EINTR) {
redraw_all();
refresh();
continue;
}
break;
}
if (FD_ISSET(STDIN_FILENO, &rfds)) {
bl = read(STDIN_FILENO, b, sizeof b);
if (bl <= 0)
break;
write(ptfd, b, bl);
}
if (FD_ISSET(ptfd, &rfds)) {
bl = read(ptfd, b, sizeof b);
if (bl <= 0)
break;
teken_input(&t, b, bl);
refresh();
}
}
endwin();
return (0);
}