CreateFuncs.hs

module G2.Language.CreateFuncs where

import qualified G2.Language.ExprEnv as E
import G2.Language.Naming
import G2.Language.Syntax
import G2.Language.Support

-- | Give a list of b's, and functions to generate names and expressions from those
-- b's, generates functions and puts them in the ExprEnv
-- All b/Name pairs are also stored in some s, using the provided function
-- The function to generate the expression gets the full s of b/Name pairs
createFuncs :: ExprEnv
            -> NameGen
            -> [b]
            -> s
            -> (b -> Name)
            -> (b -> Name -> s -> s)
            -> (s -> b -> NameGen -> (Expr, NameGen))
            -> (ExprEnv, NameGen, s)
createFuncs eenv ng genFrom store namef storef exprf =
    let
        --Generate names, put them in the store
        (ns, ng2) = freshSeededNames (map namef genFrom) ng
        genFromNames = zip genFrom ns

        fullStore = foldr (uncurry storef) store genFromNames

        --Generate functions, put them in the expression environment
        (exprfs, ng3) = mapNG (exprf fullStore) genFrom ng2
        eenv2 = foldr (uncurry E.insert) eenv (zip ns exprfs)

    in
    (eenv2, ng3, fullStore)