ecl/src/cmp/cmpmulti.lsp

281 lines
9.9 KiB
Common Lisp

;;;; Copyright (c) 1984, Taiichi Yuasa and Masami Hagiya.
;;;; Copyright (c) 1990, Giuseppe Attardi.
;;;;
;;;; This program is free software; you can redistribute it and/or
;;;; modify it under the terms of the GNU Library General Public
;;;; License as published by the Free Software Foundation; either
;;;; version 2 of the License, or (at your option) any later version.
;;;;
;;;; See file '../Copyright' for full details.
;;;; CMPMULT Multiple-value-call and Multiple-value-prog1.
(in-package "COMPILER")
(defun c1multiple-value-call (args &aux forms)
(check-args-number 'MULTIPLE-VALUE-CALL args 1)
(cond
;; (M-V-C #'FUNCTION) => (FUNCALL #'FUNCTION)
((endp (rest args)) (c1funcall args))
;; (M-V-C #'FUNCTION (VALUES A ... Z)) => (FUNCALL #'FUNCTION A ... Z)
((and (= (length args) 2)
(consp (setq forms (second args)))
(eq 'VALUES (first forms)))
(c1funcall (list* (first args) (rest forms))))
;; More complicated case.
(t (let ((funob (c1expr (first args))))
(make-c1form 'MULTIPLE-VALUE-CALL funob funob (c1args* (rest args)))))))
(defun c2multiple-value-call (funob forms)
(let* ((tot (make-lcl-var :rep-type :cl-index))
(*temp* *temp*)
(loc (maybe-save-value funob forms)))
(wt-nl "{ cl_index " tot "=0;")
(let ((*unwind-exit* `((STACK ,tot) ,@*unwind-exit*)))
(let ((*destination* 'VALUES))
(dolist (form forms)
(c2expr* form)
(wt-nl tot "+=cl_stack_push_values();")))
(c2funcall funob 'ARGS-PUSHED loc tot))
(wt "}"))
)
(defun c1multiple-value-prog1 (args)
(check-args-number 'MULTIPLE-VALUE-PROG1 args 1)
(make-c1form* 'MULTIPLE-VALUE-PROG1 :args (c1expr (first args))
(c1args* (rest args))))
(defun c2multiple-value-prog1 (form forms)
(if (eq 'TRASH *destination*)
;; dont bother saving values
(c2progn (cons form forms))
(let ((nr (make-lcl-var :type :cl-index)))
(let ((*destination* 'VALUES)) (c2expr* form))
(wt-nl "{ cl_index " nr "=cl_stack_push_values();")
(let ((*destination* 'TRASH)
(*unwind-exit* `((STACK ,nr) ,@*unwind-exit*)))
(dolist (form forms)
(c2expr* form)))
(wt-nl "cl_stack_pop_values(" nr ");}")
(unwind-exit 'VALUES))))
;;; Beppe:
;;; this is the WRONG way to handle 1 value problem.
;;; should be done in c2values, so that (values (truncate a b)) can
;;; be used to restrict to one value, so we would not have to warn
;;; if this occurred in a proclaimed fun.
(defun c1values (args)
(make-c1form* 'VALUES :args (c1args* args)))
(defun c2values (forms)
(when (and (eq *destination* 'RETURN-OBJECT)
(rest forms)
(consp *current-form*)
(eq 'DEFUN (first *current-form*)))
(cmpwarn "Trying to return multiple values. ~
~%;But ~a was proclaimed to have single value.~
~%;Only first one will be assured."
(second *current-form*)))
(cond
;; When the values are not going to be used, then just
;; process each form separately.
((eq *destination* 'TRASH)
(mapc #'c2expr forms))
;; For (VALUES) we can replace the output with either NIL (if the value
;; is actually used) and set only NVALUES when the value is the output
;; of a function.
((endp forms)
(cond ((eq *destination* 'RETURN)
(wt-nl "value0=Cnil; NVALUES=0;")
(unwind-exit 'RETURN))
((eq *destination* 'VALUES)
(wt-nl "VALUES(0)=Cnil; NVALUES=0;")
(unwind-exit 'VALUES))
(t
(unwind-exit 'NIL))))
;; For a single form, we must simply ensure that we only take a single
;; value of those that the function may output.
((endp (rest forms))
(let ((*destination* 'VALUE0))
(c2expr* (first forms)))
(unwind-exit 'VALUE0))
;; In all other cases, we store the values in the VALUES vector,
;; and force the compiler to retrieve anything out of it.
(t
(let* ((nv (length forms))
(*inline-blocks* 0)
(forms (nreverse (coerce-locs (inline-args forms)))))
;; By inlining arguments we make sure that VL has no call to funct.
;; Reverse args to avoid clobbering VALUES(0)
(wt-nl "NVALUES=" nv ";")
(do ((vl forms (rest vl))
(i (1- (length forms)) (1- i)))
((null vl))
(declare (fixnum i))
(wt-nl "VALUES(" i ")=" (first vl) ";"))
(unwind-exit 'VALUES)
(close-inline-blocks)))))
(defun c1multiple-value-setq (args &aux (info (make-info)) (vrefs nil)
(vars nil) (temp-vars nil) (late-bindings nil))
(check-args-number 'MULTIPLE-VALUE-SETQ args 2 2)
(dolist (var (reverse (first args)))
(cmpck (not (symbolp var)) "The variable ~s is not a symbol." var)
(setq var (chk-symbol-macrolet var))
(cond ((symbolp var)
(cmpck (constantp var)
"The constant ~s is being assigned a value." var)
(push var vars))
(t (let ((new-var (gensym)))
(push new-var vars)
(push new-var temp-vars)
(push `(setf ,var ,new-var) late-bindings)))))
(if temp-vars
(c1expr `(let* (,@temp-vars)
(multiple-value-setq ,vars ,@(second args))
,@late-bindings))
(let ((value (c1expr (second args))))
(dolist (var vars
(make-c1form 'MULTIPLE-VALUE-SETQ info (nreverse vrefs) value))
(setq var (c1vref var))
(push var vrefs)
(unless (type-and 'T (var-type var))
(cmpwarn "Variable ~s appeared in a MULTIPLE-VALUE-SETQ and declared to have type ~S."
(var-name var)
(var-type var)))
(push var (info-changed-vars info))))))
(defun multiple-value-check (vrefs form)
(and (rest vrefs)
(eq (c1form-name form) 'CALL-GLOBAL)
(let ((fname (c1form-arg 0 form)))
(when (and (symbolp fname)
(get-sysprop fname 'PROCLAIMED-RETURN-TYPE))
(cmpwarn "~A was proclaimed to have only one return value. ~
~%;;; But you appear to want multiple values." fname)))))
(defun c2multiple-value-setq (vrefs form)
(multiple-value-check vrefs form)
(let* ((*lcl* *lcl*)
(nr (make-lcl-var :type :int)))
(let ((*destination* 'VALUES)) (c2expr* form))
(wt-nl "{int " nr "=NVALUES;")
(do ((vs vrefs (rest vs))
(i 0 (1+ i))
(vref))
((endp vs))
(declare (fixnum i))
(setq vref (first vs))
(wt-nl "if (" nr ">0) {")
(set-var (list 'VALUE i) vref) ; (second vref) ccb
(unless (endp (rest vs)) (wt-nl nr "--;"))
(wt-nl "} else {") (set-var nil vref) ; (second vref) ccb
(wt "}"))
(unless (eq *exit* 'RETURN) (wt-nl))
(wt-nl "if (NVALUES>1) NVALUES=1;}")
(unwind-exit (if vrefs (first vrefs) '(VALUE 0)))))
(defun c1multiple-value-bind (args &aux (vars nil) (vnames nil) init-form
ss is ts body other-decls
(*vars* *vars*))
(check-args-number 'MULTIPLE-VALUE-BIND args 2)
(multiple-value-setq (body ss ts is other-decls) (c1body (cddr args) nil))
(c1add-globals ss)
(dolist (s (first args))
(push s vnames)
(push (c1make-var s ss is ts) vars))
(setq init-form (c1expr (second args)))
(dolist (v (setq vars (nreverse vars)))
(push-vars v))
(check-vdecl vnames ts is)
(setq body (c1decl-body other-decls body))
(dolist (var vars) (check-vref var))
(make-c1form* 'MULTIPLE-VALUE-BIND :type (c1form-type body)
:args vars init-form body)
)
(defun c2multiple-value-bind (vars init-form body)
(multiple-value-check vars init-form)
;; 0) Compile the form which is going to give us the values
(let ((*destination* 'VALUES)) (c2expr* init-form))
(let* ((*unwind-exit* *unwind-exit*)
(*env-lvl* *env-lvl*)
(*env* *env*)
(*lcl* *lcl*)
(labels nil)
(env-grows nil)
(nr (make-lcl-var :type :int)))
;; 1) Retrieve the number of output values
(wt-nl "{ int " nr "=NVALUES;")
;; 2) For all variables which are not special and do not belong to
;; a closure, make a local C variable.
(dolist (var vars)
(declare (type var var))
(let ((kind (local var)))
(if kind
(progn
(bind (next-lcl) var)
(wt-nl *volatile* (rep-type-name kind) " " var ";")
(wt-comment (var-name var)))
(unless env-grows (setq env-grows (var-ref-ccb var))))))
;; 3) If there are closure variables, set up an environment.
(when (setq env-grows (env-grows env-grows))
(let ((env-lvl *env-lvl*))
(wt-nl "{ volatile cl_object env" (incf *env-lvl*)
" = env" env-lvl ";")))
;; 4) Loop for assigning values to variables
(do ((vs vars (rest vs))
(i 0 (1+ i))
(value '(VALUE 0)))
((endp vs))
(declare (fixnum i))
(push (next-label) labels)
(wt-nl "if (" nr "--<=0) ") (wt-go (first labels))
(setf (second value) i)
(bind value (first vs)))
;; 5) Loop for setting default values when there are less output
;; than variables.
(let ((label (next-label)))
(wt-nl) (wt-go label)
(setq labels (nreverse labels))
(let ((*suppress-compiler-warnings* t))
;; suppress the warning by default-init
(dolist (v vars)
(wt-label (first labels))
(pop labels)
;; DEFAULT-INIT returns a LOCATION form, whose only argument
;; is the location that we pass to BIND.
(bind (c1form-arg 0 (default-init v)) v)))
(wt-label label))
;; 6) Compile the body. If there are bindings of special variables,
;; these bindings are undone here.
(c2expr body)
;; 7) Close the C expression.
(when env-grows (wt "}"))
(wt "}"))
)
;;; ----------------------------------------------------------------------
(put-sysprop 'multiple-value-call 'c1special #'c1multiple-value-call)
(put-sysprop 'multiple-value-call 'c2 #'c2multiple-value-call)
(put-sysprop 'multiple-value-prog1 'c1special #'c1multiple-value-prog1)
(put-sysprop 'multiple-value-prog1 'c2 #'c2multiple-value-prog1)
(put-sysprop 'values 'c1 #'c1values)
(put-sysprop 'values 'c2 #'c2values)
(put-sysprop 'multiple-value-setq 'c1 #'c1multiple-value-setq)
(put-sysprop 'multiple-value-setq 'c2 #'c2multiple-value-setq)
(put-sysprop 'multiple-value-bind 'c1 #'c1multiple-value-bind)
(put-sysprop 'multiple-value-bind 'c2 #'c2multiple-value-bind)