summaryrefslogtreecommitdiff
path: root/print.c
blob: 588ce0a5e3cf3f26379e6f268d32913d88daa134 (plain) (blame)
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
82
83
84
85
#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.func.code]); break;
	case HybridTag: result = runesmprint("%C", primhybridnames[d.func.code]); break;
	case MonadicOpTag: result = runesmprint("%C", primmonopnames[d.func.code]); break;
	case DyadicOpTag: result = runesmprint("%C", primdyadopnames[d.func.code]); break;
	case BoundFunctionTag: result = runesmprint("%S∘%C", pparray(d.func.left), primfuncnames[d.func.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;
	case ArrowTag: result = runestrdup(L"←"); break;
	case AssignmentTag: result = runesmprint("%S←", d.symbol->name); break;
	case NameTag: result = runestrdup(d.symbol->name); 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;
}