1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
#include <u.h>
#include <libc.h>
#include <bio.h>
#include "apl9.h"
Rune *
ppdatum(Datum d)
{
Rune *result;
switch(d.tag){
case ArrayTag: result = pparray(d.array); break;
case FunctionTag: result = runesmprint("%C", primfuncnames[d.code]); break;
case HybridTag: result = runesmprint("%C", primhybridnames[d.code]); break;
case MonadicOpTag: result = runesmprint("%C", primmonopnames[d.code]); break;
case DyadicOpTag: result = runesmprint("%C", primdyadopnames[d.code]); break;
case LParTag: result = runestrdup(L"("); break;
case RParTag: result = runestrdup(L")"); break;
case LCurlTag: result = runestrdup(L"{"); break;
case RCurlTag: result = runestrdup(L"}"); break;
case LBracketTag: result = runestrdup(L"["); break;
case RBracketTag: result = runestrdup(L"]"); break;
default: result = runesmprint("<not printable %d>", d.tag);
}
return result;
}
Rune *
ppdatums(Datum *ds, int n)
{
int i;
Rune *res = runesmprint("");
Rune *tmp;
for(i = 0; i < n; i++){
tmp = res;
res = runesmprint("%S %S", res, ppdatum(ds[i]));
free(tmp);
}
return res;
}
Rune *
pparray(Array *a)
{
Rune **elemstrs = malloc(sizeof(Rune *) * a->size);
for(int i = 0; i < a->size; i++){
if(a->type == AtypeArray){
Rune *arrstr = pparray(a->arraydata[i]);
elemstrs[i] = runesmprint("[%S]", arrstr);
free(arrstr);
}else if(a->type == AtypeInt)
elemstrs[i] = runesmprint("%lld", a->intdata[i]);
}
/* Should do some width and height padding here */
Rune *res = runesmprint("");
Rune *tmp;
for(int i = 0; i < a->size; i++){
tmp = res;
res = runesmprint("%S%S", res, elemstrs[i]);
free(tmp);
free(elemstrs[i]);
int j = 1;
int spaceprinted = 0;
for(int dim = 0; dim < a->rank && i+1 != a->size; dim++){
j *= a->shape[dim];
tmp = res;
if((i+1)%j == 0){
spaceprinted = 1;
res = runesmprint("%S\n", res);
free(tmp);
}else if(!spaceprinted){
spaceprinted = 1;
res = runesmprint("%S ", res);
free(tmp);
}
}
}
return res;
}
|