typetest

typetest.c (4765B)

 #include <unistd.h>
 #include <err.h>
 #include <termio.h>
 #include <stdarg.h>
 #include <time.h>
 #include <stdio.h>
 #include <string.h>
 #include <stdbool.h>
 #include <stdint.h>
 #include <stdlib.h>
 
 #define LINTERP(a,b,v) ((a) * (v) + (b) * (1 - (v)))
 
 enum {
 	C_EOF = 0x03, /* CTRL+C */
 	C_EOT = 0x04, /* CTRL+D */
 	C_DEL = 0x7f, /* BKSP */
 	C_SKIP = 0x17, /* CTRL+W */
 };
 
 enum {
 	STYLE_END = -1,
 
 	STYLE_RESET = 0,
 
 	STYLE_BOLD = 1,
 	STYLE_FAINT = 2,
 	STYLE_ITALIC = 3,
 	STYLE_UNDERLINE = 4,
 	STYLE_BLINK = 5,
 	STYLE_INVERT = 7,
 	STYLE_CROSSED_OUT = 9,
 
 	STYLE_BLACK_FG = 30,
 	STYLE_RED_FG = 31,
 	STYLE_GREEN_FG = 32,
 	STYLE_YELLOW_FG = 33,
 	STYLE_BLUE_FG = 34,
 	STYLE_MAGENTA_FG = 35,
 	STYLE_CYAN_FG = 36,
 	STYLE_WHITE_FG = 37,
 	STYLE_GREY_FG = 90,
 };
 
 static const int style_wrong[] = { STYLE_RED_FG, STYLE_BOLD, STYLE_END };
 static const int style_right[] = { STYLE_GREEN_FG, STYLE_END };
 static const int style_reset[] = { STYLE_RESET, STYLE_END };
 static const int style_preview[] = { STYLE_GREY_FG, STYLE_END };
 
 static struct termios old_termios;
 
 static int typed = 0;
 static float gcps;
 
 void
 print_style(const int *props)
 {
 	const int *iter;
 
 	if (!props || *props == STYLE_END) return;
 
 	printf("\x1b[");
 	for (iter = props; *iter != STYLE_END; iter++)
 		printf("%i%c", *iter, (iter[1] == STYLE_END) ? 'm' : ';');
 }
 
 void
 time_sub(struct timespec *new, struct timespec *old)
 {
 	new->tv_sec -= old->tv_sec;
 	new->tv_nsec -= old->tv_nsec;
 
 	if (new->tv_nsec < 0) {
 		new->tv_sec -= 1;
 		new->tv_nsec += 1e9;
 	}
 }
 
 int
 type(const char *text, const char *preview)
 {
 	int i, c, right, wrong, tlen;
 	struct timespec start = { 0 }, end;
 	float cps;
 
 	if (typed > 0) clock_gettime(CLOCK_REALTIME, &start);
 
 	tlen = strlen(text);
 	c = right = wrong = 0;
 	while (1) {
 		printf("\r\x1b[2K"); /* goto beginning of row and clear */
 		printf("\n\x1b[2K\x1b[1A"); /* down a row and clear */
 
 		switch (c) {
 		case C_EOF:
 			goto quit;
 		case C_EOT:
 			goto exit;
 		case C_DEL:
 			if (wrong) wrong--;
 			else if (right) right--;
 			break;
 		case C_SKIP:
 			right += wrong + 1;
 			if (right >= tlen)
 				goto exit;
 			wrong = 0;
 			break;
 		case u'\r':
 		case u'\n':
 		case u'\0':
 			break;
 		case ' ':
 			if (!wrong && right == tlen)
 				goto exit;
 		default:
 			if (!start.tv_sec)
 				clock_gettime(CLOCK_REALTIME, &start);
 			if (right + wrong == tlen) break;
 			if (c == text[right + wrong] && !wrong) right++;
 			else wrong++;
 			break;
 		}
 
 		print_style(style_right);
 		printf("%.*s", right, text);
 		print_style(style_wrong);
 		for (i = 0; i < wrong; i++)
 			putchar(text[right + i] == ' ' ? '_' : text[right + i]);
 		print_style(style_reset);
 		printf("%.*s", tlen - right - wrong, text + right + wrong);
 
 		print_style(style_preview);
 		printf("\n\r%s", preview);
 		print_style(style_reset);
 
 		printf("\x1b[1A\r");
 		if (right + wrong)
 			printf("\x1b[%iC", right + wrong);
 
 		if (!*preview && !wrong && right == tlen)
 			goto exit;
 
 		c = getchar();
 		if (c < 0) goto quit;
 	};
 
 exit:
 	printf("\r\x1b[2K"); /* goto beginning of line and clear */
 	printf("\n\x1b[2K\x1b[1A"); /* down a row and clear */
 
 	if (!start.tv_sec) return 1;
 
 	clock_gettime(CLOCK_REALTIME, &end);
 	time_sub(&end, &start);
 
 	typed += right;
 
 	end.tv_nsec /= 1000000; /* reduce precision to millis */
 	cps = right / (end.tv_sec + end.tv_nsec / 1e3f);
 	if (!typed) gcps = cps;
 	else gcps = LINTERP(cps, gcps, right * 1.f / typed);
 
 	return 0;
 
 quit:
 	return 1;
 }
 
 int
 readline(FILE *f, char *buf, size_t size)
 {
 	do {
 		if (!fgets(buf, size, f)) return 1;
 
 		for (ssize_t i = strlen(buf) - 1; i >= 0; i--) {
 			if (buf[i] == '\r' || buf[i] == '\n')
 				buf[i] = '\0';
 			else
 				break;
 		}
 		if (strlen(buf) == 0) continue;
 	} while (!*buf);
 
 	return 0;
 }
 
 void
 reset()
 {
 	tcsetattr(0, TCSANOW, &old_termios);
 }
 
 int
 main(int argc, const char **argv)
 {
 	const char *path = NULL;
 	bool retry = true;
 
 	for (const char **arg = argv + 1; *arg; arg++) {
 		if (!strcmp(*arg, "-R")) {
 			retry = false;
 		} else if (!path) {
 			path = *arg;
 		} else {
 			break;
 		}
 	}
 	if (!path) errx(1, "missing file argument");
 
 	FILE *file = fopen(path, "r");
 	if (!file) err(1, "fopen");
 
 	setvbuf(stdin, NULL, _IONBF, 0);
 	setvbuf(stdout, NULL, _IONBF, 0);
 
 	struct termios termios;
 	tcgetattr(0, &termios);
 	old_termios = termios;
 
 	cfmakeraw(&termios);
 
 	if (!tcsetattr(0, TCSANOW, &termios)) atexit(reset);
 
 	char line[256], next_line[256];
 	if (readline(file, next_line, sizeof(next_line)))
 		errx(1, "Empty input file");
 
 	for (bool eof = 0; !eof; ) {
 		strcpy(line, next_line);
 		eof |= readline(file, next_line, sizeof(next_line));
 		if (type(line, eof ? "" : next_line)) break;
 	}
 
 	if (gcps) {
 		printf("CPM: %0.2f\n\r", gcps * 60);
 		printf("WPM: %0.2f\n\r", gcps * 12);
 	}
 
 	fclose(file);
 }