! Copyright (C) 2008, 2010 Slava Pestov.
! See http://factorcode.org/license.txt for BSD license.
-USING: namespaces accessors math math.order assocs kernel sequences
-combinators classes words system cpu.architecture layouts compiler.cfg
-compiler.cfg.rpo compiler.cfg.instructions
-compiler.cfg.registers compiler.cfg.stack-frame ;
+USING: accessors compiler.cfg.instructions
+compiler.cfg.linearization cpu.architecture kernel layouts math
+math.order namespaces sequences ;
IN: compiler.cfg.build-stack-frame
-SYMBOL: frame-required?
+SYMBOLS: param-area-size allot-area-size allot-area-align ;
-GENERIC: compute-stack-frame* ( insn -- )
+GENERIC: compute-stack-frame* ( insn -- ? )
-: request-stack-frame ( stack-frame -- )
- frame-required? on
- stack-frame [ max-stack-frame ] change ;
+M:: ##local-allot compute-stack-frame* ( insn -- ? )
+ insn size>> :> s
+ insn align>> :> a
+ allot-area-align [ a max ] change
+ allot-area-size [ a align [ insn offset<< ] [ s + ] bi ] change t ;
-M: ##stack-frame compute-stack-frame*
- stack-frame>> request-stack-frame ;
+M: alien-call-insn compute-stack-frame*
+ stack-size>> param-area-size [ max ] change t ;
-: frame-required ( -- ) frame-required? on ;
-
-: vm-frame-required ( -- )
- frame-required
- stack-frame new vm-stack-space >>params request-stack-frame ;
+: vm-frame-required ( -- ? )
+ vm-stack-space param-area-size [ max ] change t ;
M: ##call-gc compute-stack-frame* drop vm-frame-required ;
M: ##box compute-stack-frame* drop vm-frame-required ;
M: ##unbox compute-stack-frame* drop vm-frame-required ;
M: ##box-long-long compute-stack-frame* drop vm-frame-required ;
-M: ##begin-callback compute-stack-frame* drop vm-frame-required ;
-M: ##end-callback compute-stack-frame* drop vm-frame-required ;
-M: ##unary-float-function compute-stack-frame* drop vm-frame-required ;
-M: ##binary-float-function compute-stack-frame* drop vm-frame-required ;
+M: ##callback-inputs compute-stack-frame* drop vm-frame-required ;
+M: ##callback-outputs compute-stack-frame* drop vm-frame-required ;
-M: ##call compute-stack-frame* drop frame-required ;
-M: ##alien-callback compute-stack-frame* drop frame-required ;
-M: ##spill compute-stack-frame* drop frame-required ;
-M: ##reload compute-stack-frame* drop frame-required ;
+M: ##call compute-stack-frame* drop t ;
+M: ##spill compute-stack-frame* drop t ;
+M: ##reload compute-stack-frame* drop t ;
M: ##float>integer compute-stack-frame*
- drop integer-float-needs-stack-frame? [ frame-required ] when ;
+ drop integer-float-needs-stack-frame? ;
M: ##integer>float compute-stack-frame*
- drop integer-float-needs-stack-frame? [ frame-required ] when ;
+ drop integer-float-needs-stack-frame? ;
+
+M: insn compute-stack-frame* drop f ;
+
+: calculate-allot-area-base ( stack-frame -- n )
+ [ params>> ] [ allot-area-align>> ] bi align ;
-M: insn compute-stack-frame* drop ;
+: calculate-spill-area-base ( stack-frame -- n )
+ [ allot-area-base>> ]
+ [ allot-area-size>> + ]
+ [ spill-area-align>> ] tri align ;
-: initial-stack-frame ( -- stack-frame )
- stack-frame new cfg get spill-area-size>> >>spill-area-size ;
+: finalize-stack-frame ( stack-frame -- stack-frame )
+ dup calculate-allot-area-base >>allot-area-base
+ dup calculate-spill-area-base >>spill-area-base
+ dup stack-frame-size >>total-size ;
-: compute-stack-frame ( cfg -- )
- initial-stack-frame stack-frame set
- [ spill-area-size>> 0 > frame-required? set ]
- [ [ instructions>> [ compute-stack-frame* ] each ] each-basic-block ] bi
- stack-frame get dup stack-frame-size >>total-size drop ;
+: compute-stack-frame ( cfg -- stack-frame/f )
+ dup cfg>insns f [ compute-stack-frame* or ] reduce [
+ stack-frame>>
+ allot-area-size get >>allot-area-size
+ allot-area-align get >>allot-area-align
+ param-area-size get >>params
+ finalize-stack-frame
+ ] [ drop f ] if ;
-: build-stack-frame ( cfg -- cfg )
- [
- [ compute-stack-frame ]
- [
- frame-required? get stack-frame get f ?
- >>stack-frame
- ] bi
- ] with-scope ;
+: build-stack-frame ( cfg -- )
+ 0 param-area-size set
+ 0 allot-area-size set
+ cell allot-area-align set
+ [ compute-stack-frame ] keep stack-frame<< ;