source: trunk/j/src/org/armedbear/lisp/defstruct.lisp @ 4823

Last change on this file since 4823 was 4823, checked in by piso, 18 years ago

Work in progress.

File size: 8.3 KB
Line 
1;;; defstruct.lisp
2;;;
3;;; Copyright (C) 2003 Peter Graves
4;;; $Id: defstruct.lisp,v 1.33 2003-11-18 02:58:20 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 "SYSTEM")
21
22(defvar *ds-name*)
23(defvar *ds-conc-name*)
24(defvar *ds-constructors*)
25(defvar *ds-copier*)
26(defvar *ds-type*)
27(defvar *ds-named*)
28(defvar *ds-initial-offset*)
29(defvar *ds-predicate*)
30(defvar *ds-print-function*)
31
32(defun define-constructor (constructor slots)
33  (let* ((constructor-name (intern (car constructor)))
34         (slot-names (mapcar #'(lambda (x) (if (atom x) x (car x))) slots))
35         (inits (mapcar #'(lambda (x) (if (atom x) nil (cadr x))) slots))
36         (slot-descriptions (mapcar #'(lambda (x y) (list x y)) slot-names inits))
37         (keys (cons '&key slot-descriptions))
38         (elements slot-names))
39    (when *ds-named*
40      (push (list 'quote *ds-name*) elements))
41    (when *ds-initial-offset*
42      (dotimes (i *ds-initial-offset*)
43        (push nil elements)))
44    (cond ((eq *ds-type* 'list)
45           `((defun ,constructor-name ,keys
46               (list ,@elements))))
47          ((or (eq *ds-type* 'vector)
48               (and (consp *ds-type*) (eq (car *ds-type*) 'vector)))
49           (let ((element-type (if (consp *ds-type*) (cadr *ds-type*) t)))
50             `((defun ,constructor-name ,keys
51                 (make-array ,(length elements)
52                             :element-type ',element-type
53                             :initial-contents (list ,@elements))))))
54          (t
55           `((defun ,constructor-name ,keys
56               (%make-structure ',*ds-name* (list ,@slot-names))))))))
57
58(defun default-constructor-name ()
59  (concatenate 'string "MAKE-" (symbol-name *ds-name*)))
60
61(defun define-constructors (slots)
62  (if *ds-constructors*
63      (let ((results ()))
64        (dolist (constructor *ds-constructors*)
65          (when (car constructor)
66            (setf results (nconc results (define-constructor constructor slots)))))
67        results)
68      (define-constructor (cons (default-constructor-name) nil) slots)))
69
70(defun define-predicate ()
71  (when (and *ds-predicate*
72             (or *ds-named* (null *ds-type*)))
73    (let ((pred (intern *ds-predicate*)))
74      (cond ((eq *ds-type* 'list)
75             `((defun ,pred (object)
76                 (and (consp object) (eq (car object) ',*ds-name*)))))
77            ((or (eq *ds-type* 'vector)
78                 (and (consp *ds-type*) (eq (car *ds-type*) 'vector)))
79             `((defun ,pred (object)
80                 (and (vectorp object)
81                      (> (length object) 0)
82                      (eq (aref object 0) ',*ds-name*)))))
83            (t
84             `((defun ,pred (object)
85                 (typep object ',*ds-name*))))))))
86
87(defun get-slot-accessor (slot)
88  (when *ds-initial-offset*
89    (incf slot *ds-initial-offset*))
90  (when *ds-named*
91    (incf slot))
92  (cond ((eq *ds-type* 'list)
93         `(lambda (instance) (elt instance ,slot)))
94        ((or (eq *ds-type* 'vector)
95             (and (consp *ds-type*) (eq (car *ds-type*) 'vector)))
96         `(lambda (instance) (aref instance ,slot)))
97        (t
98         (case slot
99           (0 #'%structure-ref-0)
100           (1 #'%structure-ref-1)
101           (2 #'%structure-ref-2)
102           (t
103            `(lambda (instance) (%structure-ref instance ,slot)))))))
104
105(defun get-slot-mutator (slot)
106  (when *ds-initial-offset*
107    (incf slot *ds-initial-offset*))
108  (when *ds-named*
109    (incf slot))
110  (cond ((eq *ds-type* 'list)
111         `(lambda (instance value) (%set-elt instance ,slot value)))
112        ((or (eq *ds-type* 'vector)
113             (and (consp *ds-type*) (eq (car *ds-type*) 'vector)))
114         `(lambda (instance value) (%aset instance ,slot value)))
115        (t
116         (case slot
117           (0 #'%structure-set-0)
118           (1 #'%structure-set-1)
119           (2 #'%structure-set-2)
120           (t
121            `(lambda (instance value) (%structure-set instance ,slot value)))))))
122
123(defun define-access-function (slot-name index)
124  (let ((accessor
125         (if *ds-conc-name*
126             (intern (concatenate 'string (symbol-name *ds-conc-name*) (symbol-name slot-name)))
127             slot-name)))
128    `((setf (symbol-function ',accessor) ,(get-slot-accessor index))
129      (%put ',accessor 'setf-inverse ,(get-slot-mutator index )))))
130
131(defun define-access-functions (slots)
132  (let ((index 0)
133        (result ()))
134    (dolist (slot slots)
135      (let ((slot-name (if (atom slot) slot (car slot))))
136        (setf result (nconc result (define-access-function slot-name index))))
137      (incf index))
138    result))
139
140(defun define-copier ()
141  (when *ds-copier*
142    (cond ((eq *ds-type* 'list)
143           `((setf (fdefinition ',*ds-copier*) #'copy-list)))
144          ((or (eq *ds-type* 'vector)
145               (and (consp *ds-type*) (eq (car *ds-type*) 'vector)))
146           `((setf (fdefinition ',*ds-copier*) #'copy-seq)))
147          (t
148           `((setf (fdefinition ',*ds-copier*) #'copy-structure))))))
149
150(defun parse-1-option (option)
151  (case (car option)
152    (:conc-name
153     (setf *ds-conc-name* (if (symbolp (cadr option))
154                              (cadr option)
155                              (make-symbol (string (cadr option))))))
156    (:constructor
157     (let* ((args (cdr option))
158            (numargs (length args))
159            name arglist)
160       (case numargs
161         (0 ; Use default name.
162          (setf name (default-constructor-name))
163          (setf arglist nil)
164          (push (list name arglist) *ds-constructors*))
165         (1
166          (if (null (car args))
167              (setf name nil) ; No constructor.
168              (setf name (symbol-name (car args))))
169          (setf arglist nil)
170          (push (list name arglist) *ds-constructors*))
171         (2))))
172    (:copier
173     (let* ((args (cdr option))
174            (numargs (length args)))
175       (when (= numargs 1)
176          (setf *ds-copier* (car args)))))
177    (:initial-offset
178     (setf *ds-initial-offset* (cadr option)))
179    (:predicate
180     (when (= (length option) 2)
181       (if (null (cadr option))
182           (setf *ds-predicate* nil)
183           (setf *ds-predicate* (symbol-name (cadr option))))))
184    (:type
185     (setf *ds-type* (cadr option)))
186    (t
187     (format t "unrecognized DEFSTRUCT option: ~S~%" (car option)))))
188
189(defun parse-name-and-options (name-and-options)
190  (setf *ds-name* (car name-and-options))
191  (setf *ds-conc-name* (make-symbol (concatenate 'string (symbol-name *ds-name*) "-")))
192  (setf *ds-copier* (intern (concatenate 'string "COPY-" (symbol-name *ds-name*))))
193  (setf *ds-predicate* (concatenate 'string (symbol-name *ds-name*) "-P"))
194  (let ((options (cdr name-and-options)))
195    (dolist (option options)
196      (cond ((consp option)
197             (parse-1-option option))
198            ((eq option :named)
199             (setf *ds-named* t))
200            ((member option '(:constructor :copier :predicate :named
201                              :conc-name))
202             (parse-1-option (list option)))
203            (t
204             (error "unrecognized DEFSTRUCT option: ~S" option))))))
205
206(defmacro defstruct (name-and-options &rest slots)
207  (let ((*ds-name* nil)
208        (*ds-conc-name* nil)
209        (*ds-constructors* nil)
210        (*ds-copier* nil)
211        (*ds-type* nil)
212        (*ds-named* nil)
213        (*ds-initial-offset* nil)
214        (*ds-predicate* nil)
215        (*ds-print-function* nil))
216    (parse-name-and-options (if (atom name-and-options)
217                                (list name-and-options)
218                                name-and-options))
219    (when (stringp (car slots))
220      (setf (documentation *ds-name* 'structure) (pop slots)))
221    `(progn
222       (make-structure-class ',*ds-name*)
223       ,@(define-constructors slots)
224       ,@(define-predicate)
225       ,@(define-access-functions slots)
226       ,@(define-copier)
227       ',*ds-name*)))
Note: See TracBrowser for help on using the repository browser.