summaryrefslogtreecommitdiffstats
path: root/lib.c
diff options
context:
space:
mode:
authorKaz Kylheku <kaz@kylheku.com>2021-12-07 06:45:05 -0800
committerKaz Kylheku <kaz@kylheku.com>2021-12-07 06:45:05 -0800
commit078819a0ae62eaa981a271e127ecbcf2ce0a435f (patch)
tree1b87721616ab2868628bed3721e69b8e2df40477 /lib.c
parent4534279ab8e0739c8d1b5eab9bdc00b829724e0f (diff)
downloadtxr-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.c42
1 files changed, 42 insertions, 0 deletions
diff --git a/lib.c b/lib.c
index 5dd21df4..9e20fb20 100644
--- a/lib.c
+++ b/lib.c
@@ -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");