source: trunk/source/compiler/nx0.lisp @ 14417

Last change on this file since 14417 was 14417, checked in by gb, 11 years ago

No more *NX-NIL*, *NX-T*.

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