feat(runtime): Add more functions to RandomInterval

platonic-io · Dec 16, 2021 · cd98cae · cd98cae
1 parent 97a424f
commit cd98cae
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/runtime-prototype/src/StuntDouble/Random.hs b/src/runtime-prototype/src/StuntDouble/Random.hs
@@ -5,9 +5,13 @@ module StuntDouble.Random
   , makeSeedIO
   , interval
   , exponential
+  , add
+  , detRandomInterval
+  , isLessThan
   )
   where
 
+import Data.Fixed (mod')
 import System.Random
 
 ------------------------------------------------------------------------
@@ -27,6 +31,15 @@ makeSeedIO = fmap makeSeed randomIO
 newtype RandomInterval = RandomInterval Double
   deriving Show
 
+detRandomInterval :: Double -> RandomInterval
+detRandomInterval d = RandomInterval $ mod' d 1
+
+add :: RandomInterval -> RandomInterval -> RandomInterval
+add (RandomInterval d) (RandomInterval d') = detRandomInterval $ d + d'
+
+isLessThan :: RandomInterval -> Double -> Bool
+isLessThan (RandomInterval d) d' = d < d'
+
 interval :: Seed -> (RandomInterval, Seed)
 interval seed =
   let