Update copyright notices for 2013.
[bpt/emacs.git] / lisp / emacs-lisp / cconv.el
CommitLineData
39605a34 1;;; cconv.el --- Closure conversion for statically scoped Emacs lisp. -*- lexical-binding: t; coding: utf-8 -*-
94d11cb5 2
ab422c4d 3;; Copyright (C) 2011-2013 Free Software Foundation, Inc.
d779e73c
SM
4
5;; Author: Igor Kuzmin <kzuminig@iro.umontreal.ca>
6;; Maintainer: FSF
7;; Keywords: lisp
8;; Package: emacs
9
10;; This file is part of GNU Emacs.
11
12;; GNU Emacs is free software: you can redistribute it and/or modify
13;; it under the terms of the GNU General Public License as published by
14;; the Free Software Foundation, either version 3 of the License, or
15;; (at your option) any later version.
16
17;; GNU Emacs is distributed in the hope that it will be useful,
18;; but WITHOUT ANY WARRANTY; without even the implied warranty of
19;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20;; GNU General Public License for more details.
21
22;; You should have received a copy of the GNU General Public License
23;; along with GNU Emacs. If not, see <http://www.gnu.org/licenses/>.
94d11cb5
IK
24
25;;; Commentary:
26
27;; This takes a piece of Elisp code, and eliminates all free variables from
28;; lambda expressions. The user entry points are cconv-closure-convert and
c7015153 29;; cconv-closure-convert-toplevel (for toplevel forms).
d779e73c
SM
30;; All macros should be expanded beforehand.
31;;
32;; Here is a brief explanation how this code works.
c7015153 33;; Firstly, we analyze the tree by calling cconv-analyse-form.
d779e73c 34;; This function finds all mutated variables, all functions that are suitable
94d11cb5
IK
35;; for lambda lifting and all variables captured by closure. It passes the tree
36;; once, returning a list of three lists.
d779e73c 37;;
c7015153 38;; Then we calculate the intersection of the first and third lists returned by
d779e73c
SM
39;; cconv-analyse form to find all mutated variables that are captured by
40;; closure.
94d11cb5 41
d779e73c 42;; Armed with this data, we call cconv-closure-convert-rec, that rewrites the
c7015153 43;; tree recursively, lifting lambdas where possible, building closures where it
94d11cb5
IK
44;; is needed and eliminating mutable variables used in closure.
45;;
46;; We do following replacements :
47;; (lambda (v1 ...) ... fv1 fv2 ...) => (lambda (v1 ... fv1 fv2 ) ... fv1 fv2 .)
48;; if the function is suitable for lambda lifting (if all calls are known)
49;;
876c194c
SM
50;; (lambda (v0 ...) ... fv0 .. fv1 ...) =>
51;; (internal-make-closure (v0 ...) (fv1 ...)
52;; ... (internal-get-closed-var 0) ... (internal-get-closed-var 1) ...)
94d11cb5
IK
53;;
54;; If the function has no free variables, we don't do anything.
94d11cb5 55;;
d779e73c 56;; If a variable is mutated (updated by setq), and it is used in a closure
876c194c 57;; we wrap its definition with list: (list val) and we also replace
d779e73c
SM
58;; var => (car var) wherever this variable is used, and also
59;; (setq var value) => (setcar var value) where it is updated.
94d11cb5 60;;
d779e73c
SM
61;; If defun argument is closure mutable, we letbind it and wrap it's
62;; definition with list.
63;; (defun foo (... mutable-arg ...) ...) =>
64;; (defun foo (... m-arg ...) (let ((m-arg (list m-arg))) ...))
94d11cb5
IK
65;;
66;;; Code:
67
ca105506 68;; TODO: (not just for cconv but also for the lexbind changes in general)
ca105506 69;; - let (e)debug find the value of lexical variables from the stack.
e4769531 70;; - make eval-region do the eval-sexp-add-defvars dance.
e2abe5a1 71;; - byte-optimize-form should be applied before cconv.
ca105506
SM
72;; OTOH, the warnings emitted by cconv-analyze need to come before optimize
73;; since afterwards they can because obnoxious (warnings about an "unused
74;; variable" should not be emitted when the variable use has simply been
75;; optimized away).
7200d79c
SM
76;; - let macros specify that some let-bindings come from the same source,
77;; so the unused warning takes all uses into account.
78;; - let interactive specs return a function to build the args (to stash into
79;; command-history).
cb9336bd
SM
80;; - canonize code in macro-expand so we don't have to handle (let (var) body)
81;; and other oddities.
3e21b6a7
SM
82;; - new byte codes for unwind-protect, catch, and condition-case so that
83;; closures aren't needed at all.
7200d79c 84;; - inline source code of different binding mode by first compiling it.
d032d5e7
SM
85;; - a reference to a var that is known statically to always hold a constant
86;; should be turned into a byte-constant rather than a byte-stack-ref.
e2abe5a1
SM
87;; Hmm... right, that's called constant propagation and could be done here,
88;; but when that constant is a function, we have to be careful to make sure
d032d5e7
SM
89;; the bytecomp only compiles it once.
90;; - Since we know here when a variable is not mutated, we could pass that
91;; info to the byte-compiler, e.g. by using a new `immutable-let'.
e2abe5a1 92;; - add tail-calls to bytecode.c and the byte compiler.
29a4dcb0 93;; - call known non-escaping functions with `goto' rather than `call'.
6c075cd7 94;; - optimize mapcar to a while loop.
d032d5e7
SM
95
96;; (defmacro dlet (binders &rest body)
97;; ;; Works in both lexical and non-lexical mode.
98;; `(progn
99;; ,@(mapcar (lambda (binder)
100;; `(defvar ,(if (consp binder) (car binder) binder)))
101;; binders)
102;; (let ,binders ,@body)))
103
104;; (defmacro llet (binders &rest body)
105;; ;; Only works in lexical-binding mode.
106;; `(funcall
107;; (lambda ,(mapcar (lambda (binder) (if (consp binder) (car binder) binder))
108;; binders)
109;; ,@body)
110;; ,@(mapcar (lambda (binder) (if (consp binder) (cadr binder)))
111;; binders)))
112
f80efb86 113(eval-when-compile (require 'cl-lib))
94d11cb5 114
d032d5e7 115(defconst cconv-liftwhen 6
d779e73c 116 "Try to do lambda lifting if the number of arguments + free variables
94d11cb5 117is less than this number.")
a9de04fa
SM
118;; List of all the variables that are both captured by a closure
119;; and mutated. Each entry in the list takes the form
120;; (BINDER . PARENTFORM) where BINDER is the (VAR VAL) that introduces the
121;; variable (or is just (VAR) for variables not introduced by let).
122(defvar cconv-captured+mutated)
d779e73c 123
a9de04fa
SM
124;; List of candidates for lambda lifting.
125;; Each candidate has the form (BINDER . PARENTFORM). A candidate
126;; is a variable that is only passed to `funcall' or `apply'.
127(defvar cconv-lambda-candidates)
d779e73c 128
a9de04fa
SM
129;; Alist associating to each function body the list of its free variables.
130(defvar cconv-freevars-alist)
94d11cb5
IK
131
132;;;###autoload
295fb2ac
SM
133(defun cconv-closure-convert (form)
134 "Main entry point for closure conversion.
94d11cb5
IK
135-- FORM is a piece of Elisp code after macroexpansion.
136-- TOPLEVEL(optional) is a boolean variable, true if we are at the root of AST
137
138Returns a form where all lambdas don't have any free variables."
b38b1ec0 139 ;; (message "Entering cconv-closure-convert...")
a9de04fa 140 (let ((cconv-freevars-alist '())
94d11cb5 141 (cconv-lambda-candidates '())
d779e73c 142 (cconv-captured+mutated '()))
c7015153 143 ;; Analyze form - fill these variables with new information.
a9de04fa
SM
144 (cconv-analyse-form form '())
145 (setq cconv-freevars-alist (nreverse cconv-freevars-alist))
6c075cd7 146 (cconv-convert form nil nil))) ; Env initially empty.
94d11cb5 147
295fb2ac 148(defconst cconv--dummy-var (make-symbol "ignored"))
b38b1ec0
SM
149
150(defun cconv--set-diff (s1 s2)
151 "Return elements of set S1 that are not in set S2."
152 (let ((res '()))
153 (dolist (x s1)
154 (unless (memq x s2) (push x res)))
155 (nreverse res)))
156
157(defun cconv--set-diff-map (s m)
158 "Return elements of set S that are not in Dom(M)."
159 (let ((res '()))
160 (dolist (x s)
161 (unless (assq x m) (push x res)))
162 (nreverse res)))
163
164(defun cconv--map-diff (m1 m2)
165 "Return the submap of map M1 that has Dom(M2) removed."
166 (let ((res '()))
167 (dolist (x m1)
168 (unless (assq (car x) m2) (push x res)))
169 (nreverse res)))
170
171(defun cconv--map-diff-elem (m x)
172 "Return the map M minus any mapping for X."
173 ;; Here we assume that X appears at most once in M.
174 (let* ((b (assq x m))
175 (res (if b (remq b m) m)))
f80efb86 176 (cl-assert (null (assq x res))) ;; Check the assumption was warranted.
b38b1ec0 177 res))
94d11cb5 178
b38b1ec0
SM
179(defun cconv--map-diff-set (m s)
180 "Return the map M minus any mapping for elements of S."
181 ;; Here we assume that X appears at most once in M.
182 (let ((res '()))
183 (dolist (b m)
184 (unless (memq (car b) s) (push b res)))
185 (nreverse res)))
186
6c075cd7 187(defun cconv--convert-function (args body env parentform)
f80efb86 188 (cl-assert (equal body (caar cconv-freevars-alist)))
6c075cd7
SM
189 (let* ((fvs (cdr (pop cconv-freevars-alist)))
190 (body-new '())
a9de04fa 191 (letbind '())
6c075cd7
SM
192 (envector ())
193 (i 0)
194 (new-env ()))
195 ;; Build the "formal and actual envs" for the closure-converted function.
196 (dolist (fv fvs)
197 (let ((exp (or (cdr (assq fv env)) fv)))
198 (pcase exp
199 ;; If `fv' is a variable that's wrapped in a cons-cell,
200 ;; we want to put the cons-cell itself in the closure,
201 ;; rather than just a copy of its current content.
202 (`(car ,iexp . ,_)
203 (push iexp envector)
204 (push `(,fv . (car (internal-get-closed-var ,i))) new-env))
205 (_
206 (push exp envector)
207 (push `(,fv . (internal-get-closed-var ,i)) new-env))))
208 (setq i (1+ i)))
209 (setq envector (nreverse envector))
210 (setq new-env (nreverse new-env))
211
212 (dolist (arg args)
213 (if (not (member (cons (list arg) parentform) cconv-captured+mutated))
214 (if (assq arg new-env) (push `(,arg) new-env))
215 (push `(,arg . (car ,arg)) new-env)
216 (push `(,arg (list ,arg)) letbind)))
ca105506 217
6c075cd7
SM
218 (setq body-new (mapcar (lambda (form)
219 (cconv-convert form new-env nil))
220 body))
221
222 (when letbind
223 (let ((special-forms '()))
224 ;; Keep special forms at the beginning of the body.
225 (while (or (stringp (car body-new)) ;docstring.
226 (memq (car-safe (car body-new)) '(interactive declare)))
227 (push (pop body-new) special-forms))
228 (setq body-new
229 `(,@(nreverse special-forms) (let ,letbind . ,body-new)))))
a9de04fa
SM
230
231 (cond
6c075cd7
SM
232 ((null envector) ;if no freevars - do nothing
233 `(function (lambda ,args . ,body-new)))
a9de04fa
SM
234 (t
235 `(internal-make-closure
6c075cd7 236 ,args ,envector . ,body-new)))))
a9de04fa 237
6c075cd7 238(defun cconv-convert (form env extend)
d779e73c 239 ;; This function actually rewrites the tree.
6c075cd7
SM
240 "Return FORM with all its lambdas changed so they are closed.
241ENV is a lexical environment mapping variables to the expression
242used to get its value. This is used for variables that are copied into
243closures, moved into cons cells, ...
244ENV is a list where each entry takes the shape either:
245 (VAR . (car EXP)): VAR has been moved into the car of a cons-cell, and EXP
246 is an expression that evaluates to this cons-cell.
247 (VAR . (internal-get-closed-var N)): VAR has been copied into the closure
248 environment's Nth slot.
249 (VAR . (apply-partially F ARG1 ARG2 ..)): VAR has been λ-lifted and takes
250 additional arguments ARGs.
251EXTEND is a list of variables which might need to be accessed even from places
252where they are shadowed, because some part of ENV causes them to be used at
253places where they originally did not directly appear."
f80efb86
SM
254 (cl-assert (not (delq nil (mapcar (lambda (mapping)
255 (if (eq (cadr mapping) 'apply-partially)
256 (cconv--set-diff (cdr (cddr mapping))
257 extend)))
258 env))))
ca105506 259
d779e73c 260 ;; What's the difference between fvrs and envs?
94d11cb5
IK
261 ;; Suppose that we have the code
262 ;; (lambda (..) fvr (let ((fvr 1)) (+ fvr 1)))
d779e73c
SM
263 ;; only the first occurrence of fvr should be replaced by
264 ;; (aref env ...).
94d11cb5
IK
265 ;; So initially envs and fvrs are the same thing, but when we descend to
266 ;; the 'let, we delete fvr from fvrs. Why we don't delete fvr from envs?
267 ;; Because in envs the order of variables is important. We use this list
d779e73c
SM
268 ;; to find the number of a specific variable in the environment vector,
269 ;; so we never touch it(unless we enter to the other closure).
270 ;;(if (listp form) (print (car form)) form)
271 (pcase form
6c075cd7 272 (`(,(and letsym (or `let* `let)) ,binders . ,body)
94d11cb5
IK
273
274 ; let and let* special forms
6c075cd7
SM
275 (let ((binders-new '())
276 (new-env env)
277 (new-extend extend))
d779e73c 278
295fb2ac
SM
279 (dolist (binder binders)
280 (let* ((value nil)
281 (var (if (not (consp binder))
a9de04fa 282 (prog1 binder (setq binder (list binder)))
295fb2ac
SM
283 (setq value (cadr binder))
284 (car binder)))
285 (new-val
286 (cond
287 ;; Check if var is a candidate for lambda lifting.
6c075cd7
SM
288 ((and (member (cons binder form) cconv-lambda-candidates)
289 (progn
f80efb86
SM
290 (cl-assert (and (eq (car value) 'function)
291 (eq (car (cadr value)) 'lambda)))
292 (cl-assert (equal (cddr (cadr value))
293 (caar cconv-freevars-alist)))
6c075cd7
SM
294 ;; Peek at the freevars to decide whether to λ-lift.
295 (let* ((fvs (cdr (car cconv-freevars-alist)))
296 (fun (cadr value))
297 (funargs (cadr fun))
298 (funcvars (append fvs funargs)))
94d11cb5 299 ; lambda lifting condition
6c075cd7
SM
300 (and fvs (>= cconv-liftwhen (length funcvars))))))
301 ; Lift.
302 (let* ((fvs (cdr (pop cconv-freevars-alist)))
303 (fun (cadr value))
304 (funargs (cadr fun))
305 (funcvars (append fvs funargs))
306 (funcbody (cddr fun))
307 (funcbody-env ()))
308 (push `(,var . (apply-partially ,var . ,fvs)) new-env)
309 (dolist (fv fvs)
f80efb86 310 (cl-pushnew fv new-extend)
6c075cd7
SM
311 (if (and (eq 'car (car-safe (cdr (assq fv env))))
312 (not (memq fv funargs)))
313 (push `(,fv . (car ,fv)) funcbody-env)))
314 `(function (lambda ,funcvars .
315 ,(mapcar (lambda (form)
316 (cconv-convert
317 form funcbody-env nil))
318 funcbody)))))
295fb2ac
SM
319
320 ;; Check if it needs to be turned into a "ref-cell".
a9de04fa 321 ((member (cons binder form) cconv-captured+mutated)
295fb2ac 322 ;; Declared variable is mutated and captured.
6c075cd7
SM
323 (push `(,var . (car ,var)) new-env)
324 `(list ,(cconv-convert value env extend)))
295fb2ac
SM
325
326 ;; Normal default case.
327 (t
6c075cd7
SM
328 (if (assq var new-env) (push `(,var) new-env))
329 (cconv-convert value env extend)))))
330
331 ;; The piece of code below letbinds free variables of a λ-lifted
332 ;; function if they are redefined in this let, example:
333 ;; (let* ((fun (lambda (x) (+ x y))) (y 1)) (funcall fun 1))
334 ;; Here we can not pass y as parameter because it is redefined.
335 ;; So we add a (closed-y y) declaration. We do that even if the
336 ;; function is not used inside this let(*). The reason why we
337 ;; ignore this case is that we can't "look forward" to see if the
338 ;; function is called there or not. To treat this case better we'd
339 ;; need to traverse the tree one more time to collect this data, and
340 ;; I think that it's not worth it.
341 (when (memq var new-extend)
342 (let ((closedsym
343 (make-symbol (concat "closed-" (symbol-name var)))))
344 (setq new-env
345 (mapcar (lambda (mapping)
346 (if (not (eq (cadr mapping) 'apply-partially))
347 mapping
f80efb86 348 (cl-assert (eq (car mapping) (nth 2 mapping)))
2ee3d7f0
SM
349 `(,(car mapping)
350 apply-partially
351 ,(car mapping)
352 ,@(mapcar (lambda (arg)
353 (if (eq var arg)
354 closedsym arg))
355 (nthcdr 3 mapping)))))
6c075cd7
SM
356 new-env))
357 (setq new-extend (remq var new-extend))
358 (push closedsym new-extend)
359 (push `(,closedsym ,var) binders-new)))
d779e73c 360
b38b1ec0
SM
361 ;; We push the element after redefined free variables are
362 ;; processed. This is important to avoid the bug when free
363 ;; variable and the function have the same name.
295fb2ac 364 (push (list var new-val) binders-new)
d779e73c 365
6c075cd7
SM
366 (when (eq letsym 'let*)
367 (setq env new-env)
368 (setq extend new-extend))
369 )) ; end of dolist over binders
370
371 `(,letsym ,(nreverse binders-new)
372 . ,(mapcar (lambda (form)
373 (cconv-convert
374 form new-env new-extend))
375 body))))
94d11cb5
IK
376 ;end of let let* forms
377
d779e73c 378 ; first element is lambda expression
6c075cd7
SM
379 (`(,(and `(lambda . ,_) fun) . ,args)
380 ;; FIXME: it's silly to create a closure just to call it.
ca105506 381 ;; Running byte-optimize-form earlier will resolve this.
6c075cd7
SM
382 `(funcall
383 ,(cconv-convert `(function ,fun) env extend)
384 ,@(mapcar (lambda (form)
385 (cconv-convert form env extend))
386 args)))
d779e73c
SM
387
388 (`(cond . ,cond-forms) ; cond special form
6c075cd7
SM
389 `(cond . ,(mapcar (lambda (branch)
390 (mapcar (lambda (form)
391 (cconv-convert form env extend))
392 branch))
393 cond-forms)))
d779e73c 394
6c075cd7
SM
395 (`(function (lambda ,args . ,body) . ,_)
396 (cconv--convert-function args body env form))
d779e73c 397
876c194c 398 (`(internal-make-closure . ,_)
6c075cd7
SM
399 (byte-compile-report-error
400 "Internal error in compiler: cconv called twice?"))
876c194c 401
6c075cd7
SM
402 (`(quote . ,_) form)
403 (`(function . ,_) form)
94d11cb5
IK
404
405 ;defconst, defvar
6c075cd7
SM
406 (`(,(and sym (or `defconst `defvar)) ,definedsymbol . ,forms)
407 `(,sym ,definedsymbol
408 . ,(mapcar (lambda (form) (cconv-convert form env extend))
409 forms)))
d779e73c 410
94d11cb5 411 ;condition-case
295fb2ac 412 (`(condition-case ,var ,protected-form . ,handlers)
6c075cd7
SM
413 (let ((newform (cconv--convert-function
414 () (list protected-form) env form)))
295fb2ac 415 `(condition-case :fun-body ,newform
876c194c
SM
416 ,@(mapcar (lambda (handler)
417 (list (car handler)
6c075cd7
SM
418 (cconv--convert-function
419 (list (or var cconv--dummy-var))
420 (cdr handler) env form)))
876c194c 421 handlers))))
295fb2ac
SM
422
423 (`(,(and head (or `catch `unwind-protect)) ,form . ,body)
6c075cd7
SM
424 `(,head ,(cconv-convert form env extend)
425 :fun-body ,(cconv--convert-function () body env form)))
295fb2ac 426
e0f57e65
SM
427 (`(track-mouse . ,body)
428 `(track-mouse
6c075cd7 429 :fun-body ,(cconv--convert-function () body env form)))
d779e73c
SM
430
431 (`(setq . ,forms) ; setq special form
6c075cd7 432 (let ((prognlist ()))
d779e73c 433 (while forms
6c075cd7
SM
434 (let* ((sym (pop forms))
435 (sym-new (or (cdr (assq sym env)) sym))
436 (value (cconv-convert (pop forms) env extend)))
437 (push (pcase sym-new
438 ((pred symbolp) `(setq ,sym-new ,value))
439 (`(car ,iexp) `(setcar ,iexp ,value))
440 ;; This "should never happen", but for variables which are
441 ;; mutated+captured+unused, we may end up trying to `setq'
442 ;; on a closed-over variable, so just drop the setq.
443 (_ ;; (byte-compile-report-error
444 ;; (format "Internal error in cconv of (setq %s ..)"
445 ;; sym-new))
446 value))
447 prognlist)))
d779e73c 448 (if (cdr prognlist)
6c075cd7 449 `(progn . ,(nreverse prognlist))
d779e73c
SM
450 (car prognlist))))
451
452 (`(,(and (or `funcall `apply) callsym) ,fun . ,args)
6c075cd7
SM
453 ;; These are not special forms but we treat them separately for the needs
454 ;; of lambda lifting.
455 (let ((mapping (cdr (assq fun env))))
456 (pcase mapping
457 (`(apply-partially ,_ . ,(and fvs `(,_ . ,_)))
f80efb86 458 (cl-assert (eq (cadr mapping) fun))
6c075cd7
SM
459 `(,callsym ,fun
460 ,@(mapcar (lambda (fv)
461 (let ((exp (or (cdr (assq fv env)) fv)))
462 (pcase exp
463 (`(car ,iexp . ,_) iexp)
464 (_ exp))))
465 fvs)
466 ,@(mapcar (lambda (arg)
467 (cconv-convert arg env extend))
468 args)))
469 (_ `(,callsym ,@(mapcar (lambda (arg)
470 (cconv-convert arg env extend))
471 (cons fun args)))))))
d779e73c 472
d032d5e7 473 (`(interactive . ,forms)
6c075cd7
SM
474 `(interactive . ,(mapcar (lambda (form)
475 (cconv-convert form nil nil))
476 forms)))
ca105506 477
ba83908c 478 (`(declare . ,_) form) ;The args don't contain code.
ca105506 479
6c075cd7
SM
480 (`(,func . ,forms)
481 ;; First element is function or whatever function-like forms are: or, and,
482 ;; if, progn, prog1, prog2, while, until
483 `(,func . ,(mapcar (lambda (form)
484 (cconv-convert form env extend))
485 forms)))
486
487 (_ (or (cdr (assq form env)) form))))
43e67019 488
a9de04fa
SM
489(unless (fboundp 'byte-compile-not-lexical-var-p)
490 ;; Only used to test the code in non-lexbind Emacs.
491 (defalias 'byte-compile-not-lexical-var-p 'boundp))
492
6c075cd7 493(defun cconv--analyse-use (vardata form varkind)
e9fce1ac 494 "Analyze the use of a variable.
d032d5e7
SM
495VARDATA should be (BINDER READ MUTATED CAPTURED CALLED).
496VARKIND is the name of the kind of variable.
497FORM is the parent form that binds this var."
a9de04fa
SM
498 ;; use = `(,binder ,read ,mutated ,captured ,called)
499 (pcase vardata
d032d5e7
SM
500 (`(,_ nil nil nil nil) nil)
501 (`((,(and (pred (lambda (var) (eq ?_ (aref (symbol-name var) 0)))) var) . ,_)
502 ,_ ,_ ,_ ,_)
6c075cd7
SM
503 (byte-compile-log-warning
504 (format "%s `%S' not left unused" varkind var))))
d032d5e7
SM
505 (pcase vardata
506 (`((,var . ,_) nil ,_ ,_ nil)
507 ;; FIXME: This gives warnings in the wrong order, with imprecise line
508 ;; numbers and without function name info.
509 (unless (or ;; Uninterned symbols typically come from macro-expansion, so
510 ;; it is often non-trivial for the programmer to avoid such
511 ;; unused vars.
512 (not (intern-soft var))
e67a13ab
CY
513 (eq ?_ (aref (symbol-name var) 0))
514 ;; As a special exception, ignore "ignore".
515 (eq var 'ignored))
d032d5e7
SM
516 (byte-compile-log-warning (format "Unused lexical %s `%S'"
517 varkind var))))
a9de04fa 518 ;; If it's unused, there's no point converting it into a cons-cell, even if
d032d5e7 519 ;; it's captured and mutated.
a9de04fa
SM
520 (`(,binder ,_ t t ,_)
521 (push (cons binder form) cconv-captured+mutated))
522 (`(,(and binder `(,_ (function (lambda . ,_)))) nil nil nil t)
6c075cd7 523 (push (cons binder form) cconv-lambda-candidates))))
a9de04fa 524
6c075cd7 525(defun cconv--analyse-function (args body env parentform)
a9de04fa
SM
526 (let* ((newvars nil)
527 (freevars (list body))
528 ;; We analyze the body within a new environment where all uses are
529 ;; nil, so we can distinguish uses within that function from uses
530 ;; outside of it.
531 (envcopy
532 (mapcar (lambda (vdata) (list (car vdata) nil nil nil nil)) env))
533 (newenv envcopy))
534 ;; Push it before recursing, so cconv-freevars-alist contains entries in
535 ;; the order they'll be used by closure-convert-rec.
536 (push freevars cconv-freevars-alist)
537 (dolist (arg args)
538 (cond
539 ((byte-compile-not-lexical-var-p arg)
06788a55 540 (byte-compile-log-warning
a9de04fa
SM
541 (format "Argument %S is not a lexical variable" arg)))
542 ((eq ?& (aref (symbol-name arg) 0)) nil) ;Ignore &rest, &optional, ...
543 (t (let ((varstruct (list arg nil nil nil nil)))
544 (push (cons (list arg) (cdr varstruct)) newvars)
545 (push varstruct newenv)))))
e9fce1ac 546 (dolist (form body) ;Analyze body forms.
a9de04fa
SM
547 (cconv-analyse-form form newenv))
548 ;; Summarize resulting data about arguments.
549 (dolist (vardata newvars)
6c075cd7 550 (cconv--analyse-use vardata parentform "argument"))
a9de04fa
SM
551 ;; Transfer uses collected in `envcopy' (via `newenv') back to `env';
552 ;; and compute free variables.
553 (while env
f80efb86 554 (cl-assert (and envcopy (eq (caar env) (caar envcopy))))
a9de04fa
SM
555 (let ((free nil)
556 (x (cdr (car env)))
557 (y (cdr (car envcopy))))
558 (while x
559 (when (car y) (setcar x t) (setq free t))
560 (setq x (cdr x) y (cdr y)))
561 (when free
2ee3d7f0
SM
562 (push (caar env) (cdr freevars))
563 (setf (nth 3 (car env)) t))
a9de04fa
SM
564 (setq env (cdr env) envcopy (cdr envcopy))))))
565
566(defun cconv-analyse-form (form env)
567 "Find mutated variables and variables captured by closure.
c7015153 568Analyze lambdas if they are suitable for lambda lifting.
d032d5e7
SM
569- FORM is a piece of Elisp code after macroexpansion.
570- ENV is an alist mapping each enclosing lexical variable to its info.
a9de04fa
SM
571 I.e. each element has the form (VAR . (READ MUTATED CAPTURED CALLED)).
572This function does not return anything but instead fills the
573`cconv-captured+mutated' and `cconv-lambda-candidates' variables
574and updates the data stored in ENV."
94d11cb5
IK
575 (pcase form
576 ; let special form
43e67019 577 (`(,(and (or `let* `let) letsym) ,binders . ,body-forms)
d779e73c 578
43e67019 579 (let ((orig-env env)
a9de04fa 580 (newvars nil)
d779e73c 581 (var nil)
43e67019
SM
582 (value nil))
583 (dolist (binder binders)
584 (if (not (consp binder))
d779e73c 585 (progn
43e67019 586 (setq var binder) ; treat the form (let (x) ...) well
a9de04fa 587 (setq binder (list binder))
43e67019
SM
588 (setq value nil))
589 (setq var (car binder))
590 (setq value (cadr binder))
591
a9de04fa 592 (cconv-analyse-form value (if (eq letsym 'let*) env orig-env)))
43e67019 593
ce5b520a 594 (unless (byte-compile-not-lexical-var-p var)
a9de04fa
SM
595 (let ((varstruct (list var nil nil nil nil)))
596 (push (cons binder (cdr varstruct)) newvars)
597 (push varstruct env))))
43e67019 598
e9fce1ac 599 (dolist (form body-forms) ; Analyze body forms.
a9de04fa 600 (cconv-analyse-form form env))
43e67019 601
a9de04fa 602 (dolist (vardata newvars)
6c075cd7 603 (cconv--analyse-use vardata form "variable"))))
43e67019 604
43e67019 605 (`(function (lambda ,vrs . ,body-forms))
6c075cd7 606 (cconv--analyse-function vrs body-forms env form))
ca105506 607
43e67019
SM
608 (`(setq . ,forms)
609 ;; If a local variable (member of env) is modified by setq then
610 ;; it is a mutated variable.
d779e73c 611 (while forms
43e67019 612 (let ((v (assq (car forms) env))) ; v = non nil if visible
2ee3d7f0 613 (when v (setf (nth 2 v) t)))
a9de04fa 614 (cconv-analyse-form (cadr forms) env)
43e67019
SM
615 (setq forms (cddr forms))))
616
0d42eb3e
SM
617 (`((lambda . ,_) . ,_) ; First element is lambda expression.
618 (byte-compile-log-warning
619 "Use of deprecated ((lambda ...) ...) form" t :warning)
d779e73c 620 (dolist (exp `((function ,(car form)) . ,(cdr form)))
a9de04fa 621 (cconv-analyse-form exp env)))
d779e73c
SM
622
623 (`(cond . ,cond-forms) ; cond special form
43e67019 624 (dolist (forms cond-forms)
d032d5e7 625 (dolist (form forms) (cconv-analyse-form form env))))
d779e73c
SM
626
627 (`(quote . ,_) nil) ; quote form
d779e73c
SM
628 (`(function . ,_) nil) ; same as quote
629
43e67019
SM
630 (`(condition-case ,var ,protected-form . ,handlers)
631 ;; FIXME: The bytecode for condition-case forces us to wrap the
ca105506
SM
632 ;; form and handlers in closures (for handlers, it's understandable
633 ;; but not for the protected form).
6c075cd7 634 (cconv--analyse-function () (list protected-form) env form)
43e67019 635 (dolist (handler handlers)
6c075cd7 636 (cconv--analyse-function (if var (list var)) (cdr handler) env form)))
43e67019
SM
637
638 ;; FIXME: The bytecode for catch forces us to wrap the body.
639 (`(,(or `catch `unwind-protect) ,form . ,body)
a9de04fa 640 (cconv-analyse-form form env)
6c075cd7 641 (cconv--analyse-function () body env form))
43e67019 642
ca105506
SM
643 ;; FIXME: The lack of bytecode for track-mouse forces us to wrap the body.
644 ;; `track-mouse' really should be made into a macro.
e0f57e65 645 (`(track-mouse . ,body)
6c075cd7 646 (cconv--analyse-function () body env form))
43e67019
SM
647
648 (`(,(or `defconst `defvar) ,var ,value . ,_)
649 (push var byte-compile-bound-variables)
a9de04fa 650 (cconv-analyse-form value env))
d779e73c
SM
651
652 (`(,(or `funcall `apply) ,fun . ,args)
43e67019
SM
653 ;; Here we ignore fun because funcall and apply are the only two
654 ;; functions where we can pass a candidate for lambda lifting as
655 ;; argument. So, if we see fun elsewhere, we'll delete it from
656 ;; lambda candidate list.
a9de04fa
SM
657 (let ((fdata (and (symbolp fun) (assq fun env))))
658 (if fdata
2ee3d7f0 659 (setf (nth 4 fdata) t)
a9de04fa 660 (cconv-analyse-form fun env)))
d032d5e7
SM
661 (dolist (form args) (cconv-analyse-form form env)))
662
663 (`(interactive . ,forms)
664 ;; These appear within the function body but they don't have access
665 ;; to the function's arguments.
666 ;; We could extend this to allow interactive specs to refer to
667 ;; variables in the function's enclosing environment, but it doesn't
668 ;; seem worth the trouble.
669 (dolist (form forms) (cconv-analyse-form form nil)))
ba83908c
SM
670
671 (`(declare . ,_) nil) ;The args don't contain code.
ca105506 672
43e67019 673 (`(,_ . ,body-forms) ; First element is a function or whatever.
d032d5e7 674 (dolist (form body-forms) (cconv-analyse-form form env)))
43e67019
SM
675
676 ((pred symbolp)
677 (let ((dv (assq form env))) ; dv = declared and visible
678 (when dv
2ee3d7f0 679 (setf (nth 1 dv) t))))))
94d11cb5
IK
680
681(provide 'cconv)
682;;; cconv.el ends here