diff options
Diffstat (limited to 'sfm-0.4/termbox.c')
| -rw-r--r-- | sfm-0.4/termbox.c | 1509 |
1 files changed, 0 insertions, 1509 deletions
diff --git a/sfm-0.4/termbox.c b/sfm-0.4/termbox.c deleted file mode 100644 index 1c7bfdd..0000000 --- a/sfm-0.4/termbox.c +++ /dev/null @@ -1,1509 +0,0 @@ -#if defined(__linux__) -#define _GNU_SOURCE -#elif defined(__APPLE__) -#define _DARWIN_C_SOURCE -#elif defined(__FreeBSD__) -#define __BSD_VISIBLE 1 -#endif -#include "termbox.h" - -#include <sys/ioctl.h> -#include <sys/select.h> -#include <sys/stat.h> -#include <sys/time.h> - -#include <assert.h> -#include <errno.h> -#include <fcntl.h> -#include <signal.h> -#include <stdbool.h> -#include <stdio.h> -#include <stdlib.h> -#include <string.h> -#include <termios.h> -#include <unistd.h> -#include <wchar.h> - -#define ENTER_MOUSE_SEQ "\x1b[?1000h\x1b[?1002h\x1b[?1015h\x1b[?1006h" -#define EXIT_MOUSE_SEQ "\x1b[?1006l\x1b[?1015l\x1b[?1002l\x1b[?1000l" -#define EUNSUPPORTED_TERM -1 -#define TI_MAGIC 0432 -#define TI_ALT_MAGIC 542 -#define TI_HEADER_LENGTH 12 -#define TB_KEYS_NUM 22 -#define CELL(buf, x, y) (buf)->cells[(y) * (buf)->width + (x)] -#define IS_CURSOR_HIDDEN(cx, cy) (cx == -1 || cy == -1) -#define LAST_COORD_INIT -1 -#define WRITE_LITERAL(X) bytebuffer_append(&output_buffer, (X), sizeof(X) - 1) -#define WRITE_INT(X) \ - bytebuffer_append(&output_buffer, buf, convertnum((X), buf)) - -struct cellbuf { - int width; - int height; - struct tb_cell *cells; -}; - -struct bytebuffer { - char *buf; - int len; - int cap; -}; - -enum { - T_ENTER_CA, - T_EXIT_CA, - T_SHOW_CURSOR, - T_HIDE_CURSOR, - T_CLEAR_SCREEN, - T_SGR0, - T_UNDERLINE, - T_BOLD, - T_BLINK, - T_REVERSE, - T_ENTER_KEYPAD, - T_EXIT_KEYPAD, - T_ENTER_MOUSE, - T_EXIT_MOUSE, - T_FUNCS_NUM, -}; - -static int init_term_builtin(void); -static char *read_file(const char *file); -static char *terminfo_try_path(const char *path, const char *term); -static char *load_terminfo(void); -static const char *terminfo_copy_string(char *data, int str, int table); -static int init_term(void); -static void shutdown_term(void); -static bool starts_with(const char *s1, int len, const char *s2); -static int parse_mouse_event(struct tb_event *event, const char *buf, int len); -static int parse_escape_seq(struct tb_event *event, const char *buf, int len); -static int convertnum(uint32_t num, char *buf); -static void write_cursor(int x, int y); -static void write_sgr(uint16_t fg, uint16_t bg); -static void cellbuf_init(struct cellbuf *buf, int width, int height); -static void cellbuf_resize(struct cellbuf *buf, int width, int height); -static void cellbuf_clear(struct cellbuf *buf); -static void cellbuf_free(struct cellbuf *buf); -static void get_term_size(int *w, int *h); -static void update_term_size(void); -static void send_attr(uint16_t fg, uint16_t bg); -static void send_char(int x, int y, uint32_t c); -static void send_clear(void); -static void sigwinch_handler(int xxx); -static void update_size(void); -static int read_up_to(int n); -static int wait_fill_event(struct tb_event *event, struct timeval *timeout); -static void bytebuffer_reserve(struct bytebuffer *b, int cap); -static void bytebuffer_init(struct bytebuffer *b, int cap); -static void bytebuffer_free(struct bytebuffer *b); -static void bytebuffer_clear(struct bytebuffer *b); -static void bytebuffer_append(struct bytebuffer *b, const char *data, int len); -static void bytebuffer_puts(struct bytebuffer *b, const char *str); -static void bytebuffer_resize(struct bytebuffer *b, int len); -static void bytebuffer_flush(struct bytebuffer *b, int fd); -static void bytebuffer_truncate(struct bytebuffer *b, int n); - -static struct termios orig_tios; -static struct cellbuf back_buffer; -static struct cellbuf front_buffer; -static struct bytebuffer output_buffer; -static struct bytebuffer input_buffer; -static int termw = -1; -static int termh = -1; -static int inputmode = TB_INPUT_ESC; -static int outputmode = TB_OUTPUT_NORMAL; -static int inout; -static int winch_fds[2]; -static int lastx = LAST_COORD_INIT; -static int lasty = LAST_COORD_INIT; -static int cursor_x = -1; -static int cursor_y = -1; -static uint16_t background = TB_DEFAULT; -static uint16_t foreground = TB_DEFAULT; -/* may happen in a different thread */ -static volatile int buffer_size_change_request; -// rxvt-256color -static const char *rxvt_256color_keys[] = { "\033[11~", "\033[12~", "\033[13~", - "\033[14~", "\033[15~", "\033[17~", "\033[18~", "\033[19~", "\033[20~", - "\033[21~", "\033[23~", "\033[24~", "\033[2~", "\033[3~", "\033[7~", - "\033[8~", "\033[5~", "\033[6~", "\033[A", "\033[B", "\033[D", "\033[C", - 0 }; -static const char *rxvt_256color_funcs[] = { - "\0337\033[?47h", - "\033[2J\033[?47l\0338", - "\033[?25h", - "\033[?25l", - "\033[H\033[2J", - "\033[m", - "\033[4m", - "\033[1m", - "\033[5m", - "\033[7m", - "\033=", - "\033>", - ENTER_MOUSE_SEQ, - EXIT_MOUSE_SEQ, -}; -// Eterm -static const char *eterm_keys[] = { "\033[11~", "\033[12~", "\033[13~", - "\033[14~", "\033[15~", "\033[17~", "\033[18~", "\033[19~", "\033[20~", - "\033[21~", "\033[23~", "\033[24~", "\033[2~", "\033[3~", "\033[7~", - "\033[8~", "\033[5~", "\033[6~", "\033[A", "\033[B", "\033[D", "\033[C", - 0 }; -static const char *eterm_funcs[] = { - "\0337\033[?47h", - "\033[2J\033[?47l\0338", - "\033[?25h", - "\033[?25l", - "\033[H\033[2J", - "\033[m", - "\033[4m", - "\033[1m", - "\033[5m", - "\033[7m", - "", - "", - "", - "", -}; -// screen -static const char *screen_keys[] = { "\033OP", "\033OQ", "\033OR", "\033OS", - "\033[15~", "\033[17~", "\033[18~", "\033[19~", "\033[20~", "\033[21~", - "\033[23~", "\033[24~", "\033[2~", "\033[3~", "\033[1~", "\033[4~", - "\033[5~", "\033[6~", "\033OA", "\033OB", "\033OD", "\033OC", 0 }; -static const char *screen_funcs[] = { - "\033[?1049h", - "\033[?1049l", - "\033[34h\033[?25h", - "\033[?25l", - "\033[H\033[J", - "\033[m", - "\033[4m", - "\033[1m", - "\033[5m", - "\033[7m", - "\033[?1h\033=", - "\033[?1l\033>", - ENTER_MOUSE_SEQ, - EXIT_MOUSE_SEQ, -}; -// rxvt-unicode -static const char *rxvt_unicode_keys[] = { "\033[11~", "\033[12~", "\033[13~", - "\033[14~", "\033[15~", "\033[17~", "\033[18~", "\033[19~", "\033[20~", - "\033[21~", "\033[23~", "\033[24~", "\033[2~", "\033[3~", "\033[7~", - "\033[8~", "\033[5~", "\033[6~", "\033[A", "\033[B", "\033[D", "\033[C", - 0 }; -static const char *rxvt_unicode_funcs[] = { - "\033[?1049h", - "\033[r\033[?1049l", - "\033[?25h", - "\033[?25l", - "\033[H\033[2J", - "\033[m\033(B", - "\033[4m", - "\033[1m", - "\033[5m", - "\033[7m", - "\033=", - "\033>", - ENTER_MOUSE_SEQ, - EXIT_MOUSE_SEQ, -}; -// linux -static const char *linux_keys[] = { "\033[[A", "\033[[B", "\033[[C", "\033[[D", - "\033[[E", "\033[17~", "\033[18~", "\033[19~", "\033[20~", "\033[21~", - "\033[23~", "\033[24~", "\033[2~", "\033[3~", "\033[1~", "\033[4~", - "\033[5~", "\033[6~", "\033[A", "\033[B", "\033[D", "\033[C", 0 }; -static const char *linux_funcs[] = { - "", - "", - "\033[?25h\033[?0c", - "\033[?25l\033[?1c", - "\033[H\033[J", - "\033[0;10m", - "\033[4m", - "\033[1m", - "\033[5m", - "\033[7m", - "", - "", - "", - "", -}; -// xterm -static const char *xterm_keys[] = { "\033OP", "\033OQ", "\033OR", "\033OS", - "\033[15~", "\033[17~", "\033[18~", "\033[19~", "\033[20~", "\033[21~", - "\033[23~", "\033[24~", "\033[2~", "\033[3~", "\033OH", "\033OF", - "\033[5~", "\033[6~", "\033OA", "\033OB", "\033OD", "\033OC", 0 }; -static const char *xterm_funcs[] = { - "\033[?1049h", - "\033[?1049l", - "\033[?12l\033[?25h", - "\033[?25l", - "\033[H\033[2J", - "\033(B\033[m", - "\033[4m", - "\033[1m", - "\033[5m", - "\033[7m", - "\033[?1h\033=", - "\033[?1l\033>", - ENTER_MOUSE_SEQ, - EXIT_MOUSE_SEQ, -}; -static struct term { - const char *name; - const char **keys; - const char **funcs; -} terms[] = { - { "rxvt-256color", rxvt_256color_keys, rxvt_256color_funcs }, - { "Eterm", eterm_keys, eterm_funcs }, - { "screen", screen_keys, screen_funcs }, - { "rxvt-unicode", rxvt_unicode_keys, rxvt_unicode_funcs }, - { "linux", linux_keys, linux_funcs }, - { "xterm", xterm_keys, xterm_funcs }, - { 0, 0, 0 }, -}; -static bool init_from_terminfo = false; -static const char **keys; -static const char **funcs; - -static int -try_compatible(const char *term, const char *name, const char **tkeys, - const char **tfuncs) -{ - if (strstr(term, name)) { - keys = tkeys; - funcs = tfuncs; - return 0; - } - - return EUNSUPPORTED_TERM; -} - -static int -init_term_builtin(void) -{ - int i; - const char *term = getenv("TERM"); - - if (term) { - for (i = 0; terms[i].name; i++) { - if (!strcmp(terms[i].name, term)) { - keys = terms[i].keys; - funcs = terms[i].funcs; - return 0; - } - } - - /* let's do some heuristic, maybe it's a compatible terminal */ - if (try_compatible(term, "xterm", xterm_keys, xterm_funcs) == 0) - return 0; - if (try_compatible(term, "rxvt", rxvt_unicode_keys, - rxvt_unicode_funcs) == 0) - return 0; - if (try_compatible(term, "linux", linux_keys, linux_funcs) == 0) - return 0; - if (try_compatible(term, "Eterm", eterm_keys, eterm_funcs) == 0) - return 0; - if (try_compatible(term, "screen", screen_keys, screen_funcs) == - 0) - return 0; - if (try_compatible(term, "tmux", screen_keys, screen_funcs) == - 0) - return 0; - /* let's assume that 'cygwin' is xterm compatible */ - if (try_compatible(term, "cygwin", xterm_keys, xterm_funcs) == - 0) - return 0; - } - - return EUNSUPPORTED_TERM; -} - -//---------------------------------------------------------------------- -// terminfo -//---------------------------------------------------------------------- - -static char * -read_file(const char *file) -{ - FILE *f = fopen(file, "rb"); - if (!f) - return 0; - - struct stat st; - if (fstat(fileno(f), &st) != 0) { - fclose(f); - return 0; - } - - char *data = malloc(st.st_size); - if (!data) { - fclose(f); - return 0; - } - - if (fread(data, 1, st.st_size, f) != (size_t)st.st_size) { - fclose(f); - free(data); - return 0; - } - - fclose(f); - return data; -} - -static char * -terminfo_try_path(const char *path, const char *term) -{ - char tmp[4096]; - snprintf(tmp, sizeof(tmp), "%s/%c/%s", path, term[0], term); - tmp[sizeof(tmp) - 1] = '\0'; - char *data = read_file(tmp); - if (data) { - return data; - } - - // fallback to darwin specific dirs structure - snprintf(tmp, sizeof(tmp), "%s/%x/%s", path, term[0], term); - tmp[sizeof(tmp) - 1] = '\0'; - return read_file(tmp); -} - -static char * -load_terminfo(void) -{ - char tmp[4096]; - const char *term = getenv("TERM"); - if (!term) { - return 0; - } - - // if TERMINFO is set, no other directory should be searched - const char *terminfo = getenv("TERMINFO"); - if (terminfo) { - return terminfo_try_path(terminfo, term); - } - - // next, consider ~/.terminfo - const char *home = getenv("HOME"); - if (home) { - snprintf(tmp, sizeof(tmp), "%s/.terminfo", home); - tmp[sizeof(tmp) - 1] = '\0'; - char *data = terminfo_try_path(tmp, term); - if (data) - return data; - } - - // next, TERMINFO_DIRS - const char *dirs = getenv("TERMINFO_DIRS"); - if (dirs) { - snprintf(tmp, sizeof(tmp), "%s", dirs); - tmp[sizeof(tmp) - 1] = '\0'; - char *dir = strtok(tmp, ":"); - while (dir) { - const char *cdir = dir; - if (strcmp(cdir, "") == 0) { - cdir = "/usr/share/terminfo"; - } - char *data = terminfo_try_path(cdir, term); - if (data) - return data; - dir = strtok(0, ":"); - } - } - - // fallback to /usr/share/terminfo - return terminfo_try_path("/usr/share/terminfo", term); -} - -static const char * -terminfo_copy_string(char *data, int str, int table) -{ - const int16_t off = *(int16_t *)(data + str); - const char *src = data + table + off; - int len = strlen(src); - char *dst = malloc(len + 1); - strcpy(dst, src); - return dst; -} - -static const int16_t ti_funcs[] = { - 28, - 40, - 16, - 13, - 5, - 39, - 36, - 27, - 26, - 34, - 89, - 88, -}; - -static const int16_t ti_keys[] = { - 66, - 68 /* apparently not a typo; 67 is F10 for whatever reason */, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 67, - 216, - 217, - 77, - 59, - 76, - 164, - 82, - 81, - 87, - 61, - 79, - 83, -}; - -static int -init_term(void) -{ - int i; - char *data = load_terminfo(); - if (!data) { - init_from_terminfo = false; - return init_term_builtin(); - } - - int16_t *header = (int16_t *)data; - - const int number_sec_len = header[0] == TI_ALT_MAGIC ? 4 : 2; - - if ((header[1] + header[2]) % 2) { - // old quirk to align everything on word boundaries - header[2] += 1; - } - - const int str_offset = TI_HEADER_LENGTH + header[1] + header[2] + - number_sec_len * header[3]; - const int table_offset = str_offset + 2 * header[4]; - - keys = malloc(sizeof(const char *) * (TB_KEYS_NUM + 1)); - for (i = 0; i < TB_KEYS_NUM; i++) { - keys[i] = terminfo_copy_string( - data, str_offset + 2 * ti_keys[i], table_offset); - } - keys[TB_KEYS_NUM] = 0; - - funcs = malloc(sizeof(const char *) * T_FUNCS_NUM); - // the last two entries are reserved for mouse. because the table offset is - // not there, the two entries have to fill in manually - for (i = 0; i < T_FUNCS_NUM - 2; i++) { - funcs[i] = terminfo_copy_string( - data, str_offset + 2 * ti_funcs[i], table_offset); - } - - funcs[T_FUNCS_NUM - 2] = ENTER_MOUSE_SEQ; - funcs[T_FUNCS_NUM - 1] = EXIT_MOUSE_SEQ; - - init_from_terminfo = true; - free(data); - return 0; -} - -static void -shutdown_term(void) -{ - if (init_from_terminfo) { - int i; - for (i = 0; i < TB_KEYS_NUM; i++) { - free((void *)keys[i]); - } - // the last two entries are reserved for mouse. because the table offset - // is not there, the two entries have to fill in manually and do not - // need to be freed. - for (i = 0; i < T_FUNCS_NUM - 2; i++) { - free((void *)funcs[i]); - } - free(keys); - free(funcs); - } -} - -// if s1 starts with s2 returns true, else false -// len is the length of s1 -// s2 should be null-terminated -static bool -starts_with(const char *s1, int len, const char *s2) -{ - int n = 0; - while (*s2 && n < len) { - if (*s1++ != *s2++) - return false; - n++; - } - return *s2 == 0; -} - -static int -parse_mouse_event(struct tb_event *event, const char *buf, int len) -{ - if (len >= 6 && starts_with(buf, len, "\033[M")) { - // X10 mouse encoding, the simplest one - // \033 [ M Cb Cx Cy - int b = buf[3] - 32; - switch (b & 3) { - case 0: - if ((b & 64) != 0) - event->key = TB_KEY_MOUSE_WHEEL_UP; - else - event->key = TB_KEY_MOUSE_LEFT; - break; - case 1: - if ((b & 64) != 0) - event->key = TB_KEY_MOUSE_WHEEL_DOWN; - else - event->key = TB_KEY_MOUSE_MIDDLE; - break; - case 2: - event->key = TB_KEY_MOUSE_RIGHT; - break; - case 3: - event->key = TB_KEY_MOUSE_RELEASE; - break; - default: - return -6; - } - event->type = TB_EVENT_MOUSE; // TB_EVENT_KEY by default - if ((b & 32) != 0) - event->mod |= TB_MOD_MOTION; - - // the coord is 1,1 for upper left - event->x = (uint8_t)buf[4] - 1 - 32; - event->y = (uint8_t)buf[5] - 1 - 32; - - return 6; - } else if (starts_with(buf, len, "\033[<") || - starts_with(buf, len, "\033[")) { - // xterm 1006 extended mode or urxvt 1015 extended mode - // xterm: \033 [ < Cb ; Cx ; Cy (M or m) - // urxvt: \033 [ Cb ; Cx ; Cy M - int i, mi = -1, starti = -1; - int isM, isU, s1 = -1, s2 = -1; - int n1 = 0, n2 = 0, n3 = 0; - - for (i = 0; i < len; i++) { - // We search the first (s1) and the last (s2) ';' - if (buf[i] == ';') { - if (s1 == -1) - s1 = i; - s2 = i; - } - - // We search for the first 'm' or 'M' - if ((buf[i] == 'm' || buf[i] == 'M') && mi == -1) { - mi = i; - break; - } - } - if (mi == -1) - return 0; - - // whether it's a capital M or not - isM = (buf[mi] == 'M'); - - if (buf[2] == '<') { - isU = 0; - starti = 3; - } else { - isU = 1; - starti = 2; - } - - if (s1 == -1 || s2 == -1 || s1 == s2) - return 0; - - n1 = strtoul(&buf[starti], NULL, 10); - n2 = strtoul(&buf[s1 + 1], NULL, 10); - n3 = strtoul(&buf[s2 + 1], NULL, 10); - - if (isU) - n1 -= 32; - - switch (n1 & 3) { - case 0: - if ((n1 & 64) != 0) { - event->key = TB_KEY_MOUSE_WHEEL_UP; - } else { - event->key = TB_KEY_MOUSE_LEFT; - } - break; - case 1: - if ((n1 & 64) != 0) { - event->key = TB_KEY_MOUSE_WHEEL_DOWN; - } else { - event->key = TB_KEY_MOUSE_MIDDLE; - } - break; - case 2: - event->key = TB_KEY_MOUSE_RIGHT; - break; - case 3: - event->key = TB_KEY_MOUSE_RELEASE; - break; - default: - return mi + 1; - } - - if (!isM) { - // on xterm mouse release is signaled by lowercase m - event->key = TB_KEY_MOUSE_RELEASE; - } - - event->type = TB_EVENT_MOUSE; // TB_EVENT_KEY by default - if ((n1 & 32) != 0) - event->mod |= TB_MOD_MOTION; - - event->x = (uint8_t)n2 - 1; - event->y = (uint8_t)n3 - 1; - - return mi + 1; - } - - return 0; -} - -// convert escape sequence to event, and return consumed bytes on success -// (failure == 0) -static int -parse_escape_seq(struct tb_event *event, const char *buf, int len) -{ - int mouse_parsed = parse_mouse_event(event, buf, len); - - if (mouse_parsed != 0) - return mouse_parsed; - - // it's pretty simple here, find 'starts_with' match and return - // success, else return failure - int i; - for (i = 0; keys[i]; i++) { - if (starts_with(buf, len, keys[i])) { - event->ch = 0; - event->key = 0xFFFF - i; - return strlen(keys[i]); - } - } - return 0; -} - -static bool -extract_event(struct tb_event *event, struct bytebuffer *inbuf, int in) -{ - const char *buf = inbuf->buf; - const int len = inbuf->len; - if (len == 0) - return false; - - if (buf[0] == '\033') { - int n = parse_escape_seq(event, buf, len); - if (n != 0) { - bool success = true; - if (n < 0) { - success = false; - n = -n; - } - bytebuffer_truncate(inbuf, n); - return success; - } else { - // it's not escape sequence, then it's ALT or ESC, - // check inputmode - if (in & TB_INPUT_ESC) { - // if we're in escape mode, fill ESC event, pop - // buffer, return success - event->ch = 0; - event->key = TB_KEY_ESC; - event->mod = 0; - bytebuffer_truncate(inbuf, 1); - return true; - } else if (in & TB_INPUT_ALT) { - // if we're in alt mode, set ALT modifier to - // event and redo parsing - event->mod = TB_MOD_ALT; - bytebuffer_truncate(inbuf, 1); - return extract_event(event, inbuf, in); - } - assert(!"never got here"); - } - } - - // if we're here, this is not an escape sequence and not an alt sequence - // so, it's a FUNCTIONAL KEY or a UNICODE character - - // first of all check if it's a functional key - if ((unsigned char)buf[0] <= TB_KEY_SPACE || - (unsigned char)buf[0] == TB_KEY_BACKSPACE2) { - // fill event, pop buffer, return success */ - event->ch = 0; - event->key = (uint16_t)buf[0]; - bytebuffer_truncate(inbuf, 1); - return true; - } - - // feh... we got utf8 here - - // check if there is all bytes - if (len >= tb_utf8_char_length(buf[0])) { - /* everything ok, fill event, pop buffer, return success */ - tb_utf8_char_to_unicode(&event->ch, buf); - event->key = 0; - bytebuffer_truncate(inbuf, tb_utf8_char_length(buf[0])); - return true; - } - - // event isn't recognized, perhaps there is not enough bytes in utf8 - // sequence - return false; -} - -/* -------------------------------------------------------- */ - -int -tb_init_fd(int inout_) -{ - inout = inout_; - if (inout == -1) { - return TB_EFAILED_TO_OPEN_TTY; - } - - if (init_term() < 0) { - close(inout); - return TB_EUNSUPPORTED_TERMINAL; - } - - if (pipe(winch_fds) < 0) { - close(inout); - return TB_EPIPE_TRAP_ERROR; - } - - struct sigaction sa; - memset(&sa, 0, sizeof(sa)); - sa.sa_handler = sigwinch_handler; - sa.sa_flags = 0; - sigaction(SIGWINCH, &sa, 0); - - tcgetattr(inout, &orig_tios); - - struct termios tios; - memcpy(&tios, &orig_tios, sizeof(tios)); - - tios.c_iflag &= ~(IGNBRK | BRKINT | PARMRK | ISTRIP | INLCR | IGNCR | - ICRNL | IXON); - tios.c_oflag &= ~OPOST; - tios.c_lflag &= ~(ECHO | ECHONL | ICANON | ISIG | IEXTEN); - tios.c_cflag &= ~(CSIZE | PARENB); - tios.c_cflag |= CS8; - tios.c_cc[VMIN] = 0; - tios.c_cc[VTIME] = 0; - tcsetattr(inout, TCSAFLUSH, &tios); - - bytebuffer_init(&input_buffer, 128); - bytebuffer_init(&output_buffer, 32 * 1024); - - bytebuffer_puts(&output_buffer, funcs[T_ENTER_CA]); - bytebuffer_puts(&output_buffer, funcs[T_ENTER_KEYPAD]); - bytebuffer_puts(&output_buffer, funcs[T_HIDE_CURSOR]); - send_clear(); - - update_term_size(); - cellbuf_init(&back_buffer, termw, termh); - cellbuf_init(&front_buffer, termw, termh); - cellbuf_clear(&back_buffer); - cellbuf_clear(&front_buffer); - - return 0; -} - -int -tb_init_file(const char *name) -{ - return tb_init_fd(open(name, O_RDWR)); -} - -int -tb_init(void) -{ - return tb_init_file("/dev/tty"); -} - -void -tb_shutdown(void) -{ - if (termw == -1) { - fputs("tb_shutdown() should not be called twice.", stderr); - abort(); - } - - bytebuffer_puts(&output_buffer, funcs[T_SHOW_CURSOR]); - bytebuffer_puts(&output_buffer, funcs[T_SGR0]); - bytebuffer_puts(&output_buffer, funcs[T_CLEAR_SCREEN]); - bytebuffer_puts(&output_buffer, funcs[T_EXIT_CA]); - bytebuffer_puts(&output_buffer, funcs[T_EXIT_KEYPAD]); - bytebuffer_puts(&output_buffer, funcs[T_EXIT_MOUSE]); - bytebuffer_flush(&output_buffer, inout); - tcsetattr(inout, TCSAFLUSH, &orig_tios); - - shutdown_term(); - close(inout); - close(winch_fds[0]); - close(winch_fds[1]); - - cellbuf_free(&back_buffer); - cellbuf_free(&front_buffer); - bytebuffer_free(&output_buffer); - bytebuffer_free(&input_buffer); - termw = termh = -1; -} - -void -tb_present(void) -{ - int x, y, w, i; - struct tb_cell *back, *front; - - /* invalidate cursor position */ - lastx = LAST_COORD_INIT; - lasty = LAST_COORD_INIT; - - if (buffer_size_change_request) { - update_size(); - buffer_size_change_request = 0; - } - - for (y = 0; y < front_buffer.height; ++y) { - for (x = 0; x < front_buffer.width;) { - back = &CELL(&back_buffer, x, y); - front = &CELL(&front_buffer, x, y); - w = wcwidth(back->ch); - if (w < 1) - w = 1; - if (memcmp(back, front, sizeof(struct tb_cell)) == 0) { - x += w; - continue; - } - memcpy(front, back, sizeof(struct tb_cell)); - send_attr(back->fg, back->bg); - if (w > 1 && x >= front_buffer.width - (w - 1)) { - // Not enough room for wide ch, so send spaces - for (i = x; i < front_buffer.width; ++i) { - send_char(i, y, ' '); - } - } else { - send_char(x, y, back->ch); - for (i = 1; i < w; ++i) { - front = &CELL(&front_buffer, x + i, y); - front->ch = 0; - front->fg = back->fg; - front->bg = back->bg; - } - } - x += w; - } - } - if (!IS_CURSOR_HIDDEN(cursor_x, cursor_y)) - write_cursor(cursor_x, cursor_y); - bytebuffer_flush(&output_buffer, inout); -} - -void -tb_set_cursor(int cx, int cy) -{ - if (IS_CURSOR_HIDDEN(cursor_x, cursor_y) && !IS_CURSOR_HIDDEN(cx, cy)) - bytebuffer_puts(&output_buffer, funcs[T_SHOW_CURSOR]); - - if (!IS_CURSOR_HIDDEN(cursor_x, cursor_y) && IS_CURSOR_HIDDEN(cx, cy)) - bytebuffer_puts(&output_buffer, funcs[T_HIDE_CURSOR]); - - cursor_x = cx; - cursor_y = cy; - if (!IS_CURSOR_HIDDEN(cursor_x, cursor_y)) - write_cursor(cursor_x, cursor_y); -} - -void -tb_put_cell(int x, int y, const struct tb_cell *cell) -{ - if ((unsigned)x >= (unsigned)back_buffer.width) - return; - if ((unsigned)y >= (unsigned)back_buffer.height) - return; - CELL(&back_buffer, x, y) = *cell; -} - -void -tb_change_cell(int x, int y, uint32_t ch, uint16_t fg, uint16_t bg) -{ - struct tb_cell c = { ch, fg, bg }; - tb_put_cell(x, y, &c); -} - -void -tb_blit(int x, int y, int w, int h, const struct tb_cell *cells) -{ - if (x + w < 0 || x >= back_buffer.width) - return; - if (y + h < 0 || y >= back_buffer.height) - return; - int xo = 0, yo = 0, ww = w, hh = h; - if (x < 0) { - xo = -x; - ww -= xo; - x = 0; - } - if (y < 0) { - yo = -y; - hh -= yo; - y = 0; - } - if (ww > back_buffer.width - x) - ww = back_buffer.width - x; - if (hh > back_buffer.height - y) - hh = back_buffer.height - y; - - int sy; - struct tb_cell *dst = &CELL(&back_buffer, x, y); - const struct tb_cell *src = cells + yo * w + xo; - size_t size = sizeof(struct tb_cell) * ww; - - for (sy = 0; sy < hh; ++sy) { - memcpy(dst, src, size); - dst += back_buffer.width; - src += w; - } -} - -struct tb_cell * -tb_cell_buffer(void) -{ - return back_buffer.cells; -} - -int -tb_poll_event(struct tb_event *event) -{ - return wait_fill_event(event, 0); -} - -int -tb_peek_event(struct tb_event *event, int timeout) -{ - struct timeval tv; - tv.tv_sec = timeout / 1000; - tv.tv_usec = (timeout - (tv.tv_sec * 1000)) * 1000; - return wait_fill_event(event, &tv); -} - -int -tb_width(void) -{ - return termw; -} - -int -tb_height(void) -{ - return termh; -} - -void -tb_clear(void) -{ - if (buffer_size_change_request) { - update_size(); - buffer_size_change_request = 0; - } - cellbuf_clear(&back_buffer); -} - -int -tb_select_input_mode(int mode) -{ - if (mode) { - if ((mode & (TB_INPUT_ESC | TB_INPUT_ALT)) == 0) - mode |= TB_INPUT_ESC; - - /* technically termbox can handle that, but let's be nice and show here - what mode is actually used */ - if ((mode & (TB_INPUT_ESC | TB_INPUT_ALT)) == - (TB_INPUT_ESC | TB_INPUT_ALT)) - mode &= ~TB_INPUT_ALT; - - inputmode = mode; - if (mode & TB_INPUT_MOUSE) { - bytebuffer_puts(&output_buffer, funcs[T_ENTER_MOUSE]); - bytebuffer_flush(&output_buffer, inout); - } else { - bytebuffer_puts(&output_buffer, funcs[T_EXIT_MOUSE]); - bytebuffer_flush(&output_buffer, inout); - } - } - return inputmode; -} - -int -tb_select_output_mode(int mode) -{ - if (mode) - outputmode = mode; - return outputmode; -} - -void -tb_set_clear_attributes(uint16_t fg, uint16_t bg) -{ - foreground = fg; - background = bg; -} - -/* -------------------------------------------------------- */ - -static int -convertnum(uint32_t num, char *buf) -{ - int i, l = 0; - int ch; - do { - buf[l++] = '0' + (num % 10); - num /= 10; - } while (num); - for (i = 0; i < l / 2; i++) { - ch = buf[i]; - buf[i] = buf[l - 1 - i]; - buf[l - 1 - i] = ch; - } - return l; -} - -static void -write_cursor(int x, int y) -{ - char buf[32]; - WRITE_LITERAL("\033["); - WRITE_INT(y + 1); - WRITE_LITERAL(";"); - WRITE_INT(x + 1); - WRITE_LITERAL("H"); -} - -static void -write_sgr(uint16_t fg, uint16_t bg) -{ - char buf[32]; - - if (fg == TB_DEFAULT && bg == TB_DEFAULT) - return; - - switch (outputmode) { - case TB_OUTPUT_256: - case TB_OUTPUT_216: - case TB_OUTPUT_GRAYSCALE: - WRITE_LITERAL("\033["); - if (fg != TB_DEFAULT) { - WRITE_LITERAL("38;5;"); - WRITE_INT(fg); - if (bg != TB_DEFAULT) { - WRITE_LITERAL(";"); - } - } - if (bg != TB_DEFAULT) { - WRITE_LITERAL("48;5;"); - WRITE_INT(bg); - } - WRITE_LITERAL("m"); - break; - case TB_OUTPUT_NORMAL: - default: - WRITE_LITERAL("\033["); - if (fg != TB_DEFAULT) { - WRITE_LITERAL("3"); - WRITE_INT(fg - 1); - if (bg != TB_DEFAULT) { - WRITE_LITERAL(";"); - } - } - if (bg != TB_DEFAULT) { - WRITE_LITERAL("4"); - WRITE_INT(bg - 1); - } - WRITE_LITERAL("m"); - break; - } -} - -static void -cellbuf_init(struct cellbuf *buf, int width, int height) -{ - buf->cells = (struct tb_cell *)malloc( - sizeof(struct tb_cell) * width * height); - assert(buf->cells); - buf->width = width; - buf->height = height; -} - -static void -cellbuf_resize(struct cellbuf *buf, int width, int height) -{ - if (buf->width == width && buf->height == height) - return; - - int oldw = buf->width; - int oldh = buf->height; - struct tb_cell *oldcells = buf->cells; - - cellbuf_init(buf, width, height); - cellbuf_clear(buf); - - int minw = (width < oldw) ? width : oldw; - int minh = (height < oldh) ? height : oldh; - int i; - - for (i = 0; i < minh; ++i) { - struct tb_cell *csrc = oldcells + (i * oldw); - struct tb_cell *cdst = buf->cells + (i * width); - memcpy(cdst, csrc, sizeof(struct tb_cell) * minw); - } - - free(oldcells); -} - -static void -cellbuf_clear(struct cellbuf *buf) -{ - int i; - int ncells = buf->width * buf->height; - - for (i = 0; i < ncells; ++i) { - buf->cells[i].ch = ' '; - buf->cells[i].fg = foreground; - buf->cells[i].bg = background; - } -} - -static void -cellbuf_free(struct cellbuf *buf) -{ - free(buf->cells); -} - -static void -get_term_size(int *w, int *h) -{ - struct winsize sz; - memset(&sz, 0, sizeof(sz)); - - ioctl(inout, TIOCGWINSZ, &sz); - - *w = sz.ws_col > 0 ? sz.ws_col : 80; - *h = sz.ws_row > 0 ? sz.ws_row : 24; -} - -static void -update_term_size(void) -{ - struct winsize sz; - memset(&sz, 0, sizeof(sz)); - - ioctl(inout, TIOCGWINSZ, &sz); - - termw = sz.ws_col > 0 ? sz.ws_col : 80; - termh = sz.ws_row > 0 ? sz.ws_row : 24; -} - -static void -send_attr(uint16_t fg, uint16_t bg) -{ -#define LAST_ATTR_INIT 0xFFFF - static uint16_t lastfg = LAST_ATTR_INIT, lastbg = LAST_ATTR_INIT; - if (fg != lastfg || bg != lastbg) { - bytebuffer_puts(&output_buffer, funcs[T_SGR0]); - - uint16_t fgcol; - uint16_t bgcol; - - switch (outputmode) { - case TB_OUTPUT_256: - fgcol = fg & 0xFF; - bgcol = bg & 0xFF; - break; - - case TB_OUTPUT_216: - fgcol = fg & 0xFF; - if (fgcol > 215) - fgcol = 7; - bgcol = bg & 0xFF; - if (bgcol > 215) - bgcol = 0; - fgcol += 0x10; - bgcol += 0x10; - break; - - case TB_OUTPUT_GRAYSCALE: - fgcol = fg & 0xFF; - if (fgcol > 23) - fgcol = 23; - bgcol = bg & 0xFF; - if (bgcol > 23) - bgcol = 0; - fgcol += 0xe8; - bgcol += 0xe8; - break; - - case TB_OUTPUT_NORMAL: - default: - fgcol = fg & 0x0F; - bgcol = bg & 0x0F; - } - - if (fg & TB_BOLD) - bytebuffer_puts(&output_buffer, funcs[T_BOLD]); - if (bg & TB_BOLD) - bytebuffer_puts(&output_buffer, funcs[T_BLINK]); - if (fg & TB_UNDERLINE) - bytebuffer_puts(&output_buffer, funcs[T_UNDERLINE]); - if ((fg & TB_REVERSE) || (bg & TB_REVERSE)) - bytebuffer_puts(&output_buffer, funcs[T_REVERSE]); - - write_sgr(fgcol, bgcol); - - lastfg = fg; - lastbg = bg; - } -} - -static void -send_char(int x, int y, uint32_t c) -{ - char buf[7]; - int bw = tb_utf8_unicode_to_char(buf, c); - if (x - 1 != lastx || y != lasty) - write_cursor(x, y); - lastx = x; - lasty = y; - if (!c) - buf[0] = ' '; // replace 0 with whitespace - bytebuffer_append(&output_buffer, buf, bw); -} - -static void -send_clear(void) -{ - send_attr(foreground, background); - bytebuffer_puts(&output_buffer, funcs[T_CLEAR_SCREEN]); - if (!IS_CURSOR_HIDDEN(cursor_x, cursor_y)) - write_cursor(cursor_x, cursor_y); - bytebuffer_flush(&output_buffer, inout); - - /* we need to invalidate cursor position too and these two vars are - * used only for simple cursor positioning optimization, cursor - * actually may be in the correct place, but we simply discard - * optimization once and it gives us simple solution for the case when - * cursor moved */ - lastx = LAST_COORD_INIT; - lasty = LAST_COORD_INIT; -} - -static void -sigwinch_handler(int xxx) -{ - (void)xxx; - const int zzz = 1; - write(winch_fds[1], &zzz, sizeof(int)); -} - -static void -update_size(void) -{ - update_term_size(); - cellbuf_resize(&back_buffer, termw, termh); - cellbuf_resize(&front_buffer, termw, termh); - cellbuf_clear(&front_buffer); - send_clear(); -} - -static int -read_up_to(int n) -{ - assert(n > 0); - const int prevlen = input_buffer.len; - bytebuffer_resize(&input_buffer, prevlen + n); - - int read_n = 0; - while (read_n <= n) { - ssize_t r = 0; - if (read_n < n) { - r = read(inout, input_buffer.buf + prevlen + read_n, - n - read_n); - } -#ifdef __CYGWIN__ - // While linux man for tty says when VMIN == 0 && VTIME == 0, read - // should return 0 when there is nothing to read, cygwin's read returns - // -1. Not sure why and if it's correct to ignore it, but let's pretend - // it's zero. - if (r < 0) - r = 0; -#endif - if (r < 0) { - // EAGAIN / EWOULDBLOCK shouldn't occur here - assert(errno != EAGAIN && errno != EWOULDBLOCK); - return -1; - } else if (r > 0) { - read_n += r; - } else { - bytebuffer_resize(&input_buffer, prevlen + read_n); - return read_n; - } - } - assert(!"unreachable"); - return 0; -} - -static int -wait_fill_event(struct tb_event *event, struct timeval *timeout) -{ - // ;-) -#define ENOUGH_DATA_FOR_PARSING 64 - fd_set events; - memset(event, 0, sizeof(struct tb_event)); - - // try to extract event from input buffer, return on success - event->type = TB_EVENT_KEY; - if (extract_event(event, &input_buffer, inputmode)) - return event->type; - - // it looks like input buffer is incomplete, let's try the short path, - // but first make sure there is enough space - int n = read_up_to(ENOUGH_DATA_FOR_PARSING); - if (n < 0) - return -1; - if (n > 0 && extract_event(event, &input_buffer, inputmode)) - return event->type; - - // n == 0, or not enough data, let's go to select - while (1) { - FD_ZERO(&events); - FD_SET(inout, &events); - FD_SET(winch_fds[0], &events); - int maxfd = (winch_fds[0] > inout) ? winch_fds[0] : inout; - int result = select(maxfd + 1, &events, 0, 0, timeout); - if (!result) - return 0; - - if (FD_ISSET(inout, &events)) { - event->type = TB_EVENT_KEY; - n = read_up_to(ENOUGH_DATA_FOR_PARSING); - if (n < 0) - return -1; - - if (n == 0) - continue; - - if (extract_event(event, &input_buffer, inputmode)) - return event->type; - } - if (FD_ISSET(winch_fds[0], &events)) { - event->type = TB_EVENT_RESIZE; - int zzz = 0; - read(winch_fds[0], &zzz, sizeof(int)); - buffer_size_change_request = 1; - get_term_size(&event->w, &event->h); - return TB_EVENT_RESIZE; - } - } -} - -static void -bytebuffer_reserve(struct bytebuffer *b, int cap) -{ - if (b->cap >= cap) { - return; - } - - // prefer doubling capacity - if (b->cap * 2 >= cap) { - cap = b->cap * 2; - } - - char *newbuf = realloc(b->buf, cap); - b->buf = newbuf; - b->cap = cap; -} - -static void -bytebuffer_init(struct bytebuffer *b, int cap) -{ - b->cap = 0; - b->len = 0; - b->buf = 0; - - if (cap > 0) { - b->cap = cap; - b->buf = malloc(cap); // just assume malloc works always - } -} - -static void -bytebuffer_free(struct bytebuffer *b) -{ - if (b->buf) - free(b->buf); -} - -static void -bytebuffer_clear(struct bytebuffer *b) -{ - b->len = 0; -} - -static void -bytebuffer_append(struct bytebuffer *b, const char *data, int len) -{ - bytebuffer_reserve(b, b->len + len); - memcpy(b->buf + b->len, data, len); - b->len += len; -} - -static void -bytebuffer_puts(struct bytebuffer *b, const char *str) -{ - bytebuffer_append(b, str, strlen(str)); -} - -static void -bytebuffer_resize(struct bytebuffer *b, int len) -{ - bytebuffer_reserve(b, len); - b->len = len; -} - -static void -bytebuffer_flush(struct bytebuffer *b, int fd) -{ - write(fd, b->buf, b->len); - bytebuffer_clear(b); -} - -static void -bytebuffer_truncate(struct bytebuffer *b, int n) -{ - if (n <= 0) - return; - if (n > b->len) - n = b->len; - const int nmove = b->len - n; - memmove(b->buf, b->buf + n, nmove); - b->len -= n; -} |