source: branches/rme-logops/compiler/nx0.lisp @ 13885

Last change on this file since 13885 was 13885, checked in by rme, 9 years ago

purported improvements to logior on natural-sized operands

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