1.0.23.38: fix bug 430 (stack alloc by nested defstruct constructors)
[sbcl.git] / src / compiler / ir1util.lisp
1 ;;;; This file contains miscellaneous utilities used for manipulating
2 ;;;; the IR1 representation.
3
4 ;;;; This software is part of the SBCL system. See the README file for
5 ;;;; more information.
6 ;;;;
7 ;;;; This software is derived from the CMU CL system, which was
8 ;;;; written at Carnegie Mellon University and released into the
9 ;;;; public domain. The software is in the public domain and is
10 ;;;; provided with absolutely no warranty. See the COPYING and CREDITS
11 ;;;; files for more information.
12
13 (in-package "SB!C")
14 \f
15 ;;;; cleanup hackery
16
17 ;;; Return the innermost cleanup enclosing NODE, or NIL if there is
18 ;;; none in its function. If NODE has no cleanup, but is in a LET,
19 ;;; then we must still check the environment that the call is in.
20 (defun node-enclosing-cleanup (node)
21   (declare (type node node))
22   (do ((lexenv (node-lexenv node)
23                (lambda-call-lexenv (lexenv-lambda lexenv))))
24       ((null lexenv) nil)
25     (let ((cup (lexenv-cleanup lexenv)))
26       (when cup (return cup)))))
27
28 ;;; Convert the FORM in a block inserted between BLOCK1 and BLOCK2 as
29 ;;; an implicit MV-PROG1. The inserted block is returned. NODE is used
30 ;;; for IR1 context when converting the form. Note that the block is
31 ;;; not assigned a number, and is linked into the DFO at the
32 ;;; beginning. We indicate that we have trashed the DFO by setting
33 ;;; COMPONENT-REANALYZE. If CLEANUP is supplied, then convert with
34 ;;; that cleanup.
35 (defun insert-cleanup-code (block1 block2 node form &optional cleanup)
36   (declare (type cblock block1 block2) (type node node)
37            (type (or cleanup null) cleanup))
38   (setf (component-reanalyze (block-component block1)) t)
39   (with-ir1-environment-from-node node
40     (with-component-last-block (*current-component*
41                                 (block-next (component-head *current-component*)))
42       (let* ((start (make-ctran))
43              (block (ctran-starts-block start))
44              (next (make-ctran))
45              (*lexenv* (if cleanup
46                            (make-lexenv :cleanup cleanup)
47                            *lexenv*)))
48         (change-block-successor block1 block2 block)
49         (link-blocks block block2)
50         (ir1-convert start next nil form)
51         (setf (block-last block) (ctran-use next))
52         (setf (node-next (block-last block)) nil)
53         block))))
54 \f
55 ;;;; lvar use hacking
56
57 ;;; Return a list of all the nodes which use LVAR.
58 (declaim (ftype (sfunction (lvar) list) find-uses))
59 (defun find-uses (lvar)
60   (let ((uses (lvar-uses lvar)))
61     (if (listp uses)
62         uses
63         (list uses))))
64
65 (declaim (ftype (sfunction (lvar) lvar) principal-lvar))
66 (defun principal-lvar (lvar)
67   (labels ((pl (lvar)
68              (let ((use (lvar-uses lvar)))
69                (if (cast-p use)
70                    (pl (cast-value use))
71                    lvar))))
72     (pl lvar)))
73
74 (defun principal-lvar-use (lvar)
75   (labels ((plu (lvar)
76              (declare (type lvar lvar))
77              (let ((use (lvar-uses lvar)))
78                (if (cast-p use)
79                    (plu (cast-value use))
80                    use))))
81     (plu lvar)))
82
83 ;;; Update lvar use information so that NODE is no longer a use of its
84 ;;; LVAR.
85 ;;;
86 ;;; Note: if you call this function, you may have to do a
87 ;;; REOPTIMIZE-LVAR to inform IR1 optimization that something has
88 ;;; changed.
89 (declaim (ftype (sfunction (node) (values))
90                 delete-lvar-use
91                 %delete-lvar-use))
92 ;;; Just delete NODE from its LVAR uses; LVAR is preserved so it may
93 ;;; be given a new use.
94 (defun %delete-lvar-use (node)
95   (let ((lvar (node-lvar node)))
96     (when lvar
97       (if (listp (lvar-uses lvar))
98           (let ((new-uses (delq node (lvar-uses lvar))))
99             (setf (lvar-uses lvar)
100                   (if (singleton-p new-uses)
101                       (first new-uses)
102                       new-uses)))
103           (setf (lvar-uses lvar) nil))
104       (setf (node-lvar node) nil)))
105   (values))
106 ;;; Delete NODE from its LVAR uses; if LVAR has no other uses, delete
107 ;;; its DEST's block, which must be unreachable.
108 (defun delete-lvar-use (node)
109   (let ((lvar (node-lvar node)))
110     (when lvar
111       (%delete-lvar-use node)
112       (if (null (lvar-uses lvar))
113           (binding* ((dest (lvar-dest lvar) :exit-if-null)
114                      (() (not (node-deleted dest)) :exit-if-null)
115                      (block (node-block dest)))
116             (mark-for-deletion block))
117           (reoptimize-lvar lvar))))
118   (values))
119
120 ;;; Update lvar use information so that NODE uses LVAR.
121 ;;;
122 ;;; Note: if you call this function, you may have to do a
123 ;;; REOPTIMIZE-LVAR to inform IR1 optimization that something has
124 ;;; changed.
125 (declaim (ftype (sfunction (node (or lvar null)) (values)) add-lvar-use))
126 (defun add-lvar-use (node lvar)
127   (aver (not (node-lvar node)))
128   (when lvar
129     (let ((uses (lvar-uses lvar)))
130       (setf (lvar-uses lvar)
131             (cond ((null uses)
132                    node)
133                   ((listp uses)
134                    (cons node uses))
135                   (t
136                    (list node uses))))
137       (setf (node-lvar node) lvar)))
138
139   (values))
140
141 ;;; Return true if LVAR destination is executed immediately after
142 ;;; NODE. Cleanups are ignored.
143 (defun immediately-used-p (lvar node)
144   (declare (type lvar lvar) (type node node))
145   (aver (eq (node-lvar node) lvar))
146   (let ((dest (lvar-dest lvar)))
147     (acond ((node-next node)
148             (eq (ctran-next it) dest))
149            (t (eq (block-start (first (block-succ (node-block node))))
150                   (node-prev dest))))))
151 \f
152 ;;;; lvar substitution
153
154 ;;; In OLD's DEST, replace OLD with NEW. NEW's DEST must initially be
155 ;;; NIL. We do not flush OLD's DEST.
156 (defun substitute-lvar (new old)
157   (declare (type lvar old new))
158   (aver (not (lvar-dest new)))
159   (let ((dest (lvar-dest old)))
160     (etypecase dest
161       ((or ref bind))
162       (cif (setf (if-test dest) new))
163       (cset (setf (set-value dest) new))
164       (creturn (setf (return-result dest) new))
165       (exit (setf (exit-value dest) new))
166       (basic-combination
167        (if (eq old (basic-combination-fun dest))
168            (setf (basic-combination-fun dest) new)
169            (setf (basic-combination-args dest)
170                  (nsubst new old (basic-combination-args dest)))))
171       (cast (setf (cast-value dest) new)))
172
173     (setf (lvar-dest old) nil)
174     (setf (lvar-dest new) dest)
175     (flush-lvar-externally-checkable-type new))
176   (values))
177
178 ;;; Replace all uses of OLD with uses of NEW, where NEW has an
179 ;;; arbitary number of uses. NEW is supposed to be "later" than OLD.
180 (defun substitute-lvar-uses (new old propagate-dx)
181   (declare (type lvar old)
182            (type (or lvar null) new)
183            (type boolean propagate-dx))
184
185   (cond (new
186          (do-uses (node old)
187            (%delete-lvar-use node)
188            (add-lvar-use node new))
189          (reoptimize-lvar new)
190          (awhen (and propagate-dx (lvar-dynamic-extent old))
191            (setf (lvar-dynamic-extent old) nil)
192            (unless (lvar-dynamic-extent new)
193              (setf (lvar-dynamic-extent new) it)
194              (setf (cleanup-info it) (substitute new old (cleanup-info it)))))
195          (when (lvar-dynamic-extent new)
196            (do-uses (node new)
197              (node-ends-block node))))
198         (t (flush-dest old)))
199
200   (values))
201 \f
202 ;;;; block starting/creation
203
204 ;;; Return the block that CTRAN is the start of, making a block if
205 ;;; necessary. This function is called by IR1 translators which may
206 ;;; cause a CTRAN to be used more than once. Every CTRAN which may be
207 ;;; used more than once must start a block by the time that anyone
208 ;;; does a USE-CTRAN on it.
209 ;;;
210 ;;; We also throw the block into the next/prev list for the
211 ;;; *CURRENT-COMPONENT* so that we keep track of which blocks we have
212 ;;; made.
213 (defun ctran-starts-block (ctran)
214   (declare (type ctran ctran))
215   (ecase (ctran-kind ctran)
216     (:unused
217      (aver (not (ctran-block ctran)))
218      (let* ((next (component-last-block *current-component*))
219             (prev (block-prev next))
220             (new-block (make-block ctran)))
221        (setf (block-next new-block) next
222              (block-prev new-block) prev
223              (block-prev next) new-block
224              (block-next prev) new-block
225              (ctran-block ctran) new-block
226              (ctran-kind ctran) :block-start)
227        (aver (not (ctran-use ctran)))
228        new-block))
229     (:block-start
230      (ctran-block ctran))))
231
232 ;;; Ensure that CTRAN is the start of a block so that the use set can
233 ;;; be freely manipulated.
234 (defun ensure-block-start (ctran)
235   (declare (type ctran ctran))
236   (let ((kind (ctran-kind ctran)))
237     (ecase kind
238       ((:block-start))
239       ((:unused)
240        (setf (ctran-block ctran)
241              (make-block-key :start ctran))
242        (setf (ctran-kind ctran) :block-start))
243       ((:inside-block)
244        (node-ends-block (ctran-use ctran)))))
245   (values))
246
247 ;;; CTRAN must be the last ctran in an incomplete block; finish the
248 ;;; block and start a new one if necessary.
249 (defun start-block (ctran)
250   (declare (type ctran ctran))
251   (aver (not (ctran-next ctran)))
252   (ecase (ctran-kind ctran)
253     (:inside-block
254      (let ((block (ctran-block ctran))
255            (node (ctran-use ctran)))
256        (aver (not (block-last block)))
257        (aver node)
258        (setf (block-last block) node)
259        (setf (node-next node) nil)
260        (setf (ctran-use ctran) nil)
261        (setf (ctran-kind ctran) :unused)
262        (setf (ctran-block ctran) nil)
263        (link-blocks block (ctran-starts-block ctran))))
264     (:block-start)))
265 \f
266 ;;;;
267
268 ;;; Filter values of LVAR through FORM, which must be an ordinary/mv
269 ;;; call. First argument must be 'DUMMY, which will be replaced with
270 ;;; LVAR. In case of an ordinary call the function should not have
271 ;;; return type NIL. We create a new "filtered" lvar.
272 ;;;
273 ;;; TODO: remove preconditions.
274 (defun filter-lvar (lvar form)
275   (declare (type lvar lvar) (type list form))
276   (let* ((dest (lvar-dest lvar))
277          (ctran (node-prev dest)))
278     (with-ir1-environment-from-node dest
279
280       (ensure-block-start ctran)
281       (let* ((old-block (ctran-block ctran))
282              (new-start (make-ctran))
283              (filtered-lvar (make-lvar))
284              (new-block (ctran-starts-block new-start)))
285
286         ;; Splice in the new block before DEST, giving the new block
287         ;; all of DEST's predecessors.
288         (dolist (block (block-pred old-block))
289           (change-block-successor block old-block new-block))
290
291         (ir1-convert new-start ctran filtered-lvar form)
292
293         ;; KLUDGE: Comments at the head of this function in CMU CL
294         ;; said that somewhere in here we
295         ;;   Set the new block's start and end cleanups to the *start*
296         ;;   cleanup of PREV's block. This overrides the incorrect
297         ;;   default from WITH-IR1-ENVIRONMENT-FROM-NODE.
298         ;; Unfortunately I can't find any code which corresponds to this.
299         ;; Perhaps it was a stale comment? Or perhaps I just don't
300         ;; understand.. -- WHN 19990521
301
302         ;; Replace 'DUMMY with the LVAR. (We can find 'DUMMY because
303         ;; no LET conversion has been done yet.) The [mv-]combination
304         ;; code from the call in the form will be the use of the new
305         ;; check lvar. We substitute for the first argument of
306         ;; this node.
307         (let* ((node (lvar-use filtered-lvar))
308                (args (basic-combination-args node))
309                (victim (first args)))
310           (aver (eq (constant-value (ref-leaf (lvar-use victim)))
311                     'dummy))
312
313           (substitute-lvar filtered-lvar lvar)
314           (substitute-lvar lvar victim)
315           (flush-dest victim))
316
317         ;; Invoking local call analysis converts this call to a LET.
318         (locall-analyze-component *current-component*))))
319   (values))
320
321 ;;; Delete NODE and VALUE. It may result in some calls becoming tail.
322 (defun delete-filter (node lvar value)
323   (aver (eq (lvar-dest value) node))
324   (aver (eq (node-lvar node) lvar))
325   (cond (lvar (collect ((merges))
326                 (when (return-p (lvar-dest lvar))
327                   (do-uses (use value)
328                     (when (and (basic-combination-p use)
329                                (eq (basic-combination-kind use) :local))
330                       (merges use))))
331                 (substitute-lvar-uses lvar value
332                                       (and lvar (eq (lvar-uses lvar) node)))
333                 (%delete-lvar-use node)
334                 (prog1
335                     (unlink-node node)
336                   (dolist (merge (merges))
337                     (merge-tail-sets merge)))))
338         (t (flush-dest value)
339            (unlink-node node))))
340
341 ;;; Make a CAST and insert it into IR1 before node NEXT.
342 (defun insert-cast-before (next lvar type policy)
343   (declare (type node next) (type lvar lvar) (type ctype type))
344   (with-ir1-environment-from-node next
345     (let* ((ctran (node-prev next))
346            (cast (make-cast lvar type policy))
347            (internal-ctran (make-ctran)))
348       (setf (ctran-next ctran) cast
349             (node-prev cast) ctran)
350       (use-ctran cast internal-ctran)
351       (link-node-to-previous-ctran next internal-ctran)
352       (setf (lvar-dest lvar) cast)
353       (reoptimize-lvar lvar)
354       (when (return-p next)
355         (node-ends-block cast))
356       (setf (block-attributep (block-flags (node-block cast))
357                               type-check type-asserted)
358             t)
359       cast)))
360 \f
361 ;;;; miscellaneous shorthand functions
362
363 ;;; Return the home (i.e. enclosing non-LET) CLAMBDA for NODE. Since
364 ;;; the LEXENV-LAMBDA may be deleted, we must chain up the
365 ;;; LAMBDA-CALL-LEXENV thread until we find a CLAMBDA that isn't
366 ;;; deleted, and then return its home.
367 (defun node-home-lambda (node)
368   (declare (type node node))
369   (do ((fun (lexenv-lambda (node-lexenv node))
370             (lexenv-lambda (lambda-call-lexenv fun))))
371       ((not (memq (functional-kind fun) '(:deleted :zombie)))
372        (lambda-home fun))
373     (when (eq (lambda-home fun) fun)
374       (return fun))))
375
376 #!-sb-fluid (declaim (inline node-block))
377 (defun node-block (node)
378   (ctran-block (node-prev node)))
379 (declaim (ftype (sfunction (node) component) node-component))
380 (defun node-component (node)
381   (block-component (node-block node)))
382 (declaim (ftype (sfunction (node) physenv) node-physenv))
383 (defun node-physenv (node)
384   (lambda-physenv (node-home-lambda node)))
385 #!-sb-fluid (declaim (inline node-dest))
386 (defun node-dest (node)
387   (awhen (node-lvar node) (lvar-dest it)))
388
389 #!-sb-fluid (declaim (inline node-stack-allocate-p))
390 (defun node-stack-allocate-p (node)
391   (awhen (node-lvar node)
392     (lvar-dynamic-extent it)))
393
394 (declaim (ftype (sfunction (node (member nil t :truly) &optional (or null component))
395                            boolean) use-good-for-dx-p))
396 (declaim (ftype (sfunction (lvar (member nil t :truly) &optional (or null component))
397                            boolean) lvar-good-for-dx-p))
398 (defun use-good-for-dx-p (use dx &optional component)
399   ;; FIXME: Can casts point to LVARs in other components?
400   ;; RECHECK-DYNAMIC-EXTENT-LVARS assumes that they can't -- that is, that the
401   ;; PRINCIPAL-LVAR is always in the same component as the original one. It
402   ;; would be either good to have an explanation of why casts don't point
403   ;; across components, or an explanation of when they do it. ...in the
404   ;; meanwhile AVER that our assumption holds true.
405   (aver (or (not component) (eq component (node-component use))))
406   (or (dx-combination-p use dx)
407       (and (cast-p use)
408            (not (cast-type-check use))
409            (lvar-good-for-dx-p (cast-value use) dx component))
410       (and (trivial-lambda-var-ref-p use)
411            (let ((uses (lvar-uses (trivial-lambda-var-ref-lvar use))))
412              (or (eq use uses)
413                  (lvar-good-for-dx-p (trivial-lambda-var-ref-lvar use) dx component))))))
414
415 (defun lvar-good-for-dx-p (lvar dx &optional component)
416   (let ((uses (lvar-uses lvar)))
417     (if (listp uses)
418         (every (lambda (use)
419                  (use-good-for-dx-p use dx component))
420                uses)
421         (use-good-for-dx-p uses dx component))))
422
423 (defun known-dx-combination-p (use dx)
424   (and (eq (combination-kind use) :known)
425        (awhen (fun-info-stack-allocate-result (combination-fun-info use))
426          (funcall it use dx))))
427
428 (defun dx-combination-p (use dx)
429   (and (combination-p use)
430        (or
431         ;; Known, and can do DX.
432         (known-dx-combination-p use dx)
433         ;; Possibly a not-yet-eliminated lambda which ends up returning the
434         ;; results of an actual known DX combination.
435         (let* ((fun (combination-fun use))
436                (ref (principal-lvar-use fun))
437                (clambda (when (ref-p ref)
438                           (ref-leaf ref)))
439                (creturn (when (lambda-p clambda)
440                           (lambda-return clambda)))
441                (result-use (when (return-p creturn)
442                              (principal-lvar-use (return-result creturn)))))
443           (when result-use
444             (if (known-dx-combination-p result-use dx)
445                 (combination-args-flow-cleanly-p use result-use dx)
446                 (dx-combination-p result-use dx)))))
447        t))
448
449 (defun combination-args-flow-cleanly-p (combination1 combination2 dx)
450   (labels ((recurse (combination)
451              (or (eq combination combination2)
452                  (if (known-dx-combination-p combination dx)
453                      (let ((dest (lvar-dest (combination-lvar combination))))
454                        (and (combination-p dest)
455                             (recurse dest)))
456                      (let* ((fun1 (combination-fun combination))
457                             (ref1 (principal-lvar-use fun1))
458                             (clambda1 (when (ref-p ref1) (ref-leaf ref1))))
459                        (when (lambda-p clambda1)
460                          (dolist (var (lambda-vars clambda1) t)
461                            (dolist (var-ref (lambda-var-refs var))
462                              (let ((dest (lvar-dest (ref-lvar var-ref))))
463                                (unless (and (combination-p dest) (recurse dest))
464                                  (return-from combination-args-flow-cleanly-p nil)))))))))))
465     (recurse combination1)))
466
467 (defun trivial-lambda-var-ref-p (use)
468   (and (ref-p use)
469        (let ((var (ref-leaf use)))
470          ;; lambda-var, no SETS
471          (when (and (lambda-var-p var) (not (lambda-var-sets var)))
472            (let ((home (lambda-var-home var))
473                  (refs (lambda-var-refs var)))
474              ;; bound by a system lambda, no other REFS
475              (when (and (lambda-system-lambda-p home)
476                         (eq use (car refs)) (not (cdr refs)))
477                ;; the LAMBDA this var is bound by has only a single REF, going
478                ;; to a combination
479                (let* ((lambda-refs (lambda-refs home))
480                       (primary (car lambda-refs)))
481                  (and (ref-p primary)
482                       (not (cdr lambda-refs))
483                       (combination-p (lvar-dest (ref-lvar primary)))))))))))
484
485 (defun trivial-lambda-var-ref-lvar (use)
486   (let* ((this (ref-leaf use))
487          (home (lambda-var-home this)))
488     (multiple-value-bind (fun vars)
489         (values home (lambda-vars home))
490       (let* ((combination (lvar-dest (ref-lvar (car (lambda-refs fun)))))
491              (args (combination-args combination)))
492         (assert (= (length vars) (length args)))
493         (loop for var in vars
494               for arg in args
495               when (eq var this)
496               return arg)))))
497
498 (declaim (inline block-to-be-deleted-p))
499 (defun block-to-be-deleted-p (block)
500   (or (block-delete-p block)
501       (eq (functional-kind (block-home-lambda block)) :deleted)))
502
503 ;;; Checks whether NODE is in a block to be deleted
504 (declaim (inline node-to-be-deleted-p))
505 (defun node-to-be-deleted-p (node)
506   (block-to-be-deleted-p (node-block node)))
507
508 (declaim (ftype (sfunction (clambda) cblock) lambda-block))
509 (defun lambda-block (clambda)
510   (node-block (lambda-bind clambda)))
511 (declaim (ftype (sfunction (clambda) component) lambda-component))
512 (defun lambda-component (clambda)
513   (block-component (lambda-block clambda)))
514
515 (declaim (ftype (sfunction (cblock) node) block-start-node))
516 (defun block-start-node (block)
517   (ctran-next (block-start block)))
518
519 ;;; Return the enclosing cleanup for environment of the first or last
520 ;;; node in BLOCK.
521 (defun block-start-cleanup (block)
522   (node-enclosing-cleanup (block-start-node block)))
523 (defun block-end-cleanup (block)
524   (node-enclosing-cleanup (block-last block)))
525
526 ;;; Return the non-LET LAMBDA that holds BLOCK's code, or NIL
527 ;;; if there is none.
528 ;;;
529 ;;; There can legitimately be no home lambda in dead code early in the
530 ;;; IR1 conversion process, e.g. when IR1-converting the SETQ form in
531 ;;;   (BLOCK B (RETURN-FROM B) (SETQ X 3))
532 ;;; where the block is just a placeholder during parsing and doesn't
533 ;;; actually correspond to code which will be written anywhere.
534 (declaim (ftype (sfunction (cblock) (or clambda null)) block-home-lambda-or-null))
535 (defun block-home-lambda-or-null (block)
536   (if (node-p (block-last block))
537       ;; This is the old CMU CL way of doing it.
538       (node-home-lambda (block-last block))
539       ;; Now that SBCL uses this operation more aggressively than CMU
540       ;; CL did, the old CMU CL way of doing it can fail in two ways.
541       ;;   1. It can fail in a few cases even when a meaningful home
542       ;;      lambda exists, e.g. in IR1-CONVERT of one of the legs of
543       ;;      an IF.
544       ;;   2. It can fail when converting a form which is born orphaned
545       ;;      so that it never had a meaningful home lambda, e.g. a form
546       ;;      which follows a RETURN-FROM or GO form.
547       (let ((pred-list (block-pred block)))
548         ;; To deal with case 1, we reason that
549         ;; previous-in-target-execution-order blocks should be in the
550         ;; same lambda, and that they seem in practice to be
551         ;; previous-in-compilation-order blocks too, so we look back
552         ;; to find one which is sufficiently initialized to tell us
553         ;; what the home lambda is.
554         (if pred-list
555             ;; We could get fancy about this, flooding through the
556             ;; graph of all the previous blocks, but in practice it
557             ;; seems to work just to grab the first previous block and
558             ;; use it.
559             (node-home-lambda (block-last (first pred-list)))
560             ;; In case 2, we end up with an empty PRED-LIST and
561             ;; have to punt: There's no home lambda.
562             nil))))
563
564 ;;; Return the non-LET LAMBDA that holds BLOCK's code.
565 (declaim (ftype (sfunction (cblock) clambda) block-home-lambda))
566 (defun block-home-lambda (block)
567   (block-home-lambda-or-null block))
568
569 ;;; Return the IR1 physical environment for BLOCK.
570 (declaim (ftype (sfunction (cblock) physenv) block-physenv))
571 (defun block-physenv (block)
572   (lambda-physenv (block-home-lambda block)))
573
574 ;;; Return the Top Level Form number of PATH, i.e. the ordinal number
575 ;;; of its original source's top level form in its compilation unit.
576 (defun source-path-tlf-number (path)
577   (declare (list path))
578   (car (last path)))
579
580 ;;; Return the (reversed) list for the PATH in the original source
581 ;;; (with the Top Level Form number last).
582 (defun source-path-original-source (path)
583   (declare (list path) (inline member))
584   (cddr (member 'original-source-start path :test #'eq)))
585
586 ;;; Return the Form Number of PATH's original source inside the Top
587 ;;; Level Form that contains it. This is determined by the order that
588 ;;; we walk the subforms of the top level source form.
589 (defun source-path-form-number (path)
590   (declare (list path) (inline member))
591   (cadr (member 'original-source-start path :test #'eq)))
592
593 ;;; Return a list of all the enclosing forms not in the original
594 ;;; source that converted to get to this form, with the immediate
595 ;;; source for node at the start of the list.
596 (defun source-path-forms (path)
597   (subseq path 0 (position 'original-source-start path)))
598
599 ;;; Return the innermost source form for NODE.
600 (defun node-source-form (node)
601   (declare (type node node))
602   (let* ((path (node-source-path node))
603          (forms (source-path-forms path)))
604     (if forms
605         (first forms)
606         (values (find-original-source path)))))
607
608 ;;; Return NODE-SOURCE-FORM, T if lvar has a single use, otherwise
609 ;;; NIL, NIL.
610 (defun lvar-source (lvar)
611   (let ((use (lvar-uses lvar)))
612     (if (listp use)
613         (values nil nil)
614         (values (node-source-form use) t))))
615
616 ;;; Return the unique node, delivering a value to LVAR.
617 #!-sb-fluid (declaim (inline lvar-use))
618 (defun lvar-use (lvar)
619   (the (not list) (lvar-uses lvar)))
620
621 #!-sb-fluid (declaim (inline lvar-has-single-use-p))
622 (defun lvar-has-single-use-p (lvar)
623   (typep (lvar-uses lvar) '(not list)))
624
625 ;;; Return the LAMBDA that is CTRAN's home, or NIL if there is none.
626 (declaim (ftype (sfunction (ctran) (or clambda null))
627                 ctran-home-lambda-or-null))
628 (defun ctran-home-lambda-or-null (ctran)
629   ;; KLUDGE: This function is a post-CMU-CL hack by WHN, and this
630   ;; implementation might not be quite right, or might be uglier than
631   ;; necessary. It appears that the original Python never found a need
632   ;; to do this operation. The obvious things based on
633   ;; NODE-HOME-LAMBDA of CTRAN-USE usually work; then if that fails,
634   ;; BLOCK-HOME-LAMBDA of CTRAN-BLOCK works, given that we
635   ;; generalize it enough to grovel harder when the simple CMU CL
636   ;; approach fails, and furthermore realize that in some exceptional
637   ;; cases it might return NIL. -- WHN 2001-12-04
638   (cond ((ctran-use ctran)
639          (node-home-lambda (ctran-use ctran)))
640         ((ctran-block ctran)
641          (block-home-lambda-or-null (ctran-block ctran)))
642         (t
643          (bug "confused about home lambda for ~S" ctran))))
644
645 ;;; Return the LAMBDA that is CTRAN's home.
646 (declaim (ftype (sfunction (ctran) clambda) ctran-home-lambda))
647 (defun ctran-home-lambda (ctran)
648   (ctran-home-lambda-or-null ctran))
649
650 (declaim (inline cast-single-value-p))
651 (defun cast-single-value-p (cast)
652   (not (values-type-p (cast-asserted-type cast))))
653
654 #!-sb-fluid (declaim (inline lvar-single-value-p))
655 (defun lvar-single-value-p (lvar)
656   (or (not lvar)
657       (let ((dest (lvar-dest lvar)))
658         (typecase dest
659           ((or creturn exit)
660            nil)
661           (mv-combination
662            (eq (basic-combination-fun dest) lvar))
663           (cast
664            (locally
665                (declare (notinline lvar-single-value-p))
666              (and (cast-single-value-p dest)
667                   (lvar-single-value-p (node-lvar dest)))))
668           (t
669            t)))))
670
671 (defun principal-lvar-end (lvar)
672   (loop for prev = lvar then (node-lvar dest)
673         for dest = (and prev (lvar-dest prev))
674         while (cast-p dest)
675         finally (return (values dest prev))))
676
677 (defun principal-lvar-single-valuify (lvar)
678   (loop for prev = lvar then (node-lvar dest)
679         for dest = (and prev (lvar-dest prev))
680         while (cast-p dest)
681         do (setf (node-derived-type dest)
682                  (make-short-values-type (list (single-value-type
683                                                 (node-derived-type dest)))))
684         (reoptimize-lvar prev)))
685 \f
686 ;;; Return a new LEXENV just like DEFAULT except for the specified
687 ;;; slot values. Values for the alist slots are NCONCed to the
688 ;;; beginning of the current value, rather than replacing it entirely.
689 (defun make-lexenv (&key (default *lexenv*)
690                          funs vars blocks tags
691                          type-restrictions
692                          (lambda (lexenv-lambda default))
693                          (cleanup (lexenv-cleanup default))
694                          (handled-conditions (lexenv-handled-conditions default))
695                          (disabled-package-locks
696                           (lexenv-disabled-package-locks default))
697                          (policy (lexenv-policy default)))
698   (macrolet ((frob (var slot)
699                `(let ((old (,slot default)))
700                   (if ,var
701                       (nconc ,var old)
702                       old))))
703     (internal-make-lexenv
704      (frob funs lexenv-funs)
705      (frob vars lexenv-vars)
706      (frob blocks lexenv-blocks)
707      (frob tags lexenv-tags)
708      (frob type-restrictions lexenv-type-restrictions)
709      lambda cleanup handled-conditions
710      disabled-package-locks policy)))
711
712 ;;; Makes a LEXENV, suitable for using in a MACROLET introduced
713 ;;; macroexpander
714 (defun make-restricted-lexenv (lexenv)
715   (flet ((fun-good-p (fun)
716            (destructuring-bind (name . thing) fun
717              (declare (ignore name))
718              (etypecase thing
719                (functional nil)
720                (global-var t)
721                (cons (aver (eq (car thing) 'macro))
722                      t))))
723          (var-good-p (var)
724            (destructuring-bind (name . thing) var
725              (declare (ignore name))
726              (etypecase thing
727                ;; The evaluator will mark lexicals with :BOGUS when it
728                ;; translates an interpreter lexenv to a compiler
729                ;; lexenv.
730                ((or leaf #!+sb-eval (member :bogus)) nil)
731                (cons (aver (eq (car thing) 'macro))
732                      t)
733                (heap-alien-info nil)))))
734     (internal-make-lexenv
735      (remove-if-not #'fun-good-p (lexenv-funs lexenv))
736      (remove-if-not #'var-good-p (lexenv-vars lexenv))
737      nil
738      nil
739      (lexenv-type-restrictions lexenv) ; XXX
740      nil
741      nil
742      (lexenv-handled-conditions lexenv)
743      (lexenv-disabled-package-locks lexenv)
744      (lexenv-policy lexenv))))
745 \f
746 ;;;; flow/DFO/component hackery
747
748 ;;; Join BLOCK1 and BLOCK2.
749 (defun link-blocks (block1 block2)
750   (declare (type cblock block1 block2))
751   (setf (block-succ block1)
752         (if (block-succ block1)
753             (%link-blocks block1 block2)
754             (list block2)))
755   (push block1 (block-pred block2))
756   (values))
757 (defun %link-blocks (block1 block2)
758   (declare (type cblock block1 block2))
759   (let ((succ1 (block-succ block1)))
760     (aver (not (memq block2 succ1)))
761     (cons block2 succ1)))
762
763 ;;; This is like LINK-BLOCKS, but we separate BLOCK1 and BLOCK2. If
764 ;;; this leaves a successor with a single predecessor that ends in an
765 ;;; IF, then set BLOCK-TEST-MODIFIED so that any test constraint will
766 ;;; now be able to be propagated to the successor.
767 (defun unlink-blocks (block1 block2)
768   (declare (type cblock block1 block2))
769   (let ((succ1 (block-succ block1)))
770     (if (eq block2 (car succ1))
771         (setf (block-succ block1) (cdr succ1))
772         (do ((succ (cdr succ1) (cdr succ))
773              (prev succ1 succ))
774             ((eq (car succ) block2)
775              (setf (cdr prev) (cdr succ)))
776           (aver succ))))
777
778   (let ((new-pred (delq block1 (block-pred block2))))
779     (setf (block-pred block2) new-pred)
780     (when (singleton-p new-pred)
781       (let ((pred-block (first new-pred)))
782         (when (if-p (block-last pred-block))
783           (setf (block-test-modified pred-block) t)))))
784   (values))
785
786 ;;; Swing the succ/pred link between BLOCK and OLD to be between BLOCK
787 ;;; and NEW. If BLOCK ends in an IF, then we have to fix up the
788 ;;; consequent/alternative blocks to point to NEW. We also set
789 ;;; BLOCK-TEST-MODIFIED so that any test constraint will be applied to
790 ;;; the new successor.
791 (defun change-block-successor (block old new)
792   (declare (type cblock new old block))
793   (unlink-blocks block old)
794   (let ((last (block-last block))
795         (comp (block-component block)))
796     (setf (component-reanalyze comp) t)
797     (typecase last
798       (cif
799        (setf (block-test-modified block) t)
800        (let* ((succ-left (block-succ block))
801               (new (if (and (eq new (component-tail comp))
802                             succ-left)
803                        (first succ-left)
804                        new)))
805          (unless (memq new succ-left)
806            (link-blocks block new))
807          (macrolet ((frob (slot)
808                       `(when (eq (,slot last) old)
809                          (setf (,slot last) new))))
810            (frob if-consequent)
811            (frob if-alternative)
812            (when (eq (if-consequent last)
813                      (if-alternative last))
814              (reoptimize-component (block-component block) :maybe)))))
815       (t
816        (unless (memq new (block-succ block))
817          (link-blocks block new)))))
818
819   (values))
820
821 ;;; Unlink a block from the next/prev chain. We also null out the
822 ;;; COMPONENT.
823 (declaim (ftype (sfunction (cblock) (values)) remove-from-dfo))
824 (defun remove-from-dfo (block)
825   (let ((next (block-next block))
826         (prev (block-prev block)))
827     (setf (block-component block) nil)
828     (setf (block-next prev) next)
829     (setf (block-prev next) prev))
830   (values))
831
832 ;;; Add BLOCK to the next/prev chain following AFTER. We also set the
833 ;;; COMPONENT to be the same as for AFTER.
834 (defun add-to-dfo (block after)
835   (declare (type cblock block after))
836   (let ((next (block-next after))
837         (comp (block-component after)))
838     (aver (not (eq (component-kind comp) :deleted)))
839     (setf (block-component block) comp)
840     (setf (block-next after) block)
841     (setf (block-prev block) after)
842     (setf (block-next block) next)
843     (setf (block-prev next) block))
844   (values))
845
846 ;;; List all NLX-INFOs which BLOCK can exit to.
847 ;;;
848 ;;; We hope that no cleanup actions are performed in the middle of
849 ;;; BLOCK, so it is enough to look only at cleanups in the block
850 ;;; end. The tricky thing is a special cleanup block; all its nodes
851 ;;; have the same cleanup info, corresponding to the start, so the
852 ;;; same approach returns safe result.
853 (defun map-block-nlxes (fun block &optional dx-cleanup-fun)
854   (loop for cleanup = (block-end-cleanup block)
855         then (node-enclosing-cleanup (cleanup-mess-up cleanup))
856         while cleanup
857         do (let ((mess-up (cleanup-mess-up cleanup)))
858              (case (cleanup-kind cleanup)
859                ((:block :tagbody)
860                 (aver (entry-p mess-up))
861                 (loop for exit in (entry-exits mess-up)
862                       for nlx-info = (exit-nlx-info exit)
863                       do (funcall fun nlx-info)))
864                ((:catch :unwind-protect)
865                 (aver (combination-p mess-up))
866                 (let* ((arg-lvar (first (basic-combination-args mess-up)))
867                        (nlx-info (constant-value (ref-leaf (lvar-use arg-lvar)))))
868                 (funcall fun nlx-info)))
869                ((:dynamic-extent)
870                 (when dx-cleanup-fun
871                   (funcall dx-cleanup-fun cleanup)))))))
872
873 ;;; Set the FLAG for all the blocks in COMPONENT to NIL, except for
874 ;;; the head and tail which are set to T.
875 (declaim (ftype (sfunction (component) (values)) clear-flags))
876 (defun clear-flags (component)
877   (let ((head (component-head component))
878         (tail (component-tail component)))
879     (setf (block-flag head) t)
880     (setf (block-flag tail) t)
881     (do-blocks (block component)
882       (setf (block-flag block) nil)))
883   (values))
884
885 ;;; Make a component with no blocks in it. The BLOCK-FLAG is initially
886 ;;; true in the head and tail blocks.
887 (declaim (ftype (sfunction () component) make-empty-component))
888 (defun make-empty-component ()
889   (let* ((head (make-block-key :start nil :component nil))
890          (tail (make-block-key :start nil :component nil))
891          (res (make-component head tail)))
892     (setf (block-flag head) t)
893     (setf (block-flag tail) t)
894     (setf (block-component head) res)
895     (setf (block-component tail) res)
896     (setf (block-next head) tail)
897     (setf (block-prev tail) head)
898     res))
899
900 ;;; Make NODE the LAST node in its block, splitting the block if necessary.
901 ;;; The new block is added to the DFO immediately following NODE's block.
902 (defun node-ends-block (node)
903   (declare (type node node))
904   (let* ((block (node-block node))
905          (start (node-next node))
906          (last (block-last block)))
907     (check-type last node)
908     (unless (eq last node)
909       (aver (and (eq (ctran-kind start) :inside-block)
910                  (not (block-delete-p block))))
911       (let* ((succ (block-succ block))
912              (new-block
913               (make-block-key :start start
914                               :component (block-component block)
915                               :succ succ :last last)))
916         (setf (ctran-kind start) :block-start)
917         (setf (ctran-use start) nil)
918         (setf (block-last block) node)
919         (setf (node-next node) nil)
920         (dolist (b succ)
921           (setf (block-pred b)
922                 (cons new-block (remove block (block-pred b)))))
923         (setf (block-succ block) ())
924         (link-blocks block new-block)
925         (add-to-dfo new-block block)
926         (setf (component-reanalyze (block-component block)) t)
927
928         (do ((ctran start (node-next (ctran-next ctran))))
929             ((not ctran))
930           (setf (ctran-block ctran) new-block))
931
932         (setf (block-type-asserted block) t)
933         (setf (block-test-modified block) t))))
934   (values))
935 \f
936 ;;;; deleting stuff
937
938 ;;; Deal with deleting the last (read) reference to a LAMBDA-VAR.
939 (defun delete-lambda-var (leaf)
940   (declare (type lambda-var leaf))
941
942   ;; Iterate over all local calls flushing the corresponding argument,
943   ;; allowing the computation of the argument to be deleted. We also
944   ;; mark the LET for reoptimization, since it may be that we have
945   ;; deleted its last variable.
946   (let* ((fun (lambda-var-home leaf))
947          (n (position leaf (lambda-vars fun))))
948     (dolist (ref (leaf-refs fun))
949       (let* ((lvar (node-lvar ref))
950              (dest (and lvar (lvar-dest lvar))))
951         (when (and (combination-p dest)
952                    (eq (basic-combination-fun dest) lvar)
953                    (eq (basic-combination-kind dest) :local))
954           (let* ((args (basic-combination-args dest))
955                  (arg (elt args n)))
956             (reoptimize-lvar arg)
957             (flush-dest arg)
958             (setf (elt args n) nil))))))
959
960   ;; The LAMBDA-VAR may still have some SETs, but this doesn't cause
961   ;; too much difficulty, since we can efficiently implement
962   ;; write-only variables. We iterate over the SETs, marking their
963   ;; blocks for dead code flushing, since we can delete SETs whose
964   ;; value is unused.
965   (dolist (set (lambda-var-sets leaf))
966     (setf (block-flush-p (node-block set)) t))
967
968   (values))
969
970 ;;; Note that something interesting has happened to VAR.
971 (defun reoptimize-lambda-var (var)
972   (declare (type lambda-var var))
973   (let ((fun (lambda-var-home var)))
974     ;; We only deal with LET variables, marking the corresponding
975     ;; initial value arg as needing to be reoptimized.
976     (when (and (eq (functional-kind fun) :let)
977                (leaf-refs var))
978       (do ((args (basic-combination-args
979                   (lvar-dest (node-lvar (first (leaf-refs fun)))))
980                  (cdr args))
981            (vars (lambda-vars fun) (cdr vars)))
982           ((eq (car vars) var)
983            (reoptimize-lvar (car args))))))
984   (values))
985
986 ;;; Delete a function that has no references. This need only be called
987 ;;; on functions that never had any references, since otherwise
988 ;;; DELETE-REF will handle the deletion.
989 (defun delete-functional (fun)
990   (aver (and (null (leaf-refs fun))
991              (not (functional-entry-fun fun))))
992   (etypecase fun
993     (optional-dispatch (delete-optional-dispatch fun))
994     (clambda (delete-lambda fun)))
995   (values))
996
997 ;;; Deal with deleting the last reference to a CLAMBDA, which means
998 ;;; that the lambda is unreachable, so that its body may be
999 ;;; deleted. We set FUNCTIONAL-KIND to :DELETED and rely on
1000 ;;; IR1-OPTIMIZE to delete its blocks.
1001 (defun delete-lambda (clambda)
1002   (declare (type clambda clambda))
1003   (let ((original-kind (functional-kind clambda))
1004         (bind (lambda-bind clambda)))
1005     (aver (not (member original-kind '(:deleted :toplevel))))
1006     (aver (not (functional-has-external-references-p clambda)))
1007     (aver (or (eq original-kind :zombie) bind))
1008     (setf (functional-kind clambda) :deleted)
1009     (setf (lambda-bind clambda) nil)
1010
1011     (labels ((delete-children (lambda)
1012                (dolist (child (lambda-children lambda))
1013                  (cond ((eq (functional-kind child) :deleted)
1014                         (delete-children child))
1015                        (t
1016                         (delete-lambda child))))
1017                (setf (lambda-children lambda) nil)
1018                (setf (lambda-parent lambda) nil)))
1019       (delete-children clambda))
1020
1021     ;; (The IF test is (FUNCTIONAL-SOMEWHAT-LETLIKE-P CLAMBDA), except
1022     ;; that we're using the old value of the KIND slot, not the
1023     ;; current slot value, which has now been set to :DELETED.)
1024     (case original-kind
1025       (:zombie)
1026       ((:let :mv-let :assignment)
1027        (let ((bind-block (node-block bind)))
1028          (mark-for-deletion bind-block))
1029        (let ((home (lambda-home clambda)))
1030          (setf (lambda-lets home) (delete clambda (lambda-lets home))))
1031        ;; KLUDGE: In presence of NLEs we cannot always understand that
1032        ;; LET's BIND dominates its body [for a LET "its" body is not
1033        ;; quite its]; let's delete too dangerous for IR2 stuff. --
1034        ;; APD, 2004-01-01
1035        (dolist (var (lambda-vars clambda))
1036          (flet ((delete-node (node)
1037                   (mark-for-deletion (node-block node))))
1038          (mapc #'delete-node (leaf-refs var))
1039          (mapc #'delete-node (lambda-var-sets var)))))
1040       (t
1041        ;; Function has no reachable references.
1042        (dolist (ref (lambda-refs clambda))
1043          (mark-for-deletion (node-block ref)))
1044        ;; If the function isn't a LET, we unlink the function head
1045        ;; and tail from the component head and tail to indicate that
1046        ;; the code is unreachable. We also delete the function from
1047        ;; COMPONENT-LAMBDAS (it won't be there before local call
1048        ;; analysis, but no matter.) If the lambda was never
1049        ;; referenced, we give a note.
1050        (let* ((bind-block (node-block bind))
1051               (component (block-component bind-block))
1052               (return (lambda-return clambda))
1053               (return-block (and return (node-block return))))
1054          (unless (leaf-ever-used clambda)
1055            (let ((*compiler-error-context* bind))
1056              (compiler-notify 'code-deletion-note
1057                               :format-control "deleting unused function~:[.~;~:*~%  ~S~]"
1058                               :format-arguments (list (leaf-debug-name clambda)))))
1059          (unless (block-delete-p bind-block)
1060            (unlink-blocks (component-head component) bind-block))
1061          (when (and return-block (not (block-delete-p return-block)))
1062            (mark-for-deletion return-block)
1063            (unlink-blocks return-block (component-tail component)))
1064          (setf (component-reanalyze component) t)
1065          (let ((tails (lambda-tail-set clambda)))
1066            (setf (tail-set-funs tails)
1067                  (delete clambda (tail-set-funs tails)))
1068            (setf (lambda-tail-set clambda) nil))
1069          (setf (component-lambdas component)
1070                (delq clambda (component-lambdas component))))))
1071
1072     ;; If the lambda is an XEP, then we null out the ENTRY-FUN in its
1073     ;; ENTRY-FUN so that people will know that it is not an entry
1074     ;; point anymore.
1075     (when (eq original-kind :external)
1076       (let ((fun (functional-entry-fun clambda)))
1077         (setf (functional-entry-fun fun) nil)
1078         (when (optional-dispatch-p fun)
1079           (delete-optional-dispatch fun)))))
1080
1081   (values))
1082
1083 ;;; Deal with deleting the last reference to an OPTIONAL-DISPATCH. We
1084 ;;; have to be a bit more careful than with lambdas, since DELETE-REF
1085 ;;; is used both before and after local call analysis. Afterward, all
1086 ;;; references to still-existing OPTIONAL-DISPATCHes have been moved
1087 ;;; to the XEP, leaving it with no references at all. So we look at
1088 ;;; the XEP to see whether an optional-dispatch is still really being
1089 ;;; used. But before local call analysis, there are no XEPs, and all
1090 ;;; references are direct.
1091 ;;;
1092 ;;; When we do delete the OPTIONAL-DISPATCH, we grovel all of its
1093 ;;; entry-points, making them be normal lambdas, and then deleting the
1094 ;;; ones with no references. This deletes any e-p lambdas that were
1095 ;;; either never referenced, or couldn't be deleted when the last
1096 ;;; reference was deleted (due to their :OPTIONAL kind.)
1097 ;;;
1098 ;;; Note that the last optional entry point may alias the main entry,
1099 ;;; so when we process the main entry, its KIND may have been changed
1100 ;;; to NIL or even converted to a LETlike value.
1101 (defun delete-optional-dispatch (leaf)
1102   (declare (type optional-dispatch leaf))
1103   (let ((entry (functional-entry-fun leaf)))
1104     (unless (and entry (leaf-refs entry))
1105       (aver (or (not entry) (eq (functional-kind entry) :deleted)))
1106       (setf (functional-kind leaf) :deleted)
1107
1108       (flet ((frob (fun)
1109                (unless (eq (functional-kind fun) :deleted)
1110                  (aver (eq (functional-kind fun) :optional))
1111                  (setf (functional-kind fun) nil)
1112                  (let ((refs (leaf-refs fun)))
1113                    (cond ((null refs)
1114                           (delete-lambda fun))
1115                          ((null (rest refs))
1116                           (or (maybe-let-convert fun)
1117                               (maybe-convert-to-assignment fun)))
1118                          (t
1119                           (maybe-convert-to-assignment fun)))))))
1120
1121         (dolist (ep (optional-dispatch-entry-points leaf))
1122           (when (promise-ready-p ep)
1123             (frob (force ep))))
1124         (when (optional-dispatch-more-entry leaf)
1125           (frob (optional-dispatch-more-entry leaf)))
1126         (let ((main (optional-dispatch-main-entry leaf)))
1127           (when entry
1128             (setf (functional-entry-fun entry) main)
1129             (setf (functional-entry-fun main) entry))
1130           (when (eq (functional-kind main) :optional)
1131             (frob main))))))
1132
1133   (values))
1134
1135 (defun note-local-functional (fun)
1136   (declare (type functional fun))
1137   (when (and (leaf-has-source-name-p fun)
1138              (eq (leaf-source-name fun) (functional-debug-name fun)))
1139     (let ((name (leaf-source-name fun)))
1140       (let ((defined-fun (gethash name *free-funs*)))
1141         (when (and defined-fun
1142                    (defined-fun-p defined-fun)
1143                    (eq (defined-fun-functional defined-fun) fun))
1144           (remhash name *free-funs*))))))
1145
1146 ;;; Do stuff to delete the semantic attachments of a REF node. When
1147 ;;; this leaves zero or one reference, we do a type dispatch off of
1148 ;;; the leaf to determine if a special action is appropriate.
1149 (defun delete-ref (ref)
1150   (declare (type ref ref))
1151   (let* ((leaf (ref-leaf ref))
1152          (refs (delq ref (leaf-refs leaf))))
1153     (setf (leaf-refs leaf) refs)
1154
1155     (cond ((null refs)
1156            (typecase leaf
1157              (lambda-var
1158               (delete-lambda-var leaf))
1159              (clambda
1160               (ecase (functional-kind leaf)
1161                 ((nil :let :mv-let :assignment :escape :cleanup)
1162                  (aver (null (functional-entry-fun leaf)))
1163                  (delete-lambda leaf))
1164                 (:external
1165                  (delete-lambda leaf))
1166                 ((:deleted :zombie :optional))))
1167              (optional-dispatch
1168               (unless (eq (functional-kind leaf) :deleted)
1169                 (delete-optional-dispatch leaf)))))
1170           ((null (rest refs))
1171            (typecase leaf
1172              (clambda (or (maybe-let-convert leaf)
1173                           (maybe-convert-to-assignment leaf)))
1174              (lambda-var (reoptimize-lambda-var leaf))))
1175           (t
1176            (typecase leaf
1177              (clambda (maybe-convert-to-assignment leaf))))))
1178
1179   (values))
1180
1181 ;;; This function is called by people who delete nodes; it provides a
1182 ;;; way to indicate that the value of a lvar is no longer used. We
1183 ;;; null out the LVAR-DEST, set FLUSH-P in the blocks containing uses
1184 ;;; of LVAR and set COMPONENT-REOPTIMIZE.
1185 (defun flush-dest (lvar)
1186   (declare (type (or lvar null) lvar))
1187   (unless (null lvar)
1188     (setf (lvar-dest lvar) nil)
1189     (flush-lvar-externally-checkable-type lvar)
1190     (do-uses (use lvar)
1191       (let ((prev (node-prev use)))
1192         (let ((block (ctran-block prev)))
1193           (reoptimize-component (block-component block) t)
1194           (setf (block-attributep (block-flags block)
1195                                   flush-p type-asserted type-check)
1196                 t)))
1197       (setf (node-lvar use) nil))
1198     (setf (lvar-uses lvar) nil))
1199   (values))
1200
1201 (defun delete-dest (lvar)
1202   (when lvar
1203     (let* ((dest (lvar-dest lvar))
1204            (prev (node-prev dest)))
1205       (let ((block (ctran-block prev)))
1206         (unless (block-delete-p block)
1207           (mark-for-deletion block))))))
1208
1209 ;;; Queue the block for deletion
1210 (defun delete-block-lazily (block)
1211   (declare (type cblock block))
1212   (unless (block-delete-p block)
1213     (setf (block-delete-p block) t)
1214     (push block (component-delete-blocks (block-component block)))))
1215
1216 ;;; Do a graph walk backward from BLOCK, marking all predecessor
1217 ;;; blocks with the DELETE-P flag.
1218 (defun mark-for-deletion (block)
1219   (declare (type cblock block))
1220   (let* ((component (block-component block))
1221          (head (component-head component)))
1222     (labels ((helper (block)
1223                (delete-block-lazily block)
1224                (dolist (pred (block-pred block))
1225                  (unless (or (block-delete-p pred)
1226                              (eq pred head))
1227                    (helper pred)))))
1228       (unless (block-delete-p block)
1229         (helper block)
1230         (setf (component-reanalyze component) t))))
1231   (values))
1232
1233 ;;; This function does what is necessary to eliminate the code in it
1234 ;;; from the IR1 representation. This involves unlinking it from its
1235 ;;; predecessors and successors and deleting various node-specific
1236 ;;; semantic information. BLOCK must be already removed from
1237 ;;; COMPONENT-DELETE-BLOCKS.
1238 (defun delete-block (block &optional silent)
1239   (declare (type cblock block))
1240   (aver (block-component block))      ; else block is already deleted!
1241   #!+high-security (aver (not (memq block (component-delete-blocks (block-component block)))))
1242   (unless silent
1243     (note-block-deletion block))
1244   (setf (block-delete-p block) t)
1245
1246   (dolist (b (block-pred block))
1247     (unlink-blocks b block)
1248     ;; In bug 147 the almost-all-blocks-have-a-successor invariant was
1249     ;; broken when successors were deleted without setting the
1250     ;; BLOCK-DELETE-P flags of their predececessors. Make sure that
1251     ;; doesn't happen again.
1252     (aver (not (and (null (block-succ b))
1253                     (not (block-delete-p b))
1254                     (not (eq b (component-head (block-component b))))))))
1255   (dolist (b (block-succ block))
1256     (unlink-blocks block b))
1257
1258   (do-nodes-carefully (node block)
1259     (when (valued-node-p node)
1260       (delete-lvar-use node))
1261     (etypecase node
1262       (ref (delete-ref node))
1263       (cif (flush-dest (if-test node)))
1264       ;; The next two cases serve to maintain the invariant that a LET
1265       ;; always has a well-formed COMBINATION, REF and BIND. We delete
1266       ;; the lambda whenever we delete any of these, but we must be
1267       ;; careful that this LET has not already been partially deleted.
1268       (basic-combination
1269        (when (and (eq (basic-combination-kind node) :local)
1270                   ;; Guards COMBINATION-LAMBDA agains the REF being deleted.
1271                   (lvar-uses (basic-combination-fun node)))
1272          (let ((fun (combination-lambda node)))
1273            ;; If our REF was the second-to-last ref, and has been
1274            ;; deleted, then FUN may be a LET for some other
1275            ;; combination.
1276            (when (and (functional-letlike-p fun)
1277                       (eq (let-combination fun) node))
1278              (delete-lambda fun))))
1279        (flush-dest (basic-combination-fun node))
1280        (dolist (arg (basic-combination-args node))
1281          (when arg (flush-dest arg))))
1282       (bind
1283        (let ((lambda (bind-lambda node)))
1284          (unless (eq (functional-kind lambda) :deleted)
1285            (delete-lambda lambda))))
1286       (exit
1287        (let ((value (exit-value node))
1288              (entry (exit-entry node)))
1289          (when value
1290            (flush-dest value))
1291          (when entry
1292            (setf (entry-exits entry)
1293                  (delq node (entry-exits entry))))))
1294       (entry
1295        (dolist (exit (entry-exits node))
1296          (mark-for-deletion (node-block exit)))
1297        (let ((home (node-home-lambda node)))
1298          (setf (lambda-entries home) (delq node (lambda-entries home)))))
1299       (creturn
1300        (flush-dest (return-result node))
1301        (delete-return node))
1302       (cset
1303        (flush-dest (set-value node))
1304        (let ((var (set-var node)))
1305          (setf (basic-var-sets var)
1306                (delete node (basic-var-sets var)))))
1307       (cast
1308        (flush-dest (cast-value node)))))
1309
1310   (remove-from-dfo block)
1311   (values))
1312
1313 ;;; Do stuff to indicate that the return node NODE is being deleted.
1314 (defun delete-return (node)
1315   (declare (type creturn node))
1316   (let* ((fun (return-lambda node))
1317          (tail-set (lambda-tail-set fun)))
1318     (aver (lambda-return fun))
1319     (setf (lambda-return fun) nil)
1320     (when (and tail-set (not (find-if #'lambda-return
1321                                       (tail-set-funs tail-set))))
1322       (setf (tail-set-type tail-set) *empty-type*)))
1323   (values))
1324
1325 ;;; If any of the VARS in FUN was never referenced and was not
1326 ;;; declared IGNORE, then complain.
1327 (defun note-unreferenced-vars (fun)
1328   (declare (type clambda fun))
1329   (dolist (var (lambda-vars fun))
1330     (unless (or (leaf-ever-used var)
1331                 (lambda-var-ignorep var))
1332       (let ((*compiler-error-context* (lambda-bind fun)))
1333         (unless (policy *compiler-error-context* (= inhibit-warnings 3))
1334           ;; ANSI section "3.2.5 Exceptional Situations in the Compiler"
1335           ;; requires this to be no more than a STYLE-WARNING.
1336           #-sb-xc-host
1337           (compiler-style-warn "The variable ~S is defined but never used."
1338                                (leaf-debug-name var))
1339           ;; There's no reason to accept this kind of equivocation
1340           ;; when compiling our own code, though.
1341           #+sb-xc-host
1342           (warn "The variable ~S is defined but never used."
1343                 (leaf-debug-name var)))
1344         (setf (leaf-ever-used var) t)))) ; to avoid repeated warnings? -- WHN
1345   (values))
1346
1347 (defvar *deletion-ignored-objects* '(t nil))
1348
1349 ;;; Return true if we can find OBJ in FORM, NIL otherwise. We bound
1350 ;;; our recursion so that we don't get lost in circular structures. We
1351 ;;; ignore the car of forms if they are a symbol (to prevent confusing
1352 ;;; function referencess with variables), and we also ignore anything
1353 ;;; inside ' or #'.
1354 (defun present-in-form (obj form depth)
1355   (declare (type (integer 0 20) depth))
1356   (cond ((= depth 20) nil)
1357         ((eq obj form) t)
1358         ((atom form) nil)
1359         (t
1360          (let ((first (car form))
1361                (depth (1+ depth)))
1362            (if (member first '(quote function))
1363                nil
1364                (or (and (not (symbolp first))
1365                         (present-in-form obj first depth))
1366                    (do ((l (cdr form) (cdr l))
1367                         (n 0 (1+ n)))
1368                        ((or (atom l) (> n 100))
1369                         nil)
1370                      (declare (fixnum n))
1371                      (when (present-in-form obj (car l) depth)
1372                        (return t)))))))))
1373
1374 ;;; This function is called on a block immediately before we delete
1375 ;;; it. We check to see whether any of the code about to die appeared
1376 ;;; in the original source, and emit a note if so.
1377 ;;;
1378 ;;; If the block was in a lambda is now deleted, then we ignore the
1379 ;;; whole block, since this case is picked off in DELETE-LAMBDA. We
1380 ;;; also ignore the deletion of CRETURN nodes, since it is somewhat
1381 ;;; reasonable for a function to not return, and there is a different
1382 ;;; note for that case anyway.
1383 ;;;
1384 ;;; If the actual source is an atom, then we use a bunch of heuristics
1385 ;;; to guess whether this reference really appeared in the original
1386 ;;; source:
1387 ;;; -- If a symbol, it must be interned and not a keyword.
1388 ;;; -- It must not be an easily introduced constant (T or NIL, a fixnum
1389 ;;;    or a character.)
1390 ;;; -- The atom must be "present" in the original source form, and
1391 ;;;    present in all intervening actual source forms.
1392 (defun note-block-deletion (block)
1393   (let ((home (block-home-lambda block)))
1394     (unless (eq (functional-kind home) :deleted)
1395       (do-nodes (node nil block)
1396         (let* ((path (node-source-path node))
1397                (first (first path)))
1398           (when (or (eq first 'original-source-start)
1399                     (and (atom first)
1400                          (or (not (symbolp first))
1401                              (let ((pkg (symbol-package first)))
1402                                (and pkg
1403                                     (not (eq pkg (symbol-package :end))))))
1404                          (not (member first *deletion-ignored-objects*))
1405                          (not (typep first '(or fixnum character)))
1406                          (every (lambda (x)
1407                                   (present-in-form first x 0))
1408                                 (source-path-forms path))
1409                          (present-in-form first (find-original-source path)
1410                                           0)))
1411             (unless (return-p node)
1412               (let ((*compiler-error-context* node))
1413                 (compiler-notify 'code-deletion-note
1414                                  :format-control "deleting unreachable code"
1415                                  :format-arguments nil)))
1416             (return))))))
1417   (values))
1418
1419 ;;; Delete a node from a block, deleting the block if there are no
1420 ;;; nodes left. We remove the node from the uses of its LVAR.
1421 ;;;
1422 ;;; If the node is the last node, there must be exactly one successor.
1423 ;;; We link all of our precedessors to the successor and unlink the
1424 ;;; block. In this case, we return T, otherwise NIL. If no nodes are
1425 ;;; left, and the block is a successor of itself, then we replace the
1426 ;;; only node with a degenerate exit node. This provides a way to
1427 ;;; represent the bodyless infinite loop, given the prohibition on
1428 ;;; empty blocks in IR1.
1429 (defun unlink-node (node)
1430   (declare (type node node))
1431   (when (valued-node-p node)
1432     (delete-lvar-use node))
1433
1434   (let* ((ctran (node-next node))
1435          (next (and ctran (ctran-next ctran)))
1436          (prev (node-prev node))
1437          (block (ctran-block prev))
1438          (prev-kind (ctran-kind prev))
1439          (last (block-last block)))
1440
1441     (setf (block-type-asserted block) t)
1442     (setf (block-test-modified block) t)
1443
1444     (cond ((or (eq prev-kind :inside-block)
1445                (and (eq prev-kind :block-start)
1446                     (not (eq node last))))
1447            (cond ((eq node last)
1448                   (setf (block-last block) (ctran-use prev))
1449                   (setf (node-next (ctran-use prev)) nil))
1450                  (t
1451                   (setf (ctran-next prev) next)
1452                   (setf (node-prev next) prev)
1453                   (when (if-p next) ; AOP wanted
1454                     (reoptimize-lvar (if-test next)))))
1455            (setf (node-prev node) nil)
1456            nil)
1457           (t
1458            (aver (eq prev-kind :block-start))
1459            (aver (eq node last))
1460            (let* ((succ (block-succ block))
1461                   (next (first succ)))
1462              (aver (singleton-p succ))
1463              (cond
1464               ((eq block (first succ))
1465                (with-ir1-environment-from-node node
1466                  (let ((exit (make-exit)))
1467                    (setf (ctran-next prev) nil)
1468                    (link-node-to-previous-ctran exit prev)
1469                    (setf (block-last block) exit)))
1470                (setf (node-prev node) nil)
1471                nil)
1472               (t
1473                (aver (eq (block-start-cleanup block)
1474                          (block-end-cleanup block)))
1475                (unlink-blocks block next)
1476                (dolist (pred (block-pred block))
1477                  (change-block-successor pred block next))
1478                (when (block-delete-p block)
1479                  (let ((component (block-component block)))
1480                    (setf (component-delete-blocks component)
1481                          (delq block (component-delete-blocks component)))))
1482                (remove-from-dfo block)
1483                (setf (block-delete-p block) t)
1484                (setf (node-prev node) nil)
1485                t)))))))
1486
1487 ;;; Return true if CTRAN has been deleted, false if it is still a valid
1488 ;;; part of IR1.
1489 (defun ctran-deleted-p (ctran)
1490   (declare (type ctran ctran))
1491   (let ((block (ctran-block ctran)))
1492     (or (not (block-component block))
1493         (block-delete-p block))))
1494
1495 ;;; Return true if NODE has been deleted, false if it is still a valid
1496 ;;; part of IR1.
1497 (defun node-deleted (node)
1498   (declare (type node node))
1499   (let ((prev (node-prev node)))
1500     (or (not prev)
1501         (ctran-deleted-p prev))))
1502
1503 ;;; Delete all the blocks and functions in COMPONENT. We scan first
1504 ;;; marking the blocks as DELETE-P to prevent weird stuff from being
1505 ;;; triggered by deletion.
1506 (defun delete-component (component)
1507   (declare (type component component))
1508   (aver (null (component-new-functionals component)))
1509   (setf (component-kind component) :deleted)
1510   (do-blocks (block component)
1511     (delete-block-lazily block))
1512   (dolist (fun (component-lambdas component))
1513     (unless (eq (functional-kind fun) :deleted)
1514       (setf (functional-kind fun) nil)
1515       (setf (functional-entry-fun fun) nil)
1516       (setf (leaf-refs fun) nil)
1517       (delete-functional fun)))
1518   (clean-component component)
1519   (values))
1520
1521 ;;; Remove all pending blocks to be deleted. Return the nearest live
1522 ;;; block after or equal to BLOCK.
1523 (defun clean-component (component &optional block)
1524   (loop while (component-delete-blocks component)
1525         ;; actual deletion of a block may queue new blocks
1526         do (let ((current (pop (component-delete-blocks component))))
1527              (when (eq block current)
1528                (setq block (block-next block)))
1529              (delete-block current)))
1530   block)
1531
1532 ;;; Convert code of the form
1533 ;;;   (FOO ... (FUN ...) ...)
1534 ;;; to
1535 ;;;   (FOO ...    ...    ...).
1536 ;;; In other words, replace the function combination FUN by its
1537 ;;; arguments. If there are any problems with doing this, use GIVE-UP
1538 ;;; to blow out of whatever transform called this. Note, as the number
1539 ;;; of arguments changes, the transform must be prepared to return a
1540 ;;; lambda with a new lambda-list with the correct number of
1541 ;;; arguments.
1542 (defun splice-fun-args (lvar fun num-args)
1543   #!+sb-doc
1544   "If LVAR is a call to FUN with NUM-ARGS args, change those arguments
1545    to feed directly to the LVAR-DEST of LVAR, which must be a
1546    combination."
1547   (declare (type lvar lvar)
1548            (type symbol fun)
1549            (type index num-args))
1550   (let ((outside (lvar-dest lvar))
1551         (inside (lvar-uses lvar)))
1552     (aver (combination-p outside))
1553     (unless (combination-p inside)
1554       (give-up-ir1-transform))
1555     (let ((inside-fun (combination-fun inside)))
1556       (unless (eq (lvar-fun-name inside-fun) fun)
1557         (give-up-ir1-transform))
1558       (let ((inside-args (combination-args inside)))
1559         (unless (= (length inside-args) num-args)
1560           (give-up-ir1-transform))
1561         (let* ((outside-args (combination-args outside))
1562                (arg-position (position lvar outside-args))
1563                (before-args (subseq outside-args 0 arg-position))
1564                (after-args (subseq outside-args (1+ arg-position))))
1565           (dolist (arg inside-args)
1566             (setf (lvar-dest arg) outside)
1567             (flush-lvar-externally-checkable-type arg))
1568           (setf (combination-args inside) nil)
1569           (setf (combination-args outside)
1570                 (append before-args inside-args after-args))
1571           (change-ref-leaf (lvar-uses inside-fun)
1572                            (find-free-fun 'list "???"))
1573           (setf (combination-fun-info inside) (info :function :info 'list)
1574                 (combination-kind inside) :known)
1575           (setf (node-derived-type inside) *wild-type*)
1576           (flush-dest lvar)
1577           (values))))))
1578
1579 (defun extract-fun-args (lvar fun num-args)
1580   (declare (type lvar lvar)
1581            (type (or symbol list) fun)
1582            (type index num-args))
1583   (let ((fun (if (listp fun) fun (list fun))))
1584     (let ((inside (lvar-uses lvar)))
1585       (unless (combination-p inside)
1586         (give-up-ir1-transform))
1587       (let ((inside-fun (combination-fun inside)))
1588         (unless (member (lvar-fun-name inside-fun) fun)
1589           (give-up-ir1-transform))
1590         (let ((inside-args (combination-args inside)))
1591           (unless (= (length inside-args) num-args)
1592             (give-up-ir1-transform))
1593           (values (lvar-fun-name inside-fun) inside-args))))))
1594
1595 (defun flush-combination (combination)
1596   (declare (type combination combination))
1597   (flush-dest (combination-fun combination))
1598   (dolist (arg (combination-args combination))
1599     (flush-dest arg))
1600   (unlink-node combination)
1601   (values))
1602
1603 \f
1604 ;;;; leaf hackery
1605
1606 ;;; Change the LEAF that a REF refers to.
1607 (defun change-ref-leaf (ref leaf)
1608   (declare (type ref ref) (type leaf leaf))
1609   (unless (eq (ref-leaf ref) leaf)
1610     (push ref (leaf-refs leaf))
1611     (delete-ref ref)
1612     (setf (ref-leaf ref) leaf)
1613     (setf (leaf-ever-used leaf) t)
1614     (let* ((ltype (leaf-type leaf))
1615            (vltype (make-single-value-type ltype)))
1616       (if (let* ((lvar (node-lvar ref))
1617                  (dest (and lvar (lvar-dest lvar))))
1618             (and (basic-combination-p dest)
1619                  (eq lvar (basic-combination-fun dest))
1620                  (csubtypep ltype (specifier-type 'function))))
1621           (setf (node-derived-type ref) vltype)
1622           (derive-node-type ref vltype)))
1623     (reoptimize-lvar (node-lvar ref)))
1624   (values))
1625
1626 ;;; Change all REFS for OLD-LEAF to NEW-LEAF.
1627 (defun substitute-leaf (new-leaf old-leaf)
1628   (declare (type leaf new-leaf old-leaf))
1629   (dolist (ref (leaf-refs old-leaf))
1630     (change-ref-leaf ref new-leaf))
1631   (values))
1632
1633 ;;; like SUBSITUTE-LEAF, only there is a predicate on the REF to tell
1634 ;;; whether to substitute
1635 (defun substitute-leaf-if (test new-leaf old-leaf)
1636   (declare (type leaf new-leaf old-leaf) (type function test))
1637   (dolist (ref (leaf-refs old-leaf))
1638     (when (funcall test ref)
1639       (change-ref-leaf ref new-leaf)))
1640   (values))
1641
1642 ;;; Return a LEAF which represents the specified constant object. If
1643 ;;; the object is not in *CONSTANTS*, then we create a new constant
1644 ;;; LEAF and enter it. If we are producing a fasl file, make sure that
1645 ;;; MAKE-LOAD-FORM gets used on any parts of the constant that it
1646 ;;; needs to be.
1647 ;;;
1648 ;;; We are allowed to coalesce things like EQUAL strings and bit-vectors
1649 ;;; when file-compiling, but not when using COMPILE.
1650 (defun find-constant (object &optional (name nil namep))
1651   (let ((faslp (producing-fasl-file)))
1652     (labels ((make-it ()
1653                (when faslp
1654                  (if namep
1655                      (maybe-emit-make-load-forms object name)
1656                      (maybe-emit-make-load-forms object)))
1657                (make-constant object))
1658              (core-coalesce-p (x)
1659                ;; True for things which retain their identity under EQUAL,
1660                ;; so we can safely share the same CONSTANT leaf between
1661                ;; multiple references.
1662                (or (typep x '(or symbol number character))
1663                    ;; Amusingly enough, we see CLAMBDAs --among other things--
1664                    ;; here, from compiling things like %ALLOCATE-CLOSUREs forms.
1665                    ;; No point in stuffing them in the hash-table.
1666                    (and (typep x 'instance)
1667                         (not (or (leaf-p x) (node-p x))))))
1668              (file-coalesce-p (x)
1669                ;; CLHS 3.2.4.2.2: We are also allowed to coalesce various
1670                ;; other things when file-compiling.
1671                (or (core-coalesce-p x)
1672                    (if (consp x)
1673                        (if (eq +code-coverage-unmarked+ (cdr x))
1674                            ;; These are already coalesced, and the CAR should
1675                            ;; always be OK, so no need to check.
1676                            t
1677                            (unless (maybe-cyclic-p x) ; safe for EQUAL?
1678                              (do ((y x (cdr y)))
1679                                  ((atom y) (file-coalesce-p y))
1680                                (unless (file-coalesce-p (car y))
1681                                  (return nil)))))
1682                        ;; We *could* coalesce base-strings as well, but we'd need
1683                        ;; a separate hash-table for that, since we are not allowed to
1684                        ;; coalesce base-strings with non-base-strings.
1685                        (typep x '(or (vector character) bit-vector)))))
1686              (coalescep (x)
1687                (if faslp (file-coalesce-p x) (core-coalesce-p x))))
1688       (if (and (boundp '*constants*) (coalescep object))
1689           (or (gethash object *constants*)
1690               (setf (gethash object *constants*)
1691                     (make-it)))
1692           (make-it)))))
1693 \f
1694 ;;; Return true if VAR would have to be closed over if environment
1695 ;;; analysis ran now (i.e. if there are any uses that have a different
1696 ;;; home lambda than VAR's home.)
1697 (defun closure-var-p (var)
1698   (declare (type lambda-var var))
1699   (let ((home (lambda-var-home var)))
1700     (cond ((eq (functional-kind home) :deleted)
1701            nil)
1702           (t (let ((home (lambda-home home)))
1703                (flet ((frob (l)
1704                         (find home l
1705                               :key #'node-home-lambda
1706                               :test #'neq)))
1707                  (or (frob (leaf-refs var))
1708                      (frob (basic-var-sets var)))))))))
1709
1710 ;;; If there is a non-local exit noted in ENTRY's environment that
1711 ;;; exits to CONT in that entry, then return it, otherwise return NIL.
1712 (defun find-nlx-info (exit)
1713   (declare (type exit exit))
1714   (let* ((entry (exit-entry exit))
1715          (cleanup (entry-cleanup entry))
1716         (block (first (block-succ (node-block exit)))))
1717     (dolist (nlx (physenv-nlx-info (node-physenv entry)) nil)
1718       (when (and (eq (nlx-info-block nlx) block)
1719                  (eq (nlx-info-cleanup nlx) cleanup))
1720         (return nlx)))))
1721
1722 (defun nlx-info-lvar (nlx)
1723   (declare (type nlx-info nlx))
1724   (node-lvar (block-last (nlx-info-target nlx))))
1725 \f
1726 ;;;; functional hackery
1727
1728 (declaim (ftype (sfunction (functional) clambda) main-entry))
1729 (defun main-entry (functional)
1730   (etypecase functional
1731     (clambda functional)
1732     (optional-dispatch
1733      (optional-dispatch-main-entry functional))))
1734
1735 ;;; RETURN true if FUNCTIONAL is a thing that can be treated like
1736 ;;; MV-BIND when it appears in an MV-CALL. All fixed arguments must be
1737 ;;; optional with null default and no SUPPLIED-P. There must be a
1738 ;;; &REST arg with no references.
1739 (declaim (ftype (sfunction (functional) boolean) looks-like-an-mv-bind))
1740 (defun looks-like-an-mv-bind (functional)
1741   (and (optional-dispatch-p functional)
1742        (do ((arg (optional-dispatch-arglist functional) (cdr arg)))
1743            ((null arg) nil)
1744          (let ((info (lambda-var-arg-info (car arg))))
1745            (unless info (return nil))
1746            (case (arg-info-kind info)
1747              (:optional
1748               (when (or (arg-info-supplied-p info) (arg-info-default info))
1749                 (return nil)))
1750              (:rest
1751               (return (and (null (cdr arg)) (null (leaf-refs (car arg))))))
1752              (t
1753               (return nil)))))))
1754
1755 ;;; Return true if function is an external entry point. This is true
1756 ;;; of normal XEPs (:EXTERNAL kind) and also of top level lambdas
1757 ;;; (:TOPLEVEL kind.)
1758 (defun xep-p (fun)
1759   (declare (type functional fun))
1760   (not (null (member (functional-kind fun) '(:external :toplevel)))))
1761
1762 ;;; If LVAR's only use is a non-notinline global function reference,
1763 ;;; then return the referenced symbol, otherwise NIL. If NOTINLINE-OK
1764 ;;; is true, then we don't care if the leaf is NOTINLINE.
1765 (defun lvar-fun-name (lvar &optional notinline-ok)
1766   (declare (type lvar lvar))
1767   (let ((use (lvar-uses lvar)))
1768     (if (ref-p use)
1769         (let ((leaf (ref-leaf use)))
1770           (if (and (global-var-p leaf)
1771                    (eq (global-var-kind leaf) :global-function)
1772                    (or (not (defined-fun-p leaf))
1773                        (not (eq (defined-fun-inlinep leaf) :notinline))
1774                        notinline-ok))
1775               (leaf-source-name leaf)
1776               nil))
1777         nil)))
1778
1779 (defun lvar-fun-debug-name (lvar)
1780   (declare (type lvar lvar))
1781   (let ((uses (lvar-uses lvar)))
1782     (flet ((name1 (use)
1783              (leaf-debug-name (ref-leaf use))))
1784       (if (ref-p uses)
1785         (name1 uses)
1786         (mapcar #'name1 uses)))))
1787
1788 ;;; Return the source name of a combination. (This is an idiom
1789 ;;; which was used in CMU CL. I gather it always works. -- WHN)
1790 (defun combination-fun-source-name (combination)
1791   (let ((ref (lvar-uses (combination-fun combination))))
1792     (leaf-source-name (ref-leaf ref))))
1793
1794 ;;; Return the COMBINATION node that is the call to the LET FUN.
1795 (defun let-combination (fun)
1796   (declare (type clambda fun))
1797   (aver (functional-letlike-p fun))
1798   (lvar-dest (node-lvar (first (leaf-refs fun)))))
1799
1800 ;;; Return the initial value lvar for a LET variable, or NIL if there
1801 ;;; is none.
1802 (defun let-var-initial-value (var)
1803   (declare (type lambda-var var))
1804   (let ((fun (lambda-var-home var)))
1805     (elt (combination-args (let-combination fun))
1806          (position-or-lose var (lambda-vars fun)))))
1807
1808 ;;; Return the LAMBDA that is called by the local CALL.
1809 (defun combination-lambda (call)
1810   (declare (type basic-combination call))
1811   (aver (eq (basic-combination-kind call) :local))
1812   (ref-leaf (lvar-uses (basic-combination-fun call))))
1813
1814 (defvar *inline-expansion-limit* 200
1815   #!+sb-doc
1816   "an upper limit on the number of inline function calls that will be expanded
1817    in any given code object (single function or block compilation)")
1818
1819 ;;; Check whether NODE's component has exceeded its inline expansion
1820 ;;; limit, and warn if so, returning NIL.
1821 (defun inline-expansion-ok (node)
1822   (let ((expanded (incf (component-inline-expansions
1823                          (block-component
1824                           (node-block node))))))
1825     (cond ((> expanded *inline-expansion-limit*) nil)
1826           ((= expanded *inline-expansion-limit*)
1827            ;; FIXME: If the objective is to stop the recursive
1828            ;; expansion of inline functions, wouldn't it be more
1829            ;; correct to look back through surrounding expansions
1830            ;; (which are, I think, stored in the *CURRENT-PATH*, and
1831            ;; possibly stored elsewhere too) and suppress expansion
1832            ;; and print this warning when the function being proposed
1833            ;; for inline expansion is found there? (I don't like the
1834            ;; arbitrary numerical limit in principle, and I think
1835            ;; it'll be a nuisance in practice if we ever want the
1836            ;; compiler to be able to use WITH-COMPILATION-UNIT on
1837            ;; arbitrarily huge blocks of code. -- WHN)
1838            (let ((*compiler-error-context* node))
1839              (compiler-notify "*INLINE-EXPANSION-LIMIT* (~W) was exceeded, ~
1840                                probably trying to~%  ~
1841                                inline a recursive function."
1842                               *inline-expansion-limit*))
1843            nil)
1844           (t t))))
1845
1846 ;;; Make sure that FUNCTIONAL is not let-converted or deleted.
1847 (defun assure-functional-live-p (functional)
1848   (declare (type functional functional))
1849   (when (and (or
1850               ;; looks LET-converted
1851               (functional-somewhat-letlike-p functional)
1852               ;; It's possible for a LET-converted function to end up
1853               ;; deleted later. In that case, for the purposes of this
1854               ;; analysis, it is LET-converted: LET-converted functionals
1855               ;; are too badly trashed to expand them inline, and deleted
1856               ;; LET-converted functionals are even worse.
1857               (memq (functional-kind functional) '(:deleted :zombie))))
1858     (throw 'locall-already-let-converted functional)))
1859
1860 (defun call-full-like-p (call)
1861   (declare (type combination call))
1862   (let ((kind (basic-combination-kind call)))
1863     (or (eq kind :full)
1864         (and (eq kind :known)
1865              (let ((info (basic-combination-fun-info call)))
1866                (and
1867                 (not (fun-info-ir2-convert info))
1868                 (dolist (template (fun-info-templates info) t)
1869                   (when (eq (template-ltn-policy template) :fast-safe)
1870                     (multiple-value-bind (val win)
1871                        (valid-fun-use call (template-type template))
1872                       (when (or val (not win)) (return nil)))))))))))
1873 \f
1874 ;;;; careful call
1875
1876 ;;; Apply a function to some arguments, returning a list of the values
1877 ;;; resulting of the evaluation. If an error is signalled during the
1878 ;;; application, then we produce a warning message using WARN-FUN and
1879 ;;; return NIL as our second value to indicate this. NODE is used as
1880 ;;; the error context for any error message, and CONTEXT is a string
1881 ;;; that is spliced into the warning.
1882 (declaim (ftype (sfunction ((or symbol function) list node function string)
1883                           (values list boolean))
1884                 careful-call))
1885 (defun careful-call (function args node warn-fun context)
1886   (values
1887    (multiple-value-list
1888     (handler-case (apply function args)
1889       (error (condition)
1890         (let ((*compiler-error-context* node))
1891           (funcall warn-fun "Lisp error during ~A:~%~A" context condition)
1892           (return-from careful-call (values nil nil))))))
1893    t))
1894
1895 ;;; Variations of SPECIFIER-TYPE for parsing possibly wrong
1896 ;;; specifiers.
1897 (macrolet
1898     ((deffrob (basic careful compiler transform)
1899        `(progn
1900           (defun ,careful (specifier)
1901             (handler-case (,basic specifier)
1902               (sb!kernel::arg-count-error (condition)
1903                 (values nil (list (format nil "~A" condition))))
1904               (simple-error (condition)
1905                 (values nil (list* (simple-condition-format-control condition)
1906                                    (simple-condition-format-arguments condition))))))
1907           (defun ,compiler (specifier)
1908             (multiple-value-bind (type error-args) (,careful specifier)
1909               (or type
1910                   (apply #'compiler-error error-args))))
1911           (defun ,transform (specifier)
1912             (multiple-value-bind (type error-args) (,careful specifier)
1913               (or type
1914                   (apply #'give-up-ir1-transform
1915                          error-args)))))))
1916   (deffrob specifier-type careful-specifier-type compiler-specifier-type ir1-transform-specifier-type)
1917   (deffrob values-specifier-type careful-values-specifier-type compiler-values-specifier-type ir1-transform-values-specifier-type))
1918
1919 \f
1920 ;;;; utilities used at run-time for parsing &KEY args in IR1
1921
1922 ;;; This function is used by the result of PARSE-DEFTRANSFORM to find
1923 ;;; the lvar for the value of the &KEY argument KEY in the list of
1924 ;;; lvars ARGS. It returns the lvar if the keyword is present, or NIL
1925 ;;; otherwise. The legality and constantness of the keywords should
1926 ;;; already have been checked.
1927 (declaim (ftype (sfunction (list keyword) (or lvar null))
1928                 find-keyword-lvar))
1929 (defun find-keyword-lvar (args key)
1930   (do ((arg args (cddr arg)))
1931       ((null arg) nil)
1932     (when (eq (lvar-value (first arg)) key)
1933       (return (second arg)))))
1934
1935 ;;; This function is used by the result of PARSE-DEFTRANSFORM to
1936 ;;; verify that alternating lvars in ARGS are constant and that there
1937 ;;; is an even number of args.
1938 (declaim (ftype (sfunction (list) boolean) check-key-args-constant))
1939 (defun check-key-args-constant (args)
1940   (do ((arg args (cddr arg)))
1941       ((null arg) t)
1942     (unless (and (rest arg)
1943                  (constant-lvar-p (first arg)))
1944       (return nil))))
1945
1946 ;;; This function is used by the result of PARSE-DEFTRANSFORM to
1947 ;;; verify that the list of lvars ARGS is a well-formed &KEY arglist
1948 ;;; and that only keywords present in the list KEYS are supplied.
1949 (declaim (ftype (sfunction (list list) boolean) check-transform-keys))
1950 (defun check-transform-keys (args keys)
1951   (and (check-key-args-constant args)
1952        (do ((arg args (cddr arg)))
1953            ((null arg) t)
1954          (unless (member (lvar-value (first arg)) keys)
1955            (return nil)))))
1956 \f
1957 ;;;; miscellaneous
1958
1959 ;;; Called by the expansion of the EVENT macro.
1960 (declaim (ftype (sfunction (event-info (or node null)) *) %event))
1961 (defun %event (info node)
1962   (incf (event-info-count info))
1963   (when (and (>= (event-info-level info) *event-note-threshold*)
1964              (policy (or node *lexenv*)
1965                      (= inhibit-warnings 0)))
1966     (let ((*compiler-error-context* node))
1967       (compiler-notify (event-info-description info))))
1968
1969   (let ((action (event-info-action info)))
1970     (when action (funcall action node))))
1971
1972 ;;;
1973 (defun make-cast (value type policy)
1974   (declare (type lvar value)
1975            (type ctype type)
1976            (type policy policy))
1977   (%make-cast :asserted-type type
1978               :type-to-check (maybe-weaken-check type policy)
1979               :value value
1980               :derived-type (coerce-to-values type)))
1981
1982 (defun cast-type-check (cast)
1983   (declare (type cast cast))
1984   (when (cast-reoptimize cast)
1985     (ir1-optimize-cast cast t))
1986   (cast-%type-check cast))
1987
1988 (defun note-single-valuified-lvar (lvar)
1989   (declare (type (or lvar null) lvar))
1990   (when lvar
1991     (let ((use (lvar-uses lvar)))
1992       (cond ((ref-p use)
1993              (let ((leaf (ref-leaf use)))
1994                (when (and (lambda-var-p leaf)
1995                           (null (rest (leaf-refs leaf))))
1996                  (reoptimize-lambda-var leaf))))
1997             ((or (listp use) (combination-p use))
1998              (do-uses (node lvar)
1999                (setf (node-reoptimize node) t)
2000                (setf (block-reoptimize (node-block node)) t)
2001                (reoptimize-component (node-component node) :maybe)))))))
2002
2003 ;;; Return true if LVAR's only use is a non-NOTINLINE reference to a
2004 ;;; global function with one of the specified NAMES.
2005 (defun lvar-fun-is (lvar names)
2006   (declare (type lvar lvar) (list names))
2007   (let ((use (lvar-uses lvar)))
2008     (and (ref-p use)
2009          (let ((leaf (ref-leaf use)))
2010            (and (global-var-p leaf)
2011                 (eq (global-var-kind leaf) :global-function)
2012                 (not (null (member (leaf-source-name leaf) names
2013                                    :test #'equal))))))))