factor/basis/compiler/cfg/linear-scan/allocation/spilling/spilling-tests.factor

USING: assocs compiler.cfg compiler.cfg.instructions
compiler.cfg.linear-scan.allocation.spilling
compiler.cfg.linear-scan.allocation.state
compiler.cfg.linear-scan.live-intervals compiler.cfg.linear-scan.ranges
compiler.cfg.registers cpu.architecture kernel namespaces sequences
tools.test ;
IN: compiler.cfg.linear-scan.allocation.spilling.tests

: test-live-interval ( -- live-interval )
    T{ live-interval-state
       { vreg 45 }
       { spill-to T{ spill-slot { n 8 } } }
       { spill-rep double-rep }
       { start 22 }
       { end 47 }
       { ranges
         T{ slice
            { from 0 }
            { to 1 }
            { seq
              {
                  T{ live-range { from 22 } { to 47 } }
                  T{ live-range { from 67 } { to 68 } }
                  T{ live-range { from 69 } { to 72 } }
              }
            }
         }
       }
       { uses
         {
             T{ vreg-use
                { n 28 }
                { use-rep double-rep }
             }
         }
       }
    } ;

! assign-spill
{ T{ spill-slot f 0 } } [
    f f <basic-block> <cfg> cfg set
    H{ } clone spill-slots set
    H{ { 45 double-2-rep } } representations set
    test-live-interval assign-spill
    { 45 8 } spill-slots get at
] unit-test

! last-use-rep
{ double-rep } [
    test-live-interval last-use-rep
] unit-test

! trim-after-ranges
{
    T{ live-interval-state
       { ranges
         {
             T{ live-range { from 25 } { to 30 } }
             T{ live-range { from 40 } { to 50 } }
         }
       }
       { uses { T{ vreg-use { n 25 } } } }
    }
} [
    T{ live-interval-state
       { ranges
         {
             T{ live-range { from 0 } { to 10 } }
             T{ live-range { from 20 } { to 30 } }
             T{ live-range { from 40 } { to 50 } }
         }
       }
       { uses { T{ vreg-use { n 25 } } } }
    } dup trim-after-ranges
] unit-test

{
    T{ live-interval-state
       { ranges { T{ live-range { from 10 } { to 23 } } } }
       { uses { T{ vreg-use { n 10 } } } }
    }
} [
    T{ live-interval-state
       { ranges { T{ live-range { from 20 } { to 23 } } } }
       { uses { T{ vreg-use { n 10 } } } }
    }
    dup trim-after-ranges
] unit-test

! trim-before-ranges
{
    T{ live-interval-state
       { ranges
         {
             T{ live-range { from 0 } { to 10 } }
             T{ live-range { from 20 } { to 21 } }
         }
       }
       { uses { T{ vreg-use { n 20 } } } }
    }
} [
    T{ live-interval-state
       { ranges
         {
             T{ live-range { from 0 } { to 10 } }
             T{ live-range { from 20 } { to 30 } }
             T{ live-range { from 40 } { to 50 } }
         }
       }
       { uses { T{ vreg-use { n 20 } } } }
    } dup trim-before-ranges
] unit-test