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", ".hyp", ".i0", ".l0",
41 ".L0", ".m0", ".n0", ".s0", ".ss", ".ssh", ".sss",
42 ".ti", ".lt", ".lt0", ".v0",
45 /* return the address of a number register */
49 return &env
->eregs
[eregs_idx
[id
]];
53 static int num_fmt(char *s
, int n
, int fmt
);
55 /* the contents of a number register (returns a static buffer) */
58 static char numbuf
[128];
59 char *s
= map_name(id
);
63 if (s
[0] == '.' && !s
[2]) {
66 sprintf(numbuf
, "%d", font_getbd(dev_font(n_f
)));
69 sprintf(numbuf
, "%d", in_lnum());
72 sprintf(numbuf
, "%d", f_hpos());
75 sprintf(numbuf
, "#%02x%02x%02x",
76 CLR_R(n_m
), CLR_G(n_m
), CLR_B(n_m
));
79 sprintf(numbuf
, "%d", f_nexttrap());
83 sprintf(numbuf
, "%s", map_name(f_divreg()));
86 sprintf(numbuf
, "%s", in_filename());
89 sprintf(numbuf
, "%d", in_nargs());
93 if (s
[0] == '.' && !strcmp(".neat", s
))
95 if (s
[0] == '.' && s
[1] == 'e' && s
[2] == 'v' && !s
[3])
96 return map_name(env_id
);
97 if (s
[0] == '$' && s
[1] == '$' && !s
[2]) {
98 sprintf(numbuf
, "%d", getpid());
101 if (s
[0] == 'y' && s
[1] == 'r' && !s
[2]) {
102 sprintf(numbuf
, "%02d", *nreg(id
));
105 if (!nregs_fmt
[id
] || num_fmt(numbuf
, *nreg(id
), nregs_fmt
[id
]))
106 sprintf(numbuf
, "%d", *nreg(id
));
110 void num_set(int id
, int val
)
117 void num_setinc(int id
, int val
)
122 void num_inc(int id
, int pos
)
124 *nreg(id
) += pos
> 0 ? nregs_inc
[id
] : -nregs_inc
[id
];
134 void str_set(int id
, char *s
)
136 int len
= strlen(s
) + 1;
139 sregs
[id
] = xmalloc(len
);
140 memcpy(sregs
[id
], s
, len
);
141 sregs_dat
[id
] = NULL
;
144 char *str_get(int id
)
149 void *str_dget(int id
)
151 return sregs_dat
[id
];
154 void str_dset(int id
, void *d
)
164 sregs_dat
[id
] = NULL
;
167 void str_rn(int src
, int dst
)
170 sregs
[dst
] = sregs
[src
];
171 sregs_dat
[dst
] = sregs_dat
[src
];
173 sregs_dat
[src
] = NULL
;
176 static struct env
*env_alloc(void)
178 struct env
*env
= xmalloc(sizeof(*env
));
179 memset(env
, 0, sizeof(*env
));
181 env
->fmt
= fmt_alloc();
185 static void env_free(struct env
*env
)
192 static void env_set(int id
)
198 envs
[id
] = env_alloc();
203 n_l
= SC_IN
* 65 / 10;
211 n_lt
= SC_IN
* 65 / 10;
217 strcpy(env
->hc
, "\\%");
218 strcpy(env
->lc
, ".");
219 for (i
= 0; i
< NTABS
; i
++)
220 env
->tabs
[i
] = i
* SC_IN
/ 2;
224 static void init_time(void)
226 time_t t
= time(NULL
);
227 struct tm
*tm
= localtime(&t
);
228 nregs
[map("dw")] = tm
->tm_wday
+ 1;
229 nregs
[map("dy")] = tm
->tm_mday
;
230 nregs
[map("mo")] = tm
->tm_mon
+ 1;
231 nregs
[map("yr")] = tm
->tm_year
% 100;
234 static void init_globals(void)
247 for (i
= 0; i
< LEN(eregs
); i
++)
248 eregs_idx
[map(eregs
[i
])] = i
+ 1;
255 for (i
= 0; i
< LEN(envs
); i
++)
260 static int oenv
[NPREV
]; /* environment stack */
263 void tr_ev(char **args
)
269 id
= nenv
? oenv
[--nenv
] : -1;
272 if (args
[1] && env
&& nenv
< NPREV
)
273 oenv
[nenv
++] = env_id
;
277 struct fmt
*env_fmt(void)
282 struct wb
*env_wb(void)
307 /* saving and restoring registers around diverted lines */
309 int f
, s
, m
, f0
, s0
, m0
;
312 static struct odiv odivs
[NPREV
]; /* state before diverted text */
315 /* begin outputting diverted line */
318 struct odiv
*o
= &odivs
[nodivs
++];
327 /* end outputting diverted line */
330 struct odiv
*o
= &odivs
[--nodivs
];
339 void tr_ta(char **args
)
343 for (i
= 0; i
< NARGS
&& args
[i
]; i
++) {
344 env
->tabs
[i
] = eval_re(args
[i
], i
> 0 ? env
->tabs
[i
- 1] : 0, 'm');
345 s
= args
[i
][0] ? strchr(args
[i
], '\0') - 1 : "";
346 env
->tabs_type
[i
] = strchr("LRC", *s
) ? *s
: 'L';
350 static int tab_idx(int pos
)
353 for (i
= 0; i
< LEN(env
->tabs
); i
++)
354 if (env
->tabs
[i
] > pos
)
359 int tab_next(int pos
)
361 int i
= tab_idx(pos
);
362 return i
>= 0 ? env
->tabs
[i
] : pos
;
365 int tab_type(int pos
)
367 int i
= tab_idx(pos
);
368 return i
>= 0 && env
->tabs_type
[i
] ? env
->tabs_type
[i
] : 'L';
371 /* number register format (.af) */
372 #define NF_LSH 8 /* number format length shifts */
373 #define NF_FMT 0x00ff /* number format mask */
375 /* the format of a number register (returns a static buffer) */
376 char *num_getfmt(int id
)
378 static char fmtbuf
[128];
381 if ((nregs_fmt
[id
] & NF_FMT
) == '0') {
383 i
= nregs_fmt
[id
] >> NF_LSH
;
386 } else if (nregs_fmt
[id
]) {
387 *s
++ = nregs_fmt
[id
] & NF_FMT
;
393 void num_setfmt(int id
, char *s
)
396 if (strchr("iIaA", s
[0])) {
397 nregs_fmt
[id
] = s
[0];
399 while (isdigit(s
[i
]))
401 nregs_fmt
[id
] = '0' | (i
<< NF_LSH
);
405 static void nf_reverse(char *s
)
411 for (i
= 0; i
< l
; i
++)
415 static void nf_roman(char *s
, int n
, char *I
, char *V
)
421 *s
++ = n
% 10 == 9 ? I
[1] : V
[0];
424 for (i
= 0; i
< n
% 5; i
++)
430 nf_roman(s
, n
/ 10, I
+ 1, V
+ 1);
433 static void nf_alpha(char *s
, int n
, int a
)
436 *s
++ = a
+ ((n
- 1) % 26);
442 /* returns nonzero on failure */
443 static int num_fmt(char *s
, int n
, int fmt
)
445 int type
= fmt
& NF_FMT
;
451 if ((type
== 'i' || type
== 'I') && n
> 0 && n
< 40000) {
453 nf_roman(s
, n
, "ixcmz", "vldw");
455 nf_roman(s
, n
, "IXCMZ", "VLDW");
459 if ((type
== 'a' || type
== 'A') && n
> 0) {
460 nf_alpha(s
, n
, type
);
467 while (len
++ < fmt
>> NF_LSH
)