mirror of
https://git.savannah.gnu.org/git/guile.git
synced 2025-05-04 14:20:26 +02:00
303 lines
11 KiB
Scheme
303 lines
11 KiB
Scheme
;;; open-coding primitive procedures
|
||
|
||
;; Copyright (C) 2009 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
|
||
|
||
;;; Code:
|
||
|
||
(define-module (language tree-il primitives)
|
||
#:use-module (system base pmatch)
|
||
#:use-module (rnrs bytevector)
|
||
#:use-module (system base syntax)
|
||
#:use-module (language tree-il)
|
||
#:use-module (srfi srfi-16)
|
||
#:export (resolve-primitives! add-interesting-primitive!
|
||
expand-primitives! effect-free-primitive?))
|
||
|
||
(define *interesting-primitive-names*
|
||
'(apply @apply
|
||
call-with-values @call-with-values
|
||
call-with-current-continuation @call-with-current-continuation
|
||
call/cc
|
||
values
|
||
eq? eqv? equal?
|
||
memq memv
|
||
= < > <= >= zero?
|
||
+ * - / 1- 1+ quotient remainder modulo
|
||
ash logand logior logxor
|
||
not
|
||
pair? null? list? acons cons cons*
|
||
|
||
list vector
|
||
|
||
car cdr
|
||
set-car! set-cdr!
|
||
|
||
caar cadr cdar cddr
|
||
|
||
caaar caadr cadar caddr cdaar cdadr cddar cdddr
|
||
|
||
caaaar caaadr caadar caaddr cadaar cadadr caddar cadddr
|
||
cdaaar cdaadr cdadar cdaddr cddaar cddadr cdddar cddddr
|
||
|
||
vector-ref vector-set!
|
||
variable-ref variable-set!
|
||
;; args of variable-set are switched; it needs special help
|
||
|
||
struct? struct-vtable make-struct
|
||
|
||
bytevector-u8-ref bytevector-u8-set!
|
||
bytevector-s8-ref bytevector-s8-set!
|
||
|
||
bytevector-u16-ref bytevector-u16-set!
|
||
bytevector-u16-native-ref bytevector-u16-native-set!
|
||
bytevector-s16-ref bytevector-s16-set!
|
||
bytevector-s16-native-ref bytevector-s16-native-set!
|
||
|
||
bytevector-u32-ref bytevector-u32-set!
|
||
bytevector-u32-native-ref bytevector-u32-native-set!
|
||
bytevector-s32-ref bytevector-s32-set!
|
||
bytevector-s32-native-ref bytevector-s32-native-set!
|
||
|
||
bytevector-u64-ref bytevector-u64-set!
|
||
bytevector-u64-native-ref bytevector-u64-native-set!
|
||
bytevector-s64-ref bytevector-s64-set!
|
||
bytevector-s64-native-ref bytevector-s64-native-set!
|
||
|
||
bytevector-ieee-single-ref bytevector-ieee-single-set!
|
||
bytevector-ieee-single-native-ref bytevector-ieee-single-native-set!
|
||
bytevector-ieee-double-ref bytevector-ieee-double-set!
|
||
bytevector-ieee-double-native-ref bytevector-ieee-double-native-set!))
|
||
|
||
(define (add-interesting-primitive! name)
|
||
(hashq-set! *interesting-primitive-vars*
|
||
(or (module-variable (current-module) name)
|
||
(error "unbound interesting primitive" name))
|
||
name))
|
||
|
||
(define *interesting-primitive-vars* (make-hash-table))
|
||
|
||
(for-each add-interesting-primitive! *interesting-primitive-names*)
|
||
|
||
(define *effect-free-primitives*
|
||
'(values
|
||
eq? eqv? equal?
|
||
= < > <= >= zero?
|
||
+ * - / 1- 1+ quotient remainder modulo
|
||
not
|
||
pair? null? list? acons cons cons*
|
||
list vector
|
||
car cdr
|
||
caar cadr cdar cddr
|
||
caaar caadr cadar caddr cdaar cdadr cddar cdddr
|
||
caaaar caaadr caadar caaddr cadaar cadadr caddar cadddr
|
||
cdaaar cdaadr cdadar cdaddr cddaar cddadr cdddar cddddr
|
||
vector-ref
|
||
struct? struct-vtable make-struct
|
||
bytevector-u8-ref bytevector-s8-ref
|
||
bytevector-u16-ref bytevector-u16-native-ref
|
||
bytevector-s16-ref bytevector-s16-native-ref
|
||
bytevector-u32-ref bytevector-u32-native-ref
|
||
bytevector-s32-ref bytevector-s32-native-ref
|
||
bytevector-u64-ref bytevector-u64-native-ref
|
||
bytevector-s64-ref bytevector-s64-native-ref
|
||
bytevector-ieee-single-ref bytevector-ieee-single-native-ref
|
||
bytevector-ieee-double-ref bytevector-ieee-double-native-ref))
|
||
|
||
|
||
(define *effect-free-primitive-table* (make-hash-table))
|
||
|
||
(for-each (lambda (x) (hashq-set! *effect-free-primitive-table* x #t))
|
||
*effect-free-primitives*)
|
||
|
||
(define (effect-free-primitive? prim)
|
||
(hashq-ref *effect-free-primitive-table* prim))
|
||
|
||
(define (resolve-primitives! x mod)
|
||
(post-order!
|
||
(lambda (x)
|
||
(record-case x
|
||
((<toplevel-ref> src name)
|
||
(and=> (hashq-ref *interesting-primitive-vars*
|
||
(module-variable mod name))
|
||
(lambda (name) (make-primitive-ref src name))))
|
||
((<module-ref> src mod name public?)
|
||
;; for the moment, we're disabling primitive resolution for
|
||
;; public refs because resolve-interface can raise errors.
|
||
(let ((m (and (not public?) (resolve-module mod))))
|
||
(and m
|
||
(and=> (hashq-ref *interesting-primitive-vars*
|
||
(module-variable m name))
|
||
(lambda (name) (make-primitive-ref src name))))))
|
||
(else #f)))
|
||
x))
|
||
|
||
|
||
|
||
(define *primitive-expand-table* (make-hash-table))
|
||
|
||
(define (expand-primitives! x)
|
||
(pre-order!
|
||
(lambda (x)
|
||
(record-case x
|
||
((<application> src proc args)
|
||
(and (primitive-ref? proc)
|
||
(let ((expand (hashq-ref *primitive-expand-table*
|
||
(primitive-ref-name proc))))
|
||
(and expand (apply expand src args)))))
|
||
(else #f)))
|
||
x))
|
||
|
||
;;; I actually did spend about 10 minutes trying to redo this with
|
||
;;; syntax-rules. Patches appreciated.
|
||
;;;
|
||
(define-macro (define-primitive-expander sym . clauses)
|
||
(define (inline-args args)
|
||
(let lp ((in args) (out '()))
|
||
(cond ((null? in) `(list ,@(reverse out)))
|
||
((symbol? in) `(cons* ,@(reverse out) ,in))
|
||
((pair? (car in))
|
||
(lp (cdr in)
|
||
(cons `(make-application src (make-primitive-ref src ',(caar in))
|
||
,(inline-args (cdar in)))
|
||
out)))
|
||
((symbol? (car in))
|
||
;; assume it's locally bound
|
||
(lp (cdr in) (cons (car in) out)))
|
||
((number? (car in))
|
||
(lp (cdr in) (cons `(make-const src ,(car in)) out)))
|
||
(else
|
||
(error "what what" (car in))))))
|
||
(define (consequent exp)
|
||
(cond
|
||
((pair? exp)
|
||
(pmatch exp
|
||
((if ,test ,then ,else)
|
||
`(if ,test
|
||
,(consequent then)
|
||
,(consequent else)))
|
||
(else
|
||
`(make-application src (make-primitive-ref src ',(car exp))
|
||
,(inline-args (cdr exp))))))
|
||
((symbol? exp)
|
||
;; assume locally bound
|
||
exp)
|
||
((number? exp)
|
||
`(make-const src ,exp))
|
||
(else (error "bad consequent yall" exp))))
|
||
`(hashq-set! *primitive-expand-table*
|
||
',sym
|
||
(case-lambda
|
||
,@(let lp ((in clauses) (out '()))
|
||
(if (null? in)
|
||
(reverse (cons '(else #f) out))
|
||
(lp (cddr in)
|
||
(cons `((src . ,(car in))
|
||
,(consequent (cadr in))) out)))))))
|
||
|
||
(define-primitive-expander zero? (x)
|
||
(= x 0))
|
||
|
||
(define-primitive-expander +
|
||
() 0
|
||
(x) x
|
||
(x y) (if (and (const? y)
|
||
(let ((y (const-exp y)))
|
||
(and (number? y) (exact? y) (= y 1))))
|
||
(1+ x)
|
||
(if (and (const? y)
|
||
(let ((y (const-exp y)))
|
||
(and (number? y) (exact? y) (= y -1))))
|
||
(1- x)
|
||
(if (and (const? x)
|
||
(let ((x (const-exp x)))
|
||
(and (number? x) (exact? x) (= x 1))))
|
||
(1+ y)
|
||
(+ x y))))
|
||
(x y z . rest) (+ x (+ y z . rest)))
|
||
|
||
(define-primitive-expander *
|
||
() 1
|
||
(x) x
|
||
(x y z . rest) (* x (* y z . rest)))
|
||
|
||
(define-primitive-expander -
|
||
(x) (- 0 x)
|
||
(x y) (if (and (const? y)
|
||
(let ((y (const-exp y)))
|
||
(and (number? y) (exact? y) (= y 1))))
|
||
(1- x)
|
||
(- x y))
|
||
(x y z . rest) (- x (+ y z . rest)))
|
||
|
||
(define-primitive-expander /
|
||
(x) (/ 1 x)
|
||
(x y z . rest) (/ x (* y z . rest)))
|
||
|
||
(define-primitive-expander caar (x) (car (car x)))
|
||
(define-primitive-expander cadr (x) (car (cdr x)))
|
||
(define-primitive-expander cdar (x) (cdr (car x)))
|
||
(define-primitive-expander cddr (x) (cdr (cdr x)))
|
||
(define-primitive-expander caaar (x) (car (car (car x))))
|
||
(define-primitive-expander caadr (x) (car (car (cdr x))))
|
||
(define-primitive-expander cadar (x) (car (cdr (car x))))
|
||
(define-primitive-expander caddr (x) (car (cdr (cdr x))))
|
||
(define-primitive-expander cdaar (x) (cdr (car (car x))))
|
||
(define-primitive-expander cdadr (x) (cdr (car (cdr x))))
|
||
(define-primitive-expander cddar (x) (cdr (cdr (car x))))
|
||
(define-primitive-expander cdddr (x) (cdr (cdr (cdr x))))
|
||
(define-primitive-expander caaaar (x) (car (car (car (car x)))))
|
||
(define-primitive-expander caaadr (x) (car (car (car (cdr x)))))
|
||
(define-primitive-expander caadar (x) (car (car (cdr (car x)))))
|
||
(define-primitive-expander caaddr (x) (car (car (cdr (cdr x)))))
|
||
(define-primitive-expander cadaar (x) (car (cdr (car (car x)))))
|
||
(define-primitive-expander cadadr (x) (car (cdr (car (cdr x)))))
|
||
(define-primitive-expander caddar (x) (car (cdr (cdr (car x)))))
|
||
(define-primitive-expander cadddr (x) (car (cdr (cdr (cdr x)))))
|
||
(define-primitive-expander cdaaar (x) (cdr (car (car (car x)))))
|
||
(define-primitive-expander cdaadr (x) (cdr (car (car (cdr x)))))
|
||
(define-primitive-expander cdadar (x) (cdr (car (cdr (car x)))))
|
||
(define-primitive-expander cdaddr (x) (cdr (car (cdr (cdr x)))))
|
||
(define-primitive-expander cddaar (x) (cdr (cdr (car (car x)))))
|
||
(define-primitive-expander cddadr (x) (cdr (cdr (car (cdr x)))))
|
||
(define-primitive-expander cdddar (x) (cdr (cdr (cdr (car x)))))
|
||
(define-primitive-expander cddddr (x) (cdr (cdr (cdr (cdr x)))))
|
||
|
||
(define-primitive-expander cons*
|
||
(x) x
|
||
(x y) (cons x y)
|
||
(x y . rest) (cons x (cons* y . rest)))
|
||
|
||
(define-primitive-expander acons (x y z)
|
||
(cons (cons x y) z))
|
||
|
||
(define-primitive-expander apply (f a0 . args)
|
||
(@apply f a0 . args))
|
||
|
||
(define-primitive-expander call-with-values (producer consumer)
|
||
(@call-with-values producer consumer))
|
||
|
||
(define-primitive-expander call-with-current-continuation (proc)
|
||
(@call-with-current-continuation proc))
|
||
|
||
(define-primitive-expander call/cc (proc)
|
||
(@call-with-current-continuation proc))
|
||
|
||
(define-primitive-expander values (x) x)
|
||
|
||
;; swap args
|
||
(define-primitive-expander variable-set! (var val)
|
||
(variable-set val var))
|