;;; srfi-9.scm --- define-record-type
;; Copyright (C) 2001, 2002, 2006, 2009, 2010, 2011, 2012,
;; 2013 Free Software Foundation, Inc.
;;
;; This library is free software; you can redistribute it and/or
;; modify it under the terms of the GNU Lesser General Public
;; License as published by the Free Software Foundation; either
;; version 3 of the License, or (at your option) any later version.
;;
;; This library is distributed in the hope that it will be useful,
;; but WITHOUT ANY WARRANTY; without even the implied warranty of
;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
;; Lesser General Public License for more details.
;;
;; You should have received a copy of the GNU Lesser General Public
;; License along with this library; if not, write to the Free Software
;; Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
;;; Commentary:
;; This module exports the syntactic form `define-record-type', which
;; is the means for creating record types defined in SRFI-9.
;;
;; The syntax of a record type definition is:
;;
;; <record type definition>
;; -> (define-record-type <type name>
;; (<constructor name> <field tag> ...)
;; <predicate name>
;; <field spec> ...)
;;
;; <field spec> -> (<field tag> <getter name>)
;; -> (<field tag> <getter name> <setter name>)
;;
;; <field tag> -> <identifier>
;; <... name> -> <identifier>
;;
;; Usage example:
;;
;; guile> (use-modules (srfi srfi-9))
;; guile> (define-record-type :foo (make-foo x) foo?
;; (x get-x) (y get-y set-y!))
;; guile> (define f (make-foo 1))
;; guile> f
;; #<:foo x: 1 y: #f>
;; guile> (get-x f)
;; 1
;; guile> (set-y! f 2)
;; 2
;; guile> (get-y f)
;; 2
;; guile> f
;; #<:foo x: 1 y: 2>
;; guile> (foo? f)
;; #t
;; guile> (foo? 1)
;; #f
;;; Code:
(define-module (srfi srfi-9)
#:use-module (srfi srfi-1)
#:use-module (system base ck)
#:export (define-record-type))
(cond-expand-provide (current-module) '(srfi-9))
;; Roll our own instead of using the public `define-inlinable'. This is
;; because the public one has a different `make-procedure-name', so
;; using it would require users to recompile code that uses SRFI-9. See
;; <http://lists.gnu.org/archive/html/guile-devel/2011-04/msg00111.html>.
;;
(define-syntax-rule (define-inlinable (name formals ...) body ...)
(define-tagged-inlinable () (name formals ...) body ...))
;; 'define-tagged-inlinable' has an additional feature: it stores a map
;; of keys to values that can be retrieved at expansion time. This is
;; currently used to retrieve the rtd id, field index, and record copier
;; macro for an arbitrary getter.
(define-syntax-rule (%%on-error err) err)
(define %%type #f) ; a private syntax literal
(define-syntax getter-type
(syntax-rules (quote)
((_ s 'getter 'err)
(getter (%%on-error err) %%type s))))
(define %%index #f) ; a private syntax literal
(define-syntax getter-index
(syntax-rules (quote)
((_ s 'getter 'err)
(getter (%%on-error err) %%index s))))
(define %%copier #f) ; a private syntax literal
(define-syntax getter-copier
(syntax-rules (quote)
((_ s 'getter 'err)
(getter (%%on-error err) %%copier s))))
(define-syntax define-tagged-inlinable
(lambda (x)
(define (make-procedure-name name)
(datum->syntax name
(symbol-append '% (syntax->datum name)
'-procedure)))
(syntax-case x ()
((_ ((key value) ...) (name formals ...) body ...)
(identifier? #'name)
(with-syntax ((proc-name (make-procedure-name #'name))
((args ...) (generate-temporaries #'(formals ...))))
#`(begin
(define (proc-name formals ...)
body ...)
(define-syntax name
(lambda (x)
(syntax-case x (%%on-error key ...)
((_ (%%on-error err) key s) #'(ck s 'value)) ...
((_ args ...)
#'((lambda (formals ...)
body ...)
args ...))
((_ a (... ...))
(syntax-violation 'name "Wrong number of arguments" x))
(_
(identifier? x)
#'proc-name))))))))))
(define (default-record-printer s p)
(display "#<" p)
(display (record-type-name (record-type-descriptor s)) p)
(let loop ((fields (record-type-fields (record-type-descriptor s)))
(off 0))
(cond
((not (null? fields))
(display " " p)
(display (car fields) p)
(display ": " p)
(write (struct-ref s off) p)
(loop (cdr fields) (+ 1 off)))))
(display ">" p))
(define (throw-bad-struct s who)
(throw 'wrong-type-arg who
"Wrong type argument: ~S" (list s)
(list s)))
(define (make-copier-id type-name)
(datum->syntax type-name
(symbol-append '%% (syntax->datum type-name)
'-set-fields)))
(define-syntax %%set-fields
(lambda (x)
(syntax-case x ()
((_ type-name (getter-id ...) check? s (getter expr) ...)
(every identifier? #'(getter ...))
(let ((copier-name (syntax->datum (make-copier-id #'type-name)))
(getter+exprs #'((getter expr) ...)))
(define (lookup id default-expr)
(let ((results
(filter (lambda (g+e)
(free-identifier=? id (car g+e)))
getter+exprs)))
(case (length results)
((0) default-expr)
((1) (cadar results))
(else (syntax-violation
copier-name "duplicate getter" x id)))))
(for-each (lambda (id)
(or (find (lambda (getter-id)
(free-identifier=? id getter-id))
#'(getter-id ...))
(syntax-violation
copier-name "unknown getter" x id)))
#'(getter ...))
(with-syntax ((unsafe-expr
#`(make-struct
type-name 0
#,@(map (lambda (getter index)
(lookup getter #`(struct-ref s #,index)))
#'(getter-id ...)
(iota (length #'(getter-id ...)))))))
(if (syntax->datum #'check?)
#`(if (eq? (struct-vtable s) type-name)
unsafe-expr
(throw-bad-struct
s '#,(datum->syntax #'here copier-name)))
#'unsafe-expr)))))))
(define-syntax %define-record-type
(lambda (x)
(define (field-identifiers field-specs)
(map (lambda (field-spec)
(syntax-case field-spec ()
((name getter) #'name)
((name getter setter) #'name)))
field-specs))
(define (getter-identifiers field-specs)
(map (lambda (field-spec)
(syntax-case field-spec ()
((name getter) #'getter)
((name getter setter) #'getter)))
field-specs))
(define (constructor form type-name constructor-spec field-names)
(syntax-case constructor-spec ()
((ctor field ...)
(every identifier? #'(field ...))
(let ((ctor-args (map (lambda (field)
(let ((name (syntax->datum field)))
(or (memq name field-names)
(syntax-violation
(syntax-case form ()
((macro . args)
(syntax->datum #'macro)))
"unknown field in constructor spec"
form field))
(cons name field)))
#'(field ...))))
#`(define-inlinable #,constructor-spec
(make-struct #,type-name 0
#,@(map (lambda (name)
(assq-ref ctor-args name))
field-names)))))))
(define (getters type-name getter-ids copier-id)
(map (lambda (getter index)
#`(define-tagged-inlinable
((%%type #,type-name)
(%%index #,index)
(%%copier #,copier-id))
(#,getter s)
(if (eq? (struct-vtable s) #,type-name)
(struct-ref s #,index)
(throw-bad-struct s '#,getter))))
getter-ids
(iota (length getter-ids))))
(define (copier type-name getter-ids copier-id)
#`(define-syntax-rule
(#,copier-id check? s (getter expr) (... ...))
(%%set-fields #,type-name #,getter-ids
check? s (getter expr) (... ...))))
(define (setters type-name field-specs)
(filter-map (lambda (field-spec index)
(syntax-case field-spec ()
((name getter) #f)
((name getter setter)
#`(define-inlinable (setter s val)
(if (eq? (struct-vtable s) #,type-name)
(struct-set! s #,index val)
(throw-bad-struct s 'setter))))))
field-specs
(iota (length field-specs))))
(define (functional-setters copier-id field-specs)
(filter-map (lambda (field-spec index)
(syntax-case field-spec ()
((name getter) #f)
((name getter setter)
#`(define-inlinable (setter s val)
(#,copier-id #t s (getter val))))))
field-specs
(iota (length field-specs))))
(define (record-layout immutable? count)
(let ((desc (if immutable? "pr" "pw")))
(string-concatenate (make-list count desc))))
(syntax-case x ()
((_ immutable? form type-name constructor-spec predicate-name
field-spec ...)
(let ()
(define (syntax-error message subform)
(syntax-violation (syntax-case #'form ()
((macro . args) (syntax->datum #'macro)))
message #'form subform))
(and (boolean? (syntax->datum #'immutable?))
(or (identifier? #'type-name)
(syntax-error "expected type name" #'type-name))
(syntax-case #'constructor-spec ()
((ctor args ...)
(every identifier? #'(ctor args ...))
#t)
(_ (syntax-error "invalid constructor spec"
#'constructor-spec)))
(or (identifier? #'predicate-name)
(syntax-error "expected predicate name" #'predicate-name))
(every (lambda (spec)
(syntax-case spec ()
((field getter) #t)
((field getter setter) #t)
(_ (syntax-error "invalid field spec" spec))))
#'(field-spec ...))))
(let* ((field-ids (field-identifiers #'(field-spec ...)))
(getter-ids (getter-identifiers #'(field-spec ...)))
(field-count (length field-ids))
(immutable? (syntax->datum #'immutable?))
(layout (record-layout immutable? field-count))
(field-names (map syntax->datum field-ids))
(ctor-name (syntax-case #'constructor-spec ()
((ctor args ...) #'ctor)))
(copier-id (make-copier-id #'type-name)))
#`(begin
#,(constructor #'form #'type-name #'constructor-spec field-names)
(define type-name
(let ((rtd (make-struct/no-tail
record-type-vtable
'#,(datum->syntax #'here (make-struct-layout layout))
default-record-printer
'type-name
'#,field-ids)))
(set-struct-vtable-name! rtd 'type-name)
(struct-set! rtd (+ 2 vtable-offset-user) #,ctor-name)
rtd))
(define-inlinable (predicate-name obj)
(and (struct? obj)
(eq? (struct-vtable obj) type-name)))
#,@(getters #'type-name getter-ids copier-id)
#,(copier #'type-name getter-ids copier-id)
#,@(if immutable?
(functional-setters copier-id #'(field-spec ...))
(setters #'type-name #'(field-spec ...))))))
((_ immutable? form . rest)
(syntax-violation
(syntax-case #'form ()
((macro . args) (syntax->datum #'macro)))
"invalid record definition syntax"
#'form)))))
(define-syntax-rule (define-record-type name ctor pred fields ...)
(%define-record-type #f (define-record-type name ctor pred fields ...)
name ctor pred fields ...))
;;; srfi-9.scm ends here