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 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
++)
280 static int oenv
[NPREV
]; /* environment stack */
283 void tr_ev(char **args
)
289 id
= nenv
? oenv
[--nenv
] : -1;
292 if (args
[1] && env
&& nenv
< NPREV
)
293 oenv
[nenv
++] = env_id
;
297 struct fmt
*env_fmt(void)
302 struct wb
*env_wb(void)
327 /* saving and restoring registers around diverted lines */
329 int f
, s
, m
, f0
, s0
, m0
;
332 static struct odiv odivs
[NPREV
]; /* state before diverted text */
335 /* begin outputting diverted line */
338 struct odiv
*o
= &odivs
[nodivs
++];
347 /* end outputting diverted line */
350 struct odiv
*o
= &odivs
[--nodivs
];
359 void tr_ta(char **args
)
363 for (i
= 0; i
< NARGS
&& args
[i
]; i
++) {
364 env
->tabs
[i
] = eval_re(args
[i
], i
> 0 ? env
->tabs
[i
- 1] : 0, 'm');
365 s
= args
[i
][0] ? strchr(args
[i
], '\0') - 1 : "";
366 env
->tabs_type
[i
] = strchr("LRC", *s
) ? *s
: 'L';
370 static int tab_idx(int pos
)
373 for (i
= 0; i
< LEN(env
->tabs
); i
++)
374 if (env
->tabs
[i
] > pos
)
379 int tab_next(int pos
)
381 int i
= tab_idx(pos
);
382 return i
>= 0 ? env
->tabs
[i
] : pos
;
385 int tab_type(int pos
)
387 int i
= tab_idx(pos
);
388 return i
>= 0 && env
->tabs_type
[i
] ? env
->tabs_type
[i
] : 'L';
391 /* number register format (.af) */
392 #define NF_LSH 8 /* number format length shifts */
393 #define NF_FMT 0x00ff /* number format mask */
395 /* the format of a number register (returns a static buffer) */
396 char *num_getfmt(int id
)
398 static char fmtbuf
[128];
401 if ((nregs_fmt
[id
] & NF_FMT
) == '0') {
403 i
= nregs_fmt
[id
] >> NF_LSH
;
406 } else if (nregs_fmt
[id
]) {
407 *s
++ = nregs_fmt
[id
] & NF_FMT
;
413 void num_setfmt(int id
, char *s
)
416 if (strchr("iIaA", s
[0])) {
417 nregs_fmt
[id
] = s
[0];
419 while (isdigit(s
[i
]))
421 nregs_fmt
[id
] = '0' | (i
<< NF_LSH
);
425 static void nf_reverse(char *s
)
431 for (i
= 0; i
< l
; i
++)
435 static void nf_roman(char *s
, int n
, char *I
, char *V
)
441 *s
++ = n
% 10 == 9 ? I
[1] : V
[0];
444 for (i
= 0; i
< n
% 5; i
++)
450 nf_roman(s
, n
/ 10, I
+ 1, V
+ 1);
453 static void nf_alpha(char *s
, int n
, int a
)
456 *s
++ = a
+ ((n
- 1) % 26);
462 /* returns nonzero on failure */
463 static int num_fmt(char *s
, int n
, int fmt
)
465 int type
= fmt
& NF_FMT
;
471 if ((type
== 'i' || type
== 'I') && n
> 0 && n
< 40000) {
473 nf_roman(s
, n
, "ixcmz", "vldw");
475 nf_roman(s
, n
, "IXCMZ", "VLDW");
479 if ((type
== 'a' || type
== 'A') && n
> 0) {
480 nf_alpha(s
, n
, type
);
487 while (len
++ < fmt
>> NF_LSH
)