roguestar-engine-0.4.0.0: src/RNG.hs

-- |
-- Don't depend on any external source of psuedo-random numbers, because
-- we want to be able to save a psuedo-random seed and know that we can
-- generate the same psuedo-random sequence when we reload it, even across
-- different environments.
--
module RNG
    (mkRNG,
     RNG,
     Random(..),
     RandomGen(..))
    where

import System.Random
import Control.Arrow (first)

newtype RNG = RNG { rng_state :: Integer }

instance RandomGen RNG where
    next g = (fromInteger $ x `quot` (2^24),RNG x)
        where x = (rng_state g * 0x5DEECE66D + 0xB) `mod` (2^48)
    split g = (mkRNG $ fromIntegral x,mkRNG $ fromIntegral y)
        where (x,g') = next g
              (y,_) = next g'
    genRange _ = (0,2^24)

instance Random RNG where
    random = first (mkRNG :: Integer -> RNG) . random
    randomR _ = random 

-- |
-- Construct an RNG from a seed.
--
mkRNG :: (Integral i) => i -> RNG
mkRNG = RNG . fromIntegral . fst . next . RNG . toInteger