add curry
This commit is contained in:
parent
cfc45d763a
commit
58c1f24161
4 changed files with 88 additions and 44 deletions
|
|
@ -1 +1,5 @@
|
|||
5 [1 +] [2 *] compose call .
|
||||
5 [1 +] [2 *] compose
|
||||
call .
|
||||
|
||||
10 5 [+] curry
|
||||
call .
|
||||
|
|
|
|||
|
|
@ -34,6 +34,7 @@ struct {
|
|||
{"if", {OP_CHOOSE, OP_CALL, 0}},
|
||||
{"call", {OP_CALL, 0}},
|
||||
{"compose", {OP_COMPOSE, 0}},
|
||||
{"curry", {OP_CURRY, 0}},
|
||||
{"?", {OP_CHOOSE, 0}},
|
||||
{"+", {OP_ADD, 0}},
|
||||
{"-", {OP_SUB, 0}},
|
||||
|
|
|
|||
13
src/gc.c
13
src/gc.c
|
|
@ -101,7 +101,6 @@ V gc_collect(Vm *vm) {
|
|||
}
|
||||
Hd *hdr = (Hd *)scan;
|
||||
switch (hdr->type) {
|
||||
// TODO: the rest of the owl
|
||||
case OBJ_STR:
|
||||
break;
|
||||
case OBJ_QUOT: {
|
||||
|
|
@ -111,6 +110,18 @@ V gc_collect(Vm *vm) {
|
|||
chunk->constants.items[i] = forward(gc, chunk->constants.items[i]);
|
||||
break;
|
||||
}
|
||||
case OBJ_COMPOSE: {
|
||||
Qo *comp = (Qo *)(hdr + 1);
|
||||
comp->first = forward(gc, comp->first);
|
||||
comp->second = forward(gc, comp->second);
|
||||
break;
|
||||
};
|
||||
case OBJ_CURRY: {
|
||||
Qc *curry = (Qc *)(hdr + 1);
|
||||
curry->value = forward(gc, curry->value);
|
||||
curry->callable = forward(gc, curry->callable);
|
||||
break;
|
||||
};
|
||||
case OBJ_USERDATA:
|
||||
break;
|
||||
case OBJ_FWD:
|
||||
|
|
|
|||
42
src/vm.c
42
src/vm.c
|
|
@ -272,6 +272,13 @@ I vm_run(Vm *vm, Bc *chunk, I offset) {
|
|||
quot = comp->first;
|
||||
goto do_call;
|
||||
}
|
||||
case TYPE_CURRY: {
|
||||
Qc *curry = (Qc *)(UNBOX(quot) + 1);
|
||||
vm_push(vm, curry->value);
|
||||
quot = curry->callable;
|
||||
goto do_call;
|
||||
break;
|
||||
}
|
||||
default:
|
||||
vm_error(vm, VM_ERR_TYPE, "attempt to call non-quotation object");
|
||||
}
|
||||
|
|
@ -304,6 +311,13 @@ I vm_run(Vm *vm, Bc *chunk, I offset) {
|
|||
quot = comp->first;
|
||||
goto do_tail_call;
|
||||
}
|
||||
case TYPE_CURRY: {
|
||||
Qc *curry = (Qc *)(UNBOX(quot) + 1);
|
||||
vm_push(vm, curry->value);
|
||||
quot = curry->callable;
|
||||
goto do_tail_call;
|
||||
break;
|
||||
}
|
||||
default:
|
||||
vm_error(vm, VM_ERR_TYPE, "attempt to call non-quotation object");
|
||||
}
|
||||
|
|
@ -319,22 +333,36 @@ I vm_run(Vm *vm, Bc *chunk, I offset) {
|
|||
}
|
||||
case OP_COMPOSE: {
|
||||
I mark = gc_mark(&vm->gc);
|
||||
O q2 = vm_pop(vm);
|
||||
O q1 = vm_pop(vm);
|
||||
gc_addroot(&vm->gc, &q1);
|
||||
gc_addroot(&vm->gc, &q2);
|
||||
if (!callable(q1) || !callable(q2))
|
||||
O c1 = vm_pop(vm);
|
||||
O c2 = vm_pop(vm);
|
||||
gc_addroot(&vm->gc, &c2);
|
||||
gc_addroot(&vm->gc, &c1);
|
||||
if (!callable(c2) || !callable(c1))
|
||||
vm_error(vm, VM_ERR_TYPE, "non-callable arguments to compose");
|
||||
Hd *hd = gc_alloc(vm, sizeof(Hd) + sizeof(Qo));
|
||||
hd->type = OBJ_COMPOSE;
|
||||
Qo *comp = (Qo *)(hd + 1);
|
||||
comp->first = q1;
|
||||
comp->second = q2;
|
||||
comp->first = c2;
|
||||
comp->second = c1;
|
||||
vm_push(vm, BOX(hd));
|
||||
gc_reset(&vm->gc, mark);
|
||||
break;
|
||||
}
|
||||
case OP_CURRY: {
|
||||
I mark = gc_mark(&vm->gc);
|
||||
O cble = vm_pop(vm);
|
||||
O value = vm_pop(vm);
|
||||
gc_addroot(&vm->gc, &cble);
|
||||
gc_addroot(&vm->gc, &value);
|
||||
if (!callable(cble))
|
||||
vm_error(vm, VM_ERR_TYPE, "non-callable argument to curry");
|
||||
Hd *hd = gc_alloc(vm, sizeof(Hd) + sizeof(Qc));
|
||||
hd->type = OBJ_CURRY;
|
||||
Qc *curry = (Qc *)(hd + 1);
|
||||
curry->value = value;
|
||||
curry->callable = cble;
|
||||
vm_push(vm, BOX(hd));
|
||||
gc_reset(&vm->gc, mark);
|
||||
break;
|
||||
}
|
||||
case OP_RETURN:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue