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

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

Work in progress.

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