From 174af2ba568de66e0d890825b8bda930b8e7bb96 Mon Sep 17 00:00:00 2001 From: Tom Smeding Date: Mon, 10 Nov 2025 21:49:45 +0100 Subject: Move module hierarchy under CHAD. --- src/CHAD/Util/IdGen.hs | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) create mode 100644 src/CHAD/Util/IdGen.hs (limited to 'src/CHAD/Util/IdGen.hs') diff --git a/src/CHAD/Util/IdGen.hs b/src/CHAD/Util/IdGen.hs new file mode 100644 index 0000000..d4fd945 --- /dev/null +++ b/src/CHAD/Util/IdGen.hs @@ -0,0 +1,19 @@ +{-# LANGUAGE DerivingStrategies #-} +{-# LANGUAGE GeneralizedNewtypeDeriving #-} +module CHAD.Util.IdGen where + +import Control.Monad.Fix +import Control.Monad.Trans.State.Strict + + +newtype IdGen a = IdGen (State Int a) + deriving newtype (Functor, Applicative, Monad, MonadFix) + +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 -- cgit v1.2.3-70-g09d2