source: trunk/j/src/org/armedbear/lisp/reduce.lisp @ 2285

Last change on this file since 2285 was 2285, checked in by piso, 19 years ago

Minor cleanup.

File size: 3.2 KB
Line 
1;;; reduce.lisp
2;;;
3;;; Copyright (C) 2003 Peter Graves
4;;; $Id: reduce.lisp,v 1.2 2003-06-11 01:02:28 piso Exp $
5;;;
6;;; This program is free software; you can redistribute it and/or
7;;; modify it under the terms of the GNU General Public License
8;;; as published by the Free Software Foundation; either version 2
9;;; of the License, or (at your option) any later version.
10;;;
11;;; This program is distributed in the hope that it will be useful,
12;;; but WITHOUT ANY WARRANTY; without even the implied warranty of
13;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14;;; GNU General Public License for more details.
15;;;
16;;; You should have received a copy of the GNU General Public License
17;;; along with this program; if not, write to the Free Software
18;;; Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
19
20(in-package "COMMON-LISP")
21
22(export 'reduce)
23
24;;; From OpenMCL.
25
26(defmacro list-reduce (function sequence start end initial-value ivp key)
27  (let ((what `(if ,key (funcall ,key (car sequence)) (car sequence))))
28    `(let ((sequence (nthcdr ,start ,sequence)))
29       (do ((count (if ,ivp ,start (1+ ,start)) (1+ count))
30            (sequence (if ,ivp sequence (cdr sequence))
31                      (cdr sequence))
32            (value (if ,ivp ,initial-value ,what)
33                   (funcall ,function value ,what)))
34           ((= count ,end) value)))))
35
36
37(defmacro list-reduce-from-end (function sequence start end
38                                         initial-value ivp key)
39  (let ((what `(if ,key (funcall ,key (car sequence)) (car sequence))))
40    `(let ((sequence (nthcdr (- (length ,sequence) ,end) (reverse ,sequence))))
41       (do ((count (if ,ivp ,start (1+ ,start)) (1+ count))
42            (sequence (if ,ivp sequence (cdr sequence))
43                      (cdr sequence))
44            (value (if ,ivp ,initial-value ,what)
45                   (funcall ,function ,what value)))
46           ((= count ,end) value)))))
47
48
49(defun reduce (function sequence &key from-end (start 0)
50                        end (initial-value nil ivp) key)
51  (unless end (setq end (length sequence)))
52  (if (= end start)
53      (if ivp initial-value (funcall function))
54      (if (listp sequence)
55          (if from-end
56              (list-reduce-from-end function sequence start end initial-value ivp key)
57              (list-reduce function sequence start end initial-value ivp key))
58          (let* ((disp (if from-end -1 1))
59                 (index (if from-end (1- end) start))
60                 (terminus (if from-end (1- start) end))
61                 (value (if ivp initial-value
62                            (let ((elt (aref sequence index)))
63                              (setq index (+ index disp))
64                              (if key (funcall key elt) elt))))
65                 (element nil))
66            (do* ()
67                 ((= index terminus) value)
68              (setq element (aref sequence index)
69                    index (+ index disp)
70                    element (if key (funcall key element) element)
71                    value (funcall function
72                                   (if from-end element value)
73                                   (if from-end value element))))))))
Note: See TracBrowser for help on using the repository browser.