Add arrset

2 files changed, 14 insertions, 0 deletions

diff --git a/functions.cpp b/functions.cpp
index 792e2cc..f4beeb9 100644
--- a/functions.cpp
+++ b/functions.cpp
@@ -421,6 +421,19 @@ unordered_map<string,function<void(vector<Stackitem>&,unordered_map<string,Stack
 		S.push_back(S.back().arrval[idx.intval]);
 	}},
 	//leaves the array on the stack
+	{"arrset",[](vector<Stackitem> &S,unordered_map<string,Stackitem> &variables){
+		BUILTIN_GUARD_STACKSIZE("arrset",3)
+		const Stackitem idx=move(S.back()); S.pop_back();
+		Stackitem val=move(S.back()); S.pop_back();
+		if(idx.type!=SIT_INT)
+			throw string("Second argument to 'arrset' not a number");
+		if(S.back().type!=SIT_ARR)
+			throw string("Top of stack is not an array in builtin 'arrset'");
+		if(idx.intval<0||idx.intval>=S.back().arrval.size())
+			throw string("Index argument to 'arrset' out of range");
+		S.back().arrval[idx.intval]=move(val);
+	}},
+	//leaves the array on the stack
 	{"arrlen",[](vector<Stackitem> &S,unordered_map<string,Stackitem> &variables){
 		BUILTIN_GUARD_STACKSIZE("arrlen",1)
 		if(S.back().type!=SIT_ARR)
diff --git a/reference.md b/reference.md
index 1938d5f..13c7b2f 100644
--- a/reference.md
+++ b/reference.md
@@ -44,6 +44,7 @@ s `ord` | Gets the ascii value of the first character in string `s` (replacing `
 `mkarray` | Creates an empty array on the stack
 [v...] n `mkarrayp` | Takes `n` elements off the stack and creates a new array with these `n` elements
 a i `arridx` | Gets index `i` from array `a`, leaving `a` on the stack
+a v i `arrset` | Sets index `i` in array `a` to value `v`, leaving `a` on the stack
 a `arrlen` | Gets the length of array `a`, leaving `a` on the stack
 a v `arrpush` | Pushes the value `v` on the end of array `a` (leaving `a` on the stack, obviously)
 a v `arrpushf` | Pushes the value `v` at the front of array `a` (leaving `a` on the stack, obviously)