summaryrefslogtreecommitdiff
path: root/operators.c
blob: ba999bdd5743bcc68cce86af72103a6e05298a7b (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
#include <u.h>
#include <libc.h>
#include <bio.h>

#include "apl9.h"

Rune primmonopnames[] = L"¨⍨⌸⌶&";
Rune primdyadopnames[] = L"⍣.∘⍤⍥@⍠⌺";

opmonad monadoperatordefs[] = {
	0, /* ¨ */
	opSwitch, /* ⍨ */
	0, /* ⌸ */
	0, /* ⌶ */
	0, /* & */
};

opdyad dyadoperatordefs[] = {
	0, /* ⍣ */
	0, /* . */
	0, /* ∘ */
	0, /* ⍤ */
	opOver, /* ⍥ */
	0, /* @ */
	0, /* ⍠ */
	0, /* ⌺ */
};

/* Monadic operators */
Array *
opSwitch(Datum *lefto, Array *left, Array *right)
{
	if(lefto->tag == ArrayTag){
		incref(lefto->array);
		return lefto->array;
	}else if(lefto->tag == FunctionTag){
		if(left)
			return runfunc(lefto->func, right, left);
		else
			return runfunc(lefto->func, right, right);
	}else
		return nil;
}

/* Dyadic operators */
Array *
opOver(Datum *lefto, Datum *righto, Array *left, Array *right)
{
	if(lefto->tag != FunctionTag || righto->tag != FunctionTag)
		return nil;
	if(left){
		Array *r = runfunc(righto->func, nil, right);
		Array *l = runfunc(righto->func, nil, left);
		Array *res = runfunc(lefto->func, l, r);
		freearray(r);
		freearray(l);
		return res;
	}else{
		Array *tmp = runfunc(righto->func, nil, right);
		Array *res = runfunc(lefto->func, nil, tmp);
		freearray(tmp);
		return res;
	}
}