WIP identity analysis

1 files changed, 18 insertions, 0 deletions

diff --git a/src/Util/IdGen.hs b/src/Util/IdGen.hs
new file mode 100644
index 0000000..fcfb6e7
--- /dev/null
+++ b/src/Util/IdGen.hs
@@ -0,0 +1,18 @@
+{-# LANGUAGE DerivingStrategies #-}
+{-# LANGUAGE GeneralizedNewtypeDeriving #-}
+module Util.IdGen where
+
+import Control.Monad.Trans.State.Strict
+
+
+newtype IdGen a = IdGen (State Int a)
+  deriving newtype (Functor, Applicative, Monad)
+
+genId :: IdGen Int
+genId = IdGen (state (\i -> (i, i + 1)))
+
+runIdGen :: Int -> IdGen a -> a
+runIdGen start (IdGen m) = evalState m start
+
+runIdGen' :: Int -> IdGen a -> (a, Int)
+runIdGen' start (IdGen m) = runState m start