From fae2b2caa0ec311821a7a7ef1f98500cfd25bb9e Mon Sep 17 00:00:00 2001 From: Bruce Hill Date: Tue, 2 Apr 2024 23:28:59 -0400 Subject: Add array:sample() --- compile.c | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'compile.c') diff --git a/compile.c b/compile.c index ff6a868d..08e2fe48 100644 --- a/compile.c +++ b/compile.c @@ -1481,6 +1481,12 @@ CORD compile(env_t *env, ast_t *ast) CORD self = compile_to_pointer_depth(env, call->self, 0, false); (void)compile_arguments(env, ast, NULL, call->args); return CORD_all("Array$random(", self, ")"); + } else if (streq(call->name, "sample")) { + CORD self = compile_to_pointer_depth(env, call->self, 0, false); + arg_t *arg_spec = new(arg_t, .name="count", .type=Type(IntType, .bits=64), + .next=new(arg_t, .name="weights", .type=self_value_t, .default_val=FakeAST(Array))); + return CORD_all("Array$sample(", self, ", ", compile_arguments(env, ast, arg_spec, call->args), ", ", + compile_type_info(env, self_value_t), ")"); } else if (streq(call->name, "shuffle")) { CORD self = compile_to_pointer_depth(env, call->self, 1, false); (void)compile_arguments(env, ast, NULL, call->args); -- cgit v1.2.3