0.6.11.40:
[sbcl.git] / src / compiler / ir1opt.lisp
1 ;;;; This file implements the IR1 optimization phase of the compiler.
2 ;;;; IR1 optimization is a grab-bag of optimizations that don't make
3 ;;;; major changes to the block-level control flow and don't use flow
4 ;;;; analysis. These optimizations can mostly be classified as
5 ;;;; "meta-evaluation", but there is a sizable top-down component as
6 ;;;; well.
7
8 ;;;; This software is part of the SBCL system. See the README file for
9 ;;;; more information.
10 ;;;;
11 ;;;; This software is derived from the CMU CL system, which was
12 ;;;; written at Carnegie Mellon University and released into the
13 ;;;; public domain. The software is in the public domain and is
14 ;;;; provided with absolutely no warranty. See the COPYING and CREDITS
15 ;;;; files for more information.
16
17 (in-package "SB!C")
18 \f
19 ;;;; interface for obtaining results of constant folding
20
21 ;;; Return true if the sole use of Cont is a reference to a constant leaf.
22 (declaim (ftype (function (continuation) boolean) constant-continuation-p))
23 (defun constant-continuation-p (cont)
24   (let ((use (continuation-use cont)))
25     (and (ref-p use)
26          (constant-p (ref-leaf use)))))
27
28 ;;; Return the constant value for a continuation whose only use is a
29 ;;; constant node.
30 (declaim (ftype (function (continuation) t) continuation-value))
31 (defun continuation-value (cont)
32   (aver (constant-continuation-p cont))
33   (constant-value (ref-leaf (continuation-use cont))))
34 \f
35 ;;;; interface for obtaining results of type inference
36
37 ;;; Return a (possibly values) type that describes what we have proven
38 ;;; about the type of Cont without taking any type assertions into
39 ;;; consideration. This is just the union of the NODE-DERIVED-TYPE of
40 ;;; all the uses. Most often people use CONTINUATION-DERIVED-TYPE or
41 ;;; CONTINUATION-TYPE instead of using this function directly.
42 (defun continuation-proven-type (cont)
43   (declare (type continuation cont))
44   (ecase (continuation-kind cont)
45     ((:block-start :deleted-block-start)
46      (let ((uses (block-start-uses (continuation-block cont))))
47        (if uses
48            (do ((res (node-derived-type (first uses))
49                      (values-type-union (node-derived-type (first current))
50                                         res))
51                 (current (rest uses) (rest current)))
52                ((null current) res))
53            *empty-type*)))
54     (:inside-block
55      (node-derived-type (continuation-use cont)))))
56
57 ;;; Our best guess for the type of this continuation's value. Note
58 ;;; that this may be Values or Function type, which cannot be passed
59 ;;; as an argument to the normal type operations. See
60 ;;; Continuation-Type. This may be called on deleted continuations,
61 ;;; always returning *.
62 ;;;
63 ;;; What we do is call CONTINUATION-PROVEN-TYPE and check whether the
64 ;;; result is a subtype of the assertion. If so, return the proven
65 ;;; type and set TYPE-CHECK to nil. Otherwise, return the intersection
66 ;;; of the asserted and proven types, and set TYPE-CHECK T. If
67 ;;; TYPE-CHECK already has a non-null value, then preserve it. Only in
68 ;;; the somewhat unusual circumstance of a newly discovered assertion
69 ;;; will we change TYPE-CHECK from NIL to T.
70 ;;;
71 ;;; The result value is cached in the CONTINUATION-%DERIVED-TYPE slot.
72 ;;; If the slot is true, just return that value, otherwise recompute
73 ;;; and stash the value there.
74 #!-sb-fluid (declaim (inline continuation-derived-type))
75 (defun continuation-derived-type (cont)
76   (declare (type continuation cont))
77   (or (continuation-%derived-type cont)
78       (%continuation-derived-type cont)))
79 (defun %continuation-derived-type (cont)
80   (declare (type continuation cont))
81   (let ((proven (continuation-proven-type cont))
82         (asserted (continuation-asserted-type cont)))
83     (cond ((values-subtypep proven asserted)
84            (setf (continuation-%type-check cont) nil)
85            (setf (continuation-%derived-type cont) proven))
86           (t
87            (unless (or (continuation-%type-check cont)
88                        (not (continuation-dest cont))
89                        (eq asserted *universal-type*))
90              (setf (continuation-%type-check cont) t))
91
92            (setf (continuation-%derived-type cont)
93                  (values-type-intersection asserted proven))))))
94
95 ;;; Call CONTINUATION-DERIVED-TYPE to make sure the slot is up to
96 ;;; date, then return it.
97 #!-sb-fluid (declaim (inline continuation-type-check))
98 (defun continuation-type-check (cont)
99   (declare (type continuation cont))
100   (continuation-derived-type cont)
101   (continuation-%type-check cont))
102
103 ;;; Return the derived type for CONT's first value. This is guaranteed
104 ;;; not to be a VALUES or FUNCTION type.
105 (declaim (ftype (function (continuation) ctype) continuation-type))
106 (defun continuation-type (cont)
107   (single-value-type (continuation-derived-type cont)))
108 \f
109 ;;;; interface routines used by optimizers
110
111 ;;; This function is called by optimizers to indicate that something
112 ;;; interesting has happened to the value of Cont. Optimizers must
113 ;;; make sure that they don't call for reoptimization when nothing has
114 ;;; happened, since optimization will fail to terminate.
115 ;;;
116 ;;; We clear any cached type for the continuation and set the
117 ;;; reoptimize flags on everything in sight, unless the continuation
118 ;;; is deleted (in which case we do nothing.)
119 ;;;
120 ;;; Since this can get called during IR1 conversion, we have to be
121 ;;; careful not to fly into space when the Dest's Prev is missing.
122 (defun reoptimize-continuation (cont)
123   (declare (type continuation cont))
124   (unless (member (continuation-kind cont) '(:deleted :unused))
125     (setf (continuation-%derived-type cont) nil)
126     (let ((dest (continuation-dest cont)))
127       (when dest
128         (setf (continuation-reoptimize cont) t)
129         (setf (node-reoptimize dest) t)
130         (let ((prev (node-prev dest)))
131           (when prev
132             (let* ((block (continuation-block prev))
133                    (component (block-component block)))
134               (when (typep dest 'cif)
135                 (setf (block-test-modified block) t))
136               (setf (block-reoptimize block) t)
137               (setf (component-reoptimize component) t))))))
138     (do-uses (node cont)
139       (setf (block-type-check (node-block node)) t)))
140   (values))
141
142 ;;; Annotate Node to indicate that its result has been proven to be
143 ;;; typep to RType. After IR1 conversion has happened, this is the
144 ;;; only correct way to supply information discovered about a node's
145 ;;; type. If you screw with the Node-Derived-Type directly, then
146 ;;; information may be lost and reoptimization may not happen.
147 ;;;
148 ;;; What we do is intersect Rtype with Node's Derived-Type. If the
149 ;;; intersection is different from the old type, then we do a
150 ;;; Reoptimize-Continuation on the Node-Cont.
151 (defun derive-node-type (node rtype)
152   (declare (type node node) (type ctype rtype))
153   (let ((node-type (node-derived-type node)))
154     (unless (eq node-type rtype)
155       (let ((int (values-type-intersection node-type rtype)))
156         (when (type/= node-type int)
157           (when (and *check-consistency*
158                      (eq int *empty-type*)
159                      (not (eq rtype *empty-type*)))
160             (let ((*compiler-error-context* node))
161               (compiler-warning
162                "New inferred type ~S conflicts with old type:~
163                 ~%  ~S~%*** Bug?"
164                (type-specifier rtype) (type-specifier node-type))))
165           (setf (node-derived-type node) int)
166           (reoptimize-continuation (node-cont node))))))
167   (values))
168
169 ;;; Similar to Derive-Node-Type, but asserts that it is an error for
170 ;;; Cont's value not to be typep to Type. If we improve the assertion,
171 ;;; we set TYPE-CHECK and TYPE-ASSERTED to guarantee that the new
172 ;;; assertion will be checked.
173 (defun assert-continuation-type (cont type)
174   (declare (type continuation cont) (type ctype type))
175   (let ((cont-type (continuation-asserted-type cont)))
176     (unless (eq cont-type type)
177       (let ((int (values-type-intersection cont-type type)))
178         (when (type/= cont-type int)
179           (setf (continuation-asserted-type cont) int)
180           (do-uses (node cont)
181             (setf (block-attributep (block-flags (node-block node))
182                                     type-check type-asserted)
183                   t))
184           (reoptimize-continuation cont)))))
185   (values))
186
187 ;;; Assert that Call is to a function of the specified Type. It is
188 ;;; assumed that the call is legal and has only constants in the
189 ;;; keyword positions.
190 (defun assert-call-type (call type)
191   (declare (type combination call) (type function-type type))
192   (derive-node-type call (function-type-returns type))
193   (let ((args (combination-args call)))
194     (dolist (req (function-type-required type))
195       (when (null args) (return-from assert-call-type))
196       (let ((arg (pop args)))
197         (assert-continuation-type arg req)))
198     (dolist (opt (function-type-optional type))
199       (when (null args) (return-from assert-call-type))
200       (let ((arg (pop args)))
201         (assert-continuation-type arg opt)))
202
203     (let ((rest (function-type-rest type)))
204       (when rest
205         (dolist (arg args)
206           (assert-continuation-type arg rest))))
207
208     (dolist (key (function-type-keywords type))
209       (let ((name (key-info-name key)))
210         (do ((arg args (cddr arg)))
211             ((null arg))
212           (when (eq (continuation-value (first arg)) name)
213             (assert-continuation-type
214              (second arg) (key-info-type key)))))))
215   (values))
216 \f
217 ;;;; IR1-OPTIMIZE
218
219 ;;; Do one forward pass over Component, deleting unreachable blocks
220 ;;; and doing IR1 optimizations. We can ignore all blocks that don't
221 ;;; have the Reoptimize flag set. If Component-Reoptimize is true when
222 ;;; we are done, then another iteration would be beneficial.
223 ;;;
224 ;;; We delete blocks when there is either no predecessor or the block
225 ;;; is in a lambda that has been deleted. These blocks would
226 ;;; eventually be deleted by DFO recomputation, but doing it here
227 ;;; immediately makes the effect available to IR1 optimization.
228 (defun ir1-optimize (component)
229   (declare (type component component))
230   (setf (component-reoptimize component) nil)
231   (do-blocks (block component)
232     (cond
233      ((or (block-delete-p block)
234           (null (block-pred block))
235           (eq (functional-kind (block-home-lambda block)) :deleted))
236       (delete-block block))
237      (t
238       (loop
239         (let ((succ (block-succ block)))
240           (unless (and succ (null (rest succ)))
241             (return)))
242         
243         (let ((last (block-last block)))
244           (typecase last
245             (cif
246              (flush-dest (if-test last))
247              (when (unlink-node last)
248                (return)))
249             (exit
250              (when (maybe-delete-exit last)
251                (return)))))
252         
253         (unless (join-successor-if-possible block)
254           (return)))
255
256       (when (and (block-reoptimize block) (block-component block))
257         (aver (not (block-delete-p block)))
258         (ir1-optimize-block block))
259
260       (when (and (block-flush-p block) (block-component block))
261         (aver (not (block-delete-p block)))
262         (flush-dead-code block)))))
263
264   (values))
265
266 ;;; Loop over the nodes in Block, looking for stuff that needs to be
267 ;;; optimized. We dispatch off of the type of each node with its
268 ;;; reoptimize flag set:
269
270 ;;; -- With a combination, we call Propagate-Function-Change whenever
271 ;;;    the function changes, and call IR1-Optimize-Combination if any
272 ;;;    argument changes.
273 ;;; -- With an Exit, we derive the node's type from the Value's type.
274 ;;;    We don't propagate Cont's assertion to the Value, since if we
275 ;;;    did, this would move the checking of Cont's assertion to the
276 ;;;    exit. This wouldn't work with Catch and UWP, where the Exit
277 ;;;    node is just a placeholder for the actual unknown exit.
278 ;;;
279 ;;; Note that we clear the node & block reoptimize flags *before*
280 ;;; doing the optimization. This ensures that the node or block will
281 ;;; be reoptimized if necessary. We leave the NODE-OPTIMIZE flag set
282 ;;; going into IR1-OPTIMIZE-RETURN, since IR1-OPTIMIZE-RETURN wants to
283 ;;; clear the flag itself.
284 (defun ir1-optimize-block (block)
285   (declare (type cblock block))
286   (setf (block-reoptimize block) nil)
287   (do-nodes (node cont block :restart-p t)
288     (when (node-reoptimize node)
289       (setf (node-reoptimize node) nil)
290       (typecase node
291         (ref)
292         (combination
293          (ir1-optimize-combination node))
294         (cif
295          (ir1-optimize-if node))
296         (creturn
297          (setf (node-reoptimize node) t)
298          (ir1-optimize-return node))
299         (mv-combination
300          (ir1-optimize-mv-combination node))
301         (exit
302          (let ((value (exit-value node)))
303            (when value
304              (derive-node-type node (continuation-derived-type value)))))
305         (cset
306          (ir1-optimize-set node)))))
307   (values))
308
309 ;;; We cannot combine with a successor block if:
310 ;;;  1. The successor has more than one predecessor.
311 ;;;  2. The last node's CONT is also used somewhere else.
312 ;;;  3. The successor is the current block (infinite loop).
313 ;;;  4. The next block has a different cleanup, and thus we may want to 
314 ;;;     insert cleanup code between the two blocks at some point.
315 ;;;  5. The next block has a different home lambda, and thus the control
316 ;;;     transfer is a non-local exit.
317 ;;;
318 ;;; If we succeed, we return true, otherwise false.
319 ;;;
320 ;;; Joining is easy when the successor's Start continuation is the
321 ;;; same from our Last's Cont. If they differ, then we can still join
322 ;;; when the last continuation has no next and the next continuation
323 ;;; has no uses. In this case, we replace the next continuation with
324 ;;; the last before joining the blocks.
325 (defun join-successor-if-possible (block)
326   (declare (type cblock block))
327   (let ((next (first (block-succ block))))
328     (when (block-start next)
329       (let* ((last (block-last block))
330              (last-cont (node-cont last))
331              (next-cont (block-start next)))
332         (cond ((or (rest (block-pred next))
333                    (not (eq (continuation-use last-cont) last))
334                    (eq next block)
335                    (not (eq (block-end-cleanup block)
336                             (block-start-cleanup next)))
337                    (not (eq (block-home-lambda block)
338                             (block-home-lambda next))))
339                nil)
340               ((eq last-cont next-cont)
341                (join-blocks block next)
342                t)
343               ((and (null (block-start-uses next))
344                     (eq (continuation-kind last-cont) :inside-block))
345                (let ((next-node (continuation-next next-cont)))
346                  ;; If next-cont does have a dest, it must be
347                  ;; unreachable, since there are no uses.
348                  ;; DELETE-CONTINUATION will mark the dest block as
349                  ;; delete-p [and also this block, unless it is no
350                  ;; longer backward reachable from the dest block.]
351                  (delete-continuation next-cont)
352                  (setf (node-prev next-node) last-cont)
353                  (setf (continuation-next last-cont) next-node)
354                  (setf (block-start next) last-cont)
355                  (join-blocks block next))
356                t)
357               (t
358                nil))))))
359
360 ;;; Join together two blocks which have the same ending/starting
361 ;;; continuation. The code in Block2 is moved into Block1 and Block2
362 ;;; is deleted from the DFO. We combine the optimize flags for the two
363 ;;; blocks so that any indicated optimization gets done.
364 (defun join-blocks (block1 block2)
365   (declare (type cblock block1 block2))
366   (let* ((last (block-last block2))
367          (last-cont (node-cont last))
368          (succ (block-succ block2))
369          (start2 (block-start block2)))
370     (do ((cont start2 (node-cont (continuation-next cont))))
371         ((eq cont last-cont)
372          (when (eq (continuation-kind last-cont) :inside-block)
373            (setf (continuation-block last-cont) block1)))
374       (setf (continuation-block cont) block1))
375
376     (unlink-blocks block1 block2)
377     (dolist (block succ)
378       (unlink-blocks block2 block)
379       (link-blocks block1 block))
380
381     (setf (block-last block1) last)
382     (setf (continuation-kind start2) :inside-block))
383
384   (setf (block-flags block1)
385         (attributes-union (block-flags block1)
386                           (block-flags block2)
387                           (block-attributes type-asserted test-modified)))
388
389   (let ((next (block-next block2))
390         (prev (block-prev block2)))
391     (setf (block-next prev) next)
392     (setf (block-prev next) prev))
393
394   (values))
395
396 ;;; Delete any nodes in BLOCK whose value is unused and have no
397 ;;; side-effects. We can delete sets of lexical variables when the set
398 ;;; variable has no references.
399 ;;;
400 ;;; [### For now, don't delete potentially flushable calls when they
401 ;;; have the CALL attribute. Someday we should look at the funcitonal
402 ;;; args to determine if they have any side-effects.]
403 (defun flush-dead-code (block)
404   (declare (type cblock block))
405   (do-nodes-backwards (node cont block)
406     (unless (continuation-dest cont)
407       (typecase node
408         (ref
409          (delete-ref node)
410          (unlink-node node))
411         (combination
412          (let ((info (combination-kind node)))
413            (when (function-info-p info)
414              (let ((attr (function-info-attributes info)))
415                (when (and (ir1-attributep attr flushable)
416                           (not (ir1-attributep attr call)))
417                  (flush-dest (combination-fun node))
418                  (dolist (arg (combination-args node))
419                    (flush-dest arg))
420                  (unlink-node node))))))
421         (mv-combination
422          (when (eq (basic-combination-kind node) :local)
423            (let ((fun (combination-lambda node)))
424              (when (dolist (var (lambda-vars fun) t)
425                      (when (or (leaf-refs var)
426                                (lambda-var-sets var))
427                        (return nil)))
428                (flush-dest (first (basic-combination-args node)))
429                (delete-let fun)))))
430         (exit
431          (let ((value (exit-value node)))
432            (when value
433              (flush-dest value)
434              (setf (exit-value node) nil))))
435         (cset
436          (let ((var (set-var node)))
437            (when (and (lambda-var-p var)
438                       (null (leaf-refs var)))
439              (flush-dest (set-value node))
440              (setf (basic-var-sets var)
441                    (delete node (basic-var-sets var)))
442              (unlink-node node)))))))
443
444   (setf (block-flush-p block) nil)
445   (values))
446 \f
447 ;;;; local call return type propagation
448
449 ;;; This function is called on RETURN nodes that have their REOPTIMIZE
450 ;;; flag set. It iterates over the uses of the RESULT, looking for
451 ;;; interesting stuff to update the TAIL-SET. If a use isn't a local
452 ;;; call, then we union its type together with the types of other such
453 ;;; uses. We assign to the RETURN-RESULT-TYPE the intersection of this
454 ;;; type with the RESULT's asserted type. We can make this
455 ;;; intersection now (potentially before type checking) because this
456 ;;; assertion on the result will eventually be checked (if
457 ;;; appropriate.)
458 ;;;
459 ;;; We call MAYBE-CONVERT-TAIL-LOCAL-CALL on each local non-MV
460 ;;; combination, which may change the succesor of the call to be the
461 ;;; called function, and if so, checks if the call can become an
462 ;;; assignment. If we convert to an assignment, we abort, since the
463 ;;; RETURN has been deleted.
464 (defun find-result-type (node)
465   (declare (type creturn node))
466   (let ((result (return-result node)))
467     (collect ((use-union *empty-type* values-type-union))
468       (do-uses (use result)
469         (cond ((and (basic-combination-p use)
470                     (eq (basic-combination-kind use) :local))
471                (aver (eq (lambda-tail-set (node-home-lambda use))
472                          (lambda-tail-set (combination-lambda use))))
473                (when (combination-p use)
474                  (when (nth-value 1 (maybe-convert-tail-local-call use))
475                    (return-from find-result-type (values)))))
476               (t
477                (use-union (node-derived-type use)))))
478       (let ((int (values-type-intersection
479                   (continuation-asserted-type result)
480                   (use-union))))
481         (setf (return-result-type node) int))))
482   (values))
483
484 ;;; Do stuff to realize that something has changed about the value
485 ;;; delivered to a return node. Since we consider the return values of
486 ;;; all functions in the tail set to be equivalent, this amounts to
487 ;;; bringing the entire tail set up to date. We iterate over the
488 ;;; returns for all the functions in the tail set, reanalyzing them
489 ;;; all (not treating Node specially.)
490 ;;;
491 ;;; When we are done, we check whether the new type is different from
492 ;;; the old TAIL-SET-TYPE. If so, we set the type and also reoptimize
493 ;;; all the continuations for references to functions in the tail set.
494 ;;; This will cause IR1-OPTIMIZE-COMBINATION to derive the new type as
495 ;;; the results of the calls.
496 (defun ir1-optimize-return (node)
497   (declare (type creturn node))
498   (let* ((tails (lambda-tail-set (return-lambda node)))
499          (funs (tail-set-functions tails)))
500     (collect ((res *empty-type* values-type-union))
501       (dolist (fun funs)
502         (let ((return (lambda-return fun)))
503           (when return
504             (when (node-reoptimize return)
505               (setf (node-reoptimize return) nil)
506               (find-result-type return))
507             (res (return-result-type return)))))
508
509       (when (type/= (res) (tail-set-type tails))
510         (setf (tail-set-type tails) (res))
511         (dolist (fun (tail-set-functions tails))
512           (dolist (ref (leaf-refs fun))
513             (reoptimize-continuation (node-cont ref)))))))
514
515   (values))
516 \f
517 ;;;; IF optimization
518
519 ;;; If the test has multiple uses, replicate the node when possible.
520 ;;; Also check whether the predicate is known to be true or false,
521 ;;; deleting the IF node in favor of the appropriate branch when this
522 ;;; is the case.
523 (defun ir1-optimize-if (node)
524   (declare (type cif node))
525   (let ((test (if-test node))
526         (block (node-block node)))
527
528     (when (and (eq (block-start block) test)
529                (eq (continuation-next test) node)
530                (rest (block-start-uses block)))
531       (do-uses (use test)
532         (when (immediately-used-p test use)
533           (convert-if-if use node)
534           (when (continuation-use test) (return)))))
535
536     (let* ((type (continuation-type test))
537            (victim
538             (cond ((constant-continuation-p test)
539                    (if (continuation-value test)
540                        (if-alternative node)
541                        (if-consequent node)))
542                   ((not (types-equal-or-intersect type (specifier-type 'null)))
543                    (if-alternative node))
544                   ((type= type (specifier-type 'null))
545                    (if-consequent node)))))
546       (when victim
547         (flush-dest test)
548         (when (rest (block-succ block))
549           (unlink-blocks block victim))
550         (setf (component-reanalyze (block-component (node-block node))) t)
551         (unlink-node node))))
552   (values))
553
554 ;;; Create a new copy of an IF Node that tests the value of the node
555 ;;; Use. The test must have >1 use, and must be immediately used by
556 ;;; Use. Node must be the only node in its block (implying that
557 ;;; block-start = if-test).
558 ;;;
559 ;;; This optimization has an effect semantically similar to the
560 ;;; source-to-source transformation:
561 ;;;    (IF (IF A B C) D E) ==>
562 ;;;    (IF A (IF B D E) (IF C D E))
563 ;;;
564 ;;; We clobber the NODE-SOURCE-PATH of both the original and the new
565 ;;; node so that dead code deletion notes will definitely not consider
566 ;;; either node to be part of the original source. One node might
567 ;;; become unreachable, resulting in a spurious note.
568 (defun convert-if-if (use node)
569   (declare (type node use) (type cif node))
570   (with-ir1-environment node
571     (let* ((block (node-block node))
572            (test (if-test node))
573            (cblock (if-consequent node))
574            (ablock (if-alternative node))
575            (use-block (node-block use))
576            (dummy-cont (make-continuation))
577            (new-cont (make-continuation))
578            (new-node (make-if :test new-cont
579                               :consequent cblock
580                               :alternative ablock))
581            (new-block (continuation-starts-block new-cont)))
582       (prev-link new-node new-cont)
583       (setf (continuation-dest new-cont) new-node)
584       (add-continuation-use new-node dummy-cont)
585       (setf (block-last new-block) new-node)
586
587       (unlink-blocks use-block block)
588       (delete-continuation-use use)
589       (add-continuation-use use new-cont)
590       (link-blocks use-block new-block)
591
592       (link-blocks new-block cblock)
593       (link-blocks new-block ablock)
594
595       (push "<IF Duplication>" (node-source-path node))
596       (push "<IF Duplication>" (node-source-path new-node))
597
598       (reoptimize-continuation test)
599       (reoptimize-continuation new-cont)
600       (setf (component-reanalyze *current-component*) t)))
601   (values))
602 \f
603 ;;;; exit IR1 optimization
604
605 ;;; This function attempts to delete an exit node, returning true if
606 ;;; it deletes the block as a consequence:
607 ;;; -- If the exit is degenerate (has no Entry), then we don't do anything,
608 ;;;    since there is nothing to be done.
609 ;;; -- If the exit node and its Entry have the same home lambda then we know
610 ;;;    the exit is local, and can delete the exit. We change uses of the
611 ;;;    Exit-Value to be uses of the original continuation, then unlink the
612 ;;;    node. If the exit is to a TR context, then we must do MERGE-TAIL-SETS
613 ;;;    on any local calls which delivered their value to this exit.
614 ;;; -- If there is no value (as in a GO), then we skip the value semantics.
615 ;;;
616 ;;; This function is also called by environment analysis, since it
617 ;;; wants all exits to be optimized even if normal optimization was
618 ;;; omitted.
619 (defun maybe-delete-exit (node)
620   (declare (type exit node))
621   (let ((value (exit-value node))
622         (entry (exit-entry node))
623         (cont (node-cont node)))
624     (when (and entry
625                (eq (node-home-lambda node) (node-home-lambda entry)))
626       (setf (entry-exits entry) (delete node (entry-exits entry)))
627       (prog1
628           (unlink-node node)
629         (when value
630           (collect ((merges))
631             (when (return-p (continuation-dest cont))
632               (do-uses (use value)
633                 (when (and (basic-combination-p use)
634                            (eq (basic-combination-kind use) :local))
635                   (merges use))))
636             (substitute-continuation-uses cont value)
637             (dolist (merge (merges))
638               (merge-tail-sets merge))))))))
639 \f
640 ;;;; combination IR1 optimization
641
642 ;;; Report as we try each transform?
643 #!+sb-show
644 (defvar *show-transforms-p* nil)
645
646 ;;; Do IR1 optimizations on a COMBINATION node.
647 (declaim (ftype (function (combination) (values)) ir1-optimize-combination))
648 (defun ir1-optimize-combination (node)
649   (when (continuation-reoptimize (basic-combination-fun node))
650     (propagate-function-change node))
651   (let ((args (basic-combination-args node))
652         (kind (basic-combination-kind node)))
653     (case kind
654       (:local
655        (let ((fun (combination-lambda node)))
656          (if (eq (functional-kind fun) :let)
657              (propagate-let-args node fun)
658              (propagate-local-call-args node fun))))
659       ((:full :error)
660        (dolist (arg args)
661          (when arg
662            (setf (continuation-reoptimize arg) nil))))
663       (t
664        (dolist (arg args)
665          (when arg
666            (setf (continuation-reoptimize arg) nil)))
667
668        (let ((attr (function-info-attributes kind)))
669          (when (and (ir1-attributep attr foldable)
670                     ;; KLUDGE: The next test could be made more sensitive,
671                     ;; only suppressing constant-folding of functions with
672                     ;; CALL attributes when they're actually passed
673                     ;; function arguments. -- WHN 19990918
674                     (not (ir1-attributep attr call))
675                     (every #'constant-continuation-p args)
676                     (continuation-dest (node-cont node))
677                     ;; Even if the function is foldable in principle,
678                     ;; it might be one of our low-level
679                     ;; implementation-specific functions. Such
680                     ;; functions don't necessarily exist at runtime on
681                     ;; a plain vanilla ANSI Common Lisp
682                     ;; cross-compilation host, in which case the
683                     ;; cross-compiler can't fold it because the
684                     ;; cross-compiler doesn't know how to evaluate it.
685                     #+sb-xc-host
686                     (let* ((ref (continuation-use (combination-fun node)))
687                            (fun (leaf-name (ref-leaf ref))))
688                       (fboundp fun)))
689            (constant-fold-call node)
690            (return-from ir1-optimize-combination)))
691
692        (let ((fun (function-info-derive-type kind)))
693          (when fun
694            (let ((res (funcall fun node)))
695              (when res
696                (derive-node-type node res)
697                (maybe-terminate-block node nil)))))
698
699        (let ((fun (function-info-optimizer kind)))
700          (unless (and fun (funcall fun node))
701            (dolist (x (function-info-transforms kind))
702              #!+sb-show 
703              (when *show-transforms-p*
704                (let* ((cont (basic-combination-fun node))
705                       (fname (continuation-function-name cont t)))
706                  (/show "trying transform" x (transform-function x) "for" fname)))
707              (unless (ir1-transform node x)
708                #!+sb-show
709                (when *show-transforms-p*
710                  (/show "quitting because IR1-TRANSFORM result was NIL"))
711                (return))))))))
712
713   (values))
714
715 ;;; If Call is to a function that doesn't return (i.e. return type is
716 ;;; NIL), then terminate the block there, and link it to the component
717 ;;; tail. We also change the call's CONT to be a dummy continuation to
718 ;;; prevent the use from confusing things.
719 ;;;
720 ;;; Except when called during IR1, we delete the continuation if it
721 ;;; has no other uses. (If it does have other uses, we reoptimize.)
722 ;;;
723 ;;; Termination on the basis of a continuation type assertion is
724 ;;; inhibited when:
725 ;;; -- The continuation is deleted (hence the assertion is spurious), or
726 ;;; -- We are in IR1 conversion (where THE assertions are subject to
727 ;;;    weakening.)
728 (defun maybe-terminate-block (call ir1-p)
729   (declare (type basic-combination call))
730   (let* ((block (node-block call))
731          (cont (node-cont call))
732          (tail (component-tail (block-component block)))
733          (succ (first (block-succ block))))
734     (unless (or (and (eq call (block-last block)) (eq succ tail))
735                 (block-delete-p block)
736                 *converting-for-interpreter*)
737       (when (or (and (eq (continuation-asserted-type cont) *empty-type*)
738                      (not (or ir1-p (eq (continuation-kind cont) :deleted))))
739                 (eq (node-derived-type call) *empty-type*))
740         (cond (ir1-p
741                (delete-continuation-use call)
742                (cond
743                 ((block-last block)
744                  (aver (and (eq (block-last block) call)
745                             (eq (continuation-kind cont) :block-start))))
746                 (t
747                  (setf (block-last block) call)
748                  (link-blocks block (continuation-starts-block cont)))))
749               (t
750                (node-ends-block call)
751                (delete-continuation-use call)
752                (if (eq (continuation-kind cont) :unused)
753                    (delete-continuation cont)
754                    (reoptimize-continuation cont))))
755         
756         (unlink-blocks block (first (block-succ block)))
757         (setf (component-reanalyze (block-component block)) t)
758         (aver (not (block-succ block)))
759         (link-blocks block tail)
760         (add-continuation-use call (make-continuation))
761         t))))
762
763 ;;; This is called both by IR1 conversion and IR1 optimization when
764 ;;; they have verified the type signature for the call, and are
765 ;;; wondering if something should be done to special-case the call. If
766 ;;; Call is a call to a global function, then see whether it defined
767 ;;; or known:
768 ;;; -- If a DEFINED-FUNCTION should be inline expanded, then convert the
769 ;;;    expansion and change the call to call it. Expansion is enabled if
770 ;;;    :INLINE or if space=0. If the FUNCTIONAL slot is true, we never expand,
771 ;;;    since this function has already been converted. Local call analysis
772 ;;;    will duplicate the definition if necessary. We claim that the parent
773 ;;;    form is LABELS for context declarations, since we don't want it to be
774 ;;;    considered a real global function.
775 ;;; -- In addition to a direct check for the function name in the table, we
776 ;;;    also must check for slot accessors. If the function is a slot accessor,
777 ;;;    then we set the combination kind to the function info of %Slot-Setter or
778 ;;;    %Slot-Accessor, as appropriate.
779 ;;; -- If it is a known function, mark it as such by setting the Kind.
780 ;;;
781 ;;; We return the leaf referenced (NIL if not a leaf) and the
782 ;;; function-info assigned.
783 (defun recognize-known-call (call ir1-p)
784   (declare (type combination call))
785   (let* ((ref (continuation-use (basic-combination-fun call)))
786          (leaf (when (ref-p ref) (ref-leaf ref)))
787          (inlinep (if (and (defined-function-p leaf)
788                            (not (byte-compiling)))
789                       (defined-function-inlinep leaf)
790                       :no-chance)))
791     (cond
792      ((eq inlinep :notinline) (values nil nil))
793      ((not (and (global-var-p leaf)
794                 (eq (global-var-kind leaf) :global-function)))
795       (values leaf nil))
796      ((and (ecase inlinep
797              (:inline t)
798              (:no-chance nil)
799              ((nil :maybe-inline) (policy call (zerop space))))
800            (defined-function-inline-expansion leaf)
801            (let ((fun (defined-function-functional leaf)))
802              (or (not fun)
803                  (and (eq inlinep :inline) (functional-kind fun))))
804            (inline-expansion-ok call))
805       (flet ((frob ()
806                (let ((res (ir1-convert-lambda-for-defun
807                            (defined-function-inline-expansion leaf)
808                            leaf t
809                            #'ir1-convert-inline-lambda)))
810                  (setf (defined-function-functional leaf) res)
811                  (change-ref-leaf ref res))))
812         (if ir1-p
813             (frob)
814             (with-ir1-environment call
815               (frob)
816               (local-call-analyze *current-component*))))
817
818       (values (ref-leaf (continuation-use (basic-combination-fun call)))
819               nil))
820      (t
821       (let* ((name (leaf-name leaf))
822              (info (info :function :info
823                          (if (slot-accessor-p leaf)
824                            (if (consp name)
825                              '%slot-setter
826                              '%slot-accessor)
827                            name))))
828         (if info
829             (values leaf (setf (basic-combination-kind call) info))
830             (values leaf nil)))))))
831
832 ;;; Check whether CALL satisfies TYPE. If so, apply the type to the
833 ;;; call, and do MAYBE-TERMINATE-BLOCK and return the values of
834 ;;; RECOGNIZE-KNOWN-CALL. If an error, set the combination kind and
835 ;;; return NIL, NIL. If the type is just FUNCTION, then skip the
836 ;;; syntax check, arg/result type processing, but still call
837 ;;; RECOGNIZE-KNOWN-CALL, since the call might be to a known lambda,
838 ;;; and that checking is done by local call analysis.
839 (defun validate-call-type (call type ir1-p)
840   (declare (type combination call) (type ctype type))
841   (cond ((not (function-type-p type))
842          (aver (multiple-value-bind (val win)
843                    (csubtypep type (specifier-type 'function))
844                  (or val (not win))))
845          (recognize-known-call call ir1-p))
846         ((valid-function-use call type
847                              :argument-test #'always-subtypep
848                              :result-test #'always-subtypep
849                              ;; KLUDGE: Common Lisp is such a dynamic
850                              ;; language that all we can do here in
851                              ;; general is issue a STYLE-WARNING. It
852                              ;; would be nice to issue a full WARNING
853                              ;; in the special case of of type
854                              ;; mismatches within a compilation unit
855                              ;; (as in section 3.2.2.3 of the spec)
856                              ;; but at least as of sbcl-0.6.11, we
857                              ;; don't keep track of whether the
858                              ;; mismatched data came from the same
859                              ;; compilation unit, so we can't do that.
860                              ;; -- WHN 2001-02-11
861                              ;;
862                              ;; FIXME: Actually, I think we could
863                              ;; issue a full WARNING if the call
864                              ;; violates a DECLAIM FTYPE.
865                              :error-function #'compiler-style-warning
866                              :warning-function #'compiler-note)
867          (assert-call-type call type)
868          (maybe-terminate-block call ir1-p)
869          (recognize-known-call call ir1-p))
870         (t
871          (setf (combination-kind call) :error)
872          (values nil nil))))
873
874 ;;; This is called by IR1-OPTIMIZE when the function for a call has
875 ;;; changed. If the call is local, we try to let-convert it, and
876 ;;; derive the result type. If it is a :FULL call, we validate it
877 ;;; against the type, which recognizes known calls, does inline
878 ;;; expansion, etc. If a call to a predicate in a non-conditional
879 ;;; position or to a function with a source transform, then we
880 ;;; reconvert the form to give IR1 another chance.
881 (defun propagate-function-change (call)
882   (declare (type combination call))
883   (let ((*compiler-error-context* call)
884         (fun-cont (basic-combination-fun call)))
885     (setf (continuation-reoptimize fun-cont) nil)
886     (case (combination-kind call)
887       (:local
888        (let ((fun (combination-lambda call)))
889          (maybe-let-convert fun)
890          (unless (member (functional-kind fun) '(:let :assignment :deleted))
891            (derive-node-type call (tail-set-type (lambda-tail-set fun))))))
892       (:full
893        (multiple-value-bind (leaf info)
894            (validate-call-type call (continuation-type fun-cont) nil)
895          (cond ((functional-p leaf)
896                 (convert-call-if-possible
897                  (continuation-use (basic-combination-fun call))
898                  call))
899                ((not leaf))
900                ((or (info :function :source-transform (leaf-name leaf))
901                     (and info
902                          (ir1-attributep (function-info-attributes info)
903                                          predicate)
904                          (let ((dest (continuation-dest (node-cont call))))
905                            (and dest (not (if-p dest))))))
906                 (let ((name (leaf-name leaf)))
907                   (when (symbolp name)
908                     (let ((dums (make-gensym-list (length
909                                                    (combination-args call)))))
910                       (transform-call call
911                                       `(lambda ,dums
912                                          (,name ,@dums))))))))))))
913   (values))
914 \f
915 ;;;; known function optimization
916
917 ;;; Add a failed optimization note to FAILED-OPTIMZATIONS for Node,
918 ;;; Fun and Args. If there is already a note for Node and Transform,
919 ;;; replace it, otherwise add a new one.
920 (defun record-optimization-failure (node transform args)
921   (declare (type combination node) (type transform transform)
922            (type (or function-type list) args))
923   (let* ((table (component-failed-optimizations *component-being-compiled*))
924          (found (assoc transform (gethash node table))))
925     (if found
926         (setf (cdr found) args)
927         (push (cons transform args) (gethash node table))))
928   (values))
929
930 ;;; Attempt to transform NODE using TRANSFORM-FUNCTION, subject to the
931 ;;; call type constraint TRANSFORM-TYPE. If we are inhibited from
932 ;;; doing the transform for some reason and FLAME is true, then we
933 ;;; make a note of the message in FAILED-OPTIMIZATIONS for IR1
934 ;;; finalize to pick up. We return true if the transform failed, and
935 ;;; thus further transformation should be attempted. We return false
936 ;;; if either the transform succeeded or was aborted.
937 (defun ir1-transform (node transform)
938   (declare (type combination node) (type transform transform))
939   (let* ((type (transform-type transform))
940          (fun (transform-function transform))
941          (constrained (function-type-p type))
942          (table (component-failed-optimizations *component-being-compiled*))
943          (flame (if (transform-important transform)
944                     (policy node (>= speed inhibit-warnings))
945                     (policy node (> speed inhibit-warnings))))
946          (*compiler-error-context* node))
947     (cond ((not (member (transform-when transform)
948                         (if *byte-compiling*
949                             '(:byte   :both)
950                             '(:native :both))))
951            ;; FIXME: Make sure that there's a transform for
952            ;; (MEMBER SYMBOL ..) into MEMQ.
953            ;; FIXME: Note that when/if I make SHARE operation to shared
954            ;; constant data between objects in the system, remember that a
955            ;; SHAREd list, or other SHAREd compound object, can be processed
956            ;; recursively, so that e.g. the two lists above can share their
957            ;; '(:BOTH) tail sublists.
958            (let ((when (transform-when transform)))
959              (not (or (eq when :both)
960                       (eq when (if *byte-compiling* :byte :native)))))
961            t)
962           ((or (not constrained)
963                (valid-function-use node type :strict-result t))
964            (multiple-value-bind (severity args)
965                (catch 'give-up-ir1-transform
966                  (transform-call node (funcall fun node))
967                  (values :none nil))
968              (ecase severity
969                (:none
970                 (remhash node table)
971                 nil)
972                (:aborted
973                 (setf (combination-kind node) :error)
974                 (when args
975                   (apply #'compiler-warning args))
976                 (remhash node table)
977                 nil)
978                (:failure
979                 (if args
980                     (when flame
981                       (record-optimization-failure node transform args))
982                     (setf (gethash node table)
983                           (remove transform (gethash node table) :key #'car)))
984                 t))))
985           ((and flame
986                 (valid-function-use node
987                                     type
988                                     :argument-test #'types-equal-or-intersect
989                                     :result-test
990                                     #'values-types-equal-or-intersect))
991            (record-optimization-failure node transform type)
992            t)
993           (t
994            t))))
995
996 ;;; Just throw the severity and args...
997 (declaim (ftype (function (&rest t) nil) give-up-ir1-transform))
998 (defun give-up-ir1-transform (&rest args)
999   #!+sb-doc
1000   "This function is used to throw out of an IR1 transform, aborting this
1001   attempt to transform the call, but admitting the possibility that this or
1002   some other transform will later succeed. If arguments are supplied, they are
1003   format arguments for an efficiency note."
1004   (throw 'give-up-ir1-transform (values :failure args)))
1005 (defun abort-ir1-transform (&rest args)
1006   #!+sb-doc
1007   "This function is used to throw out of an IR1 transform and force a normal
1008   call to the function at run time. No further optimizations will be
1009   attempted."
1010   (throw 'give-up-ir1-transform (values :aborted args)))
1011
1012 ;;; Take the lambda-expression Res, IR1 convert it in the proper
1013 ;;; environment, and then install it as the function for the call
1014 ;;; Node. We do local call analysis so that the new function is
1015 ;;; integrated into the control flow.
1016 (defun transform-call (node res)
1017   (declare (type combination node) (list res))
1018   (with-ir1-environment node
1019     (let ((new-fun (ir1-convert-inline-lambda res))
1020           (ref (continuation-use (combination-fun node))))
1021       (change-ref-leaf ref new-fun)
1022       (setf (combination-kind node) :full)
1023       (local-call-analyze *current-component*)))
1024   (values))
1025
1026 ;;; Replace a call to a foldable function of constant arguments with
1027 ;;; the result of evaluating the form. We insert the resulting
1028 ;;; constant node after the call, stealing the call's continuation. We
1029 ;;; give the call a continuation with no Dest, which should cause it
1030 ;;; and its arguments to go away. If there is an error during the
1031 ;;; evaluation, we give a warning and leave the call alone, making the
1032 ;;; call a :ERROR call.
1033 ;;;
1034 ;;; If there is more than one value, then we transform the call into a
1035 ;;; values form.
1036 (defun constant-fold-call (call)
1037   (declare (type combination call))
1038   (let* ((args (mapcar #'continuation-value (combination-args call)))
1039          (ref (continuation-use (combination-fun call)))
1040          (fun (leaf-name (ref-leaf ref))))
1041
1042     (multiple-value-bind (values win)
1043         (careful-call fun args call "constant folding")
1044       (if (not win)
1045         (setf (combination-kind call) :error)
1046         (let ((dummies (make-gensym-list (length args))))
1047           (transform-call
1048            call
1049            `(lambda ,dummies
1050               (declare (ignore ,@dummies))
1051               (values ,@(mapcar #'(lambda (x) `',x) values))))))))
1052
1053   (values))
1054 \f
1055 ;;;; local call optimization
1056
1057 ;;; Propagate Type to Leaf and its Refs, marking things changed. If
1058 ;;; the leaf type is a function type, then just leave it alone, since
1059 ;;; TYPE is never going to be more specific than that (and
1060 ;;; TYPE-INTERSECTION would choke.)
1061 (defun propagate-to-refs (leaf type)
1062   (declare (type leaf leaf) (type ctype type))
1063   (let ((var-type (leaf-type leaf)))
1064     (unless (function-type-p var-type)
1065       (let ((int (type-approx-intersection2 var-type type)))
1066         (when (type/= int var-type)
1067           (setf (leaf-type leaf) int)
1068           (dolist (ref (leaf-refs leaf))
1069             (derive-node-type ref int))))
1070       (values))))
1071
1072 ;;; Figure out the type of a LET variable that has sets. We compute
1073 ;;; the union of the initial value Type and the types of all the set
1074 ;;; values and to a PROPAGATE-TO-REFS with this type.
1075 (defun propagate-from-sets (var type)
1076   (collect ((res type type-union))
1077     (dolist (set (basic-var-sets var))
1078       (res (continuation-type (set-value set)))
1079       (setf (node-reoptimize set) nil))
1080     (propagate-to-refs var (res)))
1081   (values))
1082
1083 ;;; If a LET variable, find the initial value's type and do
1084 ;;; PROPAGATE-FROM-SETS. We also derive the VALUE's type as the node's
1085 ;;; type.
1086 (defun ir1-optimize-set (node)
1087   (declare (type cset node))
1088   (let ((var (set-var node)))
1089     (when (and (lambda-var-p var) (leaf-refs var))
1090       (let ((home (lambda-var-home var)))
1091         (when (eq (functional-kind home) :let)
1092           (let ((iv (let-var-initial-value var)))
1093             (setf (continuation-reoptimize iv) nil)
1094             (propagate-from-sets var (continuation-type iv)))))))
1095
1096   (derive-node-type node (continuation-type (set-value node)))
1097   (values))
1098
1099 ;;; Return true if the value of Ref will always be the same (and is
1100 ;;; thus legal to substitute.)
1101 (defun constant-reference-p (ref)
1102   (declare (type ref ref))
1103   (let ((leaf (ref-leaf ref)))
1104     (typecase leaf
1105       ((or constant functional) t)
1106       (lambda-var
1107        (null (lambda-var-sets leaf)))
1108       (defined-function
1109        (not (eq (defined-function-inlinep leaf) :notinline)))
1110       (global-var
1111        (case (global-var-kind leaf)
1112          (:global-function t)
1113          (:constant t))))))
1114
1115 ;;; If we have a non-set LET var with a single use, then (if possible)
1116 ;;; replace the variable reference's CONT with the arg continuation.
1117 ;;; This is inhibited when:
1118 ;;; -- CONT has other uses, or
1119 ;;; -- CONT receives multiple values, or
1120 ;;; -- the reference is in a different environment from the variable, or
1121 ;;; -- either continuation has a funky TYPE-CHECK annotation.
1122 ;;; -- the continuations have incompatible assertions, so the new asserted type
1123 ;;;    would be NIL.
1124 ;;; -- the var's DEST has a different policy than the ARG's (think safety).
1125 ;;;
1126 ;;; We change the Ref to be a reference to NIL with unused value, and
1127 ;;; let it be flushed as dead code. A side-effect of this substitution
1128 ;;; is to delete the variable.
1129 (defun substitute-single-use-continuation (arg var)
1130   (declare (type continuation arg) (type lambda-var var))
1131   (let* ((ref (first (leaf-refs var)))
1132          (cont (node-cont ref))
1133          (cont-atype (continuation-asserted-type cont))
1134          (dest (continuation-dest cont)))
1135     (when (and (eq (continuation-use cont) ref)
1136                dest
1137                (not (typep dest '(or creturn exit mv-combination)))
1138                (eq (node-home-lambda ref)
1139                    (lambda-home (lambda-var-home var)))
1140                (member (continuation-type-check arg) '(t nil))
1141                (member (continuation-type-check cont) '(t nil))
1142                (not (eq (values-type-intersection
1143                          cont-atype
1144                          (continuation-asserted-type arg))
1145                         *empty-type*))
1146                (eq (lexenv-policy (node-lexenv dest))
1147                    (lexenv-policy (node-lexenv (continuation-dest arg)))))
1148       (aver (member (continuation-kind arg)
1149                     '(:block-start :deleted-block-start :inside-block)))
1150       (assert-continuation-type arg cont-atype)
1151       (setf (node-derived-type ref) *wild-type*)
1152       (change-ref-leaf ref (find-constant nil))
1153       (substitute-continuation arg cont)
1154       (reoptimize-continuation arg)
1155       t)))
1156
1157 ;;; Delete a LET, removing the call and bind nodes, and warning about
1158 ;;; any unreferenced variables. Note that FLUSH-DEAD-CODE will come
1159 ;;; along right away and delete the REF and then the lambda, since we
1160 ;;; flush the FUN continuation.
1161 (defun delete-let (fun)
1162   (declare (type clambda fun))
1163   (aver (member (functional-kind fun) '(:let :mv-let)))
1164   (note-unreferenced-vars fun)
1165   (let ((call (let-combination fun)))
1166     (flush-dest (basic-combination-fun call))
1167     (unlink-node call)
1168     (unlink-node (lambda-bind fun))
1169     (setf (lambda-bind fun) nil))
1170   (values))
1171
1172 ;;; This function is called when one of the arguments to a LET
1173 ;;; changes. We look at each changed argument. If the corresponding
1174 ;;; variable is set, then we call PROPAGATE-FROM-SETS. Otherwise, we
1175 ;;; consider substituting for the variable, and also propagate
1176 ;;; derived-type information for the arg to all the Var's refs.
1177 ;;;
1178 ;;; Substitution is inhibited when the arg leaf's derived type isn't a
1179 ;;; subtype of the argument's asserted type. This prevents type
1180 ;;; checking from being defeated, and also ensures that the best
1181 ;;; representation for the variable can be used.
1182 ;;;
1183 ;;; Substitution of individual references is inhibited if the
1184 ;;; reference is in a different component from the home. This can only
1185 ;;; happen with closures over top-level lambda vars. In such cases,
1186 ;;; the references may have already been compiled, and thus can't be
1187 ;;; retroactively modified.
1188 ;;;
1189 ;;; If all of the variables are deleted (have no references) when we
1190 ;;; are done, then we delete the LET.
1191 ;;;
1192 ;;; Note that we are responsible for clearing the
1193 ;;; Continuation-Reoptimize flags.
1194 (defun propagate-let-args (call fun)
1195   (declare (type combination call) (type clambda fun))
1196   (loop for arg in (combination-args call)
1197         and var in (lambda-vars fun) do
1198     (when (and arg (continuation-reoptimize arg))
1199       (setf (continuation-reoptimize arg) nil)
1200       (cond
1201        ((lambda-var-sets var)
1202         (propagate-from-sets var (continuation-type arg)))
1203        ((let ((use (continuation-use arg)))
1204           (when (ref-p use)
1205             (let ((leaf (ref-leaf use)))
1206               (when (and (constant-reference-p use)
1207                          (values-subtypep (leaf-type leaf)
1208                                           (continuation-asserted-type arg)))
1209                 (propagate-to-refs var (continuation-type arg))
1210                 (let ((this-comp (block-component (node-block use))))
1211                   (substitute-leaf-if
1212                    #'(lambda (ref)
1213                        (cond ((eq (block-component (node-block ref))
1214                                   this-comp)
1215                               t)
1216                              (t
1217                               (aver (eq (functional-kind (lambda-home fun))
1218                                         :top-level))
1219                               nil)))
1220                    leaf var))
1221                 t)))))
1222        ((and (null (rest (leaf-refs var)))
1223              (not *byte-compiling*)
1224              (substitute-single-use-continuation arg var)))
1225        (t
1226         (propagate-to-refs var (continuation-type arg))))))
1227
1228   (when (every #'null (combination-args call))
1229     (delete-let fun))
1230
1231   (values))
1232
1233 ;;; This function is called when one of the args to a non-LET local
1234 ;;; call changes. For each changed argument corresponding to an unset
1235 ;;; variable, we compute the union of the types across all calls and
1236 ;;; propagate this type information to the var's refs.
1237 ;;;
1238 ;;; If the function has an XEP, then we don't do anything, since we
1239 ;;; won't discover anything.
1240 ;;;
1241 ;;; We can clear the Continuation-Reoptimize flags for arguments in
1242 ;;; all calls corresponding to changed arguments in Call, since the
1243 ;;; only use in IR1 optimization of the Reoptimize flag for local call
1244 ;;; args is right here.
1245 (defun propagate-local-call-args (call fun)
1246   (declare (type combination call) (type clambda fun))
1247
1248   (unless (or (functional-entry-function fun)
1249               (lambda-optional-dispatch fun))
1250     (let* ((vars (lambda-vars fun))
1251            (union (mapcar #'(lambda (arg var)
1252                               (when (and arg
1253                                          (continuation-reoptimize arg)
1254                                          (null (basic-var-sets var)))
1255                                 (continuation-type arg)))
1256                           (basic-combination-args call)
1257                           vars))
1258            (this-ref (continuation-use (basic-combination-fun call))))
1259
1260       (dolist (arg (basic-combination-args call))
1261         (when arg
1262           (setf (continuation-reoptimize arg) nil)))
1263
1264       (dolist (ref (leaf-refs fun))
1265         (let ((dest (continuation-dest (node-cont ref))))
1266           (unless (or (eq ref this-ref) (not dest))
1267             (setq union
1268                   (mapcar #'(lambda (this-arg old)
1269                               (when old
1270                                 (setf (continuation-reoptimize this-arg) nil)
1271                                 (type-union (continuation-type this-arg) old)))
1272                           (basic-combination-args dest)
1273                           union)))))
1274
1275       (mapc #'(lambda (var type)
1276                 (when type
1277                   (propagate-to-refs var type)))
1278             vars union)))
1279
1280   (values))
1281 \f
1282 ;;;; multiple values optimization
1283
1284 ;;; Do stuff to notice a change to a MV combination node. There are
1285 ;;; two main branches here:
1286 ;;;  -- If the call is local, then it is already a MV let, or should
1287 ;;;     become one. Note that although all :LOCAL MV calls must eventually
1288 ;;;     be converted to :MV-LETs, there can be a window when the call
1289 ;;;     is local, but has not been LET converted yet. This is because
1290 ;;;     the entry-point lambdas may have stray references (in other
1291 ;;;     entry points) that have not been deleted yet.
1292 ;;;  -- The call is full. This case is somewhat similar to the non-MV
1293 ;;;     combination optimization: we propagate return type information and
1294 ;;;     notice non-returning calls. We also have an optimization
1295 ;;;     which tries to convert MV-CALLs into MV-binds.
1296 (defun ir1-optimize-mv-combination (node)
1297   (ecase (basic-combination-kind node)
1298     (:local
1299      (let ((fun-cont (basic-combination-fun node)))
1300        (when (continuation-reoptimize fun-cont)
1301          (setf (continuation-reoptimize fun-cont) nil)
1302          (maybe-let-convert (combination-lambda node))))
1303      (setf (continuation-reoptimize (first (basic-combination-args node))) nil)
1304      (when (eq (functional-kind (combination-lambda node)) :mv-let)
1305        (unless (convert-mv-bind-to-let node)
1306          (ir1-optimize-mv-bind node))))
1307     (:full
1308      (let* ((fun (basic-combination-fun node))
1309             (fun-changed (continuation-reoptimize fun))
1310             (args (basic-combination-args node)))
1311        (when fun-changed
1312          (setf (continuation-reoptimize fun) nil)
1313          (let ((type (continuation-type fun)))
1314            (when (function-type-p type)
1315              (derive-node-type node (function-type-returns type))))
1316          (maybe-terminate-block node nil)
1317          (let ((use (continuation-use fun)))
1318            (when (and (ref-p use) (functional-p (ref-leaf use)))
1319              (convert-call-if-possible use node)
1320              (when (eq (basic-combination-kind node) :local)
1321                (maybe-let-convert (ref-leaf use))))))
1322        (unless (or (eq (basic-combination-kind node) :local)
1323                    (eq (continuation-function-name fun) '%throw))
1324          (ir1-optimize-mv-call node))
1325        (dolist (arg args)
1326          (setf (continuation-reoptimize arg) nil))))
1327     (:error))
1328   (values))
1329
1330 ;;; Propagate derived type info from the values continuation to the
1331 ;;; vars.
1332 (defun ir1-optimize-mv-bind (node)
1333   (declare (type mv-combination node))
1334   (let ((arg (first (basic-combination-args node)))
1335         (vars (lambda-vars (combination-lambda node))))
1336     (multiple-value-bind (types nvals)
1337         (values-types (continuation-derived-type arg))
1338       (unless (eq nvals :unknown)
1339         (mapc #'(lambda (var type)
1340                   (if (basic-var-sets var)
1341                       (propagate-from-sets var type)
1342                       (propagate-to-refs var type)))
1343                 vars
1344                 (append types
1345                         (make-list (max (- (length vars) nvals) 0)
1346                                    :initial-element (specifier-type 'null))))))
1347     (setf (continuation-reoptimize arg) nil))
1348   (values))
1349
1350 ;;; If possible, convert a general MV call to an MV-BIND. We can do
1351 ;;; this if:
1352 ;;; -- The call has only one argument, and
1353 ;;; -- The function has a known fixed number of arguments, or
1354 ;;; -- The argument yields a known fixed number of values.
1355 ;;;
1356 ;;; What we do is change the function in the MV-CALL to be a lambda
1357 ;;; that "looks like an MV bind", which allows
1358 ;;; IR1-OPTIMIZE-MV-COMBINATION to notice that this call can be
1359 ;;; converted (the next time around.) This new lambda just calls the
1360 ;;; actual function with the MV-BIND variables as arguments. Note that
1361 ;;; this new MV bind is not let-converted immediately, as there are
1362 ;;; going to be stray references from the entry-point functions until
1363 ;;; they get deleted.
1364 ;;;
1365 ;;; In order to avoid loss of argument count checking, we only do the
1366 ;;; transformation according to a known number of expected argument if
1367 ;;; safety is unimportant. We can always convert if we know the number
1368 ;;; of actual values, since the normal call that we build will still
1369 ;;; do any appropriate argument count checking.
1370 ;;;
1371 ;;; We only attempt the transformation if the called function is a
1372 ;;; constant reference. This allows us to just splice the leaf into
1373 ;;; the new function, instead of trying to somehow bind the function
1374 ;;; expression. The leaf must be constant because we are evaluating it
1375 ;;; again in a different place. This also has the effect of squelching
1376 ;;; multiple warnings when there is an argument count error.
1377 (defun ir1-optimize-mv-call (node)
1378   (let ((fun (basic-combination-fun node))
1379         (*compiler-error-context* node)
1380         (ref (continuation-use (basic-combination-fun node)))
1381         (args (basic-combination-args node)))
1382
1383     (unless (and (ref-p ref) (constant-reference-p ref)
1384                  args (null (rest args)))
1385       (return-from ir1-optimize-mv-call))
1386
1387     (multiple-value-bind (min max)
1388         (function-type-nargs (continuation-type fun))
1389       (let ((total-nvals
1390              (multiple-value-bind (types nvals)
1391                  (values-types (continuation-derived-type (first args)))
1392                (declare (ignore types))
1393                (if (eq nvals :unknown) nil nvals))))
1394
1395         (when total-nvals
1396           (when (and min (< total-nvals min))
1397             (compiler-warning
1398              "MULTIPLE-VALUE-CALL with ~R values when the function expects ~
1399              at least ~R."
1400              total-nvals min)
1401             (setf (basic-combination-kind node) :error)
1402             (return-from ir1-optimize-mv-call))
1403           (when (and max (> total-nvals max))
1404             (compiler-warning
1405              "MULTIPLE-VALUE-CALL with ~R values when the function expects ~
1406              at most ~R."
1407              total-nvals max)
1408             (setf (basic-combination-kind node) :error)
1409             (return-from ir1-optimize-mv-call)))
1410
1411         (let ((count (cond (total-nvals)
1412                            ((and (policy node (zerop safety))
1413                                  (eql min max))
1414                             min)
1415                            (t nil))))
1416           (when count
1417             (with-ir1-environment node
1418               (let* ((dums (make-gensym-list count))
1419                      (ignore (gensym))
1420                      (fun (ir1-convert-lambda
1421                            `(lambda (&optional ,@dums &rest ,ignore)
1422                               (declare (ignore ,ignore))
1423                               (funcall ,(ref-leaf ref) ,@dums)))))
1424                 (change-ref-leaf ref fun)
1425                 (aver (eq (basic-combination-kind node) :full))
1426                 (local-call-analyze *current-component*)
1427                 (aver (eq (basic-combination-kind node) :local)))))))))
1428   (values))
1429
1430 ;;; If we see:
1431 ;;;    (multiple-value-bind
1432 ;;;     (x y)
1433 ;;;     (values xx yy)
1434 ;;;      ...)
1435 ;;; Convert to:
1436 ;;;    (let ((x xx)
1437 ;;;       (y yy))
1438 ;;;      ...)
1439 ;;;
1440 ;;; What we actually do is convert the VALUES combination into a
1441 ;;; normal LET combination calling the original :MV-LET lambda. If
1442 ;;; there are extra args to VALUES, discard the corresponding
1443 ;;; continuations. If there are insufficient args, insert references
1444 ;;; to NIL.
1445 (defun convert-mv-bind-to-let (call)
1446   (declare (type mv-combination call))
1447   (let* ((arg (first (basic-combination-args call)))
1448          (use (continuation-use arg)))
1449     (when (and (combination-p use)
1450                (eq (continuation-function-name (combination-fun use))
1451                    'values))
1452       (let* ((fun (combination-lambda call))
1453              (vars (lambda-vars fun))
1454              (vals (combination-args use))
1455              (nvars (length vars))
1456              (nvals (length vals)))
1457         (cond ((> nvals nvars)
1458                (mapc #'flush-dest (subseq vals nvars))
1459                (setq vals (subseq vals 0 nvars)))
1460               ((< nvals nvars)
1461                (with-ir1-environment use
1462                  (let ((node-prev (node-prev use)))
1463                    (setf (node-prev use) nil)
1464                    (setf (continuation-next node-prev) nil)
1465                    (collect ((res vals))
1466                      (loop as cont = (make-continuation use)
1467                            and prev = node-prev then cont
1468                            repeat (- nvars nvals)
1469                            do (reference-constant prev cont nil)
1470                               (res cont))
1471                      (setq vals (res)))
1472                    (prev-link use (car (last vals)))))))
1473         (setf (combination-args use) vals)
1474         (flush-dest (combination-fun use))
1475         (let ((fun-cont (basic-combination-fun call)))
1476           (setf (continuation-dest fun-cont) use)
1477           (setf (combination-fun use) fun-cont))
1478         (setf (combination-kind use) :local)
1479         (setf (functional-kind fun) :let)
1480         (flush-dest (first (basic-combination-args call)))
1481         (unlink-node call)
1482         (when vals
1483           (reoptimize-continuation (first vals)))
1484         (propagate-to-args use fun))
1485       t)))
1486
1487 ;;; If we see:
1488 ;;;    (values-list (list x y z))
1489 ;;;
1490 ;;; Convert to:
1491 ;;;    (values x y z)
1492 ;;;
1493 ;;; In implementation, this is somewhat similar to
1494 ;;; CONVERT-MV-BIND-TO-LET. We grab the args of LIST and make them
1495 ;;; args of the VALUES-LIST call, flushing the old argument
1496 ;;; continuation (allowing the LIST to be flushed.)
1497 (defoptimizer (values-list optimizer) ((list) node)
1498   (let ((use (continuation-use list)))
1499     (when (and (combination-p use)
1500                (eq (continuation-function-name (combination-fun use))
1501                    'list))
1502       (change-ref-leaf (continuation-use (combination-fun node))
1503                        (find-free-function 'values "in a strange place"))
1504       (setf (combination-kind node) :full)
1505       (let ((args (combination-args use)))
1506         (dolist (arg args)
1507           (setf (continuation-dest arg) node))
1508         (setf (combination-args use) nil)
1509         (flush-dest list)
1510         (setf (combination-args node) args))
1511       t)))
1512
1513 ;;; If VALUES appears in a non-MV context, then effectively convert it
1514 ;;; to a PROG1. This allows the computation of the additional values
1515 ;;; to become dead code.
1516 (deftransform values ((&rest vals) * * :node node)
1517   (when (typep (continuation-dest (node-cont node))
1518                '(or creturn exit mv-combination))
1519     (give-up-ir1-transform))
1520   (setf (node-derived-type node) *wild-type*)
1521   (if vals
1522       (let ((dummies (make-gensym-list (length (cdr vals)))))
1523         `(lambda (val ,@dummies)
1524            (declare (ignore ,@dummies))
1525            val))
1526       nil))