1 /* registers and environments */
11 #define NENVS 64 /* number of environment registers */
14 int eregs
[NENVS
]; /* environment-specific number registers */
15 int tabs
[NTABS
]; /* tab stops */
16 char tabs_type
[NTABS
]; /* type of tabs: L, C, R */
17 struct fmt
*fmt
; /* per environment line formatting buffer */
18 struct wb wb
; /* per environment partial word */
19 char tc
[GNLEN
]; /* tab character (.tc) */
20 char lc
[GNLEN
]; /* leader character (.lc) */
21 char hc
[GNLEN
]; /* hyphenation character (.hc) */
22 char mc
[GNLEN
]; /* margin character (.mc) */
25 static int nregs
[NREGS
]; /* global number registers */
26 static int nregs_inc
[NREGS
]; /* number register auto-increment size */
27 static int nregs_fmt
[NREGS
]; /* number register format */
28 static char *sregs
[NREGS
]; /* global string registers */
29 static void *sregs_dat
[NREGS
]; /* builtin function data */
30 static struct env
*envs
[NREGS
];/* environments */
31 static struct env
*env
; /* current enviroment */
32 static int env_id
; /* current environment id */
33 static int eregs_idx
[NREGS
]; /* register environment index in eregs[] */
35 static char *eregs
[] = { /* environment-specific number registers */
36 "ln", ".f", ".i", ".j", ".l",
37 ".L", ".nI", ".nm", ".nM", ".nn",
38 ".nS", ".m", ".s", ".u", ".v",
39 ".it", ".itn", ".mc", ".mcn",
40 ".ce", ".f0", ".hy", ".hycost", ".hycost2",".hycost3", ".i0", ".l0",
41 ".L0", ".m0", ".n0", ".s0", ".ss", ".ssh", ".sss", ".pmll", ".pmllcost",
42 ".ti", ".lt", ".lt0", ".v0",
45 /* return the address of a number register */
49 return &env
->eregs
[eregs_idx
[id
]];
53 static char *directory(char *path
)
55 static char dst
[PATHLEN
];
56 char *s
= strrchr(path
, '/');
61 memcpy(dst
, path
, s
- path
);
66 static int num_fmt(char *s
, int n
, int fmt
);
68 /* the contents of a number register (returns a static buffer) */
71 static char numbuf
[128];
72 char *s
= map_name(id
);
76 if (s
[0] == '.' && !s
[2]) {
79 sprintf(numbuf
, "%d", font_getbd(dev_font(n_f
)));
82 sprintf(numbuf
, "%d", in_lnum());
85 sprintf(numbuf
, "%d", f_hpos());
88 sprintf(numbuf
, "#%02x%02x%02x",
89 CLR_R(n_m
), CLR_G(n_m
), CLR_B(n_m
));
92 sprintf(numbuf
, "%d", f_nexttrap());
96 sprintf(numbuf
, "%s", map_name(f_divreg()));
99 sprintf(numbuf
, "%s", in_filename());
102 sprintf(numbuf
, "%s", directory(in_filename()));
105 sprintf(numbuf
, "%d", in_nargs());
109 if (s
[0] == '.' && !strcmp(".neat", s
))
111 if (s
[0] == '.' && s
[1] == 'e' && s
[2] == 'v' && !s
[3])
112 return map_name(env_id
);
113 if (s
[0] == '$' && s
[1] == '$' && !s
[2]) {
114 sprintf(numbuf
, "%d", getpid());
117 if (s
[0] == 'y' && s
[1] == 'r' && !s
[2]) {
118 sprintf(numbuf
, "%02d", *nreg(id
));
121 if (!nregs_fmt
[id
] || num_fmt(numbuf
, *nreg(id
), nregs_fmt
[id
]))
122 sprintf(numbuf
, "%d", *nreg(id
));
126 void num_set(int id
, int val
)
133 void num_setinc(int id
, int val
)
138 void num_inc(int id
, int pos
)
140 *nreg(id
) += pos
> 0 ? nregs_inc
[id
] : -nregs_inc
[id
];
150 void str_set(int id
, char *s
)
152 int len
= strlen(s
) + 1;
155 sregs
[id
] = xmalloc(len
);
156 memcpy(sregs
[id
], s
, len
);
157 sregs_dat
[id
] = NULL
;
160 char *str_get(int id
)
165 void *str_dget(int id
)
167 return sregs_dat
[id
];
170 void str_dset(int id
, void *d
)
180 sregs_dat
[id
] = NULL
;
183 void str_rn(int src
, int dst
)
185 if (!sregs
[src
] && !sregs_dat
[src
])
188 sregs
[dst
] = sregs
[src
];
189 sregs_dat
[dst
] = sregs_dat
[src
];
191 sregs_dat
[src
] = NULL
;
194 static struct env
*env_alloc(void)
196 struct env
*env
= xmalloc(sizeof(*env
));
197 memset(env
, 0, sizeof(*env
));
199 env
->fmt
= fmt_alloc();
203 static void env_free(struct env
*env
)
210 static void env_set(int id
)
216 envs
[id
] = env_alloc();
221 n_l
= SC_IN
* 65 / 10;
229 n_lt
= SC_IN
* 65 / 10;
235 strcpy(env
->hc
, "\\%");
236 strcpy(env
->lc
, ".");
237 for (i
= 0; i
< NTABS
; i
++)
238 env
->tabs
[i
] = i
* SC_IN
/ 2;
242 static void init_time(void)
244 time_t t
= time(NULL
);
245 struct tm
*tm
= localtime(&t
);
246 num_set(map("dw"), tm
->tm_wday
+ 1);
247 num_set(map("dy"), tm
->tm_mday
);
248 num_set(map("mo"), tm
->tm_mon
+ 1);
249 num_set(map("yr"), tm
->tm_year
% 100);
252 static void init_globals(void)
258 num_set(map(".H"), 1);
259 num_set(map(".V"), 1);
267 for (i
= 0; i
< LEN(eregs
); i
++)
268 eregs_idx
[map(eregs
[i
])] = i
+ 1;
275 for (i
= 0; i
< LEN(envs
); i
++)
278 for (i
= 0; i
< LEN(sregs
); i
++)
282 static int oenv
[NPREV
]; /* environment stack */
285 void tr_ev(char **args
)
291 id
= nenv
? oenv
[--nenv
] : -1;
294 if (args
[1] && env
&& nenv
< NPREV
)
295 oenv
[nenv
++] = env_id
;
299 struct fmt
*env_fmt(void)
304 struct wb
*env_wb(void)
329 /* saving and restoring registers around diverted lines */
331 int f
, s
, m
, f0
, s0
, m0
;
334 static struct odiv odivs
[NPREV
]; /* state before diverted text */
337 /* begin outputting diverted line */
340 struct odiv
*o
= &odivs
[nodivs
++];
349 /* end outputting diverted line */
352 struct odiv
*o
= &odivs
[--nodivs
];
361 void tr_ta(char **args
)
365 for (i
= 0; i
< NARGS
&& args
[i
]; i
++) {
366 env
->tabs
[i
] = eval_re(args
[i
], i
> 0 ? env
->tabs
[i
- 1] : 0, 'm');
367 s
= args
[i
][0] ? strchr(args
[i
], '\0') - 1 : "";
368 env
->tabs_type
[i
] = strchr("LRC", *s
) ? *s
: 'L';
372 static int tab_idx(int pos
)
375 for (i
= 0; i
< LEN(env
->tabs
); i
++)
376 if (env
->tabs
[i
] > pos
)
381 int tab_next(int pos
)
383 int i
= tab_idx(pos
);
384 return i
>= 0 ? env
->tabs
[i
] : pos
;
387 int tab_type(int pos
)
389 int i
= tab_idx(pos
);
390 return i
>= 0 && env
->tabs_type
[i
] ? env
->tabs_type
[i
] : 'L';
393 /* number register format (.af) */
394 #define NF_LSH 8 /* number format length shifts */
395 #define NF_FMT 0x00ff /* number format mask */
397 /* the format of a number register (returns a static buffer) */
398 char *num_getfmt(int id
)
400 static char fmtbuf
[128];
403 if ((nregs_fmt
[id
] & NF_FMT
) == '0') {
405 i
= nregs_fmt
[id
] >> NF_LSH
;
408 } else if (nregs_fmt
[id
]) {
409 *s
++ = nregs_fmt
[id
] & NF_FMT
;
415 void num_setfmt(int id
, char *s
)
418 if (strchr("iIaA", s
[0])) {
419 nregs_fmt
[id
] = s
[0];
421 while (isdigit(s
[i
]))
423 nregs_fmt
[id
] = '0' | (i
<< NF_LSH
);
427 static void nf_reverse(char *s
)
433 for (i
= 0; i
< l
; i
++)
437 static void nf_roman(char *s
, int n
, char *I
, char *V
)
443 *s
++ = n
% 10 == 9 ? I
[1] : V
[0];
446 for (i
= 0; i
< n
% 5; i
++)
452 nf_roman(s
, n
/ 10, I
+ 1, V
+ 1);
455 static void nf_alpha(char *s
, int n
, int a
)
458 *s
++ = a
+ ((n
- 1) % 26);
464 /* returns nonzero on failure */
465 static int num_fmt(char *s
, int n
, int fmt
)
467 int type
= fmt
& NF_FMT
;
473 if ((type
== 'i' || type
== 'I') && n
> 0 && n
< 40000) {
475 nf_roman(s
, n
, "ixcmz", "vldw");
477 nf_roman(s
, n
, "IXCMZ", "VLDW");
481 if ((type
== 'a' || type
== 'A') && n
> 0) {
482 nf_alpha(s
, n
, type
);
489 while (len
++ < fmt
>> NF_LSH
)