Language.HERMIT.Primitive.Common

Note: this module should NOT export externals. It is for common transformations needed by the other primitive modules.

Utility Transformations

applyInContextT :: Translate c m a b -> a -> Translate c m a b

Apply a transformation to a value in the current context.

Finding function calls.

callT :: TranslateH CoreExpr (CoreExpr, [CoreExpr])

Lift GHC's collectArgs

callPredT :: (Id -> [CoreExpr] -> Bool) -> TranslateH CoreExpr (CoreExpr, [CoreExpr])

callNameT :: Name -> TranslateH CoreExpr (CoreExpr, [CoreExpr])

Succeeds if we are looking at an application of given function returning zero or more arguments to which it is applied.

callSaturatedT :: TranslateH CoreExpr (CoreExpr, [CoreExpr])

Succeeds if we are looking at a fully saturated function call.

callNameG :: Name -> TranslateH CoreExpr ()

Succeeds if we are looking at an application of given function

callDataConT :: TranslateH CoreExpr (DataCon, [Type], [CoreExpr])

Succeeds if we are looking at an application of a data constructor.

callDataConNameT :: Name -> TranslateH CoreExpr (DataCon, [Type], [CoreExpr])

Succeeds if we are looking at an application of a named data constructor.

callsR :: Name -> RewriteH CoreExpr -> RewriteH Core

Apply a rewrite to all applications of a given function in a top-down manner, pruning on success.

callsT :: Name -> TranslateH CoreExpr b -> TranslateH Core [b]

Apply a translate to all applications of a given function in a top-down manner, pruning on success, collecting the results.

Collecting variables bound at a Node

progIdsT :: TranslateH CoreProg [Id]

List all identifiers bound at the top-level in a program.

consIdsT :: TranslateH CoreProg [Id]

List the identifiers bound by the top-level binding group at the head of the program.

consRecIdsT :: TranslateH CoreProg [Id]

List the identifiers bound by a recursive top-level binding group at the head of the program.

consNonRecIdT :: TranslateH CoreProg Id

Return the identifier bound by a non-recursive top-level binding at the head of the program.

bindVarsT :: TranslateH CoreBind [Var]

List all variables bound in a binding group.

nonRecVarT :: TranslateH CoreBind Var

Return the variable bound by a non-recursive let expression.

recIdsT :: TranslateH CoreBind [Id]

List all identifiers bound in a recursive binding group.

defIdT :: TranslateH CoreDef Id

Return the identifier bound by a recursive definition.

lamVarT :: TranslateH CoreExpr Var

Return the variable bound by a lambda expression.

letVarsT :: TranslateH CoreExpr [Var]

List the variables bound by a let expression.

letRecIdsT :: TranslateH CoreExpr [Id]

List the identifiers bound by a recursive let expression.

letNonRecVarT :: TranslateH CoreExpr Var

Return the variable bound by a non-recursive let expression.

caseVarsT :: TranslateH CoreExpr [Var]

List all variables bound by a case expression (in the alternatives and the wildcard binder).

caseWildIdT :: TranslateH CoreExpr Id

Return the case wildcard binder.

caseAltVarsT :: TranslateH CoreExpr [[Var]]

List the variables bound by all alternatives in a case expression.

altVarsT :: TranslateH CoreAlt [Var]

List the variables bound by a case alternative.

Finding variables bound in the Context

boundVarsT :: TranslateH a [Var]

Lifted version of boundVars.

findBoundVarT :: Name -> TranslateH a Var

Find the unique variable bound in the context that matches the given name, failing if it is not unique.

findIdT :: Name -> TranslateH a Id

Lookup the name in the HermitC first, then, failing that, in GHC's global reader environment.

findId :: Name -> HermitC -> HermitM Id

Miscallaneous

wrongExprForm :: String -> String

Constructs a common error message. Argument String should be the desired form of the expression.

nodups :: Eq a => [a] -> Bool

Determine if a list contains no duplicated elements.

mapAlts :: (CoreExpr -> CoreExpr) -> [CoreAlt] -> [CoreAlt]