1 ! Copyright (C) 2005, 2009 Slava Pestov.
2 ! See http://factorcode.org/license.txt for BSD license.
3 USING: accessors assocs sequences kernel combinators make math
4 math.order math.ranges system namespaces locals layouts words
5 alien alien.accessors alien.c-types literals cpu.architecture
6 cpu.ppc.assembler cpu.ppc.assembler.backend compiler.cfg.registers
7 compiler.cfg.instructions compiler.cfg.comparisons
8 compiler.codegen.fixup compiler.cfg.intrinsics
9 compiler.cfg.stack-frame compiler.cfg.build-stack-frame
10 compiler.units compiler.constants compiler.codegen ;
11 FROM: cpu.ppc.assembler => B ;
14 ! PowerPC register assignments:
15 ! r2-r12: integer vregs
17 ! r30: integer scratch
21 ! Add some methods to the assembler that are useful to us
22 M: label (B) [ 0 ] 2dip (B) rc-relative-ppc-3 label-fixup ;
23 M: label BC [ 0 BC ] dip rc-relative-ppc-2 label-fixup ;
25 enable-float-intrinsics
28 \ ##integer>float t frame-required? set-word-prop
29 \ ##float>integer t frame-required? set-word-prop
32 M: ppc machine-registers
34 { int-regs $[ 2 12 [a,b] 15 29 [a,b] append ] }
35 { float-regs $[ 0 29 [a,b] ] }
38 CONSTANT: scratch-reg 30
39 CONSTANT: fp-scratch-reg 30
41 M: ppc two-operand? f ;
43 M: ppc %load-immediate ( reg n -- ) swap LOAD ;
45 M: ppc %load-reference ( reg obj -- )
46 [ 0 swap LOAD32 ] [ rc-absolute-ppc-2/2 rel-immediate ] bi* ;
48 M: ppc %alien-global ( register symbol dll -- )
49 [ 0 swap LOAD32 ] 2dip rc-absolute-ppc-2/2 rel-dlsym ;
54 GENERIC: loc-reg ( loc -- reg )
56 M: ds-loc loc-reg drop ds-reg ;
57 M: rs-loc loc-reg drop rs-reg ;
59 : loc>operand ( loc -- reg n )
60 [ loc-reg ] [ n>> cells neg ] bi ; inline
62 M: ppc %peek loc>operand LWZ ;
63 M: ppc %replace loc>operand STW ;
65 : (%inc) ( n reg -- ) dup rot cells ADDI ; inline
67 M: ppc %inc-d ( n -- ) ds-reg (%inc) ;
68 M: ppc %inc-r ( n -- ) rs-reg (%inc) ;
70 HOOK: reserved-area-size os ( -- n )
72 ! The start of the stack frame contains the size of this frame
73 ! as well as the currently executing XT
74 : factor-area-size ( -- n ) 2 cells ; foldable
75 : next-save ( n -- i ) cell - ;
76 : xt-save ( n -- i ) 2 cells - ;
78 ! Next, we have the spill area as well as the FFI parameter area.
79 ! It is safe for them to overlap, since basic blocks with FFI calls
80 ! will never spill -- indeed, basic blocks with FFI calls do not
81 ! use vregs at all, and the FFI call is a stack analysis sync point.
82 ! In the future this will change and the stack frame logic will
83 ! need to be untangled somewhat.
85 : param@ ( n -- x ) reserved-area-size + ; inline
87 : param-save-size ( -- n ) 8 cells ; foldable
90 reserved-area-size param-save-size + + ; inline
92 : spill-integer@ ( n -- offset )
93 spill-integer-offset local@ ;
95 : spill-float@ ( n -- offset )
96 spill-float-offset local@ ;
98 ! Some FP intrinsics need a temporary scratch area in the stack
99 ! frame, 8 bytes in size. This is in the param-save area so it
100 ! does not overlap with spill slots.
101 : scratch@ ( n -- offset )
102 stack-frame get total-size>>
108 : gc-root@ ( n -- offset )
109 gc-root-offset local@ ;
111 ! Finally we have the linkage area
112 HOOK: lr-save os ( -- n )
114 M: ppc stack-frame-size ( stack-frame -- i )
121 M: ppc %call ( word -- ) 0 BL rc-relative-ppc-3 rel-word-pic ;
123 M: ppc %jump ( word -- )
124 0 6 LOAD32 8 rc-absolute-ppc-2/2 rel-here
125 0 B rc-relative-ppc-3 rel-word-pic-tail ;
127 M: ppc %jump-label ( label -- ) B ;
128 M: ppc %return ( -- ) BLR ;
130 M:: ppc %dispatch ( src temp -- )
132 4 cells rc-absolute-ppc-2/2 rel-here
137 :: (%slot) ( obj slot tag temp -- reg offset )
139 temp tag neg ; inline
141 : (%slot-imm) ( obj slot tag -- reg offset )
142 [ cells ] dip - ; inline
144 M: ppc %slot ( dst obj slot tag temp -- ) (%slot) LWZ ;
145 M: ppc %slot-imm ( dst obj slot tag -- ) (%slot-imm) LWZ ;
146 M: ppc %set-slot ( src obj slot tag temp -- ) (%slot) STW ;
147 M: ppc %set-slot-imm ( src obj slot tag -- ) (%slot-imm) STW ;
149 M:: ppc %string-nth ( dst src index temp -- )
153 dst temp string-offset LBZ
156 temp src string-aux-offset LWZ
159 temp temp byte-array-offset LHZ
165 M:: ppc %set-string-nth-fast ( ch obj index temp -- )
167 ch temp string-offset STB ;
170 M: ppc %add-imm ADDI ;
171 M: ppc %sub swap SUBF ;
172 M: ppc %sub-imm SUBI ;
174 M: ppc %mul-imm MULLI ;
176 M: ppc %and-imm ANDI ;
180 M: ppc %xor-imm XORI ;
182 M: ppc %shl-imm swapd SLWI ;
184 M: ppc %shr-imm swapd SRWI ;
186 M: ppc %sar-imm SRAWI ;
189 :: overflow-template ( label dst src1 src2 insn -- )
192 dst src2 src1 insn call
195 M: ppc %fixnum-add ( label dst src1 src2 -- )
196 [ ADDO. ] overflow-template ;
198 M: ppc %fixnum-sub ( label dst src1 src2 -- )
199 [ SUBFO. ] overflow-template ;
201 M: ppc %fixnum-mul ( label dst src1 src2 -- )
202 [ MULLWO. ] overflow-template ;
204 : bignum@ ( n -- offset ) cells bignum tag-number - ; inline
206 M:: ppc %integer>bignum ( dst src temp -- )
209 dst 0 >bignum %load-reference
210 ! Is it zero? Then just go to the end and return this zero
214 dst 4 cells bignum temp %allot
217 temp dst 1 bignum@ STW
220 temp temp cell-bits 1- SRAWI
223 temp dst 2 bignum@ STW
224 ! Make negative value positive
230 temp dst 3 bignum@ STW
234 M:: ppc %bignum>integer ( dst src temp -- )
237 temp src 1 bignum@ LWZ
238 ! if the length is 1, its just the sign and nothing else,
241 0 temp 1 tag-fixnum CMPI
244 dst src 3 bignum@ LWZ
246 temp src 2 bignum@ LWZ
247 ! branchless arithmetic: we want to turn 0 into 1,
252 ! multiply value by sign
257 M: ppc %add-float FADD ;
258 M: ppc %sub-float FSUB ;
259 M: ppc %mul-float FMUL ;
260 M: ppc %div-float FDIV ;
262 M:: ppc %integer>float ( dst src -- )
263 HEX: 4330 scratch-reg LIS
264 scratch-reg 1 0 scratch@ STW
266 scratch-reg dup HEX: 8000 XORIS
267 scratch-reg 1 4 scratch@ STW
269 scratch-reg 4503601774854144.0 %load-reference
270 fp-scratch-reg scratch-reg float-offset LFD
271 dst dst fp-scratch-reg FSUB ;
273 M:: ppc %float>integer ( dst src -- )
274 fp-scratch-reg src FCTIWZ
275 fp-scratch-reg 1 0 scratch@ STFD
276 dst 1 4 scratch@ LWZ ;
278 M: ppc %copy ( dst src -- ) MR ;
280 M: ppc %copy-float ( dst src -- ) FMR ;
282 M: ppc %unbox-float ( dst src -- ) float-offset LFD ;
284 M:: ppc %box-float ( dst src temp -- )
285 dst 16 float temp %allot
286 src dst float-offset STFD ;
288 M:: ppc %unbox-any-c-ptr ( dst src temp -- )
290 { "is-byte-array" "end" "start" } [ define-label ] each
291 ! Address is computed in dst
293 ! Load object into scratch-reg
295 ! We come back here with displaced aliens
296 "start" resolve-label
298 0 scratch-reg \ f tag-number CMPI
301 ! Is the object an alien?
302 0 scratch-reg header-offset LWZ
303 0 0 alien type-number tag-fixnum CMPI
304 "is-byte-array" get BNE
305 ! If so, load the offset
306 0 scratch-reg alien-offset LWZ
307 ! Add it to address being computed
309 ! Now recurse on the underlying alien
310 scratch-reg scratch-reg underlying-alien-offset LWZ
312 "is-byte-array" resolve-label
313 ! Add byte array address to address being computed
314 dst dst scratch-reg ADD
315 ! Add an offset to start of byte array's data area
316 dst dst byte-array-offset ADDI
320 : alien@ ( n -- n' ) cells object tag-number - ;
322 M:: ppc %box-alien ( dst src temp -- )
325 dst \ f tag-number %load-immediate
328 dst 4 cells alien temp %allot
332 temp \ f tag-number %load-immediate
333 temp dst 1 alien@ STW
334 ! Store underlying-alien slot
335 temp dst 2 alien@ STW
339 M: ppc %alien-unsigned-1 0 LBZ ;
340 M: ppc %alien-unsigned-2 0 LHZ ;
342 M: ppc %alien-signed-1 dupd 0 LBZ dup EXTSB ;
343 M: ppc %alien-signed-2 0 LHA ;
345 M: ppc %alien-cell 0 LWZ ;
347 M: ppc %alien-float 0 LFS ;
348 M: ppc %alien-double 0 LFD ;
350 M: ppc %set-alien-integer-1 swap 0 STB ;
351 M: ppc %set-alien-integer-2 swap 0 STH ;
353 M: ppc %set-alien-cell swap 0 STW ;
355 M: ppc %set-alien-float swap 0 STFS ;
356 M: ppc %set-alien-double swap 0 STFD ;
358 : load-zone-ptr ( reg -- )
359 "nursery" f %alien-global ;
361 : load-allot-ptr ( nursery-ptr allot-ptr -- )
362 [ drop load-zone-ptr ] [ swap 4 LWZ ] 2bi ;
364 :: inc-allot-ptr ( nursery-ptr allot-ptr n -- )
365 scratch-reg allot-ptr n 8 align ADDI
366 scratch-reg nursery-ptr 4 STW ;
368 :: store-header ( dst class -- )
369 class type-number tag-fixnum scratch-reg LI
370 scratch-reg dst 0 STW ;
372 : store-tagged ( dst tag -- )
373 dupd tag-number ORI ;
375 M:: ppc %allot ( dst size class nursery-ptr -- )
376 nursery-ptr dst load-allot-ptr
377 nursery-ptr dst size inc-allot-ptr
378 dst class store-header
379 dst class store-tagged ;
381 : load-cards-offset ( dst -- )
382 [ "cards_offset" f %alien-global ] [ dup 0 LWZ ] bi ;
384 : load-decks-offset ( dst -- )
385 [ "decks_offset" f %alien-global ] [ dup 0 LWZ ] bi ;
387 M:: ppc %write-barrier ( src card# table -- )
388 card-mark scratch-reg LI
391 table load-cards-offset
392 src card# card-bits SRWI
393 table scratch-reg card# STBX
396 table load-decks-offset
397 src card# deck-bits SRWI
398 table scratch-reg card# STBX ;
400 M:: ppc %check-nursery ( label temp1 temp2 -- )
403 temp2 temp2 3 cells LWZ
404 ! add ALLOT_BUFFER_ZONE to here
405 temp1 temp1 1024 ADDI
410 M:: ppc %save-gc-root ( gc-root register -- )
411 register 1 gc-root gc-root@ STW ;
413 M:: ppc %load-gc-root ( gc-root register -- )
414 register 1 gc-root gc-root@ LWZ ;
416 M:: ppc %call-gc ( gc-root-count -- )
417 %prepare-alien-invoke
418 3 1 gc-root-base local@ ADDI
420 "inline_gc" f %alien-invoke ;
422 M: ppc %prologue ( n -- )
423 0 11 LOAD32 rc-absolute-ppc-2/2 rel-this
426 [ [ 1 1 ] dip neg ADDI ]
427 [ [ 11 1 ] dip xt-save STW ]
429 [ [ 11 1 ] dip next-save STW ]
430 [ [ 0 1 ] dip lr-save + STW ]
433 M: ppc %epilogue ( n -- )
434 #! At the end of each word that calls a subroutine, we store
435 #! the previous link register value in r0 by popping it off
436 #! the stack, set the link register to the contents of r0,
437 #! and jump to the link register.
438 [ [ 0 1 ] dip lr-save + LWZ ]
439 [ [ 1 1 ] dip ADDI ] bi
442 :: (%boolean) ( dst temp word -- )
444 dst \ f tag-number %load-immediate
445 "end" get word execute
446 dst \ t %load-reference
447 "end" get resolve-label ; inline
449 : %boolean ( dst temp cc -- )
451 { cc< [ \ BLT (%boolean) ] }
452 { cc<= [ \ BLE (%boolean) ] }
453 { cc> [ \ BGT (%boolean) ] }
454 { cc>= [ \ BGE (%boolean) ] }
455 { cc= [ \ BEQ (%boolean) ] }
456 { cc/= [ \ BNE (%boolean) ] }
459 : (%compare) ( src1 src2 -- ) [ 0 ] dip CMP ; inline
460 : (%compare-imm) ( src1 src2 -- ) [ 0 ] 2dip CMPI ; inline
461 : (%compare-float) ( src1 src2 -- ) [ 0 ] dip FCMPU ; inline
463 M: ppc %compare (%compare) %boolean ;
464 M: ppc %compare-imm (%compare-imm) %boolean ;
465 M: ppc %compare-float (%compare-float) %boolean ;
467 : %branch ( label cc -- )
477 M: ppc %compare-branch (%compare) %branch ;
478 M: ppc %compare-imm-branch (%compare-imm) %branch ;
479 M: ppc %compare-float-branch (%compare-float) %branch ;
481 M: ppc %spill-integer ( src n -- ) spill-integer@ 1 swap STW ;
482 M: ppc %reload-integer ( dst n -- ) spill-integer@ 1 swap LWZ ;
484 M: ppc %spill-float ( src n -- ) spill-float@ 1 swap STFD ;
485 M: ppc %reload-float ( dst n -- ) spill-float@ 1 swap LFD ;
489 M: int-regs return-reg drop 3 ;
490 M: int-regs param-regs drop { 3 4 5 6 7 8 9 10 } ;
491 M: float-regs return-reg drop 1 ;
493 M: int-regs %save-param-reg drop 1 rot local@ STW ;
494 M: int-regs %load-param-reg drop 1 rot local@ LWZ ;
496 M: single-float-rep %save-param-reg drop 1 rot local@ STFS ;
497 M: single-float-rep %load-param-reg 1 rot local@ LFS ;
499 M: double-float-rep %save-param-reg drop 1 rot local@ STFD ;
500 M: double-float-rep %load-param-reg 1 rot local@ LFD ;
502 M: stack-params %load-param-reg ( stack reg rep -- )
503 drop [ 0 1 rot local@ LWZ 0 1 ] dip param@ STW ;
505 : next-param@ ( n -- x ) param@ stack-frame get total-size>> + ;
507 M: stack-params %save-param-reg ( stack reg rep -- )
508 #! Funky. Read the parameter from the caller's stack frame.
509 #! This word is used in callbacks
511 [ 0 1 ] dip next-param@ LWZ
512 [ 0 1 ] dip local@ STW ;
514 M: ppc %prepare-unbox ( -- )
515 ! First parameter is top of stack
517 ds-reg dup cell SUBI ;
519 M: ppc %unbox ( n rep func -- )
520 ! Value must be in r3
523 ! Store the return value on the C stack
524 over [ [ reg-class-of return-reg ] keep %save-param-reg ] [ 2drop ] if ;
526 M: ppc %unbox-long-long ( n func -- )
527 ! Value must be in r3:r4
530 ! Store the return value on the C stack
532 [ [ 3 1 ] dip local@ STW ]
533 [ [ 4 1 ] dip cell + local@ STW ] bi
536 M: ppc %unbox-large-struct ( n c-type -- )
537 ! Value must be in r3
538 ! Compute destination address and load struct size
539 [ [ 4 1 ] dip local@ ADDI ] [ heap-size 5 LI ] bi*
541 "to_value_struct" f %alien-invoke ;
543 M: ppc %box ( n rep func -- )
544 ! If the source is a stack location, load it into freg #0.
545 ! If the source is f, then we assume the value is already in
547 [ over [ 0 over reg-class-of param-reg swap %load-param-reg ] [ 2drop ] if ] dip
550 M: ppc %box-long-long ( n func -- )
553 [ [ 3 1 ] dip local@ LWZ ]
554 [ [ 4 1 ] dip cell + local@ LWZ ] bi
556 ] dip f %alien-invoke ;
558 : struct-return@ ( n -- n )
559 [ stack-frame get params>> ] unless* local@ ;
561 M: ppc %prepare-box-struct ( -- )
562 #! Compute target address for value struct return
563 3 1 f struct-return@ ADDI
566 M: ppc %box-large-struct ( n c-type -- )
567 ! If n = f, then we're boxing a returned struct
568 ! Compute destination address and load struct size
569 [ [ 3 1 ] dip struct-return@ ADDI ] [ heap-size 4 LI ] bi*
571 "box_value_struct" f %alien-invoke ;
573 M: ppc %prepare-alien-invoke
574 #! Save Factor stack pointers in case the C code calls a
575 #! callback which does a GC, which must reliably trace
577 scratch-reg "stack_chain" f %alien-global
578 scratch-reg scratch-reg 0 LWZ
580 ds-reg scratch-reg 8 STW
581 rs-reg scratch-reg 12 STW ;
583 M: ppc %alien-invoke ( symbol dll -- )
584 [ 11 ] 2dip %alien-global 11 MTLR BLRL ;
586 M: ppc %alien-callback ( quot -- )
587 3 swap %load-reference "c_to_factor" f %alien-invoke ;
589 M: ppc %prepare-alien-indirect ( -- )
590 "unbox_alien" f %alien-invoke
593 M: ppc %alien-indirect ( -- )
596 M: ppc %callback-value ( ctype -- )
597 ! Save top of data stack
600 ! Restore data/call/retain stacks
601 "unnest_stacks" f %alien-invoke
602 ! Restore top of data stack
604 ! Unbox former top of data stack to return registers
607 M: ppc small-enough? ( n -- ? ) -32768 32767 between? ;
609 M: ppc return-struct-in-registers? ( c-type -- ? )
610 c-type return-in-registers?>> ;
612 M: ppc %box-small-struct ( c-type -- )
613 #! Box a <= 16-byte struct returned in r3:r4:r5:r6
615 "box_medium_struct" f %alien-invoke ;
617 : %unbox-struct-1 ( -- )
618 ! Alien must be in r3.
619 "alien_offset" f %alien-invoke
622 : %unbox-struct-2 ( -- )
623 ! Alien must be in r3.
624 "alien_offset" f %alien-invoke
628 : %unbox-struct-4 ( -- )
629 ! Alien must be in r3.
630 "alien_offset" f %alien-invoke
636 M: ppc %unbox-small-struct ( size -- )
637 #! Alien must be in EAX.
638 heap-size cell align cell /i {
639 { 1 [ %unbox-struct-1 ] }
640 { 2 [ %unbox-struct-2 ] }
641 { 4 [ %unbox-struct-4 ] }
647 { [ os macosx? ] [ "cpu.ppc.macosx" require ] }
648 { [ os linux? ] [ "cpu.ppc.linux" require ] }
651 "complex-double" c-type t >>return-in-registers? drop
655 [ alien-unsigned-4 c-bool> ] >>getter
656 [ [ >c-bool ] 2dip set-alien-unsigned-4 ] >>setter
659 "box_boolean" >>boxer
660 "to_boolean" >>unboxer
661 "bool" define-primitive-type
662 ] with-compilation-unit