source: trunk/j/src/org/armedbear/lisp/jvm.lisp @ 4653

Last change on this file since 4653 was 4653, checked in by piso, 18 years ago

Work in progress.

File size: 68.3 KB
Line 
1;;; jvm.lisp
2;;;
3;;; Copyright (C) 2003 Peter Graves
4;;; $Id: jvm.lisp,v 1.9 2003-11-05 19:34:28 piso Exp $
5;;;
6;;; This program is free software; you can redistribute it and/or
7;;; modify it under the terms of the GNU General Public License
8;;; as published by the Free Software Foundation; either version 2
9;;; of the License, or (at your option) any later version.
10;;;
11;;; This program is distributed in the hope that it will be useful,
12;;; but WITHOUT ANY WARRANTY; without even the implied warranty of
13;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14;;; GNU General Public License for more details.
15;;;
16;;; You should have received a copy of the GNU General Public License
17;;; along with this program; if not, write to the Free Software
18;;; Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
19
20(require 'transform)
21
22(in-package "JVM")
23
24(export '(jvm-compile jvm-compile-package))
25
26(shadow 'method)
27
28(defvar *instructions*
29  '(nop             aconst_null  iconst_m1       iconst_0      iconst_1     ;   0
30    iconst_2        iconst_3     iconst_4        iconst_5      lconst_0     ;   5
31    lconst_1        fconst_0     fconst_1        fconst_2      dconst_0     ;  10
32    dconst_1        bipush       sipush          ldc           ldc_w        ;  15
33    ldc2_w          iload        lload           fload         dload        ;  20
34    aload           iload_0      iload_1         iload_2       iload_3      ;  25
35    lload_0         lload_1      lload_2         lload_3       fload_0      ;  30
36    fload_1         fload_2      fload_3         dload_0       dload_1      ;  35
37    dload_2         dload_3      aload_0         aload_1       aload_2      ;  40
38    aload_3         iaload       laload          faload        daload       ;  45
39    aaload          baload       caload          saload        istore       ;  50
40    lstore          fstore       dstore          astore        istore_0     ;  55
41    istore_1        istore_2     istore_3        lstore_0      lstore_1     ;  60
42    lstore_2        lstore_3     fstore_0        fstore_1      fstore_2     ;  65
43    fstore_3        dstore_0     dstore_1        dstore_2      dstore_3     ;  70
44    astore_0        astore_1     astore_2        astore_3      iastore      ;  75
45    lastore         fastore      dastore         aastore       bastore      ;  80
46    castore         sastore      pop             pop2          dup          ;  85
47    dup_x1          dup_x2       dup2            dup2_x1       dup2_x2      ;  90
48    swap            iadd         ladd            fadd          dadd         ;  95
49    isub            lsub         fsub            dsub          imul         ; 100
50    lmul            fmul         dmul            idiv          ldiv         ; 105
51    fdiv            ddiv         irem            lrem          frem         ; 110
52    drem            ineg         lneg            fneg          dneg         ; 115
53    ishl            lshl         ishr            lshr          iushr        ; 120
54    lushr           iand         land            ior           lor          ; 125
55    ixor            lxor         iinc            i2l           i2f          ; 130
56    i2d             l2i          l2f             l2d           f2i          ; 135
57    f2l             f2d          d2i             d2l           d2f          ; 140
58    i2b             i2c          i2s             lcmp          fcmpl        ; 145
59    fcmpg           dcmpl        dcmpg           ifeq          ifne         ; 150
60    iflt            ifge         ifgt            ifle          if_icmpeq    ; 155
61    if_icmpne       if_icmplt    if_icmpge       if_icmpgt     if_icmple    ; 160
62    if_acmpeq       if_acmpne    goto            jsr           ret          ; 165
63    tableswitch     lookupswitch ireturn         lreturn       freturn      ; 170
64    dreturn         areturn      return          getstatic     putstatic    ; 175
65    getfield        putfield     invokevirtual   invokespecial invokestatic ; 180
66    invokeinterface unused       new             newarray      anewarray    ; 185
67    arraylength     athrow       checkcast       instanceof    monitorenter ; 190
68    monitorexit     wide         multianewarray  ifnull        ifnonnull    ; 195
69    goto_w          jsr_w        label                                      ; 200
70    ))
71
72(unless (vectorp *instructions*)
73  (let* ((list *instructions*)
74         (vector (make-array (length *instructions*)))
75         (index 0))
76    (dolist (instr list)
77      (setf (get instr 'opcode) index)
78      (setf (svref vector index) instr)
79      (incf index))
80    (setq *instructions* vector)))
81
82(defun instr (opcode)
83  (svref *instructions* opcode))
84
85(defparameter *opcode-size*
86  ;; 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9
87  '#(1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 2 3 2 3  ;; 000-019
88     3 2 2 2 2 2 1 1 1 1 1 1 1 1 1 1 1 1 1 1  ;; 020-039
89     1 1 1 1 1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 1  ;; 040-059
90     1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1  ;; 060-079
91     1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1  ;; 080-099
92     1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1  ;; 100-119
93     1 1 1 1 1 1 1 1 1 1 1 1 3 1 1 1 1 1 1 1  ;; 120-139
94     1 1 1 1 1 1 1 1 1 1 1 1 1 3 3 3 3 3 3 3  ;; 140-159
95     3 3 3 3 3 3 3 3 3 2 0 0 1 1 1 1 1 1 3 3  ;; 160-179
96     3 3 3 3 3 5 0 3 2 3 1 1 3 3 1 1 0 4 3 3  ;; 180-199
97     5 5 0                                    ;; 200-202
98     ))
99
100(defun opcode-size (opcode)
101  (svref *opcode-size* opcode))
102
103(defvar *pool* nil)
104
105(defun read-u1 (stream)
106  (read-byte stream))
107
108(defun read-u2 (stream)
109  (+ (ash (read-byte stream) 8) (read-byte stream)))
110
111(defun read-u4 (stream)
112  (+ (ash (read-u2 stream) 16) (read-u2 stream)))
113
114(defun lookup-utf8 (index)
115  (let ((entry (svref *pool* index)))
116    (when (eql (car entry) 1)
117      (caddr entry))))
118
119(defun read-constant-pool-entry (stream)
120  (let ((tag (read-u1 stream))
121        info)
122    (case tag
123      ((7 8)
124       (list tag (read-u2 stream)))
125      (1
126       (let* ((len (read-u2 stream))
127              (s (make-string len)))
128         (dotimes (i len)
129;;            (setf (char s i) (coerce (read-u1 stream) 'character)))
130           (setf (char s i) (code-char (read-u1 stream))))
131         (list tag len s)))
132      ((3 4)
133       (list tag (read-u4 stream)))
134      ((5 6)
135       (list tag (read-u4 stream) (read-u4 stream)))
136      ((12 9 10 11)
137       (list tag (read-u2 stream) (read-u2 stream)))
138      (t
139       (error "READ-CONSTANT-POOL-ENTRY unhandled tag ~D" tag)))))
140
141(defvar *indent* 0)
142
143(defparameter *spaces* (make-string 256 :initial-element #\space))
144
145(defmacro out (&rest args)
146  `(progn (format t (subseq *spaces* 0 *indent*)) (format t ,@args)))
147
148(defun dump-code (code)
149  (let ((code-length (length code)))
150    (do ((i 0))
151        ((>= i code-length))
152      (let* ((opcode (svref code i))
153             (size (opcode-size opcode)))
154        (out "~D (#x~X) ~A~%" opcode opcode (instr opcode))
155        (incf i)
156        (dotimes (j (1- size))
157          (let ((byte (svref code i)))
158            (out "~D (#x~X)~%" byte byte))
159          (incf i))))))
160
161(defun dump-code-attribute (stream)
162  (let ((*indent* (+ *indent* 2)))
163    (out "Stack: ~D~%" (read-u2 stream))
164    (out "Locals: ~D~%" (read-u2 stream))
165    (let* ((code-length (read-u4 stream))
166           (code (make-array code-length)))
167      (out "Code length: ~D~%" code-length)
168      (out "Code:~%")
169      (dotimes (i code-length)
170        (setf (svref code i) (read-u1 stream)))
171      (let ((*indent* (+ *indent* 2)))
172        (dump-code code)))
173    (let ((exception-table-length (read-u2 stream)))
174      (out "Exception table length: ~D~%" exception-table-length)
175      (let ((*indent* (+ *indent* 2)))
176        (dotimes (i exception-table-length)
177          (out "Start PC: ~D~%" (read-u2 stream))
178          (out "End PC: ~D~%" (read-u2 stream))
179          (out "Handler PC: ~D~%" (read-u2 stream))
180          (out "Catch type: ~D~%" (read-u2 stream)))))
181    (let ((attributes-count (read-u2 stream)))
182      (out "Number of attributes: ~D~%" attributes-count)
183      (let ((*indent* (+ *indent* 2)))
184        (dotimes (i attributes-count)
185          (read-attribute i stream))))))
186
187(defun dump-exceptions (stream)
188  )
189
190(defun read-attribute (index stream)
191  (let* ((name-index (read-u2 stream))
192         (length (read-u4 stream))
193         (*indent (+ *indent* 2)))
194    (setq name (lookup-utf8 name-index))
195    (out "Attribute ~D: Name index: ~D (~S)~%" index name-index name)
196    (out "Attribute ~D: Length: ~D~%" index length)
197    (cond ((string= name "Code")
198           (dump-code-attribute stream))
199          ((string= name "Exceptions")
200           (let ((count (read-u2 stream)))
201             (out "Attribute ~D: Number of exceptions: ~D~%" index count)
202             (let ((*indent* (+ *indent* 2)))
203               (dotimes (i count)
204                 (out "Exception ~D: ~D~%" i (read-u2 stream))))))
205          (t
206           (dotimes (i length)
207             (read-u1 stream))))))
208
209(defun read-info (index stream type)
210  (let* ((access-flags (read-u2 stream))
211         (name-index (read-u2 stream))
212         (descriptor-index (read-u2 stream))
213         (attributes-count (read-u2 stream))
214         (*indent* (+ *indent* 2))
215         (type (case type
216                 ('field "Field")
217                 ('method "Method")))
218         name)
219    (out "~A ~D: Access flags: #x~X~%" type index access-flags)
220    (out "~A ~D: Name index: ~D (~S)~%" type index name-index (lookup-utf8 name-index))
221    (out "~A ~D: Descriptor index: ~D~%" type index descriptor-index)
222    (out "~A ~D: Number of attributes: ~D~%" type index attributes-count)
223    (let ((*indent* (+ *indent* 2)))
224      (dotimes (i attributes-count)
225        (read-attribute i stream)))))
226
227(defun dump-class (filename)
228  (let ((*indent* 0)
229        (*pool* nil))
230    (with-open-file (stream filename :direction :input :element-type 'unsigned-byte)
231      (handler-bind ((end-of-file
232                      #'(lambda (c) (return-from dump-class c))))
233        (out "Magic number: #x~X~%" (read-u4 stream))
234        (let ((minor (read-u2 stream))
235              (major (read-u2 stream)))
236          (out "Version: ~D.~D~%" major minor))
237        ;; Constant pool.
238        (let ((count (read-u2 stream))
239              entry type)
240          (out "Constant pool (~D entries):~%" count)
241          (setq *pool* (make-array count))
242          (let ((*indent* (+ *indent* 2)))
243            (dotimes (index (1- count))
244              (setq entry (read-constant-pool-entry stream))
245              (setf (svref *pool* (1+ index)) entry)
246              (setq type (case (car entry)
247                           (7 'class)
248                           (9 'field)
249                           (10 'method)
250                           (11 'interface)
251                           (8 'string)
252                           (3 'integer)
253                           (4 'float)
254                           (5 'long)
255                           (6 'double)
256                           (12 'name-and-type)
257                           (1 'utf8)))
258              (out "~D: ~A ~S~%" (1+ index) type entry))))
259        (out "Access flags: #x~X~%" (read-u2 stream))
260        (out "This class: ~D~%" (read-u2 stream))
261        (out "Superclass: ~D~%" (read-u2 stream))
262        ;; Interfaces.
263        (let ((count (read-u2 stream)))
264          (cond ((zerop count)
265                 (out "No interfaces~%"))
266                (t
267                 (out "Interfaces (~D):~%" count)
268                 (dotimes (i count)
269                   (out "  ~D: ~D~%" i (read-u2 stream))))))
270        ;; Fields.
271        (let ((count (read-u2 stream)))
272          (cond ((zerop count)
273                 (out "No fields~%"))
274                (t
275                 (out "Fields (~D):~%" count)))
276          (dotimes (index count)
277            (read-info index stream 'field)))
278        ;; Methods.
279        (let ((count (read-u2 stream)))
280          (cond ((zerop count)
281                 (out "No methods~%"))
282                (t
283                 (out "Methods (~D):~%" count)))
284          (dotimes (index count)
285            (read-info index stream 'method)))
286        (let ((count (read-u2 stream)))
287          (cond ((zerop count)
288                 (out "No attributes~%"))
289                (t
290                 (out "Attributes (~D):~%" count)))))))
291  t)
292
293(defvar *stream* nil)
294(defvar *defun-name* nil)
295(defvar *this-class* nil)
296(defvar *pool-count* 1)
297
298(defvar *code* ())
299(defvar *static-code* ())
300(defvar *fields* ())
301
302(defvar *blocks* ())
303(defvar *locals* ())
304(defvar *max-locals* 0)
305
306;; (defun allocate-local ()
307;;   (let ((index (fill-pointer *locals*)))
308;;     (incf (fill-pointer *locals*))
309;;     (setf *max-locals* (fill-pointer *locals*))
310;;     index))
311
312(defvar *args* nil)
313(defvar *using-arg-array* nil)
314(defvar *hairy-arglist-p* nil)
315
316(defvar *val* nil) ; index of value register
317
318(defun clear ()
319  (setq *pool* nil
320        *pool-count* 1
321        *code* nil)
322  t)
323
324(defun dump-pool ()
325  (let ((pool (reverse *pool*))
326        entry)
327    (dotimes (index (1- *pool-count*))
328      (setq entry (car pool))
329      (setq type (case (car entry)
330                   (7 'class)
331                   (9 'field)
332                   (10 'method)
333                   (11 'interface)
334                   (8 'string)
335                   (3 'integer)
336                   (4 'float)
337                   (5 'long)
338                   (6 'double)
339                   (12 'name-and-type)
340                   (1 'utf8)))
341      (format t "~D: ~A ~S~%" (1+ index) type entry)
342      (setq pool (cdr pool))))
343  t)
344
345;; Returns index of entry (1-based).
346(defun pool-add (entry)
347  (setq *pool* (cons entry *pool*))
348  (prog1
349    *pool-count*
350    (incf *pool-count*)))
351
352;; Returns index of entry (1-based).
353(defun pool-find-entry (entry)
354  (do* ((remaining *pool* (cdr remaining))
355        (i 0 (1+ i))
356        (current (car remaining) (car remaining)))
357       ((null remaining) nil)
358    (when (equal current entry)
359      (return-from pool-find-entry (- *pool-count* 1 i)))))
360
361;; Adds entry if not already in pool. Returns index of entry (1-based).
362(defun pool-get (entry)
363  (or (pool-find-entry entry) (pool-add entry)))
364
365(defun pool-name (name)
366  (pool-get (list 1 (length name) name)))
367
368;; "org.armedbear.lisp.LispObject" => "Lorg/armedbear/lisp/LispObject;"
369;; (defun type-descriptor (type)
370;;   (unless (find #\. type)
371;;     (setq type (concatenate 'string "org.armedbear.lisp." type)))
372;;   (let ((res (concatenate 'string "L" type ";")))
373;;     (dotimes (i (length res))
374;;       (when (eql (char res i) #\.)
375;;         (setf (char res i) #\/)))
376;;     res))
377
378(defun pool-name-and-type (name type)
379  (let* ((name-index (pool-name name))
380         (type-index (pool-name type)))
381    (pool-get (list 12 name-index type-index))))
382
383(defun pool-class (class-name)
384  (let ((class-name class-name))
385    (dotimes (i (length class-name))
386      (when (eql (char class-name i) #\.)
387        (setf (char class-name i) #\/)))
388    (pool-get (list 7 (pool-name class-name)))))
389
390;; (tag class-index name-and-type-index)
391(defun pool-field (class-name field-name type-name)
392  (let* ((class-index (pool-class class-name))
393         (name-and-type-index (pool-name-and-type field-name type-name)))
394    (pool-get (list 9 class-index name-and-type-index))))
395
396;; (tag class-index name-and-type-index)
397(defun pool-method (class-name method-name type-name)
398  (let* ((class-index (pool-class class-name))
399         (name-and-type-index (pool-name-and-type method-name type-name)))
400    (pool-get (list 10 class-index name-and-type-index))))
401
402(defun pool-string (string)
403  (pool-get (list 8 (pool-name string))))
404
405(defun u2 (n)
406  (list (ash n -8) (logand n #xff)))
407
408(defstruct instruction opcode args stack)
409
410(defun inst (opcode &optional args)
411  (unless (listp args)
412    (setq args (list args)))
413  (make-instruction :opcode opcode :args args :stack 0))
414
415(defun emit (instr &rest args)
416  (unless (numberp instr)
417    (setq instr (get instr 'opcode)))
418  (setq *code* (cons (inst instr args) *code*)))
419
420(defmacro emit-store-value ()
421  `(case *val*
422     (0
423      (emit 'astore_0))
424     (1
425      (emit 'astore_1))
426     (2
427      (emit 'astore_2))
428     (3
429      (emit 'astore_3))
430     (t
431      (emit 'astore *val*))))
432
433(defmacro emit-push-value ()
434  `(case *val*
435     (0
436      (emit 'aload_0))
437     (1
438      (emit 'aload_1))
439     (2
440      (emit 'aload_2))
441     (3
442      (emit 'aload_3))
443     (t
444      (emit 'aload *val*))))
445
446(defun remove-store-value ()
447  (let* ((instruction (car *code*))
448         (opcode (instruction-opcode instruction))
449         slot)
450    (case opcode
451      (75
452       (setf slot 0))
453      (76
454       (setf slot 1))
455      (77
456       (setf slot 2))
457      (78
458       (setf slot 3))
459      (58
460       (setf slot (car (instruction-args instruction)))))
461    (when (and slot (= slot *val*))
462      (setf *code* (cdr *code*))
463      t)))
464
465(defconstant +lisp-class+ "org/armedbear/lisp/Lisp")
466
467(defconstant +lisp-object-class+ "org/armedbear/lisp/LispObject")
468(defconstant +lisp-object+ "Lorg/armedbear/lisp/LispObject;")
469
470(defconstant +lisp-string+ "Lorg/armedbear/lisp/LispString;")
471
472(defconstant +lisp-thread-class+ "org/armedbear/lisp/LispThread")
473
474(defun emit-push-nil ()
475  (emit 'getstatic
476        +lisp-class+
477        "NIL"
478        +lisp-object+))
479
480(defun emit-push-t ()
481  (emit 'getstatic
482        +lisp-class+
483        "T"
484        "Lorg/armedbear/lisp/Symbol;"))
485
486;; Index of local variable used to hold the current thread.
487(defvar *thread* nil)
488(defvar *thread-var-initialized* nil)
489
490(defun ensure-thread-var-initialized ()
491  (unless *thread-var-initialized*
492    (format t "generating thread variable initialization code~%")
493    ;; Put the code to initialize the local at the very beginning of the
494    ;; function, to guarantee that the local gets initialized even if the code
495    ;; at our current location is never executed, since the local may be
496    ;; referenced elsewhere too.
497    (let ((code *code*))
498      (setf *code* ())
499      (emit 'invokestatic
500            +lisp-thread-class+
501            "currentThread"
502            "()Lorg/armedbear/lisp/LispThread;")
503      (emit 'astore *thread*)
504      (setf *code* (append code *code*)))
505    (setf *thread-var-initialized* t)))
506
507(defun emit-clear-values ()
508  (ensure-thread-var-initialized)
509  (emit 'aload *thread*)
510  (emit 'invokevirtual
511        +lisp-thread-class+
512        "clearValues"
513        "()V"))
514
515(defun emit-invoke-method (method-name)
516  (unless (remove-store-value)
517    (emit-push-value))
518  (emit 'invokevirtual
519        +lisp-object-class+
520        method-name
521        "()Lorg/armedbear/lisp/LispObject;")
522  (emit-store-value))
523
524;; CODE is a list.
525(defun resolve-args (instruction)
526  (let ((opcode (instruction-opcode instruction))
527        (args (instruction-args instruction)))
528    (case opcode
529      ((1 ; ACONST_NULL
530        42 ; ALOAD_0
531        43 ; ALOAD_1
532        44 ; ALOAD_2
533        45 ; ALOAD_3
534        50 ; AALOAD
535        75 ; ASTORE_0
536        76 ; ASTORE_1
537        77 ; ASTORE_2
538        78 ; ASTORE_3
539        83 ; AASTORE
540        87 ; POP
541        89 ; DUP
542        95 ; SWAP
543        153 ; IFEQ
544        154 ; IFNE
545        166 ; IF_ACMPNE
546        165 ; IF_ACMPEQ
547        167 ; GOTO
548        176 ; ARETURN
549        177 ; RETURN
550        202 ; LABEL
551        )
552       instruction)
553      (25 ; ALOAD
554       (let ((index (car args)))
555         (cond ((= index 0)
556                (inst 42)) ; ALOAD_O
557               ((= index 1)
558                (inst 43)) ; ALOAD_1
559               ((= index 2)
560                (inst 44)) ; ALOAD_2
561               ((= index 3)
562                (inst 45)) ; ALOAD_3
563               ((<= 0 index 255)
564                (inst 25 index))
565               (t
566                (error "ALOAD unsupported case")))))
567      (58 ; ASTORE
568       (let ((index (car args)))
569         (cond ((= index 0)
570                (inst 75)) ; ASTORE_O
571               ((= index 1)
572                (inst 76)) ; ASTORE_1
573               ((= index 2)
574                (inst 77)) ; ASTORE_2
575               ((= index 3)
576                (inst 78)) ; ASTORE_3
577               ((<= 0 index 255)
578                (inst 58 index))
579               (t
580                (error "ASTORE unsupported case")))))
581      ((178 ; GETSTATIC class-name field-name type-name
582        179 ; PUTSTATIC class-name field-name type-name
583        )
584       (let ((index (pool-field (first args) (second args) (third args))))
585         (inst opcode (u2 index))))
586      ((182 ; INVOKEVIRTUAL class-name method-name descriptor
587        183 ; INVOKESPECIAL class-name method-name descriptor
588        184 ; INVOKESTATIC class-name method-name descriptor
589        )
590       (let ((index (pool-method (first args) (second args) (third args))))
591         (inst opcode (u2 index))))
592      ((189 ; ANEWARRAY class-name
593        )
594       (let ((index (pool-class (first args))))
595         (inst opcode (u2 index))))
596      ((16 ; BIPUSH
597        17 ; SIPUSH
598        )
599       (let ((n (first args)))
600         (cond ((= n 0)
601                (inst 3)) ; ICONST_0
602               ((= n 1)
603                (inst 4)) ; ICONST_1
604               ((= n 2)
605                (inst 5)) ; ICONST_2
606               ((= n 3)
607                (inst 6)) ; ICONST_3
608               ((= n 4)
609                (inst 7)) ; ICONST_4
610               ((= n 5)
611                (inst 8)) ; ICONST_5
612               ((<= -128 n 127)
613                (inst 16 (logand n #xff))) ; BIPUSH
614               (t ; SIPUSH
615                (inst 17 (u2 n))))))
616      (18 ; LDC
617       (unless (= (length args) 1)
618         (error "wrong number of args for LDC"))
619       (if (> (car args) 255)
620           (inst 19 (u2 (car args))) ; LDC_W
621           (inst opcode args)))
622      (t
623       (error "RESOLVE-ARGS unsupported opcode ~D" opcode)))))
624
625;; CODE is a list of INSTRUCTIONs.
626(defun resolve-opcodes (code)
627  (map 'vector #'resolve-args code))
628
629;; CODE is a list of INSTRUCTIONs.
630(defun code-bytes (code)
631  (setf code (coerce code 'vector))
632  (setf code (nreverse code))
633
634  (dotimes (i (length code))
635    (let ((instruction (svref code i)))
636      (when (and (< i (1- (length code)))
637                 (= (instruction-opcode instruction) 167) ; GOTO
638        (let ((next-instruction (svref code (1+ i))))
639          (when (and (= (instruction-opcode next-instruction) 202) ; LABEL
640                     (eq (car (instruction-args instruction))
641                         (car (instruction-args next-instruction))))
642            (setf (instruction-opcode instruction) 'nop)))))))
643
644  (setf code (delete 'nop code :key #'instruction-opcode))
645;;   (setf code (coerce code 'list))
646
647  (let ((code (resolve-opcodes code))
648        (length 0))
649    ;; Pass 1: calculate label offsets and overall length.
650    (dotimes (i (length code))
651      (let* ((instruction (aref code i))
652             (opcode (instruction-opcode instruction)))
653        (if (= opcode 202) ; LABEL
654            (let ((label (car (instruction-args instruction))))
655              (set label length))
656            (incf length (opcode-size opcode)))))
657    ;; Pass 2: replace labels with calculated offsets.
658    (let ((index 0))
659      (dotimes (i (length code))
660        (let ((instruction (aref code i)))
661          (case (instruction-opcode instruction)
662            ((153 ; IFEQ
663              154 ; IFNE
664              166 ; IF_ACMPNE
665              165 ; IF_ACMPEQ
666              167 ; GOTO
667              )
668             (let* ((label (car (instruction-args instruction)))
669                    (offset (- (symbol-value `,label) index)))
670               (setf (instruction-args instruction) (u2 offset)))))
671          (unless (= (instruction-opcode instruction) 202) ; LABEL
672            (incf index (opcode-size (instruction-opcode instruction)))))))
673    ;; FIXME Do stack analysis here!
674    ;; Convert list to vector.
675    (let ((vector (make-array length))
676          (index 0))
677      (dotimes (i (length code))
678        (let ((instruction (aref code i)))
679          (unless (= (instruction-opcode instruction) 202) ; LABEL
680            (setf (svref vector index) (instruction-opcode instruction))
681            (incf index)
682            (dolist (byte (instruction-args instruction))
683              (setf (svref vector index) byte)
684              (incf index)))))
685      vector)))
686
687(defun write-u1 (n)
688  (write-byte (logand n #xFF) *stream*))
689
690(defun write-u2 (n)
691  (write-byte (ash n -8) *stream*)
692  (write-byte (logand n #xFF) *stream*))
693
694(defun write-u4 (n)
695  (write-u2 (ash n -16))
696  (write-u2 (logand n #xFFFF)))
697
698(defun write-utf8 (string)
699  (dotimes (i (length string))
700    (write-u1 (char-int (char string i)))))
701
702(defun write-cp-entry (entry)
703  (write-u1 (first entry))
704  (case (first entry)
705    (1
706     (write-u2 (second entry))
707     (write-utf8 (third entry)))
708    ((5 6)
709     (write-u4 (second entry))
710     (write-u4 (third entry)))
711    ((9 10 11 12)
712     (write-u2 (second entry))
713     (write-u2 (third entry)))
714    ((7 8)
715     (write-u2 (second entry)))
716    (t
717     (error "WRITE-CP-ENTRY unhandled tag ~D~%" (car entry)))
718  ))
719
720(defun write-pool ()
721  (write-u2 *pool-count*)
722  (dolist (entry (reverse *pool*))
723    (write-cp-entry entry)))
724
725(defstruct field
726  access-flags
727  name
728  descriptor
729  name-index
730  descriptor-index)
731
732(defstruct method
733  access-flags
734  name
735  descriptor
736  name-index
737  descriptor-index
738  max-stack
739  max-locals
740  code)
741
742(defun make-constructor (super name args body)
743  (let* ((constructor (make-method :name "<init>"
744                                   :descriptor "()V"))
745         (*code* ()))
746    (setf (method-name-index constructor) (pool-name (method-name constructor)))
747    (setf (method-descriptor-index constructor) (pool-name (method-descriptor constructor)))
748    (setf (method-max-locals constructor) 1)
749    (cond (*hairy-arglist-p*
750           (emit 'aload_0) ;; this
751           (emit 'aconst_null) ;; name
752           (let ((s (format nil "~S" args)))
753             (emit 'ldc
754                   (pool-string s))
755             (emit 'invokestatic
756                   "org/armedbear/lisp/Lisp"
757                   "readObjectFromString"
758                   "(Ljava/lang/String;)Lorg/armedbear/lisp/LispObject;"))
759           (emit-push-nil) ;; body
760           (emit 'aconst_null) ;; environment
761           (emit 'invokespecial
762                 super
763                 "<init>"
764                 "(Ljava/lang/String;Lorg/armedbear/lisp/LispObject;Lorg/armedbear/lisp/LispObject;Lorg/armedbear/lisp/Environment;)V"))
765          (t
766           (emit 'aload_0)
767           (emit 'invokespecial
768                 super
769                 "<init>"
770                 "()V")))
771    (setq *code* (append *static-code* *code*))
772    (emit 'return)
773    (setf (method-code constructor) (code-bytes *code*))
774    constructor))
775
776(defun write-code-attr (method)
777  (let* ((name-index (pool-name "Code"))
778         (code (method-code method))
779         (code-length (length code))
780         (length (+ code-length 12))
781         (max-stack (or (method-max-stack method) 20))
782         (max-locals (or (method-max-locals method) 1)))
783    (write-u2 name-index)
784    (write-u4 length)
785    (write-u2 max-stack)
786    (write-u2 max-locals)
787    (write-u4 code-length)
788    (dotimes (i code-length)
789      (write-u1 (svref code i)))
790    (write-u2 0) ; exception table length
791    (write-u2 0) ; attributes count
792    ))
793
794(defun write-method (method)
795  (write-u2 (or (method-access-flags method) #x1)) ; access flags
796  (write-u2 (method-name-index method))
797  (write-u2 (method-descriptor-index method))
798  (write-u2 1) ; attributes count
799  (write-code-attr method))
800
801(defun write-field (field)
802  (write-u2 (or (field-access-flags field) #x1)) ; access flags
803  (write-u2 (field-name-index field))
804  (write-u2 (field-descriptor-index field))
805  (write-u2 0)) ; attributes count
806
807(defun declare-field (name descriptor)
808  (let ((field (make-field :name name :descriptor descriptor)))
809    (setf (field-access-flags field) (logior #x8 #x2)) ; private static
810    (setf (field-name-index field) (pool-name (field-name field)))
811    (setf (field-descriptor-index field) (pool-name (field-descriptor field)))
812    (setq *fields* (cons field *fields*))))
813
814(defun sanitize (symbol)
815  (let* ((input (symbol-name symbol))
816         (output (make-array (length input) :fill-pointer 0 :element-type 'character)))
817    (dotimes (i (length input))
818      (let ((c (char-upcase (char input i))))
819        (cond ((<= #.(char-code #\A) (char-code c) #.(char-code #\Z))
820               (vector-push c output))
821              ((eql c #\-)
822               (vector-push #\_ output)))))
823    (when (plusp (length output))
824      output)))
825
826(defvar *declared-symbols* ())
827(defvar *declared-functions* ())
828
829(defun declare-symbol (symbol)
830  (let ((g (gethash symbol *declared-symbols*)))
831    (unless g
832      (let ((*code* *static-code*)
833            (s (sanitize symbol)))
834        (setq g (symbol-name (gensym)))
835        (when s
836          (setq g (concatenate 'string g "_" s)))
837        (declare-field g "Lorg/armedbear/lisp/Symbol;")
838        (emit 'ldc
839              (pool-string (symbol-name symbol)))
840        (emit 'ldc
841              (pool-string (package-name (symbol-package symbol))))
842        (emit 'invokestatic
843              +lisp-class+
844              "internInPackage"
845              "(Ljava/lang/String;Ljava/lang/String;)Lorg/armedbear/lisp/Symbol;")
846        (emit 'putstatic
847              *this-class*
848              g
849              "Lorg/armedbear/lisp/Symbol;")
850        (setq *static-code* *code*)
851        (setf (gethash symbol *declared-symbols*) g)))
852    g))
853
854(defun declare-function (symbol)
855  (let ((f (gethash symbol *declared-functions*)))
856    (unless f
857      (setf f (symbol-name (gensym)))
858      (let ((s (sanitize symbol)))
859        (when s
860          (setf f (concatenate 'string f "_" s))))
861      (let ((*code* *static-code*)
862            (g (gethash symbol *declared-symbols*)))
863        (cond (g
864               (emit 'getstatic
865                     *this-class*
866                     g
867                     "Lorg/armedbear/lisp/Symbol;"))
868              (t
869               (emit 'ldc
870                     (pool-string (symbol-name symbol)))
871               (emit 'ldc
872                     (pool-string (package-name (symbol-package symbol))))
873               (emit 'invokestatic
874                     +lisp-class+
875                     "internInPackage"
876                     "(Ljava/lang/String;Ljava/lang/String;)Lorg/armedbear/lisp/Symbol;")))
877        (declare-field f "Lorg/armedbear/lisp/LispObject;")
878        (emit 'invokevirtual
879              "org/armedbear/lisp/Symbol"
880              "getSymbolFunctionOrDie"
881              "()Lorg/armedbear/lisp/LispObject;")
882        (emit 'putstatic
883              *this-class*
884              f
885              "Lorg/armedbear/lisp/LispObject;")
886        (setq *static-code* *code*)
887        (setf (gethash symbol *declared-functions*) f)))
888    f))
889
890(defun declare-keyword (symbol)
891  (let ((g (symbol-name (gensym)))
892        (*code* *static-code*))
893    (declare-field g "Lorg/armedbear/lisp/Symbol;")
894    (emit 'ldc
895          (pool-string (symbol-name symbol)))
896    (emit 'invokestatic
897          "org/armedbear/lisp/Keyword"
898          "internKeyword"
899          "(Ljava/lang/String;)Lorg/armedbear/lisp/Symbol;")
900    (emit 'putstatic
901          *this-class*
902          g
903          "Lorg/armedbear/lisp/Symbol;")
904    (setq *static-code* *code*)
905    g))
906
907(defun declare-object-as-string (obj)
908  (let ((g (symbol-name (gensym)))
909        (s (format nil "~S" obj))
910        (*code* *static-code*))
911    (declare-field g +lisp-object+)
912    (emit 'ldc
913          (pool-string s))
914    (emit 'invokestatic
915          "org/armedbear/lisp/Lisp"
916          "readObjectFromString"
917          "(Ljava/lang/String;)Lorg/armedbear/lisp/LispObject;")
918    (emit 'putstatic
919          *this-class*
920          g
921          +lisp-object+)
922    (setq *static-code* *code*)
923    g))
924
925(defun declare-object (obj)
926  (let ((key (symbol-name (gensym))))
927    (sys::remember key obj)
928    (let* ((g1 (declare-string key))
929           (g2 (symbol-name (gensym)))
930           (*code* *static-code*))
931      (declare-field g2 +lisp-object+)
932      (emit 'getstatic
933            *this-class*
934            g1
935            +lisp-string+)
936      (emit 'dup)
937      (emit 'invokestatic
938            +lisp-class+
939            "recall"
940            "(Lorg/armedbear/lisp/LispString;)Lorg/armedbear/lisp/LispObject;")
941      (emit 'putstatic
942            *this-class*
943            g2
944            +lisp-object+)
945      (emit 'invokestatic
946            +lisp-class+
947            "forget"
948            "(Lorg/armedbear/lisp/LispString;)V")
949      (setq *static-code* *code*)
950      g2)))
951
952(defun declare-string (string)
953  (let ((g (symbol-name (gensym)))
954        (*code* *static-code*))
955    (declare-field g "Lorg/armedbear/lisp/LispString;")
956    (emit 'ldc
957          (pool-string string))
958    (emit 'invokestatic
959          "org/armedbear/lisp/LispString"
960          "getInstance"
961          "(Ljava/lang/String;)Lorg/armedbear/lisp/LispString;")
962    (emit 'putstatic
963          *this-class*
964          g
965          +lisp-string+)
966    (setq *static-code* *code*)
967    g))
968
969(defun compile-constant (form)
970  (cond
971   ((sys::fixnump form)
972    (let ((n form))
973      (cond ((zerop n)
974             (emit 'getstatic
975                   "org/armedbear/lisp/Fixnum"
976                   "ZERO"
977                   "Lorg/armedbear/lisp/Fixnum;")
978             (emit-store-value))
979            ((= n 1)
980             (emit 'getstatic
981                   "org/armedbear/lisp/Fixnum"
982                   "ONE"
983                   "Lorg/armedbear/lisp/Fixnum;")
984             (emit-store-value))
985            ((= n 2)
986             (emit 'getstatic
987                   "org/armedbear/lisp/Fixnum"
988                   "TWO"
989                   "Lorg/armedbear/lisp/Fixnum;")
990             (emit-store-value))
991            (t
992             (let ((g (declare-object-as-string n)))
993               (emit 'getstatic
994                     *this-class*
995                     g
996                     "Lorg/armedbear/lisp/LispObject;")
997               (emit-store-value))))))
998   ((numberp form)
999    (let ((g (declare-object-as-string form)))
1000      (emit 'getstatic
1001            *this-class*
1002            g
1003            "Lorg/armedbear/lisp/LispObject;")
1004      (emit-store-value)))
1005   ((vectorp form)
1006    (let ((g (declare-object-as-string form)))
1007      (emit 'getstatic
1008            *this-class*
1009            g
1010            "Lorg/armedbear/lisp/LispObject;")
1011      (emit-store-value)))
1012   ((stringp form)
1013    (let ((g (declare-string form)))
1014      (emit 'getstatic
1015            *this-class*
1016            g
1017            "Lorg/armedbear/lisp/LispString;")
1018      (emit-store-value)))
1019   ((characterp form)
1020    (let ((g (declare-object-as-string form)))
1021      (emit 'getstatic
1022            *this-class*
1023            g
1024            "Lorg/armedbear/lisp/LispObject;")
1025      (emit-store-value)))
1026   ((symbolp form)
1027    (when (null (symbol-package form))
1028      ;; An uninterned symbol.
1029      (let ((g (declare-object form)))
1030        (emit 'getstatic
1031              *this-class*
1032              g
1033              "Lorg/armedbear/lisp/LispObject;")
1034        (emit-store-value))))
1035   (t
1036    (error "COMPILE-CONSTANT unhandled case ~S" form))))
1037
1038(defun compile-binary-operation (op args)
1039  (compile-form (first args))
1040  (unless (remove-store-value)
1041    (emit-push-value))
1042  (compile-form (second args))
1043  (unless (remove-store-value)
1044    (emit-push-value))
1045  (emit 'invokevirtual
1046        "org/armedbear/lisp/LispObject"
1047        op
1048        "(Lorg/armedbear/lisp/LispObject;)Lorg/armedbear/lisp/LispObject;")
1049  (emit-store-value))
1050
1051(defparameter unary-operators (make-hash-table))
1052
1053(setf (gethash '1+              unary-operators) "incr")
1054(setf (gethash '1-              unary-operators) "decr")
1055(setf (gethash 'ATOM            unary-operators) "ATOM")
1056(setf (gethash 'BIT-VECTOR-P    unary-operators) "BIT_VECTOR_P")
1057(setf (gethash 'CADR            unary-operators) "cadr")
1058(setf (gethash 'CAR             unary-operators) "car")
1059(setf (gethash 'CDDR            unary-operators) "cddr")
1060(setf (gethash 'CDR             unary-operators) "cdr")
1061(setf (gethash 'COMPLEXP        unary-operators) "COMPLEXP")
1062(setf (gethash 'CONSTANTP       unary-operators) "CONSTANTP")
1063(setf (gethash 'DENOMINATOR     unary-operators) "DENOMINATOR")
1064(setf (gethash 'ENDP            unary-operators) "ENDP")
1065(setf (gethash 'EVENP           unary-operators) "EVENP")
1066(setf (gethash 'FIRST           unary-operators) "car")
1067(setf (gethash 'FLOATP          unary-operators) "FLOATP")
1068(setf (gethash 'INTEGERP        unary-operators) "INTEGERP")
1069(setf (gethash 'LENGTH          unary-operators) "LENGTH")
1070(setf (gethash 'LISTP           unary-operators) "LISTP")
1071(setf (gethash 'MINUSP          unary-operators) "MINUSP")
1072(setf (gethash 'NOT             unary-operators) "NOT")
1073(setf (gethash 'NULL            unary-operators) "NOT")
1074(setf (gethash 'NUMBERP         unary-operators) "NUMBERP")
1075(setf (gethash 'NUMERATOR       unary-operators) "NUMERATOR")
1076(setf (gethash 'ODDP            unary-operators) "ODDP")
1077(setf (gethash 'PLUSP           unary-operators) "PLUSP")
1078(setf (gethash 'RATIONALP       unary-operators) "RATIONALP")
1079(setf (gethash 'REALP           unary-operators) "REALP")
1080(setf (gethash 'REST            unary-operators) "cdr")
1081(setf (gethash 'SECOND          unary-operators) "cadr")
1082(setf (gethash 'SIMPLE-STRING-P unary-operators) "SIMPLE_STRING_P")
1083(setf (gethash 'STRINGP         unary-operators) "STRINGP")
1084(setf (gethash 'SYMBOLP         unary-operators) "SYMBOLP")
1085(setf (gethash 'VECTORP         unary-operators) "VECTORP")
1086(setf (gethash 'ZEROP           unary-operators) "ZEROP")
1087
1088
1089(defun compile-function-call-1 (fun args)
1090  (let ((s (gethash fun unary-operators)))
1091    (when s
1092      (compile-form (first args))
1093      (emit-invoke-method s)
1094      (return-from compile-function-call-1 t)))
1095    nil)
1096
1097(defun compile-function-call-2 (fun args)
1098  (case fun
1099    (EQ
1100     (compile-form (first args))
1101     (unless (remove-store-value)
1102       (emit-push-value))
1103     (compile-form (second args))
1104     (unless (remove-store-value)
1105       (emit-push-value))
1106     (let ((label1 (gensym))
1107           (label2 (gensym)))
1108       (emit 'if_acmpeq `,label1)
1109       (emit-push-nil)
1110       (emit 'goto `,label2)
1111       (emit 'label `,label1)
1112       (emit-push-t)
1113       (emit 'label `,label2))
1114     (emit-store-value)
1115     t)
1116    (EQL
1117     (compile-binary-operation "EQL" args)
1118     t)
1119    (+
1120     (compile-binary-operation "add" args)
1121     t)
1122    (-
1123     (compile-binary-operation "subtract" args)
1124     t)
1125    (/
1126     (compile-binary-operation "divideBy" args)
1127     t)
1128    (*
1129     (compile-binary-operation "multiplyBy" args)
1130     t)
1131    (<
1132     (compile-binary-operation "IS_LT" args)
1133     t)
1134    (<=
1135     (compile-binary-operation "IS_LE" args)
1136     t)
1137    (>
1138     (compile-binary-operation "IS_GT" args)
1139     t)
1140    (>=
1141     (compile-binary-operation "IS_GE" args)
1142     t)
1143    (=
1144     (compile-binary-operation "IS_E" args)
1145     t)
1146    (/=
1147     (compile-binary-operation "IS_NE" args)
1148     t)
1149    (AREF
1150     (compile-binary-operation "AREF" args)
1151     t)
1152    (LIST
1153     (compile-form (first args))
1154     (unless (remove-store-value)
1155       (emit-push-value))
1156     (compile-form (second args))
1157     (unless (remove-store-value)
1158       (emit-push-value))
1159     (emit 'invokestatic
1160           +lisp-class+
1161           "list2"
1162           "(Lorg/armedbear/lisp/LispObject;Lorg/armedbear/lisp/LispObject;)Lorg/armedbear/lisp/Cons;")
1163     (emit-store-value)
1164     t)
1165    (SYS::SIMPLE-TYPEP
1166     (compile-binary-operation "typep" args))
1167    (t
1168     nil)))
1169
1170(defun compile-function-call-3 (fun args)
1171  (case fun
1172    (LIST
1173     (compile-form (first args))
1174     (unless (remove-store-value)
1175       (emit-push-value))
1176     (compile-form (second args))
1177     (unless (remove-store-value)
1178       (emit-push-value))
1179     (compile-form (third args))
1180     (unless (remove-store-value)
1181       (emit-push-value))
1182     (emit 'invokestatic
1183           +lisp-class+
1184           "list3"
1185           "(Lorg/armedbear/lisp/LispObject;Lorg/armedbear/lisp/LispObject;Lorg/armedbear/lisp/LispObject;)Lorg/armedbear/lisp/Cons;")
1186     (emit-store-value)
1187     t)
1188    (t
1189     nil)))
1190
1191(defconstant +cl-package+ (find-package "COMMON-LISP"))
1192(defconstant +sys-package+ (find-package "SYSTEM"))
1193
1194(defun compile-function-call (fun args &optional for-effect)
1195;;   (format t "compile-function-call fun = ~S args = ~S~%" fun args)
1196  (unless (symbolp fun)
1197    (error "COMPILE-FUNCTION-CALL ~S is not a symbol" fun))
1198  (let ((numargs (length args)))
1199    (cond ((= numargs 1)
1200           (when (compile-function-call-1 fun args)
1201             (return-from compile-function-call)))
1202          ((= numargs 2)
1203           (when (compile-function-call-2 fun args)
1204             (return-from compile-function-call)))
1205          ((= numargs 3)
1206           (when (compile-function-call-3 fun args)
1207             (return-from compile-function-call))))
1208
1209    ;; FIXME This shouldn't go here! Do this in the constructor of the
1210    ;; compiled function!
1211    (sys::resolve fun)
1212
1213    (cond
1214     ((eq fun *defun-name*)
1215      (emit 'aload 0)) ; this
1216     ((memq (symbol-package fun) '(+cl-package+ +sys-package+))
1217      (let ((f (declare-function fun)))
1218        (emit 'getstatic
1219              *this-class*
1220              f
1221              "Lorg/armedbear/lisp/LispObject;")))
1222     (t
1223      (let ((g (declare-symbol fun)))
1224        (emit 'getstatic
1225              *this-class*
1226              g
1227              "Lorg/armedbear/lisp/Symbol;"))
1228      (emit 'invokevirtual
1229            "org/armedbear/lisp/Symbol"
1230            "getSymbolFunctionOrDie"
1231            "()Lorg/armedbear/lisp/LispObject;")))
1232    (case numargs
1233      (0
1234       (emit 'invokevirtual
1235             "org/armedbear/lisp/LispObject"
1236             "execute"
1237             "()Lorg/armedbear/lisp/LispObject;"))
1238      (1
1239       (compile-form (first args))
1240       (unless (remove-store-value)
1241         (emit-push-value))
1242       (emit 'invokevirtual
1243             "org/armedbear/lisp/LispObject"
1244             "execute"
1245             "(Lorg/armedbear/lisp/LispObject;)Lorg/armedbear/lisp/LispObject;"))
1246      (2
1247       (compile-form (first args))
1248       (unless (remove-store-value)
1249         (emit-push-value))
1250       (compile-form (second args))
1251       (unless (remove-store-value)
1252         (emit-push-value))
1253       (emit 'invokevirtual
1254             "org/armedbear/lisp/LispObject"
1255             "execute"
1256             "(Lorg/armedbear/lisp/LispObject;Lorg/armedbear/lisp/LispObject;)Lorg/armedbear/lisp/LispObject;"))
1257      (3
1258       (compile-form (first args))
1259       (unless (remove-store-value)
1260         (emit-push-value))
1261       (compile-form (second args))
1262       (unless (remove-store-value)
1263         (emit-push-value))
1264       (compile-form (third args))
1265       (unless (remove-store-value)
1266         (emit-push-value))
1267       (emit 'invokevirtual
1268             "org/armedbear/lisp/LispObject"
1269             "execute"
1270             "(Lorg/armedbear/lisp/LispObject;Lorg/armedbear/lisp/LispObject;Lorg/armedbear/lisp/LispObject;)Lorg/armedbear/lisp/LispObject;"))
1271      (t
1272       (emit 'sipush (length args))
1273       (emit 'anewarray "org/armedbear/lisp/LispObject")
1274       (let ((i 0))
1275         (dolist (form args)
1276           (emit 'dup)
1277           (emit 'sipush i)
1278           (compile-form form)
1279           (unless (remove-store-value)
1280             (emit-push-value)) ; leaves value on stack
1281           (emit 'aastore) ; store value in array
1282           (incf i))) ; array left on stack here
1283       ;; Stack: function array-ref
1284       (emit 'invokevirtual
1285             "org/armedbear/lisp/LispObject"
1286             "execute"
1287             "([Lorg/armedbear/lisp/LispObject;)Lorg/armedbear/lisp/LispObject;")))
1288    (if for-effect
1289        (emit 'pop)
1290        (emit-store-value))))
1291
1292(defun compile-test (form)
1293  ;; Use a Java boolean if possible.
1294  (when (consp form)
1295    (case (length form)
1296      (2 (when (memq (car form) '(NOT NULL))
1297           (compile-form (second form))
1298           (unless (remove-store-value)
1299             (emit-push-value))
1300           (emit-push-nil)
1301           (return-from compile-test 'if_acmpne))
1302         (let ((s (cdr (assoc (car form)
1303                              '((ATOM      . "atom")
1304                                (EVENP     . "evenp")
1305                                (FLOATP    . "floatp")
1306                                (INTEGERP  . "integerp")
1307                                (MINUSP    . "minusp")
1308                                (LISTP     . "listp")
1309                                (NUMBERP   . "numberp")
1310                                (ODDP      . "oddp")
1311                                (PLUSP     . "plusp")
1312                                (RATIONALP . "rationalp")
1313                                (REALP     . "realp")
1314                                (VECTORP   . "vectorp")
1315                                (ZEROP     . "zerop")
1316                                )))))
1317           (when s
1318             (compile-form (second form))
1319             (unless (remove-store-value)
1320               (emit-push-value))
1321             (emit 'invokevirtual
1322                   +lisp-object-class+
1323                   s
1324                   "()Z")
1325             (return-from compile-test 'ifeq))))
1326      (3 (when (eq (car form) 'EQ)
1327           (compile-form (second form))
1328           (unless (remove-store-value)
1329             (emit-push-value))
1330           (compile-form (third form))
1331           (unless (remove-store-value)
1332             (emit-push-value))
1333           (return-from compile-test 'if_acmpne))
1334         (let ((s (cdr (assoc (car form)
1335                              '((=      . "isEqualTo")
1336                                (/=     . "isNotEqualTo")
1337                                (<      . "isLessThan")
1338                                (<=     . "isLessThanOrEqualTo")
1339                                (>      . "isGreaterThan")
1340                                (>=     . "isGreaterThanOrEqualTo")
1341                                (EQL    . "eql")
1342                                (EQUAL  . "equal")
1343                                (EQUALP . "equalp")
1344                                )))))
1345           (when s
1346             (compile-form (second form))
1347             (unless (remove-store-value)
1348               (emit-push-value))
1349             (compile-form (third form))
1350             (unless (remove-store-value)
1351               (emit-push-value))
1352             (emit 'invokevirtual
1353                   +lisp-object-class+
1354                   s
1355                   "(Lorg/armedbear/lisp/LispObject;)Z")
1356             (return-from compile-test 'ifeq))))))
1357  ;; Otherwise...
1358  (compile-form form)
1359  (unless (remove-store-value)
1360    (emit-push-value))
1361  (emit-push-nil)
1362  'if_acmpeq)
1363
1364(defun compile-if (form)
1365  (let* ((test (second form))
1366         (consequent (third form))
1367         (alternate (fourth form))
1368         (label1 (gensym))
1369         (label2 (gensym))
1370         (instr (compile-test test)))
1371    (emit-clear-values)
1372    (emit instr `,label1)
1373    (compile-form consequent)
1374    (emit 'goto `,label2)
1375    (emit 'label `,label1)
1376    (compile-form alternate)
1377    (emit 'label `,label2)))
1378
1379(defun compile-multiple-value-list (form)
1380  (compile-form (second form))
1381  (unless (remove-store-value)
1382    (emit-push-value))
1383  (emit 'invokestatic
1384        "org/armedbear/lisp/Lisp"
1385        "multipleValueList"
1386        "(Lorg/armedbear/lisp/LispObject;)Lorg/armedbear/lisp/LispObject;")
1387  (emit-store-value))
1388
1389(defun compile-let/let* (form)
1390  (let* ((saved-fp (fill-pointer *locals*))
1391         (varlist (second form))
1392         (specialp nil)
1393         env-var)
1394    ;; Are we going to bind any special variables?
1395    (dolist (varspec varlist)
1396      (let ((var (if (consp varspec) (car varspec) varspec)))
1397        (when (special-variable-p var)
1398          (setq specialp t)
1399          (return))))
1400    ;; If so...
1401    (when specialp
1402      ;; Save current dynamic environment.
1403      (setq env-var (vector-push nil *locals*))
1404      (setq *max-locals* (max *max-locals* (fill-pointer *locals*)))
1405      (ensure-thread-var-initialized)
1406      (emit 'aload *thread*)
1407      (emit 'invokevirtual
1408            +lisp-thread-class+
1409            "getDynamicEnvironment"
1410            "()Lorg/armedbear/lisp/Environment;")
1411      (emit 'astore env-var))
1412    (ecase (car form)
1413      (LET
1414       (compile-let-vars varlist))
1415      (LET*
1416       (compile-let*-vars varlist)))
1417    ;; Body of LET.
1418    (do ((body (cddr form) (cdr body)))
1419        ((null (cdr body))
1420         (compile-form (car body) nil))
1421      (compile-form (car body) t))
1422    (when specialp
1423      ;; Restore dynamic environment.
1424      (emit 'aload *thread*)
1425      (emit 'aload env-var)
1426      (emit 'invokevirtual
1427            +lisp-thread-class+
1428            "setDynamicEnvironment"
1429            "(Lorg/armedbear/lisp/Environment;)V"))
1430    ;; Restore fill pointer to its saved value so the slots used by these
1431    ;; bindings will again be available.
1432    (setf (fill-pointer *locals*) saved-fp)))
1433
1434(defun compile-let-vars (varlist)
1435  ;; Generate code to evaluate the initforms and leave the resulting values
1436  ;; on the stack.
1437  (dolist (varspec varlist)
1438    (let (var initform)
1439      (cond ((consp varspec)
1440             (setq var (car varspec)
1441                   initform (cadr varspec)))
1442            (t
1443             (setq var varspec
1444                   initform nil)))
1445      (cond (initform
1446             (compile-form initform)
1447             (emit-push-value))
1448            (t
1449             (emit-push-nil)))))
1450  ;; Add local variables to local variables vector.
1451  (dolist (varspec varlist)
1452    (let ((var (if (consp varspec) (car varspec) varspec)))
1453      (unless (special-variable-p var)
1454        (vector-push var *locals*))))
1455  (setq *max-locals* (max *max-locals* (fill-pointer *locals*)))
1456  ;; At this point the initial values are on the stack. Now generate code to
1457  ;; pop them off one by one and store each one in the corresponding local or
1458  ;; special variable. In order to do this, we must process the variable list
1459  ;; in reverse order.
1460  (do* ((varlist (reverse varlist) (cdr varlist))
1461        (varspec (car varlist) (car varlist))
1462        (var (if (consp varspec) (car varspec) varspec))
1463        (i (1- (fill-pointer *locals*)) (1- i)))
1464       ((null varlist))
1465    (cond ((special-variable-p var)
1466           (let ((g (declare-symbol var)))
1467             (emit 'getstatic
1468                   *this-class*
1469                   g
1470                   "Lorg/armedbear/lisp/Symbol;")
1471             (emit 'swap)
1472             (emit 'invokestatic
1473                   "org/armedbear/lisp/Lisp"
1474                   "bindSpecialVariable"
1475                   "(Lorg/armedbear/lisp/Symbol;Lorg/armedbear/lisp/LispObject;)V")))
1476          (t
1477           (emit 'astore i)))))
1478
1479(defun compile-let*-vars (varlist)
1480  ;; Generate code to evaluate initforms and bind variables.
1481  (let ((i (fill-pointer *locals*)))
1482    (dolist (varspec varlist)
1483      (let (var initform)
1484        (cond ((consp varspec)
1485               (setq var (car varspec)
1486                     initform (cadr varspec)))
1487              (t
1488               (setq var varspec
1489                     initform nil)))
1490        (cond (initform
1491               (compile-form initform)
1492               (emit-push-value))
1493              (t
1494               (emit-push-nil)))
1495        (cond ((special-variable-p var)
1496               (let ((g (declare-symbol var)))
1497                 (emit 'getstatic
1498                       *this-class*
1499                       g
1500                       "Lorg/armedbear/lisp/Symbol;")
1501                 (emit 'swap)
1502                 (emit 'invokestatic
1503                       "org/armedbear/lisp/Lisp"
1504                       "bindSpecialVariable"
1505                       "(Lorg/armedbear/lisp/Symbol;Lorg/armedbear/lisp/LispObject;)V")))
1506              (t
1507               (emit 'astore i)
1508               (vector-push var *locals*)
1509               (incf i))))))
1510  (setq *max-locals* (max *max-locals* (fill-pointer *locals*))))
1511
1512(defvar *tags* ())
1513
1514(defstruct tag name label)
1515
1516(defun label-for-tag (name)
1517  (let ((index (position name *tags* :from-end t :key #'tag-name)))
1518;;     (format t "find-tag index = ~S~%" index)
1519    (when index
1520      (tag-label (aref *tags* index)))))
1521
1522(defun compile-tagbody (form)
1523  (let ((saved-fp (fill-pointer *tags*))
1524        (body (cdr form)))
1525    ;; Scan for tags.
1526    (dolist (f body)
1527      (when (atom f)
1528        (let ((name f)
1529              (label (gensym)))
1530          (vector-push (make-tag :name name :label label) *tags*))))
1531    (dolist (f body)
1532      (cond ((atom f)
1533             (let ((label (label-for-tag f)))
1534               (unless label
1535                 (error "COMPILE-TAGBODY: tag not found: ~S" f))
1536               (emit 'label label)))
1537            (t
1538             (compile-form f t))))
1539    (setf (fill-pointer *tags*) saved-fp))
1540  ;; TAGBODY returns NIL.
1541  (emit-push-nil)
1542  (emit-store-value))
1543
1544(defun compile-go (form)
1545  (let* ((name (cadr form))
1546         (label (label-for-tag name)))
1547    (unless label
1548      (error "COMPILE-GO: tag not found: ~S" name))
1549  (emit 'goto label)))
1550
1551(defun compile-block (form)
1552   (let* ((rest (cdr form))
1553          (block-label (car rest))
1554          (block-exit (gensym))
1555          (*blocks* (acons block-label block-exit *blocks*))
1556          (forms (cdr rest)))
1557     (dolist (form forms)
1558       (compile-form form))
1559     (emit 'label `,block-exit)))
1560
1561(defun compile-progn (form)
1562  (dolist (form (cdr form))
1563    (compile-form form)))
1564
1565(defun compile-setq (form)
1566  (let* ((rest (cdr form))
1567         (len (length rest))
1568         (sym (car rest))
1569         (index (position sym *locals* :from-end t)))
1570    (unless (= len 2)
1571      (error "COMPILE-SETQ too many args for SETQ"))
1572    (when index
1573      (compile-form (cadr rest))
1574      (emit-push-value)
1575      (emit 'astore index)
1576      (return-from compile-setq))
1577    ;; index is NIL, look in *args* ...
1578    (setq index (position sym *args*))
1579    (when index
1580      (cond (*using-arg-array*
1581             (emit 'aload 1)
1582             (emit 'bipush index)
1583             (compile-form (cadr rest))
1584             (emit-push-value)
1585             (emit 'aastore))
1586            (t
1587             (compile-form (cadr rest))
1588             (emit-push-value)
1589             (emit 'astore (1+ index))))
1590      (return-from compile-setq))
1591    ;; still not found
1592    ;; must be a global variable
1593    (let ((g (declare-symbol sym)))
1594      (emit 'getstatic
1595            *this-class*
1596            g
1597            "Lorg/armedbear/lisp/Symbol;")
1598      (compile-form (cadr rest))
1599      (unless (remove-store-value)
1600        (emit-push-value))
1601      (emit 'invokestatic
1602            +lisp-class+
1603            "setSpecialVariable"
1604            "(Lorg/armedbear/lisp/Symbol;Lorg/armedbear/lisp/LispObject;)Lorg/armedbear/lisp/LispObject;")
1605      (emit-store-value))))
1606
1607(defun compile-quote (form)
1608   (let ((obj (second form)))
1609     (cond ((null obj)
1610            (emit-push-nil)
1611            (emit-store-value))
1612           ((symbolp obj)
1613            (if (symbol-package obj)
1614                (let ((g (declare-symbol obj)))
1615                  (emit 'getstatic
1616                        *this-class*
1617                        g
1618                        "Lorg/armedbear/lisp/Symbol;")
1619                  (emit-store-value))
1620                (compile-constant obj)))
1621           ((listp obj)
1622            (let ((g (declare-object-as-string obj)))
1623              (emit 'getstatic
1624                    *this-class*
1625                    g
1626                    +lisp-object+)
1627              (emit-store-value)))
1628           ((constantp obj)
1629            (compile-constant obj))
1630           (t
1631            (error "COMPILE-QUOTE: unsupported case: ~S" form)))))
1632
1633(defun compile-function (form)
1634   (let ((obj (second form)))
1635     (cond ((symbolp obj)
1636            (let ((g (declare-symbol obj)))
1637              (emit 'getstatic
1638                    *this-class*
1639                    g
1640                    "Lorg/armedbear/lisp/Symbol;")
1641              (emit 'invokevirtual
1642                    +lisp-object-class+
1643                    "getSymbolFunctionOrDie"
1644                    "()Lorg/armedbear/lisp/LispObject;")
1645              (emit-store-value)))
1646           #+nil
1647           ((and (consp obj) (eq (car obj) 'LAMBDA))
1648            ;; FIXME We need to construct a proper lexical environment here
1649            ;; and pass it to coerceToFunction().
1650            (let ((g (declare-object-as-string obj)))
1651              (emit 'getstatic
1652                    *this-class*
1653                    g
1654                    +lisp-object+)
1655              (emit 'invokestatic
1656                    +lisp-class+
1657                    "coerceToFunction"
1658                    "(Lorg/armedbear/lisp/LispObject;)Lorg/armedbear/lisp/Function;")
1659              (emit-store-value)))
1660           (t
1661            (error "COMPILE-FUNCTION: unsupported case: ~S" form)))))
1662
1663(defun compile-return (form)
1664   (let* ((rest (cdr form))
1665          (block-label nil)
1666          (block-exit (cdr (assoc block-label *blocks*)))
1667          (result-form (car rest)))
1668     (unless block-exit
1669       (error "no block named ~S is currently visible" block-label))
1670     (compile-form result-form)
1671     (emit 'goto `,block-exit)))
1672
1673(defun compile-return-from (form)
1674   (let* ((rest (cdr form))
1675          (block-label (car rest))
1676          (block-exit (cdr (assoc block-label *blocks*)))
1677          (result-form (cadr rest)))
1678     (unless block-exit
1679       (error "no block named ~S is currently visible" block-label))
1680     (compile-form result-form)
1681     (emit 'goto `,block-exit)))
1682
1683(defun compile-plus (form)
1684  (let* ((args (cdr form))
1685         (len (length args)))
1686    (case len
1687      (2
1688       (let ((first (first args))
1689             (second (second args)))
1690         (cond
1691          ((eql first 1)
1692           (compile-form second)
1693           (emit-invoke-method "incr"))
1694          ((eql second 1)
1695           (compile-form first)
1696           (emit-invoke-method "incr"))
1697          (t
1698           (compile-binary-operation "add" args)))))
1699      (t
1700       (compile-function-call '+ args)))))
1701
1702(defun compile-minus (form)
1703  (let* ((args (cdr form))
1704         (len (length args)))
1705    (case len
1706      (2
1707       (let ((first (first args))
1708             (second (second args)))
1709         (cond
1710          ((eql second 1)
1711           (compile-form first)
1712           (emit-invoke-method "decr"))
1713          (t
1714           (compile-binary-operation "subtract" args)))))
1715      (t
1716       (compile-function-call '- args)))))
1717
1718(defun compile-variable-ref (form)
1719  (let ((index (position form *locals* :from-end t)))
1720    (when index
1721      (emit 'aload index)
1722      (emit-store-value)
1723      (return-from compile-variable-ref)))
1724  ;; Not found in locals; look in args.
1725  (let ((index (position form *args*)))
1726    (when index
1727      (cond (*using-arg-array*
1728             (emit 'aload 1)
1729             (emit 'bipush index)
1730             (emit 'aaload)
1731             (emit-store-value)
1732             (return-from compile-variable-ref))
1733            (t
1734             (emit 'aload (1+ index))
1735             (emit-store-value)
1736             (return-from compile-variable-ref)))))
1737
1738  ;; Otherwise it must be a global variable.
1739  (let ((g (declare-symbol form)))
1740    (emit 'getstatic
1741          *this-class*
1742          g
1743          "Lorg/armedbear/lisp/Symbol;")
1744    (emit 'invokevirtual
1745          "org/armedbear/lisp/Symbol"
1746          "symbolValue"
1747          "()Lorg/armedbear/lisp/LispObject;")
1748    (emit-store-value)
1749    (return-from compile-variable-ref)))
1750
1751;; If for-effect is true, no value needs to be left on the stack.
1752(defun compile-form (form &optional for-effect)
1753  (cond
1754   ((consp form)
1755    (let ((first (first form))
1756          (rest (rest form)))
1757      (when (macro-function first)
1758        (compile-form (macroexpand form))
1759        (return-from compile-form))
1760      (when (symbolp first)
1761        (let ((handler (get first 'jvm-compile)))
1762          (when handler
1763            (funcall handler form)
1764            (return-from compile-form))))
1765      (cond
1766       ((special-operator-p first)
1767        (error "COMPILE-FORM unhandled special operator ~S" first))
1768       (t ; Function call.
1769        (compile-function-call first rest for-effect)))))
1770   ((eq form '())
1771    (unless for-effect
1772      (emit-push-nil)
1773      (emit-store-value)))
1774   ((eq form t)
1775    (unless for-effect
1776      (emit-push-t)
1777      (emit-store-value)))
1778   ((symbolp form)
1779    (when (keywordp form)
1780      (let ((g (declare-keyword form)))
1781        (emit 'getstatic
1782              *this-class*
1783              g
1784              "Lorg/armedbear/lisp/Symbol;"))
1785      (emit-store-value)
1786      (return-from compile-form))
1787
1788    (compile-variable-ref form))
1789   ((constantp form)
1790    (unless for-effect
1791      (compile-constant form)))
1792   (t
1793    (error "COMPILE-FORM unhandled case ~S" form))))
1794
1795;; Returns descriptor.
1796(defun analyze-args (args)
1797  (assert (not (memq '&AUX args)))
1798  (when (or (memq '&KEY args)
1799            (memq '&OPTIONAL args)
1800            (memq '&REST args))
1801    (setq *using-arg-array* t)
1802    (setq *hairy-arglist-p* t)
1803    (return-from analyze-args #.(format nil "([~A)~A" +lisp-object+ +lisp-object+)))
1804  (case (length args)
1805    (0 #.(format nil "()~A" +lisp-object+))
1806    (1 #.(format nil "(~A)~A" +lisp-object+ +lisp-object+))
1807    (2 #.(format nil "(~A~A)~A" +lisp-object+ +lisp-object+ +lisp-object+))
1808    (3 #.(format nil "(~A~A~A)~A" +lisp-object+ +lisp-object+ +lisp-object+ +lisp-object+))
1809    (t (setq *using-arg-array* t)
1810       #.(format nil "([~A)~A" +lisp-object+ +lisp-object+))))
1811
1812(defun compile-defun (name form)
1813  (unless (eq (car form) 'LAMBDA)
1814    (return-from compile-defun nil))
1815  (setq form (transform form))
1816  (let* ((*defun-name* name)
1817         (*declared-symbols* (make-hash-table))
1818         (*declared-functions* (make-hash-table))
1819         (*this-class* "org/armedbear/lisp/out")
1820         (args (cadr form))
1821         (body (cddr form))
1822         (*using-arg-array* nil)
1823         (*hairy-arglist-p* nil)
1824         (descriptor (analyze-args args))
1825         (execute-method (make-method :name "execute"
1826                                      :descriptor descriptor))
1827         (*code* ())
1828         (*static-code* ())
1829         (*fields* ())
1830         (*blocks* ())
1831         (*tags* (make-array 256 :fill-pointer 0)) ; FIXME Remove hard limit!
1832         (*args* (make-array 256 :fill-pointer 0)) ; FIXME Remove hard limit!
1833         (*locals* (make-array 256 :fill-pointer 0)) ; FIXME Remove hard limit!
1834         (*max-locals* 0)
1835         (*pool* ())
1836         (*pool-count* 1)
1837         (*val* nil)
1838         (*thread* nil)
1839         (*thread-var-initialized* nil))
1840    (setf (method-name-index execute-method)
1841          (pool-name (method-name execute-method)))
1842    (setf (method-descriptor-index execute-method)
1843          (pool-name (method-descriptor execute-method)))
1844    (if *hairy-arglist-p*
1845        (let* ((fun (sys::make-compiled-function nil args body))
1846               (vars (sys::varlist fun)))
1847          (dolist (var vars)
1848            (vector-push var *args*)))
1849        (dolist (arg args)
1850          (vector-push arg *args*)))
1851    (if *using-arg-array*
1852        ;; Using arg array: slot 0 is "this" pointer, slot 1 is arg array,
1853        ;; first available slot is 2.
1854        (setf (fill-pointer *locals*) 2)
1855        ;; Not using arg array: slot 0 is "this" pointer, next N slots are used
1856        ;; for args.
1857        (setf (fill-pointer *locals*) (1+ (length args))))
1858    ;; Reserve the next available slot for the value register.
1859    (setq *val* (fill-pointer *locals*))
1860    (incf (fill-pointer *locals*))
1861    (setf *max-locals* (fill-pointer *locals*))
1862    ;; Reserve the next available slot for the thread register.
1863    (setq *thread* (fill-pointer *locals*))
1864    (incf (fill-pointer *locals*))
1865    (setf *max-locals* (fill-pointer *locals*))
1866    (when *hairy-arglist-p*
1867      (emit 'aload_0)
1868      (emit 'aload_1)
1869      (emit 'invokevirtual
1870            *this-class*
1871            "processArgs"
1872            "([Lorg/armedbear/lisp/LispObject;)[Lorg/armedbear/lisp/LispObject;")
1873      (emit 'astore_1))
1874    (dolist (f body)
1875      (compile-form f))
1876    (unless (remove-store-value)
1877      (emit-push-value)) ; leave result on stack
1878    (emit 'areturn)
1879    (setf (method-code execute-method) (code-bytes *code*))
1880    (setf (method-max-locals execute-method) *max-locals*)
1881
1882    (let* ((super
1883            (if *hairy-arglist-p*
1884                "org.armedbear.lisp.CompiledFunction"
1885                (case (length args)
1886                  (0 "org.armedbear.lisp.Primitive0")
1887                  (1 "org.armedbear.lisp.Primitive1")
1888                  (2 "org.armedbear.lisp.Primitive2")
1889                  (3 "org.armedbear.lisp.Primitive3")
1890                  (t "org.armedbear.lisp.Primitive"))))
1891           (this-index (pool-class *this-class*))
1892           (super-index (pool-class super))
1893           (constructor (make-constructor super *defun-name* args body)))
1894      (pool-name "Code") ; Must be in pool!
1895
1896      ;; Write class file (out.class in current directory).
1897      (with-open-file (*stream* "out.class"
1898                                :direction :output
1899                                :element-type 'unsigned-byte
1900                                :if-exists :supersede)
1901        (write-u4 #xCAFEBABE)
1902        (write-u2 3)
1903        (write-u2 45)
1904        (write-pool)
1905        ;; access flags
1906        (write-u2 #x21)
1907        (write-u2 this-index)
1908        (write-u2 super-index)
1909        ;; interfaces count
1910        (write-u2 0)
1911        ;; fields count
1912        (write-u2 (length *fields*))
1913        ;; fields
1914        (dolist (field *fields*)
1915          (write-field field))
1916        ;; methods count
1917        (write-u2 2)
1918        ;; methods
1919        (write-method execute-method)
1920        (write-method constructor)
1921        ;; attributes count
1922        (write-u2 0))))
1923  (sys::load-compiled-function "out.class"))
1924
1925(defun get-lambda-to-compile (definition-designator)
1926  (if (and (consp definition-designator)
1927           (eq (car definition-designator) 'LAMBDA))
1928      definition-designator
1929      (multiple-value-bind (lambda-expression closure-p)
1930        (function-lambda-expression definition-designator)
1931        (when closure-p
1932          (error "unable to compile function defined in non-null lexical environment"))
1933  (unless lambda-expression
1934    (error "can't find a definition"))
1935        lambda-expression)))
1936
1937(defun load-verbose-prefix ()
1938  (with-output-to-string (s)
1939    (princ #\; s)
1940    (dotimes (i (1- sys::*load-depth*))
1941      (princ #\space s))))
1942
1943(defun jvm-compile (name &optional definition)
1944  (let ((prefix (load-verbose-prefix)))
1945    (when name
1946      (format t "~A Compiling ~S ...~%" prefix name)
1947      (when (and (fboundp name) (typep (fdefinition name) 'generic-function))
1948        (format t "~A Unable to compile generic function ~S~%" prefix name)
1949        (return-from jvm-compile (values name nil t)))
1950      (unless (symbolp name)
1951        (format t "~A Unable to compile ~S~%" prefix name)
1952        (return-from jvm-compile (values name nil t))))
1953    (unless definition
1954      (sys::resolve name)
1955      (setf definition (fdefinition name))
1956      (when (compiled-function-p definition)
1957        (when name
1958          (format t "~A Already compiled ~S~%" prefix name))
1959        (return-from jvm-compile (values name nil nil))))
1960    (handler-case
1961        (let* ((*package* (if (and name (symbol-package name))
1962                              (symbol-package name)
1963                              *package*))
1964               (expr (get-lambda-to-compile definition))
1965               (compiled-definition (compile-defun name expr)))
1966          (when (and name (functionp compiled-definition))
1967            (sys::%set-lambda-name compiled-definition name)
1968            (sys::%set-call-count compiled-definition (sys::%call-count definition))
1969            (sys::%set-arglist compiled-definition (sys::arglist definition))
1970            (if (macro-function name)
1971                (setf (fdefinition name) (sys::make-macro compiled-definition))
1972                (setf (fdefinition name) compiled-definition)))
1973          (when name
1974            (format t "~A Compiled ~S~%" prefix name))
1975          (values (or name compiled-definition) nil nil))
1976      (error (c)
1977             (format t "Error: ~S~%" c)
1978             (when name (format t "~A Unable to compile ~S~%" prefix name))
1979             (values (or name (sys::coerce-to-function definition)) nil t)))))
1980
1981(defun jvm-compile-package (package-designator)
1982  (let ((pkg (if (packagep package-designator)
1983                 package-designator
1984                 (find-package package-designator))))
1985      (dolist (sym (sys::package-symbols pkg))
1986        (when (fboundp sym)
1987          (unless (or (special-operator-p sym) (macro-function sym))
1988            ;; Force autoload to be resolved.
1989            (sys::resolve sym)
1990            (let ((f (fdefinition sym)))
1991              (unless (compiled-function-p f)
1992                (jvm-compile sym)))))))
1993  t)
1994
1995(defun install-handler (fun &optional handler)
1996  (let ((handler (or handler
1997                     (find-symbol (concatenate 'string "COMPILE-" (symbol-name fun)) 'jvm))))
1998    (unless (and handler (fboundp handler))
1999      (error "no handler for ~S" fun))
2000    (setf (get fun 'jvm-compile) handler)))
2001
2002(mapc #'install-handler '(block
2003                          function
2004                          go
2005                          if
2006                          multiple-value-list
2007                          progn
2008                          quote
2009                          return
2010                          return-from
2011                          setq
2012                          tagbody))
2013
2014(install-handler 'let  'compile-let/let*)
2015(install-handler 'let* 'compile-let/let*)
2016(install-handler '+    'compile-plus)
2017(install-handler '-    'compile-minus)
2018
2019#+nil
2020(defun compile (name &optional definition)
2021  (jvm:jvm-compile name definition))
2022
2023(defmacro defun (name lambda-list &rest body)
2024  `(progn
2025     (sys::%defun ',name ',lambda-list ',body)
2026     (compiler::%compile ',name)
2027     (when *auto-compile*
2028       (jvm-compile ',name))
2029     ',name))
2030
2031(mapc #'jvm-compile '(pool-add
2032                      pool-find-entry
2033                      pool-name
2034                      pool-get
2035                      compile-form))
Note: See TracBrowser for help on using the repository browser.