factor/extra/benchmark/nsieve-bits/nsieve-bits.factor

38 lines
986 B
Factor
Raw Normal View History

2007-09-20 18:09:08 -04:00
USING: math math.parser sequences sequences.private kernel
bit-arrays make io math.ranges multiline fry locals ;
IN: benchmark.nsieve-bits
2007-09-20 18:09:08 -04:00
: clear-flags ( step i seq -- )
2dup length >= [
3drop
] [
2008-12-17 20:28:07 -05:00
f 2over set-nth-unsafe [ over + ] dip clear-flags
2008-08-24 04:59:22 -04:00
] if ; inline recursive
2007-09-20 18:09:08 -04:00
: (nsieve-bits) ( count i seq -- count )
2008-01-12 21:38:22 -05:00
2dup length < [
2007-09-20 18:09:08 -04:00
2dup nth-unsafe [
over dup 2 * pick clear-flags
[ 1 + ] 2dip ! increment count
] when [ 1 + ] dip (nsieve-bits)
2007-09-20 18:09:08 -04:00
] [
2drop
2008-08-24 04:59:22 -04:00
] if ; inline recursive
2007-09-20 18:09:08 -04:00
: nsieve-bits ( m -- count )
[ 0 2 ] dip 1 + <bit-array> dup set-bits (nsieve-bits) ;
2007-09-20 18:09:08 -04:00
: nsieve-bits. ( m -- )
[ "Primes up to " % dup # " " % nsieve-bits # ] "" make
print ; inline
2007-09-20 18:09:08 -04:00
: nsieve-bits-main ( n -- )
[ 2^ 10000 * nsieve-bits. ]
[ 1 - 2^ 10000 * nsieve-bits. ]
[ 2 - 2^ 10000 * nsieve-bits. ]
tri ;
2007-09-20 18:09:08 -04:00
: nsieve-bits-benchmark ( -- ) 11 nsieve-bits-main ;
2007-09-20 18:09:08 -04:00
MAIN: nsieve-bits-benchmark