source: branches/working-0711/ccl/compiler/nx0.lisp @ 11823

Last change on this file since 11823 was 11823, checked in by gz, 12 years ago

record source transformation when wrap THE

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 106.8 KB
Line 
1;;; -*- Mode: Lisp; Package: CCL -*-
2;;;
3;;;   Copyright (C) 1994-2001 Digitool, Inc
4;;;   This file is part of OpenMCL. 
5;;;
6;;;   OpenMCL is licensed under the terms of the Lisp Lesser GNU Public
7;;;   License , known as the LLGPL and distributed with OpenMCL as the
8;;;   file "LICENSE".  The LLGPL consists of a preamble and the LGPL,
9;;;   which is distributed with OpenMCL as the file "LGPL".  Where these
10;;;   conflict, the preamble takes precedence. 
11;;;
12;;;   OpenMCL is referenced in the preamble as the "LIBRARY."
13;;;
14;;;   The LLGPL is also available online at
15;;;   http://opensource.franz.com/preamble.html
16
17
18
19(in-package "CCL")
20
21;; :compiler:nx0.lisp - part of the compiler
22
23
24(defstruct pending-declarations
25  vdecls
26  fdecls
27  mdecls)
28
29; Phony AFUNC "defstruct":
30(defun make-afunc (&aux (v (%make-afunc)))
31  (setf (afunc-fn-refcount v) 0)
32  (setf (afunc-fn-downward-refcount v) 0)
33  (setf (afunc-bits v) 0)
34  v)
35
36(defvar *compile-code-coverage* nil "True to instrument for code coverage")
37
38(defvar *nx-blocks* nil)
39(defvar *nx-tags* nil)
40(defvar *nx-parent-function* nil)
41(defvar *nx-current-function* nil)
42(defvar *nx-lexical-environment* nil)
43(defvar *nx-symbol-macros* nil)
44(defvar *nx-inner-functions* nil)
45(defvar *nx-cur-func-name* nil)
46(defvar *nx-current-note* nil)
47(defparameter *nx-source-note-map* nil) ;; there might be external refs, from macros.
48(defvar *nx-form-type* t)
49;(defvar *nx-proclaimed-inline* nil)
50;(defvar *nx-proclaimed-inline* (make-hash-table :size 400 :test #'eq))
51(defvar *nx-proclaimed-ignore* nil)
52(defvar *nx-parsing-lambda-decls* nil) ; el grosso.
53(defparameter *nx-standard-declaration-handlers* nil)
54(defparameter *nx-hoist-declarations* t)
55(defparameter *nx-loop-nesting-level* 0)
56(defvar *nx-break-on-program-errors* t)
57
58(defvar *nx1-vcells* nil)
59(defvar *nx1-fcells* nil)
60
61(defvar *nx1-operators* (make-hash-table :size 300 :test #'eq))
62
63
64; The compiler can (generally) use temporary vectors for VARs.
65(defun nx-cons-var (name &optional (bits 0))
66  (%istruct 'var name bits nil nil nil nil 0 nil))
67
68
69
70
71(defvar *nx-lambdalist* (make-symbol "lambdalist"))
72(defvar *nx-nil* (list (make-symbol "nil")))
73(defvar *nx-t* (list (make-symbol "t")))
74
75(defparameter *nx-current-compiler-policy* (%default-compiler-policy))
76
77(defvar *nx-next-method-var* nil)
78(defvar *nx-call-next-method-function* nil)
79
80(defvar *nx-sfname* nil)
81(defvar *nx-operators* ())
82(defvar *nx-warnings* nil)
83
84(defvar *nx1-compiler-special-forms* nil "Real special forms")
85
86(defmacro without-compiling-code-coverage (&body body)
87  "Disable code coverage in the lexical scope of the form"
88  `(compiler-let ((*nx-current-code-note* nil))
89     ,@body))
90
91(defparameter *nx-never-tail-call*
92  '(error cerror break warn type-error file-error
93    signal-program-error signal-simple-program-error
94    print-call-history
95    #-bccl %get-frame-pointer
96    #-bccl break-loop)
97  "List of functions which never return multiple values and
98   should never be tail-called.")
99
100(defvar *cross-compiling* nil "bootstrapping")
101
102
103(defparameter *nx-operator-result-types*
104  '((#.(%nx1-operator list) . list)
105    (#.(%nx1-operator memq) . list)
106    (#.(%nx1-operator %temp-list) . list)
107    (#.(%nx1-operator assq) . list)
108    (#.(%nx1-operator cons) . cons)
109    (#.(%nx1-operator rplaca) . cons)
110    (#.(%nx1-operator %rplaca) . cons)
111    (#.(%nx1-operator rplacd) . cons)
112    (#.(%nx1-operator %rplacd) . cons)
113    (#.(%nx1-operator %temp-cons) . cons)
114    (#.(%nx1-operator %i+) . fixnum)
115    (#.(%nx1-operator %i-) . fixnum)
116    (#.(%nx1-operator %i*) . fixnum)
117    (#.(%nx1-operator %ilsl) . fixnum)
118    (#.(%nx1-operator %ilsr) . fixnum)
119    (#.(%nx1-operator %iasr) . fixnum)
120    (#.(%nx1-operator %ilogior2) . fixnum)
121    (#.(%nx1-operator %ilogand2) . fixnum)
122    (#.(%nx1-operator %ilogxor2) . fixnum)
123    (#.(%nx1-operator %code-char) . character)
124    (#.(%nx1-operator schar) . character)
125    (#.(%nx1-operator length) . fixnum)
126    (#.(%nx1-operator uvsize) . fixnum)
127    (#.(%nx1-operator %double-float/-2) . double-float)
128    (#.(%nx1-operator %double-float/-2!) . double-float) ; no such operator
129    (#.(%nx1-operator %double-float+-2) . double-float)
130    (#.(%nx1-operator %double-float+-2!) . double-float)
131    (#.(%nx1-operator %double-float--2) . double-float)
132    (#.(%nx1-operator %double-float--2!) . double-float)
133    (#.(%nx1-operator %double-float*-2) . double-float)
134    (#.(%nx1-operator %double-float*-2!) . double-float)
135    (#.(%nx1-operator %short-float/-2) . double-float)
136    (#.(%nx1-operator %short-float+-2) . double-float)
137    (#.(%nx1-operator %short-float--2) . double-float)
138    (#.(%nx1-operator %short-float*-2) . double-float)
139    (#.(%nx1-operator %double-to-single) . single-float)
140    (#.(%nx1-operator %single-to-double) . double-float)
141    (#.(%nx1-operator %fixnum-to-single) . single-float)
142    (#.(%nx1-operator %fixnum-to-double) . double-float)
143    (#.(%nx1-operator char-code) . #.`(integer 0 (,char-code-limit)))
144   ))
145
146(defparameter *nx-operator-result-types-by-name*
147  '((%ilognot . fixnum)
148    (%ilogxor . fixnum)
149    (%ilogand . fixnum)
150    (%ilogior . fixnum)
151    (char-code . #. `(integer 0 (,char-code-limit)))))
152
153(setq *nx-known-declarations*
154  '(special inline notinline type ftype function ignore optimize dynamic-extent ignorable
155    ignore-if-unused settable unsettable
156     notspecial global-function-name debugging-function-name resident))
157
158(defun find-optimize-quantity (name env)
159  (let ((pair ()))
160    (loop
161      (when (listp env) (return))
162      (when (setq pair (assq name (lexenv.mdecls env)))
163        (return (%cdr pair)))
164      (setq env (lexenv.parent-env env)))))
165   
166(defun debug-optimize-quantity (env)
167  (or (find-optimize-quantity 'debug env)
168      *nx-debug*))
169
170(defun space-optimize-quantity (env)
171  (or (find-optimize-quantity 'space env)
172      *nx-space*))
173
174(defun safety-optimize-quantity (env)
175  (or (find-optimize-quantity 'safety env)
176      *nx-safety*))
177
178(defun speed-optimize-quantity (env)
179  (or (find-optimize-quantity 'speed env)
180      *nx-speed*))
181
182(defun compilation-speed-optimize-quantity (env)
183  (or (find-optimize-quantity 'compilation-speed env)
184      *nx-cspeed*))
185
186(defvar *nx-ignore-if-unused* ())
187(defvar *nx-new-p2decls* ())
188(defvar *nx-inlined-self* t)
189(defvar *nx-all-vars* nil)
190(defvar *nx-bound-vars* nil)
191(defvar *nx-punted-vars* nil)
192(defvar *nx-inline-expansions* nil)
193(defparameter *nx-compile-time-compiler-macros* nil)
194(defvar *nx-global-function-name* nil)
195(defvar *nx-can-constant-fold* ())
196(defvar *nx-synonyms* ())
197(defvar *nx-load-time-eval-token* ())
198
199(define-condition compiler-function-overflow (condition) ())
200
201(defun compiler-function-overflow ()
202  (signal 'compiler-function-overflow)
203  (error "Function size exceeds compiler limitation."))
204
205(defvar *compiler-macros* (make-hash-table :size 100 :test #'eq))
206
207;;; Just who was responsible for the "FUNCALL" nonsense ?
208;;; Whoever it is deserves a slow and painful death ...
209
210(defmacro define-compiler-macro  (name arglist &body body &environment env)
211  "Define a compiler-macro for NAME."
212  (let* ((block-name name)
213         (def-name (validate-function-name name)))
214    (unless (eq def-name block-name)
215      (setq block-name (cadr block-name)))
216    (let ((body (parse-macro-1 block-name arglist body env)))
217      `(eval-when (:compile-toplevel :load-toplevel :execute)
218         (eval-when (:load-toplevel :execute)
219           (record-source-file ',name 'compiler-macro))
220         (setf (compiler-macro-function ',name)
221               (nfunction (compiler-macro-function ,name)  ,body))
222         ',name))))
223
224;;; This is silly (as may be the whole idea of actually -using-
225;;; compiler-macros).  Compiler-macroexpand-1 will return a second
226;;; value of NIL if the value returned by the expansion function is EQ
227;;; to the original form.  This differs from the behavior of
228;;; macroexpand-1, but users are not encouraged to write macros which
229;;; return their &whole args (as the DEFINE-COMPILER-MACRO issue
230;;; encourages them to do ...)  Cheer up! Neither of these things have
231;;; to exist!
232(defun compiler-macroexpand-1 (form &optional env)
233  (let ((expander nil)
234        (newdef nil))
235    (if (and (consp form)
236             (symbolp (car form))
237             (setq expander (compiler-macro-function (car form) env)))
238      (values (setq newdef (funcall *macroexpand-hook* expander form env)) (neq newdef form))
239      (values form nil))))
240
241; ... If this exists, it should probably be exported.
242(defun compiler-macroexpand (form &optional env)
243  (multiple-value-bind (new win) (compiler-macroexpand-1 form env)
244    (do* ((won-at-least-once win))
245         ((null win) (values new won-at-least-once))
246      (multiple-value-setq (new win) (compiler-macroexpand-1 new env)))))
247
248
249
250
251(defun compiler-macro-function (name &optional env)
252  "If NAME names a compiler-macro in ENV, return the expansion function, else
253   return NIL. Can be set with SETF when ENV is NIL."
254  (setq name (validate-function-name name))
255  (unless (nx-lexical-finfo name env)
256    (or (cdr (assq name *nx-compile-time-compiler-macros*))
257        (values (gethash name *compiler-macros*)))))
258
259(defun set-compiler-macro-function (name def)
260  (setq name (validate-function-name name))
261  (if def
262    (setf (gethash name *compiler-macros*) def)
263    (remhash name *compiler-macros*))
264  def)
265
266(defsetf compiler-macro-function set-compiler-macro-function)
267
268(defparameter *nx-add-xref-entry-hook* nil
269  "When non-NIL, assumed to be a function of 3 arguments
270which asserts that the specied relation from the current
271function to the indicated name is true.")
272
273;; Cross-referencing
274(defun nx-record-xref-info (relation name)
275  (when (fboundp '%add-xref-entry)
276    (funcall '%add-xref-entry relation *nx-cur-func-name* name)))
277
278
279
280(defun nx-apply-env-hook (hook env &rest args)
281  (declare (dynamic-extent args))
282  (when (fixnump hook) (setq hook (uvref *nx-current-compiler-policy* hook)))
283  (if hook
284    (if (functionp hook)
285      (apply hook env args)
286      t)))
287
288(defun nx-self-calls-inlineable (env)
289  (nx-apply-env-hook policy.inline-self-calls env))
290
291(defun nx-allow-register-allocation (env)
292  (not (nx-apply-env-hook policy.inhibit-register-allocation env)))
293
294(defun nx-trust-declarations (env)
295  (unless (eq (safety-optimize-quantity env) 3)
296    (nx-apply-env-hook policy.trust-declarations env)))
297
298(defun nx-open-code-in-line (env)
299  (nx-apply-env-hook policy.open-code-inline env))
300
301(defun nx-inline-car-cdr (env)
302  (unless (eq (safety-optimize-quantity env) 3)
303    (nx-apply-env-hook policy.inhibit-safety-checking env)))
304
305(defun nx-inhibit-safety-checking (env)
306  (unless (eq (safety-optimize-quantity env) 3)
307    (nx-apply-env-hook policy.inhibit-safety-checking env)))
308
309(defun nx-tailcalls (env)
310  (nx-apply-env-hook policy.allow-tail-recursion-elimination env))
311
312(defun nx-allow-transforms (env)
313  (nx-apply-env-hook policy.allow-transforms env))
314
315(defun nx-force-boundp-checks (var env)
316  (or (eq (safety-optimize-quantity env) 3)
317      (nx-apply-env-hook policy.force-boundp-checks var env)))
318
319(defun nx-substititute-constant-value (symbol value env)
320  (nx-apply-env-hook policy.allow-constant-substitution symbol value env))
321
322#-BOOTSTRAPPED
323(eval-when (compile)
324  (unless (boundp 'policy.declarations-typecheck)
325    (load "ccl:library;lispequ.lisp")))
326
327(defun nx-declarations-typecheck (env)
328  (nx-apply-env-hook policy.declarations-typecheck env))
329
330
331#-bccl
332(defun nx1-default-operator ()
333 (or (gethash *nx-sfname* *nx1-operators*)
334     (error "Bug - operator not found for  ~S" *nx-sfname*)))
335
336(defun nx-new-temp-var (pending &optional (pname "COMPILER-VAR"))
337  (let ((var (nx-new-var pending (make-symbol pname))))
338    (nx-set-var-bits var (%ilogior (%ilsl $vbitignoreunused 1)
339                                   (%ilsl $vbittemporary 1)
340                                   (nx-var-bits var)))
341    var))
342
343(defun nx-new-vdecl (pending name class &optional info)
344  (push (cons name (cons class info)) (pending-declarations-vdecls pending)))
345
346(defun nx-new-fdecl (pending name class &optional info)
347  (push (cons name (cons class info)) (pending-declarations-fdecls pending)))
348
349(defun nx-new-var (pending sym &optional (check t))
350  (nx-init-var pending (nx-cons-var (nx-need-var sym check) 0)))
351                   
352(defun nx-proclaimed-special-p (sym)
353  (setq sym (nx-need-sym sym))
354  (let* ((defenv (definition-environment *nx-lexical-environment*))
355         (specials (if defenv (defenv.specials defenv))))
356    (or (assq sym specials)
357        (proclaimed-special-p sym))))
358
359(defun nx-proclaimed-parameter-p (sym)
360  (or (constantp sym)
361      (multiple-value-bind (special-p info) (nx-lex-info sym t)
362        (or 
363         (and (eq special-p :special) info)
364         (let* ((defenv (definition-environment *nx-lexical-environment*)))
365           (if defenv 
366             (or (%cdr (assq sym (defenv.specials defenv)))
367                 (assq sym (defenv.constants defenv)))))))))
368
369(defun nx-process-declarations (pending decls &optional (env *nx-lexical-environment*) &aux s f)
370  (dolist (decl decls pending)
371    (dolist (spec (%cdr decl))
372      (if (memq (setq s (car spec)) *nx-known-declarations*)
373        (if (setq f (getf *nx-standard-declaration-handlers* s))
374          (funcall f pending spec env))
375        ; Any type name is now (ANSI CL) a valid declaration.
376        ; We should probably do something to distinguish "type names" from "typo names",
377        ; so that (declare (inliMe foo)) warns unless the compiler has some reason to
378        ; believe that 'inliMe' (inlemon) has been DEFTYPEd.
379        (dolist (var (%cdr spec))
380          (if (symbolp var)
381            (nx-new-vdecl pending var 'type s)))))))
382
383; Put all variable decls for the symbol VAR into effect in environment ENV.  Now.
384; Returns list of all new vdecls pertaining to VAR.
385(defun nx-effect-vdecls (pending var env)
386  (let ((vdecls (lexenv.vdecls env))
387        (own nil))
388    (dolist (decl (pending-declarations-vdecls pending) (setf (lexenv.vdecls env) vdecls))
389      (when (eq (car decl) var) 
390        (when (eq (cadr decl) 'type)
391          (let* ((newtype (cddr decl))
392                 (merged-type (nx1-type-intersect var newtype (nx-declared-type var env))))
393             (unless (eq merged-type newtype)
394              (rplacd (cdr decl) merged-type))))
395        (push decl vdecls)
396        (push (cdr decl) own)))
397    own))
398
399
400(defun nx1-typed-var-initform (pending sym form &optional (env *nx-lexical-environment*))
401  (let* ((type t)
402         (form-type (dolist (decl (pending-declarations-vdecls pending)  type)
403                      (when (and (eq (car decl) sym) (eq (cadr decl) 'type))
404                        (setq type (nx1-type-intersect sym (nx-target-type type) (cddr decl))))))
405         (*nx-form-type* (if (nx-trust-declarations env)
406                           form-type
407                           t)))
408    (when (nx-declarations-typecheck env)
409      (setq form `(the ,form-type ,form))
410      (nx-note-source-transformation (caddr form) form))
411    (nx1-typed-form form env)))
412
413; Guess.
414(defun nx-effect-fdecls (pending var env)
415  (let ((fdecls (lexenv.fdecls env))
416        (own nil))
417    (dolist (decl (pending-declarations-fdecls pending) (setf (lexenv.fdecls env) fdecls))
418      (when (eq (car decl) var) 
419        (push decl fdecls)
420        (push (cdr decl) own)))
421    own))
422
423
424
425
426(defun nx-acode-form-typep (form type env)
427  (acode-form-typep form type  (nx-trust-declarations env)))
428
429(defun acode-form-typep (form type trust-decls)
430  (if (acode-p form)
431    (let* ((op (acode-operator form))
432           (opval-p (or (eq op (%nx1-operator fixnum)) (eq op (%nx1-operator immediate))))
433           (optype (acode-form-type form trust-decls)))
434      (values
435       (if optype 
436         (subtypep optype (nx-target-type type))
437         (if opval-p (typep (%cadr form) (nx-target-type type))))))))
438
439(defun nx-acode-form-type (form env)
440  (acode-form-type form (nx-trust-declarations env)))
441
442(defparameter *numeric-acode-ops*
443  (list (%nx1-operator add2)
444        (%nx1-operator sub2)
445        (%nx1-operator mul2)))
446
447
448(defun acode-form-type (form trust-decls)
449  (nx-target-type 
450   (if (acode-p form)
451     (let* ((op (acode-operator form)))
452       (if (eq op (%nx1-operator fixnum))
453         'fixnum
454         (if (eq op (%nx1-operator immediate))
455           (type-of (%cadr form))
456           (and trust-decls
457                (if (eq op (%nx1-operator typed-form))
458                  (if (eq (%cadr form) 'number)
459                    (or (acode-form-type (nx-untyped-form form) trust-decls)
460                        'number)
461                    (%cadr form))
462                  (if (eq op (%nx1-operator lexical-reference))
463                    (let* ((var (cadr form))
464                           (bits (nx-var-bits var))
465                           (punted (logbitp $vbitpunted bits)))
466                      (if (or punted
467                              (eql 0 (%ilogand $vsetqmask bits)))
468                        (var-inittype var)))
469                    (if (or (eq op (%nx1-operator %aref1))
470                            (eq op (%nx1-operator simple-typed-aref2))
471                            (eq op (%nx1-operator general-aref2))
472                            (eq op (%nx1-operator simple-typed-aref3))
473                            (eq op (%nx1-operator general-aref3)))
474                      (let* ((atype (acode-form-type (cadr form) t))
475                             (actype (if atype (specifier-type atype))))
476                        (if (typep actype 'array-ctype)
477                          (type-specifier (array-ctype-specialized-element-type
478                                           actype))))
479                      (if (member op *numeric-acode-ops*)
480                        (multiple-value-bind (f1 f2)
481                            (nx-binop-numeric-contagion (cadr form)
482                                                        (caddr form)
483                                                        trust-decls)
484                          (if (and (acode-form-typep f1 'float trust-decls)
485                                   (acode-form-typep f2 'float trust-decls))
486
487                            (if (or (acode-form-typep f1 'double-float trust-decls)
488                                    (acode-form-typep f2 'double-float trust-decls))
489                            'double-float
490                            'single-float)))
491                        (cdr (assq op *nx-operator-result-types*)))))))))))))
492
493(defun nx-binop-numeric-contagion (form1 form2 trust-decls)
494  (cond ((acode-form-typep form1 'double-float trust-decls)
495         (if (acode-form-typep form2 'double-float trust-decls)
496           (values form1 form2)
497           (let* ((c2 (acode-real-constant-p form2)))
498             (if c2
499               (values form1 (make-acode (%nx1-operator immediate)
500                                         (float c2 0.0d0)))
501               (if (acode-form-typep form2 'fixnum trust-decls)
502                 (values form1 (make-acode (%nx1-operator %fixnum-to-double)
503                                           form2))
504                 (values form1 form2))))))
505        ((acode-form-typep form2 'double-float trust-decls)
506         (let* ((c1 (acode-real-constant-p form1)))
507           (if c1
508             (values (make-acode (%nx1-operator immediate)
509                                 (float c1 0.0d0)) form2)
510             (if (acode-form-typep form1 'fixnum trust-decls)
511               (values (make-acode (%nx1-operator %fixnum-to-double)
512                                   form1) form2)
513               (values form1 form2)))))
514        ((acode-form-typep form1 'single-float trust-decls)
515         (if (acode-form-typep form2 'single-float trust-decls)
516           (values form1 form2)
517           (let* ((c2 (acode-real-constant-p form2)))
518             (if c2
519               (values form1 (make-acode (%nx1-operator immediate)
520                                         (float c2 0.0f0)))
521               (if (acode-form-typep form2 'fixnum trust-decls)
522                 (values form1 (make-acode (%nx1-operator %fixnum-to-single)
523                                           form2))
524                 (values form1 form2))))))
525        ((acode-form-typep form2 'single-float trust-decls)
526         (let* ((c1 (acode-real-constant-p form1)))
527           (if c1
528             (values (make-acode (%nx1-operator immediate)
529                                 (float c1 0.0f0)) form2)
530             (if (acode-form-typep form1 'fixnum trust-decls)
531               (values (make-acode (%nx1-operator %fixnum-to-single)
532                                   form1) form2)
533               (values form1 form2)))))
534        (t
535         (values form1 form2))))
536
537(defun acode-punted-var-p (var)
538  (let ((bits (nx-var-bits var)))
539    (and (%ilogbitp $vbitpunted bits)
540         (not (%ilogbitp $vbitspecial bits)))))
541
542;; Use acode-unwrapped-form-value to reason about the value of a form at
543;; compile time.   To actually generate code, use acode-unwrapped-form.
544(defun acode-unwrapped-form-value (form)
545  (setq form (acode-unwrapped-form form))
546  (when (and (acode-p form)
547             (eq (acode-operator form) (%nx1-operator with-code-note)))
548    (setq form (acode-unwrapped-form-value (caddr form))))
549  form)
550
551; Strip off any type info or "punted" lexical references.
552; ??? Is it true that the "value" of the punted reference is unwrapped ? ???
553(defun acode-unwrapped-form (form) 
554  (while (and (consp (setq form (nx-untyped-form form)))
555           (eq (%car form) (%nx1-operator lexical-reference))
556           (acode-punted-var-p (cadr form)))
557    (setq form (var-ea (cadr form))))
558  form)
559
560(defun acode-fixnum-form-p (x)
561  (setq x (acode-unwrapped-form-value x))
562  (if (acode-p x)
563    (if (eq (acode-operator x) (%nx1-operator fixnum)) 
564      (cadr x))))
565
566(defun acode-integer-constant-p (x bits)
567  (let* ((int (or (acode-fixnum-form-p x)
568                  (progn
569                    (setq x (acode-unwrapped-form x))
570                    (if (acode-p x)
571                      (if (and (eq (acode-operator x) (%nx1-operator immediate))
572                               (typep (cadr x) 'fixnum))
573                        (cadr x)))))))
574    (and int
575         (or
576           (typep int `(signed-byte ,bits))
577           (typep int `(unsigned-byte ,bits)))
578         int)))
579
580(defun acode-real-constant-p (x)
581  (or (acode-fixnum-form-p x)
582      (progn
583        (setq x (acode-unwrapped-form x))
584        (if (acode-p x)
585          (if (and (eq (acode-operator x) (%nx1-operator immediate))
586                   (typep (cadr x) 'real))
587            (cadr x))))))
588
589
590
591(defun nx-lookup-target-uvector-subtag (name)
592  (or (cdr (assoc name (arch::target-uvector-subtags (backend-target-arch *target-backend*))))
593      (nx-error "Type ~s not supported on target ~s"
594                name (backend-target-arch-name *target-backend*))))
595
596(defun nx-target-uvector-subtag-name (subtag)
597  (or (car (rassoc subtag (arch::target-uvector-subtags (backend-target-arch *target-backend*))))
598      (nx-error "Subtag ~s not native on target ~s"
599                subtag (backend-target-arch-name *target-backend*))))
600
601(defun nx-error-for-simple-2d-array-type (type-keyword)
602  (ecase type-keyword
603    (:simple-vector arch::error-object-not-simple-array-t-2d)
604    (:simple-string arch::error-object-not-simple-array-char-2d)
605    (:bit-vector arch::error-object-not-simple-array-bit-2d)
606    (:unsigned-8-bit-vector arch::error-object-not-simple-array-u8-2d)
607    (:signed-8-bit-vector arch::error-object-not-simple-array-s8-2d)
608    (:unsigned-16-bit-vector arch::error-object-not-simple-array-u16-2d)
609    (:signed-16-bit-vector arch::error-object-not-simple-array-s16-2d)
610    (:unsigned-32-bit-vector arch::error-object-not-simple-array-u32-2d)
611    (:signed-32-bit-vector arch::error-object-not-simple-array-s32-2d)
612    (:unsigned-64-bit-vector arch::error-object-not-simple-array-u64-2d)
613    (:signed-64-bit-vector arch::error-object-not-simple-array-s64-2d)
614    (:double-float-vector arch::error-object-not-simple-array-double-float-2d)
615    (:single-float-vector arch::error-object-not-simple-array-double-float-2d)
616    (:fixnum-vector arch::error-object-not-simple-array-fixnum-2d)))
617
618(defun nx-error-for-simple-3d-array-type (type-keyword)
619  (ecase type-keyword
620    (:simple-vector arch::error-object-not-simple-array-t-3d)
621    (:simple-string arch::error-object-not-simple-array-char-3d)
622    (:bit-vector arch::error-object-not-simple-array-bit-3d)
623    (:unsigned-8-bit-vector arch::error-object-not-simple-array-u8-3d)
624    (:signed-8-bit-vector arch::error-object-not-simple-array-s8-3d)
625    (:unsigned-16-bit-vector arch::error-object-not-simple-array-u16-3d)
626    (:signed-16-bit-vector arch::error-object-not-simple-array-s16-3d)
627    (:unsigned-32-bit-vector arch::error-object-not-simple-array-u32-3d)
628    (:signed-32-bit-vector arch::error-object-not-simple-array-s32-3d)
629    (:unsigned-64-bit-vector arch::error-object-not-simple-array-u64-3d)
630    (:signed-64-bit-vector arch::error-object-not-simple-array-s64-3d)
631    (:double-float-vector arch::error-object-not-simple-array-double-float-3d)
632    (:single-float-vector arch::error-object-not-simple-array-double-float-3d)
633    (:fixnum-vector arch::error-object-not-simple-array-fixnum-3d)))
634
635(defun acode-s16-constant-p (x)
636  (setq x (acode-unwrapped-form x))
637  (if (acode-p x)
638    (let* ((op (acode-operator x)))
639      (if (eql op (%nx1-operator fixnum))
640        (let* ((val (cadr x)))
641          (if (target-word-size-case
642               (32 (typep val '(signed-byte #.(- 16 2))))
643               (64 (typep val '(signed-byte #.(- 16 3)))))
644            (ash val (target-word-size-case
645                      (32 2)
646                      (64 3)))))
647        (if (eql op (%nx1-operator %unbound-marker))
648          (arch::target-unbound-marker-value
649           (backend-target-arch *target-backend*))
650          (if (eql op (%nx1-operator %slot-unbound-marker))
651            (arch::target-slot-unbound-marker-value
652             (backend-target-arch *target-backend*))))))))
653
654(defun acode-s32-constant-p (x)
655  (setq x (acode-unwrapped-form x))
656  (if (acode-p x)
657    (let* ((op (acode-operator x)))
658      (if (eql op (%nx1-operator fixnum))
659        (let* ((val (cadr x)))
660          (if (target-word-size-case
661               (32 (typep val '(signed-byte #.(- 32 2))))
662               (64 (typep val '(signed-byte #.(- 32 3)))))
663            (ash val (target-word-size-case
664                      (32 2)
665                      (64 3)))))
666        (if (eql op (%nx1-operator %unbound-marker))
667          (arch::target-unbound-marker-value
668           (backend-target-arch *target-backend*))
669          (if (eql op (%nx1-operator %slot-unbound-marker))
670            (arch::target-slot-unbound-marker-value
671             (backend-target-arch *target-backend*))))))))
672
673(defun acode-fixnum-type-p (form trust-decls)
674  (or (acode-fixnum-form-p form)
675      (and trust-decls
676           (acode-p form)
677           (eq (acode-operator form) (%nx1-operator typed-form))
678           (subtypep (cadr form) 'fixnum))))
679
680
681(defun nx-acode-fixnum-type-p (form env)
682    (acode-fixnum-type-p form (nx-trust-declarations env)))
683
684; Is acode-expression the result of alphatizing (%int-to-ptr <integer>) ?
685(defun acode-absolute-ptr-p (acode-expression &optional skip)
686  (and (acode-p acode-expression)
687       (or skip (prog1 (eq (acode-operator acode-expression) (%nx1-operator %macptrptr%))
688                  (setq acode-expression (%cadr acode-expression))))
689       (eq (acode-operator acode-expression) (%nx1-operator %consmacptr%))
690       (eq (acode-operator (setq acode-expression (%cadr acode-expression))) 
691           (%nx1-operator %immediate-int-to-ptr))
692       (let ((op (acode-operator (setq acode-expression (%cadr acode-expression)))))
693         (if (or (eq op (%nx1-operator fixnum))
694                 (and (eq op (%nx1-operator immediate))
695                      (integerp (%cadr acode-expression))))
696           (%cadr acode-expression)))))
697
698(defun nx-check-vdecl-var-ref (decl)
699  (unless (eq (cadr decl) 'special)
700    (let* ((sym (car decl))
701           (info (nx-lex-info sym)))
702      (when (or (eq info :symbol-macro)
703                (and (null info) (not (nx-proclaimed-special-p sym))))
704        (nx1-whine :unknown-declaration-variable (cadr decl) sym)))))
705
706
707(defun nx-effect-other-decls (pending env)
708  (flet ((merge-decls (new old)
709                      (dolist (decl new old) (pushnew decl old :test #'eq))))
710    (let ((vdecls (pending-declarations-vdecls pending))
711          (fdecls (pending-declarations-fdecls pending))
712          (mdecls (pending-declarations-mdecls pending)))
713      (when vdecls
714        (let ((env-vdecls (lexenv.vdecls env)))
715          (dolist (decl vdecls (setf (lexenv.vdecls env) env-vdecls))
716            (unless (memq decl env-vdecls)
717              (nx-check-vdecl-var-ref decl)
718              (when (eq (cadr decl) 'type)
719                (let* ((var (car decl))
720                       (newtype (cddr decl))
721                       (merged-type (nx1-type-intersect var newtype (nx-declared-type var env))))
722                  (unless (eq merged-type newtype)
723                    (rplacd (cdr decl) merged-type))))
724              (push decl env-vdecls)))))
725      (when fdecls (setf (lexenv.fdecls env) (merge-decls fdecls (lexenv.vdecls env))))
726      (when mdecls (setf (lexenv.mdecls env) (merge-decls mdecls (lexenv.mdecls env))))
727      (setq *nx-inlined-self* (and (nx-self-calls-inlineable env) 
728                                   (let ((name *nx-global-function-name*)) 
729                                     (and name (not (nx-declared-notinline-p name env))))))
730      (unless (nx-allow-register-allocation env)
731        (nx-inhibit-register-allocation))
732      (setq *nx-new-p2decls*
733            (if (eql (safety-optimize-quantity env) 3)
734              (logior $decl_full_safety
735                      (if (nx-tailcalls env) $decl_tailcalls 0))
736              (%ilogior
737                (if (nx-tailcalls env) $decl_tailcalls 0)
738                (if (nx-open-code-in-line env) $decl_opencodeinline 0)
739                (if (nx-inhibit-safety-checking env) $decl_unsafe 0)
740                (if (nx-trust-declarations env) $decl_trustdecls 0)))))))
741
742#|     
743(defun nx-find-misc-decl (declname env)
744  (loop
745    (unless (and env (eq (uvref env 0) 'lexical-environment)) (return))
746    (dolist (mdecl (lexenv.mdecls env))
747      (if (atom mdecl)
748        (if (eq mdecl declname)
749          (return-from nx-find-misc-decl t))
750        (if (eq (%car mdecl) declname)
751          (return-from nx-find-misc-decl (%cdr mdecl)))))
752    (setq env (lexenv.parent-env env))))
753|#
754
755
756(defun nx-bad-decls (decls)
757  (nx1-whine :unknown-declaration decls))
758
759
760
761(defnxdecl special (pending decl env)
762  (declare (ignore env))
763  (dolist (s (%cdr decl))
764    (if (symbolp s) 
765      (nx-new-vdecl pending s 'special)
766      (nx-bad-decls decl))))
767
768(defnxdecl dynamic-extent (pending decl env)
769  (declare (ignore env))
770  (dolist (s (%cdr decl))
771    (if (symbolp s) 
772      (nx-new-vdecl pending s 'dynamic-extent t)
773      (if (and (consp s)
774               (eq (%car s) 'function)
775               (consp (%cdr s))
776               (valid-function-name-p (cadr s))
777               (setq s (validate-function-name (cadr s))))
778        (nx-new-fdecl pending s 'dynamic-extent t)
779        (nx-bad-decls decl)))))
780
781(defnxdecl ignorable (pending decl env)
782  (declare (ignore env))
783  (dolist (s (%cdr decl))
784    (if (symbolp s) 
785      (nx-new-vdecl pending s 'ignore-if-unused t)
786      (if (and (consp s)
787               (eq (%car s) 'function)
788               (consp (%cdr s))
789               (valid-function-name-p (cadr s))
790               (setq s (validate-function-name (cadr s))))
791        (nx-new-fdecl pending s 'ignore-if-unused t)
792        (nx-bad-decls decl)))))
793
794(defnxdecl ftype (pending decl env)
795  (declare (ignore env))
796  (destructuring-bind (type &rest fnames) (%cdr decl)
797    (dolist (s fnames)
798      (nx-new-fdecl pending s 'ftype type))))
799
800(defnxdecl settable (pending decl env)
801  (nx-settable-decls pending decl env t))
802
803(defnxdecl unsettable (pending decl env)
804  (nx-settable-decls pending decl env nil))
805
806(defun nx-settable-decls (pending decl env val)
807  (declare (ignore env))
808  (dolist (s (%cdr decl))
809    (if (symbolp s)
810      (nx-new-vdecl pending s 'settable val)
811      (nx-bad-decls decl))))
812
813(defnxdecl type (pending decl env)
814  (declare (ignore env))
815  (labels ((kludge (type) ; 0 => known, 1 => unknown, 2=> illegal
816             (cond ((type-specifier-p type)
817                    0)
818                   ((and (consp type)
819                         (member (car type) '(and or))
820                         (not (null (list-length type))))
821                    (do ((result 0 (max result (kludge (car tail))))
822                         (tail (cdr type) (cdr tail)))
823                        ((null tail)
824                         result)))
825                   ((not (symbolp type))
826                    ;;>>>> nx-bad-decls shouldn't signal a fatal error!!!!
827                    ;;>>>> Most callers of nx-bad-decls should just ignore the
828                    ;;>>>> losing decl element and proceed with the rest
829                    ;;>>>>  (ie (declare (ignore foo (bar) baz)) should
830                    ;;>>>>   have the effect of ignoring foo and baz as well
831                    ;;>>>>   as WARNING about the mal-formed declaration.)
832                    (nx-bad-decls decl)
833                    2)
834                   (t 1))))
835    (let* ((spec (%cdr decl))
836           (type (car spec)))
837      (case (kludge type)
838        ((0)
839         (dolist (sym (cdr spec))
840           (if (symbolp sym)
841             (nx-new-vdecl pending sym 'type type)
842             (nx-bad-decls decl))))
843        ((1)
844         (dolist (sym (cdr spec))
845           (unless (symbolp sym)
846             (nx-bad-decls decl))))
847        ((2)
848         (nx-bad-decls decl))))))
849
850
851
852(defnxdecl global-function-name (pending decl env)
853  (declare (ignore pending))
854  (when *nx-parsing-lambda-decls*
855    (let ((name (cadr decl)))
856      (setq *nx-global-function-name* (setf (afunc-name *nx-current-function*) name))
857      (setq *nx-inlined-self* (not (nx-declared-notinline-p name env))))))
858
859(defnxdecl debugging-function-name (pending decl env)
860  (declare (ignore pending env))
861  (when *nx-parsing-lambda-decls*
862    (setf (afunc-name *nx-current-function*) (cadr decl))))
863
864(defnxdecl resident (pending decl env)
865  (declare (ignore env pending))
866  (declare (ignore decl))
867  (nx-decl-set-fbit $fbitresident))
868
869
870(defun nx-inline-decl (pending decl val &aux valid-name)
871  (dolist (s (%cdr decl))
872    (multiple-value-setq (valid-name s) (valid-function-name-p s))
873    (if valid-name
874      (progn
875        (if (nx-self-call-p s nil t)
876          (setq *nx-inlined-self* val))
877        (nx-new-fdecl pending s 'inline (if val 'inline 'notinline)))
878      (nx-bad-decls decl))))
879
880(defnxdecl inline (pending decl env)
881  (declare (ignore env))
882  (nx-inline-decl pending decl t))
883
884(defnxdecl notinline (pending decl env)
885  (declare (ignore env))
886  (nx-inline-decl pending decl nil))
887
888(defnxdecl ignore (pending decl env)
889  (declare (ignore env))
890  (dolist (s (%cdr decl))
891    (if (symbolp s)     
892      (nx-new-vdecl pending s 'ignore t)
893      (if (and (consp s)
894               (eq (%car s) 'function)
895               (consp (%cdr s))
896               (valid-function-name-p (cadr s))
897               (setq s (validate-function-name (cadr s))))
898        (nx-new-fdecl pending s 'ignore t)
899        (nx-bad-decls decl)))))
900
901(defnxdecl ignore-if-unused (pending decl env)
902  (declare (ignore env))
903  (dolist (s (%cdr decl))
904    (if (symbolp s) 
905      (nx-new-vdecl pending s 'ignore-if-unused)
906      (nx-bad-decls decl))))
907
908(defun nx-self-call-p (name &optional ignore-lexical (allow *nx-inlined-self*))
909  (when (and name (symbolp name))
910    (let ((current-afunc *nx-current-function*)
911          (target-afunc (unless ignore-lexical (nth-value 1 (nx-lexical-finfo name)))))
912      (or (eq current-afunc target-afunc)
913          (and allow
914               (eq name *nx-global-function-name*)
915               (null target-afunc)
916               (null (afunc-parent current-afunc)))))))
917
918(defun nx-check-var-usage (var)
919  (let* ((sym (var-name var))
920         (bits (nx-var-bits var))
921         (expansion (var-ea var))
922         (setqed (%ilogbitp $vbitsetq bits))
923         (reffed (%ilogbitp $vbitreffed bits))
924         (closed (%ilogbitp $vbitclosed bits))
925         (special (%ilogbitp $vbitspecial bits))
926         (ignored (%ilogbitp $vbitignore bits))
927         (ignoreunused (%ilogbitp $vbitignoreunused bits)))
928    (if (or special reffed closed)
929      (progn
930        (if ignored (nx1-whine :ignore sym))
931        (nx-set-var-bits var (%ilogand (nx-check-downward-vcell var bits) (%ilognot (%ilsl $vbitignore 1)))))
932      (progn
933        (if (and setqed ignored) (nx1-whine :ignore sym))
934        (or ignored ignoreunused 
935            (progn (and (consp expansion) (eq (car expansion) :symbol-macro) (setq sym (list :symbol-macro sym))) (nx1-whine :unused sym)))
936        (when (%izerop (%ilogand bits (%ilogior $vrefmask $vsetqmask)))
937          (nx-set-var-bits var (%ilogior (%ilsl $vbitignore 1) bits)))))))
938
939; if an inherited var isn't setqed, it gets no vcell.  If it -is- setqed, but
940; all inheritors are downward, the vcell can be stack-consed.  Set a bit so that
941; the right thing happens when the var is bound.
942; Set the bit for the next-method var even if it is not setqed.
943(defun nx-check-downward-vcell (v bits)
944  (if (and (%ilogbitp $vbitclosed bits)
945           (or (%ilogbitp $vbitsetq bits)
946               (eq v *nx-next-method-var*))
947           (nx-afuncs-downward-p v (afunc-inner-functions *nx-current-function*)))
948    (%ilogior (%ilsl $vbitcloseddownward 1) bits)
949    bits))
950
951; afunc is "downward wrt v" if it doesn't inherit v or if all refs to afunc
952; are "downward" and no inner function of afunc is not downward with respect to v.
953(defun nx-afunc-downward-p (v afunc)
954  (or (dolist (i (afunc-inherited-vars afunc) t)
955        (when (eq (nx-root-var i) v) (return nil)))
956      (if (nx-afuncs-downward-p v (afunc-inner-functions afunc))
957        (eq (afunc-fn-refcount afunc)
958            (afunc-fn-downward-refcount afunc)))))
959
960(defun nx-afuncs-downward-p (v afuncs)
961  (dolist (afunc afuncs t)
962    (unless (nx-afunc-downward-p v afunc) (return nil))))
963
964(defun nx1-punt-bindings (vars initforms)
965  (dolist (v vars)
966    (nx1-punt-var v (pop initforms))))
967
968;;; at the beginning of a binding construct, note which lexical
969;;; variables are bound to other variables and the number of setqs
970;;; done so far on the initform.  After executing the body, if neither
971;;; variable has been closed over, the new variable hasn't been
972;;; setq'ed, and the old guy wasn't setq'ed in the body, the binding
973;;; can be punted.
974(defun nx1-note-var-binding (var initform)
975  (let* ((init (nx-untyped-form initform))
976         (inittype (nx-acode-form-type initform *nx-lexical-environment*))
977         (bits (nx-var-bits var)))
978    (when (%ilogbitp $vbitspecial bits) (nx-record-xref-info :binds (var-name var)))
979    (when inittype (setf (var-inittype var) inittype))
980    (when (and (not (%ilogbitp $vbitspecial bits))
981               (consp init))
982      (let* ((op (acode-operator init)))
983        (if (eq op (%nx1-operator lexical-reference))
984          (let* ((target (%cadr init))
985                 (setq-count (%ilsr 8 (%ilogand $vsetqmask (nx-var-bits target)))))
986            (unless (eq setq-count (%ilsr 8 $vsetqmask))
987              (cons var (cons setq-count target))))
988          (if (and (%ilogbitp $vbitdynamicextent bits)
989                   (or (eq op (%nx1-operator closed-function))
990                       (eq op (%nx1-operator simple-function))))
991            (let* ((afunc (%cadr init)))
992              (setf (afunc-fn-downward-refcount afunc)
993                    (afunc-fn-refcount afunc)
994                    (afunc-bits afunc) (logior (ash 1 $fbitdownward) (ash 1 $fbitbounddownward)
995                                               (the fixnum (afunc-bits afunc))))
996              nil)))))))
997
998
999;;; Process entries involving variables bound to other variables at
1000;;; the end of a binding construct.  Each entry is of the form
1001;;; (source-var setq-count . target-var), where setq-count is the
1002;;; assignment count of TARGET-VAR at the time that the binding's
1003;;; initform was evaluated (not, in the case of LET, at the time that
1004;;; the bindinw was established.).  If the target isn't closed-over
1005;;; and SETQed (somewhere), and wasn't setqed in the body (e.g.,
1006;;; still has the same assignment-count as it had when the initform
1007;;; was executed), then we can "punt" the source (and replace references
1008;;; to it with references to the target.)
1009;;; It obviously makes no sense to do this if the source is SPECIAL;
1010;;; in some cases (LET), we create the source variable and add it to
1011;;; this alist before it's known whether or not the source variable
1012;;; is SPECIAL. so we have to ignore that case here.
1013(defun nx1-check-var-bindings (alist)
1014  (dolist (pair alist)
1015    (let* ((var (car pair))
1016           (target (cddr pair))
1017           (vbits (nx-var-bits var))
1018           (target-bits (nx-var-bits target)))
1019      (unless (or
1020               ;; var can't be special, setq'ed or closed; target can't be
1021               ;; setq'ed AND closed.
1022               (neq (%ilogand vbits (%ilogior (%ilsl $vbitsetq 1)
1023                                              (%ilsl $vbitclosed 1)
1024                                              (%ilsl $vbitspecial 1))) 0)
1025               (eq (%ilogior (%ilsl $vbitsetq 1) (%ilsl $vbitclosed 1)) 
1026                   (%ilogand
1027                     (%ilogior (%ilsl $vbitsetq 1) (%ilsl $vbitclosed 1))
1028                     target-bits))
1029               (neq (%ilsr 8 (%ilogand $vsetqmask target-bits)) (cadr pair)))
1030             (push (cons var target) *nx-punted-vars*)))))
1031
1032(defun nx1-punt-var (var initform)
1033  (let* ((bits (nx-var-bits var))
1034         (mask (%ilogior (%ilsl $vbitsetq 1) (ash -1 $vbitspecial) (%ilsl $vbitclosed 1)))
1035         (nrefs (%ilogand $vrefmask bits))
1036         (val (nx-untyped-form initform))
1037         (op (if (acode-p val) (acode-operator val))))
1038    (when (%izerop (%ilogand mask bits))
1039      (if
1040        (or 
1041         (nx-t val)
1042         (nx-null val)
1043         (and (eql nrefs 1) (not (logbitp $vbitdynamicextent bits)) ( acode-absolute-ptr-p val t))
1044         (eq op (%nx1-operator fixnum))
1045         (eq op (%nx1-operator immediate)))
1046        (progn
1047          (nx-set-var-bits var (%ilogior (%ilsl $vbitpuntable 1) bits)))))
1048    (when (and (%ilogbitp $vbitdynamicextent bits)
1049               (or (eq op (%nx1-operator closed-function))
1050                   (eq op (%nx1-operator simple-function))))
1051      (let* ((afunc (cadr val)))
1052        (setf (afunc-bits afunc) (%ilogior (%ilsl $fbitbounddownward 1) (afunc-bits afunc))
1053              (afunc-fn-downward-refcount afunc) 1))) 
1054    nil))
1055           
1056(defnxdecl optimize (pending specs env)
1057  (declare (ignore env))
1058  (let* ((q nil)
1059         (v nil)
1060         (mdecls (pending-declarations-mdecls pending)))
1061    (dolist (spec (%cdr specs) (setf (pending-declarations-mdecls pending) mdecls))
1062      (if (atom spec)
1063        (setq q spec v 3)
1064        (setq q (%car spec) v (cadr spec)))
1065      (if (and (fixnump v) (<= 0 v 3) (memq q '(speed space compilation-speed safety debug)))
1066        (push (cons q v) mdecls)
1067        (nx-bad-decls specs)))))
1068
1069(defun %proclaim-optimize (specs &aux q v)
1070 (dolist (spec specs)
1071  (if (atom spec)
1072   (setq q spec v 3)
1073   (setq q (%car spec) v (cadr spec)))
1074  (when (and (fixnump v) (<= 0 v 3))
1075   (if (eq q 'speed)
1076    (setq *nx-speed* v)
1077    (if (eq q 'space)
1078     (setq *nx-space* v)
1079     (if (eq q 'compilation-speed)
1080      (setq *nx-cspeed* v)
1081      (if (eq q 'safety)
1082       (setq *nx-safety* v)
1083       (if (eq q 'debug)
1084         (setq *nx-debug* v)))))))))
1085
1086(defun nx-lexical-finfo (sym &optional (env *nx-lexical-environment*))
1087  (let* ((info nil)
1088         (barrier-crossed nil))
1089    (if env
1090      (loop
1091        (when (eq 'barrier (lexenv.variables env))
1092          (setq barrier-crossed t))
1093        (when (setq info (%cdr (assq sym (lexenv.functions env))))
1094          (return (values info (if (and (eq (car info) 'function)
1095                                        (consp (%cdr info)))
1096                                 (progn
1097                                   (when barrier-crossed
1098                                     (nx-error "Illegal reference to lexically-defined function ~S." sym))
1099                                   (%cadr info))))))
1100        (if (listp (setq env (lexenv.parent-env env)))
1101          (return (values nil nil))))
1102      (values nil nil))))
1103
1104(defun nx-inline-expansion (sym &optional (env *nx-lexical-environment*) global-only)
1105  (let* ((lambda-form nil)
1106         (containing-env nil)
1107         (token nil))
1108    (if (and (nx-declared-inline-p sym env)
1109             (not (gethash sym *nx1-alphatizers*))
1110             (not *nx-current-code-note*))
1111      (multiple-value-bind (info afunc) (unless global-only (nx-lexical-finfo sym env))
1112        (if info (setq token afunc 
1113                       containing-env (afunc-environment afunc)
1114                       lambda-form (afunc-lambdaform afunc)))
1115        (setq info (cdr (retrieve-environment-function-info sym env)))
1116        (if (def-info.lambda info)
1117            (setq lambda-form (def-info.lambda info)
1118                  token sym
1119                  containing-env (new-lexical-environment (definition-environment env)))
1120            (unless info
1121              (if (cdr (setq info (assq sym *nx-globally-inline*)))
1122                (setq lambda-form (%cdr info)
1123                      token sym
1124                      containing-env (new-lexical-environment (new-definition-environment nil))))))))
1125    (values lambda-form (nx-closed-environment env containing-env) token)))
1126
1127(defun nx-closed-environment (current-env target)
1128  (when target
1129    (let* ((intervening-functions nil))
1130      (do* ((env current-env (lexenv.parent-env env)))
1131           ((or (eq env target) (null env) (istruct-typep env 'definition-environment)))
1132        (let* ((fn (lexenv.lambda env)))
1133          (when fn (push fn intervening-functions))))
1134      (let* ((result target))
1135        (dolist (fn intervening-functions result)
1136          (setf (lexenv.lambda (setq result (new-lexical-environment result))) fn))))))
1137
1138(defun nx-root-var (v)
1139  (do* ((v v bits)
1140        (bits (var-bits v) (var-bits v)))
1141       ((fixnump bits) v)))
1142
1143(defun nx-reconcile-inherited-vars (more)
1144  (let ((last nil)) ; Bop 'til ya drop.
1145    (loop
1146      (setq last more more nil)
1147      (dolist (callee last)
1148        (dolist (caller (afunc-callers callee))
1149          (unless (or (eq caller callee)
1150                      (eq caller (afunc-parent callee)))
1151            (dolist (v (afunc-inherited-vars callee))
1152              (let ((root-v (nx-root-var v)))
1153                (unless (dolist (caller-v (afunc-inherited-vars caller))
1154                          (when (eq root-v (nx-root-var caller-v))
1155                            (return t)))
1156                  ; caller must inherit root-v in order to call callee without using closure.
1157                  ; can't just bind afunc & call nx-lex-info here, 'cause caller may have
1158                  ; already shadowed another var with same name.  So:
1159                  ; 1) find the ancestor of callee which bound v; this afunc is also an ancestor
1160                  ;    of caller
1161                  ; 2) ensure that each afunc on the inheritance path from caller to this common
1162                  ;    ancestor inherits root-v.
1163                  (let ((ancestor (afunc-parent callee))
1164                        (inheritors (list caller)))
1165                    (until (eq (setq v (var-bits v)) root-v)
1166                      (setq ancestor (afunc-parent ancestor)))
1167                    (do* ((p (afunc-parent caller) (afunc-parent p)))
1168                         ((eq p ancestor))
1169                      (push p inheritors))
1170                    (dolist (f inheritors)
1171                      (setq v (nx-cons-var (var-name v) v))
1172                      (unless (dolist (i (afunc-inherited-vars f))
1173                                (when (eq root-v (nx-root-var i))
1174                                  (return (setq v i))))
1175                        (pushnew f more)
1176                        (push v (afunc-inherited-vars f))
1177                        ; change shared structure of all refs in acode with one swell foop.
1178                        (nx1-afunc-ref f))))))))))   
1179      (unless more (return)))))
1180
1181(defun nx-inherit-var (var binder current)
1182  (if (eq binder current)
1183    (progn
1184      (nx-set-var-bits var (%ilogior2 (%ilsl $vbitclosed 1) (nx-var-bits var)))
1185      var)
1186    (let ((sym (var-name var)))
1187      (or (dolist (already (afunc-inherited-vars current))
1188            (when (eq sym (var-name already)) (return already)))
1189          (progn
1190            (setq var (nx-cons-var sym (nx-inherit-var var binder (afunc-parent current))))
1191            (push var (afunc-inherited-vars current))
1192            var)))))
1193
1194(defun nx-lex-info (sym &optional current-only)
1195  (let* ((current-function *nx-current-function*)
1196         (catch nil)
1197         (barrier-crossed nil))
1198    (multiple-value-bind 
1199      (info afunc)
1200      (do* ((env *nx-lexical-environment* (lexenv.parent-env env))
1201            (continue env (and env (not (istruct-typep env 'definition-environment))))
1202            (binder current-function (or (if continue (lexenv.lambda env)) binder)))
1203           ((or (not continue) (and (neq binder current-function) current-only)) 
1204            (values nil nil))
1205        (let ((vars (lexenv.variables env)))
1206          (if (eq vars 'catch) 
1207            (setq catch t)
1208            (if (eq vars 'barrier)
1209              (setq barrier-crossed t)
1210              (let ((v (dolist (var vars)
1211                         (when (eq (var-name var) sym) (return var)))))
1212                (when v (return (values v binder)))
1213                (dolist (decl (lexenv.vdecls env))
1214                  (when (and (eq (car decl) sym)
1215                             (eq (cadr decl) 'special))
1216                    (return-from nx-lex-info (values :special nil nil)))))))))
1217      (if info
1218        (if (var-expansion info)
1219          (values :symbol-macro (cdr (var-expansion info)) info)
1220          (if (%ilogbitp $vbitspecial (nx-var-bits info))
1221            (values :special info nil)
1222            (if barrier-crossed
1223              (nx-error "Illegal reference to lexically defined variable ~S." sym)
1224              (if (eq afunc current-function)
1225                (values info nil catch)
1226                (values (nx-inherit-var info afunc current-function) t catch)))))
1227        (values nil nil nil)))))
1228
1229
1230(defun nx-block-info (blockname &optional (afunc *nx-current-function*) &aux
1231  blocks
1232  parent
1233  (toplevel (eq afunc *nx-current-function*))
1234  blockinfo)
1235 (when afunc
1236  (setq
1237   blocks (if toplevel *nx-blocks* (afunc-blocks afunc))
1238   blockinfo (assq blockname blocks)
1239   parent (afunc-parent afunc))
1240  (if blockinfo
1241   (values blockinfo nil)
1242   (when parent
1243    (when (setq blockinfo (nx-block-info blockname parent))
1244     (values blockinfo t))))))
1245
1246(defun nx-tag-info (tagname &optional (afunc *nx-current-function*) &aux
1247                            tags
1248                            parent
1249                            index
1250                            counter
1251                            (toplevel (eq afunc *nx-current-function*))
1252                            taginfo)
1253  (when afunc
1254    (setq
1255     tags (if toplevel *nx-tags* (afunc-tags afunc))
1256     taginfo (assoc tagname tags)
1257     parent (afunc-parent afunc))
1258    (if taginfo
1259      (values taginfo nil)
1260      (when (and parent (setq taginfo (nx-tag-info tagname parent)))
1261        (unless (setq index (cadr taginfo))
1262          (setq counter (caddr taginfo))
1263          (%rplaca counter (%i+ (%car counter) 1))
1264          (setq index (%car counter))
1265          (%rplaca (%cdr taginfo) index))
1266        (values taginfo index)))))
1267
1268(defun nx1-transitively-punt-bindings (pairs) 
1269  (dolist (pair (nreverse pairs))
1270    (let* ((var         (%car pair))
1271           (boundto     (%cdr pair))
1272           (varbits     (nx-var-bits var))
1273           (boundtobits (nx-var-bits boundto)))
1274      (declare (fixnum varbits boundtobits))
1275      (unless (eq (%ilogior
1276                    (%ilsl $vbitsetq 1)
1277                    (%ilsl $vbitclosed 1))
1278                  (%ilogand
1279                    (%ilogior
1280                      (%ilsl $vbitsetq 1)
1281                      (%ilsl $vbitclosed 1))
1282                    boundtobits))
1283        ;; Can't happen -
1284        (unless (%izerop (%ilogand (%ilogior
1285                                     (%ilsl $vbitsetq 1) 
1286                                     (ash -1 $vbitspecial)
1287                                     (%ilsl $vbitclosed 1)) varbits))
1288          (error "Bug-o-rama - \"punted\" var had bogus bits. ~
1289Or something. Right? ~s ~s" var varbits))
1290        (let* ((varcount     (%ilogand $vrefmask varbits)) 
1291               (boundtocount (%ilogand $vrefmask boundtobits)))
1292          (nx-set-var-bits var (%ilogior
1293                                 (%ilsl $vbitpuntable 1)
1294                                 (%i- varbits varcount)))
1295          (nx-set-var-bits
1296           boundto
1297           (%i+ (%i- boundtobits boundtocount)
1298                (%ilogand $vrefmask
1299                          (%i+ (%i- boundtocount 1) varcount)))))))))
1300
1301;;; Home-baked handler-case replacement.  About 10 times as fast as full handler-case.
1302;;;(LET ((S 0)) (DOTIMES (I 1000000) (INCF S))) took 45,678 microseconds
1303;;;(LET ((S 0)) (DOTIMES (I 1000000) (BLOCK X (ERROR (CATCH 'X (RETURN-FROM X (INCF S))))))) took 57,485
1304;;;(LET ((S 0)) (DOTIMES (I 1000000) (HANDLER-CASE (INCF S) (ERROR (C) C)))) took 168,947
1305(defmacro with-program-error-handler (handler &body body)
1306  (let ((tag (gensym)))
1307    `(block ,tag
1308       (,handler (catch 'program-error-handler (return-from ,tag (progn ,@body)))))))
1309
1310(defun runtime-program-error-form (c)
1311  `(signal-program-error "Invalid program: ~a" ,(princ-to-string c)))
1312
1313(defun nx1-compile-lambda (name lambda-form &optional
1314                                 (p (make-afunc))
1315                                 q
1316                                 parent-env
1317                                 (policy *default-compiler-policy*)
1318                                 load-time-eval-token)
1319
1320  (if q
1321     (setf (afunc-parent p) q))
1322
1323  ;; In the case of a method function, the name will get reset at load time to the
1324  ;; method object.  However, during compilation, we want any inner functions to use
1325  ;; the fully qualified method name, so store that.
1326  (when (method-lambda-p lambda-form)
1327    (setq name (or *nx-method-warning-name* name)))
1328
1329  (setf (afunc-name p)
1330        (let ((parent-name (and (afunc-parent p) (afunc-name (afunc-parent p)))))
1331          (if parent-name
1332            (if (and (consp parent-name) (eq (%car parent-name) :internal))
1333              (if name
1334                `(:internal ,name ,@(cdr parent-name))
1335                parent-name)
1336              (if name
1337                `(:internal ,name ,parent-name)
1338                `(:internal ,parent-name)))
1339            name)))
1340
1341  (unless (lambda-expression-p lambda-form)
1342    (nx-error "~S is not a valid lambda expression." lambda-form))
1343
1344  (let* ((*nx-current-function* p)
1345         (*nx-parent-function* q)
1346         (*nx-current-note* (or *nx-current-note* (nx-source-note lambda-form)))
1347         (*nx-lexical-environment* (new-lexical-environment parent-env))
1348         (*nx-load-time-eval-token* load-time-eval-token)
1349         (*nx-all-vars* nil)
1350         (*nx-bound-vars* nil)
1351         (*nx-punted-vars* nil)
1352         (*nx-current-compiler-policy* policy)
1353         (*nx-blocks* nil)
1354         (*nx-tags* nil)
1355         (*nx-loop-nesting-level* 0)
1356         (*nx-inner-functions* nil)
1357         (*nx-global-function-name* nil)
1358         (*nx-warnings* nil)
1359         (*nx1-fcells* nil)
1360         (*nx1-vcells* nil)
1361         (*nx-inline-expansions* nil)
1362         (*nx-parsing-lambda-decls* nil)
1363         (*nx-next-method-var* (if q *nx-next-method-var*))
1364         (*nx-call-next-method-function* (if q *nx-call-next-method-function*))
1365         (*nx-cur-func-name* name))
1366    (if (%non-empty-environment-p *nx-lexical-environment*)
1367      (setf (afunc-bits p) (logior (ash 1 $fbitnonnullenv) (the fixnum (afunc-bits p)))))
1368
1369    (setf (afunc-lambdaform p) lambda-form)
1370
1371    (when *nx-current-note*
1372      (setf (afunc-lfun-info p)
1373            (list* 'function-source-note *nx-current-note* (afunc-lfun-info p))))
1374
1375    (with-program-error-handler
1376        (lambda (c)
1377          (setf (afunc-acode p) (nx1-lambda '(&rest args) `(args ,(runtime-program-error-form c)) nil)))
1378      (handler-bind ((warning (lambda (c)
1379                                (nx1-whine :program-error c)
1380                                (muffle-warning c)))
1381                     (program-error (lambda (c)
1382                                      (when *nx-break-on-program-errors*
1383                                        (cerror "continue compilation ignoring this form" c))
1384                                      (when (typep c 'compile-time-program-error)
1385                                        (setq c (make-condition 'simple-program-error
1386                                                                :format-control (simple-condition-format-control c)
1387                                                                :format-arguments (simple-condition-format-arguments c))))
1388                                      (unless *nx-break-on-program-errors*
1389                                        (nx1-whine :program-error c))
1390                                      (throw 'program-error-handler c))))
1391        (multiple-value-bind (body decls)
1392            (with-program-error-handler (lambda (c) (runtime-program-error-form c))
1393              (parse-body (%cddr lambda-form) *nx-lexical-environment* t))
1394          (setf (afunc-acode p) (nx1-lambda (%cadr lambda-form) body decls)))))
1395
1396    (when *record-pc-mapping*
1397      (when *nx-current-note*
1398        (setf (acode-note (afunc-acode p)) *nx-current-note*)))
1399
1400    (nx1-transitively-punt-bindings *nx-punted-vars*)
1401    (setf (afunc-blocks p) *nx-blocks*)
1402    (setf (afunc-tags p) *nx-tags*)
1403    (setf (afunc-inner-functions p) *nx-inner-functions*)
1404    (setf (afunc-all-vars p) *nx-all-vars*)
1405    (setf (afunc-vcells p) *nx1-vcells*)
1406    (setf (afunc-fcells p) *nx1-fcells*)
1407    (let* ((warnings (merge-compiler-warnings *nx-warnings*))
1408           (name *nx-cur-func-name*))       
1409      (dolist (inner *nx-inner-functions*)
1410        (dolist (w (afunc-warnings inner))
1411          (push name (compiler-warning-function-name w))
1412          (push w warnings)))
1413      (setf (afunc-warnings p) warnings))
1414    p))
1415
1416(defun method-lambda-p (form)
1417  (and (consp form)
1418       (consp (setq form (%cdr form)))       
1419       (eq (caar form) '&method)))
1420
1421
1422(defun nx1-lambda (ll body decls &aux (l ll) methvar)
1423  (let* ((old-env *nx-lexical-environment*)
1424         (*nx-bound-vars* *nx-bound-vars*))
1425    (with-nx-declarations (pending)
1426      (let* ((*nx-parsing-lambda-decls* t))
1427        (nx-process-declarations pending decls))
1428      (when (eq (car l) '&lap)
1429        (let ((bits nil))
1430          (unless (and (eq (length (%cdr l)) 1) (fixnump (setq bits (%cadr l))))
1431            (unless (setq bits (encode-lambda-list (%cdr l)))
1432              (nx-error "invalid lambda-list  - ~s" l)))
1433          (return-from nx1-lambda
1434                       (make-acode
1435                        (%nx1-operator lambda-list)
1436                        (list (cons '&lap bits))
1437                        nil
1438                        nil
1439                        nil
1440                        nil
1441                        (nx1-env-body body old-env)
1442                        *nx-new-p2decls*))))
1443      (when (eq (car l) '&method)
1444        (setf (afunc-bits *nx-current-function*)
1445              (%ilogior (%ilsl $fbitmethodp 1)
1446                        (afunc-bits *nx-current-function*)))
1447        (setq *nx-inlined-self* nil)
1448        (setq *nx-next-method-var* (setq methvar (let ((var (nx-new-var
1449                                                             pending
1450                                                             (%cadr ll))))
1451                                                   (nx-set-var-bits var (%ilogior 
1452                                                                          (%ilsl $vbitignoreunused 1) 
1453                                                                          ;(%ilsl $vbitnoreg 1)
1454                                                                          (nx-var-bits var)))
1455                                                   var)))
1456                                                   
1457        (setq ll (%cddr ll)))
1458      (multiple-value-bind (req opt rest keys auxen lexpr)
1459                           (nx-parse-simple-lambda-list pending ll)
1460        (nx-effect-other-decls pending *nx-lexical-environment*)
1461        (setq body (nx1-env-body body old-env))
1462        (nx1-punt-bindings (%car auxen) (%cdr auxen))         
1463        (when methvar
1464          (push methvar req)
1465          (unless (eq 0 (%ilogand (%ilogior (%ilsl $vbitreffed 1)
1466                                            (%ilsl $vbitclosed 1)
1467                                            (%ilsl $vbitsetq 1))
1468                                  (nx-var-bits methvar)))
1469            (setf (afunc-bits *nx-current-function*)
1470                  (%ilogior 
1471                   (%ilsl $fbitnextmethp 1)
1472                   (afunc-bits *nx-current-function*)))))
1473        (let ((acode (make-acode
1474                      (%nx1-operator lambda-list) 
1475                      req
1476                      opt 
1477                      (if lexpr (list rest) rest)
1478                      keys
1479                      auxen
1480                      body
1481                      *nx-new-p2decls*
1482                      *nx-current-code-note*)))
1483          acode)))))
1484
1485(defun nx-parse-simple-lambda-list (pending ll &aux
1486                                              req
1487                                              opt
1488                                              rest
1489                                              keys
1490                                              lexpr
1491                                              sym)
1492  (multiple-value-bind (ok reqsyms opttail resttail keytail auxtail)
1493                       (verify-lambda-list ll)
1494    (unless ok (nx-error "Bad lambda list : ~S" ll))
1495    (dolist (var reqsyms)
1496      (push (nx-new-var pending var t) req))
1497    (when (eq (pop opttail) '&optional)
1498      (let* (optvars optinits optsuppliedp)
1499        (until (eq opttail resttail) 
1500          (setq sym (pop opttail))
1501          (let* ((var sym)
1502                 (initform nil)
1503                 (spvar nil))
1504            (when (consp var)
1505              (setq sym (pop var) initform (pop var) spvar (%car var)))
1506            (push (nx1-typed-var-initform pending sym initform) optinits)
1507            (push (nx-new-var pending sym t) optvars)
1508            (push (if spvar (nx-new-var pending spvar t)) optsuppliedp)))
1509        (if optvars
1510          (setq opt (list (nreverse optvars) (nreverse optinits) (nreverse optsuppliedp)))
1511          (nx1-whine :lambda ll))))
1512    (let ((temp (pop resttail)))
1513      (when (or (eq temp '&rest)
1514                (setq lexpr (eq temp '&lexpr)))
1515        (setq rest (nx-new-var pending (%car resttail) t))))
1516    (when (eq (%car keytail) '&key) 
1517      (setq keytail (%cdr keytail))
1518      (let* ((keysyms ())
1519             (keykeys ())
1520             (keyinits ())
1521             (keysupp ())
1522             (kallowother (not (null (memq '&allow-other-keys ll))))
1523             (kvar ())
1524             (kkey ())
1525             (kinit ())
1526             (ksupp))
1527        (until (eq keytail auxtail)
1528          (unless (eq (setq sym (pop keytail)) '&allow-other-keys)     
1529            (setq kinit *nx-nil* ksupp nil)
1530            (if (atom sym)
1531              (setq kvar sym kkey (make-keyword sym))
1532              (progn
1533                (if (consp (%car sym))
1534                  (setq kkey (%caar sym) kvar (%cadar sym))
1535                  (progn
1536                    (setq kvar (%car sym))
1537                    (setq kkey (make-keyword kvar))))
1538                (setq kinit (nx1-typed-var-initform pending kvar (%cadr sym)))
1539                (setq ksupp (%caddr sym))))
1540            (push (nx-new-var pending kvar t) keysyms)
1541            (push kkey keykeys)
1542            (push kinit keyinits)
1543            (push (if ksupp (nx-new-var pending ksupp t)) keysupp)))
1544        (setq 
1545         keys
1546         (list
1547          kallowother
1548          (nreverse keysyms)
1549          (nreverse keysupp)
1550          (nreverse keyinits)
1551          (apply #'vector (nreverse keykeys))))))
1552    (let (auxvals auxvars)
1553      (dolist (pair (%cdr auxtail))
1554        (let* ((auxvar (nx-pair-name pair))
1555               (auxval (nx1-typed-var-initform pending auxvar (nx-pair-initform pair))))
1556          (push auxval auxvals)
1557          (push (nx-new-var pending auxvar t) auxvars)))
1558      (values
1559       (nreverse req) 
1560       opt 
1561       rest
1562       keys
1563       (list (nreverse auxvars) (nreverse auxvals))
1564       lexpr))))
1565
1566(defun nx-new-structured-var (pending sym)
1567  (if sym
1568    (nx-new-var pending sym t)
1569    (nx-new-temp-var pending)))
1570
1571(defun nx-parse-structured-lambda-list (pending ll &optional no-acode whole-p &aux
1572                                           req
1573                                           opt
1574                                           rest
1575                                           keys
1576                                           sym)
1577  (multiple-value-bind (ok reqsyms opttail resttail keytail auxtail all whole structured-p)
1578                       (verify-lambda-list ll t whole-p nil)
1579    (declare (ignore all))
1580    (unless ok (nx-error "Bad lambda list : ~S" ll))
1581    (if (or whole (and whole-p structured-p)) (setq whole (nx-new-structured-var pending whole)))
1582    (dolist (var reqsyms)
1583      (push (if (symbolp var)
1584                    (nx-new-structured-var pending var)
1585                    (nx-structured-lambda-form pending var no-acode))
1586                  req))
1587    (when (eq (pop opttail) '&optional)
1588      (let* (optvars optinits optsuppliedp)
1589        (until (eq opttail resttail) 
1590          (setq sym (pop opttail))
1591          (let* ((var sym)
1592                 (initform nil)
1593                 (spvar nil))
1594            (when (consp var)
1595              (setq sym (pop var) initform (pop var) spvar (%car var)))
1596            (push (if no-acode initform (nx1-form initform)) optinits)
1597            (push (if (symbolp sym)
1598                          (nx-new-structured-var pending sym)
1599                          (nx-structured-lambda-form pending sym no-acode))
1600                        optvars)
1601            (push (if spvar (nx-new-var pending spvar)) optsuppliedp)))
1602        (if optvars
1603          (setq opt (list (nreverse optvars) (nreverse optinits) (nreverse optsuppliedp)))
1604          (nx1-whine :lambda ll))))
1605    (let ((var (pop resttail)))
1606      (when (or (eq var '&rest)
1607                (eq var '&body))
1608        (setq var (pop resttail)
1609              rest (if (symbolp var)
1610                     (nx-new-structured-var pending var)
1611                     (nx-structured-lambda-form pending var no-acode)))))
1612    (when (eq (%car keytail) '&key) 
1613      (setq keytail (%cdr keytail))
1614      (let* ((keysyms ())
1615             (keykeys ())
1616             (keyinits ())
1617             (keysupp ())
1618             (kallowother (not (null (memq '&allow-other-keys ll))))
1619             (kvar ())
1620             (kkey ())
1621             (kinit ())
1622             (ksupp))
1623        (until (eq keytail auxtail)
1624          (unless (eq (setq sym (pop keytail)) '&allow-other-keys)     
1625            (setq kinit *nx-nil* ksupp nil)
1626            (if (atom sym)
1627              (setq kvar sym kkey (make-keyword sym))
1628              (progn
1629                (if (consp (%car sym))
1630                  (setq kkey (%caar sym) kvar (%cadar sym))
1631                  (progn
1632                    (setq kvar (%car sym))
1633                    (setq kkey (make-keyword kvar))))
1634                (setq kinit (if no-acode (%cadr sym) (nx1-form (%cadr sym))))
1635                (setq ksupp (%caddr sym))))
1636            (push (if (symbolp kvar)
1637                          (nx-new-structured-var pending kvar)
1638                          (nx-structured-lambda-form pending kvar no-acode))
1639                        keysyms)
1640            (push kkey keykeys)
1641            (push kinit keyinits)
1642            (push (if ksupp (nx-new-var pending ksupp)) keysupp)))
1643        (setq 
1644         keys
1645         (list
1646          kallowother
1647          (nreverse keysyms)
1648          (nreverse keysupp)
1649          (nreverse keyinits)
1650          (apply #'vector (nreverse keykeys))))))
1651    (let (auxvals auxvars)
1652      (dolist (pair (%cdr auxtail))
1653        (let ((auxvar (nx-pair-name pair))
1654              (auxval (nx-pair-initform pair)))
1655          (push (if no-acode auxval (nx1-form auxval)) auxvals)
1656          (push (nx-new-var pending auxvar) auxvars)))
1657      (values
1658       (nreverse req) 
1659       opt 
1660       rest 
1661       keys
1662       (list (nreverse auxvars) (nreverse auxvals))
1663       whole))))
1664
1665(defun nx-structured-lambda-form (pending l &optional no-acode)
1666  (multiple-value-bind (req opt rest keys auxen whole)
1667                       (nx-parse-structured-lambda-list pending l no-acode t)
1668    (list (%nx1-operator lambda-list) whole req opt rest keys auxen)))
1669
1670(defun nx1-form (form &optional (*nx-lexical-environment* *nx-lexical-environment*))
1671  (let* ((*nx-form-type* (if (and (consp form) (eq (car form) 'the))
1672                           (nx-target-type (cadr form))
1673                           t)))
1674    (nx1-typed-form form *nx-lexical-environment*)))
1675
1676(defun nx1-typed-form (original env)
1677  (with-program-error-handler
1678      (lambda (c)
1679        (let ((replacement (runtime-program-error-form c)))
1680          (nx-note-source-transformation original replacement)
1681          (nx1-transformed-form (nx-transform replacement env) env original)))
1682    (nx1-transformed-form (nx-transform original env) env original)))
1683
1684(defun nx1-transformed-form (form env &optional original)
1685  (let* ((*nx-current-note* (or (nx-source-note form) *nx-current-note*))
1686         (*nx-current-code-note*  (and *nx-current-code-note*
1687                                       (or (nx-ensure-code-note form original *nx-current-code-note*)
1688                                           (compiler-bug "No source note for ~s -> ~s" original form))))
1689         (acode (if (consp form)
1690                  (nx1-combination form env)
1691                  (let* ((symbolp (non-nil-symbol-p form))
1692                         (constant-value (unless symbolp form))
1693                         (constant-symbol-p nil))
1694                    (if symbolp 
1695                      (multiple-value-setq (constant-value constant-symbol-p) 
1696                        (nx-transform-defined-constant form env)))
1697                    (if (and symbolp (not constant-symbol-p))
1698                      (nx1-symbol form env)
1699                      (nx1-immediate (nx-unquote constant-value)))))))
1700    (when *record-pc-mapping*
1701      (setf (acode-note acode) (nx-source-note form)))
1702    (if *nx-current-code-note*
1703      (make-acode (%nx1-operator with-code-note)
1704                  *nx-current-code-note*
1705                  acode)
1706      acode)))
1707
1708(defun nx1-prefer-areg (form env)
1709  (nx1-form form env))
1710
1711(defun nx1-target-fixnump (form)
1712  (when (typep form 'integer)
1713       (let* ((target (backend-target-arch *target-backend*)))
1714         (and
1715          (>= form (arch::target-most-negative-fixnum target))
1716          (<= form (arch::target-most-positive-fixnum target))))))
1717
1718
1719(defun nx1-immediate (form)
1720  (if (or (eq form t) (null form))
1721    (nx1-sysnode form)
1722    (make-acode 
1723     (if (nx1-target-fixnump form) 
1724       (%nx1-operator fixnum)
1725        (%nx1-operator immediate))   ; Screw: chars
1726     form)))
1727
1728(defun nx-constant-form-p (form)
1729  (setq form (nx-untyped-form form))
1730  (and (or (nx-null form)
1731           (nx-t form)
1732           (and (acode-p form)
1733                (or (eq (acode-operator form) (%nx1-operator immediate))
1734                    (eq (acode-operator form) (%nx1-operator fixnum))
1735                    (eq (acode-operator form) (%nx1-operator simple-function))
1736                    (and (eq (acode-operator form) (%nx1-operator with-code-note))
1737                         (setq form (nx-constant-form-p (%caddr form)))))))
1738       form))
1739
1740(defun nx-natural-constant-p (form)
1741  (setq form (nx-untyped-form form))
1742  (if (consp form)
1743    (let* ((val (if (or (eq (acode-operator form) (%nx1-operator fixnum))
1744                        (eq (acode-operator form) (%nx1-operator immediate)))
1745                  (cadr form))))
1746      (target-word-size-case
1747       (32 (and (typep val '(unsigned-byte 32)) val))
1748       (64 (and (typep val '(unsigned-byte 64)) val))))))
1749
1750(defun nx-u32-constant-p (form)
1751  (setq form (nx-untyped-form form))
1752  (if (consp form)
1753    (let* ((val (if (or (eq (acode-operator form) (%nx1-operator fixnum))
1754                        (eq (acode-operator form) (%nx1-operator immediate)))
1755                  (cadr form))))
1756      (and (typep val '(unsigned-byte 32)) val))))
1757
1758(defun nx-u31-constant-p (form)
1759  (setq form (nx-untyped-form form))
1760  (if (consp form)
1761    (let* ((val (if (or (eq (acode-operator form) (%nx1-operator fixnum))
1762                        (eq (acode-operator form) (%nx1-operator immediate)))
1763                  (cadr form))))
1764      (and (typep val '(unsigned-byte 31)) val))))
1765
1766
1767;;; Reference-count vcell, fcell refs.
1768(defun nx1-note-vcell-ref (sym)
1769  (let* ((there (assq sym *nx1-vcells*))
1770         (count (expt 4 *nx-loop-nesting-level*)))
1771    (if there
1772      (%rplacd there (%i+ (%cdr there) count))
1773      (push (cons sym count) *nx1-vcells*)))
1774  sym)
1775
1776(defun nx1-note-fcell-ref (sym)
1777  (let* ((there (assq sym *nx1-fcells*))
1778         (count (expt 4 *nx-loop-nesting-level*)))
1779    (if there
1780      (%rplacd there (%i+ (%cdr there) count))
1781      (push (cons sym count) *nx1-fcells*))
1782    sym))
1783
1784; Note that "simple lexical refs" may not be; that's the whole problem ...
1785(defun nx1-symbol (form &optional (env *nx-lexical-environment*))
1786  (let* ((type (nx-declared-type form))
1787         (form
1788          (multiple-value-bind (info inherited-p more)
1789                               (nx-lex-info form)
1790            (if (and info (neq info :special))
1791              (if (eq info :symbol-macro)
1792                (progn
1793                  (nx-set-var-bits more (%ilogior (%ilsl $vbitreffed 1) (nx-var-bits more)))
1794                  (if (eq type t)
1795                    (nx1-form inherited-p)
1796                    (nx1-form `(the ,(prog1 type (setq type t)) ,inherited-p))))
1797                (progn
1798                  (when (not inherited-p)
1799                    (nx-set-var-bits info (%ilogior2 (%ilsl $vbitreffed 1) (nx-var-bits info)))
1800                    (nx-adjust-ref-count info))
1801                  (make-acode (%nx1-operator lexical-reference) info)))
1802              (make-acode
1803               (if (nx1-check-special-ref form info)
1804                   (progn
1805                     (nx-record-xref-info :references form)
1806                     (if (nx-global-p form env)
1807                         (%nx1-operator global-ref)
1808                         (if (and (not (nx-force-boundp-checks form env))
1809                                  (or (nx-proclaimed-parameter-p form)
1810                                  (assq form *nx-compile-time-types*)
1811                                  (assq form *nx-proclaimed-types*)
1812                                  (nx-open-code-in-line env)))
1813                             (%nx1-operator bound-special-ref)
1814                             (%nx1-operator special-ref))))
1815                   (%nx1-operator free-reference))
1816               (nx1-note-vcell-ref form))))))
1817    (if (eq type t)
1818        form
1819      (make-acode (%nx1-operator typed-form) type form))))
1820
1821(defun nx1-check-special-ref (form auxinfo)
1822  (or (eq auxinfo :special) 
1823      (nx-proclaimed-special-p form)
1824      (let ((defenv (definition-environment *nx-lexical-environment*)))
1825        (unless (and defenv (eq (car (defenv.type defenv)) :execute) (boundp form))
1826          (nx1-whine :special form))
1827        nil)))
1828
1829
1830
1831(defun nx1-whine (about &rest forms)
1832  (if #-BOOTSTRAPPED (fboundp 'compiler-warning-source-note) #+BOOTSTRAPPED T
1833    (push (make-condition (or (cdr (assq about *compiler-whining-conditions*)) 'compiler-warning)
1834                          :function-name (list *nx-cur-func-name*)
1835                          :source-note *nx-current-note*
1836                          :warning-type about
1837                          :args (or forms (list nil)))
1838          *nx-warnings*)
1839   
1840    (push (make-condition (or (cdr (assq about *compiler-whining-conditions*)) 'compiler-warning)
1841                          :function-name (list *nx-cur-func-name*)
1842                          :warning-type about
1843                          :args (or forms (list nil)))
1844          *nx-warnings*))
1845  nil)
1846
1847(defun p2-whine (afunc about &rest forms)
1848  (let* ((warning (make-condition (or (cdr (assq about *compiler-whining-conditions*)) 'compiler-warning)
1849                                  :function-name (list (afunc-name afunc))
1850                                  :warning-type about
1851                                  :args (or forms (list nil)))))
1852    (push warning (afunc-warnings afunc))
1853    (do* ((p (afunc-parent afunc) (afunc-parent p)))
1854         ((null p) warning)
1855      (let* ((pname (afunc-name p)))
1856        (push pname (compiler-warning-function-name warning))
1857        (push warning (afunc-warnings p))))))
1858
1859(defun nx1-type-intersect (form type1 type2 &optional env)
1860  (declare (ignore env)) ; use it when deftype records info in env.  Fix this then ...
1861  (let* ((ctype1 (if (typep type1 'ctype) type1 (specifier-type type1)))
1862         (ctype2 (if (typep type2 'ctype) type2 (specifier-type type2)))
1863         (intersection (type-intersection ctype1 ctype2)))
1864    (if (eq intersection *empty-type*)
1865      (let ((type1 (if (typep type1 'ctype)
1866                     (type-specifier type1)
1867                     type1))
1868            (type2 (if (typep type2 'ctype)
1869                     (type-specifier type2)
1870                     type2)))
1871        (nx1-whine :type-conflict form type1 type2)))
1872    (type-specifier intersection)))
1873                 
1874
1875
1876(defun nx-declared-notinline-p (sym env)
1877  (setq sym (maybe-setf-function-name sym))
1878  (loop
1879    (when (listp env)
1880      (return (and (symbolp sym)
1881                   (proclaimed-notinline-p sym))))
1882    (dolist (decl (lexenv.fdecls env))
1883      (when (and (eq (car decl) sym)
1884                 (eq (cadr decl) 'inline))
1885         (return-from nx-declared-notinline-p (eq (cddr decl) 'notinline))))
1886    (setq env (lexenv.parent-env env))))
1887
1888
1889
1890(defun nx1-combination (form env)
1891  (destructuring-bind (sym &rest args)
1892                      form
1893    (if (symbolp sym)
1894      (let* ((*nx-sfname* sym) special)
1895        (if (and (setq special (gethash sym *nx1-alphatizers*))
1896                 (or (not (functionp (fboundp sym)))
1897                     (memq sym '(apply funcall ;; see bug #285
1898                                 %defun        ;; see bug #295
1899                                 ))
1900                     (< (safety-optimize-quantity env) 3))
1901                 ;(not (nx-lexical-finfo sym env))
1902                 (not (nx-declared-notinline-p sym *nx-lexical-environment*)))
1903          (funcall special form env) ; pass environment arg ...
1904          (progn           
1905            (nx1-typed-call sym args))))
1906      (if (lambda-expression-p sym)
1907        (nx1-lambda-bind (%cadr sym) args (%cddr sym))
1908      (nx-error "~S is not a symbol or lambda expression in the form ~S ." sym form)))))
1909
1910(defun nx1-treat-as-call (args)
1911  (nx1-typed-call (car args) (%cdr args)))
1912
1913(defun nx1-typed-call (sym args)
1914  (multiple-value-bind (type errors-p) (nx1-call-result-type sym args)
1915    (let ((form (nx1-call sym args nil nil errors-p)))
1916      (if (eq type t)
1917        form
1918        (make-acode (%nx1-operator typed-form) type form)))))
1919
1920(defvar *format-arg-functions* '((format . 1) (format-to-string . 1) (error . 0) (warn . 0)
1921                                 (y-or-n-p . 0) (yes-or-no-p . 0)
1922                                 (signal-simple-program-error . 0)
1923                                 (signal-simple-condition . 1)
1924                                 (signal-reader-error . 1)
1925                                 (%method-combination-error . 0)
1926                                 (%invalid-method-error . 1)
1927                                 (nx-compile-time-error . 0)
1928                                 (nx-error . 0)
1929                                 (compiler-bug . 0)))
1930
1931#-BOOTSTRAPPED (unless (fboundp 'nx1-check-format-call) (fset 'nx1-check-format-call (lambda (&rest x) (declare (ignore x)))))
1932
1933;;; Wimpy.
1934(defun nx1-call-result-type (sym &optional (args nil args-p) spread-p)
1935  (let* ((env *nx-lexical-environment*)
1936         (global-def nil)
1937         (lexenv-def nil)
1938         (defenv-def nil)
1939         (somedef nil)
1940         (whined nil))
1941    (when (and sym 
1942               (symbolp sym)
1943               (not (find-ftype-decl sym env))
1944               (not (setq lexenv-def (nth-value 1 (nx-lexical-finfo sym))))
1945               (null (setq defenv-def (retrieve-environment-function-info sym env)))
1946               (neq sym *nx-global-function-name*)
1947               (not (functionp (setq global-def (fboundp sym)))))
1948      (if args-p
1949        (nx1-whine :undefined-function sym args spread-p)
1950        (nx1-whine :undefined-function sym))
1951      (setq whined t))
1952    (when (and args-p
1953               (not spread-p)
1954               (setq somedef (cdr (assq sym *format-arg-functions*)))
1955               (setq somedef (nthcdr somedef args))
1956               (stringp (car somedef)))
1957      (when (nx1-check-format-call (car somedef) (cdr somedef) env)
1958        (setq whined t)))
1959    (when (and args-p (setq somedef (or lexenv-def defenv-def global-def)))
1960      (multiple-value-bind (deftype reason)
1961          (nx1-check-call-args somedef args spread-p)
1962        (when deftype
1963          (nx1-whine deftype sym reason args spread-p)
1964          (setq whined t))))
1965    (values (nx-target-type *nx-form-type*) whined)))
1966
1967(defun find-ftype-decl (sym env)
1968  (setq sym (maybe-setf-function-name sym))
1969  (loop 
1970    (when (listp env)
1971      (return (and (symbolp sym)
1972                   (proclaimed-ftype sym))))
1973    (dolist (fdecl (lexenv.fdecls env))
1974      (declare (list fdecl))
1975      (when (and (eq (car fdecl) sym)
1976                 (eq (car (the list (cdr fdecl))) 'ftype))
1977        (return-from find-ftype-decl (cdr (the list (cdr fdecl))))))
1978    (setq env (lexenv.parent-env env))))
1979
1980(defun innermost-lfun-bits-keyvect (def)
1981  (declare (notinline innermost-lfun-bits-keyvect))
1982  (let* ((inner-def (closure-function (find-unencapsulated-definition def)))
1983         (bits (lfun-bits inner-def))
1984         (keys (lfun-keyvect inner-def)))
1985    (declare (fixnum bits))
1986    (when (and (eq (ash 1 $lfbits-gfn-bit)
1987                   (logand bits (logior (ash 1 $lfbits-gfn-bit)
1988                                        (ash 1 $lfbits-method-bit))))
1989               (logbitp $lfbits-keys-bit bits))
1990      (setq bits (logior (ash 1 $lfbits-aok-bit) bits)
1991            keys nil))
1992    (values bits keys)))
1993
1994
1995(defun nx1-check-call-args (def arglist spread-p)
1996  (let* ((deftype (if (functionp def) 
1997                    :global-mismatch
1998                    (if (istruct-typep def 'afunc)
1999                      :lexical-mismatch
2000                      :environment-mismatch)))
2001         (reason nil))
2002    (multiple-value-bind (bits keyvect)
2003                         (case deftype
2004                           (:global-mismatch (innermost-lfun-bits-keyvect def))
2005                           (:environment-mismatch
2006                              (values (def-info.lfbits (cdr def)) (def-info.keyvect (cdr def))))
2007                           (t (let* ((lambda-form (afunc-lambdaform def)))
2008                                (if (lambda-expression-p lambda-form)
2009                                  (encode-lambda-list (cadr lambda-form))))))
2010      (when bits
2011        (unless (typep bits 'fixnum) (error "Bug: Bad bits ~s!" bits))
2012        (let* ((nargs (length arglist))
2013               (minargs (if spread-p (1- nargs) nargs))
2014               (maxargs (if spread-p nil nargs))
2015               (required (ldb $lfbits-numreq bits))
2016               (max (if (logtest (logior (ash 1 $lfbits-rest-bit) (ash 1 $lfbits-restv-bit) (ash 1 $lfbits-keys-bit)) bits)
2017                      nil
2018                      (+ required (ldb $lfbits-numopt bits)))))
2019          ;; If the (apparent) number of args in the call doesn't
2020          ;; match the definition, complain.  If "spread-p" is true,
2021          ;; we can only be sure of the case when more than the
2022          ;; required number of args have been supplied.
2023          (if (or (if (and (not spread-p) (< minargs required))
2024                    (setq reason `(:toofew ,minargs ,required)))
2025                  (if (and max (or (> minargs max)) (if maxargs (> maxargs max)))
2026                    (setq reason (list :toomany (if (> minargs max) minargs maxargs) max)))
2027                  (setq reason (nx1-find-bogus-keywords arglist spread-p bits keyvect)))
2028            (values deftype reason)))))))
2029
2030(defun nx1-find-bogus-keywords (args spread-p bits keyvect)
2031  (declare (fixnum bits))
2032  (when (logbitp $lfbits-aok-bit bits)
2033    (setq keyvect nil))                 ; only check for even length tail
2034  (when (and (logbitp $lfbits-keys-bit bits) 
2035             (not spread-p))     ; Can't be sure, last argform may contain :allow-other-keys
2036    (do* ((key-values (nthcdr (+ (ldb $lfbits-numreq bits)  (ldb $lfbits-numopt bits)) args))
2037          (key-args key-values  (cddr key-args)))
2038         ((null key-args))
2039      (if (null (cdr key-args))
2040        (return (list :odd-keywords key-values))
2041        (when keyvect
2042          (let* ((keyword (%car key-args)))
2043            (unless (constantp keyword)
2044              (return nil))
2045            (unless (eq keyword :allow-other-keys)
2046              (unless (position (nx-unquote keyword) keyvect)               
2047                (return (list :unknown-keyword
2048                              (nx-unquote keyword)
2049                              (coerce keyvect 'list)))))))))))
2050
2051;;; we can save some space by going through subprims to call "builtin"
2052;;; functions for us.
2053(defun nx1-builtin-function-offset (name)
2054   (arch::builtin-function-name-offset name))
2055
2056(defun nx1-call-form (global-name afunc arglist spread-p  &optional (env *nx-lexical-environment*))
2057  (if afunc
2058    (make-acode (%nx1-operator lexical-function-call) afunc (nx1-arglist arglist (if spread-p 1 (backend-num-arg-regs *target-backend*))) spread-p)
2059    (let* ((builtin (unless (or spread-p
2060                                (eql 3 (safety-optimize-quantity env)))
2061                      (nx1-builtin-function-offset global-name))))
2062      (if (and builtin
2063               (let* ((bits (lfun-bits (fboundp global-name))))
2064                 (and bits (eql (logand $lfbits-args-mask bits)
2065                                (dpb (length arglist)
2066                                     $lfbits-numreq
2067                                     0)))))
2068        (make-acode (%nx1-operator builtin-call) 
2069                    (make-acode (%nx1-operator fixnum) builtin)
2070                    (nx1-arglist arglist))
2071        (make-acode (%nx1-operator call)
2072                     (if (symbolp global-name)
2073                       (nx1-immediate (nx1-note-fcell-ref global-name))
2074                       global-name)
2075                     (nx1-arglist arglist (if spread-p 1 (backend-num-arg-regs *target-backend*)))
2076                     spread-p)))))
2077 
2078;;; If "sym" is an expression (not a symbol which names a function),
2079;;; the caller has already alphatized it.
2080(defun nx1-call (sym args &optional spread-p global-only inhibit-inline)
2081  (nx1-verify-length args 0 nil)
2082  (let ((args-in-regs (if spread-p 1 (backend-num-arg-regs *target-backend*))))
2083    (if (nx-self-call-p sym global-only)
2084      ;; Should check for downward functions here as well.
2085      (multiple-value-bind (deftype reason)
2086                           (nx1-check-call-args *nx-current-function* args spread-p)
2087        (when deftype
2088          (nx1-whine deftype sym reason args spread-p))
2089        (make-acode (%nx1-operator self-call) (nx1-arglist args args-in-regs) spread-p))
2090      (multiple-value-bind (lambda-form containing-env token) (nx-inline-expansion sym *nx-lexical-environment* global-only)
2091        (or (and (not inhibit-inline)
2092                 (nx1-expand-inline-call lambda-form containing-env token args spread-p *nx-lexical-environment*))
2093            (multiple-value-bind (info afunc) (if (and  (symbolp sym) (not global-only)) (nx-lexical-finfo sym))
2094              (when (eq 'macro (car info))
2095                (nx-error "Can't call macro function ~s" sym))
2096              (nx-record-xref-info :direct-calls sym)
2097              (if (and afunc (%ilogbitp $fbitruntimedef (afunc-bits afunc)))
2098                (let ((sym (var-name (afunc-lfun afunc))))
2099                  (nx1-form 
2100                   (if spread-p
2101                     `(,(if (eql spread-p 0) 'applyv 'apply) ,sym ,args)
2102                     `(funcall ,sym ,@args))))
2103                (let* ((val (nx1-call-form sym afunc args spread-p)))
2104                    (when afunc
2105                      (let ((callers (afunc-callers afunc))
2106                            (self *nx-current-function*))
2107                        (unless (or (eq self afunc) (memq self callers))
2108                          (setf (afunc-callers afunc) (cons self callers)))))
2109                    (if (and (null afunc) (memq sym *nx-never-tail-call*))
2110                      (make-acode (%nx1-operator values) (list val))
2111                      val)))))))))
2112
2113(defun nx1-expand-inline-call (lambda-form env token args spread-p old-env)
2114  (if (and (or (null spread-p) (eq (length args) 1)))
2115    (if (and token (not (memq token *nx-inline-expansions*)))
2116      (with-program-error-handler (lambda (c) (declare (ignore c)) nil)
2117        (let* ((*nx-inline-expansions* (cons token *nx-inline-expansions*))
2118               (lambda-list (cadr lambda-form))
2119               (body (cddr lambda-form))
2120               (new-env (new-lexical-environment env)))
2121          (setf (lexenv.mdecls new-env)
2122                `((speed . ,(speed-optimize-quantity old-env))
2123                  (space . ,(space-optimize-quantity old-env))
2124                  (safety . ,(space-optimize-quantity old-env))
2125                  (compilation-speed . ,(compilation-speed-optimize-quantity old-env))
2126                  (debug . ,(debug-optimize-quantity old-env))))
2127          (if spread-p
2128            (nx1-destructure lambda-list (car args) nil nil body new-env)
2129            (nx1-lambda-bind lambda-list args body new-env)))))))
2130             
2131; note that regforms are reversed: arg_z is always in the car
2132(defun nx1-arglist (args &optional (nregargs (backend-num-arg-regs *target-backend*)))
2133  (declare (fixnum nregargs))
2134  (let* ((stkforms nil)
2135         (regforms nil)
2136         (nstkargs (%i- (length args) nregargs)))
2137    (declare (fixnum nstkargs))
2138      (list
2139       (dotimes (i nstkargs (nreverse stkforms))
2140         (declare (fixnum i))
2141         (push (nx1-form (%car args)) stkforms)
2142         (setq args (%cdr args)))
2143       (dolist (arg args regforms)
2144         (push (nx1-form arg) regforms)))))
2145
2146(defun nx1-formlist (args)
2147  (let* ((a nil))
2148    (dolist (arg args)
2149      (push (nx1-form arg) a))
2150    (nreverse a)))
2151
2152(defun nx1-verify-length (forms min max &aux (len (list-length forms)))
2153 (if (or (null len)
2154         (%i> min len)
2155         (and max (%i> len max)))
2156     (nx-error "Wrong number of args in form ~S." (cons *nx-sfname* forms))
2157     len))
2158
2159(defun nx-unquote (form)
2160  (if (nx-quoted-form-p form)
2161    (%cadr form)
2162    form))
2163
2164(defun nx-quoted-form-p (form &aux (f form))
2165 (and (consp form)
2166      (eq (pop form) 'quote)
2167      (or
2168       (and (consp form)
2169            (not (%cdr form)))
2170       (nx-error "Illegally quoted form ~S." f))))
2171
2172; Returns two values: expansion & win
2173; win is true if expansion is not EQ to form.
2174; This is a bootstrapping version.
2175; The real one is in "ccl:compiler;optimizers.lisp".
2176(unless (fboundp 'maybe-optimize-slot-accessor-form)
2177
2178(defun maybe-optimize-slot-accessor-form (form environment)
2179  (declare (ignore environment))
2180  (values form nil))
2181
2182)
2183
2184(defun nx-transform (form &optional (environment *nx-lexical-environment*) (source-note-map *nx-source-note-map*))
2185  (macrolet ((form-changed (form)
2186               `(progn
2187                  (unless source (setq source (gethash ,form source-note-map)))
2188                  (setq changed t))))
2189    (prog (sym transforms lexdefs changed enabled macro-function compiler-macro (source t))
2190       (when source-note-map
2191         (setq source (gethash form source-note-map)))
2192       (go START)
2193     LOOP
2194       (form-changed form)
2195       (when (and (consp form) 
2196                  (or (eq (%car form) 'the)
2197                      (and sym (eq (%car form) sym))))
2198         (go DONE))
2199     START
2200       (when (non-nil-symbol-p form)
2201         (multiple-value-bind (newform win) (nx-transform-symbol form environment)
2202           (unless win (go DONE))
2203           (setq form newform)
2204           (go LOOP)))
2205       (when (atom form) (go DONE))
2206       (unless (symbolp (setq sym (%car form)))
2207         (go DONE))
2208       (when (eq sym 'the)
2209         (destructuring-bind (typespec thing) (cdr form)
2210           (if (constantp thing)
2211             (progn
2212               (setq form thing)
2213               (go LOOP))
2214             (multiple-value-bind (newform win) (nx-transform thing environment source-note-map)
2215               (when win
2216                 (form-changed newform)
2217                 (if (and (self-evaluating-p newform)
2218                          (typep newform typespec))
2219                   (setq form newform)
2220                   (setq form `(the ,typespec ,newform)))
2221                 (go DONE))))))
2222       (when (nx-quoted-form-p form)
2223         (when (self-evaluating-p (%cadr form))
2224           (setq form (%cadr form)))
2225         (go DONE))
2226       (when (setq lexdefs (nx-lexical-finfo sym environment))
2227         (if (eq 'function (%car lexdefs))
2228           (go DONE)))
2229       (setq transforms (setq compiler-macro (compiler-macro-function sym environment))
2230             macro-function (macro-function sym environment)
2231             enabled (nx-allow-transforms environment))
2232       (unless macro-function
2233         (let* ((win nil))
2234           (when (and enabled (functionp (fboundp sym)))
2235             (multiple-value-setq (form win) (nx-transform-arglist form environment source-note-map))
2236             (when win
2237               (form-changed form)))))
2238       (when (and enabled
2239                  (not (nx-declared-notinline-p sym environment)))
2240         (multiple-value-bind (value folded) (nx-constant-fold form environment)
2241           (when folded
2242             (setq form value)
2243             (form-changed form)
2244             (unless (and (consp form) (eq (car form) sym)) (go START))))
2245         (when compiler-macro
2246           (multiple-value-bind (newform win) (compiler-macroexpand-1 form environment)
2247             (when win
2248               (when (and (consp newform) (eq (car newform) sym) (functionp (fboundp sym)))
2249                 (setq sym nil))
2250               (setq form newform)
2251               (go LOOP))))
2252         (multiple-value-bind (newform win) (maybe-optimize-slot-accessor-form form environment)
2253           (when win
2254             (setq sym nil)
2255             (setq form newform)
2256             (go START)))
2257         (unless macro-function
2258           (when (setq transforms (or (environment-structref-info sym environment)
2259                                      (and (boundp '%structure-refs%)
2260                                           (gethash sym %structure-refs%))))
2261             (setq form (defstruct-ref-transform transforms (%cdr form)))
2262             (form-changed form)
2263             (go START))
2264           (when (setq transforms (assq sym *nx-synonyms*))
2265             (setq form (cons (%cdr transforms) (setq sym (%cdr form))))
2266             (go LOOP))))
2267       (when (and macro-function
2268                  (or lexdefs
2269                      (not (and (gethash sym *nx1-alphatizers*) (not (nx-declared-notinline-p sym environment))))))
2270         (nx-record-xref-info :macro-calls (function-name macro-function))
2271         (setq form (macroexpand-1 form environment))
2272         (form-changed form)
2273         (go START))
2274     DONE
2275       (when (and source (neq source t) (not (gethash form source-note-map)))
2276         (unless (and (consp form)
2277                      (eq (%car form) 'the)
2278                      (eq source (gethash (caddr form) source-note-map)))
2279           (unless (or (eq form (%unbound-marker))
2280                       (eq form (%slot-unbound-marker)))
2281             (setf (gethash form source-note-map) source))))
2282       (return (values form changed)))))
2283
2284; Transform all of the arguments to the function call form.
2285; If any of them won, return a new call form (with the same operator as the original), else return the original
2286; call form unchanged.
2287(defun nx-transform-arglist (callform env source-note-map)
2288  (let* ((any-wins nil)
2289         (transformed-call (cons (car callform) nil))
2290         (ptr transformed-call)
2291         (win nil))
2292    (declare (type cons ptr))
2293    (dolist (form (cdr callform) (if any-wins (values (copy-list transformed-call) t) (values callform nil)))
2294      (multiple-value-setq (form win) (nx-transform form env source-note-map))
2295      (rplacd ptr (setq ptr (cons form nil)))
2296      (if win (setq any-wins t)))))
2297
2298;This is needed by (at least) SETF.
2299(defun nxenv-local-function-p (name macro-env)
2300  (multiple-value-bind (type local-p) (function-information name macro-env)
2301    (and local-p (eq :function type))))
2302
2303           
2304;;; This guy has to return multiple values.  The arguments have
2305;;; already been transformed; if they're all constant (or quoted), try
2306;;; to evaluate the expression at compile-time.
2307(defun nx-constant-fold (original-call &optional (environment *nx-lexical-environment*) &aux 
2308                                       (fn (car original-call)) form mv foldable foldfn)
2309  (flet ((quotify (x) (if (self-evaluating-p x) x (list 'quote x))))
2310    (if (and (nx-allow-transforms environment)
2311             (let* ((bits (if (symbolp fn) (%symbol-bits fn) 0)))
2312               (declare (fixnum bits))
2313               (if (setq foldable (logbitp $sym_fbit_constant_fold bits))
2314                 (if (logbitp $sym_fbit_fold_subforms bits)
2315                   (setq foldfn 'fold-constant-subforms))
2316                 (setq foldable (assq fn *nx-can-constant-fold*)
2317                       foldfn (cdr foldable)))
2318               foldable))
2319      (if foldfn
2320        (funcall foldfn original-call environment)
2321        (progn
2322          (let ((args nil))
2323            (dolist (arg (cdr original-call) (setq args (nreverse args)))
2324              (if (quoted-form-p arg)
2325                (setq arg (%cadr arg))
2326                (unless (self-evaluating-p arg) (return-from nx-constant-fold (values original-call nil))))
2327              (push arg args))
2328            (if (nx1-check-call-args (fboundp fn) args nil)
2329              (return-from nx-constant-fold (values original-call nil))
2330              (setq form (multiple-value-list 
2331                             (handler-case (apply fn args)
2332                               (error (condition)
2333                                      (warn "Error: \"~A\" ~&signalled during compile-time evaluation of ~S ."
2334                                            condition original-call)
2335                                      (return-from nx-constant-fold
2336                                        (values `(locally (declare (notinline ,fn))
2337                                                  ,original-call)
2338                                                t))))))))
2339          (if form
2340            (if (null (%cdr form))
2341              (setq form (%car form))
2342              (setq mv (setq form (cons 'values (mapcar #'quotify form))))))
2343          (values (if mv form (quotify form)) T)))
2344      (values original-call nil))))
2345
2346(defun nx-transform-symbol (sym &optional (env *nx-lexical-environment*))
2347; Gak.  Can't call NX-LEX-INFO without establishing *nx-lexical-environment*.
2348; NX-LEX-INFO should take env arg!.
2349  (let* ((*nx-lexical-environment* env))
2350    (multiple-value-bind (expansion win) (macroexpand-1 sym env)
2351      (if win
2352        (let ((type (nx-declared-type sym))
2353              (var (nth-value 2 (nx-lex-info sym))))
2354          (unless (eq t type) (setq expansion `(the ,type ,expansion)))
2355          (if var (nx-set-var-bits var (%ilogior (%ilsl $vbitreffed 1) (nx-var-bits var)))))
2356        (progn
2357          (multiple-value-setq (expansion win)
2358            (nx-transform-defined-constant sym env))
2359          (if win (setq win (neq sym expansion)))))
2360      (values expansion win))))
2361
2362; if sym has a substitutable constant value in env (or globally), return
2363; (values <value> t), else (values nil nil)
2364(defun nx-transform-defined-constant (sym env)
2365  (let* ((defenv (definition-environment env))
2366         (val (if defenv (assq sym (defenv.constants defenv))))
2367         (constant-value-p val))
2368    (if val
2369      (setq val (%cdr val))
2370      (if (constant-symbol-p sym)
2371        (setq constant-value-p t val (%sym-global-value sym))))
2372    (if (and (neq val (%unbound-marker-8))
2373             constant-value-p 
2374             (nx-substititute-constant-value sym val env))
2375      (values (if (self-evaluating-p val) val (list 'quote val)) t)
2376      (values nil nil))))
2377
2378
2379(defun nx-var-bits (var)
2380  (do* ((var var bits)
2381        (bits (var-bits var) (var-bits var)))
2382       ((fixnump bits) bits)))
2383
2384(defun nx-set-var-bits (var newbits)
2385  (do* ((var var bits)
2386        (bits (var-bits var) (var-bits var)))
2387       ((fixnump bits) (setf (var-bits var) newbits))))
2388
2389(defun nx-adjust-ref-count (var)
2390  (let* ((bits (nx-var-bits var))
2391         (temp-p (%ilogbitp $vbittemporary bits))
2392         (by (if temp-p 1 (expt  4 *nx-loop-nesting-level*)))
2393         (new (%imin (%i+ (%ilogand2 $vrefmask bits) by) 255)))
2394    (nx-set-var-bits var (%ilogior (%ilogand (%ilognot $vrefmask) bits) new))
2395    new))
2396
2397;;; Treat (VALUES x . y) as X if it appears in a THE form
2398(defun nx-form-type (form &optional (env *nx-lexical-environment*))
2399  (if (quoted-form-p form)
2400    (type-of (nx-unquote form))
2401    (if (self-evaluating-p form)
2402      (type-of form)
2403      (if (and (consp form)             ; Kinda bogus now, but require-type
2404               (eq (%car form) 'require-type) ; should be special some day
2405               (quoted-form-p (caddr form)))
2406        (%cadr (%caddr form))
2407        (if (nx-trust-declarations env)
2408          (if (symbolp form)
2409            (nx-target-type (nx-declared-type form env))
2410            (if (consp form)
2411              (if (eq (%car form) 'the)
2412                (destructuring-bind (typespec val) (%cdr form)
2413                  (declare (ignore val))
2414                  (let* ((ctype (values-specifier-type typespec)))
2415                    (if (typep ctype 'values-ctype)
2416                      (let* ((req (values-ctype-required ctype)))
2417                        (if req
2418                          (nx-target-type (type-specifier (car req)))
2419                          '*))
2420                      (nx-target-type (type-specifier ctype)))))
2421                (if (eq (%car form) 'setq)
2422                  (nx-declared-type (cadr form) env)
2423                  (let* ((op (gethash (%car form) *nx1-operators*)))
2424                    (or (and op (cdr (assq op *nx-operator-result-types*)))
2425                        (and (not op)(cdr (assq (car form) *nx-operator-result-types-by-name*)))
2426                        (and (memq (car form) *numeric-ops*)
2427                             (grovel-numeric-form form env))
2428                        (and (memq (car form) *logical-ops*)
2429                             (grovel-logical-form form env))
2430                        ;; Sort of the right idea, but this should be done
2431                        ;; in a more general way.
2432                        (when (or (eq (car form) 'aref)
2433                                  (eq (car form) 'uvref))
2434                          (let* ((atype (nx-form-type (cadr form) env))
2435                                 (a-ctype (specifier-type atype)))
2436                            (when (array-ctype-p a-ctype)
2437                              (type-specifier (array-ctype-specialized-element-type
2438                                               a-ctype)))))
2439                        t))))
2440              t))
2441          t)))))
2442
2443(defparameter *numeric-ops* '(+ -  / * +-2 --2 *-2 /-2))
2444
2445(defparameter *logical-ops* '(logxor-2 logior-2 logand-2  lognot logxor))
2446
2447(defun numeric-type-p (type &optional not-complex)
2448  (or (memq type '(fixnum integer double-float single-float float))
2449      (let ((ctype (specifier-type type)))
2450        (and (numeric-ctype-p ctype)
2451             (or (not not-complex)
2452                 (neq (numeric-ctype-complexp ctype) :complex))))))
2453
2454(defun grovel-numeric-form (form env)
2455  (let* ((op (car form))
2456         (args (cdr form)))
2457    (if (every #'(lambda (x) (nx-form-typep x 'float env)) args)
2458      (if (some #'(lambda (x) (nx-form-typep x 'double-float env)) args)
2459        'double-float
2460        'single-float)
2461      (if (every #'(lambda (x) (nx-form-typep x 'integer env)) args)
2462        (if (or (eq op '/) (eq op '/-2))
2463          t
2464          'integer)))))
2465
2466;; now e.g. logxor of 3 known fixnums is inline as is (logior a (logxor b c))
2467;; and (the fixnum (+ a (logxor b c)))
2468
2469(defun grovel-logical-form (form env)
2470  (when (nx-trust-declarations env)
2471    (let (;(op (car form))
2472          type)
2473      (dolist (arg (cdr form))
2474        (let ((it (nx-form-type arg env)))         
2475          (if (not (subtypep it 'fixnum))
2476            (return (setq type nil))
2477            (setq type 'fixnum))))
2478      type)))
2479
2480(defun nx-form-typep (arg type &optional (env *nx-lexical-environment*))
2481  (setq type (nx-target-type (type-expand type)))
2482  (if (constantp arg)
2483    (typep (nx-unquote arg) type env)
2484    (subtypep (nx-form-type arg env) type env)))
2485
2486
2487(defun nx-binary-fixnum-op-p (form1 form2 env &optional ignore-result-type)
2488  (setq form1 (nx-transform form1 env)
2489        form2 (nx-transform form2 env))
2490  (and
2491   (target-word-size-case
2492    (32 (nx-form-typep form1 '(signed-byte 30) env))
2493    (64 (nx-form-typep form1 '(signed-byte 61) env)))
2494   (target-word-size-case
2495    (32 (nx-form-typep form2 '(signed-byte 30) env))
2496    (64 (nx-form-typep form2 '(signed-byte 61) env)))
2497   (or ignore-result-type
2498        (and (nx-trust-declarations env)
2499                (target-word-size-case
2500                 (32 (subtypep *nx-form-type* '(signed-byte 30)))
2501                 (64 (subtypep *nx-form-type* '(signed-byte 61))))))))
2502
2503
2504(defun nx-binary-natural-op-p (form1 form2 env &optional (ignore-result-type t))
2505  (and
2506   (target-word-size-case
2507    (32
2508     (and (nx-form-typep form1 '(unsigned-byte 32)  env)
2509          (nx-form-typep form2 '(unsigned-byte 32)  env)))
2510    (64
2511     (and (nx-form-typep form1 '(unsigned-byte 64)  env)
2512          (nx-form-typep form2 '(unsigned-byte 64)  env))))
2513   (or ignore-result-type
2514       (and (nx-trust-declarations env)
2515            (target-word-size-case
2516             (32 (subtypep *nx-form-type* '(unsigned-byte 32)))
2517             (64 (subtypep *nx-form-type* '(unsigned-byte 64))))))))
2518
2519   
2520
2521
2522(defun nx-binary-boole-op (whole env arg-1 arg-2 fixop intop naturalop)
2523  (let* ((use-fixop (nx-binary-fixnum-op-p arg-1 arg-2 env t))
2524         (use-naturalop (nx-binary-natural-op-p arg-1 arg-2 env)))
2525    (if (or use-fixop use-naturalop intop)
2526      (make-acode (if use-fixop fixop (if use-naturalop naturalop intop))
2527                  (nx1-form arg-1)
2528                  (nx1-form arg-2))
2529      (nx1-treat-as-call whole))))
2530
2531(defun nx-global-p (sym &optional (env *nx-lexical-environment*))
2532  (or 
2533   (logbitp $sym_vbit_global (the fixnum (%symbol-bits sym)))
2534   (let* ((defenv (definition-environment env)))
2535     (if defenv 
2536       (eq :global (%cdr (assq sym (defenv.specials defenv))))))))
2537 
2538(defun nx-need-var (sym &optional (check-bindable t))
2539  (if (and (nx-need-sym sym)
2540           (not (constantp sym))
2541           (let* ((defenv (definition-environment *nx-lexical-environment*)))
2542             (or (null defenv)
2543                 (not (assq sym (defenv.constants defenv)))))) ; check compile-time-constants, too
2544    (if (and check-bindable (nx-global-p sym))
2545      (nx-error "~S is declared static and can not be bound" sym)
2546      sym)
2547    (nx-error "Can't bind or assign to constant ~S." sym)))
2548
2549(defun nx-need-sym (sym)
2550  (if (symbolp sym)
2551    sym
2552    (nx-error "~S is not a symbol." sym)))
2553
2554(defun nx-need-function-name (name)
2555  (multiple-value-bind (valid nm) (valid-function-name-p name)
2556    (if valid nm (nx-error "Invalid function name ~S" name))))
2557
2558(defun nx-pair-name (form)
2559  (nx-need-sym (if (consp form) (%car form) form)))
2560
2561(defun nx-pair-initform (form)
2562  (if (atom form)
2563    nil
2564    (if (and (listp (%cdr form)) (null (%cddr form)))
2565      (%cadr form)
2566      (nx-error "Bad initialization form: ~S." form))))
2567
2568; some callers might assume that this guy errors out if it can't conjure up
2569; a fixnum.  I certainly did ...
2570(defun nx-get-fixnum (form &aux (trans (nx-transform form *nx-lexical-environment*)))
2571 (if (fixnump trans)
2572  trans
2573  form))
2574 
2575(defun nx1-func-name (gizmo)
2576  (and (consp gizmo)
2577       (or (eq (%car gizmo) 'function) (eq (%car gizmo) 'quote))
2578       (consp (%cdr gizmo))
2579       (null (%cddr gizmo))
2580       (nth-value 1 (valid-function-name-p (%cadr gizmo)))))
2581
2582; distinguish between program errors & incidental ones.
2583(defun nx-error (format-string &rest args)
2584  (error (make-condition 'compile-time-program-error 
2585                :context (nx-error-context)
2586                :format-control format-string
2587                :format-arguments args)))
2588
2589(defun nx-compile-time-error (format-string &rest args)
2590  (error (make-condition 'compile-time-program-error 
2591                :context (nx-error-context)
2592                :format-control format-string
2593                :format-arguments args)))
2594
2595; Should return information about file being compiled, nested functions, etc. ...
2596(defun nx-error-context ()
2597  (or *nx-cur-func-name* "an anonymous function"))
2598
2599(defparameter *warn-if-function-result-ignored*
2600  '(sort stable-sort delete delete-if delete-if-not remf nreverse
2601    nunion nset-intersection)
2602  "Names of functions whos result(s) should ordinarily be used, because of their side-effects or lack of them.")
Note: See TracBrowser for help on using the repository browser.