From 8580001ef44f7a8da3b97c439473cb204a0e34f5 Mon Sep 17 00:00:00 2001 From: Leah Neukirchen Date: Fri, 16 Dec 2022 00:21:29 +0100 Subject: add for/into --- mew.el | 2 ++ mew.scm | 14 +++++++++++++- mew.svnwiki | 4 ++++ tests/test.mew | 6 ++++++ 4 files changed, 25 insertions(+), 1 deletion(-) diff --git a/mew.el b/mew.el index 888032d..a635f2c 100644 --- a/mew.el +++ b/mew.el @@ -12,6 +12,8 @@ (put 'until 'scheme-indent-function 1) (put 'if 'scheme-indent-function 1) +(put 'for 'scheme-indent-function 1) +(put 'for/into 'scheme-indent-function 2) (put 'match 'scheme-indent-function 1) (put 'accumulate 'scheme-indent-function 1) diff --git a/mew.scm b/mew.scm index 17be96d..7707851 100644 --- a/mew.scm +++ b/mew.scm @@ -5,7 +5,7 @@ comp cross-product dec def del-at div dup empty? eof esc - fail fin final for fun* + fail fin final for for/into fun* gconcatenate gen generator-xfold generic-for-each genumerate get gfix giterate gmatch gpick group-by-accumulator gslice-when gsplit gsplit-on gwindow @@ -435,6 +435,18 @@ (let ((o obj)) ((generic-for-each o) (match-lambda (i body ...)) o))))) + (define-syntax for-into-regroup + (syntax-rules () + ((_ acc () ((i obj) ...) body ...) + (into acc (gmap (match-lambda* ((i ...) body ...)) (gen obj) ...))) + ((_ acc (i obj rest ...) (j ...) body ...) + (for-into-regroup acc (rest ...) (j ... (i obj)) body ...)))) + + (define-syntax for/into + (syntax-rules () + ((_ acc (rest ...) body ...) + (for-into-regroup acc (rest ...) () body ...)))) + (define (eof) #!eof) (define (void? x) diff --git a/mew.svnwiki b/mew.svnwiki index 958da33..0099046 100644 --- a/mew.svnwiki +++ b/mew.svnwiki @@ -388,6 +388,7 @@ up to a level of {{}}, or infinitely by default. (for ( ...) ...) (for (( . ) ...) ...) +(for/into ( ...) ...) If {{}} is a list or a vector, iterate over its elements. If {{}} is a procedure, consider it a SRFI-158 generator @@ -397,6 +398,9 @@ If {{}} is a hash-table, iterate over its keys and values. Multiple {{}}-pairs may be specified, then {{for}} iterates over these in *parallel*, stopping after the shortest ends. +The variant {{for/into}} accumulates the values of the {{}} +and returns the result of the accumulation. + (search ?) Returns the offset of the sequence (string/list/vector) {{}} diff --git a/tests/test.mew b/tests/test.mew index 3aa70bb..b7e5da7 100644 --- a/tests/test.mew +++ b/tests/test.mew @@ -395,6 +395,12 @@ y (vector 6 7)) (c x))))) +(test-group "for/into" + (test '(4 5 6) (for/into '() (x (vector 3 4 5)) + (inc x))) + (test '#(#(3 6) #(4 7)) (for/into #() (x (vector 3 4 5) + y (vector 6 7)) + (vector x y)))) (test-group "eof" (test-assert (eof-object? (eof)))) -- cgit 1.4.1