2008-03-19 17:18:03 -04:00
|
|
|
! Copyright (C) 2005, 2008 Doug Coleman.
|
|
|
|
! See http://factorcode.org/license.txt for BSD license.
|
2015-07-18 23:14:22 -04:00
|
|
|
! mersenne twister based on
|
2008-03-19 17:18:03 -04:00
|
|
|
! http://www.math.sci.hiroshima-u.ac.jp/~m-mat/MT/MT2002/CODES/mt19937ar.c
|
2014-02-07 14:56:29 -05:00
|
|
|
USING: accessors alien.c-types alien.data fry init kernel math
|
|
|
|
math.bitwise namespaces random sequences sequences.private
|
2014-02-14 21:48:25 -05:00
|
|
|
specialized-arrays system ;
|
2009-09-09 23:33:34 -04:00
|
|
|
SPECIALIZED-ARRAY: uint
|
2008-03-19 17:18:03 -04:00
|
|
|
IN: random.mersenne-twister
|
|
|
|
|
|
|
|
<PRIVATE
|
|
|
|
|
2008-12-03 11:44:41 -05:00
|
|
|
TUPLE: mersenne-twister { seq uint-array } { i fixnum } ;
|
2008-03-19 17:18:03 -04:00
|
|
|
|
2009-02-22 20:13:08 -05:00
|
|
|
CONSTANT: n 624
|
|
|
|
CONSTANT: m 397
|
2011-11-23 21:49:33 -05:00
|
|
|
CONSTANT: a uint-array{ 0 0x9908b0df }
|
2008-03-19 17:18:03 -04:00
|
|
|
|
2014-02-07 14:56:29 -05:00
|
|
|
: mt-step ( k+m k+1 k seq -- )
|
2008-04-03 14:04:24 -04:00
|
|
|
[
|
2014-02-07 14:56:29 -05:00
|
|
|
[ nth-unsafe ] curry tri@
|
|
|
|
[ 31 bits ] [ 31 mask-bit ] bi* bitor
|
|
|
|
[ 2/ ] [ 1 bitand a nth ] bi bitxor bitxor
|
2008-12-05 02:50:30 -05:00
|
|
|
] 2keep set-nth-unsafe ; inline
|
2008-03-19 17:18:03 -04:00
|
|
|
|
2014-02-09 14:00:57 -05:00
|
|
|
: mt-steps ( k+m k+1 k n seq -- )
|
2014-02-07 14:56:29 -05:00
|
|
|
[ mt-step ] curry [ 3keep [ 1 + ] tri@ ] curry times 3drop ; inline
|
|
|
|
|
2008-03-19 17:18:03 -04:00
|
|
|
: mt-generate ( mt -- )
|
2008-04-03 14:04:24 -04:00
|
|
|
[
|
2008-12-05 02:50:30 -05:00
|
|
|
seq>>
|
2014-02-07 14:56:29 -05:00
|
|
|
[ [ m 1 0 n m - ] dip mt-steps ]
|
|
|
|
[ [ 0 n m - 1 + n m - m 1 - ] dip mt-steps ]
|
|
|
|
[ [ m 1 - 0 n 1 - ] dip mt-step ]
|
|
|
|
tri
|
2008-12-03 11:44:41 -05:00
|
|
|
] [ 0 >>i drop ] bi ; inline
|
2008-03-19 17:18:03 -04:00
|
|
|
|
2008-04-03 14:04:24 -04:00
|
|
|
: init-mt-formula ( i seq -- f(seq[i]) )
|
2014-02-25 18:59:51 -05:00
|
|
|
dupd nth dup -30 shift bitxor 1812433253 * + 1 w+ ; inline
|
2008-03-19 17:18:03 -04:00
|
|
|
|
|
|
|
: init-mt-rest ( seq -- )
|
2009-08-13 20:21:44 -04:00
|
|
|
n 1 - swap '[
|
|
|
|
_ [ init-mt-formula ] [ [ 1 + ] dip set-nth ] 2bi
|
2010-01-14 10:10:13 -05:00
|
|
|
] each-integer ; inline
|
2008-03-19 17:18:03 -04:00
|
|
|
|
|
|
|
: init-mt-seq ( seed -- seq )
|
2011-09-25 14:49:27 -04:00
|
|
|
32 bits n uint <c-array>
|
2008-12-03 19:15:14 -05:00
|
|
|
[ set-first ] [ init-mt-rest ] [ ] tri ; inline
|
2008-03-19 17:18:03 -04:00
|
|
|
|
|
|
|
: mt-temper ( y -- yt )
|
|
|
|
dup -11 shift bitxor
|
2011-11-23 21:49:33 -05:00
|
|
|
dup 7 shift 0x9d2c5680 bitand bitxor
|
|
|
|
dup 15 shift 0xefc60000 bitand bitxor
|
2008-03-19 17:18:03 -04:00
|
|
|
dup -18 shift bitxor ; inline
|
|
|
|
|
2008-04-03 14:04:24 -04:00
|
|
|
: next-index ( mt -- i )
|
2008-12-05 02:50:30 -05:00
|
|
|
dup i>> dup n < [ nip ] [ drop mt-generate 0 ] if ; inline
|
2008-04-03 14:04:24 -04:00
|
|
|
|
2008-03-19 17:18:03 -04:00
|
|
|
PRIVATE>
|
|
|
|
|
|
|
|
: <mersenne-twister> ( seed -- obj )
|
2008-04-13 16:06:27 -04:00
|
|
|
init-mt-seq 0 mersenne-twister boa
|
2008-03-19 17:18:03 -04:00
|
|
|
dup mt-generate ;
|
|
|
|
|
2014-02-06 21:17:03 -05:00
|
|
|
M: mersenne-twister seed-random
|
2009-09-30 04:22:11 -04:00
|
|
|
init-mt-seq >>seq
|
2016-07-11 17:00:15 -04:00
|
|
|
dup mt-generate ;
|
2008-03-19 17:18:03 -04:00
|
|
|
|
2014-02-06 21:17:03 -05:00
|
|
|
M: mersenne-twister random-32*
|
2008-04-03 14:04:24 -04:00
|
|
|
[ next-index ]
|
2008-12-05 02:50:30 -05:00
|
|
|
[ seq>> nth-unsafe mt-temper ]
|
2009-08-13 20:21:44 -04:00
|
|
|
[ [ 1 + ] change-i drop ] tri ;
|
2008-11-06 02:30:59 -05:00
|
|
|
|
2009-09-30 04:22:11 -04:00
|
|
|
: default-mersenne-twister ( -- mersenne-twister )
|
2014-02-14 21:48:25 -05:00
|
|
|
nano-count <mersenne-twister> ;
|
2009-10-07 22:41:12 -04:00
|
|
|
|
|
|
|
[
|
|
|
|
default-mersenne-twister random-generator set-global
|
2009-10-19 22:17:02 -04:00
|
|
|
] "bootstrap.random" add-startup-hook
|