sys/teken/demo/teken_demo.c

   1 /*-
   2  * Copyright (c) 2008-2009 Ed Schouten <ed@FreeBSD.org>
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  24  * SUCH DAMAGE.
  25  *
  26  * $FreeBSD$
  27  */
  28
  29 #include <sys/ioctl.h>
  30
  31 #include <assert.h>
  32 #include <errno.h>
  33 #include <inttypes.h>
  34 #include <locale.h>
  35 #include <stdio.h>
  36 #include <stdlib.h>
  37 #include <unistd.h>
  38
  39 #include <ncurses.h>
  40 #if defined(__FreeBSD__)
  41 #include <libutil.h>
  42 #elif defined(__linux__)
  43 #include <pty.h>
  44 #else
  45 #include <util.h>
  46 #endif
  47
  48 #include <teken.h>
  49
  50 static tf_bell_t        test_bell;
  51 static tf_cursor_t      test_cursor;
  52 static tf_putchar_t     test_putchar;
  53 static tf_fill_t        test_fill;
  54 static tf_copy_t        test_copy;
  55 static tf_param_t       test_param;
  56 static tf_respond_t     test_respond;
  57
  58 static teken_funcs_t tf = {
  59         .tf_bell        = test_bell,
  60         .tf_cursor      = test_cursor,
  61         .tf_putchar     = test_putchar,
  62         .tf_fill        = test_fill,
  63         .tf_copy        = test_copy,
  64         .tf_param       = test_param,
  65         .tf_respond     = test_respond,
  66 };
  67
  68 struct pixel {
  69         teken_char_t    c;
  70         teken_attr_t    a;
  71 };
  72
  73 #define NCOLS   80
  74 #define NROWS   24
  75 struct pixel buffer[NCOLS][NROWS];
  76
  77 static int ptfd;
  78
  79 static void
  80 printchar(const teken_pos_t *p)
  81 {
  82         int y, x, attr = 0;
  83         struct pixel *px;
  84         char str[5] = { 0 };
  85
  86         assert(p->tp_row < NROWS);
  87         assert(p->tp_col < NCOLS);
  88
  89         getyx(stdscr, y, x);
  90
  91         px = &buffer[p->tp_col][p->tp_row];
  92
  93         /* Convert Unicode to UTF-8. */
  94         if (px->c < 0x80) {
  95                 str[0] = px->c;
  96         } else if (px->c < 0x800) {
  97                 str[0] = 0xc0 | (px->c >> 6);
  98                 str[1] = 0x80 | (px->c & 0x3f);
  99         } else if (px->c < 0x10000) {
 100                 str[0] = 0xe0 | (px->c >> 12);
 101                 str[1] = 0x80 | ((px->c >> 6) & 0x3f);
 102                 str[2] = 0x80 | (px->c & 0x3f);
 103         } else {
 104                 str[0] = 0xf0 | (px->c >> 18);
 105                 str[1] = 0x80 | ((px->c >> 12) & 0x3f);
 106                 str[2] = 0x80 | ((px->c >> 6) & 0x3f);
 107                 str[3] = 0x80 | (px->c & 0x3f);
 108         }
 109
 110         if (px->a.ta_format & TF_BOLD)
 111                 attr |= A_BOLD;
 112         if (px->a.ta_format & TF_UNDERLINE)
 113                 attr |= A_UNDERLINE;
 114         if (px->a.ta_format & TF_BLINK)
 115                 attr |= A_BLINK;
 116         if (px->a.ta_format & TF_REVERSE)
 117                 attr |= A_REVERSE;
 118
 119         bkgdset(attr | COLOR_PAIR(teken_256to8(px->a.ta_fgcolor) +
 120               8 * teken_256to8(px->a.ta_bgcolor)));
 121         mvaddstr(p->tp_row, p->tp_col, str);
 122
 123         move(y, x);
 124 }
 125
 126 static void
 127 test_bell(void *s __unused)
 128 {
 129
 130         beep();
 131 }
 132
 133 static void
 134 test_cursor(void *s __unused, const teken_pos_t *p)
 135 {
 136
 137         move(p->tp_row, p->tp_col);
 138 }
 139
 140 static void
 141 test_putchar(void *s __unused, const teken_pos_t *p, teken_char_t c,
 142     const teken_attr_t *a)
 143 {
 144
 145         buffer[p->tp_col][p->tp_row].c = c;
 146         buffer[p->tp_col][p->tp_row].a = *a;
 147         printchar(p);
 148 }
 149
 150 static void
 151 test_fill(void *s, const teken_rect_t *r, teken_char_t c,
 152     const teken_attr_t *a)
 153 {
 154         teken_pos_t p;
 155
 156         /* Braindead implementation of fill() - just call putchar(). */
 157         for (p.tp_row = r->tr_begin.tp_row; p.tp_row < r->tr_end.tp_row; p.tp_row++)
 158                 for (p.tp_col = r->tr_begin.tp_col; p.tp_col < r->tr_end.tp_col; p.tp_col++)
 159                         test_putchar(s, &p, c, a);
 160 }
 161
 162 static void
 163 test_copy(void *s __unused, const teken_rect_t *r, const teken_pos_t *p)
 164 {
 165         int nrow, ncol, x, y; /* Has to be signed - >= 0 comparison */
 166         teken_pos_t d;
 167
 168         /*
 169          * Copying is a little tricky. We must make sure we do it in
 170          * correct order, to make sure we don't overwrite our own data.
 171          */
 172
 173         nrow = r->tr_end.tp_row - r->tr_begin.tp_row;
 174         ncol = r->tr_end.tp_col - r->tr_begin.tp_col;
 175
 176         if (p->tp_row < r->tr_begin.tp_row) {
 177                 /* Copy from top to bottom. */
 178                 if (p->tp_col < r->tr_begin.tp_col) {
 179                         /* Copy from left to right. */
 180                         for (y = 0; y < nrow; y++) {
 181                                 d.tp_row = p->tp_row + y;
 182                                 for (x = 0; x < ncol; x++) {
 183                                         d.tp_col = p->tp_col + x;
 184                                         buffer[d.tp_col][d.tp_row] =
 185                                             buffer[r->tr_begin.tp_col + x][r->tr_begin.tp_row + y];
 186                                         printchar(&d);
 187                                 }
 188                         }
 189                 } else {
 190                         /* Copy from right to left. */
 191                         for (y = 0; y < nrow; y++) {
 192                                 d.tp_row = p->tp_row + y;
 193                                 for (x = ncol - 1; x >= 0; x--) {
 194                                         d.tp_col = p->tp_col + x;
 195                                         buffer[d.tp_col][d.tp_row] =
 196                                             buffer[r->tr_begin.tp_col + x][r->tr_begin.tp_row + y];
 197                                         printchar(&d);
 198                                 }
 199                         }
 200                 }
 201         } else {
 202                 /* Copy from bottom to top. */
 203                 if (p->tp_col < r->tr_begin.tp_col) {
 204                         /* Copy from left to right. */
 205                         for (y = nrow - 1; y >= 0; y--) {
 206                                 d.tp_row = p->tp_row + y;
 207                                 for (x = 0; x < ncol; x++) {
 208                                         d.tp_col = p->tp_col + x;
 209                                         buffer[d.tp_col][d.tp_row] =
 210                                             buffer[r->tr_begin.tp_col + x][r->tr_begin.tp_row + y];
 211                                         printchar(&d);
 212                                 }
 213                         }
 214                 } else {
 215                         /* Copy from right to left. */
 216                         for (y = nrow - 1; y >= 0; y--) {
 217                                 d.tp_row = p->tp_row + y;
 218                                 for (x = ncol - 1; x >= 0; x--) {
 219                                         d.tp_col = p->tp_col + x;
 220                                         buffer[d.tp_col][d.tp_row] =
 221                                             buffer[r->tr_begin.tp_col + x][r->tr_begin.tp_row + y];
 222                                         printchar(&d);
 223                                 }
 224                         }
 225                 }
 226         }
 227 }
 228
 229 static void
 230 test_param(void *s __unused, int cmd, unsigned int value)
 231 {
 232
 233         switch (cmd) {
 234         case TP_SHOWCURSOR:
 235                 curs_set(value);
 236                 break;
 237         case TP_KEYPADAPP:
 238                 keypad(stdscr, value ? TRUE : FALSE);
 239                 break;
 240         }
 241 }
 242
 243 static void
 244 test_respond(void *s __unused, const void *buf, size_t len)
 245 {
 246
 247         write(ptfd, buf, len);
 248 }
 249
 250 static void
 251 redraw_border(void)
 252 {
 253         unsigned int i;
 254
 255         for (i = 0; i < NROWS; i++)
 256                 mvaddch(i, NCOLS, '|');
 257         for (i = 0; i < NCOLS; i++)
 258                 mvaddch(NROWS, i, '-');
 259
 260         mvaddch(NROWS, NCOLS, '+');
 261 }
 262
 263 static void
 264 redraw_all(void)
 265 {
 266         teken_pos_t tp;
 267
 268         for (tp.tp_row = 0; tp.tp_row < NROWS; tp.tp_row++)
 269                 for (tp.tp_col = 0; tp.tp_col < NCOLS; tp.tp_col++)
 270                         printchar(&tp);
 271
 272         redraw_border();
 273 }
 274
 275 int
 276 main(int argc __unused, char *argv[] __unused)
 277 {
 278         struct winsize ws;
 279         teken_t t;
 280         teken_pos_t tp;
 281         fd_set rfds;
 282         char b[256];
 283         ssize_t bl;
 284         const int ccolors[8] = {
 285             COLOR_BLACK, COLOR_RED, COLOR_GREEN, COLOR_YELLOW,
 286             COLOR_BLUE, COLOR_MAGENTA, COLOR_CYAN, COLOR_WHITE
 287         };
 288         int i, j;
 289
 290         setlocale(LC_CTYPE, "UTF-8");
 291
 292         tp.tp_row = ws.ws_row = NROWS;
 293         tp.tp_col = ws.ws_col = NCOLS;
 294
 295         switch (forkpty(&ptfd, NULL, NULL, &ws)) {
 296         case -1:
 297                 perror("forkpty");
 298                 exit(1);
 299         case 0:
 300                 setenv("TERM", "xterm", 1);
 301                 setenv("LC_CTYPE", "UTF-8", 0);
 302                 execlp("zsh", "-zsh", NULL);
 303                 execlp("bash", "-bash", NULL);
 304                 execlp("sh", "-sh", NULL);
 305                 _exit(1);
 306         }
 307
 308         teken_init(&t, &tf, NULL);
 309         teken_set_winsize(&t, &tp);
 310
 311         initscr();
 312         raw();
 313         start_color();
 314         for (i = 0; i < 8; i++)
 315                 for (j = 0; j < 8; j++)
 316                         init_pair(i + 8 * j, ccolors[i], ccolors[j]);
 317
 318         redraw_border();
 319
 320         FD_ZERO(&rfds);
 321
 322         for (;;) {
 323                 FD_SET(STDIN_FILENO, &rfds);
 324                 FD_SET(ptfd, &rfds);
 325
 326                 if (select(ptfd + 1, &rfds, NULL, NULL, NULL) < 0) {
 327                         if (errno == EINTR) {
 328                                 redraw_all();
 329                                 refresh();
 330                                 continue;
 331                         }
 332                         break;
 333                 }
 334
 335                 if (FD_ISSET(STDIN_FILENO, &rfds)) {
 336                         bl = read(STDIN_FILENO, b, sizeof b);
 337                         if (bl <= 0)
 338                                 break;
 339                         write(ptfd, b, bl);
 340                 }
 341
 342                 if (FD_ISSET(ptfd, &rfds)) {
 343                         bl = read(ptfd, b, sizeof b);
 344                         if (bl <= 0)
 345                                 break;
 346                         teken_input(&t, b, bl);
 347                         refresh();
 348                 }
 349         }
 350
 351         endwin();
 352
 353         return (0);
 354 }