source: branches/arm/compiler/optimizers.lisp @ 13948

Last change on this file since 13948 was 13948, checked in by gb, 9 years ago

The point of the checkpoint in the last commit: change nil_value on ARM
from #x10000005 to #x04000001. That obviously means that we invert
arm::fulltag-cons and arm::fulltag-nil, and change a few things (the link-
inverting marker, the LISTP compiler-macro) that were more sensitive to
that than they probably should have been.

OTOH, we get ~192M more address space to play with.

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 95.9 KB
Line 
1;;;-*- Mode: Lisp; Package: CCL -*-
2;;;
3;;;   Copyright (C) 2009 Clozure Associates
4;;;   Copyright (C) 1994-2001 Digitool, Inc
5;;;   This file is part of Clozure CL.
6;;;
7;;;   Clozure CL is licensed under the terms of the Lisp Lesser GNU Public
8;;;   License , known as the LLGPL and distributed with Clozure CL as the
9;;;   file "LICENSE".  The LLGPL consists of a preamble and the LGPL,
10;;;   which is distributed with Clozure CL as the file "LGPL".  Where these
11;;;   conflict, the preamble takes precedence.
12;;;
13;;;   Clozure CL is referenced in the preamble as the "LIBRARY."
14;;;
15;;;   The LLGPL is also available online at
16;;;   http://opensource.franz.com/preamble.html
17
18; Optimizers.lisp - compiler optimizers
19
20(in-package "CCL")
21
22(eval-when (eval compile)
23  (require'backquote)
24  (require'lispequ)
25  (require "ARCH"))
26
27(declaim (special *nx-can-constant-fold* *nx-synonyms*))
28
29(defvar *dont-find-class-optimize* nil) ; t means dont
30
31#|
32;;; can-constant-fold had a bug in the way it called #'proclaim-inline
33|#
34
35;;; There seems to be some confusion about what #'proclaim-inline does.
36;;; The value of the alist entry in *nx-proclaimed-inline* indicates
37;;; whether or not the compiler is allowed to use any special knowledge
38;;; about the symbol in question.  That's a necessary but not sufficient
39;;; condition to enable inline expansion; that's governed by declarations
40;;; in the compile-time environment.
41;;; If someone observed a symptom whereby calling CAN-CONSTANT-FOLD
42;;; caused unintended inline-expansion, the bug's elsewhere ...
43;;; The bug is that nx-declared-inline-p calls proclaimed-inline-p
44;;;  which looks at what proclaim-inline sets.  Presumably, that
45;;;  means that someone fixed it because it showed evidence of
46;;;  being broken.
47;;; The two concepts (the compiler should/should not make assumptions about
48;;;  the signature of known functions, the compiler should/should not arrange
49;;;  to keep the lambda expression around) need to be sorted out.
50
51(defun can-constant-fold (names &aux handler inlines)
52  (dolist (name names)
53    (if (atom name)
54      (setq handler nil)
55      (setq handler (cdr name) name (car name)))
56    (when (and handler (not (eq handler 'fold-constant-subforms)))
57      (warn "Unknown constant-fold handler : ~s" handler)
58      (setq handler nil))
59    (let* ((bits (%symbol-bits name)))
60      (declare (fixnum bits))
61      (%symbol-bits name (logior
62                          (if handler (logior (ash 1 $sym_fbit_fold_subforms) (ash 1 $sym_fbit_constant_fold))
63                              (ash 1 $sym_fbit_constant_fold))
64                          bits)))
65     (push name inlines))
66  '(apply #'proclaim-inline t inlines)
67)
68
69;;; There's a bit somewhere.  This is very partial.  Should be a bit
70;;; somewhere, there are too many of these to keep on a list.
71(can-constant-fold '(specfier-type %ilsl %ilsr 1- 1+ eql eq
72                     byte make-point - / (+ . fold-constant-subforms) (* . fold-constant-subforms) ash character
73                     char-code code-char lsh
74                     (logior . fold-constant-subforms) (logand . fold-constant-subforms)
75                     (logxor . fold-constant-subforms) logcount logorc2 listp consp expt
76                     logorc1 logtest lognand logeqv lognor lognot logandc2 logandc1
77                     numerator denominator ldb-test byte-position byte-size isqrt gcd
78                     floor mod truncate rem round boole max min ldb dpb mask-field deposit-field
79                     length aref svref char schar bit sbit getf identity list-length
80                     car cdr cadr cddr nth nthcdr last load-byte deposit-byte byte-mask
81                     member search count position assoc rassoc integer-length
82                         float not null char-int expt abs
83                     = /= < <= > >=))
84
85(defun %binop-cassoc (call)
86  (unless (and (cddr call) (null (cdr (%cddr call))))
87    (return-from %binop-cassoc call))
88  (let ((func (%car call))
89        (arg1 (%cadr call))
90        (arg2 (%caddr call))
91        (val))
92    (cond ((and (fixnump arg1) (fixnump arg2))
93           (funcall func arg1 arg2))
94          ((or (fixnump arg1) (fixnump arg2))
95           (if (fixnump arg2) (psetq arg1 arg2 arg2 arg1))
96           (if (and (consp arg2)
97                    (eq (%car arg2) func)
98                    (cddr arg2)
99                    (null (cdr (%cddr arg2)))
100                    (or (fixnump (setq val (%cadr arg2)))
101                        (fixnump (setq val (%caddr arg2)))))
102             (list func
103                   (funcall func arg1 val)
104                   (if (eq val (%cadr arg2)) (%caddr arg2) (%cadr arg2)))
105             call))
106          (t call))))
107
108(defun fixnumify (args op &aux (len (length args)))
109  (if (eq len 2)
110    (cons op args)
111    (list op (%car args) (fixnumify (%cdr args) op))))
112
113(defun generic-to-fixnum-n (call env op &aux (args (%cdr call)) targs)
114  (block nil
115    (if (and (%i> (length args) 1)
116             (and (nx-trust-declarations env)
117                  (or (neq op '%i+) (subtypep *nx-form-type* 'fixnum))))
118      (if (dolist (arg args t)
119            (if (nx-form-typep arg 'fixnum env)
120              (push arg targs)
121              (return)))
122        (return
123         (fixnumify (nreverse targs) op))))
124    call))
125
126;;; True if arg is an alternating list of keywords and args, only
127;;; recognizes keywords in keyword package.  Historical note: this
128;;; used to try to ensure that the keyword appeared at most once.  Why
129;;; ? (Even before destructuring, pl-search/getf would have dtrt.)
130;;; Side effects: it's not the right thing to simply pick the value
131;;; associated with the first occurrence of a keyword if the value
132;;; associated with subsequent occurrence could have a side-effect.
133;;; (We -can- ignore a duplicate key if the associated value is
134;;; side-effect free.)
135(defun constant-keywords-p (keys)
136  (when (plistp keys)
137    (do* ((seen ())
138          (keys keys (cddr keys)))
139         ((null keys) t)
140      (let* ((key (car keys)))
141        (if (or (not (keywordp key))
142                (and (memq key seen)
143                     (not (constantp (cadr keys)))))
144          (return))
145        (push key seen)))))
146
147
148(defun remove-explicit-test-keyword-from-test-testnot-key (item list keys default alist testonly)
149  (if (null keys)
150    `(,default ,item ,list)
151     (if (constant-keywords-p keys)
152        (destructuring-bind (&key (test nil test-p)
153                                  (test-not nil test-not-p)
154                                  (key nil key-p))
155                            keys
156          (declare (ignore test-not))
157          (if (and test-p
158                   (not test-not-p)
159                   (or (not key-p)
160                       (and (consp key)
161                            (consp (%cdr key))
162                            (null (%cddr key))
163                            (or (eq (%car key) 'function)
164                                (eq (%car key) 'quote))
165                            (eq (%cadr key) 'identity)))
166                   (consp test)
167                   (consp (%cdr test))
168                   (null (%cddr test))
169                   (or (eq (%car test) 'function)
170                       (eq (%car test) 'quote)))
171            (let* ((testname (%cadr test))
172                   (reduced (cdr (assoc testname alist))))
173              (if reduced
174                `(,reduced ,item ,list)
175                `(,testonly ,item ,list ,test))))))))
176
177
178(defun eql-iff-eq-p (thing env)
179  (if (nx-form-constant-p thing env)
180    (setq thing (nx-form-constant-value thing env))
181    (return-from eql-iff-eq-p
182      (or (nx-form-typep thing  'symbol env)
183          (nx-form-typep thing 'character env)
184          (nx-form-typep thing
185                         '(or fixnum
186                           #+64-bit-target single-float
187                           symbol character
188                           (and (not number) (not macptr))) env))))
189  (or (fixnump thing) #+64-bit-target (typep thing 'single-float)
190      (symbolp thing) (characterp thing)
191      (and (not (numberp thing)) (not (macptrp thing)))))
192
193(defun equal-iff-eql-p (thing env)
194  (if (nx-form-constant-p thing env)
195    (setq thing (nx-form-constant-value thing env))
196    (return-from equal-iff-eql-p
197      (nx-form-typep thing
198                     '(and (not cons) (not string) (not bit-vector) (not pathname)) env)))
199  (not (typep thing '(or cons string bit-vector pathname))))
200
201
202(defun fold-constant-subforms (call env)
203    (let* ((constants nil)
204           (forms nil))
205      (declare (list constants forms))
206      (dolist (form (cdr call))
207        (setq form (nx-transform form env))
208        (if (numberp form)
209          (setq constants (%temp-cons form constants))
210          (setq forms (%temp-cons form forms))))
211      (if constants
212        (let* ((op (car call))
213               (constant (if (cdr constants) (handler-case (apply op constants)
214                                               (error (c) (declare (ignore c))
215                                                      (return-from fold-constant-subforms (values call t))))
216                             (car constants))))
217          (values (if forms (cons op (cons constant (reverse forms))) constant) t))
218        (values call nil))))
219
220;;; inline some, etc. in some cases
221;;; in all cases, add dynamic-extent declarations
222(defun some-xx-transform (call env)
223  (destructuring-bind (func predicate sequence &rest args) call
224    (multiple-value-bind (func-constant end-value loop-test)
225                         (case func
226                           (some (values $some nil 'when))
227                           (notany (values $notany t 'when))
228                           (every (values $every t 'unless))
229                           (notevery (values $notevery nil 'unless)))
230      (if args
231        (let ((func-sym (gensym))
232              (seq-sym (gensym))
233              (list-sym (gensym)))
234          `(let ((,func-sym ,predicate)
235                 (,seq-sym ,sequence)
236                 (,list-sym (list ,@args)))
237             (declare (dynamic-extent ,func-sym ,list-sym ,seq-sym))
238             (some-xx-multi ,func-constant ,end-value ,func-sym ,seq-sym ,list-sym)))
239        (let ((loop-function (nx-form-sequence-iterator sequence env)))
240          ;; inline if we know the type of the sequence and if
241          ;; the predicate is a lambda expression
242          ;; otherwise, it blows up the code for not much gain
243          (if (and loop-function
244                   (function-form-p predicate)
245                   (lambda-expression-p (second predicate)))
246            (let ((elt-var (gensym)))
247              (case func
248                (some
249                 `(,loop-function (,elt-var ,sequence ,end-value)
250                                  (let ((result (funcall ,predicate ,elt-var)))
251                                    (when result (return result)))))
252                ((every notevery notany)
253                 `(,loop-function (,elt-var ,sequence ,end-value)
254                                  (,loop-test (funcall ,predicate ,elt-var)
255                                              (return ,(not end-value)))))))
256            (let ((func-sym (gensym))
257                  (seq-sym (gensym)))
258              `(let ((,func-sym ,predicate)
259                     (,seq-sym ,sequence))
260                 (declare (dynamic-extent ,func-sym ,seq-sym))
261                 (some-xx-one ,func-constant ,end-value ,func-sym ,seq-sym)))))))))
262
263
264;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
265;;;
266;;; The new (roughly alphabetical) order.
267;;;
268;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
269
270;;; Compiler macros on functions can assume that their arguments have
271;;; already been transformed.
272
273
274(defun transform-real-n-ary-comparision (whole binary-name)
275  (destructuring-bind (n0 &optional (n1 0 n1-p) &rest more) (cdr whole)
276    (if more
277      (if (cdr more)
278        whole
279        (let* ((n2 (car more))
280               (n (gensym)))
281          `(let* ((,n ,n0))
282            (if (,binary-name ,n (setq ,n ,n1))
283              (,binary-name ,n ,n2)))))
284      (if (not n1-p)
285        `(require-type ,n0 'real)
286        `(,binary-name ,n0 ,n1)))))
287
288
289
290(define-compiler-macro < (&whole whole &rest ignore)
291  (declare (ignore ignore))
292  (transform-real-n-ary-comparision whole '<-2))
293
294(define-compiler-macro > (&whole whole &rest ignore)
295  (declare (ignore ignore))
296  (transform-real-n-ary-comparision whole '>-2))
297
298(define-compiler-macro <= (&whole whole &rest ignore)
299  (declare (ignore ignore))
300  (transform-real-n-ary-comparision whole '<=-2))
301
302(define-compiler-macro >= (&whole whole &rest ignore)
303  (declare (ignore ignore))
304  (transform-real-n-ary-comparision whole '>=-2))
305
306
307(define-compiler-macro 1- (x)
308  `(- ,x 1))
309
310(define-compiler-macro 1+ (x)
311  `(+ ,x 1))
312
313(define-compiler-macro append  (&whole call
314                                       &optional arg0
315                                       &rest
316                                       (&whole tail
317                                               &optional (junk nil arg1-p)
318                                               &rest more))
319  ;(append (list x y z) A) -> (list* x y z A)
320  (if (and arg1-p
321           (null more)
322           (consp arg0)
323           (eq (%car arg0) 'list))
324    (cons 'list* (append (%cdr arg0) tail))
325    (if (and arg1-p (null more))
326      `(append-2 ,arg0 ,junk)
327      call)))
328
329
330(define-compiler-macro apply  (&whole call fn arg0 &rest args)
331  ;; Special-case (apply #'make-instance 'name ...)
332  ;; Might be good to make this a little more general, e.g., there
333  ;; may be other things that can be strength-reduced even if we can't
334  ;; get rid of the APPLY.
335  (if (and (consp fn)
336           (or (eq (car fn) 'quote)
337               (eq (car fn) 'function))
338           (consp (cdr fn))
339           (null (cddr fn))
340           (eq (cadr fn) 'make-instance)
341           (consp arg0)
342           (eq (car arg0) 'quote)
343           (consp (cdr arg0))
344           (symbolp (cadr arg0)))
345    (let* ((name (cadr arg0))
346           (class-cell (gensym)))
347      `(let* ((,class-cell (load-time-value (find-class-cell ',name t))))
348        (apply (class-cell-instantiate ,class-cell) ,class-cell ,@args)))
349    (let ((original-fn fn))
350      (if (and arg0
351               (null args)
352               (consp fn)
353               (eq (%car fn) 'function)
354               (null (cdr (%cdr fn)))
355               (consp (setq fn (%cadr fn)))
356               (eq (%car fn) 'lambda))
357        (destructuring-bind (lambda-list &body body) (%cdr fn)
358          `(destructuring-bind ,lambda-list ,arg0 ,@body))
359        (let ((last (%car (last (push arg0 args)))))
360          (if (and (consp last) (memq (%car last) '(cons list* list)))
361            (cons (if (eq (%car last) 'list) 'funcall 'apply)
362                  (cons
363                   original-fn
364                   (nreconc (cdr (reverse args)) (%cdr last))))
365            (if (and (consp last)
366                     (eq (car last) 'quote)
367                     (proper-list-p (cadr last)))
368              (flet ((quotify (arg)
369                       (if (self-evaluating-p arg)
370                         arg
371                         (list 'quote arg))))
372                (cons 'funcall (cons original-fn
373                                     (nreconc (cdr (reverse args)) (mapcar #'quotify (%cadr last))))))
374              call)))))))
375
376
377
378
379(define-compiler-macro assoc (&whole call item list &rest keys)
380  (or (remove-explicit-test-keyword-from-test-testnot-key item list keys 'asseql '((eq . assq) (eql . asseql) (equal . assequal)) 'assoc-test)
381      call))
382
383(define-compiler-macro assequal (&whole call &environment env item list)
384  (if (or (equal-iff-eql-p item env)
385          (and (quoted-form-p list)
386               (proper-list-p (%cadr list))
387               (every (lambda (x) (equal-iff-eql-p (car x) env)) (%cadr list))))
388    `(asseql ,item ,list)
389    call))
390
391(define-compiler-macro asseql (&whole call &environment env item list)
392  (if (or (eql-iff-eq-p item env)
393          (and (quoted-form-p list)
394               (proper-list-p (%cadr list))
395               (every (lambda (x) (eql-iff-eq-p (car x) env)) (%cadr list))))
396    `(assq ,item ,list)
397    call))
398
399(define-compiler-macro assq (item list)
400  (let* ((itemx (gensym))
401         (listx (gensym))
402         (pair (gensym)))
403    `(let* ((,itemx ,item)
404            (,listx ,list))
405      (dolist (,pair ,listx)
406        (when (and ,pair (eq (car ,pair) ,itemx)) (return ,pair))))))
407
408(define-compiler-macro caar (form)
409  `(car (car ,form)))
410
411(define-compiler-macro cadr (form)
412  `(car (cdr ,form)))
413
414(define-compiler-macro cdar (form)
415  `(cdr (car ,form)))
416
417(define-compiler-macro cddr (form)
418  `(cdr (cdr ,form)))
419
420(define-compiler-macro caaar (form)
421  `(car (caar ,form)))
422
423(define-compiler-macro caadr (form)
424  `(car (cadr ,form)))
425
426(define-compiler-macro cadar (form)
427  `(car (cdar ,form)))
428
429(define-compiler-macro caddr (form)
430  `(car (cddr ,form)))
431
432(define-compiler-macro cdaar (form)
433  `(cdr (caar ,form)))
434
435(define-compiler-macro cdadr (form)
436  `(cdr (cadr ,form)))
437
438(define-compiler-macro cddar (form)
439  `(cdr (cdar ,form)))
440
441(define-compiler-macro cdddr (form)
442  `(cdr (cddr ,form)))
443
444(define-compiler-macro caaaar (form)
445  `(car (caaar ,form)))
446
447(define-compiler-macro caaadr (form)
448  `(car (caadr ,form)))
449
450(define-compiler-macro caadar (form)
451  `(car (cadar ,form)))
452
453(define-compiler-macro caaddr (form)
454  `(car (caddr ,form)))
455
456(define-compiler-macro cadaar (form)
457  `(car (cdaar ,form)))
458
459(define-compiler-macro cadadr (form)
460  `(car (cdadr ,form)))
461
462(define-compiler-macro caddar (form)
463  `(car (cddar ,form)))
464
465(define-compiler-macro cadddr (form)
466  `(car (cdddr ,form)))
467
468(define-compiler-macro cdaaar (form)
469  `(cdr (caaar ,form)))
470
471(define-compiler-macro cdaadr (form)
472  `(cdr (caadr ,form)))
473
474(define-compiler-macro cdadar (form)
475  `(cdr (cadar ,form)))
476
477(define-compiler-macro cdaddr (form)
478  `(cdr (caddr ,form)))
479
480(define-compiler-macro cddaar (form)
481  `(cdr (cdaar ,form)))
482
483(define-compiler-macro cddadr (form)
484  `(cdr (cdadr ,form)))
485
486(define-compiler-macro cdddar (form)
487  `(cdr (cddar ,form)))
488
489(define-compiler-macro cddddr (form)
490  `(cdr (cdddr ,form)))
491
492
493
494
495(define-compiler-macro cons (&whole call x y &aux dcall ddcall)
496   (if (consp (setq dcall y))
497     (cond
498      ((or (eq (%car dcall) 'list) (eq (%car dcall) 'list*))
499       ;(CONS A (LIST[*] . args)) -> (LIST[*] A . args)
500       (list* (%car dcall) x (%cdr dcall)))
501      ((or (neq (%car dcall) 'cons) (null (cddr dcall)) (cdddr dcall))
502       call)
503      ((null (setq ddcall (%caddr dcall)))
504       ;(CONS A (CONS B NIL)) -> (LIST A B)
505       `(list ,x ,(%cadr dcall)))
506      ((and (consp ddcall)
507            (eq (%car ddcall) 'cons)
508            (eq (list-length ddcall) 3))
509       ;(CONS A (CONS B (CONS C D))) -> (LIST* A B C D)
510       (list* 'list* x (%cadr dcall) (%cdr ddcall)))
511      (t call))
512     call))
513
514(define-compiler-macro dotimes (&whole call (i n &optional result)
515                                       &body body
516                                       &environment env)
517  (multiple-value-bind (body decls) (parse-body body env)
518    (if (nx-form-typep (setq n (nx-transform n env)) 'fixnum env)
519        (let* ((limit (gensym))
520               (upper (if (nx-form-constant-p n env) (nx-form-constant-value n env) most-positive-fixnum))
521               (top (gensym))
522               (test (gensym)))
523          `(let* ((,limit ,n) (,i 0))
524             ,@decls
525             (declare (fixnum ,limit)
526                      (type (integer 0 ,(if (<= upper 0) 0 upper)) ,i)
527                      (unsettable ,i))
528             (block nil
529               (tagbody
530                 (go ,test)
531                 ,top
532                 ,@body
533                 (locally
534                   (declare (settable ,i))
535                   (setq ,i (1+ ,i)))
536                 ,test
537                 (when (< ,i ,limit) (go ,top)))
538               ,result)))
539        call)))
540
541(define-compiler-macro dpb (&whole call value byte integer)
542  (cond ((and (integerp byte) (> byte 0))
543         (if (integerp value)
544           `(logior ,(dpb value byte 0) (logand ,(lognot byte) ,integer))
545           `(deposit-field (ash ,value ,(byte-position byte)) ,byte ,integer)))
546        ((and (consp byte)
547              (eq (%car byte) 'byte)
548              (eq (list-length (%cdr byte)) 2))
549         `(deposit-byte ,value ,(%cadr byte) ,(%caddr byte) ,integer))
550        (t call)))
551
552(define-compiler-macro eql (&whole call &environment env v1 v2)
553  (if (or (eql-iff-eq-p v1 env) (eql-iff-eq-p v2 env))
554    `(eq ,v1 ,v2)
555    call))
556
557(define-compiler-macro every (&whole call &environment env &rest ignore)
558  (declare (ignore ignore))
559  (some-xx-transform call env))
560
561
562(define-compiler-macro identity (form) form)
563
564(define-compiler-macro if (&whole call test true &optional false &environment env)
565  (let ((test-val (nx-transform test env)))
566    (if (nx-form-constant-p test-val env)
567      (if (nx-form-constant-value test-val env)
568        true
569        false)
570      call)))
571
572(define-compiler-macro %ilsr (&whole call shift value)
573  (if (eql shift 0)
574    value
575    (if (eql value 0)
576      `(progn ,shift 0)
577      call)))
578
579(defun string-designator-p (object)
580  (typecase object
581    (character t)
582    (symbol t)
583    (string t)))
584
585(define-compiler-macro ldb (&whole call &environment env byte integer)
586   (cond ((and (integerp byte) (> byte 0))
587          (let ((size (byte-size byte))
588                (position (byte-position byte)))
589            (cond ((nx-form-typep integer 'fixnum env)
590                   `(logand ,(byte-mask size)
591                     (the fixnum (ash ,integer ,(- position)))))
592                  (t `(load-byte ,size ,position ,integer)))))
593         ((and (consp byte)
594               (eq (%car byte) 'byte)
595               (eq (list-length (%cdr byte)) 2))
596          (let ((size (%cadr byte))
597                (position (%caddr byte)))
598            (if (and (nx-form-typep integer 'fixnum env) (fixnump position))
599              ;; I'm not sure this is worth doing
600              `(logand (byte-mask ,size) (the fixnum (ash ,integer ,(- position))))
601              ;; this IS worth doing
602              `(load-byte ,size ,position ,integer))))
603         (t call)))
604
605(define-compiler-macro length (&whole call &environment env seq)
606  (if (nx-form-typep seq '(simple-array * (*)) env)
607    `(uvsize ,seq)
608    call))
609
610(define-compiler-macro let (&whole call (&optional (first nil first-p) &rest rest) &body body)
611  (if first-p
612    (if rest
613      call
614      `(let* (,first) ,@body))
615    `(locally ,@body)))
616
617(define-compiler-macro let* (&whole call (&rest bindings) &body body)
618  (if bindings
619    call
620    `(locally ,@body)))
621
622(define-compiler-macro list* (&whole call &rest rest  &aux (n (list-length rest)) last)
623  (cond ((%izerop n) nil)
624        ((null (setq last (%car (last call))))
625         (cons 'list (nreverse (cdr (reverse (cdr call))))))
626        ((and (consp last) (memq (%car last) '(list* list cons)))
627         (cons (if (eq (%car last) 'cons) 'list* (%car last))
628                                 (nreconc (cdr (reverse (cdr call))) (%cdr last))))
629        ((eq n 1) (list 'values last))
630        ((eq n 2) (cons 'cons (%cdr call)))
631        (t call)))
632
633
634
635;;;(CONS X NIL) is same size as (LIST X) and faster.
636(define-compiler-macro list  (&whole call &optional (first nil first-p) &rest more)
637  (if more
638    call
639    (if first-p
640      `(cons ,first nil))))
641
642
643(define-compiler-macro locally (&whole call &body body &environment env)
644  (multiple-value-bind (body decls) (parse-body body env nil)
645    (if decls
646      call
647      `(progn ,@body))))
648
649(defun target-element-type-type-keyword (typespec &optional env)
650  (let ((ctype (specifier-type-if-known `(array ,typespec) env)))
651    (when ctype
652      (funcall (arch::target-array-type-name-from-ctype-function
653                (backend-target-arch *target-backend*))
654               ctype))))
655
656(defun infer-array-type (dims element-type element-type-p displaced-to-p fill-pointer-p adjustable-p env)
657  (let* ((ctype (make-array-ctype :complexp (or displaced-to-p fill-pointer-p adjustable-p))))
658    (if (nx-form-constant-p dims env)
659      (let* ((dims (nx-form-constant-value dims env)))
660        (if (listp dims)
661          (progn
662            (unless (every #'fixnump dims)
663              (warn "Funky-looking array dimensions ~s in MAKE-ARRAY call" dims))
664            (setf (array-ctype-dimensions ctype) dims))
665          (progn
666            (unless (typep dims 'fixnum)
667              (warn "Funky-looking array dimensions ~s in MAKE-ARRAY call" dims))
668            (setf (array-ctype-dimensions ctype) (list dims)))))
669      (if (atom dims)
670        (if (nx-form-typep dims 'fixnum env)
671          (setf (array-ctype-dimensions ctype)
672                (if (typep (setq dims (nx-transform dims env)) 'fixnum)
673                  (list dims)
674                  (list '*)))
675          (setf (array-ctype-dimensions ctype) '*))
676        (if (eq (car dims) 'list)
677          (setf (array-ctype-dimensions ctype)
678                (mapcar #'(lambda (d)
679                            (if (typep (setq d (nx-transform d env)) 'fixnum)
680                              d
681                              '*))
682                        (cdr dims)))
683          ;; Wimp out
684          (setf (array-ctype-dimensions ctype)
685                '*))))
686    (let* ((typespec (if element-type-p
687                       (if (nx-form-constant-p element-type env)
688                         (nx-form-constant-value element-type env)
689                         '*)
690                       t))
691           (element-type (specifier-type-if-known typespec env :whine t)))
692      (setf (array-ctype-element-type ctype) (or element-type *wild-type*))
693      (specialize-array-type ctype))
694    (type-specifier ctype)))
695
696
697
698(define-compiler-macro make-array (&whole call &environment env dims &rest keys)
699  (if (constant-keywords-p keys)
700    (destructuring-bind (&key (element-type t element-type-p)
701                              (displaced-to () displaced-to-p)
702                              (displaced-index-offset () displaced-index-offset-p)
703                              (adjustable () adjustable-p)
704                              (fill-pointer () fill-pointer-p)
705                              (initial-element () initial-element-p)
706                              (initial-contents () initial-contents-p))
707        keys
708      (declare (ignorable element-type element-type-p
709                          displaced-to displaced-to-p
710                          displaced-index-offset displaced-index-offset-p
711                          adjustable adjustable-p
712                          fill-pointer fill-pointer-p
713                          initial-element initial-element-p
714                          initial-contents initial-contents-p))
715      (let* ((element-type-keyword nil)
716             (expansion
717              (cond ((and initial-element-p initial-contents-p)
718                     (signal-program-error  "Incompatible arguments :INITIAL-ELEMENT and :INITIAL-CONTENTS in ~s" call)
719                     call)
720                    (displaced-to-p
721                     (if (or initial-element-p initial-contents-p element-type-p)
722                       (comp-make-array-1 dims keys)
723                       (comp-make-displaced-array dims keys)))
724                    ((or displaced-index-offset-p
725                         (not (nx-form-constant-p element-type env))
726                         (null (setq element-type-keyword
727                                     (target-element-type-type-keyword
728                                      (nx-form-constant-value element-type env) env))))
729                     (comp-make-array-1 dims keys))
730                    ((and (typep element-type-keyword 'keyword)
731                          (nx-form-typep dims 'fixnum env)
732                          (null (or adjustable fill-pointer initial-contents
733                                    initial-contents-p)))
734                     (if
735                       (or (null initial-element-p)
736                           (cond ((eql element-type-keyword :double-float-vector)
737                                  (eql initial-element 0.0d0))
738                                 ((eql element-type-keyword :single-float-vector)
739                                  (eql initial-element 0.0s0))
740                                 ((eql element-type :simple-string)
741                                  (eql initial-element #\Null))
742                                 (t (eql initial-element 0))))
743                       `(allocate-typed-vector ,element-type-keyword ,dims)
744                       `(allocate-typed-vector ,element-type-keyword ,dims ,initial-element)))
745                    (t                        ;Should do more here
746                     (comp-make-uarray dims keys (type-keyword-code element-type-keyword)))))
747             (type (if (nx-trust-declarations env)
748                     (infer-array-type dims element-type element-type-p displaced-to-p fill-pointer-p adjustable-p env)
749                     t)))
750        `(the ,type ,expansion)))
751
752        call))
753
754(defun comp-make-displaced-array (dims keys)
755  (let* ((call-list (make-list 4 :initial-element nil))
756         (dims-var (make-symbol "DIMS"))
757         (let-list (comp-nuke-keys keys
758                                   '((:displaced-to 0)
759                                     (:fill-pointer 1)
760                                     (:adjustable 2)
761                                     (:displaced-index-offset 3))
762                                   call-list
763                                   `((,dims-var ,dims)))))
764
765    `(let ,let-list
766       (%make-displaced-array ,dims-var ,@call-list t))))
767
768(defun comp-make-uarray (dims keys subtype)
769  (if (null keys)
770    `(%make-simple-array ,subtype ,dims)
771    (let* ((call-list (make-list 6))
772           (dims-var (make-symbol "DIMS"))
773           (let-list (comp-nuke-keys keys
774                                     '((:adjustable 0)
775                                       (:fill-pointer 1)
776                                       (:initial-element 2 3)
777                                       (:initial-contents 4 5))
778                                     call-list
779                                     `((,dims-var ,dims)))))
780      `(let ,let-list
781        (make-uarray-1 ,subtype ,dims-var ,@call-list nil nil)))))
782
783(defun comp-make-array-1 (dims keys)
784  (let* ((call-list (make-list 10 :initial-element nil))
785         (dims-var (make-symbol "DIMS"))
786         (let-list (comp-nuke-keys keys
787                                   '((:element-type 0 1)
788                                     (:displaced-to 2)
789                                     (:displaced-index-offset 3)
790                                     (:adjustable 4)
791                                     (:fill-pointer 5)
792                                     (:initial-element 6 7)
793                                     (:initial-contents 8 9))
794                                   call-list
795                                   `((,dims-var ,dims)))))
796    `(let ,let-list
797       (make-array-1 ,dims-var ,@call-list nil))))
798
799(defun comp-nuke-keys (keys key-list call-list &optional required-bindings)
800  ; side effects call list, returns a let-list
801  (let* ((let-list (reverse required-bindings))
802         (seen nil))
803    (do ((lst keys (cddr lst)))
804        ((null lst) nil)
805      (let* ((key (car lst))
806             (val (cadr lst))
807             (ass (assq key key-list))
808             (vpos (cadr ass))
809             (ppos (caddr ass)))
810        (when ass
811          (unless (memq vpos seen)
812            (push vpos seen)
813            (when (not (constantp val))
814              (let ((gen (gensym)))
815                (setq let-list (cons (list gen val) let-list)) ; reverse him
816                (setq val gen)))
817            (rplaca (nthcdr vpos call-list) val)
818            (if ppos (rplaca (nthcdr ppos call-list) t))))))
819    (nreverse let-list)))
820
821(define-compiler-macro make-instance (&whole call class &rest initargs)
822  (if (and (listp class)
823           (eq (car class) 'quote)
824           (symbolp (cadr class))
825           (null (cddr class)))
826    (let* ((cell (gensym)))
827      `(let* ((,cell (load-time-value (find-class-cell ,class t))))
828        (funcall (class-cell-instantiate ,cell) ,cell ,@initargs)))
829    call))
830
831
832
833
834
835
836
837(define-compiler-macro mapc  (&whole call fn lst &rest more)
838  (if more
839    call
840    (let* ((temp-var (gensym))
841           (elt-var (gensym))
842           (fn-var (gensym)))
843       `(let* ((,fn-var ,fn)
844               (,temp-var ,lst))
845          (dolist (,elt-var ,temp-var ,temp-var)
846            (funcall ,fn-var ,elt-var))
847          ))))
848
849(define-compiler-macro mapcar (&whole call fn lst &rest more)
850  (if more
851    call
852    (let* ((temp-var (gensym))
853           (result-var (gensym))
854           (elt-var (gensym))
855           (fn-var (gensym)))
856      `(let* ((,temp-var (cons nil nil))
857              (,result-var ,temp-var)
858              (,fn-var ,fn))
859         (declare (dynamic-extent ,temp-var)
860                  (type cons ,temp-var ,result-var))
861         (dolist (,elt-var ,lst (cdr ,result-var))
862           (setq ,temp-var (setf (cdr ,temp-var) (list (funcall ,fn-var ,elt-var)))))))))
863
864(define-compiler-macro member (&whole call item list &rest keys)
865  (or (remove-explicit-test-keyword-from-test-testnot-key item list keys 'memeql '((eq . memq) (eql . memeql) (equal . memequal)) 'member-test)
866      call))
867
868(define-compiler-macro memequal (&whole call &environment env item list)
869  (if (or (equal-iff-eql-p item env)
870          (and (quoted-form-p list)
871               (proper-list-p (%cadr list))
872               (every (lambda (elt) (equal-iff-eql-p elt env)) (%cadr list))))
873    `(memeql ,item ,list)
874    call))
875
876(define-compiler-macro memeql (&whole call &environment env item list)
877  (if (or (eql-iff-eq-p item env)
878          (and (quoted-form-p list)
879               (proper-list-p (%cadr list))
880               (every (lambda (elt) (eql-iff-eq-p elt env)) (%cadr list))))
881    `(memq ,item ,list)
882    call))
883
884(define-compiler-macro memq (item list)
885  ;;(memq x '(y)) => (if (eq x 'y) '(y))
886  ;;Would it be worth making a two elt list into an OR?  Maybe if
887  ;;optimizing for speed...
888   (if (and (or (quoted-form-p list)
889                (null list))
890            (null (cdr (%cadr list))))
891     (if list `(if (eq ,item ',(%caadr list)) ,list))
892     (let* ((x (gensym))
893            (tail (gensym)))
894       `(do* ((,x ,item)
895              (,tail ,list (cdr (the list ,tail))))
896         ((null ,tail))
897         (if (eq (car ,tail) ,x) (return ,tail))))))
898
899(define-compiler-macro minusp (x)
900  `(< ,x 0))
901
902(define-compiler-macro notany (&whole call &environment env &rest ignore)
903  (declare (ignore ignore))
904  (some-xx-transform call env))
905
906(define-compiler-macro notevery (&whole call &environment env &rest ignore)
907  (declare (ignore ignore))
908  (some-xx-transform call env))
909
910(define-compiler-macro nth  (count list)
911   (if (and (fixnump count)
912            (%i>= count 0)
913            (%i< count 3))
914     `(,(svref '#(car cadr caddr) count) ,list)
915     `(car (nthcdr ,count ,list))))
916
917(define-compiler-macro nthcdr (count list)
918  (if (and (fixnump count)
919           (%i>= count 0)
920           (%i< count 4))
921     (if (%izerop count)
922       `(require-type ,list 'list)
923       `(,(svref '#(cdr cddr cdddr) (%i- count 1)) ,list))
924    (let* ((i (gensym))
925           (n (gensym))                 ; evaluation order
926           (tail (gensym)))
927      `(let* ((,n (require-type ,count 'unsigned-byte))
928              (,tail (require-type ,list 'list)))
929        (dotimes (,i ,n ,tail)
930          (unless (setq ,tail (cdr ,tail))
931            (return nil)))))))
932
933(define-compiler-macro plusp (x)
934  `(> ,x 0))
935
936(define-compiler-macro progn (&whole call &optional (first nil first-p) &rest rest)
937  (if first-p
938    (if rest call first)))
939
940;;; This isn't quite right... The idea is that (car (require-type foo
941;;; 'list)) ;can become just (<typechecking-car> foo) [regardless of
942;;; optimize settings], ;but I don't think this can be done just with
943;;; optimizers... For now, at least try to get it to become (%car
944;;; (<typecheck> foo)).
945(define-compiler-macro require-type (&whole call &environment env arg type &aux ctype)
946  (cond ((and (or (eq type t)
947                  (and (nx-form-constant-p type env)
948                       (setq type (nx-form-constant-value type env))))
949              (setq ctype (specifier-type-if-known type env :whine t)))
950         (cond ((nx-form-typep arg type env) arg)
951               ((and (nx-trust-declarations env) ;; if don't trust declarations, don't bother.
952                     (cond ((eq type 'simple-vector)
953                            `(the simple-vector (require-simple-vector ,arg)))
954                           ((eq type 'simple-string)
955                            `(the simple-string (require-simple-string ,arg)))
956                           ((eq type 'integer)
957                            `(the integer (require-integer ,arg)))
958                           ((eq type 'fixnum)
959                            `(the fixnum (require-fixnum ,arg)))
960                           ((eq type 'real)
961                            `(the real (require-real ,arg)))
962                           ((eq type 'list)
963                            `(the list (require-list ,arg)))
964                           ((eq type 'character)
965                            `(the character (require-character ,arg)))
966                           ((eq type 'number)
967                            `(the number (require-number ,arg)))
968                           ((eq type 'symbol)
969                            `(the symbol (require-symbol ,arg)))
970                           ((type= ctype
971                                   (specifier-type '(signed-byte 8)))
972                            `(the (signed-byte 8) (require-s8 ,arg)))
973                           ((type= ctype
974                                   (specifier-type '(unsigned-byte 8)))
975                            `(the (unsigned-byte 8) (require-u8 ,arg)))
976                           ((type= ctype
977                                   (specifier-type '(signed-byte 16)))
978                            `(the (signed-byte 16) (require-s16 ,arg)))
979                           ((type= ctype
980                                   (specifier-type '(unsigned-byte 16)))
981                            `(the (unsigned-byte 16) (require-u16 ,arg)))
982                           ((type= ctype
983                                   (specifier-type '(signed-byte 32)))
984                            `(the (signed-byte 32) (require-s32 ,arg)))
985                           ((type= ctype
986                                   (specifier-type '(unsigned-byte 32)))
987                            `(the (unsigned-byte 32) (require-u32 ,arg)))
988                           ((type= ctype
989                                   (specifier-type '(signed-byte 64)))
990                            `(the (signed-byte 64) (require-s64 ,arg)))
991                           ((type= ctype
992                                   (specifier-type '(unsigned-byte 64)))
993                            `(the (unsigned-byte 64) (require-u64 ,arg)))
994                           #+nil
995                           ((and (symbolp type)
996                                 (let ((simpler (type-predicate type)))
997                                   (if simpler `(the ,type (%require-type ,arg ',simpler))))))
998                           #+nil
999                           ((and (symbolp type)(find-class type nil env))
1000                            `(%require-type-class-cell ,arg (load-time-value (find-class-cell ',type t))))
1001                           (t (let* ((val (gensym)))
1002                                `(the ,type
1003                                   (let* ((,val ,arg))
1004                                     (if (typep ,val ',type)
1005                                       ,val
1006                                       (%kernel-restart $xwrongtype ,val ',type)))))))))
1007               (t (let* ((val (gensym)))
1008                    `(let* ((,val ,arg))
1009                       (if (typep ,val ',type)
1010                         ,val
1011                         (%kernel-restart $xwrongtype ,val ',type)))))))
1012        (t call)))
1013
1014(define-compiler-macro proclaim (&whole call decl)
1015   (if (and (quoted-form-p decl)
1016            (eq (car (setq decl (%cadr decl))) 'special))
1017       (do ((vars (%cdr decl) (%cdr vars)) (decls ()))
1018           ((null vars)
1019            (cons 'progn (nreverse decls)))
1020         (unless (and (car vars)
1021                      (neq (%car vars) t)
1022                      (symbolp (%car vars)))
1023            (return call))
1024         (push (list '%proclaim-special (list 'quote (%car vars))) decls))
1025       call))
1026
1027
1028(define-compiler-macro some (&whole call &environment env &rest ignore)
1029  (declare (ignore ignore))
1030  (some-xx-transform call env))
1031
1032(define-compiler-macro struct-ref (&whole call &environment env struct offset)
1033   (if (nx-inhibit-safety-checking env)
1034    `(%svref ,struct ,offset)
1035    call))
1036
1037;;; expand find-if and find-if-not
1038
1039(define-compiler-macro find-if (test sequence &rest keys)
1040  `(find ,test ,sequence
1041        :test #'funcall
1042        ,@keys))
1043
1044(define-compiler-macro find-if-not (test sequence &rest keys)
1045  `(find ,test ,sequence
1046        :test-not #'funcall
1047        ,@keys))
1048
1049;;; inline some cases, and use a positional function in others
1050
1051(define-compiler-macro find (&whole call &environment env
1052                                    item sequence &rest keys)
1053  (if (constant-keywords-p keys)
1054    (destructuring-bind (&key from-end test test-not (start 0) end key) keys
1055      (if (and (eql start 0)
1056               (null end)
1057               (null from-end)
1058               (not (and test test-not)))
1059        (let ((find-test (or test test-not '#'eql))
1060              (loop-test (if test-not 'unless 'when))
1061              (loop-function (nx-form-sequence-iterator sequence env)))
1062          (if loop-function
1063            (let ((item-var (unless (or (nx-form-constant-p item env)
1064                                        (and (equal find-test '#'funcall)
1065                                             (function-form-p item)))
1066                              (gensym)))
1067                  (elt-var (gensym)))
1068              `(let (,@(when item-var `((,item-var ,item))))
1069                 (,loop-function (,elt-var ,sequence)
1070                                 (,loop-test (funcall ,find-test ,(or item-var item)
1071                                                      (funcall ,(or key '#'identity) ,elt-var))
1072                                             (return ,elt-var)))))
1073            (let ((find-function (if test-not 'find-positional-test-not-key 'find-positional-test-key))
1074                  (item-var (gensym))
1075                  (sequence-var (gensym))
1076                  (test-var (gensym))
1077                  (key-var (gensym)))
1078              `(let ((,item-var ,item)
1079                     (,sequence-var ,sequence)
1080                     (,test-var ,(or test test-not))
1081                     (,key-var ,key))
1082                 (declare (dynamic-extent ,item-var ,sequence-var ,test-var ,key-var))
1083                 (,find-function ,item-var ,sequence-var ,test-var ,key-var)))))
1084        call))
1085      call))
1086
1087;;; expand position-if and position-if-not
1088
1089(define-compiler-macro position-if (test sequence &rest keys)
1090  `(position ,test ,sequence
1091             :test #'funcall
1092             ,@keys))
1093
1094(define-compiler-macro position-if-not (test sequence &rest keys)
1095  `(position ,test ,sequence
1096             :test-not #'funcall
1097             ,@keys))
1098
1099;;; inline some cases, and use positional functions for others
1100
1101(define-compiler-macro position (&whole call &environment env
1102                                        item sequence &rest keys)
1103  (if (constant-keywords-p keys)
1104    (destructuring-bind (&key from-end test test-not (start 0) end key) keys
1105      (if (and (eql start 0)
1106               (null end)
1107               (null from-end)
1108               (not (and test test-not)))
1109        (let ((position-test (or test test-not '#'eql))
1110              (loop-test (if test-not 'unless 'when)))
1111          (cond ((nx-form-typep sequence 'list env)
1112                 (let ((item-var (unless (or (nx-form-constant-p item env)
1113                                             (and (equal position-test '#'funcall)
1114                                                  (function-form-p item)))
1115                                   (gensym)))
1116                       (elt-var (gensym))
1117                       (position-var (gensym)))
1118                   `(let (,@(when item-var `((,item-var ,item)))
1119                          (,position-var 0))
1120                      (dolist (,elt-var ,sequence)
1121                        (,loop-test (funcall ,position-test ,(or item-var item)
1122                                             (funcall ,(or key '#'identity) ,elt-var))
1123                                    (return ,position-var))
1124                        (incf ,position-var)))))
1125                ((nx-form-typep sequence 'vector env)
1126                 (let ((item-var (unless (or (nx-form-constant-p item env)
1127                                             (and (equal position-test '#'funcall)
1128                                                  (function-form-p item)))
1129                                   (gensym)))
1130                       (sequence-var (gensym))
1131                       (position-var (gensym)))
1132                   `(let (,@(when item-var `((,item-var ,item)))
1133                          (,sequence-var ,sequence))
1134                      ,@(let ((type (nx-form-type sequence env)))
1135                          (unless (eq type t)
1136                            `((declare (type ,type ,sequence-var)))))
1137                      (dotimes (,position-var (length ,sequence-var))
1138                        (,loop-test (funcall ,position-test ,(or item-var item)
1139                                             (funcall ,(or key '#'identity)
1140                                                      (locally (declare (optimize (speed 3) (safety 0)))
1141                                                        (aref ,sequence ,position-var))))
1142                                    (return ,position-var))))))
1143                (t
1144                 (let ((position-function (if test-not
1145                                            'position-positional-test-not-key
1146                                            'position-positional-test-key))
1147                       (item-var (gensym))
1148                       (sequence-var (gensym))
1149                       (test-var (gensym))
1150                       (key-var (gensym)))
1151                   `(let ((,item-var ,item)
1152                          (,sequence-var ,sequence)
1153                          (,test-var ,(or test test-not))
1154                          (,key-var ,key))
1155                      (declare (dynamic-extent ,sequence-var ,test-var ,key-var))
1156                      (,position-function ,item-var ,sequence-var ,test-var ,key-var))))))
1157        call))
1158    call))
1159
1160;;; inline some cases of remove-if and remove-if-not
1161
1162(define-compiler-macro remove-if (&whole call &environment env &rest ignore)
1163  (declare (ignore ignore))
1164  (remove-if-transform call env))
1165
1166(define-compiler-macro remove-if-not (&whole call &environment env &rest ignore)
1167  (declare (ignore ignore))
1168  (remove-if-transform call env))
1169
1170(defun remove-if-transform (call env)
1171  (destructuring-bind (function test sequence &rest keys) call
1172    (if (constant-keywords-p keys)
1173      (destructuring-bind (&key from-end (start 0) end count (key '#'identity)) keys
1174        (if (and (eql start 0)
1175                 (null end)
1176                 (null from-end)
1177                 (null count)
1178                 (nx-form-typep sequence 'list env))
1179          ;; only do the list case, since it's hard to collect vector results
1180          (let ((temp-var (gensym))
1181                (result-var (gensym))
1182                (elt-var (gensym))
1183                (loop-test (ecase function (remove-if 'unless) (remove-if-not 'when))))
1184            `(the list
1185               (let* ((,temp-var (cons nil nil))
1186                      (,result-var ,temp-var))
1187                 (declare (dynamic-extent ,temp-var))
1188                 (dolist (,elt-var ,sequence (%cdr ,result-var))
1189                   (,loop-test (funcall ,test (funcall ,key ,elt-var))
1190                               (setq ,temp-var
1191                                     (%cdr
1192                                      (%rplacd ,temp-var (list ,elt-var)))))))))
1193          call))
1194      call)))
1195
1196
1197
1198(define-compiler-macro struct-set (&whole call &environment env struct offset new)
1199  (if (nx-inhibit-safety-checking env)
1200    `(%svset ,struct ,offset ,new)
1201    call))
1202
1203(define-compiler-macro zerop (arg &environment env)
1204  (let* ((z (if (nx-form-typep arg 'float env)
1205              (coerce 0 (nx-form-type arg env))
1206              0)))
1207    `(= ,arg ,z)))
1208
1209
1210(define-compiler-macro = (&whole w n0 &optional (n1 nil n1p) &rest more)
1211  (if (not n1p)
1212    `(require-type ,n0 'number)
1213    (if more
1214      w
1215      `(=-2 ,n0 ,n1))))
1216
1217(define-compiler-macro /= (&whole w n0 &optional (n1 nil n1p) &rest more)
1218  (if (not n1p)
1219    `(require-type ,n0 'number)
1220    (if more
1221      w
1222      `(/=-2 ,n0 ,n1))))
1223
1224(define-compiler-macro + (&optional (n0 nil n0p) (n1 nil n1p) &rest more &environment env)
1225  (if more
1226    (if (and (nx-trust-declarations env)
1227             (subtypep *nx-form-type* 'fixnum)
1228             (nx-form-typep n0 'fixnum env)
1229             (nx-form-typep n1 'fixnum env)
1230             (dolist (m more t)
1231               (unless (nx-form-typep m 'fixnum env)
1232                 (return nil))))
1233      `(+-2 ,n0 (the fixnum (+ ,n1 ,@more)))
1234      `(+ (+-2 ,n0 ,n1) ,@more))
1235    (if n1p
1236      `(+-2 ,n0 ,n1)
1237      (if n0p
1238        `(require-type ,n0 'number)
1239        0))))
1240
1241(define-compiler-macro - (n0 &optional (n1 nil n1p) &rest more)
1242  (if more
1243    `(- (--2 ,n0 ,n1) ,@more)
1244    (if n1p
1245      `(--2 ,n0 ,n1)
1246      `(%negate ,n0))))
1247
1248(define-compiler-macro * (&optional (n0 nil n0p) (n1 nil n1p) &rest more)
1249  (if more
1250    `(*-2 ,n0 (* ,n1 ,@more))
1251    (if n1p
1252      `(*-2 ,n0 ,n1)
1253      (if n0p
1254        `(require-type ,n0 'number)
1255        1))))
1256
1257(define-compiler-macro / (&whole w n0 &optional (n1 nil n1p) &rest more)
1258  (if more
1259    w
1260    (if n1p
1261      `(/-2 ,n0 ,n1)
1262      `(%quo-1 ,n0))))
1263
1264;;; beware of limits - truncate of most-negative-fixnum & -1 ain't a
1265;;; fixnum - too bad
1266(define-compiler-macro truncate (&whole w &environment env n0 &optional (n1 nil n1p))
1267  (let ((*nx-form-type* t))
1268    (if (nx-form-typep n0 'fixnum env)
1269      (if (not n1p)
1270        n0
1271        (if (nx-form-typep n1 'fixnum env)
1272          `(%fixnum-truncate ,n0 ,n1)
1273          w))
1274      w)))
1275
1276(define-compiler-macro floor (&whole w &environment env n0 &optional (n1 nil n1p))
1277  (let ((*nx-form-type* t))
1278    (if (nx-form-typep n0 'fixnum env)
1279      (if (not n1p)
1280        n0
1281        (if (nx-form-typep n1 'fixnum env)
1282          `(%fixnum-floor ,n0 ,n1)
1283          w))
1284      w)))
1285
1286(define-compiler-macro round (&whole w &environment env n0 &optional (n1 nil n1p))
1287  (let ((*nx-form-type* t)) ; it doesn't matter what the result type is declared to be
1288    (if (nx-form-typep n0 'fixnum env)
1289      (if (not n1p)
1290        n0
1291        (if (nx-form-typep n1 'fixnum env)
1292          `(%fixnum-round ,n0 ,n1)
1293          w))
1294      w)))
1295
1296(define-compiler-macro ceiling (&whole w &environment env n0 &optional (n1 nil n1p))
1297  (let ((*nx-form-type* t))
1298    (if (nx-form-typep n0 'fixnum env)
1299      (if (not n1p)
1300        n0
1301        (if (nx-form-typep n1 'fixnum env)
1302          `(%fixnum-ceiling ,n0 ,n1)
1303          w))
1304      w)))
1305
1306(define-compiler-macro oddp (&whole w &environment env n0)
1307  (if (nx-form-typep n0 'fixnum env)
1308    `(logbitp 0 (the fixnum ,n0))
1309    w))
1310
1311(define-compiler-macro evenp (&whole w &environment env n0)
1312  (if (nx-form-typep n0 'fixnum env)
1313    `(not (logbitp 0 (the fixnum ,n0)))
1314    w))
1315
1316
1317(define-compiler-macro logandc2 (n0 n1)
1318  (let ((n1var (gensym))
1319        (n0var (gensym)))
1320    `(let ((,n0var ,n0)
1321           (,n1var ,n1))
1322       (logandc1 ,n1var ,n0var))))
1323
1324(define-compiler-macro logorc2 (n0 n1)
1325  (let ((n1var (gensym))
1326        (n0var (gensym)))
1327    `(let ((,n0var ,n0)
1328           (,n1var ,n1))
1329       (logorc1 ,n1var ,n0var))))
1330
1331(define-compiler-macro lognand (n0 n1)
1332  `(lognot (logand ,n0 ,n1)))
1333
1334(define-compiler-macro lognor (n0 n1)
1335  `(lognot (logior ,n0 ,n1)))
1336
1337
1338(defun transform-logop (whole identity binop &optional (transform-complement t))
1339  (destructuring-bind (op &optional (n0 nil n0p) (n1 nil n1p) &rest more) whole
1340    (if (and n1p (eql n0 identity))
1341      `(,op ,n1 ,@more)
1342      (if (and transform-complement n1p (eql n0 (lognot identity)))
1343        `(progn
1344           (,op ,n1 ,@more)
1345           ,(lognot identity))
1346        (if more
1347          (if (cdr more)
1348            whole
1349            `(,binop ,n0 (,binop ,n1 ,(car more))))
1350          (if n1p
1351            `(,binop ,n0 ,n1)
1352            (if n0p
1353              `(require-type ,n0 'integer)
1354              identity)))))))
1355
1356(define-compiler-macro logand (&whole w &rest all)
1357  (declare (ignore all))
1358  (transform-logop w -1 'logand-2))
1359
1360(define-compiler-macro logior (&whole w &rest all)
1361  (declare (ignore all))
1362  (transform-logop w 0 'logior-2))
1363
1364(define-compiler-macro logxor (&whole w &rest all)
1365  (declare (ignore all))
1366  (transform-logop w 0 'logxor-2 nil))
1367
1368(define-compiler-macro lognot (&whole w &environment env n1)
1369  (if (nx-form-typep n1 'fixnum env)
1370    `(%ilognot ,n1)
1371    w))
1372
1373(define-compiler-macro logtest (&whole w &environment env n1 n2)
1374  (if (and (nx-form-typep n1 'fixnum env)
1375           (nx-form-typep n2 'fixnum env))
1376    `(not (eql 0 (logand ,n1 ,n2)))
1377    w))
1378
1379
1380(defmacro defsynonym (from to)
1381  ;Should maybe check for circularities.
1382  `(progn
1383     (setf (compiler-macro-function ',from) nil)
1384     (let ((pair (assq ',from *nx-synonyms*)))
1385       (if pair (rplacd pair ',to)
1386           (push (cons ',from ',to)
1387                 *nx-synonyms*))
1388       ',to)))
1389
1390(defsynonym first car)
1391(defsynonym second cadr)
1392(defsynonym third caddr)
1393(defsynonym fourth cadddr)
1394(defsynonym rest cdr)
1395
1396
1397(defsynonym functionp lfunp)
1398(defsynonym null not)
1399(defsynonym char-int char-code)
1400
1401;;; Improvemets file by Bob Cassels
1402;;; Just what are "Improvemets", anyway ?
1403
1404;;; Optimize some CL sequence functions, mostly by inlining them in
1405;;; simple cases when the type of the sequence is known.  In some
1406;;; cases, dynamic-extent declarations are automatically inserted.
1407;;; For some sequence functions, if the type of the sequence is known
1408;;; at compile time, the function is inlined.  If the type isn't known
1409;;; but the call is "simple", a call to a faster (positional-arg)
1410;;; function is substituted.
1411
1412
1413(defun nx-form-sequence-iterator (sequence-form env)
1414  (cond ((nx-form-typep sequence-form 'vector env) 'dovector)
1415        ((nx-form-typep sequence-form 'list env) 'dolist)))
1416
1417(defun function-form-p (form)
1418   ;; c.f. quoted-form-p
1419   (and (consp form)
1420        (eq (%car form) 'function)
1421        (consp (%cdr form))
1422        (null (%cdr (%cdr form)))))
1423
1424
1425;; Return a form that checks to see if THING is if type CTYPE, or
1426;; NIL if we can't do that for some reason.
1427(defun optimize-ctypep (thing ctype)
1428  (when (eq *target-backend* *host-backend*)
1429    (typecase ctype
1430      (numeric-ctype
1431       (cond ((eq :real (numeric-ctype-complexp ctype))
1432              (let* ((low (numeric-ctype-low ctype))
1433                     (high (numeric-ctype-high ctype))
1434                     (class (numeric-ctype-class ctype))
1435                     (format (numeric-ctype-format ctype))
1436                     (type (if (eq class 'float)
1437                             (or format class)
1438                             (or class 'real))))
1439                (cond ((and low (eql low high) (or (not (eq class 'float))
1440                                                   format))
1441                       `(eql ,thing ,low))
1442                      ((and (eq type 'float)
1443                            (or low high)
1444                            (or (null low)
1445                                (typep low 'single-float)
1446                                (not (null (ignore-errors
1447                                             (coerce (if (atom low)
1448                                                       low
1449                                                       (car low))
1450                                                     'single-float)))))
1451                            (or (null high)
1452                                (typep high 'single-float)
1453                                (not (null (ignore-errors
1454                                             (coerce (if (atom high)
1455                                                       high
1456                                                       (car high))
1457                                                     'single-float))))))
1458                       (let* ((temp (gensym)))
1459                         (flet ((bounded-float (type low high)
1460                                  `(,type
1461                                    ,(if low
1462                                         (if (listp low)
1463                                           (list (coerce (car low) type))
1464                                           (coerce low type))
1465                                         '*)
1466                                    ,(if high
1467                                         (if (listp high)
1468                                           (list (coerce (car high) type))
1469                                           (coerce high type))
1470                                         '*))))
1471                         `(let* ((,temp ,thing))
1472                           (or (typep ,temp ',(bounded-float 'single-float low high))
1473                            (typep ,temp ',(bounded-float 'double-float low high)))))))
1474                      (t
1475                       (let* ((temp (gensym)))
1476                         (if (and (typep low 'fixnum) (typep high 'fixnum)
1477                                  (eq class 'integer))
1478                           (setq type 'fixnum))
1479                         (if (or low high)
1480                           `(let* ((,temp ,thing))
1481                             (and (typep ,temp ',type)
1482                              ,@(if low `((,(if (consp low) '> '>=) (the ,type ,temp) ,(if (consp low) (car low) low))))
1483                              ,@(if high `((,(if (consp high) '< '<=) (the ,type ,temp) ,(if (consp high) (car high) high))))))
1484                           `(typep ,thing ',type)))))))
1485             (t `(numeric-%%typep ,thing ,ctype))))
1486      (array-ctype
1487       (or
1488        (let* ((typecode (array-ctype-typecode ctype))
1489               (dims (array-ctype-dimensions ctype)))
1490          (cond ((and typecode (consp dims) (null (cdr dims)))
1491                 (case (array-ctype-complexp ctype)
1492                   ((nil)
1493                    (if (eq (car dims) '*)
1494                      `(eql (typecode ,thing) ,typecode)
1495                      (let* ((temp (gensym)))
1496                        `(let* ((,temp ,thing))
1497                          (and (eql (typecode ,temp) ,typecode)
1498                           (eq (uvsize ,temp) ,(car dims)))))))
1499                   ((* :maybe)
1500                    (let* ((temp (gensym))
1501                           (tempcode (gensym)))
1502                      `(let* ((,temp ,thing)
1503                              (,tempcode (typecode ,temp)))
1504                        (or (and (eql ,tempcode ,typecode)
1505                             ,@(unless (eq (car dims) '*)
1506                                       `((eq (uvsize ,temp) ,(car dims)))))
1507                         (and (eql ,tempcode target::subtag-vectorH)
1508                          (eql (ldb target::arrayH.flags-cell-subtag-byte (the fixnum (%svref ,temp target::arrayH.flags-cell))) ,typecode)
1509                          ,@(unless (eq (car dims) '*)
1510                                    `((eq (%svref ,temp target::vectorH.logsize-cell) ,(car dims)))))))))))))
1511        `(values (array-%%typep ,thing ,ctype)))))))
1512
1513
1514(defun optimize-typep (thing type env)
1515  ;; returns a new form, or nil if it can't optimize
1516  (let ((ctype (specifier-type-if-known type env :whine t)))
1517    (when ctype
1518      (let* ((type (type-specifier ctype))
1519             (predicate (if (typep type 'symbol) (type-predicate type))))
1520        (if (and predicate (symbolp predicate))
1521          `(,predicate ,thing)
1522          (let* ((pair (assq type *istruct-cells*))
1523                 (class (and pair (%wrapper-class (istruct-cell-info pair)))))
1524            (if (and class (not (%class-direct-subclasses class)))
1525              `(istruct-typep ,thing ',type)             
1526              (or (optimize-ctypep thing ctype)
1527                  (cond ((symbolp type)
1528                         (cond ((%deftype-expander type)
1529                                ;; recurse here, rather than returning the
1530                                ;; partially-expanded form mostly since it doesn't
1531                                ;; seem to further optimize the result otherwise
1532                                (let ((expanded-type (type-expand type)))
1533                                  (or (optimize-typep thing expanded-type env)
1534                                      ;; at least do the first expansion
1535                                      `(typep ,thing ',expanded-type))))
1536                               ((structure-class-p type env)
1537                                `(structure-typep ,thing ',(find-class-cell type t)))
1538                               ((find-class type nil env)
1539                                ;; If we know for sure that the class
1540                                ;; is one whose instances are all
1541                                ;; STANDARD-INSTANCEs (not funcallable,
1542                                ;; not foreign), we can use
1543                                ;; STD-INSTANCE-CLASS-CELL-TYPEP, which
1544                                ;; can be a little faster then the more
1545                                ;; general CLASS-CELL-TYPEP.  We can
1546                                ;; only be sure of that if the class
1547                                ;; exists (as a non-COMPILE-TIME-CLASS)
1548                                (let* ((class (find-class type nil nil))
1549                                       (fname 
1550                                        (if (and class
1551                                                 (subtypep class 'standard-object)
1552                                                 (not (subtypep class 'foreign-standard-object))
1553                                                 (not (subtypep class 'funcallable-standard-object)))
1554                                          'std-instance-class-cell-typep
1555                                          'class-cell-typep)))
1556                                  `(,fname ,thing (load-time-value (find-class-cell ',type t)))))
1557                               ((info-type-builtin type) ; bootstrap troubles here?
1558                                `(builtin-typep ,thing (load-time-value (find-builtin-cell ',type))))
1559                               (t nil)))
1560                        ((consp type)
1561                         (cond
1562                           ((info-type-builtin type) ; byte types
1563                            `(builtin-typep ,thing (load-time-value (find-builtin-cell ',type))))
1564                           (t
1565                            (case (%car type)
1566                              (satisfies `(funcall ',(cadr type) ,thing))
1567                              (eql `(eql ,thing ',(cadr type)))
1568                              (member `(not (null (member ,thing ',(%cdr type)))))
1569                              (not `(not (typep ,thing ',(cadr type))))
1570                              ((or and)
1571                               (let ((thing-sym (gensym)))
1572                                 `(let ((,thing-sym ,thing))
1573                                   (,(%car type)
1574                                    ,@(mapcar #'(lambda (type-spec)
1575                                                  (or (optimize-typep thing-sym type-spec env)
1576                                                      `(typep ,thing-sym ',type-spec)))
1577                                              (%cdr type))))))
1578                              ((signed-byte unsigned-byte integer mod) ; more byte types
1579                               `(builtin-typep ,thing (load-time-value (find-builtin-cell ',type))))
1580                              (t nil)))))
1581                        (t nil))))))))))
1582
1583(define-compiler-macro typep  (&whole call &environment env thing type &optional e)
1584  (if (nx-form-constant-p type env)
1585    (let ((type-val (nx-form-constant-value type env)))
1586      (if (eq type-val t)
1587        `(progn ,thing t)
1588        (if (and (nx-form-constant-p thing env)
1589                 (specifier-type-if-known type-val env))
1590          (typep (nx-form-constant-value thing env) type-val env)
1591          (or (and (null e) (optimize-typep thing type-val env))
1592              call))))
1593    call))
1594
1595(define-compiler-macro structure-typep (&whole w thing type)
1596  (if (not (quoted-form-p type))
1597    (progn
1598      (warn "Non-quoted structure-type in ~s" w)
1599      w)
1600    (let* ((type (nx-unquote type)))
1601      (if (symbolp type)
1602        `(structure-typep ,thing ',(find-class-cell type t))
1603        w))))
1604
1605(define-compiler-macro true (&rest args)
1606  `(progn
1607    ,@args
1608    t))
1609
1610
1611(define-compiler-macro false (&rest args)
1612  `(progn
1613    ,@args
1614    nil))
1615
1616(define-compiler-macro find-class (&whole call type &optional (errorp t) env)
1617  (if (and (quoted-form-p type)(not *dont-find-class-optimize*)(not env))
1618      `(class-cell-find-class (load-time-value (find-class-cell ,type t)) ,errorp)
1619    call))
1620
1621
1622(define-compiler-macro gcd (&whole call &optional (n0 nil n0-p) (n1 nil n1-p) &rest rest)
1623  (if rest
1624    call
1625    (if n1-p
1626      `(gcd-2 ,n0 ,n1)
1627      (if n0-p
1628        `(%integer-abs ,n0)
1629        0))))
1630
1631(define-compiler-macro lcm (&whole call &optional (n0 nil n0-p) (n1 nil n1-p) &rest rest)
1632  (if rest
1633    call
1634    (if n1-p
1635      `(lcm-2 ,n0 ,n1)
1636      (if n0-p
1637        `(%integer-abs ,n0)
1638        1))))
1639
1640(define-compiler-macro max (&whole call &environment env n0 &optional (n1 nil n1-p) &rest rest)
1641  (if rest
1642    call
1643    (if n1-p
1644      (if (and (nx-form-typep n0 'fixnum env)(nx-form-typep n1 'fixnum env))
1645        `(imax-2 ,n0 ,n1)
1646        `(max-2 ,n0 ,n1))
1647      `(require-type ,n0 'real))))
1648
1649(define-compiler-macro max-2 (n0 n1)
1650  (let* ((g0 (gensym))
1651         (g1 (gensym)))
1652   `(let* ((,g0 ,n0)
1653           (,g1 ,n1))
1654      (if (> ,g0 ,g1) ,g0 ,g1))))
1655
1656(define-compiler-macro imax-2 (n0 n1)
1657  (let* ((g0 (gensym))
1658         (g1 (gensym)))
1659   `(let* ((,g0 ,n0)
1660           (,g1 ,n1))
1661      (if (%i> ,g0 ,g1) ,g0 ,g1))))
1662
1663
1664
1665
1666(define-compiler-macro min (&whole call &environment env n0 &optional (n1 nil n1-p) &rest rest)
1667  (if rest
1668    call
1669    (if n1-p
1670      (if (and (nx-form-typep n0 'fixnum env)(nx-form-typep n1 'fixnum env))
1671        `(imin-2 ,n0 ,n1)
1672        `(min-2 ,n0 ,n1))
1673      `(require-type ,n0 'real))))
1674
1675(define-compiler-macro min-2 (n0 n1)
1676  (let* ((g0 (gensym))
1677         (g1 (gensym)))
1678   `(let* ((,g0 ,n0)
1679           (,g1 ,n1))
1680      (if (< ,g0 ,g1) ,g0 ,g1))))
1681
1682(define-compiler-macro imin-2 (n0 n1)
1683  (let* ((g0 (gensym))
1684         (g1 (gensym)))
1685   `(let* ((,g0 ,n0)
1686           (,g1 ,n1))
1687      (if (%i< ,g0 ,g1) ,g0 ,g1))))
1688
1689
1690(defun eq-test-p (test)
1691  (or (equal test ''eq) (equal test '#'eq)))
1692
1693(defun eql-test-p (test)
1694  (or (equal test ''eql) (equal test '#'eql)))
1695
1696(define-compiler-macro adjoin (&whole whole elt list &rest keys)
1697  (if (constant-keywords-p keys)
1698    (destructuring-bind (&key (test ''eql) test-not key) keys
1699      (or (and (null test-not)
1700               (null key)
1701               (cond ((eq-test-p test)
1702                      `(adjoin-eq ,elt ,list))
1703                     ((eql-test-p test)
1704                      `(adjoin-eql ,elt ,list))
1705                     (t nil)))
1706          whole))
1707    whole))
1708
1709(define-compiler-macro union (&whole whole list1 list2 &rest keys)
1710  (if (constant-keywords-p keys)
1711    (destructuring-bind (&key (test ''eql) test-not key) keys
1712      (or (and (null test-not)
1713               (null key)
1714               (cond ((eq-test-p test)
1715                      `(union-eq ,list1 ,list2))
1716                     ((eql-test-p test)
1717                      `(union-eql ,list1 ,list2))
1718                     (t nil)))
1719          whole))
1720    whole))
1721
1722(define-compiler-macro slot-value (&whole whole &environment env
1723                                          instance slot-name-form)
1724  (declare (ignore env))
1725  (let* ((name (and (quoted-form-p slot-name-form)
1726                    (typep (cadr slot-name-form) 'symbol)
1727                    (cadr slot-name-form))))
1728    (if name
1729      `(slot-id-value ,instance (load-time-value (ensure-slot-id ',name)))
1730      whole)))
1731
1732
1733(define-compiler-macro set-slot-value (&whole whole &environment env
1734                                          instance slot-name-form value-form)
1735  (declare (ignore env))
1736  (let* ((name (and (quoted-form-p slot-name-form)
1737                    (typep (cadr slot-name-form) 'symbol)
1738                    (cadr slot-name-form))))
1739    (if name
1740      `(set-slot-id-value
1741        ,instance
1742        (load-time-value (ensure-slot-id ',name))
1743        ,value-form)
1744      whole)))
1745
1746
1747(define-compiler-macro slot-boundp (&whole whole instance slot-name-form)
1748  (let* ((name (and (quoted-form-p slot-name-form)
1749                    (typep (cadr slot-name-form) 'symbol)
1750                    (cadr slot-name-form))))
1751    (if name
1752      `(slot-id-boundp ,instance (load-time-value (ensure-slot-id ',name)))
1753      whole)))
1754
1755(defsynonym %get-unsigned-byte %get-byte)
1756(defsynonym %get-unsigned-word %get-word)
1757(defsynonym %get-signed-long %get-long)
1758
1759
1760
1761
1762(define-compiler-macro arrayp (arg)
1763  `(>= (the fixnum (typecode ,arg))
1764    ,(nx-lookup-target-uvector-subtag :array-header)))
1765
1766(define-compiler-macro vectorp (arg)
1767  `(>= (the fixnum (typecode ,arg))
1768    ,(nx-lookup-target-uvector-subtag :vector-header)))
1769
1770
1771
1772(define-compiler-macro fixnump (arg)
1773  (let* ((fixnum-tag
1774          (arch::target-fixnum-tag (backend-target-arch *target-backend*))))
1775    `(eql (lisptag ,arg) ,fixnum-tag)))
1776
1777
1778
1779(define-compiler-macro double-float-p (n)
1780  (let* ((tag (arch::target-double-float-tag (backend-target-arch *target-backend*))))
1781    `(eql (typecode ,n) ,tag)))
1782
1783
1784(define-compiler-macro short-float-p (n)
1785  (let* ((arch (backend-target-arch *target-backend*))
1786         (tag (arch::target-single-float-tag arch))
1787         (op (if (arch::target-single-float-tag-is-subtag arch)
1788               'typecode
1789               'fulltag)))
1790    `(eql (,op ,n) ,tag)))
1791
1792
1793(define-compiler-macro floatp (n)
1794  (let* ((typecode (make-symbol "TYPECODE"))
1795         (arch (backend-target-arch *target-backend*))
1796         (single (arch::target-single-float-tag arch))
1797         (double (arch::target-double-float-tag arch)))
1798    `(let* ((,typecode (typecode ,n)))
1799       (declare (fixnum ,typecode))
1800       (or (= ,typecode ,single)
1801           (= ,typecode ,double)))))
1802
1803(define-compiler-macro functionp (n)
1804  (let* ((arch (backend-target-arch *target-backend*))
1805         (tag (arch::target-function-tag arch))
1806         (op (if (arch::target-function-tag-is-subtag arch)
1807               'typecode
1808               'fulltag)))
1809    `(eql (,op  ,n) ,tag)))
1810
1811(define-compiler-macro symbolp (s)
1812  (let* ((arch (backend-target-arch *target-backend*))
1813         (symtag (arch::target-symbol-tag arch))
1814         (op (if (arch::target-symbol-tag-is-subtag arch)
1815               'typecode
1816               'fulltag))
1817         (niltag (arch::target-null-tag arch)))
1818    (if (eql niltag symtag)
1819      `(eql (,op ,s) ,symtag)
1820      (let* ((sym (gensym)))
1821        `(let* ((,sym ,s))
1822          (if ,sym (eql (,op ,sym) ,symtag) t))))))
1823
1824;;; If NIL isn't tagged as a symbol, assume that LISPTAG only looks
1825;;; at bits that NIL shares with a cons.
1826(define-compiler-macro listp (n)
1827  (let* ((arch (backend-target-arch *target-backend*))
1828         (list-tag (logand (arch::target-cons-tag arch)
1829                           (1- (ash 1 (arch::target-nlisptagbits arch)))))
1830         (nil-tag  (arch::target-null-tag arch))
1831         (symbol-tag (arch::target-symbol-tag arch)))
1832    (if (= nil-tag symbol-tag)
1833      (let* ((nvar (gensym)))
1834        `(let* ((,nvar ,n))
1835          (if ,nvar (consp ,nvar) t)))
1836      `(eql (lisptag ,n) ,list-tag))))
1837
1838(define-compiler-macro consp (&whole call n)
1839  (let* ((arch (backend-target-arch *target-backend*))
1840         (cons-tag (arch::target-cons-tag arch))
1841         (nil-tag (arch::target-null-tag arch)))
1842    (if (= nil-tag cons-tag)
1843      call
1844      `(eql (fulltag ,n) ,cons-tag))))
1845
1846(define-compiler-macro bignump (n)
1847  `(eql (typecode ,n) ,(nx-lookup-target-uvector-subtag :bignum)))
1848
1849(define-compiler-macro ratiop (n)
1850  `(eql (typecode ,n) ,(nx-lookup-target-uvector-subtag :ratio)))
1851
1852(define-compiler-macro complexp (n)
1853  `(eql (typecode ,n) ,(nx-lookup-target-uvector-subtag :complex)))
1854
1855(define-compiler-macro macptrp (n)
1856  `(eql (typecode ,n) ,(nx-lookup-target-uvector-subtag :macptr)))
1857
1858(define-compiler-macro basic-stream-p (n)
1859  `(eql (typecode ,n) ,(nx-lookup-target-uvector-subtag :basic-stream)))
1860
1861(define-compiler-macro aref (&whole call a &rest subscripts &environment env)
1862  (let* ((ctype (if (nx-form-typep a 'array env)
1863                  (specifier-type (nx-form-type a env) env)))
1864         (ectype (typecase ctype
1865                   (array-ctype (array-ctype-specialized-element-type ctype))
1866                   (union-ctype (when (every #'array-ctype-p (union-ctype-types ctype))
1867                                  (%type-union
1868                                   (mapcar (lambda (ct) (array-ctype-specialized-element-type ct))
1869                                           (union-ctype-types ctype)))))))
1870         (etype (and ectype (type-specifier ectype)))
1871         (useful (unless (or (eq etype *) (eq etype t))
1872                   etype)))
1873    (if (= 2 (length subscripts))
1874      (setq call `(%aref2 ,a ,@subscripts))
1875      (if (= 3 (length subscripts))
1876        (setq call `(%aref3 ,a ,@subscripts))))
1877    (if useful
1878      `(the ,useful ,call)
1879      call)))
1880
1881
1882(define-compiler-macro aset (&whole call a &rest subs&val)
1883  (if (= 3 (length subs&val))
1884    `(%aset2 ,a ,@subs&val)
1885    (if (= 4 (length subs&val))
1886      `(%aset3 ,a ,@subs&val)
1887      call)))
1888
1889
1890(define-compiler-macro make-sequence (&whole call typespec len &rest keys &key initial-element)
1891  (declare (ignore typespec len keys initial-element))
1892  call)
1893
1894(define-compiler-macro make-string (&whole call &environment env size &rest keys)
1895  (if (constant-keywords-p keys)
1896    (destructuring-bind (&key (element-type () element-type-p)
1897                              (initial-element () initial-element-p))
1898                        keys
1899      (if (and element-type-p
1900               (nx-form-constant-p element-type env))
1901        (let* ((element-type (nx-form-constant-value element-type env)))
1902          (if (subtypep element-type 'base-char)
1903            `(allocate-typed-vector :simple-string ,size ,@(if initial-element-p `(,initial-element)))
1904            call))
1905        (if (not element-type-p)
1906          `(allocate-typed-vector :simple-string ,size ,@(if initial-element-p `(,initial-element)))
1907          call)))
1908    call))
1909
1910(define-compiler-macro make-string-output-stream (&whole whole &rest keys)
1911  (if (null keys)
1912    '(make-simple-string-output-stream)
1913    whole))
1914
1915
1916(define-compiler-macro write-string (&environment env &whole call
1917                                                  string &optional (stream nil) &rest keys)
1918  (if (nx-form-typep string 'simple-string env)
1919    (if keys
1920      `((lambda (string stream &key (start 0) end)
1921          (write-simple-string string stream start end))
1922        ,string ,stream ,@keys)
1923      `(write-simple-string ,string ,stream 0 nil))
1924    call))
1925
1926(define-compiler-macro format (&environment env &whole call stream string &rest args)
1927  (if (stringp string)
1928    (cond ((and (string-equal string "~a") args (null (cdr args)))
1929           (destructuring-bind (object) args
1930             (cond ((null stream)
1931                    `(princ-to-string ,object))
1932                   ((or (eq stream t) (nx-form-typep stream 'stream env))
1933                    `(progn (princ ,object ,(and (neq stream t) stream)) nil))
1934                   (t `(let ((stream ,stream)
1935                             (object ,object))
1936                         (if (or (null stream) (stringp stream))
1937                           (format-to-string stream ,string object)
1938                           (progn (princ object (and (neq stream t) stream)) nil)))))))
1939          ((and (string-equal string "~s") args (null (cdr args)))
1940           (destructuring-bind (object) args
1941             (cond ((null stream)
1942                    `(prin1-to-string ,object))
1943                   ((or (eq stream t) (nx-form-typep stream 'stream env))
1944                    `(progn (prin1 ,object ,(and (neq stream t) stream)) nil))
1945                   (t `(let ((stream ,stream)
1946                             (object ,object))
1947                         (if (or (null stream) (stringp stream))
1948                           (format-to-string stream ,string object)
1949                           (progn (prin1 object (and (neq stream t) stream)) nil)))))))
1950          ((and (null (position #\~ string)) (null args))
1951           (cond ((null stream)
1952                  string)
1953                 ((or (eq stream t) (nx-form-typep stream 'stream env))
1954                  `(progn (write-string ,string ,(and (neq stream t) stream)) nil))
1955                 (t `(let ((stream ,stream))
1956                       (if (or (null stream) (stringp stream))
1957                         (format-to-string stream ,string)
1958                         (progn (write-string ,string (and (neq stream t) stream)) nil))))))
1959          ((let ((new (format-string-sans~newlines string)))
1960             (and (neq new string) (setq string new)))
1961           `(format ,stream ,string ,@args))
1962          ((optimize-format-call stream string args env))
1963          (t call))
1964    call))
1965
1966(defun format-string-sans~newlines (string)
1967  (loop as pos = 0 then (position #\Newline string :start pos) while pos
1968        as ch = (and (> pos 0) (schar string (1- pos)))
1969        do (cond ((not (or (eq ch #\~)
1970                           (and (or (eq ch #\:) (eq ch #\@))
1971                                (> pos 1) (eq (schar string (- pos 2)) #\~))))
1972                  (incf pos))
1973                 ((eq ch #\:)
1974                  (decf pos 2)
1975                  (setq string (%str-cat (subseq string 0 pos) (subseq string (+ pos 3)))))
1976                 ((eq ch #\@)
1977                  (setq string (%str-cat (subseq string 0 (- pos 2))
1978                                         "~%"
1979                                         (subseq string (or
1980                                                         (position-if-not #'whitespacep string
1981                                                                          :start (1+ pos))
1982                                                         (1+ pos))))))
1983                  ((eq ch #\~)
1984                  (decf pos)
1985                  (setq string (%str-cat (subseq string 0 pos)
1986                                         (subseq string (or (position-if-not #'whitespacep string
1987                                                                         :start (1+ pos))
1988                                                            (1+ pos))))))))
1989  string)
1990
1991(defun count-known-format-args (string start end)
1992  (declare (fixnum end))
1993  (loop with count = 0
1994        do (setq start (position #\~ string :start start :end end))
1995        when (null start)
1996          do (return count)
1997        unless (< (incf start) end)
1998          do (return nil)
1999        do (let ((ch (aref string start)))
2000             (cond ((memq ch '(#\a #\A #\s #\S)) (incf count))
2001                   ((memq ch '(#\~ #\% #\&)))
2002                   (t (return nil)))
2003             (incf start))))
2004
2005(defun optimize-format-call (stream string args env)
2006  (let* ((start (or (search "~/" string)
2007                    (return-from optimize-format-call nil)))
2008         (ipos (+ start 2))
2009         (epos (or (position #\/ string :start ipos)
2010                   (return-from optimize-format-call nil)))
2011         (nargs (or (count-known-format-args string 0 start)
2012                    (return-from optimize-format-call nil))))
2013    (when (and
2014           ;; Must be able to split args
2015           (< nargs (length args))
2016           ;; Don't deal with packages
2017           (not (position #\: string :start ipos :end epos)))
2018      (let* ((func (intern (string-upcase (subseq string ipos epos)) :cl-user))
2019             (prev (and (< 0 start) (subseq string 0 start)))
2020             (prev-args (subseq args 0 nargs))
2021             (rest (and (< (1+ epos) (length string)) (subseq string (1+ epos))))
2022             (rest-args (nthcdr nargs args))
2023             (obj (pop rest-args))
2024             (stream-var (gensym))
2025             (body `(,@(and prev `((format ,stream-var ,prev ,@prev-args)))
2026                       (,func ,stream-var ,obj nil nil)
2027                       ,(if rest `(format ,stream-var ,rest ,@rest-args) `nil))))
2028        (cond ((null stream)
2029               `(with-output-to-string (,stream-var)
2030                  (declare (type stream ,stream-var))
2031                  ,@body))
2032              ((or (eq stream t) (nx-form-typep stream 'stream env))
2033               `(let ((,stream-var ,(if (eq stream t) '*standard-output* stream)))
2034                  (declare (type stream ,stream-var))
2035                  ,@body))
2036              (t
2037               `(let ((,stream-var ,stream))
2038                  (if (or (null ,stream-var) (stringp ,stream-var))
2039                    (format-to-string ,stream-var ,string ,@args)
2040                    (let ((,stream-var
2041                           (if (eq ,stream-var t) *standard-output* ,stream-var)))
2042                      ;; For the purposes of body, it's ok to assume stream-var
2043                      ;; is a stream. method dispatch will signal any errors
2044                      ;; at runtime if it's not true...
2045                      (declare (type stream ,stream-var))
2046                      ,@body)))))))))
2047
2048
2049(define-compiler-macro sbit (&whole call v &optional sub0 &rest others)
2050  (if (and sub0 (null others))
2051    `(aref (the simple-bit-vector ,v) ,sub0)
2052    call))
2053
2054(define-compiler-macro %sbitset (&whole call v sub0 &optional (newval nil newval-p) &rest newval-was-really-sub1)
2055  (if (and newval-p (not newval-was-really-sub1) )
2056    `(setf (aref (the simple-bit-vector ,v) ,sub0) ,newval)
2057    call))
2058
2059(define-compiler-macro simple-base-string-p (thing)
2060  `(= (the fixnum (typecode ,thing)) ,(nx-lookup-target-uvector-subtag :simple-string)))
2061
2062(define-compiler-macro simple-string-p (thing)
2063  `(simple-base-string-p ,thing))
2064
2065(define-compiler-macro stringp (thing)
2066  `(base-string-p  ,thing))
2067
2068(define-compiler-macro base-string-p (thing)
2069  (let* ((gthing (gensym))
2070         (gtype (gensym)))
2071    `(let* ((,gthing ,thing)
2072            (,gtype (typecode ,gthing)))
2073      (declare (type (unsigned-byte 8) ,gtype))
2074      (if (= ,gtype ,(nx-lookup-target-uvector-subtag :vector-header))
2075        (= (the (unsigned-byte 8)
2076             (ldb target::arrayH.flags-cell-subtag-byte (the fixnum (%svref ,gthing target::arrayH.flags-cell))))
2077           ,(nx-lookup-target-uvector-subtag :simple-string))
2078        (= ,gtype ,(nx-lookup-target-uvector-subtag :simple-string))))))
2079
2080
2081
2082(defsetf %misc-ref %misc-set)
2083
2084(define-compiler-macro lockp (lock)
2085  (let* ((tag (nx-lookup-target-uvector-subtag :lock)))
2086    `(eq ,tag (typecode ,lock))))
2087
2088(define-compiler-macro structurep (s)
2089  (let* ((tag (nx-lookup-target-uvector-subtag :struct)))
2090    `(eq ,tag (typecode ,s))))
2091 
2092
2093(define-compiler-macro integerp (thing)
2094  (let* ((typecode (gensym))
2095         (fixnum-tag (arch::target-fixnum-tag (backend-target-arch *target-backend*)))
2096         (bignum-tag (nx-lookup-target-uvector-subtag :bignum)))
2097    `(let* ((,typecode (typecode ,thing)))
2098      (declare (fixnum ,typecode))
2099      (if (= ,typecode ,fixnum-tag)
2100        t
2101        (= ,typecode ,bignum-tag)))))
2102
2103(define-compiler-macro realp (&whole call x)
2104  (if (not (eq *host-backend* *target-backend*))
2105    call
2106    (let* ((typecode (gensym)))
2107      `(let* ((,typecode (typecode ,x)))
2108        (declare (type (unsigned-byte 8) ,typecode))
2109        #+(or ppc32-target x8632-target arm-target)
2110        (or (= ,typecode target::tag-fixnum)
2111         (and (>= ,typecode target::min-numeric-subtag)
2112          (<= ,typecode target::max-real-subtag)))
2113        #+ppc64-target
2114        (if (<= ,typecode ppc64::subtag-double-float)
2115          (logbitp (the (integer 0 #.ppc64::subtag-double-float) ,typecode)
2116                   (logior (ash 1 ppc64::tag-fixnum)
2117                           (ash 1 ppc64::subtag-single-float)
2118                           (ash 1 ppc64::subtag-double-float)
2119                           (ash 1 ppc64::subtag-bignum)
2120                           (ash 1 ppc64::subtag-ratio))))
2121        #+x8664-target
2122        (if (<= ,typecode x8664::subtag-double-float)
2123          (logbitp (the (integer 0 #.x8664::subtag-double-float) ,typecode)
2124                   (logior (ash 1 x8664::tag-fixnum)
2125                           (ash 1 x8664::subtag-bignum)
2126                           (ash 1 x8664::tag-single-float)
2127                           (ash 1 x8664::subtag-double-float)
2128                           (ash 1 x8664::subtag-ratio))))))))
2129
2130(define-compiler-macro %composite-pointer-ref (size pointer offset)
2131  (if (constantp size)
2132    `(%inc-ptr ,pointer ,offset)
2133    `(progn
2134      ,size
2135      (%inc-ptr ,pointer ,offset))))
2136
2137
2138(define-compiler-macro char= (&whole call ch &optional (other nil other-p) &rest others)
2139  (if (null others)
2140    (if other-p
2141      `(eq (char-code ,ch) (char-code ,other))
2142      `(progn (char-code ,ch) t))
2143    (if (null (cdr others))
2144      (let* ((third (car others))
2145             (code (gensym))
2146             (code2 (gensym))
2147             (code3 (gensym)))
2148        `(let* ((,code (char-code ,ch))
2149                (,code2 (char-code ,other))
2150                (,code3 (char-code ,third)))
2151          (and (eq ,code ,code2)
2152           (eq ,code2 ,code3))))
2153      call)))
2154
2155(define-compiler-macro char-equal (&whole call ch &optional (other nil other-p) &rest others)
2156  (if (null others)
2157    (if other-p
2158      `(eq (%char-code-upcase (char-code ,ch)) (%char-code-upcase (char-code ,other)))
2159      `(progn (char-code ,ch) t))
2160    (if (null (cdr others))
2161      (let* ((third (car others))
2162             (code (gensym))
2163             (code2 (gensym))
2164             (code3 (gensym)))
2165        `(let* ((,code (%char-code-upcase (char-code ,ch)))
2166                (,code2 (%char-code-upcase (char-code ,other)))
2167                (,code3 (%char-code-upcase (char-code ,third))))
2168          (and (eq ,code ,code2)
2169           (eq ,code ,code3))))
2170      call)))
2171
2172(define-compiler-macro char/= (&whole call ch &optional (other nil other-p) &rest others)
2173  (if (null others)
2174    (if other-p
2175      `(not (eq (char-code ,ch) (char-code ,other)))
2176      `(progn (char-code ,ch) t))
2177    call))
2178
2179
2180(define-compiler-macro char< (&whole call ch &optional (other nil other-p) &rest others)
2181  (if (null others)
2182    (if other-p
2183      `(< (the fixnum (char-code ,ch)) (the fixnum (char-code ,other)))
2184      `(progn (char-code ,ch) t))
2185    (if (null (cdr others))
2186      (let* ((third (car others))
2187             (code (gensym))
2188             (code2 (gensym))
2189             (code3 (gensym)))
2190        ;; We have to evaluate all forms for side-effects.
2191        ;; Hopefully, there won't be any
2192        `(let* ((,code (char-code ,ch))
2193                (,code2 (char-code ,other))
2194                (,code3 (char-code ,third)))
2195          (declare (fixnum ,code ,code2 ,code3))
2196          (and (< ,code ,code2)
2197           (< ,code2 ,code3))))
2198      call)))
2199
2200(define-compiler-macro char<= (&whole call ch &optional (other nil other-p) &rest others)
2201  (if (null others)
2202    (if other-p
2203      `(<= (the fixnum (char-code ,ch)) (the fixnum (char-code ,other)))
2204      `(progn (char-code ,ch) t))
2205    (if (null (cdr others))
2206      (let* ((third (car others))
2207             (code (gensym))
2208             (code2 (gensym))
2209             (code3 (gensym)))
2210        `(let* ((,code (char-code ,ch))
2211                (,code2 (char-code ,other))
2212                (,code3 (char-code ,third)))
2213          (declare (fixnum ,code ,code2 ,code3))
2214          (and (<= ,code ,code2)
2215           (<= ,code2 ,code3))))
2216      call)))
2217
2218(define-compiler-macro char> (&whole call ch &optional (other nil other-p) &rest others)
2219  (if (null others)
2220    (if other-p
2221      `(> (the fixnum (char-code ,ch)) (the fixnum (char-code ,other)))
2222      `(progn (char-code ,ch) t))
2223    (if (null (cdr others))
2224      (let* ((third (car others))
2225             (code (gensym))
2226             (code2 (gensym))
2227             (code3 (gensym)))
2228        `(let* ((,code (char-code ,ch))
2229                (,code2 (char-code ,other))
2230                (,code3 (char-code ,third)))
2231          (declare (fixnum ,code ,code2 ,code3))
2232          (and (> ,code ,code2)
2233           (> ,code2 ,code3))))
2234      call)))
2235
2236(define-compiler-macro char>= (&whole call ch &optional (other nil other-p) &rest others)
2237  (if (null others)
2238    (if other-p
2239      `(>= (the fixnum (char-code ,ch)) (the fixnum (char-code ,other)))
2240      `(progn (char-code ,ch) t))
2241    (if (null (cdr others))
2242      (let* ((third (car others))
2243             (code (gensym))
2244             (code2 (gensym))
2245             (code3 (gensym)))
2246        `(let* ((,code (char-code ,ch))
2247                (,code2 (char-code ,other))
2248                (,code3 (char-code ,third)))
2249          (declare (fixnum ,code ,code2 ,code3))
2250          (and (>= ,code ,code2)
2251           (>= ,code2 ,code3))))
2252      call)))
2253
2254(define-compiler-macro float (&whole call number &optional (other 0.0f0 other-p) &environment env)
2255
2256  (cond ((and (typep other 'single-float)
2257              (nx-form-typep number 'double-float env))
2258         `(the single-float (%double-to-single ,number)))
2259        ((and (typep other 'double-float)
2260              (nx-form-typep number 'single-float env))
2261         `(the double-float (%single-to-double ,number)))
2262        ((and other-p (typep other 'single-float))
2263         `(the single-float (%short-float ,number)))
2264        ((typep other 'double-float)
2265         `(the double-float (%double-float ,number)))
2266        ((null other-p)
2267         (let* ((temp (gensym)))
2268           `(let* ((,temp ,number))
2269             (if (typep ,temp 'double-float)
2270               ,temp
2271               (the single-float (%short-float ,temp))))))
2272        (t call)))
2273
2274(define-compiler-macro coerce (&whole call &environment env thing type)
2275  (cond ((nx-form-constant-p type env)
2276         (setq type (nx-form-constant-value type env))
2277         (let ((ctype (specifier-type-if-known type env :whine t)))
2278           (if ctype
2279             (if (csubtypep ctype (specifier-type 'single-float))
2280                 `(float ,thing 0.0f0)
2281                 (if (csubtypep ctype (specifier-type 'double-float))
2282                     `(float ,thing 0.0d0)
2283                     (let ((simple nil)
2284                           (extra nil))
2285                       (if (and (typep ctype 'array-ctype)
2286                                (equal (array-ctype-dimensions ctype) '(*)))
2287                           (if (eq (array-ctype-specialized-element-type ctype)
2288                                   (specifier-type 'character))
2289                               (setq simple '%coerce-to-string)
2290                               (if (and (eq *host-backend* *target-backend*)
2291                                        (array-ctype-typecode ctype))
2292                                   (setq simple '%coerce-to-vector
2293                                         extra (list (array-ctype-typecode ctype)))))
2294                           (if (eq ctype (specifier-type 'list))
2295                               (setq simple '%coerce-to-list)))
2296                       (if simple
2297                           (let* ((temp (gensym)))
2298                             `(let* ((,temp ,thing))
2299                                (if (typep ,temp ',(type-specifier ctype))
2300                                    ,temp
2301                                    (,simple ,temp ,@extra))))
2302                           call))))
2303             call)))
2304        (t call)))
2305
2306(define-compiler-macro equal (&whole call x y &environment env)
2307  (if (or (equal-iff-eql-p x env)
2308          (equal-iff-eql-p y env))
2309    `(eql ,x ,y)
2310    call))
2311
2312(define-compiler-macro instance-slots (instance &environment env)
2313  (if (and (nx-form-constant-p instance env)
2314           (eql (typecode (nx-form-constant-value instance env)) (nx-lookup-target-uvector-subtag :instance)))
2315    `(instance.slots ,instance)
2316    (let* ((itemp (gensym))
2317           (typecode (gensym)))
2318      `(let* ((,itemp ,instance)
2319              (,typecode (typecode ,itemp)))
2320        (declare (type (unsigned-byte 8) ,typecode))
2321        (if (eql ,typecode ,(nx-lookup-target-uvector-subtag :instance))
2322          (instance.slots ,itemp)
2323          (%non-standard-instance-slots ,itemp ,typecode))))))
2324
2325(define-compiler-macro instance-class-wrapper (instance)
2326  (let* ((itemp (gensym)))
2327    `(let* ((,itemp ,instance))
2328      (if (eql (the (unsigned-byte 8) (typecode ,itemp))
2329               ,(nx-lookup-target-uvector-subtag :instance))
2330        (instance.class-wrapper ,itemp)
2331        (non-standard-instance-class-wrapper ,itemp)))))
2332
2333;; Instance must be a standard-instance.
2334(define-compiler-macro %class-of-instance (instance)
2335  `(%wrapper-class (instance.class-wrapper ,instance)))
2336
2337(define-compiler-macro standard-object-p (thing)
2338  (let* ((temp (gensym))
2339         (typecode (gensym)))
2340    `(let* ((,temp ,thing)
2341            (,typecode (typecode ,temp)))
2342      (declare (type (unsigned-byte 8) ,typecode))
2343      (if (= ,typecode ,(nx-lookup-target-uvector-subtag :instance))
2344        (instance.class-wrapper ,temp)
2345        (if (= ,typecode ,(nx-lookup-target-uvector-subtag :macptr))
2346          (foreign-instance-class-wrapper ,temp))))))
2347
2348(define-compiler-macro %class-ordinal (class &optional error)
2349  (let* ((temp (gensym)))
2350    `(let* ((,temp ,class))
2351      (if (eql (the (unsigned-byte 8) (typecode ,temp))
2352               ,(nx-lookup-target-uvector-subtag :instance))
2353        (instance.hash ,temp)
2354        (funcall '%class-ordinal ,temp ,error)))))
2355
2356(define-compiler-macro native-class-p (class)
2357  (let* ((temp (gensym)))
2358    `(let* ((,temp ,class))
2359      (if (eql (the (unsigned-byte 8) (typecode ,temp))
2360               ,(nx-lookup-target-uvector-subtag :instance))
2361        (< (the fixnum (instance.hash ,temp)) max-class-ordinal)))))
2362 
2363
2364
2365(define-compiler-macro unsigned-byte-p (x)
2366  (if (typep (nx-unquote x) 'unsigned-byte)
2367    t
2368    (let* ((val (gensym)))
2369      `(let* ((,val ,x))
2370        (and (integerp ,val) (not (< ,val 0)))))))
2371
2372(define-compiler-macro subtypep (&whole w t1 t2 &optional rtenv)
2373  (if (and (consp t1)
2374           (consp (cdr t1))
2375           (null (cddr t1))
2376           (eq (car t1) 'type-of))
2377    ;; People really write code like this.  I've seen it.
2378    `(typep ,(cadr t1) ,t2 ,@(and rtenv `(,rtenv)))
2379    (if (and (null rtenv) (quoted-form-p t2))
2380      `(cell-csubtypep-2 ,t1 (load-time-value (register-type-cell ,t2)))
2381      w)))
2382
2383
2384(define-compiler-macro string-equal (s1 s2 &rest keys)
2385  (if (null keys)
2386    `(%fixed-string-equal ,s1 ,s2)
2387    (let* ((s1-arg (gensym))
2388           (s2-arg (gensym)))
2389      `(funcall
2390        (lambda (,s1-arg ,s2-arg &key start1 end1 start2 end2)
2391          (%bounded-string-equal ,s1-arg ,s2-arg start1 end1 start2 end2))
2392        ,s1 ,s2 ,@keys))))
2393
2394;;; Try to use "package-references" to speed up package lookup when
2395;;; a package name is used as a constant argument to some functions.
2396
2397(defun package-ref-form (arg env)
2398  (when (and arg (nx-form-constant-p arg env)
2399             (typep (setq arg (nx-form-constant-value arg env))
2400                    '(or symbol string)))
2401    `(load-time-value (register-package-ref ,(string arg)))))
2402
2403
2404
2405(define-compiler-macro intern (&whole w string &optional package &environment env)
2406  (let* ((ref (package-ref-form package env)))
2407    (if (or ref
2408            (setq ref (and (consp package)
2409                           (eq (car package) 'find-package)
2410                           (consp (cdr package))
2411                           (null (cddr package))
2412                           (package-ref-form (cadr package) env))))
2413      `(%pkg-ref-intern ,string ,ref)
2414      w)))
2415
2416(define-compiler-macro find-symbol (&whole w string &optional package &environment env)
2417  (let* ((ref (package-ref-form package env)))
2418    (if (or ref
2419            (setq ref (and (consp package)
2420                           (eq (car package) 'find-package)
2421                           (consp (cdr package))
2422                           (null (cddr package))
2423                           (package-ref-form (cadr package) env))))
2424      `(%pkg-ref-find-symbol ,string ,ref)
2425      w)))
2426
2427(define-compiler-macro find-package (&whole w package &environment env)
2428  (let* ((ref (package-ref-form package env)))
2429    (if ref
2430      `(package-ref.pkg ,ref)
2431      w)))
2432
2433(define-compiler-macro pkg-arg (&whole w package &optional allow-deleted &environment env)
2434  (let* ((ref (unless allow-deleted (package-ref-form package env))))
2435    (if ref
2436      (let* ((r (gensym)))
2437        `(let* ((,r ,ref))
2438          (or (package-ref.pkg ,ref)
2439           (%kernel-restart $xnopkg (package-ref.pkg ,r)))))
2440      w)))
2441
2442
2443;;; In practice, things that're STREAMP are almost always
2444;;; BASIC-STREAMs or FUNDAMENTAL-STREAMs, but STREAMP is a generic
2445;;; function.
2446(define-compiler-macro streamp (arg)
2447  (let* ((s (gensym)))
2448    `(let* ((,s ,arg))
2449      (or (typep ,s 'basic-stream)
2450       (typep ,s 'fundamental-stream)
2451       ;; Don't recurse
2452       (funcall 'streamp ,s)))))
2453
2454
2455(define-compiler-macro %char-code-case-fold (&whole w code vector &environment env)
2456  (if (nx-open-code-in-line env)
2457    (let* ((c (gensym))
2458           (table (gensym)))
2459      `(let* ((,c ,code)
2460              (,table ,vector))
2461        (declare (type (mod #x110000) ,c)
2462                 (type (simple-array (signed-byte 16) (*)) ,table))
2463        (if (< ,c (length ,table))
2464          (the fixnum (+ ,c (the (signed-byte 16)
2465                              (locally (declare (optimize (speed 3) (safety 0)))
2466                                (aref ,table ,c)))))
2467          ,c)))
2468    w))
2469       
2470(define-compiler-macro %char-code-upcase (code)
2471  (if (typep code '(mod #x110000))
2472    (%char-code-upcase code)
2473    `(%char-code-case-fold ,code *lower-to-upper*)))
2474
2475(define-compiler-macro %char-code-downcase (code)
2476  (if (typep code '(mod #x110000))
2477    (%char-code-downcase code)
2478    `(%char-code-case-fold ,code *upper-to-lower*)))
2479
2480(define-compiler-macro char-upcase (char)
2481  `(code-char (the valid-char-code (%char-code-upcase (char-code ,char)))))
2482
2483(define-compiler-macro char-downcase (char)
2484  `(code-char (the valid-char-code (%char-code-downcase (char-code ,char)))))
2485
2486
2487(define-compiler-macro register-istruct-cell (&whole w arg &environment env)
2488  (if (and (nx-form-constant-p arg env)
2489           (setq arg (nx-form-constant-value arg env))
2490           (symbolp arg))
2491    `',(register-istruct-cell arg)
2492    w))
2493
2494(define-compiler-macro get-character-encoding (&whole w name)
2495  (or (if (typep name 'keyword) (lookup-character-encoding name))
2496      w))
2497
2498(define-compiler-macro read-char (&optional stream (eof-error-p t) eof-value recursive-p)
2499  `(read-char-internal ,stream ,eof-error-p (values ,eof-value ,recursive-p)))
2500
2501
2502(provide "OPTIMIZERS")
Note: See TracBrowser for help on using the repository browser.