diff options
author | Kaz Kylheku <kaz@kylheku.com> | 2021-12-07 06:45:05 -0800 |
---|---|---|
committer | Kaz Kylheku <kaz@kylheku.com> | 2021-12-07 06:45:05 -0800 |
commit | 078819a0ae62eaa981a271e127ecbcf2ce0a435f (patch) | |
tree | 1b87721616ab2868628bed3721e69b8e2df40477 /lib.c | |
parent | 4534279ab8e0739c8d1b5eab9bdc00b829724e0f (diff) | |
download | txr-078819a0ae62eaa981a271e127ecbcf2ce0a435f.tar.gz txr-078819a0ae62eaa981a271e127ecbcf2ce0a435f.tar.bz2 txr-078819a0ae62eaa981a271e127ecbcf2ce0a435f.zip |
rot, nrot: new functions.
* eval.c (eval_init): nrot, rot intrinsics registered.
* lib.c (nrot, rot): New functions.
* lib.h (nrot, rot): Declared.
* tests/012/seq.tl: New test cases.
* txr.1: Documented.
* stdlib/doc-syms.tl: Updated.
Diffstat (limited to 'lib.c')
-rw-r--r-- | lib.c | 42 |
1 files changed, 42 insertions, 0 deletions
@@ -10750,6 +10750,48 @@ val grade(val seq, val lessfun, val keyfun_in) return nil; } +val nrot(val seq, val n_in) +{ + val len = length(seq); + + if (len != zero && len != one) { + val n = mod(default_arg(n_in, one), len); + + if (n == one) { + val head = ref(seq, zero); + seq = replace(seq, cons(head, nil), len, len); + seq = replace(seq, nil, zero, one); + } else if (n != zero) { + val head = sub(seq, zero, n); + seq = replace(seq, head, len, len); + seq = replace(seq, nil, zero, n); + } + } + + return seq; +} + +val rot(val seq, val n_in) +{ + val seq_orig = seq; + val len = length(seq); + + if (len != zero && len != one) { + val n = mod(default_arg(n_in, one), len); + + if (n != zero) { + val head = sub(seq, zero, n); + seq = sub(seq, n, t); + seq = replace(seq, head, t, t); + } + } + + if (seq == seq_orig) + seq = copy(seq); + + return seq; +} + val find(val item, val seq, val testfun, val keyfun) { val self = lit("find"); |