From 8bee541c76ebea46ea1c809805270bb69595a56a Mon Sep 17 00:00:00 2001 From: Bruce Hill Date: Fri, 23 Feb 2024 13:47:06 -0500 Subject: Support ++= item --- compile.c | 19 +++++++++++++++---- tests/arrays.nl | 3 +++ 2 files changed, 18 insertions(+), 4 deletions(-) diff --git a/compile.c b/compile.c index 56e8b5a6..19df60f1 100644 --- a/compile.c +++ b/compile.c @@ -336,6 +336,8 @@ CORD compile(env_t *env, ast_t *ast) operand_t = lhs_t; else if (can_promote(lhs_t, rhs_t)) operand_t = rhs_t; + else if (lhs_t->tag == ArrayType && can_promote(rhs_t, Match(lhs_t, ArrayType)->item_type)) + operand_t = lhs_t; else code_err(ast, "I can't do operations between %T and %T", lhs_t, rhs_t); @@ -377,10 +379,19 @@ CORD compile(env_t *env, ast_t *ast) if (operand_t->tag == StringType) { return CORD_asprintf("%r = CORD_cat(%r, %r);", lhs, lhs, rhs); } else if (operand_t->tag == ArrayType) { - if (update->lhs->tag == Var) - return CORD_all("Array__insert_all(&", lhs, ", ", rhs, ", 0, ", compile_type_info(env, operand_t), ")"); - else - return CORD_all(lhs, "Array__concat(", lhs, ", ", rhs, ", ", compile_type_info(env, operand_t), ")"); + if (can_promote(rhs_t, Match(lhs_t, ArrayType)->item_type)) { + // arr ++= item + if (update->lhs->tag == Var) + return CORD_all("Array__insert(&", lhs, ", $stack(", rhs, "), 0, ", compile_type_info(env, operand_t), ")"); + else + return CORD_all(lhs, "Array__concat(", lhs, ", $Array(", rhs, "), ", compile_type_info(env, operand_t), ")"); + } else { + // arr ++= [...] + if (update->lhs->tag == Var) + return CORD_all("Array__insert_all(&", lhs, ", ", rhs, ", 0, ", compile_type_info(env, operand_t), ")"); + else + return CORD_all(lhs, "Array__concat(", lhs, ", ", rhs, ", ", compile_type_info(env, operand_t), ")"); + } } else { code_err(ast, "'++=' is not implemented for %T types", operand_t); } diff --git a/tests/arrays.nl b/tests/arrays.nl index 0c118364..5484d5e2 100644 --- a/tests/arrays.nl +++ b/tests/arrays.nl @@ -28,3 +28,6 @@ for i,x in arr >> arr2 = [10, 20, 30, 40, 50, 60] +>> arr2 ++= 70 +>> arr2 += [10, 20, 30, 40, 50, 60, 70] -- cgit v1.2.3