diff options
Diffstat (limited to 'counter.c')
| -rw-r--r-- | counter.c | 258 |
1 files changed, 258 insertions, 0 deletions
diff --git a/counter.c b/counter.c new file mode 100644 index 0000000..df9c3f2 --- /dev/null +++ b/counter.c @@ -0,0 +1,258 @@ +/* See LICENSE file for copyright and license details. */ +#include "common.h" + +/* TODO add buttons */ + +USAGE("[-2s]"); + + +#define INCLUDE_LARGE_MINUS +#include "large.h" + + +static volatile sig_atomic_t caught_sigterm = 0; +static volatile sig_atomic_t caught_sigwinch = 1; +static volatile sig_atomic_t caught_sigio = 1; +static int with_small = 0; +static int with_quad = 0; + +static const char **large_digits[] = { + large_0, large_1, large_2, large_3, large_4, + large_5, large_6, large_7, large_8, large_9 +}; + +static const char **small_digits[] = { + small_0, small_1, small_2, small_3, small_4, + small_5, small_6, small_7, small_8, small_9 +}; + + + +static void +sigterm(int signo) +{ + caught_sigterm = 1; + (void) signo; +} + +static void +sigwinch(int signo) +{ + caught_sigwinch = 1; + (void) signo; +} + +static void +sigio(int signo) +{ + caught_sigio = 1; + (void) signo; +} + + +static void +print_number(const char ***str, int left, size_t height) +{ + size_t r, c; + + for (r = 0;;) { + printf("%*.s", left, ""); + for (c = 0; str[c]; c++) + fprintf(stdout, "%s", str[c][r]); + if (++r == height) { + fprintf(stdout, "\033[K"); + break; + } else { + fprintf(stdout, "\033[K\n"); + } + } +} + + +static int +display_counter(void) +{ + char buf[sizeof(intmax_t) * 3 + 2]; + const char **digits[sizeof(intmax_t) * 3 + 2]; + intmax_t counter = {0}; + int r, i, x = 0, y = 0, was_quad = 0, len, width; + struct winsize winsize; + char c; + ssize_t rd; + + while (!caught_sigterm) { + if (caught_sigwinch) { + if (ioctl(STDOUT_FILENO, (unsigned long)TIOCGWINSZ, &winsize) < 0) { + if (errno == EINTR) + continue; + goto fail; + } + caught_sigwinch = 0; + y = winsize.ws_row; + x = winsize.ws_col; + if (was_quad) { + was_quad = 0; + for (r = 0; r < y; r++) + printf("%s\033#5", r ? "\n" : "\033[H"); + } + } + + if (caught_sigio) { + caught_sigio = 0; + for (;;) { + rd = read(STDIN_FILENO, &c, 1); + if (rd <= 0) { + if (!rd) + goto out; + if (errno == EAGAIN) + break; + if (errno == EINTR) + continue; + goto fail; + } + if (c == 'q') + goto out; + else if (c == 'r') + counter = 0; + else if (c == '+' || c == 'C') + counter += 1; + else if (c == '-' || c == 'D') + counter -= 1; + } + } + + len = sprintf(buf, "%ji", counter); + + printf("\033[H"); + + if (x >= (width = len * LARGE_X + (counter < 0 ? LARGE_XM - LARGE_X : 0)) && y >= LARGE_Y) { + for (r = 0; r < (y - LARGE_Y) / 2; r++) + printf("\033[K\n"); + for (i = 0; i < len; i++) { + if (isdigit(buf[i])) + digits[i] = large_digits[buf[i] & 15]; + else + digits[i] = large_minus; + } + digits[i] = NULL; + print_number(digits, (x - width) / 2, LARGE_Y); + } else if (with_small && x >= (width = len * SMALL_X + (counter < 0 ? SMALL_XM - SMALL_X : 0)) && y > SMALL_Y + 1) { + for (r = 0; r < (y - SMALL_Y) / 2; r++) + printf("\033[K\n"); + for (i = 0; i < len; i++) { + if (isdigit(buf[i])) + digits[i] = small_digits[buf[i] & 15]; + else + digits[i] = small_minus; + } + digits[i] = NULL; + print_number(digits, (x - width) / 2, SMALL_Y); + } else if (with_quad && x > len * 2 && y > 2) { + printf("\033#3%*.s%s\033[K\n", (x / 2 - len) / 2, "", buf); + printf("\033#4%*.s%s\033[K\n", (x / 2 - len) / 2, "", buf); + printf("\033#5"); + was_quad = 1; + } else { + printf("%*.s%s\033[K", (x - len) / 2, "", buf); + } + + printf("\033[J"); + fflush(stdout); + + pause(); + } + +out: + return 0; + +fail: + return -1; +} + + +int +main(int argc, char *argv[]) +{ + int tcset = 0, old_flags = -1, old_sig = 0, owner_set = 0; + struct sigaction sigact; + struct termios stty, saved_stty; + struct f_owner_ex old_owner, new_owner; + + ARGBEGIN { + case '2': + with_quad = 1; + break; + case 's': + with_small = 1; + break; + default: + usage(); + } ARGEND; + + if (argc) + usage(); + + tzset(); + + fprintf(stdout, "\033[?1049h\033[?25l"); + + memset(&sigact, 0, sizeof(sigact)); + + sigact.sa_handler = sigterm; + sigaction(SIGTERM, &sigact, NULL); + sigaction(SIGQUIT, &sigact, NULL); + sigaction(SIGINT, &sigact, NULL); + + sigact.sa_handler = sigwinch; + sigaction(SIGWINCH, &sigact, NULL); + + sigact.sa_handler = sigio; + sigaction(SIGIO, &sigact, NULL); + sigaction(SIGURG, &sigact, NULL); + + if (fcntl(STDIN_FILENO, F_GETOWN_EX, &old_owner)) + goto fail; + memset(&new_owner, 0, sizeof(new_owner)); + new_owner.type = F_OWNER_PID; + new_owner.pid = getpid(); + if (fcntl(STDIN_FILENO, F_SETOWN_EX, &new_owner)) + goto fail; + owner_set = 1; + old_flags = fcntl(STDIN_FILENO, F_GETFL); + fcntl(STDIN_FILENO, F_SETFL, old_flags | FASYNC | O_NONBLOCK); + fcntl(STDIN_FILENO, F_GETSIG, &old_sig); + if (old_sig) + fcntl(STDIN_FILENO, F_SETSIG, 0); + + if (!tcgetattr(STDIN_FILENO, &stty)) { + saved_stty = stty; + stty.c_lflag &= (tcflag_t)~(ECHO | ICANON); + tcsetattr(STDIN_FILENO, TCSAFLUSH, &stty); + tcset = 1; + } + + if (display_counter()) + goto fail; + + fprintf(stdout, "\033[?25h\n\033[?1049l"); + fflush(stdout); + fcntl(STDIN_FILENO, F_SETOWN_EX, &old_owner); + fcntl(STDIN_FILENO, F_SETFL, old_flags); + fcntl(STDIN_FILENO, F_SETSIG, old_sig); + tcsetattr(STDIN_FILENO, TCSAFLUSH, &saved_stty); + return 0; + +fail: + perror(argv0 ? argv0 : "counter"); + fprintf(stdout, "\033[?25h\n\033[?1049l"); + fflush(stdout); + if (owner_set) + fcntl(STDIN_FILENO, F_SETOWN_EX, &old_owner); + if (old_flags != -1) + fcntl(STDIN_FILENO, F_SETFL, old_flags); + if (old_sig) + fcntl(STDIN_FILENO, F_SETSIG, old_sig); + if (tcset) + tcsetattr(STDIN_FILENO, TCSAFLUSH, &saved_stty); + return 1; +} |