X-Git-Url: http://repo.macrolet.net/gitweb/?a=blobdiff_plain;f=ecmalisp.lisp;h=149a7779fbb38d971d882b9e9e08aceacc87abaa;hb=4e103a7d0be6bf1e3cd0ed39934afb36e525a243;hp=9e6fa12b4fe2093bd3525988185859cca68cf8f4;hpb=68cd13d873bcaf0367d5327e80bb4e995b773250;p=jscl.git diff --git a/ecmalisp.lisp b/ecmalisp.lisp index 9e6fa12..149a777 100644 --- a/ecmalisp.lisp +++ b/ecmalisp.lisp @@ -42,10 +42,17 @@ `(eval-when-compile ,@(mapcar (lambda (decl) `(!proclaim ',decl)) decls))) - (declaim (constant nil t) (special t nil)) - (setq nil 'nil) + (defmacro defconstant (name value &optional docstring) + `(progn + (declaim (special ,name)) + (declaim (constant ,name)) + (setq ,name ,value) + ,@(when (stringp docstring) `((oset ',name "vardoc" ,docstring))) + ',name)) + + (defconstant t 't) + (defconstant nil 'nil) (js-vset "nil" nil) - (setq t 't) (defmacro lambda (args &body body) `(function (lambda ,args ,@body))) @@ -133,6 +140,7 @@ (defun second (x) (cadr x)) (defun third (x) (caddr x)) (defun fourth (x) (cadddr x)) + (defun rest (x) (cdr x)) (defun list (&rest args) args) (defun atom (x) @@ -348,14 +356,45 @@ (defun concat-two (s1 s2) (concat-two s1 s2)) - (defun mapcar (func list) - (if (null list) - '() - (cons (funcall func (car list)) - (mapcar func (cdr list))))) + (defmacro with-collect (&body body) + (let ((head (gensym)) + (tail (gensym))) + `(let* ((,head (cons 'sentinel nil)) + (,tail ,head)) + (flet ((collect (x) + (rplacd ,tail (cons x nil)) + (setq ,tail (cdr ,tail)) + x)) + ,@body) + (cdr ,head)))) + + (defun map1 (func list) + (with-collect + (while list + (collect (funcall func (car list))) + (setq list (cdr list))))) + + (defmacro loop (&body body) + `(while t ,@body)) + + (defun mapcar (func list &rest lists) + (let ((lists (cons list lists))) + (with-collect + (block loop + (loop + (let ((elems (map1 #'car lists))) + (do ((tail lists (cdr tail))) + ((null tail)) + (when (null (car tail)) (return-from loop)) + (rplaca tail (cdar tail))) + (collect (apply func elems)))))))) (defun identity (x) x) + (defun constantly (x) + (lambda (&rest args) + x)) + (defun copy-list (x) (mapcar #'identity x)) @@ -428,6 +467,10 @@ (- x #\0) nil)) + (defun digit-char (weight) + (and (<= 0 weight 9) + (char "0123456789" weight))) + (defun subseq (seq a &optional b) (cond ((stringp seq) @@ -437,34 +480,6 @@ (t (error "Unsupported argument.")))) - (defun parse-integer (string) - (block nil - (let ((value 0) - (index 0) - (size (length string)) - (sign 1)) - (when (zerop size) (return (values nil 0))) - ;; Optional sign - (case (char string 0) - (#\+ (incf index)) - (#\- (setq sign -1) - (incf index))) - ;; First digit - (unless (and (< index size) - (setq value (digit-char-p (char string index)))) - (values nil index)) - (incf index) - ;; Other digits - (while (< index size) - (let ((digit (digit-char-p (char string index)))) - (unless digit (return)) - (setq value (+ (* value 10) digit)) - (incf index))) - (if (or (= index size) - (char= (char string index) #\space)) - (values (* sign value) index) - (values nil index))))) - (defun some (function seq) (cond ((stringp seq) @@ -538,7 +553,83 @@ (error "Wrong argument type! it should be a symbol")) (oget x "vardoc")))) - ;; Packages + (defmacro multiple-value-bind (variables value-from &body body) + `(multiple-value-call (lambda (&optional ,@variables &rest ,(gensym)) + ,@body) + ,value-from)) + + (defmacro multiple-value-list (value-from) + `(multiple-value-call #'list ,value-from)) + + + ;;; Generalized references (SETF) + + (defvar *setf-expanders* nil) + + (defun get-setf-expansion (place) + (if (symbolp place) + (let ((value (gensym))) + (values nil + nil + `(,value) + `(setq ,place ,value) + place)) + (let* ((access-fn (car place)) + (expander (cdr (assoc access-fn *setf-expanders*)))) + (when (null expander) + (error "Unknown generalized reference.")) + (apply expander (cdr place))))) + + (defmacro define-setf-expander (access-fn lambda-list &body body) + (unless (symbolp access-fn) + (error "ACCESS-FN must be a symbol.")) + `(progn (push (cons ',access-fn (lambda ,lambda-list ,@body)) + *setf-expanders*) + ',access-fn)) + + (defmacro setf (&rest pairs) + (cond + ((null pairs) + nil) + ((null (cdr pairs)) + (error "Odd number of arguments to setf.")) + ((null (cddr pairs)) + (let ((place (first pairs)) + (value (second pairs))) + (multiple-value-bind (vars vals store-vars writer-form reader-form) + (get-setf-expansion place) + ;; TODO: Optimize the expansion a little bit to avoid let* + ;; or multiple-value-bind when unnecesary. + `(let* ,(mapcar #'list vars vals) + (multiple-value-bind ,store-vars + ,value + ,writer-form))))) + (t + `(progn + ,@(do ((pairs pairs (cddr pairs)) + (result '() (cons `(setf ,(car pairs) ,(cadr pairs)) result))) + ((null pairs) + (reverse result))))))) + + (define-setf-expander car (x) + (let ((cons (gensym)) + (new-value (gensym))) + (values (list cons) + (list x) + (list new-value) + `(progn (rplaca ,cons ,new-value) ,new-value) + `(car ,cons)))) + + (define-setf-expander cdr (x) + (let ((cons (gensym)) + (new-value (gensym))) + (values (list cons) + (list x) + (list new-value) + `(progn (rplacd ,cons ,new-value) ,new-value) + `(car ,cons)))) + + ;;; Packages (defvar *package-list* nil) @@ -607,29 +698,37 @@ ;; This function is used internally to initialize the CL package ;; with the symbols built during bootstrap. (defun %intern-symbol (symbol) - (let ((symbols (%package-symbols *common-lisp-package*))) - (oset symbol "package" *common-lisp-package*) + (let* ((package + (if (in "package" symbol) + (find-package-or-fail (oget symbol "package")) + *common-lisp-package*)) + (symbols (%package-symbols package))) + (oset symbol "package" package) + (when (eq package *keyword-package*) + (oset symbol "value" symbol)) (oset symbols (symbol-name symbol) symbol))) - (defun %find-symbol (name package) - (let ((package (find-package-or-fail package))) - (let ((symbols (%package-symbols package))) - (if (in name symbols) - (cons (oget symbols name) t) - (dolist (used (package-use-list package) (cons nil nil)) - (let ((exports (%package-external-symbols used))) - (when (in name exports) - (return-from %find-symbol - (cons (oget exports name) t))))))))) - (defun find-symbol (name &optional (package *package*)) - (car (%find-symbol name package))) + (let* ((package (find-package-or-fail package)) + (externals (%package-external-symbols package)) + (symbols (%package-symbols package))) + (cond + ((in name externals) + (values (oget externals name) :external)) + ((in name symbols) + (values (oget symbols name) :internal)) + (t + (dolist (used (package-use-list package) (values nil nil)) + (let ((exports (%package-external-symbols used))) + (when (in name exports) + (return (values (oget exports name) :inherit))))))))) (defun intern (name &optional (package *package*)) (let ((package (find-package-or-fail package))) - (let ((result (%find-symbol name package))) - (if (cdr result) - (car result) + (multiple-value-bind (symbol foundp) + (find-symbol name package) + (if foundp + (values symbol foundp) (let ((symbols (%package-symbols package))) (oget symbols name) (let ((symbol (make-symbol name))) @@ -637,7 +736,8 @@ (when (eq package *keyword-package*) (oset symbol "value" symbol) (export (list symbol) package)) - (oset symbols name symbol))))))) + (oset symbols name symbol) + (values symbol nil))))))) (defun symbol-package (symbol) (unless (symbolp symbol) @@ -725,49 +825,7 @@ (values-array (list-to-vector list))) (defun values (&rest args) - (values-list args)) - - (defmacro multiple-value-bind (variables value-from &body body) - `(multiple-value-call (lambda (&optional ,@variables &rest ,(gensym)) - ,@body) - ,value-from)) - - (defmacro multiple-value-list (value-from) - `(multiple-value-call #'list ,value-from))) - - -;;; Like CONCAT, but prefix each line with four spaces. Two versions -;;; of this function are available, because the Ecmalisp version is -;;; very slow and bootstraping was annoying. - -#+ecmalisp -(defun indent (&rest string) - (let ((input (join string))) - (let ((output "") - (index 0) - (size (length input))) - (when (plusp (length input)) (concatf output " ")) - (while (< index size) - (let ((str - (if (and (char= (char input index) #\newline) - (< index (1- size)) - (not (char= (char input (1+ index)) #\newline))) - (concat (string #\newline) " ") - (string (char input index))))) - (concatf output str)) - (incf index)) - output))) - -#+common-lisp -(defun indent (&rest string) - (with-output-to-string (*standard-output*) - (with-input-from-string (input (join string)) - (loop - for line = (read-line input nil) - while line - do (write-string " ") - do (write-line line))))) - + (values-list args))) (defun integer-to-string (x) (cond @@ -780,22 +838,8 @@ (while (not (zerop x)) (push (mod x 10) digits) (setq x (truncate x 10))) - (join (mapcar (lambda (d) (string (char "0123456789" d))) - digits)))))) - - -;;; Wrap X with a Javascript code to convert the result from -;;; Javascript generalized booleans to T or NIL. -(defun js!bool (x) - (concat "(" x "?" (ls-compile t) ": " (ls-compile nil) ")")) - -;;; Concatenate the arguments and wrap them with a self-calling -;;; Javascript anonymous function. It is used to make some Javascript -;;; statements valid expressions and provide a private scope as well. -;;; It could be defined as function, but we could do some -;;; preprocessing in the future. -(defmacro js!selfcall (&body body) - `(concat "(function(){" *newline* (indent ,@body) "})()")) + (mapconcat (lambda (x) (string (digit-char x))) + digits))))) ;;; Printer @@ -805,15 +849,17 @@ (defun prin1-to-string (form) (cond ((symbolp form) - (if (cdr (%find-symbol (symbol-name form) *package*)) - (symbol-name form) - (let ((package (symbol-package form)) - (name (symbol-name form))) - (concat (cond - ((null package) "#") - ((eq package (find-package "KEYWORD")) "") - (t (package-name package))) - ":" name)))) + (multiple-value-bind (symbol foundp) + (find-symbol (symbol-name form) *package*) + (if (and foundp (eq symbol form)) + (symbol-name form) + (let ((package (symbol-package form)) + (name (symbol-name form))) + (concat (cond + ((null package) "#") + ((eq package (find-package "KEYWORD")) "") + (t (package-name package))) + ":" name))))) ((integerp form) (integer-to-string form)) ((stringp form) (concat "\"" (escape-string form) "\"")) ((functionp form) @@ -991,6 +1037,40 @@ (intern name package) (find-symbol name package)))) + +(defun !parse-integer (string junk-allow) + (block nil + (let ((value 0) + (index 0) + (size (length string)) + (sign 1)) + (when (zerop size) (return (values nil 0))) + ;; Optional sign + (case (char string 0) + (#\+ (incf index)) + (#\- (setq sign -1) + (incf index))) + ;; First digit + (unless (and (< index size) + (setq value (digit-char-p (char string index)))) + (return (values nil index))) + (incf index) + ;; Other digits + (while (< index size) + (let ((digit (digit-char-p (char string index)))) + (unless digit (return)) + (setq value (+ (* value 10) digit)) + (incf index))) + (if (or junk-allow + (= index size) + (char= (char string index) #\space)) + (values (* sign value) index) + (values nil index))))) + +#+ecmalisp +(defun parse-integer (string) + (!parse-integer string nil)) + (defvar *eof* (gensym)) (defun ls-read (stream) (skip-whitespaces-and-comments stream) @@ -1019,9 +1099,8 @@ (read-sharp stream)) (t (let ((string (read-until stream #'terminalp))) - (if (every #'digit-char-p string) - (parse-integer string) - (read-symbol string))))))) + (or (values (!parse-integer string nil)) + (read-symbol string))))))) (defun ls-read-from-string (string) (ls-read (make-string-stream string))) @@ -1034,6 +1113,61 @@ ;;; too. The respective real functions are defined in the target (see ;;; the beginning of this file) as well as some primitive functions. +(defun code (&rest args) + (mapconcat (lambda (arg) + (cond + ((null arg) "") + ((integerp arg) (integer-to-string arg)) + ((stringp arg) arg) + (t (error "Unknown argument.")))) + args)) + +;;; Wrap X with a Javascript code to convert the result from +;;; Javascript generalized booleans to T or NIL. +(defun js!bool (x) + (code "(" x "?" (ls-compile t) ": " (ls-compile nil) ")")) + +;;; Concatenate the arguments and wrap them with a self-calling +;;; Javascript anonymous function. It is used to make some Javascript +;;; statements valid expressions and provide a private scope as well. +;;; It could be defined as function, but we could do some +;;; preprocessing in the future. +(defmacro js!selfcall (&body body) + `(code "(function(){" *newline* (indent ,@body) "})()")) + +;;; Like CODE, but prefix each line with four spaces. Two versions +;;; of this function are available, because the Ecmalisp version is +;;; very slow and bootstraping was annoying. + +#+ecmalisp +(defun indent (&rest string) + (let ((input (apply #'code string))) + (let ((output "") + (index 0) + (size (length input))) + (when (plusp (length input)) (concatf output " ")) + (while (< index size) + (let ((str + (if (and (char= (char input index) #\newline) + (< index (1- size)) + (not (char= (char input (1+ index)) #\newline))) + (concat (string #\newline) " ") + (string (char input index))))) + (concatf output str)) + (incf index)) + output))) + +#+common-lisp +(defun indent (&rest string) + (with-output-to-string (*standard-output*) + (with-input-from-string (input (apply #'code string)) + (loop + for line = (read-line input nil) + while line + do (write-string " ") + do (write-line line))))) + + ;;; A Form can return a multiple values object calling VALUES, like ;;; values(arg1, arg2, ...). It will work in any context, as well as ;;; returning an individual object. However, if the special variable @@ -1042,7 +1176,6 @@ ;;; function call. (defvar *multiple-value-p* nil) - (defun make-binding (name type value &optional declarations) (list name type value declarations)) @@ -1090,7 +1223,7 @@ (defvar *variable-counter* 0) (defun gvarname (symbol) - (concat "v" (integer-to-string (incf *variable-counter*)))) + (code "v" (incf *variable-counter*))) (defun translate-variable (symbol) (binding-value (lookup-in-lexenv symbol *environment* 'variable))) @@ -1158,40 +1291,64 @@ *compilations*)) (define-compilation if (condition true false) - (concat "(" (ls-compile condition) " !== " (ls-compile nil) - " ? " (ls-compile true *multiple-value-p*) - " : " (ls-compile false *multiple-value-p*) - ")")) + (code "(" (ls-compile condition) " !== " (ls-compile nil) + " ? " (ls-compile true *multiple-value-p*) + " : " (ls-compile false *multiple-value-p*) + ")")) -(defvar *lambda-list-keywords* '(&optional &rest)) +(defvar *lambda-list-keywords* '(&optional &rest &key)) (defun list-until-keyword (list) (if (or (null list) (member (car list) *lambda-list-keywords*)) nil (cons (car list) (list-until-keyword (cdr list))))) +(defun lambda-list-section (keyword lambda-list) + (list-until-keyword (cdr (member keyword lambda-list)))) + (defun lambda-list-required-arguments (lambda-list) (list-until-keyword lambda-list)) (defun lambda-list-optional-arguments-with-default (lambda-list) - (mapcar #'ensure-list (list-until-keyword (cdr (member '&optional lambda-list))))) + (mapcar #'ensure-list (lambda-list-section '&optional lambda-list))) (defun lambda-list-optional-arguments (lambda-list) (mapcar #'car (lambda-list-optional-arguments-with-default lambda-list))) (defun lambda-list-rest-argument (lambda-list) - (let ((rest (list-until-keyword (cdr (member '&rest lambda-list))))) + (let ((rest (lambda-list-section '&rest lambda-list))) (when (cdr rest) (error "Bad lambda-list")) (car rest))) +(defun lambda-list-keyword-arguments-canonical (lambda-list) + (flet ((canonalize (keyarg) + ;; Build a canonical keyword argument descriptor, filling + ;; the optional fields. The result is a list of the form + ;; ((keyword-name var) init-form). + (let* ((arg (ensure-list keyarg)) + (init-form (cadr arg)) + var + keyword-name) + (if (listp (car arg)) + (setq var (cadr (car arg)) + keyword-name (car (car arg))) + (setq var (car arg) + keyword-name (intern (symbol-name (car arg)) "KEYWORD"))) + `((,keyword-name ,var) ,init-form)))) + (mapcar #'canonalize (lambda-list-section '&key lambda-list)))) + +(defun lambda-list-keyword-arguments (lambda-list) + (mapcar (lambda (keyarg) (second (first keyarg))) + (lambda-list-keyword-arguments-canonical lambda-list))) + (defun lambda-docstring-wrapper (docstring &rest strs) (if docstring (js!selfcall "var func = " (join strs) ";" *newline* "func.docstring = '" docstring "';" *newline* "return func;" *newline*) - (join strs))) + (apply #'code strs))) (defun lambda-check-argument-count (n-required-arguments n-optional-arguments rest-p) @@ -1202,20 +1359,107 @@ (block nil ;; Special case: a positive exact number of arguments. (when (and (< 1 min) (eql min max)) - (return (concat "checkArgs(arguments, " (integer-to-string min) ");" *newline*))) + (return (code "checkArgs(arguments, " min ");" *newline*))) ;; General case: - (concat - (if (< 1 min) - (concat "checkArgsAtLeast(arguments, " (integer-to-string min) ");" *newline*) - "") - (if (numberp max) - (concat "checkArgsAtMost(arguments, " (integer-to-string max) ");" *newline*) - ""))))) + (code + (when (< 1 min) + (code "checkArgsAtLeast(arguments, " min ");" *newline*)) + (when (numberp max) + (code "checkArgsAtMost(arguments, " max ");" *newline*)))))) + +(defun compile-lambda-optional (lambda-list) + (let* ((optional-arguments (lambda-list-optional-arguments lambda-list)) + (n-required-arguments (length (lambda-list-required-arguments lambda-list))) + (n-optional-arguments (length optional-arguments))) + (when optional-arguments + (code "switch(arguments.length-1){" *newline* + (let ((optional-and-defaults + (lambda-list-optional-arguments-with-default lambda-list)) + (cases nil) + (idx 0)) + (progn + (while (< idx n-optional-arguments) + (let ((arg (nth idx optional-and-defaults))) + (push (code "case " (+ idx n-required-arguments) ":" *newline* + (translate-variable (car arg)) + "=" + (ls-compile (cadr arg)) + ";" *newline*) + cases) + (incf idx))) + (push (code "default: break;" *newline*) cases) + (join (reverse cases)))) + "}" *newline*)))) + +(defun compile-lambda-rest (lambda-list) + (let ((n-required-arguments (length (lambda-list-required-arguments lambda-list))) + (n-optional-arguments (length (lambda-list-optional-arguments lambda-list))) + (rest-argument (lambda-list-rest-argument lambda-list))) + (when rest-argument + (let ((js!rest (translate-variable rest-argument))) + (code "var " js!rest "= " (ls-compile nil) ";" *newline* + "for (var i = arguments.length-1; i>=" + (+ 1 n-required-arguments n-optional-arguments) + "; i--)" *newline* + (indent js!rest " = {car: arguments[i], cdr: ") js!rest "};" + *newline*))))) + +(defun compile-lambda-parse-keywords (lambda-list) + (let ((n-required-arguments + (length (lambda-list-required-arguments lambda-list))) + (n-optional-arguments + (length (lambda-list-optional-arguments lambda-list))) + (keyword-arguments + (lambda-list-keyword-arguments-canonical lambda-list))) + (code + ;; Declare variables + (mapconcat (lambda (arg) + (let ((var (second (car arg)))) + (code "var " (translate-variable var) "; " *newline*))) + keyword-arguments) + ;; Parse keywords + (flet ((parse-keyword (keyarg) + ;; ((keyword-name var) init-form) + (code "for (i=" (+ 1 n-required-arguments n-optional-arguments) + "; i=" - (integer-to-string (+ 1 n-required-arguments n-optional-arguments)) - "; i--)" *newline* - (indent js!rest " = " - "{car: arguments[i], cdr: ") js!rest "};" - *newline*)) - "") - ;; Body - (let ((*multiple-value-p* t)) (ls-compile-block body t))) + (or rest-argument keyword-arguments)) + (compile-lambda-optional lambda-list) + (compile-lambda-rest lambda-list) + (compile-lambda-parse-keywords lambda-list) + (let ((*multiple-value-p* t)) + (ls-compile-block body t))) "})")))) + (defun setq-pair (var val) (let ((b (lookup-in-lexenv var *environment* 'variable))) (if (and (eq (binding-type b) 'variable) (not (member 'special (binding-declarations b))) (not (member 'constant (binding-declarations b)))) - (concat (binding-value b) " = " (ls-compile val)) + (code (binding-value b) " = " (ls-compile val)) (ls-compile `(set ',var ,val))))) (define-compilation setq (&rest pairs) @@ -1299,15 +1515,14 @@ (concat (setq-pair (car pairs) (cadr pairs)) (if (null (cddr pairs)) "" ", "))) (setq pairs (cddr pairs))))) - (concat "(" result ")"))) + (code "(" result ")"))) ;;; FFI Variable accessors (define-compilation js-vref (var) var) (define-compilation js-vset (var val) - (concat "(" var " = " (ls-compile val) ")")) - + (code "(" var " = " (ls-compile val) ")")) ;;; Literals @@ -1331,37 +1546,42 @@ (defvar *literal-counter* 0) (defun genlit () - (concat "l" (integer-to-string (incf *literal-counter*)))) + (code "l" (incf *literal-counter*))) (defun literal (sexp &optional recursive) (cond ((integerp sexp) (integer-to-string sexp)) - ((stringp sexp) (concat "\"" (escape-string sexp) "\"")) + ((stringp sexp) (code "\"" (escape-string sexp) "\"")) ((symbolp sexp) (or (cdr (assoc sexp *literal-symbols*)) (let ((v (genlit)) - (s #+common-lisp (concat "{name: \"" (escape-string (symbol-name sexp)) "\"}") - #+ecmalisp - (let ((package (symbol-package sexp))) - (if (null package) - (concat "{name: \"" (escape-string (symbol-name sexp)) "\"}") - (ls-compile `(intern ,(symbol-name sexp) ,(package-name package))))))) + (s #+common-lisp + (let ((package (symbol-package sexp))) + (if (eq package (find-package "KEYWORD")) + (code "{name: \"" (escape-string (symbol-name sexp)) + "\", 'package': '" (package-name package) "'}") + (code "{name: \"" (escape-string (symbol-name sexp)) "\"}"))) + #+ecmalisp + (let ((package (symbol-package sexp))) + (if (null package) + (code "{name: \"" (escape-string (symbol-name sexp)) "\"}") + (ls-compile `(intern ,(symbol-name sexp) ,(package-name package))))))) (push (cons sexp v) *literal-symbols*) - (toplevel-compilation (concat "var " v " = " s)) + (toplevel-compilation (code "var " v " = " s)) v))) ((consp sexp) (let* ((head (butlast sexp)) (tail (last sexp)) - (c (concat "QIList(" - (join-trailing (mapcar (lambda (x) (literal x t)) head) ",") - (literal (car tail) t) - "," - (literal (cdr tail) t) - ")"))) + (c (code "QIList(" + (join-trailing (mapcar (lambda (x) (literal x t)) head) ",") + (literal (car tail) t) + "," + (literal (cdr tail) t) + ")"))) (if recursive c (let ((v (genlit))) - (toplevel-compilation (concat "var " v " = " c)) + (toplevel-compilation (code "var " v " = " c)) v)))) ((arrayp sexp) (let ((elements (vector-to-list sexp))) @@ -1369,7 +1589,7 @@ (if recursive c (let ((v (genlit))) - (toplevel-compilation (concat "var " v " = " c)) + (toplevel-compilation (code "var " v " = " c)) v))))))) (define-compilation quote (sexp) @@ -1387,7 +1607,52 @@ ((and (listp x) (eq (car x) 'lambda)) (compile-lambda (cadr x) (cddr x))) ((symbolp x) - (ls-compile `(symbol-function ',x))))) + (let ((b (lookup-in-lexenv x *environment* 'function))) + (if b + (binding-value b) + (ls-compile `(symbol-function ',x))))))) + + +(defun make-function-binding (fname) + (make-binding fname 'function (gvarname fname))) + +(defun compile-function-definition (list) + (compile-lambda (car list) (cdr list))) + +(defun translate-function (name) + (let ((b (lookup-in-lexenv name *environment* 'function))) + (binding-value b))) + +(define-compilation flet (definitions &rest body) + (let* ((fnames (mapcar #'car definitions)) + (fbody (mapcar #'cdr definitions)) + (cfuncs (mapcar #'compile-function-definition fbody)) + (*environment* + (extend-lexenv (mapcar #'make-function-binding fnames) + *environment* + 'function))) + (code "(function(" + (join (mapcar #'translate-function fnames) ",") + "){" *newline* + (let ((body (ls-compile-block body t))) + (indent body)) + "})(" (join cfuncs ",") ")"))) + +(define-compilation labels (definitions &rest body) + (let* ((fnames (mapcar #'car definitions)) + (*environment* + (extend-lexenv (mapcar #'make-function-binding fnames) + *environment* + 'function))) + (js!selfcall + (mapconcat (lambda (func) + (code "var " (translate-function (car func)) + " = " (compile-lambda (cadr func) (cddr func)) + ";" *newline*)) + definitions) + (ls-compile-block body t)))) + + (defvar *compiling-file* nil) (define-compilation eval-when-compile (&rest body) @@ -1417,15 +1682,15 @@ (defun let-binding-wrapper (bindings body) (when (null bindings) (return-from let-binding-wrapper body)) - (concat + (code "try {" *newline* (indent "var tmp;" *newline* (mapconcat (lambda (b) (let ((s (ls-compile `(quote ,(car b))))) - (concat "tmp = " s ".value;" *newline* - s ".value = " (cdr b) ";" *newline* - (cdr b) " = tmp;" *newline*))) + (code "tmp = " s ".value;" *newline* + s ".value = " (cdr b) ";" *newline* + (cdr b) " = tmp;" *newline*))) bindings) body *newline*) "}" *newline* @@ -1433,7 +1698,7 @@ (indent (mapconcat (lambda (b) (let ((s (ls-compile `(quote ,(car b))))) - (concat s ".value" " = " (cdr b) ";" *newline*))) + (code s ".value" " = " (cdr b) ";" *newline*))) bindings)) "}" *newline*)) @@ -1443,19 +1708,19 @@ (cvalues (mapcar #'ls-compile (mapcar #'second bindings))) (*environment* (extend-local-env (remove-if #'special-variable-p variables))) (dynamic-bindings)) - (concat "(function(" - (join (mapcar (lambda (x) - (if (special-variable-p x) - (let ((v (gvarname x))) - (push (cons x v) dynamic-bindings) - v) - (translate-variable x))) - variables) - ",") - "){" *newline* - (let ((body (ls-compile-block body t))) - (indent (let-binding-wrapper dynamic-bindings body))) - "})(" (join cvalues ",") ")"))) + (code "(function(" + (join (mapcar (lambda (x) + (if (special-variable-p x) + (let ((v (gvarname x))) + (push (cons x v) dynamic-bindings) + v) + (translate-variable x))) + variables) + ",") + "){" *newline* + (let ((body (ls-compile-block body t))) + (indent (let-binding-wrapper dynamic-bindings body))) + "})(" (join cvalues ",") ")"))) ;;; Return the code to initialize BINDING, and push it extending the @@ -1464,10 +1729,10 @@ (let ((var (first binding)) (value (second binding))) (if (special-variable-p var) - (concat (ls-compile `(setq ,var ,value)) ";" *newline*) + (code (ls-compile `(setq ,var ,value)) ";" *newline*) (let* ((v (gvarname var)) (b (make-binding var 'variable v))) - (prog1 (concat "var " v " = " (ls-compile value) ";" *newline*) + (prog1 (code "var " v " = " (ls-compile value) ";" *newline*) (push-to-lexenv b *environment* 'variable)))))) ;;; Wrap BODY to restore the symbol values of SYMBOLS after body. It @@ -1478,12 +1743,12 @@ (return-from let*-binding-wrapper body)) (let ((store (mapcar (lambda (s) (cons s (gvarname s))) (remove-if-not #'special-variable-p symbols)))) - (concat + (code "try {" *newline* (indent (mapconcat (lambda (b) (let ((s (ls-compile `(quote ,(car b))))) - (concat "var " (cdr b) " = " s ".value;" *newline*))) + (code "var " (cdr b) " = " s ".value;" *newline*))) store) body) "}" *newline* @@ -1491,7 +1756,7 @@ (indent (mapconcat (lambda (b) (let ((s (ls-compile `(quote ,(car b))))) - (concat s ".value" " = " (cdr b) ";" *newline*))) + (code s ".value" " = " (cdr b) ";" *newline*))) store)) "}" *newline*))) @@ -1508,7 +1773,7 @@ (defvar *block-counter* 0) (define-compilation block (name &rest body) - (let* ((tr (integer-to-string (incf *block-counter*))) + (let* ((tr (incf *block-counter*)) (b (make-binding name 'block tr))) (when *multiple-value-p* (push-binding-declaration 'multiple-value b)) @@ -1537,9 +1802,7 @@ (error (concat "Unknown block `" (symbol-name name) "'."))) (push-binding-declaration 'used b) (js!selfcall - (if multiple-value-p - (concat "var values = mv;" *newline*) - "") + (when multiple-value-p (code "var values = mv;" *newline*)) "throw ({" "type: 'block', " "id: " (binding-value b) ", " @@ -1598,7 +1861,7 @@ (unless (go-tag-p (car body)) (push (gensym "START") body)) ;; Tagbody compilation - (let ((tbidx (integer-to-string *tagbody-counter*))) + (let ((tbidx *tagbody-counter*)) (let ((*environment* (declare-tagbody-tags tbidx body)) initag) (let ((b (lookup-in-lexenv (first body) *environment* 'gotag))) @@ -1609,17 +1872,17 @@ "while (true) {" *newline* (indent "try {" *newline* (indent (let ((content "")) - (concat "switch(tagbody_" tbidx "){" *newline* - "case " initag ":" *newline* - (dolist (form (cdr body) content) - (concatf content - (if (not (go-tag-p form)) - (indent (ls-compile form) ";" *newline*) - (let ((b (lookup-in-lexenv form *environment* 'gotag))) - (concat "case " (second (binding-value b)) ":" *newline*))))) - "default:" *newline* - " break tbloop;" *newline* - "}" *newline*))) + (code "switch(tagbody_" tbidx "){" *newline* + "case " initag ":" *newline* + (dolist (form (cdr body) content) + (concatf content + (if (not (go-tag-p form)) + (indent (ls-compile form) ";" *newline*) + (let ((b (lookup-in-lexenv form *environment* 'gotag))) + (code "case " (second (binding-value b)) ":" *newline*))))) + "default:" *newline* + " break tbloop;" *newline* + "}" *newline*))) "}" *newline* "catch (jump) {" *newline* " if (jump.type == 'tagbody' && jump.id == " tbidx ")" *newline* @@ -1635,15 +1898,15 @@ (n (cond ((symbolp label) (symbol-name label)) ((integerp label) (integer-to-string label))))) - (if b - (js!selfcall - "throw ({" - "type: 'tagbody', " - "id: " (first (binding-value b)) ", " - "label: " (second (binding-value b)) ", " - "message: 'Attempt to GO to non-existing tag " n "'" - "})" *newline*) - (error (concat "Unknown tag `" n "'."))))) + (when (null b) + (error (concat "Unknown tag `" n "'."))) + (js!selfcall + "throw ({" + "type: 'tagbody', " + "id: " (first (binding-value b)) ", " + "label: " (second (binding-value b)) ", " + "message: 'Attempt to GO to non-existing tag " n "'" + "})" *newline*))) (define-compilation unwind-protect (form &rest clean-up) (js!selfcall @@ -1664,11 +1927,11 @@ "var values = mv;" *newline* "var vs;" *newline* (mapconcat (lambda (form) - (concat "vs = " (ls-compile form t) ";" *newline* - "if (typeof vs === 'object' && 'multiple-value' in vs)" *newline* - (indent "args = args.concat(vs);" *newline*) - "else" *newline* - (indent "args.push(vs);" *newline*))) + (code "vs = " (ls-compile form t) ";" *newline* + "if (typeof vs === 'object' && 'multiple-value' in vs)" *newline* + (indent "args = args.concat(vs);" *newline*) + "else" *newline* + (indent "args.push(vs);" *newline*))) forms) "return func.apply(window, args);" *newline*) ";" *newline*)) @@ -1736,18 +1999,18 @@ (defmacro type-check (decls &body body) `(js!selfcall ,@(mapcar (lambda (decl) - `(concat "var " ,(first decl) " = " ,(third decl) ";" *newline*)) - decls) + `(code "var " ,(first decl) " = " ,(third decl) ";" *newline*)) + decls) ,@(mapcar (lambda (decl) - `(concat "if (typeof " ,(first decl) " != '" ,(second decl) "')" *newline* - (indent "throw 'The value ' + " - ,(first decl) - " + ' is not a type " - ,(second decl) - ".';" - *newline*))) + `(code "if (typeof " ,(first decl) " != '" ,(second decl) "')" *newline* + (indent "throw 'The value ' + " + ,(first decl) + " + ' is not a type " + ,(second decl) + ".';" + *newline*))) decls) - (concat "return " (progn ,@body) ";" *newline*))) + (code "return " (progn ,@body) ";" *newline*))) ;;; VARIABLE-ARITY compiles variable arity operations. ARGS stands for ;;; a variable which holds a list of forms. It will compile them and @@ -1759,16 +2022,18 @@ (unless (consp args) (error "ARGS must be a non-empty list")) (let ((counter 0) - (variables '()) + (fargs '()) (prelude "")) (dolist (x args) - (let ((v (concat "x" (integer-to-string (incf counter))))) - (push v variables) - (concatf prelude - (concat "var " v " = " (ls-compile x) ";" *newline* - "if (typeof " v " !== 'number') throw 'Not a number!';" - *newline*)))) - (js!selfcall prelude (funcall function (reverse variables))))) + (if (numberp x) + (push (integer-to-string x) fargs) + (let ((v (code "x" (incf counter)))) + (push v fargs) + (concatf prelude + (code "var " v " = " (ls-compile x) ";" *newline* + "if (typeof " v " !== 'number') throw 'Not a number!';" + *newline*))))) + (js!selfcall prelude (funcall function (reverse fargs))))) (defmacro variable-arity (args &body body) @@ -1776,11 +2041,11 @@ (error "Bad usage of VARIABLE-ARITY, you must pass a symbol")) `(variable-arity-call ,args (lambda (,args) - (concat "return " ,@body ";" *newline*)))) + (code "return " ,@body ";" *newline*)))) (defun num-op-num (x op y) (type-check (("x" "number" x) ("y" "number" y)) - (concat "x" op "y"))) + (code "x" op "y"))) (define-raw-builtin + (&rest numbers) (if (null numbers) @@ -1835,14 +2100,14 @@ (define-builtin-comparison = "==") (define-builtin numberp (x) - (js!bool (concat "(typeof (" x ") == \"number\")"))) + (js!bool (code "(typeof (" x ") == \"number\")"))) (define-builtin floor (x) (type-check (("x" "number" x)) "Math.floor(x)")) (define-builtin cons (x y) - (concat "({car: " x ", cdr: " y "})")) + (code "({car: " x ", cdr: " y "})")) (define-builtin consp (x) (js!bool @@ -1866,11 +2131,11 @@ (define-builtin rplaca (x new) (type-check (("x" "object" x)) - (concat "(x.car = " new ", x)"))) + (code "(x.car = " new ", x)"))) (define-builtin rplacd (x new) (type-check (("x" "object" x)) - (concat "(x.cdr = " new ", x)"))) + (code "(x.cdr = " new ", x)"))) (define-builtin symbolp (x) (js!bool @@ -1883,16 +2148,16 @@ "({name: name})")) (define-builtin symbol-name (x) - (concat "(" x ").name")) + (code "(" x ").name")) (define-builtin set (symbol value) - (concat "(" symbol ").value = " value)) + (code "(" symbol ").value = " value)) (define-builtin fset (symbol value) - (concat "(" symbol ").fvalue = " value)) + (code "(" symbol ").fvalue = " value)) (define-builtin boundp (x) - (js!bool (concat "(" x ".value !== undefined)"))) + (js!bool (code "(" x ".value !== undefined)"))) (define-builtin symbol-value (x) (js!selfcall @@ -1909,20 +2174,20 @@ "return func;" *newline*)) (define-builtin symbol-plist (x) - (concat "((" x ").plist || " (ls-compile nil) ")")) + (code "((" x ").plist || " (ls-compile nil) ")")) (define-builtin lambda-code (x) - (concat "(" x ").toString()")) + (code "(" x ").toString()")) -(define-builtin eq (x y) (js!bool (concat "(" x " === " y ")"))) -(define-builtin equal (x y) (js!bool (concat "(" x " == " y ")"))) +(define-builtin eq (x y) (js!bool (code "(" x " === " y ")"))) +(define-builtin equal (x y) (js!bool (code "(" x " == " y ")"))) (define-builtin char-to-string (x) (type-check (("x" "number" x)) "String.fromCharCode(x)")) (define-builtin stringp (x) - (js!bool (concat "(typeof(" x ") == \"string\")"))) + (js!bool (code "(typeof(" x ") == \"string\")"))) (define-builtin string-upcase (x) (type-check (("x" "string" x)) @@ -1937,9 +2202,7 @@ "var str = " (ls-compile string) ";" *newline* "var a = " (ls-compile a) ";" *newline* "var b;" *newline* - (if b - (concat "b = " (ls-compile b) ";" *newline*) - "") + (when b (code "b = " (ls-compile b) ";" *newline*)) "return str.slice(a,b);" *newline*)) (define-builtin char (string index) @@ -1953,15 +2216,15 @@ "string1.concat(string2)")) (define-raw-builtin funcall (func &rest args) - (concat "(" (ls-compile func) ")(" - (join (cons (if *multiple-value-p* "values" "pv") - (mapcar #'ls-compile args)) - ", ") - ")")) + (code "(" (ls-compile func) ")(" + (join (cons (if *multiple-value-p* "values" "pv") + (mapcar #'ls-compile args)) + ", ") + ")")) (define-raw-builtin apply (func &rest args) (if (null args) - (concat "(" (ls-compile func) ")()") + (code "(" (ls-compile func) ")()") (let ((args (butlast args)) (last (car (last args)))) (js!selfcall @@ -1995,7 +2258,7 @@ (define-builtin new () "{}") (define-builtin objectp (x) - (js!bool (concat "(typeof (" x ") === 'object')"))) + (js!bool (code "(typeof (" x ") === 'object')"))) (define-builtin oget (object key) (js!selfcall @@ -2003,13 +2266,13 @@ "return tmp == undefined? " (ls-compile nil) ": tmp ;" *newline*)) (define-builtin oset (object key value) - (concat "((" object ")[" key "] = " value ")")) + (code "((" object ")[" key "] = " value ")")) (define-builtin in (key object) - (js!bool (concat "((" key ") in (" object "))"))) + (js!bool (code "((" key ") in (" object "))"))) (define-builtin functionp (x) - (js!bool (concat "(typeof " x " == 'function')"))) + (js!bool (code "(typeof " x " == 'function')"))) (define-builtin write-string (x) (type-check (("x" "string" x)) @@ -2042,17 +2305,17 @@ "return x[i] = " value ";" *newline*)) (define-builtin get-unix-time () - (concat "(Math.round(new Date() / 1000))")) + (code "(Math.round(new Date() / 1000))")) (define-builtin values-array (array) (if *multiple-value-p* - (concat "values.apply(this, " array ")") - (concat "pv.apply(this, " array ")"))) + (code "values.apply(this, " array ")") + (code "pv.apply(this, " array ")"))) (define-raw-builtin values (&rest args) (if *multiple-value-p* - (concat "values(" (join (mapcar #'ls-compile args) ", ") ")") - (concat "pv(" (join (mapcar #'ls-compile args) ", ") ")"))) + (code "values(" (join (mapcar #'ls-compile args) ", ") ")") + (code "pv(" (join (mapcar #'ls-compile args) ", ") ")"))) (defun macro (x) (and (symbolp x) @@ -2079,23 +2342,22 @@ form))) (defun compile-funcall (function args) - (let ((values-funcs (if *multiple-value-p* "values" "pv"))) - (if (and (symbolp function) - #+ecmalisp (eq (symbol-package function) (find-package "COMMON-LISP")) - #+common-lisp t) - (concat (ls-compile `',function) ".fvalue(" - (join (cons values-funcs (mapcar #'ls-compile args)) - ", ") - ")") - (concat (ls-compile `#',function) "(" - (join (cons values-funcs (mapcar #'ls-compile args)) - ", ") - ")")))) + (let* ((values-funcs (if *multiple-value-p* "values" "pv")) + (arglist (concat "(" (join (cons values-funcs (mapcar #'ls-compile args)) ", ") ")"))) + (cond + ((translate-function function) + (concat (translate-function function) arglist)) + ((and (symbolp function) + #+ecmalisp (eq (symbol-package function) (find-package "COMMON-LISP")) + #+common-lisp t) + (code (ls-compile `',function) ".fvalue" arglist)) + (t + (code (ls-compile `#',function) arglist))))) (defun ls-compile-block (sexps &optional return-last-p) (if return-last-p - (concat (ls-compile-block (butlast sexps)) - "return " (ls-compile (car (last sexps)) *multiple-value-p*) ";") + (code (ls-compile-block (butlast sexps)) + "return " (ls-compile (car (last sexps)) *multiple-value-p*) ";") (join-trailing (remove-if #'null-or-empty-p (mapcar #'ls-compile sexps)) (concat ";" *newline*)))) @@ -2110,11 +2372,11 @@ (binding-value b)) ((or (keywordp sexp) (member 'constant (binding-declarations b))) - (concat (ls-compile `',sexp) ".value")) + (code (ls-compile `',sexp) ".value")) (t (ls-compile `(symbol-value ',sexp)))))) ((integerp sexp) (integer-to-string sexp)) - ((stringp sexp) (concat "\"" (escape-string sexp) "\"")) + ((stringp sexp) (code "\"" (escape-string sexp) "\"")) ((arrayp sexp) (literal sexp)) ((listp sexp) (let ((name (car sexp)) @@ -2146,11 +2408,10 @@ (join (remove-if #'null-or-empty-p subs)))) (t (let ((code (ls-compile sexp multiple-value-p))) - (concat (join-trailing (get-toplevel-compilations) - (concat ";" *newline*)) - (if code - (concat code ";" *newline*) - ""))))))) + (code (join-trailing (get-toplevel-compilations) + (code ";" *newline*)) + (when code + (code code ";" *newline*)))))))) ;;; Once we have the compiler, we define the runtime environment and @@ -2162,26 +2423,31 @@ (defun eval (x) (js-eval (ls-compile-toplevel x t))) - (export '(&rest &optional &body * *gensym-counter* *package* + - / 1+ 1- < <= = - = > >= and append apply aref arrayp aset assoc atom block boundp - boundp butlast caar cadddr caddr cadr car car case catch cdar cdddr - cddr cdr cdr char char-code char= code-char cond cons consp copy-list - decf declaim defparameter defun defmacro defvar digit-char-p - disassemble do do* documentation dolist dotimes ecase eq eql equal - error eval every export fdefinition find-package find-symbol first - fourth fset funcall function functionp gensym get-universal-time go - identity if in-package incf integerp integerp intern keywordp lambda - last length let let* list-all-packages list listp make-array - make-package make-symbol mapcar member minusp mod multiple-value-bind - multiple-value-call multiple-value-list multiple-value-prog1 nil not - nth nthcdr null numberp or package-name package-use-list packagep - parse-integer plusp prin1-to-string print proclaim prog1 prog2 progn - psetq push quote remove remove-if remove-if-not return return-from - revappend reverse rplaca rplacd second set setq some string-upcase - string string= stringp subseq symbol-function symbol-name symbol-package - symbol-plist symbol-value symbolp t tagbody third throw truncate - unless unwind-protect values values-list variable warn when write-line - write-string zerop)) + (export '(&rest &key &optional &body * *gensym-counter* *package* + + - / 1+ 1- < <= = = > >= and append apply aref arrayp assoc + atom block boundp boundp butlast caar cadddr caddr cadr + car car case catch cdar cdddr cddr cdr cdr char char-code + char= code-char cond cons consp constantly copy-list decf + declaim defconstant defparameter defun defmacro defvar + digit-char digit-char-p disassemble do do* documentation + dolist dotimes ecase eq eql equal error eval every export + fdefinition find-package find-symbol first flet fourth + fset funcall function functionp gensym get-universal-time + go identity if in-package incf integerp integerp intern + keywordp labels lambda last length let let* + list-all-packages list listp make-array make-package + make-symbol mapcar member minusp mod multiple-value-bind + multiple-value-call multiple-value-list + multiple-value-prog1 nil not nth nthcdr null numberp or + package-name package-use-list packagep parse-integer plusp + prin1-to-string print proclaim prog1 prog2 progn psetq + push quote remove remove-if remove-if-not return + return-from revappend reverse rplaca rplacd second set setf + setq some string-upcase string string= stringp subseq + symbol-function symbol-name symbol-package symbol-plist + symbol-value symbolp t tagbody third throw truncate unless + unwind-protect values values-list variable warn when + write-line write-string zerop)) (setq *package* *user-package*) @@ -2198,12 +2464,15 @@ ;; environment at this point of the compilation. (eval-when-compile (toplevel-compilation + (ls-compile `(setq *environment* ',*environment*)))) + + (eval-when-compile + (toplevel-compilation (ls-compile `(progn ,@(mapcar (lambda (s) `(%intern-symbol (js-vref ,(cdr s)))) *literal-symbols*) (setq *literal-symbols* ',*literal-symbols*) - (setq *environment* ',*environment*) (setq *variable-counter* ,*variable-counter*) (setq *gensym-counter* ,*gensym-counter*) (setq *block-counter* ,*block-counter*)))))