1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
{-# LANGUAGE CPP #-}
module Main (main) where
import System.Environment (getArgs)
import Data.List (foldl')
import Data.Word (Word32)
import qualified System.Random as R
import qualified System.Random.SplitMix as SM
import qualified System.Random.SplitMix32 as SM32
newGen :: a -> (a -> g) -> IO g -> IO g
#if 0
newGen _ _ new = new
#else
newGen seed mk _ = return (mk seed)
#endif
main :: IO ()
main = do
putStrLn "Summing randoms..."
getArgs >>= \args -> case args of
"splitmix" : _ -> newGen 33 SM.mkSMGen SM.newSMGen >>= \g -> print $ benchSum g SM.nextTwoWord32
"splitmix32" : _ -> newGen 33 SM32.mkSMGen SM32.newSMGen >>= \g -> print $ benchSum g SM32.nextTwoWord32
"random" : _ -> R.newStdGen >>= \g -> print $ benchSum g randomNextTwoWord32
"sm-integer" : _ -> SM.newSMGen >>= \g -> print $ benchSumInteger g (SM.nextInteger two64 (two64 * 5))
"r-integer" : _ -> R.newStdGen >>= \g -> print $ benchSumInteger g (R.randomR (two64, two64 * 5))
-- after Closure Compiler getArgs return [] always?
-- _ -> newGen 33 SM.mkSMGen SM.newSMGen >>= \g -> print $ benchSum g SM.nextTwoWord32
_ -> newGen 33 SM32.mkSMGen SM32.newSMGen >>= \g -> print $ benchSum g SM32.nextTwoWord32
benchSum :: g -> (g -> (Word32, Word32, g)) -> Word32
benchSum g next = foldl' (+) 0 $ take 10000000 $ unfoldr2 next g
benchSumInteger :: g -> (g -> (Integer, g)) -> Integer
benchSumInteger g next = foldl' (+) 0 $ take 10000000 $ unfoldr next g
-- | Infinite unfoldr with two element generator
unfoldr2 :: (s -> (a, a, s)) -> s -> [a]
unfoldr2 f = go where
go s = let (x, y, s') = f s in x : y : go s'
-- | Infinite unfoldr with one element generator
unfoldr :: (s -> (a, s)) -> s -> [a]
unfoldr f = go where
go s = let (x, s') = f s in x : go s'
randomNextTwoWord32 :: R.StdGen -> (Word32, Word32, R.StdGen)
randomNextTwoWord32 s0 = (x, y, s2) where
(x, s1) = R.random s0
(y, s2) = R.random s1
two64 :: Integer
two64 = 2 ^ (64 :: Int)
|
