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

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

Assorted code coverage fixes:

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