source: trunk/j/src/org/armedbear/lisp/compiler.lisp @ 4665

;;; compiler.lisp
;;;
;;; Copyright (C) 2003 Peter Graves
;;; $Id: compiler.lisp,v 1.59 2003-11-07 18:26:32 piso Exp $
;;;
;;; This program is free software; you can redistribute it and/or
;;; modify it under the terms of the GNU General Public License
;;; as published by the Free Software Foundation; either version 2
;;; of the License, or (at your option) any later version.
;;;
;;; This program 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 General Public License for more details.
;;;
;;; You should have received a copy of the GNU General Public License
;;; along with this program; if not, write to the Free Software
;;; Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

(unless (find-package "COMPILER")
  (make-package "COMPILER" :nicknames '("C") :use '("COMMON-LISP")))

;; (in-package "COMMON-LISP")

;; (export 'compile)

(in-package "COMPILER")

(defun compile-progn (forms)
  (mapcar #'compile-sexp forms))

(defun compile-setq (exprs)
  (when (oddp (length exprs))
    (error "odd number of arguments to SETQ"))
  (if (= 2 (length exprs))
      (list 'SETQ (car exprs) (compile-sexp (cadr exprs)))
      (do* ((result '(setq))
            (sym (car exprs) (car exprs))
            (val (cadr exprs) (cadr exprs)))
           ((null exprs) result)
        (setq result (append result (list sym) (list (compile-sexp val))))
        (setq exprs (cddr exprs)))))

(defun compile-cond (clauses)
  (let ((result nil))
    (dolist (clause clauses)
      (setq result (nconc result (list (compile-cond-clause clause)))))
    result))

(defun compile-cond-clause (clause)
  (let ((test (car clause))
        (forms (cdr clause)))
    (nconc (list (compile-sexp test)) (compile-progn forms))))

(defun compile-case (keyform clauses)
  (let ((result (list (compile-sexp keyform))))
    (dolist (clause clauses)
      (setq result (nconc result (list (compile-case-clause clause)))))
    result))

(defun compile-case-clause (clause)
  (let ((keys (car clause))
        (forms (cdr clause)))
    (nconc (list keys) (compile-progn forms))))

(defun compile-tagbody (body)
  (let ((rest body)
        (result ()))
    (do () ((null rest) result)
      (if (atom (car rest))
          (setq result (nconc result (list (car rest))))
          (setq result (append result (list (compile-sexp (car rest))))))
      (setq rest (cdr rest)))))

(defun compile-locals (locals)
  (let ((result nil))
    (dolist (local locals)
      (setq result (append result (list (compile-local-def local)))))
    result))

(defun compile-local-def (def)
  (let ((name (car def))
        (arglist (cadr def))
        (body (cddr def)))
    (list* name arglist (compile-progn body))))

(defun compile-let-vars (vars)
  (let ((result nil))
    (dolist (var vars)
      (if (consp var)
          (let* ((v (car var))
                 (expr (cadr var)))
            (unless (symbolp v)
              (error 'type-error))
            (setq result (append result (list (list v (compile-sexp expr))))))
          (setq result (append result (list var)))))
    result))

;; (defun define-local-macro (name lambda-list &rest body)
;;   (let* ((form (gensym))
;;          (env (gensym))
;;          (body (sys::parse-defmacro lambda-list form body name 'macrolet
;;                                     :environment env))
;;          (expander `(lambda (,form ,env) (block ,name ,body))))
;;     (format t "expander = ~S~%" expander)
;;     (sys::make-macro expander)))

(defun define-local-macro (name lambda-list body)
  (let* ((form (gensym))
         (env (gensym))
         (body (sys::parse-defmacro lambda-list form body name 'macrolet
                                    :environment env))
         (expander `(lambda (,form ,env) (block ,name ,body)))
         (compiled-expander (%compile nil expander)))
;;     (format t "expander = ~S~%" expander)
;;     (format t "compiled-expander = ~S~%" compiled-expander)
    (or compiled-expander expander)))

(defparameter *local-macros* ())

(defun local-macro-function (name)
  (getf *local-macros* name))

(defun expand-local-macro (form)
  (let ((expansion (funcall (local-macro-function (car form)) form nil)))
    ;; If the expansion turns out to be a bare symbol, wrap it with PROGN so it
    ;; won't be mistaken for a tag in an enclosing TAGBODY.
    (if (symbolp expansion)
        (list 'progn expansion)
        expansion)))

(defun compile-macrolet (form)
  (let ((*local-macros* *local-macros*)
        (macros (cadr form))
        (body (cddr form))
        (res ())
        compiled-body)
    (dolist (macro macros)
      (let ((name (car macro))
            (lambda-list (cadr macro))
            (forms (cddr macro)))
        (push (define-local-macro name lambda-list forms) *local-macros*)
        (push name *local-macros*)
        (push (list* name lambda-list (compile-progn forms)) res)))
    (setf compiled-body (compile-progn body))
    (setf res (list* 'progn compiled-body))
    res))

(defun compile-special (form)
  (let ((first (car form)))
    (case first
      (BLOCK
       (unless (>= (length form) 2)
         (error "wrong number of arguments for BLOCK"))
       (unless (symbolp (cadr form))
         (error 'type-error))
       (list* 'block (cadr form) (mapcar #'compile-sexp (cddr form))))
      (COND
       (cons 'cond (compile-cond (cdr form))))
      (QUOTE
       form)
      ((AND OR)
       (cons first
             (mapcar #'compile-sexp (cdr form))))
      (FUNCTION
       (if (and (consp (cadr form)) (eq (caadr form) 'setf))
           form
           (cons 'function (list (compile-sexp (cadr form))))))
      (WHEN
       (cons 'when (mapcar #'compile-sexp (cdr form))))
      ((LET LET*)
       (list* first (compile-let-vars (cadr form)) (mapcar #'compile-sexp (cddr form))))
      (SETQ
       (compile-setq (cdr form)))
      (PROGN
       (let ((body (cdr form)))
         (if (= (length body) 1)
             (let ((res (compile-sexp (car body))))
               ;; If the result turns out to be a bare symbol, leave it wrapped
               ;; with PROGN so it won't be mistaken for a tag in an enclosing
               ;; TAGBODY.
               (if (symbolp res)
                   (list 'progn res)
                   res))
             (cons 'progn (mapcar #'compile-sexp body)))))
      (IF
       (unless (<= 2 (length (cdr form)) 3)
         (error "wrong number of arguments for IF"))
       (cons 'if (mapcar #'compile-sexp (cdr form))))
      ((CASE ECASE)
       (cons first (compile-case (cadr form) (cddr form))))
      (DOLIST
       (let ((args (cadr form))
             (body (cddr form)))
         (cons first (cons args (compile-progn body)))))
      ((DO DO*)
       (let ((second (second form))
             (third (third form))
             (body (cdddr form)))
         (list* first second third (mapcar #'compile-sexp body))))
      (DOTIMES
       (let ((args (cadr form))
             (body (cddr form)))
         (list* first args (compile-progn body))))
      (TAGBODY
       (let ((body (cdr form)))
         (cons 'tagbody (compile-tagbody body))))
      ((LABELS FLET)
;;        (format t "LABELS *local-macros* = ~S~%" *local-macros*)
       (let* ((locals (cadr form))
              (body (cddr form))
              (compiled-locals (compile-locals locals))
              (compiled-body (compile-progn body)))
;;          (format t "body          = ~S~%" body)
;;          (format t "compiled-body = ~S~%" compiled-body)
;;          (append '(labels) (list compiled-locals) compiled-body)))
         (list* first compiled-locals compiled-body)))
      (RETURN
       (if (cdr form)
           (cons 'return (list (compile-sexp (cadr form))))
           form))
      (RETURN-FROM form)
      (UNLESS
       (cons 'unless (mapcar #'compile-sexp (cdr form))))
      (UNWIND-PROTECT
       (list* 'unwind-protect (compile-sexp (cadr form)) (mapcar #'compile-sexp (cddr form))))
      (MULTIPLE-VALUE-PROG1
       (list* 'unwind-protect (compile-sexp (cadr form)) (mapcar #'compile-sexp (cddr form))))
      (THE
       (compile-sexp (caddr form)))
      (GO form)
      (MACROLET
       (compile-macrolet form))
      (MULTIPLE-VALUE-BIND
       (let ((vars (second form))
             (values-form (third form))
             (body (cdddr form)))
         (list* 'multiple-value-bind vars (compile-sexp values-form)
                (mapcar #'compile-sexp body))))
      (MULTIPLE-VALUE-SETQ
       (list 'multiple-value-setq (second form) (compile-sexp (third form))))
      (t
;;        (format t "COMPILE-SPECIAL skipping ~S~%" first)
       form))))

;; EXPAND-MACRO is like MACROEXPAND, but EXPAND-MACRO quits if it encounters a
;; macro that's also implemented as a special operator, so interpreted code can
;; use the (faster) special operator implementation.
(defun expand-macro (form)
  (loop
    (multiple-value-bind (result expanded) (macroexpand-1 form)
      (unless expanded (return-from expand-macro result))
      (when (and (consp result)
                 (symbolp (car result))
                 (special-operator-p (car result)))
        (return-from expand-macro result))
      (setq form result))))

(defun compile-sexp (form)
  (if (atom form) form
      (let ((first (car form)))
        (when (symbolp first)
          (cond ((local-macro-function first)
                 (return-from compile-sexp (expand-local-macro form)))
                ((eq first 'LAMBDA)
                 (return-from compile-sexp (list* 'LAMBDA (second form)
                                                  (mapcar #'compile-sexp (cddr form)))))
                ((special-operator-p first)
                 (return-from compile-sexp (compile-special form)))
                ((macro-function first)
                 (return-from compile-sexp (compile-sexp (expand-macro form))))))
        (cons first (mapcar #'compile-sexp (cdr form))))))

(defun %compile (name &optional definition)
  (unless definition
    (setq definition (or (and (symbolp name) (macro-function name))
                         (fdefinition name))))
  (let (expr result)
    (cond ((functionp definition)
           (multiple-value-bind (form closure-p)
             (function-lambda-expression definition)
             (unless form
               (format t "; No lambda expression available for ~S.~%" name)
               (return-from %compile (values nil t t)))
             (when closure-p
               (format t "; Unable to compile function ~S defined in non-null lexical environment.~%" name)
               (finish-output)
               (return-from %compile (values nil t t)))
             (setq expr form)))
          ((and (consp definition) (eq (car definition) 'lambda))
           (setq expr definition))
          (t
           (error 'type-error)))
    (setq result (sys::coerce-to-function (compile-sexp expr)))
    (when (and name (functionp result))
      (sys::%set-lambda-name result name)
      (sys::%set-call-count result (sys::%call-count definition))
      (sys::%set-arglist result (sys::arglist definition))
      (if (and (symbolp name) (macro-function name))
          (setf (fdefinition name) (sys::make-macro result))
          (setf (fdefinition name) result)))
    (values (or name result) nil nil)))

(defun compile-package (pkg &key verbose)
  (dolist (sym (sys::package-symbols pkg))
    (when (fboundp sym)
      ;;       (unless (or (special-operator-p sym) (macro-function sym))
      (unless (special-operator-p sym)
        (let ((f (fdefinition sym)))
          (unless (compiled-function-p f)
            (when verbose
              (format t "compiling ~S~%" sym)
              (finish-output))
            (%compile sym))))))
  t)

(compile-package :compiler)
(compile-package :sys)
(compile-package :cl)

(in-package :cl)

(defun compile (name &optional definition)
  (if (and name (fboundp name) (typep (symbol-function name) 'generic-function))
      (values name nil nil)
      (c::%compile name definition)))

;; Redefine DEFMACRO to compile the expansion function on the fly.
(defmacro defmacro (name lambda-list &rest body)
  (let* ((form (gensym))
         (env (gensym))
         (body (sys::parse-defmacro lambda-list form body name 'defmacro
                                    :environment env))
         (expander `(lambda (,form ,env) (block ,name ,body))))
    `(progn
       (if (special-operator-p ',name)
         (sys::%put ',name
                    'sys::macroexpand-macro
                    (sys::make-macro (or (c::%compile nil ,expander) ,expander)))
         (sys::fset ',name
                    (sys::make-macro (or (c::%compile nil ,expander) ,expander))))
       ',name)))

;; Make an exception just this one time...
(sys::fset 'defmacro (get 'defmacro 'sys::macroexpand-macro))

;; Redefine DEFUN to compile the definition on the fly.
(defmacro defun (name lambda-list &rest body &environment env)
  `(progn
     (sys::%defun ',name ',lambda-list ',body ,env)
     (compiler::%compile ',name)
     ',name))