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

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

Work in progress.

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