Ensure new procedures are compiled only once

1 files changed, 13 insertions, 1 deletions

diff --git a/src/ir.h b/src/ir.h
index d92f912..cd2100c 100644
--- a/src/ir.h
+++ b/src/ir.h
@@ -60,7 +60,7 @@ typedef enum Op {
     // - Prints the last object in the stack.
     OP_PRINT,
     // Procedures.
-    // - Requires a function name as parameter.
+    // - Consumes the last value in the stack, which must be a lambda.
     OP_CALL,
     // - Return position is on a know location of the stack based on the offset
     // of locals, parameters, etc.
@@ -135,6 +135,7 @@ typedef struct Procedure {
 
 typedef struct ProgramIr {
     Procedure **procedures;
+    Object **lambdas;
     size_t labels;
 } ProgramIr;
 
@@ -360,6 +361,9 @@ void
 compile_proc_call(ProgramIr *program, Procedure *proc, Object *obj) {
     if (IS_BUILTIN(obj->head)) {
         compile_builtin(program, proc, obj);
+    } else if (IS_LAMBDA(obj->head)) {
+        compile_object(program, proc, obj->head);
+        INST_SIMPLE(proc, OP_CALL, obj->line, obj->col);
     } else {
         assert(false && "compile_proc_call: not implemented");
     }
@@ -395,6 +399,13 @@ compile_def(ProgramIr *program, Procedure *proc, Object *obj) {
 
 void
 compile_lambda(ProgramIr *program, Procedure *proc, Object *obj) {
+    for (size_t i = 0; i < array_size(program->lambdas); ++i) {
+        if (object_equal(program->lambdas[i], obj)) {
+            INST_ARG(proc, OP_PUSH, obj, obj->line, obj->col);
+            return;
+        }
+    }
+    array_push(program->lambdas, obj);
     Procedure *lambda = proc_alloc(program, STRING("lambda"), proc);
     for (size_t i = 0; i < array_size(obj->body) - 1; i++) {
         compile_object(program, lambda, obj->body[i]);
@@ -474,6 +485,7 @@ ProgramIr
 compile(Program program) {
     ProgramIr program_ir = {0};
     array_init(program_ir.procedures, 0);
+    array_init(program_ir.lambdas, 0);
     Procedure *main = proc_alloc(&program_ir, STRING("main"), NULL);
 
     for (size_t i = 0; i < array_size(program.roots); i++) {