source: trunk/j/src/org/armedbear/lisp/bit-array-ops.lisp @ 8556

Last change on this file since 8556 was 8556, checked in by piso, 16 years ago

Work in progress.

File size: 10.4 KB
Line 
1;;; bit-array-ops.lisp
2;;;
3;;; Copyright (C) 2003-2005 Peter Graves
4;;; $Id: bit-array-ops.lisp,v 1.4 2005-02-13 04:02:56 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;;; Adapted from CMUCL.
21
22(in-package #:system)
23
24(defun bit-array-same-dimensions-p (array1 array2)
25  (declare (type (array bit) array1 array2))
26  (and (= (array-rank array1)
27    (array-rank array2))
28       (dotimes (index (array-rank array1) t)
29   (when (/= (array-dimension array1 index)
30       (array-dimension array2 index))
31     (return nil)))))
32
33(defun require-same-dimensions (array1 array2)
34  (unless (bit-array-same-dimensions-p array1 array2)
35    (error 'program-error
36           "~S and ~S do not have the same dimensions."
37           array1 array2)))
38
39(defun pick-result-array (result-bit-array bit-array-1)
40  (case result-bit-array
41    ((t) bit-array-1)
42    ((nil) (make-array (array-dimensions bit-array-1)
43           :element-type 'bit
44           :initial-element 0))
45    (t
46     (require-same-dimensions bit-array-1 result-bit-array)
47     result-bit-array)))
48
49(defmacro def-bit-array-op (name function)
50  `(defun ,name (bit-array-1 bit-array-2 &optional result-bit-array)
51     ,(format nil
52        "Perform a bit-wise ~A on the elements of BIT-ARRAY-1 and ~
53               BIT-ARRAY-2,~%  putting the results in RESULT-BIT-ARRAY.  ~
54               If RESULT-BIT-ARRAY is T,~%  BIT-ARRAY-1 is used.  If ~
55               RESULT-BIT-ARRAY is NIL or omitted, a new array is~%  created.  ~
56               All the arrays must have the same rank and dimensions."
57        (symbol-name function))
58     (require-same-dimensions bit-array-1 bit-array-2)
59     (let ((result-bit-array (pick-result-array result-bit-array bit-array-1)))
60       (dotimes (i (array-total-size result-bit-array) result-bit-array)
61         (setf (row-major-aref result-bit-array i)
62               (logand (,function (row-major-aref bit-array-1 i)
63                                  (row-major-aref bit-array-2 i))
64                       1))))))
65
66;; (def-bit-array-op bit-and   logand)
67;; (def-bit-array-op bit-ior   logior)
68;; (def-bit-array-op bit-xor   logxor)
69;; (def-bit-array-op bit-eqv   logeqv)
70;; (def-bit-array-op bit-nand  lognand)
71;; (def-bit-array-op bit-nor   lognor)
72;; (def-bit-array-op bit-andc1 logandc1)
73;; (def-bit-array-op bit-andc2 logandc2)
74;; (def-bit-array-op bit-orc1  logorc1)
75;; (def-bit-array-op bit-orc2  logorc2)
76
77(defun bit-and (bit-array-1 bit-array-2 &optional result-bit-array)
78  (require-same-dimensions bit-array-1 bit-array-2)
79  (let ((result-bit-array (pick-result-array result-bit-array bit-array-1)))
80    (if (and (simple-bit-vector-p bit-array-1)
81             (simple-bit-vector-p bit-array-2)
82             (simple-bit-vector-p result-bit-array))
83        (%simple-bit-vector-bit-and bit-array-1 bit-array-2 result-bit-array)
84        (dotimes (i (array-total-size result-bit-array) result-bit-array)
85          (setf (row-major-aref result-bit-array i)
86                (logand (row-major-aref bit-array-1 i)
87                        (row-major-aref bit-array-2 i)))))))
88
89(defun bit-ior (bit-array-1 bit-array-2 &optional result-bit-array)
90  (require-same-dimensions bit-array-1 bit-array-2)
91  (let ((result-bit-array (pick-result-array result-bit-array bit-array-1)))
92    (if (and (simple-bit-vector-p bit-array-1)
93             (simple-bit-vector-p bit-array-2)
94             (simple-bit-vector-p result-bit-array))
95        (%simple-bit-vector-bit-ior bit-array-1 bit-array-2 result-bit-array)
96        (dotimes (i (array-total-size result-bit-array) result-bit-array)
97          (setf (row-major-aref result-bit-array i)
98                (logior (row-major-aref bit-array-1 i)
99                        (row-major-aref bit-array-2 i)))))))
100
101(defun bit-xor (bit-array-1 bit-array-2 &optional result-bit-array)
102  (require-same-dimensions bit-array-1 bit-array-2)
103  (let ((result-bit-array (pick-result-array result-bit-array bit-array-1)))
104    (if (and (simple-bit-vector-p bit-array-1)
105             (simple-bit-vector-p bit-array-2)
106             (simple-bit-vector-p result-bit-array))
107        (%simple-bit-vector-bit-xor bit-array-1 bit-array-2 result-bit-array)
108        (dotimes (i (array-total-size result-bit-array) result-bit-array)
109          (setf (row-major-aref result-bit-array i)
110                (logxor (row-major-aref bit-array-1 i)
111                        (row-major-aref bit-array-2 i)))))))
112
113(defun bit-eqv (bit-array-1 bit-array-2 &optional result-bit-array)
114  (require-same-dimensions bit-array-1 bit-array-2)
115  (let ((result-bit-array (pick-result-array result-bit-array bit-array-1)))
116    (if (and (simple-bit-vector-p bit-array-1)
117             (simple-bit-vector-p bit-array-2)
118             (simple-bit-vector-p result-bit-array))
119        (%simple-bit-vector-bit-eqv bit-array-1 bit-array-2 result-bit-array)
120        (dotimes (i (array-total-size result-bit-array) result-bit-array)
121          (setf (row-major-aref result-bit-array i)
122                (logand (logeqv (row-major-aref bit-array-1 i)
123                                (row-major-aref bit-array-2 i))
124                        1))))))
125
126(defun bit-nand (bit-array-1 bit-array-2 &optional result-bit-array)
127  (require-same-dimensions bit-array-1 bit-array-2)
128  (let ((result-bit-array (pick-result-array result-bit-array bit-array-1)))
129    (if (and (simple-bit-vector-p bit-array-1)
130             (simple-bit-vector-p bit-array-2)
131             (simple-bit-vector-p result-bit-array))
132        (%simple-bit-vector-bit-nand bit-array-1 bit-array-2 result-bit-array)
133        (dotimes (i (array-total-size result-bit-array) result-bit-array)
134          (setf (row-major-aref result-bit-array i)
135                (logand (lognand (row-major-aref bit-array-1 i)
136                                 (row-major-aref bit-array-2 i))
137                        1))))))
138
139(defun bit-nor (bit-array-1 bit-array-2 &optional result-bit-array)
140  (require-same-dimensions bit-array-1 bit-array-2)
141  (let ((result-bit-array (pick-result-array result-bit-array bit-array-1)))
142    (if (and (simple-bit-vector-p bit-array-1)
143             (simple-bit-vector-p bit-array-2)
144             (simple-bit-vector-p result-bit-array))
145        (%simple-bit-vector-bit-nor bit-array-1 bit-array-2 result-bit-array)
146        (dotimes (i (array-total-size result-bit-array) result-bit-array)
147          (setf (row-major-aref result-bit-array i)
148                (logand (lognor (row-major-aref bit-array-1 i)
149                                (row-major-aref bit-array-2 i))
150                        1))))))
151
152(defun bit-andc1 (bit-array-1 bit-array-2 &optional result-bit-array)
153  (require-same-dimensions bit-array-1 bit-array-2)
154  (let ((result-bit-array (pick-result-array result-bit-array bit-array-1)))
155    (if (and (simple-bit-vector-p bit-array-1)
156             (simple-bit-vector-p bit-array-2)
157             (simple-bit-vector-p result-bit-array))
158        (%simple-bit-vector-bit-andc1 bit-array-1 bit-array-2 result-bit-array)
159        (dotimes (i (array-total-size result-bit-array) result-bit-array)
160          (setf (row-major-aref result-bit-array i)
161                (logand (logandc1 (row-major-aref bit-array-1 i)
162                                  (row-major-aref bit-array-2 i))
163                        1))))))
164
165(defun bit-andc2 (bit-array-1 bit-array-2 &optional result-bit-array)
166  (require-same-dimensions bit-array-1 bit-array-2)
167  (let ((result-bit-array (pick-result-array result-bit-array bit-array-1)))
168    (if (and (simple-bit-vector-p bit-array-1)
169             (simple-bit-vector-p bit-array-2)
170             (simple-bit-vector-p result-bit-array))
171        (%simple-bit-vector-bit-andc2 bit-array-1 bit-array-2 result-bit-array)
172        (dotimes (i (array-total-size result-bit-array) result-bit-array)
173          (setf (row-major-aref result-bit-array i)
174                (logand (logandc2 (row-major-aref bit-array-1 i)
175                                  (row-major-aref bit-array-2 i))
176                        1))))))
177
178(defun bit-orc1 (bit-array-1 bit-array-2 &optional result-bit-array)
179  (require-same-dimensions bit-array-1 bit-array-2)
180  (let ((result-bit-array (pick-result-array result-bit-array bit-array-1)))
181    (if (and (simple-bit-vector-p bit-array-1)
182             (simple-bit-vector-p bit-array-2)
183             (simple-bit-vector-p result-bit-array))
184        (%simple-bit-vector-bit-orc1 bit-array-1 bit-array-2 result-bit-array)
185        (dotimes (i (array-total-size result-bit-array) result-bit-array)
186          (setf (row-major-aref result-bit-array i)
187                (logand (logorc1 (row-major-aref bit-array-1 i)
188                                 (row-major-aref bit-array-2 i))
189                        1))))))
190
191(defun bit-orc2 (bit-array-1 bit-array-2 &optional result-bit-array)
192  (require-same-dimensions bit-array-1 bit-array-2)
193  (let ((result-bit-array (pick-result-array result-bit-array bit-array-1)))
194    (if (and (simple-bit-vector-p bit-array-1)
195             (simple-bit-vector-p bit-array-2)
196             (simple-bit-vector-p result-bit-array))
197        (%simple-bit-vector-bit-orc2 bit-array-1 bit-array-2 result-bit-array)
198        (dotimes (i (array-total-size result-bit-array) result-bit-array)
199          (setf (row-major-aref result-bit-array i)
200                (logand (logorc2 (row-major-aref bit-array-1 i)
201                                 (row-major-aref bit-array-2 i))
202                        1))))))
203
204(defun bit-not (bit-array &optional result-bit-array)
205  "Performs a bit-wise logical NOT on the elements of BIT-ARRAY,
206   putting the results in RESULT-BIT-ARRAY.  If RESULT-BIT-ARRAY is T,
207   BIT-ARRAY is used.  If RESULT-BIT-ARRAY is NIL or omitted, a new array is
208   created.  Both arrays must have the same rank and dimensions."
209  (let ((result-bit-array (pick-result-array result-bit-array bit-array)))
210    (dotimes (i (array-total-size result-bit-array) result-bit-array)
211      (setf (row-major-aref result-bit-array i)
212            (logxor (row-major-aref bit-array i) 1)))))
Note: See TracBrowser for help on using the repository browser.