tr.c

   1 /* built-in troff requests */
   2 #include <ctype.h>
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5 #include <string.h>
   6 #include "roff.h"
   7
   8 static int tr_nl = 1;           /* just read a newline */
   9 static int c_pc = '%';          /* page number character */
  10 int c_ec = '\\';
  11 int c_cc = '.';
  12 int c_c2 = '\'';
  13
  14 /* skip everything until the end of line */
  15 static void jmp_eol(void)
  16 {
  17         int c;
  18         do {
  19                 c = cp_next();
  20         } while (c >= 0 && c != '\n');
  21 }
  22
  23 static void tr_vs(char **args)
  24 {
  25         int vs = args[1] ? eval_re(args[1], n_v, 'p') : n_v0;
  26         n_v0 = n_v;
  27         n_v = MAX(0, vs);
  28 }
  29
  30 static void tr_ls(char **args)
  31 {
  32         int ls = args[1] ? eval_re(args[1], n_L, 0) : n_L0;
  33         n_L0 = n_L;
  34         n_L = MAX(1, ls);
  35 }
  36
  37 static void tr_pl(char **args)
  38 {
  39         int n = eval_re(args[1] ? args[1] : "11i", n_p, 'v');
  40         n_p = MAX(0, n);
  41 }
  42
  43 static void tr_nr(char **args)
  44 {
  45         int id;
  46         if (!args[2])
  47                 return;
  48         id = map(args[1]);
  49         num_set(id, eval_re(args[2], num_get(id), 'u'));
  50         num_setinc(id, args[3] ? eval(args[3], 'u') : 0);
  51 }
  52
  53 static void tr_rr(char **args)
  54 {
  55         int i;
  56         for (i = 1; i <= NARGS; i++)
  57                 if (args[i])
  58                         num_del(map(args[i]));
  59 }
  60
  61 static void tr_af(char **args)
  62 {
  63         if (args[2])
  64                 num_setfmt(map(args[1]), args[2]);
  65 }
  66
  67 static void tr_ds(char **args)
  68 {
  69         if (args[2])
  70                 str_set(map(args[1]), args[2]);
  71 }
  72
  73 static void tr_as(char **args)
  74 {
  75         int reg;
  76         char *s1, *s2, *s;
  77         if (!args[2])
  78                 return;
  79         reg = map(args[1]);
  80         s1 = str_get(reg) ? str_get(reg) : "";
  81         s2 = args[2];
  82         s = xmalloc(strlen(s1) + strlen(s2) + 1);
  83         strcpy(s, s1);
  84         strcat(s, s2);
  85         str_set(reg, s);
  86         free(s);
  87 }
  88
  89 static void tr_rm(char **args)
  90 {
  91         int i;
  92         for (i = 1; i <= NARGS; i++)
  93                 if (args[i])
  94                         str_rm(map(args[i]));
  95 }
  96
  97 static void tr_rn(char **args)
  98 {
  99         if (!args[2])
 100                 return;
 101         str_rn(map(args[1]), map(args[2]));
 102 }
 103
 104 static void tr_po(char **args)
 105 {
 106         int po = args[1] ? eval_re(args[1], n_o, 'm') : n_o0;
 107         n_o0 = n_o;
 108         n_o = MAX(0, po);
 109 }
 110
 111 static void read_regname(char *s)
 112 {
 113         int c = cp_next();
 114         int n = n_cp ? 2 : NMLEN - 1;
 115         while (c == ' ' || c == '\t')
 116                 c = cp_next();
 117         while (c >= 0 && c != ' ' && c != '\t' && c != '\n' && --n >= 0) {
 118                 *s++ = c;
 119                 c = cp_next();
 120         }
 121         if (c >= 0)
 122                 cp_back(c);
 123         *s = '\0';
 124 }
 125
 126 static void macrobody(struct sbuf *sbuf, char *end)
 127 {
 128         char buf[NMLEN];
 129         int i, c;
 130         int first = 1;
 131         cp_back('\n');
 132         cp_copymode(1);
 133         while ((c = cp_next()) >= 0) {
 134                 if (sbuf && !first)
 135                         sbuf_add(sbuf, c);
 136                 first = 0;
 137                 if (c == '\n') {
 138                         if ((c = cp_next()) != '.') {
 139                                 cp_back(c);
 140                                 continue;
 141                         }
 142                         read_regname(buf);
 143                         if ((n_cp && end[0] == buf[0] && end[1] == buf[1]) ||
 144                                                 !strcmp(end, buf)) {
 145                                 jmp_eol();
 146                                 break;
 147                         }
 148                         if (sbuf) {
 149                                 sbuf_add(sbuf, '.');
 150                                 for (i = 0; buf[i]; i++)
 151                                         sbuf_add(sbuf, (unsigned char) buf[i]);
 152                         }
 153                 }
 154         }
 155         cp_copymode(0);
 156 }
 157
 158 static void tr_de(char **args)
 159 {
 160         struct sbuf sbuf;
 161         int id;
 162         if (!args[1])
 163                 return;
 164         id = map(args[1]);
 165         sbuf_init(&sbuf);
 166         if (args[0][1] == 'a' && args[0][2] == 'm' && str_get(id))
 167                 sbuf_append(&sbuf, str_get(id));
 168         macrobody(&sbuf, args[2] ? args[2] : ".");
 169         str_set(id, sbuf_buf(&sbuf));
 170         sbuf_done(&sbuf);
 171 }
 172
 173 static void tr_ig(char **args)
 174 {
 175         macrobody(NULL, args[1] ? args[1] : ".");
 176 }
 177
 178 /* read into sbuf until stop; if stop is NULL, stop at whitespace */
 179 static int read_until(struct sbuf *sbuf, char *stop,
 180                         int (*next)(void), void (*back)(int))
 181 {
 182         char cs[GNLEN], cs2[GNLEN];
 183         int c;
 184         while ((c = next()) >= 0) {
 185                 back(c);
 186                 if (c == '\n')
 187                         return 1;
 188                 if (!stop && (c == ' ' || c == '\t'))
 189                         return 0;
 190                 charnext(cs, next, back);
 191                 if (stop && !strcmp(stop, cs))
 192                         return 0;
 193                 charnext_str(cs2, cs);
 194                 sbuf_append(sbuf, cs2);
 195         }
 196         return 1;
 197 }
 198
 199 /* evaluate .if strcmp (i.e. 'str'str') */
 200 static int if_strcmp(int (*next)(void), void (*back)(int))
 201 {
 202         char delim[GNLEN];
 203         struct sbuf s1, s2;
 204         int ret;
 205         charnext(delim, next, back);
 206         sbuf_init(&s1);
 207         sbuf_init(&s2);
 208         read_until(&s1, delim, next, back);
 209         read_until(&s2, delim, next, back);
 210         ret = !strcmp(sbuf_buf(&s1), sbuf_buf(&s2));
 211         sbuf_done(&s1);
 212         sbuf_done(&s2);
 213         return ret;
 214 }
 215
 216 /* evaluate .if condition letters */
 217 static int if_cond(int (*next)(void), void (*back)(int))
 218 {
 219         switch (cp_next()) {
 220         case 'o':
 221                 return n_pg % 2;
 222         case 'e':
 223                 return !(n_pg % 2);
 224         case 't':
 225                 return 1;
 226         case 'n':
 227                 return 0;
 228         }
 229         return 0;
 230 }
 231
 232 /* evaluate .if condition */
 233 static int if_eval(int (*next)(void), void (*back)(int))
 234 {
 235         struct sbuf sbuf;
 236         int ret;
 237         sbuf_init(&sbuf);
 238         read_until(&sbuf, NULL, next, back);
 239         ret = eval(sbuf_buf(&sbuf), '\0') > 0;
 240         sbuf_done(&sbuf);
 241         return ret;
 242 }
 243
 244 static int eval_if(int (*next)(void), void (*back)(int))
 245 {
 246         int neg = 0;
 247         int ret;
 248         int c;
 249         do {
 250                 c = next();
 251         } while (c == ' ' || c == '\t');
 252         if (c == '!') {
 253                 neg = 1;
 254                 c = next();
 255         }
 256         back(c);
 257         if (strchr("oetn", c)) {
 258                 ret = if_cond(next, back);
 259         } else if (!isdigit(c) && !strchr("-+*/%<=>&:.|()", c)) {
 260                 ret = if_strcmp(next, back);
 261         } else {
 262                 ret = if_eval(next, back);
 263         }
 264         return ret != neg;
 265 }
 266
 267 static int ie_cond[NIES];       /* .ie condition stack */
 268 static int ie_depth;
 269
 270 static void tr_if(char **args)
 271 {
 272         int c = eval_if(cp_next, cp_back);
 273         if (args[0][1] == 'i' && args[0][2] == 'e')     /* .ie command */
 274                 if (ie_depth < NIES)
 275                         ie_cond[ie_depth++] = c;
 276         cp_blk(!c);
 277 }
 278
 279 static void tr_el(char **args)
 280 {
 281         cp_blk(ie_depth > 0 ? ie_cond[--ie_depth] : 1);
 282 }
 283
 284 static void tr_na(char **args)
 285 {
 286         n_na = 1;
 287 }
 288
 289 static int adjmode(int c, int def)
 290 {
 291         switch (c) {
 292         case 'l':
 293                 return AD_L;
 294         case 'r':
 295                 return AD_R;
 296         case 'c':
 297                 return AD_C;
 298         case 'b':
 299         case 'n':
 300                 return AD_B;
 301         }
 302         return def;
 303 }
 304
 305 static void tr_ad(char **args)
 306 {
 307         char *s = args[1];
 308         n_na = 0;
 309         if (!s)
 310                 return;
 311         if (isdigit(s[0]))
 312                 n_j = atoi(s) & 15;
 313         else
 314                 n_j = s[0] == 'p' ? AD_P | adjmode(s[1], AD_B) : adjmode(s[0], n_j);
 315 }
 316
 317 static void tr_tm(char **args)
 318 {
 319         fprintf(stderr, "%s\n", args[1]);
 320 }
 321
 322 static void tr_so(char **args)
 323 {
 324         if (args[1])
 325                 in_so(args[1]);
 326 }
 327
 328 static void tr_nx(char **args)
 329 {
 330         in_nx(args[1]);
 331 }
 332
 333 static void tr_ex(char **args)
 334 {
 335         in_ex();
 336 }
 337
 338 static void tr_sy(char **args)
 339 {
 340         system(args[1]);
 341 }
 342
 343 static void tr_lt(char **args)
 344 {
 345         int lt = args[1] ? eval_re(args[1], n_lt, 'm') : n_t0;
 346         n_t0 = n_t0;
 347         n_lt = MAX(0, lt);
 348 }
 349
 350 static void tr_pc(char **args)
 351 {
 352         c_pc = args[1] ? args[1][0] : -1;
 353 }
 354
 355 static int tl_next(void)
 356 {
 357         int c = cp_next();
 358         if (c >= 0 && c == c_pc) {
 359                 in_push(num_str(map("%")), NULL);
 360                 c = cp_next();
 361         }
 362         return c;
 363 }
 364
 365 static void tr_tl(char **args)
 366 {
 367         int c;
 368         do {
 369                 c = cp_next();
 370         } while (c >= 0 && (c == ' ' || c == '\t'));
 371         cp_back(c);
 372         ren_tl(tl_next, cp_back);
 373         do {
 374                 c = cp_next();
 375         } while (c >= 0 && c != '\n');
 376 }
 377
 378 static void tr_ec(char **args)
 379 {
 380         c_ec = args[1] ? args[1][0] : '\\';
 381 }
 382
 383 static void tr_cc(char **args)
 384 {
 385         c_ec = args[1] ? args[1][0] : '.';
 386 }
 387
 388 static void tr_c2(char **args)
 389 {
 390         c_ec = args[1] ? args[1][0] : '\'';
 391 }
 392
 393 static void tr_eo(char **args)
 394 {
 395         c_ec = -1;
 396 }
 397
 398 static void tr_hc(char **args)
 399 {
 400         char *s = args[1];
 401         if (!s || charread(&s, c_hc) < 0)
 402                 strcpy(c_hc, "\\%");
 403 }
 404
 405 static void tr_nh(char **args)
 406 {
 407         n_hy = 0;
 408 }
 409
 410 static void tr_hy(char **args)
 411 {
 412         n_hy = args[1] ? atoi(args[1]) : 1;
 413 }
 414
 415 static void tr_hyp(char **args)
 416 {
 417         n_hyp = args[1] ? atoi(args[1]) : 1;
 418 }
 419
 420 static void tr_lg(char **args)
 421 {
 422         if (args[1])
 423                 n_lg = atoi(args[1]);
 424 }
 425
 426 static void tr_kn(char **args)
 427 {
 428         if (args[1])
 429                 n_kn = atoi(args[1]);
 430 }
 431
 432 static void tr_cp(char **args)
 433 {
 434         if (args[1])
 435                 n_cp = atoi(args[1]);
 436 }
 437
 438 static void tr_ss(char **args)
 439 {
 440         if (args[1]) {
 441                 n_ss = eval_re(args[1], n_ss, 0);
 442                 n_sss = args[2] ? eval_re(args[2], n_sss, 0) : n_ss;
 443         }
 444 }
 445
 446 static void tr_ssh(char **args)
 447 {
 448         n_ssh = args[1] ? eval_re(args[1], n_ssh, 0) : 0;
 449 }
 450
 451 static void tr_cs(char **args)
 452 {
 453         struct font *fn = args[1] ? dev_font(dev_pos(args[1])) : NULL;
 454         if (fn)
 455                 font_setcs(fn, args[2] ? eval(args[2], 0) : 0,
 456                                 args[3] ? eval(args[3], 0) : 0);
 457 }
 458
 459 static void tr_fzoom(char **args)
 460 {
 461         struct font *fn = args[1] ? dev_font(dev_pos(args[1])) : NULL;
 462         if (fn)
 463                 font_setzoom(fn, args[2] ? eval(args[2], 0) : 0);
 464 }
 465
 466 static void tr_ff(char **args)
 467 {
 468         struct font *fn = args[1] ? dev_font(dev_pos(args[1])) : NULL;
 469         int i;
 470         for (i = 2; i <= NARGS; i++)
 471                 if (fn && args[i] && args[i][0] && args[i][1])
 472                         font_feat(fn, args[i] + 1, args[i][0] == '+');
 473 }
 474
 475 static void tr_nm(char **args)
 476 {
 477         if (!args[1]) {
 478                 n_nm = 0;
 479                 return;
 480         }
 481         n_nm = 1;
 482         n_ln = eval_re(args[1], n_ln, 0);
 483         n_ln = MAX(0, n_ln);
 484         if (args[2] && isdigit(args[2][0]))
 485                 n_nM = MAX(1, eval(args[2], 0));
 486         if (args[3] && isdigit(args[3][0]))
 487                 n_nS = MAX(0, eval(args[3], 0));
 488         if (args[4] && isdigit(args[4][0]))
 489                 n_nI = MAX(0, eval(args[4], 0));
 490 }
 491
 492 static void tr_nn(char **args)
 493 {
 494         n_nn = args[1] ? eval(args[1], 0) : 1;
 495 }
 496
 497 static void tr_bd(char **args)
 498 {
 499         if (!args[1] || !strcmp("S", args[1]))
 500                 return;
 501         font_setbd(dev_font(dev_pos(args[1])), args[2] ? eval(args[2], 'u') : 0);
 502 }
 503
 504 static void tr_it(char **args)
 505 {
 506         if (args[2]) {
 507                 n_it = map(args[2]);
 508                 n_itn = eval(args[1], 0);
 509         } else {
 510                 n_it = 0;
 511         }
 512 }
 513
 514 static void tr_mc(char **args)
 515 {
 516         char *s = args[1];
 517         if (s && charread(&s, c_mc) >= 0) {
 518                 n_mc = 1;
 519                 n_mcn = args[2] ? eval(args[2], 'm') : SC_EM;
 520         } else {
 521                 n_mc = 0;
 522         }
 523 }
 524
 525 static void tr_tc(char **args)
 526 {
 527         char *s = args[1];
 528         if (!s || charread(&s, c_tc) < 0)
 529                 strcpy(c_tc, "");
 530 }
 531
 532 static void tr_lc(char **args)
 533 {
 534         char *s = args[1];
 535         if (!s || charread(&s, c_lc) < 0)
 536                 strcpy(c_lc, "");
 537 }
 538
 539 static void tr_lf(char **args)
 540 {
 541         if (args[1])
 542                 in_lf(args[2], eval(args[1], 0));
 543 }
 544
 545 static void tr_chop(char **args)
 546 {
 547         struct sbuf sbuf;
 548         int id;
 549         if (args[1])
 550                 in_lf(args[2], eval(args[1], 0));
 551         id = map(args[1]);
 552         if (str_get(id)) {
 553                 sbuf_init(&sbuf);
 554                 sbuf_append(&sbuf, str_get(id));
 555                 if (!sbuf_empty(&sbuf)) {
 556                         sbuf_cut(&sbuf, sbuf_len(&sbuf) - 1);
 557                         str_set(id, sbuf_buf(&sbuf));
 558                 }
 559                 sbuf_done(&sbuf);
 560         }
 561 }
 562
 563 /* character translation (.tr) */
 564 static struct dict cmap;                /* character mapping */
 565 static char cmap_src[NCMAPS][GNLEN];    /* source character */
 566 static char cmap_dst[NCMAPS][GNLEN];    /* character mapping */
 567 static int cmap_n;                      /* number of translated character */
 568
 569 void cmap_add(char *c1, char *c2)
 570 {
 571         int i = dict_get(&cmap, c1);
 572         if (i >= 0) {
 573                 strcpy(cmap_dst[i], c2);
 574         } else if (cmap_n < NCMAPS) {
 575                 strcpy(cmap_src[cmap_n], c1);
 576                 strcpy(cmap_dst[cmap_n], c2);
 577                 dict_put(&cmap, cmap_src[cmap_n], cmap_n);
 578                 cmap_n++;
 579         }
 580 }
 581
 582 char *cmap_map(char *c)
 583 {
 584         int i = dict_get(&cmap, c);
 585         return i >= 0 ? cmap_dst[i] : c;
 586 }
 587
 588 static void tr_tr(char **args)
 589 {
 590         char *s = args[1];
 591         char c1[GNLEN], c2[GNLEN];
 592         while (s && charread(&s, c1) >= 0) {
 593                 if (charread(&s, c2) < 0)
 594                         strcpy(c2, " ");
 595                 cmap_add(c1, c2);
 596         }
 597 }
 598
 599 /* character definition (.char) */
 600 static char cdef_src[NCDEFS][GNLEN];    /* source character */
 601 static char *cdef_dst[NCDEFS];          /* character definition */
 602 static int cdef_fn[NCDEFS];             /* owning font */
 603 static int cdef_n;                      /* number of defined characters */
 604 static int cdef_expanding;              /* inside cdef_expand() call */
 605
 606 static int cdef_find(char *c, int fn)
 607 {
 608         int i;
 609         for (i = 0; i < cdef_n; i++)
 610                 if ((!cdef_fn[i] || cdef_fn[i] == fn) && !strcmp(cdef_src[i], c))
 611                         return i;
 612         return -1;
 613 }
 614
 615 /* return the definition of the given character */
 616 char *cdef_map(char *c, int fn)
 617 {
 618         int i = cdef_find(c, fn);
 619         return !cdef_expanding && i >= 0 ? cdef_dst[i] : NULL;
 620 }
 621
 622 int cdef_expand(struct wb *wb, char *s, int fn)
 623 {
 624         char *d = cdef_map(s, fn);
 625         if (!d)
 626                 return 1;
 627         cdef_expanding = 1;
 628         ren_parse(wb, d);
 629         cdef_expanding = 0;
 630         return 0;
 631 }
 632
 633 static void cdef_add(char *fn, char *cs, char *def)
 634 {
 635         char c[GNLEN];
 636         int i;
 637         if (!def || charread(&cs, c) < 0)
 638                 return;
 639         i = cdef_find(c, -1);
 640         if (i < 0 && cdef_n < NCDEFS)
 641                 i = cdef_n++;
 642         if (i >= 0) {
 643                 strncpy(cdef_src[i], c, sizeof(cdef_src[i]) - 1);
 644                 cdef_dst[i] = xmalloc(strlen(def) + 1);
 645                 strcpy(cdef_dst[i], def);
 646                 cdef_fn[i] = fn ? dev_pos(fn) : 0;
 647         }
 648 }
 649
 650 static void cdef_remove(char *cs)
 651 {
 652         char c[GNLEN];
 653         int i;
 654         if (!cs || charread(&cs, c) < 0)
 655                 return;
 656         for (i = 0; i < cdef_n; i++) {
 657                 if (!strcmp(cdef_src[i], c)) {
 658                         free(cdef_dst[i]);
 659                         cdef_dst[i] = NULL;
 660                         cdef_src[i][0] = '\0';
 661                 }
 662         }
 663 }
 664
 665 static void tr_char(char **args)
 666 {
 667         cdef_add(NULL, args[1], args[2]);
 668 }
 669
 670 static void tr_rchar(char **args)
 671 {
 672         int i;
 673         for (i = 1; i <= NARGS; i++)
 674                 if (args[i])
 675                         cdef_remove(args[i]);
 676 }
 677
 678 static void tr_ochar(char **args)
 679 {
 680         cdef_add(args[1], args[2], args[3]);
 681 }
 682
 683 static void tr_fmap(char **args)
 684 {
 685         struct font *fn = args[1] ? dev_font(dev_pos(args[1])) : NULL;
 686         if (fn && args[2])
 687                 font_map(fn, args[2], args[3]);
 688 }
 689
 690 static void arg_regname(struct sbuf *sbuf)
 691 {
 692         char reg[NMLEN];
 693         read_regname(reg);
 694         sbuf_append(sbuf, reg);
 695         sbuf_add(sbuf, 0);
 696 }
 697
 698 static void arg_string(struct sbuf *sbuf)
 699 {
 700         int c;
 701         while ((c = cp_next()) == ' ')
 702                 ;
 703         if (c == '"')
 704                 c = cp_next();
 705         while (c > 0 && c != '\n') {
 706                 sbuf_add(sbuf, c);
 707                 c = cp_next();
 708         }
 709         sbuf_add(sbuf, 0);
 710         if (c >= 0)
 711                 cp_back(c);
 712 }
 713
 714 static int mkargs_arg(struct sbuf *sbuf, int (*next)(void), void (*back)(int))
 715 {
 716         int quoted = 0;
 717         int c;
 718         c = next();
 719         while (c == ' ')
 720                 c = next();
 721         if (c == '\n')
 722                 back(c);
 723         if (c < 0 || c == '\n')
 724                 return 1;
 725         if (c == '"') {
 726                 quoted = 1;
 727                 c = next();
 728         }
 729         while (c >= 0 && c != '\n') {
 730                 if (!quoted && c == ' ')
 731                         break;
 732                 if (quoted && c == '"') {
 733                         c = next();
 734                         if (c != '"')
 735                                 break;
 736                 }
 737                 sbuf_add(sbuf, c);
 738                 c = next();
 739         }
 740         sbuf_add(sbuf, 0);
 741         if (c >= 0)
 742                 back(c);
 743         return 0;
 744 }
 745
 746 /* read macro arguments */
 747 int tr_readargs(char **args, struct sbuf *sbuf, int (*next)(void), void (*back)(int))
 748 {
 749         int idx[NARGS];
 750         int i, n = 0;
 751         while (n < NARGS) {
 752                 idx[n] = sbuf_len(sbuf);
 753                 if (mkargs_arg(sbuf, next, back))
 754                         break;
 755                 n++;
 756         }
 757         for (i = 0; i < n; i++)
 758                 args[i] = sbuf_buf(sbuf) + idx[i];
 759         return n;
 760 }
 761
 762 /* read macro arguments; trims tabs if rmtabs is nonzero */
 763 static int mkargs(char **args, struct sbuf *sbuf)
 764 {
 765         int n = tr_readargs(args, sbuf, cp_next, cp_back);
 766         jmp_eol();
 767         return n;
 768 }
 769
 770 /* read request arguments; trims tabs too */
 771 static int mkargs_req(char **args, struct sbuf *sbuf)
 772 {
 773         int idx[NARGS];
 774         int i, n = 0;
 775         int c;
 776         c = cp_next();
 777         while (n < NARGS) {
 778                 idx[n] = sbuf_len(sbuf);
 779                 while (c == ' ' || c == '\t')
 780                         c = cp_next();
 781                 while (c >= 0 && c != '\n' && c != ' ' && c != '\t') {
 782                         sbuf_add(sbuf, c);
 783                         c = cp_next();
 784                 }
 785                 if (sbuf_len(sbuf) > idx[n])
 786                         n++;
 787                 sbuf_add(sbuf, 0);
 788                 if (c == '\n')
 789                         cp_back(c);
 790                 if (c < 0 || c == '\n')
 791                         break;
 792         }
 793         for (i = 0; i < n; i++)
 794                 args[i] = sbuf_buf(sbuf) + idx[i];
 795         jmp_eol();
 796         return n;
 797 }
 798
 799 /* read arguments for .ds */
 800 static int mkargs_ds(char **args, struct sbuf *sbuf)
 801 {
 802         int idx[NARGS];
 803         int i, n = 0;
 804         idx[n++] = sbuf_len(sbuf);
 805         arg_regname(sbuf);
 806         idx[n++] = sbuf_len(sbuf);
 807         cp_copymode(1);
 808         arg_string(sbuf);
 809         cp_copymode(0);
 810         jmp_eol();
 811         for (i = 0; i < n; i++)
 812                 args[i] = sbuf_buf(sbuf) + idx[i];
 813         return n;
 814 }
 815
 816 /* read arguments for commands .nr that expect a register name */
 817 static int mkargs_reg1(char **args, struct sbuf *sbuf)
 818 {
 819         int n;
 820         int idx0 = sbuf_len(sbuf);
 821         arg_regname(sbuf);
 822         n = mkargs_req(args + 1, sbuf) + 1;
 823         args[0] = sbuf_buf(sbuf) + idx0;
 824         return n;
 825 }
 826
 827 /* do not read arguments; for .if, .ie and .el */
 828 static int mkargs_null(char **args, struct sbuf *sbuf)
 829 {
 830         return 0;
 831 }
 832
 833 /* read the whole line for .tm */
 834 static int mkargs_eol(char **args, struct sbuf *sbuf)
 835 {
 836         int idx0 = sbuf_len(sbuf);
 837         int c;
 838         c = cp_next();
 839         while (c == ' ')
 840                 c = cp_next();
 841         while (c >= 0 && c != '\n') {
 842                 sbuf_add(sbuf, c);
 843                 c = cp_next();
 844         }
 845         args[0] = sbuf_buf(sbuf) + idx0;
 846         return 1;
 847 }
 848
 849 static struct cmd {
 850         char *id;
 851         void (*f)(char **args);
 852         int (*args)(char **args, struct sbuf *sbuf);
 853 } cmds[] = {
 854         {TR_DIVBEG, tr_divbeg},
 855         {TR_DIVEND, tr_divend},
 856         {TR_POPREN, tr_popren},
 857         {"ab", tr_ab, mkargs_eol},
 858         {"ad", tr_ad},
 859         {"af", tr_af},
 860         {"am", tr_de, mkargs_reg1},
 861         {"as", tr_as, mkargs_ds},
 862         {"bd", tr_bd},
 863         {"bp", tr_bp},
 864         {"br", tr_br},
 865         {"c2", tr_c2},
 866         {"cc", tr_cc},
 867         {"ochar", tr_ochar},
 868         {"ce", tr_ce},
 869         {"ch", tr_ch},
 870         {"char", tr_char, mkargs_ds},
 871         {"chop", tr_chop, mkargs_reg1},
 872         {"cl", tr_cl},
 873         {"cp", tr_cp},
 874         {"cs", tr_cs},
 875         {"da", tr_di},
 876         {"de", tr_de, mkargs_reg1},
 877         {"di", tr_di},
 878         {"ds", tr_ds, mkargs_ds},
 879         {"dt", tr_dt},
 880         {"ec", tr_ec},
 881         {"el", tr_el, mkargs_null},
 882         {"em", tr_em},
 883         {"eo", tr_eo},
 884         {"ev", tr_ev},
 885         {"ex", tr_ex},
 886         {"fc", tr_fc},
 887         {"ff", tr_ff},
 888         {"fi", tr_fi},
 889         {"fmap", tr_fmap},
 890         {"fp", tr_fp},
 891         {"fspecial", tr_fspecial},
 892         {"ft", tr_ft},
 893         {"fzoom", tr_fzoom},
 894         {"hc", tr_hc},
 895         {"hcode", tr_hcode},
 896         {"hpf", tr_hpf},
 897         {"hpfa", tr_hpfa},
 898         {"hy", tr_hy},
 899         {"hyp", tr_hyp},
 900         {"hw", tr_hw},
 901         {"ie", tr_if, mkargs_null},
 902         {"if", tr_if, mkargs_null},
 903         {"ig", tr_ig},
 904         {"in", tr_in},
 905         {"it", tr_it},
 906         {"kn", tr_kn},
 907         {"lc", tr_lc},
 908         {"lf", tr_lf},
 909         {"lg", tr_lg},
 910         {"ll", tr_ll},
 911         {"ls", tr_ls},
 912         {"lt", tr_lt},
 913         {"mc", tr_mc},
 914         {"mk", tr_mk},
 915         {"na", tr_na},
 916         {"ne", tr_ne},
 917         {"nf", tr_nf},
 918         {"nh", tr_nh},
 919         {"nm", tr_nm},
 920         {"nn", tr_nn},
 921         {"nr", tr_nr, mkargs_reg1},
 922         {"ns", tr_ns},
 923         {"nx", tr_nx},
 924         {"os", tr_os},
 925         {"pc", tr_pc},
 926         {"pl", tr_pl},
 927         {"pn", tr_pn},
 928         {"po", tr_po},
 929         {"ps", tr_ps},
 930         {"rchar", tr_rchar, mkargs_ds},
 931         {"rm", tr_rm},
 932         {"rn", tr_rn},
 933         {"rr", tr_rr},
 934         {"rs", tr_rs},
 935         {"rt", tr_rt},
 936         {"so", tr_so},
 937         {"sp", tr_sp},
 938         {"ss", tr_ss},
 939         {"ssh", tr_ssh},
 940         {"sv", tr_sv},
 941         {"sy", tr_sy, mkargs_eol},
 942         {"ta", tr_ta},
 943         {"tc", tr_tc},
 944         {"ti", tr_ti},
 945         {"tl", tr_tl, mkargs_null},
 946         {"tm", tr_tm, mkargs_eol},
 947         {"tr", tr_tr, mkargs_eol},
 948         {"vs", tr_vs},
 949         {"wh", tr_wh},
 950 };
 951
 952 /* read the next troff request; return zero if a request was executed. */
 953 int tr_nextreq(void)
 954 {
 955         char *args[NARGS + 3] = {NULL};
 956         char cmd[RNLEN + 1];
 957         struct cmd *req;
 958         struct sbuf sbuf;
 959         int c;
 960         if (!tr_nl)
 961                 return 1;
 962         c = cp_next();
 963         if (c < 0 || (c != c_cc && c != c_c2)) {
 964                 cp_back(c);
 965                 return 1;
 966         }
 967         memset(args, 0, sizeof(args));
 968         args[0] = cmd;
 969         cmd[0] = c;
 970         req = NULL;
 971         cp_reqline();
 972         read_regname(cmd + 1);
 973         sbuf_init(&sbuf);
 974         req = str_dget(map(cmd + 1));
 975         if (req) {
 976                 if (req->args)
 977                         req->args(args + 1, &sbuf);
 978                 else
 979                         mkargs_req(args + 1, &sbuf);
 980                 req->f(args);
 981         } else {
 982                 cp_copymode(1);
 983                 mkargs(args + 1, &sbuf);
 984                 cp_copymode(0);
 985                 if (str_get(map(cmd + 1)))
 986                         in_push(str_get(map(cmd + 1)), args + 1);
 987         }
 988         sbuf_done(&sbuf);
 989         return 0;
 990 }
 991
 992 int tr_next(void)
 993 {
 994         int c;
 995         while (!tr_nextreq())
 996                 ;
 997         c = cp_next();
 998         tr_nl = c == '\n' || c < 0;
 999         return c;
1000 }
1001
1002 void tr_init(void)
1003 {
1004         int i;
1005         for (i = 0; i < LEN(cmds); i++)
1006                 str_dset(map(cmds[i].id), &cmds[i]);
1007         dict_init(&cmap, NCMAPS, -1, 0, 0);
1008 }