1
0
mirror of https://github.com/janet-lang/janet synced 2024-10-31 22:16:16 +00:00

Merge pull request #894 from pepe/fix-numarray-example

Improve numarray example
This commit is contained in:
Calvin Rose 2021-12-09 14:09:11 -06:00 committed by GitHub
commit e1602618c3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 1 deletions

View File

@ -76,9 +76,16 @@ void num_array_put(void *p, Janet key, Janet value) {
} }
} }
static Janet num_array_length(int32_t argc, Janet *argv) {
janet_fixarity(argc, 1);
num_array *array = (num_array *)janet_getabstract(argv, 0, &num_array_type);
return janet_wrap_number(array->size);
}
static const JanetMethod methods[] = { static const JanetMethod methods[] = {
{"scale", num_array_scale}, {"scale", num_array_scale},
{"sum", num_array_sum}, {"sum", num_array_sum},
{"length", num_array_length},
{NULL, NULL} {NULL, NULL}
}; };
@ -109,6 +116,11 @@ static const JanetReg cfuns[] = {
"(numarray/scale numarray factor)\n\n" "(numarray/scale numarray factor)\n\n"
"scale numarray by factor" "scale numarray by factor"
}, },
{
"sum", num_array_sum,
"(numarray/sum numarray)\n\n"
"sums numarray"
},
{NULL, NULL, NULL} {NULL, NULL, NULL}
}; };

View File

@ -1,4 +1,4 @@
(import build/numarray) (import /build/numarray)
(def a (numarray/new 30)) (def a (numarray/new 30))
(print (get a 20)) (print (get a 20))