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

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

Make nx-form-typep not err with compile-time classes

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