factor/library/test/benchmark/vectors.factor

USE: vectors
USE: kernel
USE: math
USE: compiler
USE: test

! http://inferno.bell-labs.com/cm/cs/who/bwk/interps/pap.html

: fill-vector ( n -- vector )
    dup <vector> swap [ [ dup pick set-vector-nth ] keep ] repeat ; compiled

: copy-elt ( vec-y vec-x n -- )
    #! Copy nth element from vec-x to vec-y.
    rot >r tuck >r vector-nth r> r> set-vector-nth ; compiled

: copy-vector ( vec-y vec-x n -- )
    #! Copy first n-1 elements from vec-x to vec-y.
    [ [ >r 2dup r> copy-elt ] keep ] repeat 2drop ; compiled

: vector-benchmark ( n -- )
    0 <vector> over fill-vector rot copy-vector ; compiled

[ ] [ 4000000 vector-benchmark ] unit-test