shithub: neatroff

ref: a579ab3890031df70e0c05a9f301f6798991969e
dir: /tr.c/

#include <ctype.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "xroff.h"

#define NARGS		10
#define LINEL		1024
#define LEN(a)		(sizeof(a) / sizeof((a)[0]))

static int tr_nl = 1;

static int unit_scale(int c, int n, int mag)
{
	int mul = 1;
	int div = 1;
	switch (c) {
	case 'i':
		mul = SC_IN;
		break;
	case 'c':
		mul = SC_IN * 50;
		div = 127;
		break;
	case 'p':
		mul = SC_IN;
		div = 72;
		break;
	case 'P':
		mul = SC_IN;
		div = 6;
		break;
	case 'v':
		mul = n_v;
		break;
	case 'm':
		mul = n_s * SC_IN;
		div = 72;
		break;
	case 'n':
		mul = n_s * SC_IN;
		div = 144;
		break;
	}
	/* it may overflow */
	return n * mul / div / mag;
}

int tr_int(char *s, int orig, int unit)
{
	int n = 0;		/* the result */
	int mag = 0;		/* n should be divided by mag */
	int rel = 0;		/* n should be added to orig */
	int neg = *s == '-';	/* n should be negated */
	if (*s == '+' || *s == '-') {
		rel = 1;
		s++;
	}
	while (isdigit(*s) || *s == '.') {
		if (*s == '.') {
			mag = 1;
			s++;
			continue;
		}
		mag *= 10;
		n = n * 10 + *s++ - '0';
	}
	if (!mag)
		mag = 1;
	if (unit)
		n = unit_scale(*s ? *s : unit, n, mag);
	else
		n /= mag;
	if (neg)
		n = -n;
	return rel ? orig + n : n;
}

static void tr_ll(int argc, char **args)
{
	if (argc >= 2)
		n_l = tr_int(args[1], n_l, 'm');
}

static void tr_vs(int argc, char **args)
{
	if (argc >= 2)
		n_v = tr_int(args[1], n_v, 'p');
}

static void tr_ds(int argc, char **args)
{
	if (argc < 3)
		return;
	str_set(N_ID(args[1][0], args[1][1]), args[2]);
}

static void tr_de(int argc, char **args)
{
	struct sbuf sbuf;
	int c;
	if (argc <= 1)
		return;
	sbuf_init(&sbuf);
	while ((c = cp_next()) >= 0) {
		sbuf_add(&sbuf, c);
		if (c == '\n') {
			c = cp_next();
			if (c == '.') {
				c = cp_next();
				if (c == '.')
					break;
				sbuf_add(&sbuf, '.');
				if (c >= 0)
					sbuf_add(&sbuf, c);
			} else {
				if (c >= 0)
					sbuf_add(&sbuf, c);
			}
		}
	}
	str_set(N_ID(args[1][0], args[1][1]), sbuf_buf(&sbuf));
	sbuf_done(&sbuf);
}

static void tr_in(int argc, char **args)
{
	if (argc >= 2)
		n_i = tr_int(args[1], n_i, 'm');
}

static void tr_na(int argc, char **args)
{
	n_ad = 0;
}

static void tr_readcmd(char *s)
{
	int i = 0;
	int c;
	while ((c = cp_next()) == ' ')
		;
	while (i < 2 && c > 0 && isprint(c)) {
		s[i++] = c;
		c = cp_next();
	}
	s[i] = '\0';
	if (!isprint(c))
		cp_back(c);
}

/* skip everything until the end of line */
static void jmp_eol(void)
{
	int c;
	do {
		c = cp_next();
	} while (c >= 0 && c != '\n');
}

/* read macro arguments */
static int tr_args(char **args, int maxargs, char *buf, int len)
{
	char *s = buf;
	char *e = buf + len - 1;
	int c;
	int n = 0;
	while ((c = cp_next()) == ' ')
		;
	while (n < maxargs && s < e && c > 0 && c != '\n') {
		args[n++] = s;
		while (s < e && c > 0 && c != ' ' && c != '\n') {
			*s++ = c;
			c = cp_next();
		}
		*s++ = '\0';
		while (c == ' ')
			c = cp_next();
	}
	if (c >= 0 && c != '\n')
		jmp_eol();
	return n;
}

/* read arguments for .ds */
static int tr_args_ds(char **args, int maxargs, char *buf, int len)
{
	char *s = buf;
	char *e = buf + len - 1;
	int c;
	while ((c = cp_next()) == ' ')
		;
	if (c == '\n' || c < 0)
		return 0;
	args[0] = s;
	*s++ = c;
	c = cp_next();
	if (c >= 0 && c != '\n' && c != ' ')
		*s++ = c;
	*s++ = '\0';
	while ((c = cp_next()) == ' ')
		;
	if (c == '"')
		c = cp_next();
	args[1] = s;
	while (s < e && c > 0 && c != '\n') {
		*s++ = c;
		c = cp_next();
	}
	*s = '\0';
	return 2;
}

static struct cmd {
	char *id;
	void (*f)(int argc, char **args);
	int (*args)(char **args, int maxargs, char *buf, int len);
} cmds[] = {
	{"br", tr_br},
	{"de", tr_de},
	{"ds", tr_ds, tr_args_ds},
	{"fp", tr_fp},
	{"ft", tr_ft},
	{"in", tr_in},
	{"ll", tr_ll},
	{"na", tr_na},
	{"nr", tr_nr},
	{"ps", tr_ps},
	{"sp", tr_sp},
	{"vs", tr_vs},
};

int tr_next(void)
{
	int c = cp_next();
	int nl = c == '\n';
	char *args[NARGS];
	char buf[LINEL];
	char cmd[LINEL];
	struct cmd *req = NULL;
	int argc;
	int i;
	while (tr_nl && (c == '.' || c == '\'')) {
		nl = 1;
		args[0] = cmd;
		cmd[0] = c;
		tr_readcmd(cmd + 1);
		for (i = 0; i < LEN(cmds); i++)
			if (!strcmp(cmd + 1, cmds[i].id))
				req = &cmds[i];
		if (req) {
			if (req->args)
				argc = req->args(args + 1, NARGS - 1, buf, LINEL);
			else
				argc = tr_args(args + 1, NARGS - 1, buf, LINEL);
			req->f(argc + 1, args);
		} else {
			jmp_eol();
		}
		c = cp_next();
	}
	tr_nl = nl;
	return c;
}