factor/apps/benchmarks/vectors.factor

20 lines
576 B
Factor

USING: compiler kernel math sequences test vectors ;
! http://inferno.bell-labs.com/cm/cs/who/bwk/interps/pap.html
: fill-vector ( n -- vector )
dup <vector> swap [ dup pick set-nth ] each ;
: copy-elt ( vec-y vec-x n -- )
#! Copy nth element from vec-x to vec-y.
rot >r tuck >r nth r> r> set-nth ;
: 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 ;
: vector-benchmark ( n -- )
0 <vector> over fill-vector rot copy-vector ;
[ ] [ 400000 vector-benchmark ] unit-test