Make setNth n-dimensional

2022-07-09 14:46:09 +00:00
parent 367479693e
commit b60e60a8ed
1 changed files with 4 additions and 3 deletions
--- a/src/kiss/Macros.hx
+++ b/src/kiss/Macros.hx
@@ -315,13 +315,14 @@ class Macros {

        function arraySet(wholeExp:ReaderExp, exps:Array<ReaderExp>, k:KissState) {
            var b = wholeExp.expBuilder();
+            var value = exps.pop();
            return b.call(
                b.symbol("set"), [
-                    b.call(b.symbol("nth"), [exps[0], exps[1]]),
-                    exps[2]
+                    b.call(b.symbol("nth"), exps),
+                    value
                ]);
        }
-        k.doc("setNth", 3, 3, "(setNth <list> <index> <value>)");
+        k.doc("setNth", 3, null, "(setNth <list> <index> <?n-dimensional indices...> <value>)");
        macros["setNth"] = (wholeExp:ReaderExp, exps:Array<ReaderExp>, k:KissState) -> {
            arraySet(wholeExp, exps, k);
        };