🐐 Array functions

centvrion/ast_nodes.py

@@ -1496,6 +1496,32 @@ class BuiltIn(Node):
       case "EVERRE":
         print("\033[2J\033[H", end="", flush=True)
         return vtable, ValNul()
+      case "ADDE":
+        if len(params) != 2:
+          raise CentvrionError("ADDE takes exactly II arguments")
+        if not isinstance(params[0], ValList):
+          raise CentvrionError("ADDE requires an array")
+        return vtable, ValList(list(params[0].value()) + [params[1]])
+      case "TOLLE":
+        if len(params) != 2:
+          raise CentvrionError("TOLLE takes exactly II arguments")
+        if not isinstance(params[0], ValList):
+          raise CentvrionError("TOLLE requires an array")
+        items = list(params[0].value())
+        idx = _to_index_int(params[1])
+        if idx < 1 or idx > len(items):
+          raise CentvrionError(f"Index {idx} out of range for array of length {len(items)}")
+        return vtable, ValList(items[:idx - 1] + items[idx:])
+      case "INSERE":
+        if len(params) != 3:
+          raise CentvrionError("INSERE takes exactly III arguments")
+        if not isinstance(params[0], ValList):
+          raise CentvrionError("INSERE requires an array")
+        items = list(params[0].value())
+        idx = _to_index_int(params[1])
+        if idx < 1 or idx > len(items) + 1:
+          raise CentvrionError(f"Index {idx} out of range for array of length {len(items)}")
+        return vtable, ValList(items[:idx - 1] + [params[2]] + items[idx - 1:])
       case "ORDINA":
         if not isinstance(params[0], ValList):
           raise CentvrionError("ORDINA requires an array")

centvrion/compiler/emit_expr.py

@@ -313,6 +313,15 @@ def _emit_builtin(node, ctx):
     case "ORDINA":
       lines.append(f"CentValue {tmp} = cent_ordina({param_vars[0]});")
 
+    case "ADDE":
+      lines.append(f"CentValue {tmp} = cent_adde({param_vars[0]}, {param_vars[1]});")
+
+    case "TOLLE":
+      lines.append(f"CentValue {tmp} = cent_tolle({param_vars[0]}, {param_vars[1]});")
+
+    case "INSERE":
+      lines.append(f"CentValue {tmp} = cent_insere({param_vars[0]}, {param_vars[1]}, {param_vars[2]});")
+
     case "EVERRE":
       lines.append("cent_everre();")
       lines.append(f"CentValue {tmp} = cent_null();")

centvrion/compiler/runtime/cent_runtime.c

@@ -848,6 +848,56 @@ CentValue cent_ordina(CentValue lst) {
   return result;
 }
 
+static long _index_arg(CentValue idx, const char *name) {
+  if (idx.type == CENT_INT)
+    return idx.ival;
+  if (idx.type == CENT_FRAC && idx.fval.den == 1)
+    return idx.fval.num;
+  cent_type_error(name);
+  return 0;
+}
+
+CentValue cent_adde(CentValue lst, CentValue v) {
+  if (lst.type != CENT_LIST)
+    cent_type_error("'ADDE' requires a list");
+  int len = lst.lval.len;
+  CentValue result = cent_list_new(len + 1);
+  for (int i = 0; i < len; i++)
+    cent_list_push(&result, lst.lval.items[i]);
+  cent_list_push(&result, v);
+  return result;
+}
+
+CentValue cent_tolle(CentValue lst, CentValue idx) {
+  if (lst.type != CENT_LIST)
+    cent_type_error("'TOLLE' requires a list");
+  long i = _index_arg(idx, "'TOLLE' index must be an integer");
+  int len = lst.lval.len;
+  if (i < 1 || i > len)
+    cent_runtime_error("'TOLLE' index out of range");
+  CentValue result = cent_list_new(len - 1);
+  for (int j = 0; j < len; j++)
+    if (j != i - 1)
+      cent_list_push(&result, lst.lval.items[j]);
+  return result;
+}
+
+CentValue cent_insere(CentValue lst, CentValue idx, CentValue v) {
+  if (lst.type != CENT_LIST)
+    cent_type_error("'INSERE' requires a list");
+  long i = _index_arg(idx, "'INSERE' index must be an integer");
+  int len = lst.lval.len;
+  if (i < 1 || i > len + 1)
+    cent_runtime_error("'INSERE' index out of range");
+  CentValue result = cent_list_new(len + 1);
+  for (int j = 0; j < i - 1; j++)
+    cent_list_push(&result, lst.lval.items[j]);
+  cent_list_push(&result, v);
+  for (int j = i - 1; j < len; j++)
+    cent_list_push(&result, lst.lval.items[j]);
+  return result;
+}
+
 /* ------------------------------------------------------------------ */
 /* Array helpers                                                        */
 /* ------------------------------------------------------------------ */

centvrion/compiler/runtime/cent_runtime.h

@@ -239,6 +239,9 @@ CentValue cent_senatus(CentValue *args, int n);                /* SENATVS */
 CentValue cent_typvs(CentValue v);                              /* TYPVS */
 void      cent_dormi(CentValue n);                              /* DORMI */
 CentValue cent_ordina(CentValue lst);                           /* ORDINA */
+CentValue cent_adde(CentValue lst, CentValue v);                /* ADDE */
+CentValue cent_tolle(CentValue lst, CentValue idx);             /* TOLLE */
+CentValue cent_insere(CentValue lst, CentValue idx, CentValue v); /* INSERE */
 CentValue cent_lege(CentValue path);                            /* LEGE */
 void      cent_scribe(CentValue path, CentValue content);       /* SCRIBE */
 void      cent_adivnge(CentValue path, CentValue content);      /* ADIVNGE */

centvrion/lexer.py

@@ -46,6 +46,7 @@ keyword_tokens = [("KEYWORD_"+i, i) for i in [
 ]]
 
 builtin_tokens = [("BUILTIN", i) for i in [
+  "ADDE",
   "AVDI_NVMERVS",
   "AVDI",
   "CLAVES",
@@ -55,6 +56,7 @@ builtin_tokens = [("BUILTIN", i) for i in [
   "EVERRE",
   "FORTVITVS_NVMERVS",
   "FORTVITA_ELECTIO",
+  "INSERE",
   "LITTERA",
   "LONGITVDO",
   "MAIVSCVLA",
@@ -63,6 +65,7 @@ builtin_tokens = [("BUILTIN", i) for i in [
   "ORDINA",
   "SEMEN",
   "SENATVS",
+  "TOLLE",
   "TYPVS",
   "LEGE",
   "SCRIBE",