{- (c) The University of Glasgow 2006 (c) The GRASP/AQUA Project, Glasgow University, 1992-1998 -} {-# LANGUAGE CPP, DeriveDataTypeable, FlexibleContexts #-} {-# LANGUAGE NamedFieldPuns #-} {-# LANGUAGE BangPatterns #-} -- | CoreSyn holds all the main data types for use by for the Glasgow Haskell Compiler midsection module CoreSyn ( -- * Main data types Expr(..), Alt, Bind(..), AltCon(..), Arg, Tickish(..), TickishScoping(..), TickishPlacement(..), CoreProgram, CoreExpr, CoreAlt, CoreBind, CoreArg, CoreBndr, TaggedExpr, TaggedAlt, TaggedBind, TaggedArg, TaggedBndr(..), deTagExpr, -- * In/Out type synonyms InId, InBind, InExpr, InAlt, InArg, InType, InKind, InBndr, InVar, InCoercion, InTyVar, InCoVar, OutId, OutBind, OutExpr, OutAlt, OutArg, OutType, OutKind, OutBndr, OutVar, OutCoercion, OutTyVar, OutCoVar, MOutCoercion, -- ** 'Expr' construction mkLet, mkLets, mkLetNonRec, mkLetRec, mkLams, mkApps, mkTyApps, mkCoApps, mkVarApps, mkTyArg, mkIntLit, mkIntLitInt, mkWordLit, mkWordLitWord, mkWord64LitWord64, mkInt64LitInt64, mkCharLit, mkStringLit, mkFloatLit, mkFloatLitFloat, mkDoubleLit, mkDoubleLitDouble, mkConApp, mkConApp2, mkTyBind, mkCoBind, varToCoreExpr, varsToCoreExprs, isId, cmpAltCon, cmpAlt, ltAlt, -- ** Simple 'Expr' access functions and predicates bindersOf, bindersOfBinds, rhssOfBind, rhssOfAlts, collectBinders, collectTyBinders, collectTyAndValBinders, collectNBinders, collectArgs, stripNArgs, collectArgsTicks, flattenBinds, exprToType, exprToCoercion_maybe, applyTypeToArg, isValArg, isTypeArg, isCoArg, isTyCoArg, valArgCount, valBndrCount, isRuntimeArg, isRuntimeVar, -- * Tick-related functions tickishCounts, tickishScoped, tickishScopesLike, tickishFloatable, tickishCanSplit, mkNoCount, mkNoScope, tickishIsCode, tickishPlace, tickishContains, -- * Unfolding data types Unfolding(..), UnfoldingGuidance(..), UnfoldingSource(..), -- ** Constructing 'Unfolding's noUnfolding, bootUnfolding, evaldUnfolding, mkOtherCon, unSaturatedOk, needSaturated, boringCxtOk, boringCxtNotOk, -- ** Predicates and deconstruction on 'Unfolding' unfoldingTemplate, expandUnfolding_maybe, maybeUnfoldingTemplate, otherCons, isValueUnfolding, isEvaldUnfolding, isCheapUnfolding, isExpandableUnfolding, isConLikeUnfolding, isCompulsoryUnfolding, isStableUnfolding, isFragileUnfolding, hasSomeUnfolding, isBootUnfolding, canUnfold, neverUnfoldGuidance, isStableSource, -- * Annotated expression data types AnnExpr, AnnExpr'(..), AnnBind(..), AnnAlt, -- ** Operations on annotated expressions collectAnnArgs, collectAnnArgsTicks, -- ** Operations on annotations deAnnotate, deAnnotate', deAnnAlt, deAnnBind, collectAnnBndrs, collectNAnnBndrs, -- * Orphanhood IsOrphan(..), isOrphan, notOrphan, chooseOrphanAnchor, -- * Core rule data types CoreRule(..), RuleBase, RuleName, RuleFun, IdUnfoldingFun, InScopeEnv, RuleEnv(..), mkRuleEnv, emptyRuleEnv, -- ** Operations on 'CoreRule's ruleArity, ruleName, ruleIdName, ruleActivation, setRuleIdName, ruleModule, isBuiltinRule, isLocalRule, isAutoRule, ) where #include "HsVersions.h" import GhcPrelude import CostCentre import VarEnv( InScopeSet ) import Var import Type import Coercion import Name import NameSet import NameEnv( NameEnv, emptyNameEnv ) import Literal import DataCon import Module import BasicTypes import DynFlags import Outputable import Util import UniqSet import SrcLoc ( RealSrcSpan, containsSpan ) import Binary import Data.Data hiding (TyCon) import Data.Int import Data.Word infixl 4 `mkApps`, `mkTyApps`, `mkVarApps`, `App`, `mkCoApps` -- Left associative, so that we can say (f `mkTyApps` xs `mkVarApps` ys) {- ************************************************************************ * * \subsection{The main data types} * * ************************************************************************ These data types are the heart of the compiler -} -- | This is the data type that represents GHCs core intermediate language. Currently -- GHC uses System FC <https://www.microsoft.com/en-us/research/publication/system-f-with-type-equality-coercions/> for this purpose, -- which is closely related to the simpler and better known System F <http://en.wikipedia.org/wiki/System_F>. -- -- We get from Haskell source to this Core language in a number of stages: -- -- 1. The source code is parsed into an abstract syntax tree, which is represented -- by the data type 'HsExpr.HsExpr' with the names being 'RdrName.RdrNames' -- -- 2. This syntax tree is /renamed/, which attaches a 'Unique.Unique' to every 'RdrName.RdrName' -- (yielding a 'Name.Name') to disambiguate identifiers which are lexically identical. -- For example, this program: -- -- @ -- f x = let f x = x + 1 -- in f (x - 2) -- @ -- -- Would be renamed by having 'Unique's attached so it looked something like this: -- -- @ -- f_1 x_2 = let f_3 x_4 = x_4 + 1 -- in f_3 (x_2 - 2) -- @ -- But see Note [Shadowing] below. -- -- 3. The resulting syntax tree undergoes type checking (which also deals with instantiating -- type class arguments) to yield a 'HsExpr.HsExpr' type that has 'Id.Id' as it's names. -- -- 4. Finally the syntax tree is /desugared/ from the expressive 'HsExpr.HsExpr' type into -- this 'Expr' type, which has far fewer constructors and hence is easier to perform -- optimization, analysis and code generation on. -- -- The type parameter @b@ is for the type of binders in the expression tree. -- -- The language consists of the following elements: -- -- * Variables -- See Note [Variable occurrences in Core] -- -- * Primitive literals -- -- * Applications: note that the argument may be a 'Type'. -- See Note [CoreSyn let/app invariant] -- See Note [Levity polymorphism invariants] -- -- * Lambda abstraction -- See Note [Levity polymorphism invariants] -- -- * Recursive and non recursive @let@s. Operationally -- this corresponds to allocating a thunk for the things -- bound and then executing the sub-expression. -- -- See Note [CoreSyn letrec invariant] -- See Note [CoreSyn let/app invariant] -- See Note [Levity polymorphism invariants] -- See Note [CoreSyn type and coercion invariant] -- -- * Case expression. Operationally this corresponds to evaluating -- the scrutinee (expression examined) to weak head normal form -- and then examining at most one level of resulting constructor (i.e. you -- cannot do nested pattern matching directly with this). -- -- The binder gets bound to the value of the scrutinee, -- and the 'Type' must be that of all the case alternatives -- -- #case_invariants# -- This is one of the more complicated elements of the Core language, -- and comes with a number of restrictions: -- -- 1. The list of alternatives may be empty; -- See Note [Empty case alternatives] -- -- 2. The 'DEFAULT' case alternative must be first in the list, -- if it occurs at all. -- -- 3. The remaining cases are in order of increasing -- tag (for 'DataAlts') or -- lit (for 'LitAlts'). -- This makes finding the relevant constructor easy, -- and makes comparison easier too. -- -- 4. The list of alternatives must be exhaustive. An /exhaustive/ case -- does not necessarily mention all constructors: -- -- @ -- data Foo = Red | Green | Blue -- ... case x of -- Red -> True -- other -> f (case x of -- Green -> ... -- Blue -> ... ) ... -- @ -- -- The inner case does not need a @Red@ alternative, because @x@ -- can't be @Red@ at that program point. -- -- 5. Floating-point values must not be scrutinised against literals. -- See Trac #9238 and Note [Rules for floating-point comparisons] -- in PrelRules for rationale. -- -- * Cast an expression to a particular type. -- This is used to implement @newtype@s (a @newtype@ constructor or -- destructor just becomes a 'Cast' in Core) and GADTs. -- -- * Notes. These allow general information to be added to expressions -- in the syntax tree -- -- * A type: this should only show up at the top level of an Arg -- -- * A coercion -- If you edit this type, you may need to update the GHC formalism -- See Note [GHC Formalism] in coreSyn/CoreLint.hs data Expr b = Var Id | Lit Literal | App (Expr b) (Arg b) | Lam b (Expr b) | Let (Bind b) (Expr b) | Case (Expr b) b Type [Alt b] -- See #case_invariants# | Cast (Expr b) Coercion | Tick (Tickish Id) (Expr b) | Type Type | Coercion Coercion deriving Typeable (Expr b) DataType Constr Typeable (Expr b) => (forall (c :: * -> *). (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Expr b -> c (Expr b)) -> (forall (c :: * -> *). (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Expr b)) -> (Expr b -> Constr) -> (Expr b -> DataType) -> (forall (t :: * -> *) (c :: * -> *). Typeable t => (forall d. Data d => c (t d)) -> Maybe (c (Expr b))) -> (forall (t :: * -> * -> *) (c :: * -> *). Typeable t => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c (Expr b))) -> ((forall b. Data b => b -> b) -> Expr b -> Expr b) -> (forall r r'. (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Expr b -> r) -> (forall r r'. (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Expr b -> r) -> (forall u. (forall d. Data d => d -> u) -> Expr b -> [u]) -> (forall u. Int -> (forall d. Data d => d -> u) -> Expr b -> u) -> (forall (m :: * -> *). Monad m => (forall d. Data d => d -> m d) -> Expr b -> m (Expr b)) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> Expr b -> m (Expr b)) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> Expr b -> m (Expr b)) -> Data (Expr b) Expr b -> DataType Expr b -> Constr (forall d. Data d => c (t d)) -> Maybe (c (Expr b)) (forall b. Data b => b -> b) -> Expr b -> Expr b (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Expr b -> c (Expr b) (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Expr b) forall b. Data b => Typeable (Expr b) forall b. Data b => Expr b -> DataType forall b. Data b => Expr b -> Constr forall b. Data b => (forall b. Data b => b -> b) -> Expr b -> Expr b forall b u. Data b => Int -> (forall d. Data d => d -> u) -> Expr b -> u forall b u. Data b => (forall d. Data d => d -> u) -> Expr b -> [u] forall b r r'. Data b => (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Expr b -> r forall b r r'. Data b => (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Expr b -> r forall b (m :: * -> *). (Data b, Monad m) => (forall d. Data d => d -> m d) -> Expr b -> m (Expr b) forall b (m :: * -> *). (Data b, MonadPlus m) => (forall d. Data d => d -> m d) -> Expr b -> m (Expr b) forall b (c :: * -> *). Data b => (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Expr b) forall b (c :: * -> *). Data b => (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Expr b -> c (Expr b) forall b (t :: * -> *) (c :: * -> *). (Data b, Typeable t) => (forall d. Data d => c (t d)) -> Maybe (c (Expr b)) forall b (t :: * -> * -> *) (c :: * -> *). (Data b, Typeable t) => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c (Expr b)) forall a. Typeable a => (forall (c :: * -> *). (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> a -> c a) -> (forall (c :: * -> *). (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c a) -> (a -> Constr) -> (a -> DataType) -> (forall (t :: * -> *) (c :: * -> *). Typeable t => (forall d. Data d => c (t d)) -> Maybe (c a)) -> (forall (t :: * -> * -> *) (c :: * -> *). Typeable t => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c a)) -> ((forall b. Data b => b -> b) -> a -> a) -> (forall r r'. (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> a -> r) -> (forall r r'. (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> a -> r) -> (forall u. (forall d. Data d => d -> u) -> a -> [u]) -> (forall u. Int -> (forall d. Data d => d -> u) -> a -> u) -> (forall (m :: * -> *). Monad m => (forall d. Data d => d -> m d) -> a -> m a) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> a -> m a) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> a -> m a) -> Data a forall u. Int -> (forall d. Data d => d -> u) -> Expr b -> u forall u. (forall d. Data d => d -> u) -> Expr b -> [u] forall r r'. (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Expr b -> r forall r r'. (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Expr b -> r forall (m :: * -> *). Monad m => (forall d. Data d => d -> m d) -> Expr b -> m (Expr b) forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> Expr b -> m (Expr b) forall (c :: * -> *). (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Expr b) forall (c :: * -> *). (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Expr b -> c (Expr b) forall (t :: * -> *) (c :: * -> *). Typeable t => (forall d. Data d => c (t d)) -> Maybe (c (Expr b)) forall (t :: * -> * -> *) (c :: * -> *). Typeable t => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c (Expr b)) $cCoercion :: Constr $cType :: Constr $cTick :: Constr $cCast :: Constr $cCase :: Constr $cLet :: Constr $cLam :: Constr $cApp :: Constr $cLit :: Constr $cVar :: Constr $tExpr :: DataType gmapMo :: (forall d. Data d => d -> m d) -> Expr b -> m (Expr b) $cgmapMo :: forall b (m :: * -> *). (Data b, MonadPlus m) => (forall d. Data d => d -> m d) -> Expr b -> m (Expr b) gmapMp :: (forall d. Data d => d -> m d) -> Expr b -> m (Expr b) $cgmapMp :: forall b (m :: * -> *). (Data b, MonadPlus m) => (forall d. Data d => d -> m d) -> Expr b -> m (Expr b) gmapM :: (forall d. Data d => d -> m d) -> Expr b -> m (Expr b) $cgmapM :: forall b (m :: * -> *). (Data b, Monad m) => (forall d. Data d => d -> m d) -> Expr b -> m (Expr b) gmapQi :: Int -> (forall d. Data d => d -> u) -> Expr b -> u $cgmapQi :: forall b u. Data b => Int -> (forall d. Data d => d -> u) -> Expr b -> u gmapQ :: (forall d. Data d => d -> u) -> Expr b -> [u] $cgmapQ :: forall b u. Data b => (forall d. Data d => d -> u) -> Expr b -> [u] gmapQr :: (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Expr b -> r $cgmapQr :: forall b r r'. Data b => (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Expr b -> r gmapQl :: (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Expr b -> r $cgmapQl :: forall b r r'. Data b => (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Expr b -> r gmapT :: (forall b. Data b => b -> b) -> Expr b -> Expr b $cgmapT :: forall b. Data b => (forall b. Data b => b -> b) -> Expr b -> Expr b dataCast2 :: (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c (Expr b)) $cdataCast2 :: forall b (t :: * -> * -> *) (c :: * -> *). (Data b, Typeable t) => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c (Expr b)) dataCast1 :: (forall d. Data d => c (t d)) -> Maybe (c (Expr b)) $cdataCast1 :: forall b (t :: * -> *) (c :: * -> *). (Data b, Typeable t) => (forall d. Data d => c (t d)) -> Maybe (c (Expr b)) dataTypeOf :: Expr b -> DataType $cdataTypeOf :: forall b. Data b => Expr b -> DataType toConstr :: Expr b -> Constr $ctoConstr :: forall b. Data b => Expr b -> Constr gunfold :: (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Expr b) $cgunfold :: forall b (c :: * -> *). Data b => (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Expr b) gfoldl :: (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Expr b -> c (Expr b) $cgfoldl :: forall b (c :: * -> *). Data b => (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Expr b -> c (Expr b) $cp1Data :: forall b. Data b => Typeable (Expr b) Data -- | Type synonym for expressions that occur in function argument positions. -- Only 'Arg' should contain a 'Type' at top level, general 'Expr' should not type Arg b = Expr b -- | A case split alternative. Consists of the constructor leading to the alternative, -- the variables bound from the constructor, and the expression to be executed given that binding. -- The default alternative is @(DEFAULT, [], rhs)@ -- If you edit this type, you may need to update the GHC formalism -- See Note [GHC Formalism] in coreSyn/CoreLint.hs type Alt b = (AltCon, [b], Expr b) -- | A case alternative constructor (i.e. pattern match) -- If you edit this type, you may need to update the GHC formalism -- See Note [GHC Formalism] in coreSyn/CoreLint.hs data AltCon = DataAlt DataCon -- ^ A plain data constructor: @case e of { Foo x -> ... }@. -- Invariant: the 'DataCon' is always from a @data@ type, and never from a @newtype@ | LitAlt Literal -- ^ A literal: @case e of { 1 -> ... }@ -- Invariant: always an *unlifted* literal -- See Note [Literal alternatives] | DEFAULT -- ^ Trivial alternative: @case e of { _ -> ... }@ deriving (AltCon -> AltCon -> Bool (AltCon -> AltCon -> Bool) -> (AltCon -> AltCon -> Bool) -> Eq AltCon forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a /= :: AltCon -> AltCon -> Bool $c/= :: AltCon -> AltCon -> Bool == :: AltCon -> AltCon -> Bool $c== :: AltCon -> AltCon -> Bool Eq, Typeable AltCon DataType Constr Typeable AltCon => (forall (c :: * -> *). (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> AltCon -> c AltCon) -> (forall (c :: * -> *). (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c AltCon) -> (AltCon -> Constr) -> (AltCon -> DataType) -> (forall (t :: * -> *) (c :: * -> *). Typeable t => (forall d. Data d => c (t d)) -> Maybe (c AltCon)) -> (forall (t :: * -> * -> *) (c :: * -> *). Typeable t => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c AltCon)) -> ((forall b. Data b => b -> b) -> AltCon -> AltCon) -> (forall r r'. (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> AltCon -> r) -> (forall r r'. (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> AltCon -> r) -> (forall u. (forall d. Data d => d -> u) -> AltCon -> [u]) -> (forall u. Int -> (forall d. Data d => d -> u) -> AltCon -> u) -> (forall (m :: * -> *). Monad m => (forall d. Data d => d -> m d) -> AltCon -> m AltCon) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> AltCon -> m AltCon) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> AltCon -> m AltCon) -> Data AltCon AltCon -> DataType AltCon -> Constr (forall b. Data b => b -> b) -> AltCon -> AltCon (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> AltCon -> c AltCon (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c AltCon forall a. Typeable a => (forall (c :: * -> *). (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> a -> c a) -> (forall (c :: * -> *). (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c a) -> (a -> Constr) -> (a -> DataType) -> (forall (t :: * -> *) (c :: * -> *). Typeable t => (forall d. Data d => c (t d)) -> Maybe (c a)) -> (forall (t :: * -> * -> *) (c :: * -> *). Typeable t => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c a)) -> ((forall b. Data b => b -> b) -> a -> a) -> (forall r r'. (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> a -> r) -> (forall r r'. (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> a -> r) -> (forall u. (forall d. Data d => d -> u) -> a -> [u]) -> (forall u. Int -> (forall d. Data d => d -> u) -> a -> u) -> (forall (m :: * -> *). Monad m => (forall d. Data d => d -> m d) -> a -> m a) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> a -> m a) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> a -> m a) -> Data a forall u. Int -> (forall d. Data d => d -> u) -> AltCon -> u forall u. (forall d. Data d => d -> u) -> AltCon -> [u] forall r r'. (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> AltCon -> r forall r r'. (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> AltCon -> r forall (m :: * -> *). Monad m => (forall d. Data d => d -> m d) -> AltCon -> m AltCon forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> AltCon -> m AltCon forall (c :: * -> *). (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c AltCon forall (c :: * -> *). (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> AltCon -> c AltCon forall (t :: * -> *) (c :: * -> *). Typeable t => (forall d. Data d => c (t d)) -> Maybe (c AltCon) forall (t :: * -> * -> *) (c :: * -> *). Typeable t => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c AltCon) $cDEFAULT :: Constr $cLitAlt :: Constr $cDataAlt :: Constr $tAltCon :: DataType gmapMo :: (forall d. Data d => d -> m d) -> AltCon -> m AltCon $cgmapMo :: forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> AltCon -> m AltCon gmapMp :: (forall d. Data d => d -> m d) -> AltCon -> m AltCon $cgmapMp :: forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> AltCon -> m AltCon gmapM :: (forall d. Data d => d -> m d) -> AltCon -> m AltCon $cgmapM :: forall (m :: * -> *). Monad m => (forall d. Data d => d -> m d) -> AltCon -> m AltCon gmapQi :: Int -> (forall d. Data d => d -> u) -> AltCon -> u $cgmapQi :: forall u. Int -> (forall d. Data d => d -> u) -> AltCon -> u gmapQ :: (forall d. Data d => d -> u) -> AltCon -> [u] $cgmapQ :: forall u. (forall d. Data d => d -> u) -> AltCon -> [u] gmapQr :: (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> AltCon -> r $cgmapQr :: forall r r'. (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> AltCon -> r gmapQl :: (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> AltCon -> r $cgmapQl :: forall r r'. (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> AltCon -> r gmapT :: (forall b. Data b => b -> b) -> AltCon -> AltCon $cgmapT :: (forall b. Data b => b -> b) -> AltCon -> AltCon dataCast2 :: (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c AltCon) $cdataCast2 :: forall (t :: * -> * -> *) (c :: * -> *). Typeable t => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c AltCon) dataCast1 :: (forall d. Data d => c (t d)) -> Maybe (c AltCon) $cdataCast1 :: forall (t :: * -> *) (c :: * -> *). Typeable t => (forall d. Data d => c (t d)) -> Maybe (c AltCon) dataTypeOf :: AltCon -> DataType $cdataTypeOf :: AltCon -> DataType toConstr :: AltCon -> Constr $ctoConstr :: AltCon -> Constr gunfold :: (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c AltCon $cgunfold :: forall (c :: * -> *). (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c AltCon gfoldl :: (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> AltCon -> c AltCon $cgfoldl :: forall (c :: * -> *). (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> AltCon -> c AltCon $cp1Data :: Typeable AltCon Data) -- This instance is a bit shady. It can only be used to compare AltCons for -- a single type constructor. Fortunately, it seems quite unlikely that we'll -- ever need to compare AltCons for different type constructors. -- The instance adheres to the order described in [CoreSyn case invariants] instance Ord AltCon where compare :: AltCon -> AltCon -> Ordering compare (DataAlt con1 :: DataCon con1) (DataAlt con2 :: DataCon con2) = ASSERT( dataConTyCon con1 == dataConTyCon con2 ) Int -> Int -> Ordering forall a. Ord a => a -> a -> Ordering compare (DataCon -> Int dataConTag DataCon con1) (DataCon -> Int dataConTag DataCon con2) compare (DataAlt _) _ = Ordering GT compare _ (DataAlt _) = Ordering LT compare (LitAlt l1 :: Literal l1) (LitAlt l2 :: Literal l2) = Literal -> Literal -> Ordering forall a. Ord a => a -> a -> Ordering compare Literal l1 Literal l2 compare (LitAlt _) DEFAULT = Ordering GT compare DEFAULT DEFAULT = Ordering EQ compare DEFAULT _ = Ordering LT -- | Binding, used for top level bindings in a module and local bindings in a @let@. -- If you edit this type, you may need to update the GHC formalism -- See Note [GHC Formalism] in coreSyn/CoreLint.hs data Bind b = NonRec b (Expr b) | Rec [(b, (Expr b))] deriving Typeable (Bind b) DataType Constr Typeable (Bind b) => (forall (c :: * -> *). (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Bind b -> c (Bind b)) -> (forall (c :: * -> *). (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Bind b)) -> (Bind b -> Constr) -> (Bind b -> DataType) -> (forall (t :: * -> *) (c :: * -> *). Typeable t => (forall d. Data d => c (t d)) -> Maybe (c (Bind b))) -> (forall (t :: * -> * -> *) (c :: * -> *). Typeable t => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c (Bind b))) -> ((forall b. Data b => b -> b) -> Bind b -> Bind b) -> (forall r r'. (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Bind b -> r) -> (forall r r'. (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Bind b -> r) -> (forall u. (forall d. Data d => d -> u) -> Bind b -> [u]) -> (forall u. Int -> (forall d. Data d => d -> u) -> Bind b -> u) -> (forall (m :: * -> *). Monad m => (forall d. Data d => d -> m d) -> Bind b -> m (Bind b)) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> Bind b -> m (Bind b)) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> Bind b -> m (Bind b)) -> Data (Bind b) Bind b -> DataType Bind b -> Constr (forall d. Data d => c (t d)) -> Maybe (c (Bind b)) (forall b. Data b => b -> b) -> Bind b -> Bind b (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Bind b -> c (Bind b) (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Bind b) forall b. Data b => Typeable (Bind b) forall b. Data b => Bind b -> DataType forall b. Data b => Bind b -> Constr forall b. Data b => (forall b. Data b => b -> b) -> Bind b -> Bind b forall b u. Data b => Int -> (forall d. Data d => d -> u) -> Bind b -> u forall b u. Data b => (forall d. Data d => d -> u) -> Bind b -> [u] forall b r r'. Data b => (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Bind b -> r forall b r r'. Data b => (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Bind b -> r forall b (m :: * -> *). (Data b, Monad m) => (forall d. Data d => d -> m d) -> Bind b -> m (Bind b) forall b (m :: * -> *). (Data b, MonadPlus m) => (forall d. Data d => d -> m d) -> Bind b -> m (Bind b) forall b (c :: * -> *). Data b => (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Bind b) forall b (c :: * -> *). Data b => (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Bind b -> c (Bind b) forall b (t :: * -> *) (c :: * -> *). (Data b, Typeable t) => (forall d. Data d => c (t d)) -> Maybe (c (Bind b)) forall b (t :: * -> * -> *) (c :: * -> *). (Data b, Typeable t) => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c (Bind b)) forall a. Typeable a => (forall (c :: * -> *). (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> a -> c a) -> (forall (c :: * -> *). (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c a) -> (a -> Constr) -> (a -> DataType) -> (forall (t :: * -> *) (c :: * -> *). Typeable t => (forall d. Data d => c (t d)) -> Maybe (c a)) -> (forall (t :: * -> * -> *) (c :: * -> *). Typeable t => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c a)) -> ((forall b. Data b => b -> b) -> a -> a) -> (forall r r'. (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> a -> r) -> (forall r r'. (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> a -> r) -> (forall u. (forall d. Data d => d -> u) -> a -> [u]) -> (forall u. Int -> (forall d. Data d => d -> u) -> a -> u) -> (forall (m :: * -> *). Monad m => (forall d. Data d => d -> m d) -> a -> m a) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> a -> m a) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> a -> m a) -> Data a forall u. Int -> (forall d. Data d => d -> u) -> Bind b -> u forall u. (forall d. Data d => d -> u) -> Bind b -> [u] forall r r'. (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Bind b -> r forall r r'. (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Bind b -> r forall (m :: * -> *). Monad m => (forall d. Data d => d -> m d) -> Bind b -> m (Bind b) forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> Bind b -> m (Bind b) forall (c :: * -> *). (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Bind b) forall (c :: * -> *). (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Bind b -> c (Bind b) forall (t :: * -> *) (c :: * -> *). Typeable t => (forall d. Data d => c (t d)) -> Maybe (c (Bind b)) forall (t :: * -> * -> *) (c :: * -> *). Typeable t => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c (Bind b)) $cRec :: Constr $cNonRec :: Constr $tBind :: DataType gmapMo :: (forall d. Data d => d -> m d) -> Bind b -> m (Bind b) $cgmapMo :: forall b (m :: * -> *). (Data b, MonadPlus m) => (forall d. Data d => d -> m d) -> Bind b -> m (Bind b) gmapMp :: (forall d. Data d => d -> m d) -> Bind b -> m (Bind b) $cgmapMp :: forall b (m :: * -> *). (Data b, MonadPlus m) => (forall d. Data d => d -> m d) -> Bind b -> m (Bind b) gmapM :: (forall d. Data d => d -> m d) -> Bind b -> m (Bind b) $cgmapM :: forall b (m :: * -> *). (Data b, Monad m) => (forall d. Data d => d -> m d) -> Bind b -> m (Bind b) gmapQi :: Int -> (forall d. Data d => d -> u) -> Bind b -> u $cgmapQi :: forall b u. Data b => Int -> (forall d. Data d => d -> u) -> Bind b -> u gmapQ :: (forall d. Data d => d -> u) -> Bind b -> [u] $cgmapQ :: forall b u. Data b => (forall d. Data d => d -> u) -> Bind b -> [u] gmapQr :: (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Bind b -> r $cgmapQr :: forall b r r'. Data b => (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Bind b -> r gmapQl :: (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Bind b -> r $cgmapQl :: forall b r r'. Data b => (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Bind b -> r gmapT :: (forall b. Data b => b -> b) -> Bind b -> Bind b $cgmapT :: forall b. Data b => (forall b. Data b => b -> b) -> Bind b -> Bind b dataCast2 :: (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c (Bind b)) $cdataCast2 :: forall b (t :: * -> * -> *) (c :: * -> *). (Data b, Typeable t) => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c (Bind b)) dataCast1 :: (forall d. Data d => c (t d)) -> Maybe (c (Bind b)) $cdataCast1 :: forall b (t :: * -> *) (c :: * -> *). (Data b, Typeable t) => (forall d. Data d => c (t d)) -> Maybe (c (Bind b)) dataTypeOf :: Bind b -> DataType $cdataTypeOf :: forall b. Data b => Bind b -> DataType toConstr :: Bind b -> Constr $ctoConstr :: forall b. Data b => Bind b -> Constr gunfold :: (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Bind b) $cgunfold :: forall b (c :: * -> *). Data b => (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Bind b) gfoldl :: (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Bind b -> c (Bind b) $cgfoldl :: forall b (c :: * -> *). Data b => (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Bind b -> c (Bind b) $cp1Data :: forall b. Data b => Typeable (Bind b) Data {- Note [Shadowing] ~~~~~~~~~~~~~~~~ While various passes attempt to rename on-the-fly in a manner that avoids "shadowing" (thereby simplifying downstream optimizations), neither the simplifier nor any other pass GUARANTEES that shadowing is avoided. Thus, all passes SHOULD work fine even in the presence of arbitrary shadowing in their inputs. In particular, scrutinee variables `x` in expressions of the form `Case e x t` are often renamed to variables with a prefix "wild_". These "wild" variables may appear in the body of the case-expression, and further, may be shadowed within the body. So the Unique in a Var is not really unique at all. Still, it's very useful to give a constant-time equality/ordering for Vars, and to give a key that can be used to make sets of Vars (VarSet), or mappings from Vars to other things (VarEnv). Moreover, if you do want to eliminate shadowing, you can give a new Unique to an Id without changing its printable name, which makes debugging easier. Note [Literal alternatives] ~~~~~~~~~~~~~~~~~~~~~~~~~~~ Literal alternatives (LitAlt lit) are always for *un-lifted* literals. We have one literal, a literal Integer, that is lifted, and we don't allow in a LitAlt, because LitAlt cases don't do any evaluation. Also (see Trac #5603) if you say case 3 of S# x -> ... J# _ _ -> ... (where S#, J# are the constructors for Integer) we don't want the simplifier calling findAlt with argument (LitAlt 3). No no. Integer literals are an opaque encoding of an algebraic data type, not of an unlifted literal, like all the others. Also, we do not permit case analysis with literal patterns on floating-point types. See Trac #9238 and Note [Rules for floating-point comparisons] in PrelRules for the rationale for this restriction. -------------------------- CoreSyn INVARIANTS --------------------------- Note [Variable occurrences in Core] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ Variable /occurrences/ are never CoVars, though /bindings/ can be. All CoVars appear in Coercions. For example \(c :: Age~#Int) (d::Int). d |> (sym c) Here 'c' is a CoVar, which is lambda-bound, but it /occurs/ in a Coercion, (sym c). Note [CoreSyn letrec invariant] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ The right hand sides of all top-level and recursive @let@s /must/ be of lifted type (see "Type#type_classification" for the meaning of /lifted/ vs. /unlifted/). There is one exception to this rule, top-level @let@s are allowed to bind primitive string literals: see Note [CoreSyn top-level string literals]. Note [CoreSyn top-level string literals] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ As an exception to the usual rule that top-level binders must be lifted, we allow binding primitive string literals (of type Addr#) of type Addr# at the top level. This allows us to share string literals earlier in the pipeline and crucially allows other optimizations in the Core2Core pipeline to fire. Consider, f n = let a::Addr# = "foo"# in \x -> blah In order to be able to inline `f`, we would like to float `a` to the top. Another option would be to inline `a`, but that would lead to duplicating string literals, which we want to avoid. See Trac #8472. The solution is simply to allow top-level unlifted binders. We can't allow arbitrary unlifted expression at the top-level though, unlifted binders cannot be thunks, so we just allow string literals. We allow the top-level primitive string literals to be wrapped in Ticks in the same way they can be wrapped when nested in an expression. CoreToSTG currently discards Ticks around top-level primitive string literals. See Trac #14779. Also see Note [Compilation plan for top-level string literals]. Note [Compilation plan for top-level string literals] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ Here is a summary on how top-level string literals are handled by various parts of the compilation pipeline. * In the source language, there is no way to bind a primitive string literal at the top level. * In Core, we have a special rule that permits top-level Addr# bindings. See Note [CoreSyn top-level string literals]. Core-to-core passes may introduce new top-level string literals. * In STG, top-level string literals are explicitly represented in the syntax tree. * A top-level string literal may end up exported from a module. In this case, in the object file, the content of the exported literal is given a label with the _bytes suffix. Note [CoreSyn let/app invariant] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ The let/app invariant the right hand side of a non-recursive 'Let', and the argument of an 'App', /may/ be of unlifted type, but only if the expression is ok-for-speculation or the 'Let' is for a join point. This means that the let can be floated around without difficulty. For example, this is OK: y::Int# = x +# 1# But this is not, as it may affect termination if the expression is floated out: y::Int# = fac 4# In this situation you should use @case@ rather than a @let@. The function 'CoreUtils.needsCaseBinding' can help you determine which to generate, or alternatively use 'MkCore.mkCoreLet' rather than this constructor directly, which will generate a @case@ if necessary The let/app invariant is initially enforced by mkCoreLet and mkCoreApp in coreSyn/MkCore. For discussion of some implications of the let/app invariant primops see Note [Checking versus non-checking primops] in PrimOp. Note [CoreSyn type and coercion invariant] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ We allow a /non-recursive/, /non-top-level/ let to bind type and coercion variables. These can be very convenient for postponing type substitutions until the next run of the simplifier. * A type variable binding must have a RHS of (Type ty) * A coercion variable binding must have a RHS of (Coercion co) It is possible to have terms that return a coercion, but we use case-binding for those; e.g. case (eq_sel d) of (co :: a ~# b) -> blah where eq_sel :: (a~b) -> (a~#b) Or even even case (df @Int) of (co :: a ~# b) -> blah Which is very exotic, and I think never encountered; but see Note [Equality superclasses in quantified constraints] in TcCanonical Note [CoreSyn case invariants] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ See #case_invariants# Note [Levity polymorphism invariants] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ The levity-polymorphism invariants are these (as per "Levity Polymorphism", PLDI '17): * The type of a term-binder must not be levity-polymorphic, unless it is a let(rec)-bound join point (see Note [Invariants on join points]) * The type of the argument of an App must not be levity-polymorphic. A type (t::TYPE r) is "levity polymorphic" if 'r' has any free variables. For example \(r::RuntimeRep). \(a::TYPE r). \(x::a). e is illegal because x's type has kind (TYPE r), which has 'r' free. See Note [Levity polymorphism checking] in DsMonad to see where these invariants are established for user-written code. Note [CoreSyn let goal] ~~~~~~~~~~~~~~~~~~~~~~~ * The simplifier tries to ensure that if the RHS of a let is a constructor application, its arguments are trivial, so that the constructor can be inlined vigorously. Note [Type let] ~~~~~~~~~~~~~~~ See #type_let# Note [Empty case alternatives] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ The alternatives of a case expression should be exhaustive. But this exhaustive list can be empty! * A case expression can have empty alternatives if (and only if) the scrutinee is bound to raise an exception or diverge. When do we know this? See Note [Bottoming expressions] in CoreUtils. * The possibility of empty alternatives is one reason we need a type on the case expression: if the alternatives are empty we can't get the type from the alternatives! * In the case of empty types (see Note [Bottoming expressions]), say data T we do NOT want to replace case (x::T) of Bool {} --> error Bool "Inaccessible case" because x might raise an exception, and *that*'s what we want to see! (Trac #6067 is an example.) To preserve semantics we'd have to say x `seq` error Bool "Inaccessible case" but the 'seq' is just a case, so we are back to square 1. Or I suppose we could say x |> UnsafeCoerce T Bool but that loses all trace of the fact that this originated with an empty set of alternatives. * We can use the empty-alternative construct to coerce error values from one type to another. For example f :: Int -> Int f n = error "urk" g :: Int -> (# Char, Bool #) g x = case f x of { 0 -> ..., n -> ... } Then if we inline f in g's RHS we get case (error Int "urk") of (# Char, Bool #) { ... } and we can discard the alternatives since the scrutinee is bottom to give case (error Int "urk") of (# Char, Bool #) {} This is nicer than using an unsafe coerce between Int ~ (# Char,Bool #), if for no other reason that we don't need to instantiate the (~) at an unboxed type. * We treat a case expression with empty alternatives as trivial iff its scrutinee is (see CoreUtils.exprIsTrivial). This is actually important; see Note [Empty case is trivial] in CoreUtils * An empty case is replaced by its scrutinee during the CoreToStg conversion; remember STG is un-typed, so there is no need for the empty case to do the type conversion. Note [Join points] ~~~~~~~~~~~~~~~~~~ In Core, a *join point* is a specially tagged function whose only occurrences are saturated tail calls. A tail call can appear in these places: 1. In the branches (not the scrutinee) of a case 2. Underneath a let (value or join point) 3. Inside another join point We write a join-point declaration as join j @a @b x y = e1 in e2, like a let binding but with "join" instead (or "join rec" for "let rec"). Note that we put the parameters before the = rather than using lambdas; this is because it's relevant how many parameters the join point takes *as a join point.* This number is called the *join arity,* distinct from arity because it counts types as well as values. Note that a join point may return a lambda! So join j x = x + 1 is different from join j = \x -> x + 1 The former has join arity 1, while the latter has join arity 0. The identifier for a join point is called a join id or a *label.* An invocation is called a *jump.* We write a jump using the jump keyword: jump j 3 The words *label* and *jump* are evocative of assembly code (or Cmm) for a reason: join points are indeed compiled as labeled blocks, and jumps become actual jumps (plus argument passing and stack adjustment). There is no closure allocated and only a fraction of the function-call overhead. Hence we would like as many functions as possible to become join points (see OccurAnal) and the type rules for join points ensure we preserve the properties that make them efficient. In the actual AST, a join point is indicated by the IdDetails of the binder: a local value binding gets 'VanillaId' but a join point gets a 'JoinId' with its join arity. For more details, see the paper: Luke Maurer, Paul Downen, Zena Ariola, and Simon Peyton Jones. "Compiling without continuations." Submitted to PLDI'17. https://www.microsoft.com/en-us/research/publication/compiling-without-continuations/ Note [Invariants on join points] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ Join points must follow these invariants: 1. All occurrences must be tail calls. Each of these tail calls must pass the same number of arguments, counting both types and values; we call this the "join arity" (to distinguish from regular arity, which only counts values). See Note [Join points are less general than the paper] 2. For join arity n, the right-hand side must begin with at least n lambdas. No ticks, no casts, just lambdas! C.f. CoreUtils.joinRhsArity. 2a. Moreover, this same constraint applies to any unfolding of the binder. Reason: if we want to push a continuation into the RHS we must push it into the unfolding as well. 3. If the binding is recursive, then all other bindings in the recursive group must also be join points. 4. The binding's type must not be polymorphic in its return type (as defined in Note [The polymorphism rule of join points]). However, join points have simpler invariants in other ways 5. A join point can have an unboxed type without the RHS being ok-for-speculation (i.e. drop the let/app invariant) e.g. let j :: Int# = factorial x in ... 6. A join point can have a levity-polymorphic RHS e.g. let j :: r :: TYPE l = fail void# in ... This happened in an intermediate program Trac #13394 Examples: join j1 x = 1 + x in jump j (jump j x) -- Fails 1: non-tail call join j1' x = 1 + x in if even a then jump j1 a else jump j1 a b -- Fails 1: inconsistent calls join j2 x = flip (+) x in j2 1 2 -- Fails 2: not enough lambdas join j2' x = \y -> x + y in j3 1 -- Passes: extra lams ok join j @a (x :: a) = x -- Fails 4: polymorphic in ret type Invariant 1 applies to left-hand sides of rewrite rules, so a rule for a join point must have an exact call as its LHS. Strictly speaking, invariant 3 is redundant, since a call from inside a lazy binding isn't a tail call. Since a let-bound value can't invoke a free join point, then, they can't be mutually recursive. (A Core binding group *can* include spurious extra bindings if the occurrence analyser hasn't run, so invariant 3 does still need to be checked.) For the rigorous definition of "tail call", see Section 3 of the paper (Note [Join points]). Invariant 4 is subtle; see Note [The polymorphism rule of join points]. Core Lint will check these invariants, anticipating that any binder whose OccInfo is marked AlwaysTailCalled will become a join point as soon as the simplifier (or simpleOptPgm) runs. Note [Join points are less general than the paper] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ In the paper "Compiling without continuations", this expression is perfectly valid: join { j = \_ -> e } in (case blah of ) ( True -> j void# ) arg ( False -> blah ) assuming 'j' has arity 1. Here the call to 'j' does not look like a tail call, but actually everything is fine. See Section 3, "Managing \Delta" in the paper. In GHC, however, we adopt a slightly more restrictive subset, in which join point calls must be tail calls. I think we /could/ loosen it up, but in fact the simplifier ensures that we always get tail calls, and it makes the back end a bit easier I think. Generally, just less to think about; nothing deeper than that. Note [The type of a join point] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ A join point has the same type it would have as a function. That is, if it takes an Int and a Bool and its body produces a String, its type is `Int -> Bool -> String`. Natural as this may seem, it can be awkward. A join point shouldn't be thought to "return" in the same sense a function does---a jump is one-way. This is crucial for understanding how case-of-case interacts with join points: case (join j :: Int -> Bool -> String j x y = ... in jump j z w) of "" -> True _ -> False The simplifier will pull the case into the join point (see Note [Case-of-case and join points] in Simplify): join j :: Int -> Bool -> Bool -- changed! j x y = case ... of "" -> True _ -> False in jump j z w The body of the join point now returns a Bool, so the label `j` has to have its type updated accordingly. Inconvenient though this may be, it has the advantage that 'CoreUtils.exprType' can still return a type for any expression, including a jump. This differs from the paper (see Note [Invariants on join points]). In the paper, we instead give j the type `Int -> Bool -> forall a. a`. Then each jump carries the "return type" as a parameter, exactly the way other non-returning functions like `error` work: case (join j :: Int -> Bool -> forall a. a j x y = ... in jump j z w @String) of "" -> True _ -> False Now we can move the case inward and we only have to change the jump: join j :: Int -> Bool -> forall a. a j x y = case ... of "" -> True _ -> False in jump j z w @Bool (Core Lint would still check that the body of the join point has the right type; that type would simply not be reflected in the join id.) Note [The polymorphism rule of join points] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ Invariant 4 of Note [Invariants on join points] forbids a join point to be polymorphic in its return type. That is, if its type is forall a1 ... ak. t1 -> ... -> tn -> r where its join arity is k+n, none of the type parameters ai may occur free in r. In some way, this falls out of the fact that given join j @a1 ... @ak x1 ... xn = e1 in e2 then all calls to `j` are in tail-call positions of `e`, and expressions in tail-call positions in `e` have the same type as `e`. Therefore the type of `e1` -- the return type of the join point -- must be the same as the type of e2. Since the type variables aren't bound in `e2`, its type can't include them, and thus neither can the type of `e1`. This unfortunately prevents the `go` in the following code from being a join-point: iter :: forall a. Int -> (a -> a) -> a -> a iter @a n f x = go @a n f x where go :: forall a. Int -> (a -> a) -> a -> a go @a 0 _ x = x go @a n f x = go @a (n-1) f (f x) In this case, a static argument transformation would fix that (see ticket #14620): iter :: forall a. Int -> (a -> a) -> a -> a iter @a n f x = go' @a n f x where go' :: Int -> (a -> a) -> a -> a go' 0 _ x = x go' n f x = go' (n-1) f (f x) In general, loopification could be employed to do that (see #14068.) Can we simply drop the requirement, and allow `go` to be a join-point? We could, and it would work. But we could not longer apply the case-of-join-point transformation universally. This transformation would do: case (join go @a n f x = case n of 0 -> x n -> go @a (n-1) f (f x) in go @Bool n neg True) of True -> e1; False -> e2 ===> join go @a n f x = case n of 0 -> case x of True -> e1; False -> e2 n -> go @a (n-1) f (f x) in go @Bool n neg True but that is ill-typed, as `x` is type `a`, not `Bool`. This also justifies why we do not consider the `e` in `e |> co` to be in tail position: A cast changes the type, but the type must be the same. But operationally, casts are vacuous, so this is a bit unfortunate! See #14610 for ideas how to fix this. ************************************************************************ * * In/Out type synonyms * * ********************************************************************* -} {- Many passes apply a substitution, and it's very handy to have type synonyms to remind us whether or not the substitution has been applied -} -- Pre-cloning or substitution type InBndr = CoreBndr type InType = Type type InKind = Kind type InBind = CoreBind type InExpr = CoreExpr type InAlt = CoreAlt type InArg = CoreArg type InCoercion = Coercion -- Post-cloning or substitution type OutBndr = CoreBndr type OutType = Type type OutKind = Kind type OutCoercion = Coercion type OutBind = CoreBind type OutExpr = CoreExpr type OutAlt = CoreAlt type OutArg = CoreArg type MOutCoercion = MCoercion {- ********************************************************************* * * Ticks * * ************************************************************************ -} -- | Allows attaching extra information to points in expressions -- If you edit this type, you may need to update the GHC formalism -- See Note [GHC Formalism] in coreSyn/CoreLint.hs data Tickish id = -- | An @{-# SCC #-}@ profiling annotation, either automatically -- added by the desugarer as a result of -auto-all, or added by -- the user. ProfNote { Tickish id -> CostCentre profNoteCC :: CostCentre, -- ^ the cost centre Tickish id -> Bool profNoteCount :: !Bool, -- ^ bump the entry count? Tickish id -> Bool profNoteScope :: !Bool -- ^ scopes over the enclosed expression -- (i.e. not just a tick) } -- | A "tick" used by HPC to track the execution of each -- subexpression in the original source code. | HpcTick { Tickish id -> Module tickModule :: Module, Tickish id -> Int tickId :: !Int } -- | A breakpoint for the GHCi debugger. This behaves like an HPC -- tick, but has a list of free variables which will be available -- for inspection in GHCi when the program stops at the breakpoint. -- -- NB. we must take account of these Ids when (a) counting free variables, -- and (b) substituting (don't substitute for them) | Breakpoint { Tickish id -> Int breakpointId :: !Int , Tickish id -> [id] breakpointFVs :: [id] -- ^ the order of this list is important: -- it matches the order of the lists in the -- appropriate entry in HscTypes.ModBreaks. -- -- Careful about substitution! See -- Note [substTickish] in CoreSubst. } -- | A source note. -- -- Source notes are pure annotations: Their presence should neither -- influence compilation nor execution. The semantics are given by -- causality: The presence of a source note means that a local -- change in the referenced source code span will possibly provoke -- the generated code to change. On the flip-side, the functionality -- of annotated code *must* be invariant against changes to all -- source code *except* the spans referenced in the source notes -- (see "Causality of optimized Haskell" paper for details). -- -- Therefore extending the scope of any given source note is always -- valid. Note that it is still undesirable though, as this reduces -- their usefulness for debugging and profiling. Therefore we will -- generally try only to make use of this property where it is -- necessary to enable optimizations. | SourceNote { Tickish id -> RealSrcSpan sourceSpan :: RealSrcSpan -- ^ Source covered , Tickish id -> String sourceName :: String -- ^ Name for source location -- (uses same names as CCs) } deriving (Tickish id -> Tickish id -> Bool (Tickish id -> Tickish id -> Bool) -> (Tickish id -> Tickish id -> Bool) -> Eq (Tickish id) forall id. Eq id => Tickish id -> Tickish id -> Bool forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a /= :: Tickish id -> Tickish id -> Bool $c/= :: forall id. Eq id => Tickish id -> Tickish id -> Bool == :: Tickish id -> Tickish id -> Bool $c== :: forall id. Eq id => Tickish id -> Tickish id -> Bool Eq, Eq (Tickish id) Eq (Tickish id) => (Tickish id -> Tickish id -> Ordering) -> (Tickish id -> Tickish id -> Bool) -> (Tickish id -> Tickish id -> Bool) -> (Tickish id -> Tickish id -> Bool) -> (Tickish id -> Tickish id -> Bool) -> (Tickish id -> Tickish id -> Tickish id) -> (Tickish id -> Tickish id -> Tickish id) -> Ord (Tickish id) Tickish id -> Tickish id -> Bool Tickish id -> Tickish id -> Ordering Tickish id -> Tickish id -> Tickish id forall a. Eq a => (a -> a -> Ordering) -> (a -> a -> Bool) -> (a -> a -> Bool) -> (a -> a -> Bool) -> (a -> a -> Bool) -> (a -> a -> a) -> (a -> a -> a) -> Ord a forall id. Ord id => Eq (Tickish id) forall id. Ord id => Tickish id -> Tickish id -> Bool forall id. Ord id => Tickish id -> Tickish id -> Ordering forall id. Ord id => Tickish id -> Tickish id -> Tickish id min :: Tickish id -> Tickish id -> Tickish id $cmin :: forall id. Ord id => Tickish id -> Tickish id -> Tickish id max :: Tickish id -> Tickish id -> Tickish id $cmax :: forall id. Ord id => Tickish id -> Tickish id -> Tickish id >= :: Tickish id -> Tickish id -> Bool $c>= :: forall id. Ord id => Tickish id -> Tickish id -> Bool > :: Tickish id -> Tickish id -> Bool $c> :: forall id. Ord id => Tickish id -> Tickish id -> Bool <= :: Tickish id -> Tickish id -> Bool $c<= :: forall id. Ord id => Tickish id -> Tickish id -> Bool < :: Tickish id -> Tickish id -> Bool $c< :: forall id. Ord id => Tickish id -> Tickish id -> Bool compare :: Tickish id -> Tickish id -> Ordering $ccompare :: forall id. Ord id => Tickish id -> Tickish id -> Ordering $cp1Ord :: forall id. Ord id => Eq (Tickish id) Ord, Typeable (Tickish id) DataType Constr Typeable (Tickish id) => (forall (c :: * -> *). (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Tickish id -> c (Tickish id)) -> (forall (c :: * -> *). (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Tickish id)) -> (Tickish id -> Constr) -> (Tickish id -> DataType) -> (forall (t :: * -> *) (c :: * -> *). Typeable t => (forall d. Data d => c (t d)) -> Maybe (c (Tickish id))) -> (forall (t :: * -> * -> *) (c :: * -> *). Typeable t => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c (Tickish id))) -> ((forall b. Data b => b -> b) -> Tickish id -> Tickish id) -> (forall r r'. (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Tickish id -> r) -> (forall r r'. (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Tickish id -> r) -> (forall u. (forall d. Data d => d -> u) -> Tickish id -> [u]) -> (forall u. Int -> (forall d. Data d => d -> u) -> Tickish id -> u) -> (forall (m :: * -> *). Monad m => (forall d. Data d => d -> m d) -> Tickish id -> m (Tickish id)) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> Tickish id -> m (Tickish id)) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> Tickish id -> m (Tickish id)) -> Data (Tickish id) Tickish id -> DataType Tickish id -> Constr (forall d. Data d => c (t d)) -> Maybe (c (Tickish id)) (forall b. Data b => b -> b) -> Tickish id -> Tickish id (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Tickish id -> c (Tickish id) (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Tickish id) forall id. Data id => Typeable (Tickish id) forall id. Data id => Tickish id -> DataType forall id. Data id => Tickish id -> Constr forall id. Data id => (forall b. Data b => b -> b) -> Tickish id -> Tickish id forall id u. Data id => Int -> (forall d. Data d => d -> u) -> Tickish id -> u forall id u. Data id => (forall d. Data d => d -> u) -> Tickish id -> [u] forall id r r'. Data id => (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Tickish id -> r forall id r r'. Data id => (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Tickish id -> r forall id (m :: * -> *). (Data id, Monad m) => (forall d. Data d => d -> m d) -> Tickish id -> m (Tickish id) forall id (m :: * -> *). (Data id, MonadPlus m) => (forall d. Data d => d -> m d) -> Tickish id -> m (Tickish id) forall id (c :: * -> *). Data id => (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Tickish id) forall id (c :: * -> *). Data id => (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Tickish id -> c (Tickish id) forall id (t :: * -> *) (c :: * -> *). (Data id, Typeable t) => (forall d. Data d => c (t d)) -> Maybe (c (Tickish id)) forall id (t :: * -> * -> *) (c :: * -> *). (Data id, Typeable t) => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c (Tickish id)) forall a. Typeable a => (forall (c :: * -> *). (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> a -> c a) -> (forall (c :: * -> *). (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c a) -> (a -> Constr) -> (a -> DataType) -> (forall (t :: * -> *) (c :: * -> *). Typeable t => (forall d. Data d => c (t d)) -> Maybe (c a)) -> (forall (t :: * -> * -> *) (c :: * -> *). Typeable t => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c a)) -> ((forall b. Data b => b -> b) -> a -> a) -> (forall r r'. (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> a -> r) -> (forall r r'. (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> a -> r) -> (forall u. (forall d. Data d => d -> u) -> a -> [u]) -> (forall u. Int -> (forall d. Data d => d -> u) -> a -> u) -> (forall (m :: * -> *). Monad m => (forall d. Data d => d -> m d) -> a -> m a) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> a -> m a) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> a -> m a) -> Data a forall u. Int -> (forall d. Data d => d -> u) -> Tickish id -> u forall u. (forall d. Data d => d -> u) -> Tickish id -> [u] forall r r'. (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Tickish id -> r forall r r'. (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Tickish id -> r forall (m :: * -> *). Monad m => (forall d. Data d => d -> m d) -> Tickish id -> m (Tickish id) forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> Tickish id -> m (Tickish id) forall (c :: * -> *). (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Tickish id) forall (c :: * -> *). (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Tickish id -> c (Tickish id) forall (t :: * -> *) (c :: * -> *). Typeable t => (forall d. Data d => c (t d)) -> Maybe (c (Tickish id)) forall (t :: * -> * -> *) (c :: * -> *). Typeable t => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c (Tickish id)) $cSourceNote :: Constr $cBreakpoint :: Constr $cHpcTick :: Constr $cProfNote :: Constr $tTickish :: DataType gmapMo :: (forall d. Data d => d -> m d) -> Tickish id -> m (Tickish id) $cgmapMo :: forall id (m :: * -> *). (Data id, MonadPlus m) => (forall d. Data d => d -> m d) -> Tickish id -> m (Tickish id) gmapMp :: (forall d. Data d => d -> m d) -> Tickish id -> m (Tickish id) $cgmapMp :: forall id (m :: * -> *). (Data id, MonadPlus m) => (forall d. Data d => d -> m d) -> Tickish id -> m (Tickish id) gmapM :: (forall d. Data d => d -> m d) -> Tickish id -> m (Tickish id) $cgmapM :: forall id (m :: * -> *). (Data id, Monad m) => (forall d. Data d => d -> m d) -> Tickish id -> m (Tickish id) gmapQi :: Int -> (forall d. Data d => d -> u) -> Tickish id -> u $cgmapQi :: forall id u. Data id => Int -> (forall d. Data d => d -> u) -> Tickish id -> u gmapQ :: (forall d. Data d => d -> u) -> Tickish id -> [u] $cgmapQ :: forall id u. Data id => (forall d. Data d => d -> u) -> Tickish id -> [u] gmapQr :: (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Tickish id -> r $cgmapQr :: forall id r r'. Data id => (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> Tickish id -> r gmapQl :: (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Tickish id -> r $cgmapQl :: forall id r r'. Data id => (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> Tickish id -> r gmapT :: (forall b. Data b => b -> b) -> Tickish id -> Tickish id $cgmapT :: forall id. Data id => (forall b. Data b => b -> b) -> Tickish id -> Tickish id dataCast2 :: (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c (Tickish id)) $cdataCast2 :: forall id (t :: * -> * -> *) (c :: * -> *). (Data id, Typeable t) => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c (Tickish id)) dataCast1 :: (forall d. Data d => c (t d)) -> Maybe (c (Tickish id)) $cdataCast1 :: forall id (t :: * -> *) (c :: * -> *). (Data id, Typeable t) => (forall d. Data d => c (t d)) -> Maybe (c (Tickish id)) dataTypeOf :: Tickish id -> DataType $cdataTypeOf :: forall id. Data id => Tickish id -> DataType toConstr :: Tickish id -> Constr $ctoConstr :: forall id. Data id => Tickish id -> Constr gunfold :: (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Tickish id) $cgunfold :: forall id (c :: * -> *). Data id => (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c (Tickish id) gfoldl :: (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Tickish id -> c (Tickish id) $cgfoldl :: forall id (c :: * -> *). Data id => (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> Tickish id -> c (Tickish id) $cp1Data :: forall id. Data id => Typeable (Tickish id) Data) -- | A "counting tick" (where tickishCounts is True) is one that -- counts evaluations in some way. We cannot discard a counting tick, -- and the compiler should preserve the number of counting ticks as -- far as possible. -- -- However, we still allow the simplifier to increase or decrease -- sharing, so in practice the actual number of ticks may vary, except -- that we never change the value from zero to non-zero or vice versa. tickishCounts :: Tickish id -> Bool tickishCounts :: Tickish id -> Bool tickishCounts n :: Tickish id n@ProfNote{} = Tickish id -> Bool forall id. Tickish id -> Bool profNoteCount Tickish id n tickishCounts HpcTick{} = Bool True tickishCounts Breakpoint{} = Bool True tickishCounts _ = Bool False -- | Specifies the scoping behaviour of ticks. This governs the -- behaviour of ticks that care about the covered code and the cost -- associated with it. Important for ticks relating to profiling. data TickishScoping = -- | No scoping: The tick does not care about what code it -- covers. Transformations can freely move code inside as well as -- outside without any additional annotation obligations NoScope -- | Soft scoping: We want all code that is covered to stay -- covered. Note that this scope type does not forbid -- transformations from happening, as long as all results of -- the transformations are still covered by this tick or a copy of -- it. For example -- -- let x = tick<...> (let y = foo in bar) in baz -- ===> -- let x = tick<...> bar; y = tick<...> foo in baz -- -- Is a valid transformation as far as "bar" and "foo" is -- concerned, because both still are scoped over by the tick. -- -- Note though that one might object to the "let" not being -- covered by the tick any more. However, we are generally lax -- with this - constant costs don't matter too much, and given -- that the "let" was effectively merged we can view it as having -- lost its identity anyway. -- -- Also note that this scoping behaviour allows floating a tick -- "upwards" in pretty much any situation. For example: -- -- case foo of x -> tick<...> bar -- ==> -- tick<...> case foo of x -> bar -- -- While this is always leagl, we want to make a best effort to -- only make us of this where it exposes transformation -- opportunities. | SoftScope -- | Cost centre scoping: We don't want any costs to move to other -- cost-centre stacks. This means we not only want no code or cost -- to get moved out of their cost centres, but we also object to -- code getting associated with new cost-centre ticks - or -- changing the order in which they get applied. -- -- A rule of thumb is that we don't want any code to gain new -- annotations. However, there are notable exceptions, for -- example: -- -- let f = \y -> foo in tick<...> ... (f x) ... -- ==> -- tick<...> ... foo[x/y] ... -- -- In-lining lambdas like this is always legal, because inlining a -- function does not change the cost-centre stack when the -- function is called. | CostCentreScope deriving (TickishScoping -> TickishScoping -> Bool (TickishScoping -> TickishScoping -> Bool) -> (TickishScoping -> TickishScoping -> Bool) -> Eq TickishScoping forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a /= :: TickishScoping -> TickishScoping -> Bool $c/= :: TickishScoping -> TickishScoping -> Bool == :: TickishScoping -> TickishScoping -> Bool $c== :: TickishScoping -> TickishScoping -> Bool Eq) -- | Returns the intended scoping rule for a Tickish tickishScoped :: Tickish id -> TickishScoping tickishScoped :: Tickish id -> TickishScoping tickishScoped n :: Tickish id n@ProfNote{} | Tickish id -> Bool forall id. Tickish id -> Bool profNoteScope Tickish id n = TickishScoping CostCentreScope | Bool otherwise = TickishScoping NoScope tickishScoped HpcTick{} = TickishScoping NoScope tickishScoped Breakpoint{} = TickishScoping CostCentreScope -- Breakpoints are scoped: eventually we're going to do call -- stacks, but also this helps prevent the simplifier from moving -- breakpoints around and changing their result type (see #1531). tickishScoped SourceNote{} = TickishScoping SoftScope -- | Returns whether the tick scoping rule is at least as permissive -- as the given scoping rule. tickishScopesLike :: Tickish id -> TickishScoping -> Bool tickishScopesLike :: Tickish id -> TickishScoping -> Bool tickishScopesLike t :: Tickish id t scope :: TickishScoping scope = Tickish id -> TickishScoping forall id. Tickish id -> TickishScoping tickishScoped Tickish id t TickishScoping -> TickishScoping -> Bool `like` TickishScoping scope where NoScope like :: TickishScoping -> TickishScoping -> Bool `like` _ = Bool True _ `like` NoScope = Bool False SoftScope `like` _ = Bool True _ `like` SoftScope = Bool False CostCentreScope `like` _ = Bool True -- | Returns @True@ for ticks that can be floated upwards easily even -- where it might change execution counts, such as: -- -- Just (tick<...> foo) -- ==> -- tick<...> (Just foo) -- -- This is a combination of @tickishSoftScope@ and -- @tickishCounts@. Note that in principle splittable ticks can become -- floatable using @mkNoTick@ -- even though there's currently no -- tickish for which that is the case. tickishFloatable :: Tickish id -> Bool tickishFloatable :: Tickish id -> Bool tickishFloatable t :: Tickish id t = Tickish id t Tickish id -> TickishScoping -> Bool forall id. Tickish id -> TickishScoping -> Bool `tickishScopesLike` TickishScoping SoftScope Bool -> Bool -> Bool && Bool -> Bool not (Tickish id -> Bool forall id. Tickish id -> Bool tickishCounts Tickish id t) -- | Returns @True@ for a tick that is both counting /and/ scoping and -- can be split into its (tick, scope) parts using 'mkNoScope' and -- 'mkNoTick' respectively. tickishCanSplit :: Tickish id -> Bool tickishCanSplit :: Tickish id -> Bool tickishCanSplit ProfNote{profNoteScope :: forall id. Tickish id -> Bool profNoteScope = Bool True, profNoteCount :: forall id. Tickish id -> Bool profNoteCount = Bool True} = Bool True tickishCanSplit _ = Bool False mkNoCount :: Tickish id -> Tickish id mkNoCount :: Tickish id -> Tickish id mkNoCount n :: Tickish id n | Bool -> Bool not (Tickish id -> Bool forall id. Tickish id -> Bool tickishCounts Tickish id n) = Tickish id n | Bool -> Bool not (Tickish id -> Bool forall id. Tickish id -> Bool tickishCanSplit Tickish id n) = String -> Tickish id forall a. String -> a panic "mkNoCount: Cannot split!" mkNoCount n :: Tickish id n@ProfNote{} = Tickish id n {profNoteCount :: Bool profNoteCount = Bool False} mkNoCount _ = String -> Tickish id forall a. String -> a panic "mkNoCount: Undefined split!" mkNoScope :: Tickish id -> Tickish id mkNoScope :: Tickish id -> Tickish id mkNoScope n :: Tickish id n | Tickish id -> TickishScoping forall id. Tickish id -> TickishScoping tickishScoped Tickish id n TickishScoping -> TickishScoping -> Bool forall a. Eq a => a -> a -> Bool == TickishScoping NoScope = Tickish id n | Bool -> Bool not (Tickish id -> Bool forall id. Tickish id -> Bool tickishCanSplit Tickish id n) = String -> Tickish id forall a. String -> a panic "mkNoScope: Cannot split!" mkNoScope n :: Tickish id n@ProfNote{} = Tickish id n {profNoteScope :: Bool profNoteScope = Bool False} mkNoScope _ = String -> Tickish id forall a. String -> a panic "mkNoScope: Undefined split!" -- | Return @True@ if this source annotation compiles to some backend -- code. Without this flag, the tickish is seen as a simple annotation -- that does not have any associated evaluation code. -- -- What this means that we are allowed to disregard the tick if doing -- so means that we can skip generating any code in the first place. A -- typical example is top-level bindings: -- -- foo = tick<...> \y -> ... -- ==> -- foo = \y -> tick<...> ... -- -- Here there is just no operational difference between the first and -- the second version. Therefore code generation should simply -- translate the code as if it found the latter. tickishIsCode :: Tickish id -> Bool tickishIsCode :: Tickish id -> Bool tickishIsCode SourceNote{} = Bool False tickishIsCode _tickish :: Tickish id _tickish = Bool True -- all the rest for now -- | Governs the kind of expression that the tick gets placed on when -- annotating for example using @mkTick@. If we find that we want to -- put a tickish on an expression ruled out here, we try to float it -- inwards until we find a suitable expression. data TickishPlacement = -- | Place ticks exactly on run-time expressions. We can still -- move the tick through pure compile-time constructs such as -- other ticks, casts or type lambdas. This is the most -- restrictive placement rule for ticks, as all tickishs have in -- common that they want to track runtime processes. The only -- legal placement rule for counting ticks. PlaceRuntime -- | As @PlaceRuntime@, but we float the tick through all -- lambdas. This makes sense where there is little difference -- between annotating the lambda and annotating the lambda's code. | PlaceNonLam -- | In addition to floating through lambdas, cost-centre style -- tickishs can also be moved from constructors, non-function -- variables and literals. For example: -- -- let x = scc<...> C (scc<...> y) (scc<...> 3) in ... -- -- Neither the constructor application, the variable or the -- literal are likely to have any cost worth mentioning. And even -- if y names a thunk, the call would not care about the -- evaluation context. Therefore removing all annotations in the -- above example is safe. | PlaceCostCentre deriving (TickishPlacement -> TickishPlacement -> Bool (TickishPlacement -> TickishPlacement -> Bool) -> (TickishPlacement -> TickishPlacement -> Bool) -> Eq TickishPlacement forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a /= :: TickishPlacement -> TickishPlacement -> Bool $c/= :: TickishPlacement -> TickishPlacement -> Bool == :: TickishPlacement -> TickishPlacement -> Bool $c== :: TickishPlacement -> TickishPlacement -> Bool Eq) -- | Placement behaviour we want for the ticks tickishPlace :: Tickish id -> TickishPlacement tickishPlace :: Tickish id -> TickishPlacement tickishPlace n :: Tickish id n@ProfNote{} | Tickish id -> Bool forall id. Tickish id -> Bool profNoteCount Tickish id n = TickishPlacement PlaceRuntime | Bool otherwise = TickishPlacement PlaceCostCentre tickishPlace HpcTick{} = TickishPlacement PlaceRuntime tickishPlace Breakpoint{} = TickishPlacement PlaceRuntime tickishPlace SourceNote{} = TickishPlacement PlaceNonLam -- | Returns whether one tick "contains" the other one, therefore -- making the second tick redundant. tickishContains :: Eq b => Tickish b -> Tickish b -> Bool tickishContains :: Tickish b -> Tickish b -> Bool tickishContains (SourceNote sp1 :: RealSrcSpan sp1 n1 :: String n1) (SourceNote sp2 :: RealSrcSpan sp2 n2 :: String n2) = RealSrcSpan -> RealSrcSpan -> Bool containsSpan RealSrcSpan sp1 RealSrcSpan sp2 Bool -> Bool -> Bool && String n1 String -> String -> Bool forall a. Eq a => a -> a -> Bool == String n2 -- compare the String last tickishContains t1 :: Tickish b t1 t2 :: Tickish b t2 = Tickish b t1 Tickish b -> Tickish b -> Bool forall a. Eq a => a -> a -> Bool == Tickish b t2 {- ************************************************************************ * * Orphans * * ************************************************************************ -} -- | Is this instance an orphan? If it is not an orphan, contains an 'OccName' -- witnessing the instance's non-orphanhood. -- See Note [Orphans] data IsOrphan = IsOrphan | NotOrphan OccName -- The OccName 'n' witnesses the instance's non-orphanhood -- In that case, the instance is fingerprinted as part -- of the definition of 'n's definition deriving Typeable IsOrphan DataType Constr Typeable IsOrphan => (forall (c :: * -> *). (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> IsOrphan -> c IsOrphan) -> (forall (c :: * -> *). (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c IsOrphan) -> (IsOrphan -> Constr) -> (IsOrphan -> DataType) -> (forall (t :: * -> *) (c :: * -> *). Typeable t => (forall d. Data d => c (t d)) -> Maybe (c IsOrphan)) -> (forall (t :: * -> * -> *) (c :: * -> *). Typeable t => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c IsOrphan)) -> ((forall b. Data b => b -> b) -> IsOrphan -> IsOrphan) -> (forall r r'. (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> IsOrphan -> r) -> (forall r r'. (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> IsOrphan -> r) -> (forall u. (forall d. Data d => d -> u) -> IsOrphan -> [u]) -> (forall u. Int -> (forall d. Data d => d -> u) -> IsOrphan -> u) -> (forall (m :: * -> *). Monad m => (forall d. Data d => d -> m d) -> IsOrphan -> m IsOrphan) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> IsOrphan -> m IsOrphan) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> IsOrphan -> m IsOrphan) -> Data IsOrphan IsOrphan -> DataType IsOrphan -> Constr (forall b. Data b => b -> b) -> IsOrphan -> IsOrphan (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> IsOrphan -> c IsOrphan (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c IsOrphan forall a. Typeable a => (forall (c :: * -> *). (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> a -> c a) -> (forall (c :: * -> *). (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c a) -> (a -> Constr) -> (a -> DataType) -> (forall (t :: * -> *) (c :: * -> *). Typeable t => (forall d. Data d => c (t d)) -> Maybe (c a)) -> (forall (t :: * -> * -> *) (c :: * -> *). Typeable t => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c a)) -> ((forall b. Data b => b -> b) -> a -> a) -> (forall r r'. (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> a -> r) -> (forall r r'. (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> a -> r) -> (forall u. (forall d. Data d => d -> u) -> a -> [u]) -> (forall u. Int -> (forall d. Data d => d -> u) -> a -> u) -> (forall (m :: * -> *). Monad m => (forall d. Data d => d -> m d) -> a -> m a) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> a -> m a) -> (forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> a -> m a) -> Data a forall u. Int -> (forall d. Data d => d -> u) -> IsOrphan -> u forall u. (forall d. Data d => d -> u) -> IsOrphan -> [u] forall r r'. (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> IsOrphan -> r forall r r'. (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> IsOrphan -> r forall (m :: * -> *). Monad m => (forall d. Data d => d -> m d) -> IsOrphan -> m IsOrphan forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> IsOrphan -> m IsOrphan forall (c :: * -> *). (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c IsOrphan forall (c :: * -> *). (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> IsOrphan -> c IsOrphan forall (t :: * -> *) (c :: * -> *). Typeable t => (forall d. Data d => c (t d)) -> Maybe (c IsOrphan) forall (t :: * -> * -> *) (c :: * -> *). Typeable t => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c IsOrphan) $cNotOrphan :: Constr $cIsOrphan :: Constr $tIsOrphan :: DataType gmapMo :: (forall d. Data d => d -> m d) -> IsOrphan -> m IsOrphan $cgmapMo :: forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> IsOrphan -> m IsOrphan gmapMp :: (forall d. Data d => d -> m d) -> IsOrphan -> m IsOrphan $cgmapMp :: forall (m :: * -> *). MonadPlus m => (forall d. Data d => d -> m d) -> IsOrphan -> m IsOrphan gmapM :: (forall d. Data d => d -> m d) -> IsOrphan -> m IsOrphan $cgmapM :: forall (m :: * -> *). Monad m => (forall d. Data d => d -> m d) -> IsOrphan -> m IsOrphan gmapQi :: Int -> (forall d. Data d => d -> u) -> IsOrphan -> u $cgmapQi :: forall u. Int -> (forall d. Data d => d -> u) -> IsOrphan -> u gmapQ :: (forall d. Data d => d -> u) -> IsOrphan -> [u] $cgmapQ :: forall u. (forall d. Data d => d -> u) -> IsOrphan -> [u] gmapQr :: (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> IsOrphan -> r $cgmapQr :: forall r r'. (r' -> r -> r) -> r -> (forall d. Data d => d -> r') -> IsOrphan -> r gmapQl :: (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> IsOrphan -> r $cgmapQl :: forall r r'. (r -> r' -> r) -> r -> (forall d. Data d => d -> r') -> IsOrphan -> r gmapT :: (forall b. Data b => b -> b) -> IsOrphan -> IsOrphan $cgmapT :: (forall b. Data b => b -> b) -> IsOrphan -> IsOrphan dataCast2 :: (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c IsOrphan) $cdataCast2 :: forall (t :: * -> * -> *) (c :: * -> *). Typeable t => (forall d e. (Data d, Data e) => c (t d e)) -> Maybe (c IsOrphan) dataCast1 :: (forall d. Data d => c (t d)) -> Maybe (c IsOrphan) $cdataCast1 :: forall (t :: * -> *) (c :: * -> *). Typeable t => (forall d. Data d => c (t d)) -> Maybe (c IsOrphan) dataTypeOf :: IsOrphan -> DataType $cdataTypeOf :: IsOrphan -> DataType toConstr :: IsOrphan -> Constr $ctoConstr :: IsOrphan -> Constr gunfold :: (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c IsOrphan $cgunfold :: forall (c :: * -> *). (forall b r. Data b => c (b -> r) -> c r) -> (forall r. r -> c r) -> Constr -> c IsOrphan gfoldl :: (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> IsOrphan -> c IsOrphan $cgfoldl :: forall (c :: * -> *). (forall d b. Data d => c (d -> b) -> d -> c b) -> (forall g. g -> c g) -> IsOrphan -> c IsOrphan $cp1Data :: Typeable IsOrphan Data -- | Returns true if 'IsOrphan' is orphan. isOrphan :: IsOrphan -> Bool isOrphan :: IsOrphan -> Bool isOrphan IsOrphan = Bool True isOrphan _ = Bool False -- | Returns true if 'IsOrphan' is not an orphan. notOrphan :: IsOrphan -> Bool notOrphan :: IsOrphan -> Bool notOrphan NotOrphan{} = Bool True notOrphan _ = Bool False chooseOrphanAnchor :: NameSet -> IsOrphan -- Something (rule, instance) is relate to all the Names in this -- list. Choose one of them to be an "anchor" for the orphan. We make -- the choice deterministic to avoid gratuitious changes in the ABI -- hash (Trac #4012). Specifically, use lexicographic comparison of -- OccName rather than comparing Uniques -- -- NB: 'minimum' use Ord, and (Ord OccName) works lexicographically -- chooseOrphanAnchor :: NameSet -> IsOrphan chooseOrphanAnchor local_names :: NameSet local_names | NameSet -> Bool isEmptyNameSet NameSet local_names = IsOrphan IsOrphan | Bool otherwise = OccName -> IsOrphan NotOrphan ([OccName] -> OccName forall (t :: * -> *) a. (Foldable t, Ord a) => t a -> a minimum [OccName] occs) where occs :: [OccName] occs = (Name -> OccName) -> [Name] -> [OccName] forall a b. (a -> b) -> [a] -> [b] map Name -> OccName nameOccName ([Name] -> [OccName]) -> [Name] -> [OccName] forall a b. (a -> b) -> a -> b $ NameSet -> [Name] forall elt. UniqSet elt -> [elt] nonDetEltsUniqSet NameSet local_names -- It's OK to use nonDetEltsUFM here, see comments above instance Binary IsOrphan where put_ :: BinHandle -> IsOrphan -> IO () put_ bh :: BinHandle bh IsOrphan = BinHandle -> Word8 -> IO () putByte BinHandle bh 0 put_ bh :: BinHandle bh (NotOrphan n :: OccName n) = do BinHandle -> Word8 -> IO () putByte BinHandle bh 1 BinHandle -> OccName -> IO () forall a. Binary a => BinHandle -> a -> IO () put_ BinHandle bh OccName n get :: BinHandle -> IO IsOrphan get bh :: BinHandle bh = do Word8 h <- BinHandle -> IO Word8 getByte BinHandle bh case Word8 h of 0 -> IsOrphan -> IO IsOrphan forall (m :: * -> *) a. Monad m => a -> m a return IsOrphan IsOrphan _ -> do OccName n <- BinHandle -> IO OccName forall a. Binary a => BinHandle -> IO a get BinHandle bh IsOrphan -> IO IsOrphan forall (m :: * -> *) a. Monad m => a -> m a return (IsOrphan -> IO IsOrphan) -> IsOrphan -> IO IsOrphan forall a b. (a -> b) -> a -> b $ OccName -> IsOrphan NotOrphan OccName n {- Note [Orphans] ~~~~~~~~~~~~~~ Class instances, rules, and family instances are divided into orphans and non-orphans. Roughly speaking, an instance/rule is an orphan if its left hand side mentions nothing defined in this module. Orphan-hood has two major consequences * A module that contains orphans is called an "orphan module". If the module being compiled depends (transitively) on an oprhan module M, then M.hi is read in regardless of whether M is oherwise needed. This is to ensure that we don't miss any instance decls in M. But it's painful, because it means we need to keep track of all the orphan modules below us. * A non-orphan is not finger-printed separately. Instead, for fingerprinting purposes it is treated as part of the entity it mentions on the LHS. For example data T = T1 | T2 instance Eq T where .... The instance (Eq T) is incorprated as part of T's fingerprint. In contrast, orphans are all fingerprinted together in the mi_orph_hash field of the ModIface. See MkIface.addFingerprints. Orphan-hood is computed * For class instances: when we make a ClsInst (because it is needed during instance lookup) * For rules and family instances: when we generate an IfaceRule (MkIface.coreRuleToIfaceRule) or IfaceFamInst (MkIface.instanceToIfaceInst) -} {- ************************************************************************ * * \subsection{Transformation rules} * * ************************************************************************ The CoreRule type and its friends are dealt with mainly in CoreRules, but CoreFVs, Subst, PprCore, CoreTidy also inspect the representation. -} -- | Gathers a collection of 'CoreRule's. Maps (the name of) an 'Id' to its rules type RuleBase = NameEnv [CoreRule] -- The rules are unordered; -- we sort out any overlaps on lookup -- | A full rule environment which we can apply rules from. Like a 'RuleBase', -- but it also includes the set of visible orphans we use to filter out orphan -- rules which are not visible (even though we can see them...) data RuleEnv = RuleEnv { RuleEnv -> RuleBase re_base :: RuleBase , RuleEnv -> ModuleSet re_visible_orphs :: ModuleSet } mkRuleEnv :: RuleBase -> [Module] -> RuleEnv mkRuleEnv :: RuleBase -> [Module] -> RuleEnv mkRuleEnv rules :: RuleBase rules vis_orphs :: [Module] vis_orphs = RuleBase -> ModuleSet -> RuleEnv RuleEnv RuleBase rules ([Module] -> ModuleSet mkModuleSet [Module] vis_orphs) emptyRuleEnv :: RuleEnv emptyRuleEnv :: RuleEnv emptyRuleEnv = RuleBase -> ModuleSet -> RuleEnv RuleEnv RuleBase forall a. NameEnv a emptyNameEnv ModuleSet emptyModuleSet -- | A 'CoreRule' is: -- -- * \"Local\" if the function it is a rule for is defined in the -- same module as the rule itself. -- -- * \"Orphan\" if nothing on the LHS is defined in the same module -- as the rule itself data CoreRule = Rule { CoreRule -> RuleName ru_name :: RuleName, -- ^ Name of the rule, for communication with the user CoreRule -> Activation ru_act :: Activation, -- ^ When the rule is active -- Rough-matching stuff -- see comments with InstEnv.ClsInst( is_cls, is_rough ) CoreRule -> Name ru_fn :: Name, -- ^ Name of the 'Id.Id' at the head of this rule CoreRule -> [Maybe Name] ru_rough :: [Maybe Name], -- ^ Name at the head of each argument to the left hand side -- Proper-matching stuff -- see comments with InstEnv.ClsInst( is_tvs, is_tys ) CoreRule -> [CoreBndr] ru_bndrs :: [CoreBndr], -- ^ Variables quantified over CoreRule -> [CoreExpr] ru_args :: [CoreExpr], -- ^ Left hand side arguments -- And the right-hand side CoreRule -> CoreExpr ru_rhs :: CoreExpr, -- ^ Right hand side of the rule -- Occurrence info is guaranteed correct -- See Note [OccInfo in unfoldings and rules] -- Locality CoreRule -> Bool ru_auto :: Bool, -- ^ @True@ <=> this rule is auto-generated -- (notably by Specialise or SpecConstr) -- @False@ <=> generated at the user's behest -- See Note [Trimming auto-rules] in TidyPgm -- for the sole purpose of this field. CoreRule -> Module ru_origin :: !Module, -- ^ 'Module' the rule was defined in, used -- to test if we should see an orphan rule. CoreRule -> IsOrphan ru_orphan :: !IsOrphan, -- ^ Whether or not the rule is an orphan. CoreRule -> Bool ru_local :: Bool -- ^ @True@ iff the fn at the head of the rule is -- defined in the same module as the rule -- and is not an implicit 'Id' (like a record selector, -- class operation, or data constructor). This -- is different from 'ru_orphan', where a rule -- can avoid being an orphan if *any* Name in -- LHS of the rule was defined in the same -- module as the rule. } -- | Built-in rules are used for constant folding -- and suchlike. They have no free variables. -- A built-in rule is always visible (there is no such thing as -- an orphan built-in rule.) | BuiltinRule { ru_name :: RuleName, -- ^ As above ru_fn :: Name, -- ^ As above CoreRule -> Int ru_nargs :: Int, -- ^ Number of arguments that 'ru_try' consumes, -- if it fires, including type arguments CoreRule -> RuleFun ru_try :: RuleFun -- ^ This function does the rewrite. It given too many -- arguments, it simply discards them; the returned 'CoreExpr' -- is just the rewrite of 'ru_fn' applied to the first 'ru_nargs' args } -- See Note [Extra args in rule matching] in Rules.hs type RuleFun = DynFlags -> InScopeEnv -> Id -> [CoreExpr] -> Maybe CoreExpr type InScopeEnv = (InScopeSet, IdUnfoldingFun) type IdUnfoldingFun = Id -> Unfolding -- A function that embodies how to unfold an Id if you need -- to do that in the Rule. The reason we need to pass this info in -- is that whether an Id is unfoldable depends on the simplifier phase isBuiltinRule :: CoreRule -> Bool isBuiltinRule :: CoreRule -> Bool isBuiltinRule (BuiltinRule {}) = Bool True isBuiltinRule _ = Bool False isAutoRule :: CoreRule -> Bool isAutoRule :: CoreRule -> Bool isAutoRule (BuiltinRule {}) = Bool False isAutoRule (Rule { ru_auto :: CoreRule -> Bool ru_auto = Bool is_auto }) = Bool is_auto -- | The number of arguments the 'ru_fn' must be applied -- to before the rule can match on it ruleArity :: CoreRule -> Int ruleArity :: CoreRule -> Int ruleArity (BuiltinRule {ru_nargs :: CoreRule -> Int ru_nargs = Int n}) = Int n ruleArity (Rule {ru_args :: CoreRule -> [CoreExpr] ru_args = [CoreExpr] args}) = [CoreExpr] -> Int forall (t :: * -> *) a. Foldable t => t a -> Int length [CoreExpr] args ruleName :: CoreRule -> RuleName ruleName :: CoreRule -> RuleName ruleName = CoreRule -> RuleName ru_name ruleModule :: CoreRule -> Maybe Module ruleModule :: CoreRule -> Maybe Module ruleModule Rule { Module ru_origin :: Module ru_origin :: CoreRule -> Module ru_origin } = Module -> Maybe Module forall a. a -> Maybe a Just Module ru_origin ruleModule BuiltinRule {} = Maybe Module forall a. Maybe a Nothing ruleActivation :: CoreRule -> Activation ruleActivation :: CoreRule -> Activation ruleActivation (BuiltinRule { }) = Activation AlwaysActive ruleActivation (Rule { ru_act :: CoreRule -> Activation ru_act = Activation act }) = Activation act -- | The 'Name' of the 'Id.Id' at the head of the rule left hand side ruleIdName :: CoreRule -> Name ruleIdName :: CoreRule -> Name ruleIdName = CoreRule -> Name ru_fn isLocalRule :: CoreRule -> Bool isLocalRule :: CoreRule -> Bool isLocalRule = CoreRule -> Bool ru_local -- | Set the 'Name' of the 'Id.Id' at the head of the rule left hand side setRuleIdName :: Name -> CoreRule -> CoreRule setRuleIdName :: Name -> CoreRule -> CoreRule setRuleIdName nm :: Name nm ru :: CoreRule ru = CoreRule ru { ru_fn :: Name ru_fn = Name nm } {- ************************************************************************ * * Unfoldings * * ************************************************************************ The @Unfolding@ type is declared here to avoid numerous loops -} -- | Records the /unfolding/ of an identifier, which is approximately the form the -- identifier would have if we substituted its definition in for the identifier. -- This type should be treated as abstract everywhere except in "CoreUnfold" data Unfolding = NoUnfolding -- ^ We have no information about the unfolding. | BootUnfolding -- ^ We have no information about the unfolding, because -- this 'Id' came from an @hi-boot@ file. -- See Note [Inlining and hs-boot files] in ToIface -- for what this is used for. | OtherCon [AltCon] -- ^ It ain't one of these constructors. -- @OtherCon xs@ also indicates that something has been evaluated -- and hence there's no point in re-evaluating it. -- @OtherCon []@ is used even for non-data-type values -- to indicated evaluated-ness. Notably: -- -- > data C = C !(Int -> Int) -- > case x of { C f -> ... } -- -- Here, @f@ gets an @OtherCon []@ unfolding. | DFunUnfolding { -- The Unfolding of a DFunId -- See Note [DFun unfoldings] -- df = /\a1..am. \d1..dn. MkD t1 .. tk -- (op1 a1..am d1..dn) -- (op2 a1..am d1..dn) Unfolding -> [CoreBndr] df_bndrs :: [Var], -- The bound variables [a1..m],[d1..dn] Unfolding -> DataCon df_con :: DataCon, -- The dictionary data constructor (never a newtype datacon) Unfolding -> [CoreExpr] df_args :: [CoreExpr] -- Args of the data con: types, superclasses and methods, } -- in positional order | CoreUnfolding { -- An unfolding for an Id with no pragma, -- or perhaps a NOINLINE pragma -- (For NOINLINE, the phase, if any, is in the -- InlinePragInfo for this Id.) Unfolding -> CoreExpr uf_tmpl :: CoreExpr, -- Template; occurrence info is correct Unfolding -> UnfoldingSource uf_src :: UnfoldingSource, -- Where the unfolding came from Unfolding -> Bool uf_is_top :: Bool, -- True <=> top level binding Unfolding -> Bool uf_is_value :: Bool, -- exprIsHNF template (cached); it is ok to discard -- a `seq` on this variable Unfolding -> Bool uf_is_conlike :: Bool, -- True <=> applicn of constructor or CONLIKE function -- Cached version of exprIsConLike Unfolding -> Bool uf_is_work_free :: Bool, -- True <=> doesn't waste (much) work to expand -- inside an inlining -- Cached version of exprIsCheap Unfolding -> Bool uf_expandable :: Bool, -- True <=> can expand in RULE matching -- Cached version of exprIsExpandable Unfolding -> UnfoldingGuidance uf_guidance :: UnfoldingGuidance -- Tells about the *size* of the template. } -- ^ An unfolding with redundant cached information. Parameters: -- -- uf_tmpl: Template used to perform unfolding; -- NB: Occurrence info is guaranteed correct: -- see Note [OccInfo in unfoldings and rules] -- -- uf_is_top: Is this a top level binding? -- -- uf_is_value: 'exprIsHNF' template (cached); it is ok to discard a 'seq' on -- this variable -- -- uf_is_work_free: Does this waste only a little work if we expand it inside an inlining? -- Basically this is a cached version of 'exprIsWorkFree' -- -- uf_guidance: Tells us about the /size/ of the unfolding template ------------------------------------------------ data UnfoldingSource = -- See also Note [Historical note: unfoldings for wrappers] InlineRhs -- The current rhs of the function -- Replace uf_tmpl each time around | InlineStable -- From an INLINE or INLINABLE pragma -- INLINE if guidance is UnfWhen -- INLINABLE if guidance is UnfIfGoodArgs/UnfoldNever -- (well, technically an INLINABLE might be made -- UnfWhen if it was small enough, and then -- it will behave like INLINE outside the current -- module, but that is the way automatic unfoldings -- work so it is consistent with the intended -- meaning of INLINABLE). -- -- uf_tmpl may change, but only as a result of -- gentle simplification, it doesn't get updated -- to the current RHS during compilation as with -- InlineRhs. -- -- See Note [InlineStable] | InlineCompulsory -- Something that *has* no binding, so you *must* inline it -- Only a few primop-like things have this property -- (see MkId.hs, calls to mkCompulsoryUnfolding). -- Inline absolutely always, however boring the context. -- | 'UnfoldingGuidance' says when unfolding should take place data UnfoldingGuidance = UnfWhen { -- Inline without thinking about the *size* of the uf_tmpl -- Used (a) for small *and* cheap unfoldings -- (b) for INLINE functions -- See Note [INLINE for small functions] in CoreUnfold UnfoldingGuidance -> Int ug_arity :: Arity, -- Number of value arguments expected UnfoldingGuidance -> Bool ug_unsat_ok :: Bool, -- True <=> ok to inline even if unsaturated UnfoldingGuidance -> Bool ug_boring_ok :: Bool -- True <=> ok to inline even if the context is boring -- So True,True means "always" } | UnfIfGoodArgs { -- Arose from a normal Id; the info here is the -- result of a simple analysis of the RHS UnfoldingGuidance -> [Int] ug_args :: [Int], -- Discount if the argument is evaluated. -- (i.e., a simplification will definitely -- be possible). One elt of the list per *value* arg. UnfoldingGuidance -> Int ug_size :: Int, -- The "size" of the unfolding. UnfoldingGuidance -> Int ug_res :: Int -- Scrutinee discount: the discount to substract if the thing is in } -- a context (case (thing args) of ...), -- (where there are the right number of arguments.) | UnfNever -- The RHS is big, so don't inline it deriving (UnfoldingGuidance -> UnfoldingGuidance -> Bool (UnfoldingGuidance -> UnfoldingGuidance -> Bool) -> (UnfoldingGuidance -> UnfoldingGuidance -> Bool) -> Eq UnfoldingGuidance forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a /= :: UnfoldingGuidance -> UnfoldingGuidance -> Bool $c/= :: UnfoldingGuidance -> UnfoldingGuidance -> Bool == :: UnfoldingGuidance -> UnfoldingGuidance -> Bool $c== :: UnfoldingGuidance -> UnfoldingGuidance -> Bool Eq) {- Note [Historical note: unfoldings for wrappers] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ We used to have a nice clever scheme in interface files for wrappers. A wrapper's unfolding can be reconstructed from its worker's id and its strictness. This decreased .hi file size (sometimes significantly, for modules like GHC.Classes with many high-arity w/w splits) and had a slight corresponding effect on compile times. However, when we added the second demand analysis, this scheme lead to some Core lint errors. The second analysis could change the strictness signatures, which sometimes resulted in a wrapper's regenerated unfolding applying the wrapper to too many arguments. Instead of repairing the clever .hi scheme, we abandoned it in favor of simplicity. The .hi sizes are usually insignificant (excluding the +1M for base libraries), and compile time barely increases (~+1% for nofib). The nicer upshot is that the UnfoldingSource no longer mentions an Id, so, eg, substitutions need not traverse them. Note [DFun unfoldings] ~~~~~~~~~~~~~~~~~~~~~~ The Arity in a DFunUnfolding is total number of args (type and value) that the DFun needs to produce a dictionary. That's not necessarily related to the ordinary arity of the dfun Id, esp if the class has one method, so the dictionary is represented by a newtype. Example class C a where { op :: a -> Int } instance C a -> C [a] where op xs = op (head xs) The instance translates to $dfCList :: forall a. C a => C [a] -- Arity 2! $dfCList = /\a.\d. $copList {a} d |> co $copList :: forall a. C a => [a] -> Int -- Arity 2! $copList = /\a.\d.\xs. op {a} d (head xs) Now we might encounter (op (dfCList {ty} d) a1 a2) and we want the (op (dfList {ty} d)) rule to fire, because $dfCList has all its arguments, even though its (value) arity is 2. That's why we record the number of expected arguments in the DFunUnfolding. Note that although it's an Arity, it's most convenient for it to give the *total* number of arguments, both type and value. See the use site in exprIsConApp_maybe. -} -- Constants for the UnfWhen constructor needSaturated, unSaturatedOk :: Bool needSaturated :: Bool needSaturated = Bool False unSaturatedOk :: Bool unSaturatedOk = Bool True boringCxtNotOk, boringCxtOk :: Bool boringCxtOk :: Bool boringCxtOk = Bool True boringCxtNotOk :: Bool boringCxtNotOk = Bool False ------------------------------------------------ noUnfolding :: Unfolding -- ^ There is no known 'Unfolding' evaldUnfolding :: Unfolding -- ^ This unfolding marks the associated thing as being evaluated noUnfolding :: Unfolding noUnfolding = Unfolding NoUnfolding evaldUnfolding :: Unfolding evaldUnfolding = [AltCon] -> Unfolding OtherCon [] -- | There is no known 'Unfolding', because this came from an -- hi-boot file. bootUnfolding :: Unfolding bootUnfolding :: Unfolding bootUnfolding = Unfolding BootUnfolding mkOtherCon :: [AltCon] -> Unfolding mkOtherCon :: [AltCon] -> Unfolding mkOtherCon = [AltCon] -> Unfolding OtherCon isStableSource :: UnfoldingSource -> Bool -- Keep the unfolding template isStableSource :: UnfoldingSource -> Bool isStableSource InlineCompulsory = Bool True isStableSource InlineStable = Bool True isStableSource InlineRhs = Bool False -- | Retrieves the template of an unfolding: panics if none is known unfoldingTemplate :: Unfolding -> CoreExpr unfoldingTemplate :: Unfolding -> CoreExpr unfoldingTemplate = Unfolding -> CoreExpr uf_tmpl -- | Retrieves the template of an unfolding if possible -- maybeUnfoldingTemplate is used mainly wnen specialising, and we do -- want to specialise DFuns, so it's important to return a template -- for DFunUnfoldings maybeUnfoldingTemplate :: Unfolding -> Maybe CoreExpr maybeUnfoldingTemplate :: Unfolding -> Maybe CoreExpr maybeUnfoldingTemplate (CoreUnfolding { uf_tmpl :: Unfolding -> CoreExpr uf_tmpl = CoreExpr expr }) = CoreExpr -> Maybe CoreExpr forall a. a -> Maybe a Just CoreExpr expr maybeUnfoldingTemplate (DFunUnfolding { df_bndrs :: Unfolding -> [CoreBndr] df_bndrs = [CoreBndr] bndrs, df_con :: Unfolding -> DataCon df_con = DataCon con, df_args :: Unfolding -> [CoreExpr] df_args = [CoreExpr] args }) = CoreExpr -> Maybe CoreExpr forall a. a -> Maybe a Just ([CoreBndr] -> CoreExpr -> CoreExpr forall b. [b] -> Expr b -> Expr b mkLams [CoreBndr] bndrs (CoreExpr -> [CoreExpr] -> CoreExpr forall b. Expr b -> [Expr b] -> Expr b mkApps (CoreBndr -> CoreExpr forall b. CoreBndr -> Expr b Var (DataCon -> CoreBndr dataConWorkId DataCon con)) [CoreExpr] args)) maybeUnfoldingTemplate _ = Maybe CoreExpr forall a. Maybe a Nothing -- | The constructors that the unfolding could never be: -- returns @[]@ if no information is available otherCons :: Unfolding -> [AltCon] otherCons :: Unfolding -> [AltCon] otherCons (OtherCon cons :: [AltCon] cons) = [AltCon] cons otherCons _ = [] -- | Determines if it is certainly the case that the unfolding will -- yield a value (something in HNF): returns @False@ if unsure isValueUnfolding :: Unfolding -> Bool -- Returns False for OtherCon isValueUnfolding :: Unfolding -> Bool isValueUnfolding (CoreUnfolding { uf_is_value :: Unfolding -> Bool uf_is_value = Bool is_evald }) = Bool is_evald isValueUnfolding _ = Bool False -- | Determines if it possibly the case that the unfolding will -- yield a value. Unlike 'isValueUnfolding' it returns @True@ -- for 'OtherCon' isEvaldUnfolding :: Unfolding -> Bool -- Returns True for OtherCon isEvaldUnfolding :: Unfolding -> Bool isEvaldUnfolding (OtherCon _) = Bool True isEvaldUnfolding (CoreUnfolding { uf_is_value :: Unfolding -> Bool uf_is_value = Bool is_evald }) = Bool is_evald isEvaldUnfolding _ = Bool False -- | @True@ if the unfolding is a constructor application, the application -- of a CONLIKE function or 'OtherCon' isConLikeUnfolding :: Unfolding -> Bool isConLikeUnfolding :: Unfolding -> Bool isConLikeUnfolding (OtherCon _) = Bool True isConLikeUnfolding (CoreUnfolding { uf_is_conlike :: Unfolding -> Bool uf_is_conlike = Bool con }) = Bool con isConLikeUnfolding _ = Bool False -- | Is the thing we will unfold into certainly cheap? isCheapUnfolding :: Unfolding -> Bool isCheapUnfolding :: Unfolding -> Bool isCheapUnfolding (CoreUnfolding { uf_is_work_free :: Unfolding -> Bool uf_is_work_free = Bool is_wf }) = Bool is_wf isCheapUnfolding _ = Bool False isExpandableUnfolding :: Unfolding -> Bool isExpandableUnfolding :: Unfolding -> Bool isExpandableUnfolding (CoreUnfolding { uf_expandable :: Unfolding -> Bool uf_expandable = Bool is_expable }) = Bool is_expable isExpandableUnfolding _ = Bool False expandUnfolding_maybe :: Unfolding -> Maybe CoreExpr -- Expand an expandable unfolding; this is used in rule matching -- See Note [Expanding variables] in Rules.hs -- The key point here is that CONLIKE things can be expanded expandUnfolding_maybe :: Unfolding -> Maybe CoreExpr expandUnfolding_maybe (CoreUnfolding { uf_expandable :: Unfolding -> Bool uf_expandable = Bool True, uf_tmpl :: Unfolding -> CoreExpr uf_tmpl = CoreExpr rhs }) = CoreExpr -> Maybe CoreExpr forall a. a -> Maybe a Just CoreExpr rhs expandUnfolding_maybe _ = Maybe CoreExpr forall a. Maybe a Nothing isCompulsoryUnfolding :: Unfolding -> Bool isCompulsoryUnfolding :: Unfolding -> Bool isCompulsoryUnfolding (CoreUnfolding { uf_src :: Unfolding -> UnfoldingSource uf_src = UnfoldingSource InlineCompulsory }) = Bool True isCompulsoryUnfolding _ = Bool False isStableUnfolding :: Unfolding -> Bool -- True of unfoldings that should not be overwritten -- by a CoreUnfolding for the RHS of a let-binding isStableUnfolding :: Unfolding -> Bool isStableUnfolding (CoreUnfolding { uf_src :: Unfolding -> UnfoldingSource uf_src = UnfoldingSource src }) = UnfoldingSource -> Bool isStableSource UnfoldingSource src isStableUnfolding (DFunUnfolding {}) = Bool True isStableUnfolding _ = Bool False -- | Only returns False if there is no unfolding information available at all hasSomeUnfolding :: Unfolding -> Bool hasSomeUnfolding :: Unfolding -> Bool hasSomeUnfolding NoUnfolding = Bool False hasSomeUnfolding BootUnfolding = Bool False hasSomeUnfolding _ = Bool True isBootUnfolding :: Unfolding -> Bool isBootUnfolding :: Unfolding -> Bool isBootUnfolding BootUnfolding = Bool True isBootUnfolding _ = Bool False neverUnfoldGuidance :: UnfoldingGuidance -> Bool neverUnfoldGuidance :: UnfoldingGuidance -> Bool neverUnfoldGuidance UnfNever = Bool True neverUnfoldGuidance _ = Bool False isFragileUnfolding :: Unfolding -> Bool -- An unfolding is fragile if it mentions free variables or -- is otherwise subject to change. A robust one can be kept. -- See Note [Fragile unfoldings] isFragileUnfolding :: Unfolding -> Bool isFragileUnfolding (CoreUnfolding {}) = Bool True isFragileUnfolding (DFunUnfolding {}) = Bool True isFragileUnfolding _ = Bool False -- NoUnfolding, BootUnfolding, OtherCon are all non-fragile canUnfold :: Unfolding -> Bool canUnfold :: Unfolding -> Bool canUnfold (CoreUnfolding { uf_guidance :: Unfolding -> UnfoldingGuidance uf_guidance = UnfoldingGuidance g }) = Bool -> Bool not (UnfoldingGuidance -> Bool neverUnfoldGuidance UnfoldingGuidance g) canUnfold _ = Bool False {- Note [Fragile unfoldings] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~ An unfolding is "fragile" if it mentions free variables (and hence would need substitution) or might be affected by optimisation. The non-fragile ones are NoUnfolding, BootUnfolding OtherCon {} If we know this binder (say a lambda binder) will be bound to an evaluated thing, we want to retain that info in simpleOptExpr; see Trac #13077. We consider even a StableUnfolding as fragile, because it needs substitution. Note [InlineStable] ~~~~~~~~~~~~~~~~~ When you say {-# INLINE f #-} f x = <rhs> you intend that calls (f e) are replaced by <rhs>[e/x] So we should capture (\x.<rhs>) in the Unfolding of 'f', and never meddle with it. Meanwhile, we can optimise <rhs> to our heart's content, leaving the original unfolding intact in Unfolding of 'f'. For example all xs = foldr (&&) True xs any p = all . map p {-# INLINE any #-} We optimise any's RHS fully, but leave the InlineRule saying "all . map p", which deforests well at the call site. So INLINE pragma gives rise to an InlineRule, which captures the original RHS. Moreover, it's only used when 'f' is applied to the specified number of arguments; that is, the number of argument on the LHS of the '=' sign in the original source definition. For example, (.) is now defined in the libraries like this {-# INLINE (.) #-} (.) f g = \x -> f (g x) so that it'll inline when applied to two arguments. If 'x' appeared on the left, thus (.) f g x = f (g x) it'd only inline when applied to three arguments. This slightly-experimental change was requested by Roman, but it seems to make sense. See also Note [Inlining an InlineRule] in CoreUnfold. Note [OccInfo in unfoldings and rules] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ In unfoldings and rules, we guarantee that the template is occ-analysed, so that the occurrence info on the binders is correct. This is important, because the Simplifier does not re-analyse the template when using it. If the occurrence info is wrong - We may get more simplifier iterations than necessary, because once-occ info isn't there - More seriously, we may get an infinite loop if there's a Rec without a loop breaker marked ************************************************************************ * * AltCon * * ************************************************************************ -} -- The Ord is needed for the FiniteMap used in the lookForConstructor -- in SimplEnv. If you declared that lookForConstructor *ignores* -- constructor-applications with LitArg args, then you could get -- rid of this Ord. instance Outputable AltCon where ppr :: AltCon -> SDoc ppr (DataAlt dc :: DataCon dc) = DataCon -> SDoc forall a. Outputable a => a -> SDoc ppr DataCon dc ppr (LitAlt lit :: Literal lit) = Literal -> SDoc forall a. Outputable a => a -> SDoc ppr Literal lit ppr DEFAULT = String -> SDoc text "__DEFAULT" cmpAlt :: (AltCon, a, b) -> (AltCon, a, b) -> Ordering cmpAlt :: (AltCon, a, b) -> (AltCon, a, b) -> Ordering cmpAlt (con1 :: AltCon con1, _, _) (con2 :: AltCon con2, _, _) = AltCon con1 AltCon -> AltCon -> Ordering `cmpAltCon` AltCon con2 ltAlt :: (AltCon, a, b) -> (AltCon, a, b) -> Bool ltAlt :: (AltCon, a, b) -> (AltCon, a, b) -> Bool ltAlt a1 :: (AltCon, a, b) a1 a2 :: (AltCon, a, b) a2 = ((AltCon, a, b) a1 (AltCon, a, b) -> (AltCon, a, b) -> Ordering forall a b. (AltCon, a, b) -> (AltCon, a, b) -> Ordering `cmpAlt` (AltCon, a, b) a2) Ordering -> Ordering -> Bool forall a. Eq a => a -> a -> Bool == Ordering LT cmpAltCon :: AltCon -> AltCon -> Ordering -- ^ Compares 'AltCon's within a single list of alternatives -- DEFAULT comes out smallest, so that sorting by AltCon -- puts alternatives in the order required by #case_invariants# cmpAltCon :: AltCon -> AltCon -> Ordering cmpAltCon DEFAULT DEFAULT = Ordering EQ cmpAltCon DEFAULT _ = Ordering LT cmpAltCon (DataAlt d1 :: DataCon d1) (DataAlt d2 :: DataCon d2) = DataCon -> Int dataConTag DataCon d1 Int -> Int -> Ordering forall a. Ord a => a -> a -> Ordering `compare` DataCon -> Int dataConTag DataCon d2 cmpAltCon (DataAlt _) DEFAULT = Ordering GT cmpAltCon (LitAlt l1 :: Literal l1) (LitAlt l2 :: Literal l2) = Literal l1 Literal -> Literal -> Ordering forall a. Ord a => a -> a -> Ordering `compare` Literal l2 cmpAltCon (LitAlt _) DEFAULT = Ordering GT cmpAltCon con1 :: AltCon con1 con2 :: AltCon con2 = WARN( True, text "Comparing incomparable AltCons" <+> ppr con1 <+> ppr con2 ) Ordering LT {- ************************************************************************ * * \subsection{Useful synonyms} * * ************************************************************************ Note [CoreProgram] ~~~~~~~~~~~~~~~~~~ The top level bindings of a program, a CoreProgram, are represented as a list of CoreBind * Later bindings in the list can refer to earlier ones, but not vice versa. So this is OK NonRec { x = 4 } Rec { p = ...q...x... ; q = ...p...x } Rec { f = ...p..x..f.. } NonRec { g = ..f..q...x.. } But it would NOT be ok for 'f' to refer to 'g'. * The occurrence analyser does strongly-connected component analysis on each Rec binding, and splits it into a sequence of smaller bindings where possible. So the program typically starts life as a single giant Rec, which is then dependency-analysed into smaller chunks. -} -- If you edit this type, you may need to update the GHC formalism -- See Note [GHC Formalism] in coreSyn/CoreLint.hs type CoreProgram = [CoreBind] -- See Note [CoreProgram] -- | The common case for the type of binders and variables when -- we are manipulating the Core language within GHC type CoreBndr = Var -- | Expressions where binders are 'CoreBndr's type CoreExpr = Expr CoreBndr -- | Argument expressions where binders are 'CoreBndr's type CoreArg = Arg CoreBndr -- | Binding groups where binders are 'CoreBndr's type CoreBind = Bind CoreBndr -- | Case alternatives where binders are 'CoreBndr's type CoreAlt = Alt CoreBndr {- ************************************************************************ * * \subsection{Tagging} * * ************************************************************************ -} -- | Binders are /tagged/ with a t data TaggedBndr t = TB CoreBndr t -- TB for "tagged binder" type TaggedBind t = Bind (TaggedBndr t) type TaggedExpr t = Expr (TaggedBndr t) type TaggedArg t = Arg (TaggedBndr t) type TaggedAlt t = Alt (TaggedBndr t) instance Outputable b => Outputable (TaggedBndr b) where ppr :: TaggedBndr b -> SDoc ppr (TB b :: CoreBndr b l :: b l) = Char -> SDoc char '<' SDoc -> SDoc -> SDoc <> CoreBndr -> SDoc forall a. Outputable a => a -> SDoc ppr CoreBndr b SDoc -> SDoc -> SDoc <> SDoc comma SDoc -> SDoc -> SDoc <> b -> SDoc forall a. Outputable a => a -> SDoc ppr b l SDoc -> SDoc -> SDoc <> Char -> SDoc char '>' deTagExpr :: TaggedExpr t -> CoreExpr deTagExpr :: TaggedExpr t -> CoreExpr deTagExpr (Var v :: CoreBndr v) = CoreBndr -> CoreExpr forall b. CoreBndr -> Expr b Var CoreBndr v deTagExpr (Lit l :: Literal l) = Literal -> CoreExpr forall b. Literal -> Expr b Lit Literal l deTagExpr (Type ty :: Type ty) = Type -> CoreExpr forall b. Type -> Expr b Type Type ty deTagExpr (Coercion co :: Coercion co) = Coercion -> CoreExpr forall b. Coercion -> Expr b Coercion Coercion co deTagExpr (App e1 :: TaggedExpr t e1 e2 :: TaggedExpr t e2) = CoreExpr -> CoreExpr -> CoreExpr forall b. Expr b -> Expr b -> Expr b App (TaggedExpr t -> CoreExpr forall t. TaggedExpr t -> CoreExpr deTagExpr TaggedExpr t e1) (TaggedExpr t -> CoreExpr forall t. TaggedExpr t -> CoreExpr deTagExpr TaggedExpr t e2) deTagExpr (Lam (TB b :: CoreBndr b _) e :: TaggedExpr t e) = CoreBndr -> CoreExpr -> CoreExpr forall b. b -> Expr b -> Expr b Lam CoreBndr b (TaggedExpr t -> CoreExpr forall t. TaggedExpr t -> CoreExpr deTagExpr TaggedExpr t e) deTagExpr (Let bind :: Bind (TaggedBndr t) bind body :: TaggedExpr t body) = Bind CoreBndr -> CoreExpr -> CoreExpr forall b. Bind b -> Expr b -> Expr b Let (Bind (TaggedBndr t) -> Bind CoreBndr forall t. TaggedBind t -> Bind CoreBndr deTagBind Bind (TaggedBndr t) bind) (TaggedExpr t -> CoreExpr forall t. TaggedExpr t -> CoreExpr deTagExpr TaggedExpr t body) deTagExpr (Case e :: TaggedExpr t e (TB b :: CoreBndr b _) ty :: Type ty alts :: [Alt (TaggedBndr t)] alts) = CoreExpr -> CoreBndr -> Type -> [Alt CoreBndr] -> CoreExpr forall b. Expr b -> b -> Type -> [Alt b] -> Expr b Case (TaggedExpr t -> CoreExpr forall t. TaggedExpr t -> CoreExpr deTagExpr TaggedExpr t e) CoreBndr b Type ty ((Alt (TaggedBndr t) -> Alt CoreBndr) -> [Alt (TaggedBndr t)] -> [Alt CoreBndr] forall a b. (a -> b) -> [a] -> [b] map Alt (TaggedBndr t) -> Alt CoreBndr forall t. TaggedAlt t -> Alt CoreBndr deTagAlt [Alt (TaggedBndr t)] alts) deTagExpr (Tick t :: Tickish CoreBndr t e :: TaggedExpr t e) = Tickish CoreBndr -> CoreExpr -> CoreExpr forall b. Tickish CoreBndr -> Expr b -> Expr b Tick Tickish CoreBndr t (TaggedExpr t -> CoreExpr forall t. TaggedExpr t -> CoreExpr deTagExpr TaggedExpr t e) deTagExpr (Cast e :: TaggedExpr t e co :: Coercion co) = CoreExpr -> Coercion -> CoreExpr forall b. Expr b -> Coercion -> Expr b Cast (TaggedExpr t -> CoreExpr forall t. TaggedExpr t -> CoreExpr deTagExpr TaggedExpr t e) Coercion co deTagBind :: TaggedBind t -> CoreBind deTagBind :: TaggedBind t -> Bind CoreBndr deTagBind (NonRec (TB b :: CoreBndr b _) rhs :: Expr (TaggedBndr t) rhs) = CoreBndr -> CoreExpr -> Bind CoreBndr forall b. b -> Expr b -> Bind b NonRec CoreBndr b (Expr (TaggedBndr t) -> CoreExpr forall t. TaggedExpr t -> CoreExpr deTagExpr Expr (TaggedBndr t) rhs) deTagBind (Rec prs :: [(TaggedBndr t, Expr (TaggedBndr t))] prs) = [(CoreBndr, CoreExpr)] -> Bind CoreBndr forall b. [(b, Expr b)] -> Bind b Rec [(CoreBndr b, Expr (TaggedBndr t) -> CoreExpr forall t. TaggedExpr t -> CoreExpr deTagExpr Expr (TaggedBndr t) rhs) | (TB b :: CoreBndr b _, rhs :: Expr (TaggedBndr t) rhs) <- [(TaggedBndr t, Expr (TaggedBndr t))] prs] deTagAlt :: TaggedAlt t -> CoreAlt deTagAlt :: TaggedAlt t -> Alt CoreBndr deTagAlt (con :: AltCon con, bndrs :: [TaggedBndr t] bndrs, rhs :: Expr (TaggedBndr t) rhs) = (AltCon con, [CoreBndr b | TB b :: CoreBndr b _ <- [TaggedBndr t] bndrs], Expr (TaggedBndr t) -> CoreExpr forall t. TaggedExpr t -> CoreExpr deTagExpr Expr (TaggedBndr t) rhs) {- ************************************************************************ * * \subsection{Core-constructing functions with checking} * * ************************************************************************ -} -- | Apply a list of argument expressions to a function expression in a nested fashion. Prefer to -- use 'MkCore.mkCoreApps' if possible mkApps :: Expr b -> [Arg b] -> Expr b -- | Apply a list of type argument expressions to a function expression in a nested fashion mkTyApps :: Expr b -> [Type] -> Expr b -- | Apply a list of coercion argument expressions to a function expression in a nested fashion mkCoApps :: Expr b -> [Coercion] -> Expr b -- | Apply a list of type or value variables to a function expression in a nested fashion mkVarApps :: Expr b -> [Var] -> Expr b -- | Apply a list of argument expressions to a data constructor in a nested fashion. Prefer to -- use 'MkCore.mkCoreConApps' if possible mkConApp :: DataCon -> [Arg b] -> Expr b mkApps :: Expr b -> [Expr b] -> Expr b mkApps f :: Expr b f args :: [Expr b] args = (Expr b -> Expr b -> Expr b) -> Expr b -> [Expr b] -> Expr b forall (t :: * -> *) b a. Foldable t => (b -> a -> b) -> b -> t a -> b foldl' Expr b -> Expr b -> Expr b forall b. Expr b -> Expr b -> Expr b App Expr b f [Expr b] args mkCoApps :: Expr b -> [Coercion] -> Expr b mkCoApps f :: Expr b f args :: [Coercion] args = (Expr b -> Coercion -> Expr b) -> Expr b -> [Coercion] -> Expr b forall (t :: * -> *) b a. Foldable t => (b -> a -> b) -> b -> t a -> b foldl' (\ e :: Expr b e a :: Coercion a -> Expr b -> Expr b -> Expr b forall b. Expr b -> Expr b -> Expr b App Expr b e (Coercion -> Expr b forall b. Coercion -> Expr b Coercion Coercion a)) Expr b f [Coercion] args mkVarApps :: Expr b -> [CoreBndr] -> Expr b mkVarApps f :: Expr b f vars :: [CoreBndr] vars = (Expr b -> CoreBndr -> Expr b) -> Expr b -> [CoreBndr] -> Expr b forall (t :: * -> *) b a. Foldable t => (b -> a -> b) -> b -> t a -> b foldl' (\ e :: Expr b e a :: CoreBndr a -> Expr b -> Expr b -> Expr b forall b. Expr b -> Expr b -> Expr b App Expr b e (CoreBndr -> Expr b forall b. CoreBndr -> Expr b varToCoreExpr CoreBndr a)) Expr b f [CoreBndr] vars mkConApp :: DataCon -> [Arg b] -> Arg b mkConApp con :: DataCon con args :: [Arg b] args = Arg b -> [Arg b] -> Arg b forall b. Expr b -> [Expr b] -> Expr b mkApps (CoreBndr -> Arg b forall b. CoreBndr -> Expr b Var (DataCon -> CoreBndr dataConWorkId DataCon con)) [Arg b] args mkTyApps :: Expr b -> [Type] -> Expr b mkTyApps f :: Expr b f args :: [Type] args = (Expr b -> Type -> Expr b) -> Expr b -> [Type] -> Expr b forall (t :: * -> *) b a. Foldable t => (b -> a -> b) -> b -> t a -> b foldl' (\ e :: Expr b e a :: Type a -> Expr b -> Expr b -> Expr b forall b. Expr b -> Expr b -> Expr b App Expr b e (Type -> Expr b forall b. Type -> Expr b mkTyArg Type a)) Expr b f [Type] args mkConApp2 :: DataCon -> [Type] -> [Var] -> Expr b mkConApp2 :: DataCon -> [Type] -> [CoreBndr] -> Expr b mkConApp2 con :: DataCon con tys :: [Type] tys arg_ids :: [CoreBndr] arg_ids = CoreBndr -> Expr b forall b. CoreBndr -> Expr b Var (DataCon -> CoreBndr dataConWorkId DataCon con) Expr b -> [Expr b] -> Expr b forall b. Expr b -> [Expr b] -> Expr b `mkApps` (Type -> Expr b) -> [Type] -> [Expr b] forall a b. (a -> b) -> [a] -> [b] map Type -> Expr b forall b. Type -> Expr b Type [Type] tys Expr b -> [Expr b] -> Expr b forall b. Expr b -> [Expr b] -> Expr b `mkApps` (CoreBndr -> Expr b) -> [CoreBndr] -> [Expr b] forall a b. (a -> b) -> [a] -> [b] map CoreBndr -> Expr b forall b. CoreBndr -> Expr b varToCoreExpr [CoreBndr] arg_ids mkTyArg :: Type -> Expr b mkTyArg :: Type -> Expr b mkTyArg ty :: Type ty | Just co :: Coercion co <- Type -> Maybe Coercion isCoercionTy_maybe Type ty = Coercion -> Expr b forall b. Coercion -> Expr b Coercion Coercion co | Bool otherwise = Type -> Expr b forall b. Type -> Expr b Type Type ty -- | Create a machine integer literal expression of type @Int#@ from an @Integer@. -- If you want an expression of type @Int@ use 'MkCore.mkIntExpr' mkIntLit :: DynFlags -> Integer -> Expr b -- | Create a machine integer literal expression of type @Int#@ from an @Int@. -- If you want an expression of type @Int@ use 'MkCore.mkIntExpr' mkIntLitInt :: DynFlags -> Int -> Expr b mkIntLit :: DynFlags -> Integer -> Expr b mkIntLit dflags :: DynFlags dflags n :: Integer n = Literal -> Expr b forall b. Literal -> Expr b Lit (DynFlags -> Integer -> Literal mkLitInt DynFlags dflags Integer n) mkIntLitInt :: DynFlags -> Int -> Expr b mkIntLitInt dflags :: DynFlags dflags n :: Int n = Literal -> Expr b forall b. Literal -> Expr b Lit (DynFlags -> Integer -> Literal mkLitInt DynFlags dflags (Int -> Integer forall a. Integral a => a -> Integer toInteger Int n)) -- | Create a machine word literal expression of type @Word#@ from an @Integer@. -- If you want an expression of type @Word@ use 'MkCore.mkWordExpr' mkWordLit :: DynFlags -> Integer -> Expr b -- | Create a machine word literal expression of type @Word#@ from a @Word@. -- If you want an expression of type @Word@ use 'MkCore.mkWordExpr' mkWordLitWord :: DynFlags -> Word -> Expr b mkWordLit :: DynFlags -> Integer -> Expr b mkWordLit dflags :: DynFlags dflags w :: Integer w = Literal -> Expr b forall b. Literal -> Expr b Lit (DynFlags -> Integer -> Literal mkLitWord DynFlags dflags Integer w) mkWordLitWord :: DynFlags -> Word -> Expr b mkWordLitWord dflags :: DynFlags dflags w :: Word w = Literal -> Expr b forall b. Literal -> Expr b Lit (DynFlags -> Integer -> Literal mkLitWord DynFlags dflags (Word -> Integer forall a. Integral a => a -> Integer toInteger Word w)) mkWord64LitWord64 :: Word64 -> Expr b mkWord64LitWord64 :: Word64 -> Expr b mkWord64LitWord64 w :: Word64 w = Literal -> Expr b forall b. Literal -> Expr b Lit (Integer -> Literal mkLitWord64 (Word64 -> Integer forall a. Integral a => a -> Integer toInteger Word64 w)) mkInt64LitInt64 :: Int64 -> Expr b mkInt64LitInt64 :: Int64 -> Expr b mkInt64LitInt64 w :: Int64 w = Literal -> Expr b forall b. Literal -> Expr b Lit (Integer -> Literal mkLitInt64 (Int64 -> Integer forall a. Integral a => a -> Integer toInteger Int64 w)) -- | Create a machine character literal expression of type @Char#@. -- If you want an expression of type @Char@ use 'MkCore.mkCharExpr' mkCharLit :: Char -> Expr b -- | Create a machine string literal expression of type @Addr#@. -- If you want an expression of type @String@ use 'MkCore.mkStringExpr' mkStringLit :: String -> Expr b mkCharLit :: Char -> Expr b mkCharLit c :: Char c = Literal -> Expr b forall b. Literal -> Expr b Lit (Char -> Literal mkLitChar Char c) mkStringLit :: String -> Expr b mkStringLit s :: String s = Literal -> Expr b forall b. Literal -> Expr b Lit (String -> Literal mkLitString String s) -- | Create a machine single precision literal expression of type @Float#@ from a @Rational@. -- If you want an expression of type @Float@ use 'MkCore.mkFloatExpr' mkFloatLit :: Rational -> Expr b -- | Create a machine single precision literal expression of type @Float#@ from a @Float@. -- If you want an expression of type @Float@ use 'MkCore.mkFloatExpr' mkFloatLitFloat :: Float -> Expr b mkFloatLit :: Rational -> Expr b mkFloatLit f :: Rational f = Literal -> Expr b forall b. Literal -> Expr b Lit (Rational -> Literal mkLitFloat Rational f) mkFloatLitFloat :: Float -> Expr b mkFloatLitFloat f :: Float f = Literal -> Expr b forall b. Literal -> Expr b Lit (Rational -> Literal mkLitFloat (Float -> Rational forall a. Real a => a -> Rational toRational Float f)) -- | Create a machine double precision literal expression of type @Double#@ from a @Rational@. -- If you want an expression of type @Double@ use 'MkCore.mkDoubleExpr' mkDoubleLit :: Rational -> Expr b -- | Create a machine double precision literal expression of type @Double#@ from a @Double@. -- If you want an expression of type @Double@ use 'MkCore.mkDoubleExpr' mkDoubleLitDouble :: Double -> Expr b mkDoubleLit :: Rational -> Expr b mkDoubleLit d :: Rational d = Literal -> Expr b forall b. Literal -> Expr b Lit (Rational -> Literal mkLitDouble Rational d) mkDoubleLitDouble :: Double -> Expr b mkDoubleLitDouble d :: Double d = Literal -> Expr b forall b. Literal -> Expr b Lit (Rational -> Literal mkLitDouble (Double -> Rational forall a. Real a => a -> Rational toRational Double d)) -- | Bind all supplied binding groups over an expression in a nested let expression. Assumes -- that the rhs satisfies the let/app invariant. Prefer to use 'MkCore.mkCoreLets' if -- possible, which does guarantee the invariant mkLets :: [Bind b] -> Expr b -> Expr b -- | Bind all supplied binders over an expression in a nested lambda expression. Prefer to -- use 'MkCore.mkCoreLams' if possible mkLams :: [b] -> Expr b -> Expr b mkLams :: [b] -> Expr b -> Expr b mkLams binders :: [b] binders body :: Expr b body = (b -> Expr b -> Expr b) -> Expr b -> [b] -> Expr b forall (t :: * -> *) a b. Foldable t => (a -> b -> b) -> b -> t a -> b foldr b -> Expr b -> Expr b forall b. b -> Expr b -> Expr b Lam Expr b body [b] binders mkLets :: [Bind b] -> Expr b -> Expr b mkLets binds :: [Bind b] binds body :: Expr b body = (Bind b -> Expr b -> Expr b) -> Expr b -> [Bind b] -> Expr b forall (t :: * -> *) a b. Foldable t => (a -> b -> b) -> b -> t a -> b foldr Bind b -> Expr b -> Expr b forall b. Bind b -> Expr b -> Expr b mkLet Expr b body [Bind b] binds mkLet :: Bind b -> Expr b -> Expr b -- The desugarer sometimes generates an empty Rec group -- which Lint rejects, so we kill it off right away mkLet :: Bind b -> Expr b -> Expr b mkLet (Rec []) body :: Expr b body = Expr b body mkLet bind :: Bind b bind body :: Expr b body = Bind b -> Expr b -> Expr b forall b. Bind b -> Expr b -> Expr b Let Bind b bind Expr b body -- | @mkLetNonRec bndr rhs body@ wraps @body@ in a @let@ binding @bndr@. mkLetNonRec :: b -> Expr b -> Expr b -> Expr b mkLetNonRec :: b -> Expr b -> Expr b -> Expr b mkLetNonRec b :: b b rhs :: Expr b rhs body :: Expr b body = Bind b -> Expr b -> Expr b forall b. Bind b -> Expr b -> Expr b Let (b -> Expr b -> Bind b forall b. b -> Expr b -> Bind b NonRec b b Expr b rhs) Expr b body -- | @mkLetRec binds body@ wraps @body@ in a @let rec@ with the given set of -- @binds@ if binds is non-empty. mkLetRec :: [(b, Expr b)] -> Expr b -> Expr b mkLetRec :: [(b, Expr b)] -> Expr b -> Expr b mkLetRec [] body :: Expr b body = Expr b body mkLetRec bs :: [(b, Expr b)] bs body :: Expr b body = Bind b -> Expr b -> Expr b forall b. Bind b -> Expr b -> Expr b Let ([(b, Expr b)] -> Bind b forall b. [(b, Expr b)] -> Bind b Rec [(b, Expr b)] bs) Expr b body -- | Create a binding group where a type variable is bound to a type. Per "CoreSyn#type_let", -- this can only be used to bind something in a non-recursive @let@ expression mkTyBind :: TyVar -> Type -> CoreBind mkTyBind :: CoreBndr -> Type -> Bind CoreBndr mkTyBind tv :: CoreBndr tv ty :: Type ty = CoreBndr -> CoreExpr -> Bind CoreBndr forall b. b -> Expr b -> Bind b NonRec CoreBndr tv (Type -> CoreExpr forall b. Type -> Expr b Type Type ty) -- | Create a binding group where a type variable is bound to a type. Per "CoreSyn#type_let", -- this can only be used to bind something in a non-recursive @let@ expression mkCoBind :: CoVar -> Coercion -> CoreBind mkCoBind :: CoreBndr -> Coercion -> Bind CoreBndr mkCoBind cv :: CoreBndr cv co :: Coercion co = CoreBndr -> CoreExpr -> Bind CoreBndr forall b. b -> Expr b -> Bind b NonRec CoreBndr cv (Coercion -> CoreExpr forall b. Coercion -> Expr b Coercion Coercion co) -- | Convert a binder into either a 'Var' or 'Type' 'Expr' appropriately varToCoreExpr :: CoreBndr -> Expr b varToCoreExpr :: CoreBndr -> Expr b varToCoreExpr v :: CoreBndr v | CoreBndr -> Bool isTyVar CoreBndr v = Type -> Expr b forall b. Type -> Expr b Type (CoreBndr -> Type mkTyVarTy CoreBndr v) | CoreBndr -> Bool isCoVar CoreBndr v = Coercion -> Expr b forall b. Coercion -> Expr b Coercion (CoreBndr -> Coercion mkCoVarCo CoreBndr v) | Bool otherwise = ASSERT( isId v ) Var v varsToCoreExprs :: [CoreBndr] -> [Expr b] varsToCoreExprs :: [CoreBndr] -> [Expr b] varsToCoreExprs vs :: [CoreBndr] vs = (CoreBndr -> Expr b) -> [CoreBndr] -> [Expr b] forall a b. (a -> b) -> [a] -> [b] map CoreBndr -> Expr b forall b. CoreBndr -> Expr b varToCoreExpr [CoreBndr] vs {- ************************************************************************ * * Getting a result type * * ************************************************************************ These are defined here to avoid a module loop between CoreUtils and CoreFVs -} applyTypeToArg :: Type -> CoreExpr -> Type -- ^ Determines the type resulting from applying an expression with given type -- to a given argument expression applyTypeToArg :: Type -> CoreExpr -> Type applyTypeToArg fun_ty :: Type fun_ty arg :: CoreExpr arg = HasDebugCallStack => Type -> Type -> Type Type -> Type -> Type piResultTy Type fun_ty (CoreExpr -> Type exprToType CoreExpr arg) -- | If the expression is a 'Type', converts. Otherwise, -- panics. NB: This does /not/ convert 'Coercion' to 'CoercionTy'. exprToType :: CoreExpr -> Type exprToType :: CoreExpr -> Type exprToType (Type ty :: Type ty) = Type ty exprToType _bad :: CoreExpr _bad = String -> SDoc -> Type forall a. HasCallStack => String -> SDoc -> a pprPanic "exprToType" SDoc empty -- | If the expression is a 'Coercion', converts. exprToCoercion_maybe :: CoreExpr -> Maybe Coercion exprToCoercion_maybe :: CoreExpr -> Maybe Coercion exprToCoercion_maybe (Coercion co :: Coercion co) = Coercion -> Maybe Coercion forall a. a -> Maybe a Just Coercion co exprToCoercion_maybe _ = Maybe Coercion forall a. Maybe a Nothing {- ************************************************************************ * * \subsection{Simple access functions} * * ************************************************************************ -} -- | Extract every variable by this group bindersOf :: Bind b -> [b] -- If you edit this function, you may need to update the GHC formalism -- See Note [GHC Formalism] in coreSyn/CoreLint.hs bindersOf :: Bind b -> [b] bindersOf (NonRec binder :: b binder _) = [b binder] bindersOf (Rec pairs :: [(b, Expr b)] pairs) = [b binder | (binder :: b binder, _) <- [(b, Expr b)] pairs] -- | 'bindersOf' applied to a list of binding groups bindersOfBinds :: [Bind b] -> [b] bindersOfBinds :: [Bind b] -> [b] bindersOfBinds binds :: [Bind b] binds = (Bind b -> [b] -> [b]) -> [b] -> [Bind b] -> [b] forall (t :: * -> *) a b. Foldable t => (a -> b -> b) -> b -> t a -> b foldr ([b] -> [b] -> [b] forall a. [a] -> [a] -> [a] (++) ([b] -> [b] -> [b]) -> (Bind b -> [b]) -> Bind b -> [b] -> [b] forall b c a. (b -> c) -> (a -> b) -> a -> c . Bind b -> [b] forall b. Bind b -> [b] bindersOf) [] [Bind b] binds rhssOfBind :: Bind b -> [Expr b] rhssOfBind :: Bind b -> [Expr b] rhssOfBind (NonRec _ rhs :: Expr b rhs) = [Expr b rhs] rhssOfBind (Rec pairs :: [(b, Expr b)] pairs) = [Expr b rhs | (_,rhs :: Expr b rhs) <- [(b, Expr b)] pairs] rhssOfAlts :: [Alt b] -> [Expr b] rhssOfAlts :: [Alt b] -> [Expr b] rhssOfAlts alts :: [Alt b] alts = [Expr b e | (_,_,e :: Expr b e) <- [Alt b] alts] -- | Collapse all the bindings in the supplied groups into a single -- list of lhs\/rhs pairs suitable for binding in a 'Rec' binding group flattenBinds :: [Bind b] -> [(b, Expr b)] flattenBinds :: [Bind b] -> [(b, Expr b)] flattenBinds (NonRec b :: b b r :: Expr b r : binds :: [Bind b] binds) = (b b,Expr b r) (b, Expr b) -> [(b, Expr b)] -> [(b, Expr b)] forall a. a -> [a] -> [a] : [Bind b] -> [(b, Expr b)] forall b. [Bind b] -> [(b, Expr b)] flattenBinds [Bind b] binds flattenBinds (Rec prs1 :: [(b, Expr b)] prs1 : binds :: [Bind b] binds) = [(b, Expr b)] prs1 [(b, Expr b)] -> [(b, Expr b)] -> [(b, Expr b)] forall a. [a] -> [a] -> [a] ++ [Bind b] -> [(b, Expr b)] forall b. [Bind b] -> [(b, Expr b)] flattenBinds [Bind b] binds flattenBinds [] = [] -- | We often want to strip off leading lambdas before getting down to -- business. Variants are 'collectTyBinders', 'collectValBinders', -- and 'collectTyAndValBinders' collectBinders :: Expr b -> ([b], Expr b) collectTyBinders :: CoreExpr -> ([TyVar], CoreExpr) collectValBinders :: CoreExpr -> ([Id], CoreExpr) collectTyAndValBinders :: CoreExpr -> ([TyVar], [Id], CoreExpr) -- | Strip off exactly N leading lambdas (type or value). Good for use with -- join points. collectNBinders :: Int -> Expr b -> ([b], Expr b) collectBinders :: Expr b -> ([b], Expr b) collectBinders expr :: Expr b expr = [b] -> Expr b -> ([b], Expr b) forall a. [a] -> Expr a -> ([a], Expr a) go [] Expr b expr where go :: [a] -> Expr a -> ([a], Expr a) go bs :: [a] bs (Lam b :: a b e :: Expr a e) = [a] -> Expr a -> ([a], Expr a) go (a ba -> [a] -> [a] forall a. a -> [a] -> [a] :[a] bs) Expr a e go bs :: [a] bs e :: Expr a e = ([a] -> [a] forall a. [a] -> [a] reverse [a] bs, Expr a e) collectTyBinders :: CoreExpr -> ([CoreBndr], CoreExpr) collectTyBinders expr :: CoreExpr expr = [CoreBndr] -> CoreExpr -> ([CoreBndr], CoreExpr) go [] CoreExpr expr where go :: [CoreBndr] -> CoreExpr -> ([CoreBndr], CoreExpr) go tvs :: [CoreBndr] tvs (Lam b :: CoreBndr b e :: CoreExpr e) | CoreBndr -> Bool isTyVar CoreBndr b = [CoreBndr] -> CoreExpr -> ([CoreBndr], CoreExpr) go (CoreBndr bCoreBndr -> [CoreBndr] -> [CoreBndr] forall a. a -> [a] -> [a] :[CoreBndr] tvs) CoreExpr e go tvs :: [CoreBndr] tvs e :: CoreExpr e = ([CoreBndr] -> [CoreBndr] forall a. [a] -> [a] reverse [CoreBndr] tvs, CoreExpr e) collectValBinders :: CoreExpr -> ([CoreBndr], CoreExpr) collectValBinders expr :: CoreExpr expr = [CoreBndr] -> CoreExpr -> ([CoreBndr], CoreExpr) go [] CoreExpr expr where go :: [CoreBndr] -> CoreExpr -> ([CoreBndr], CoreExpr) go ids :: [CoreBndr] ids (Lam b :: CoreBndr b e :: CoreExpr e) | CoreBndr -> Bool isId CoreBndr b = [CoreBndr] -> CoreExpr -> ([CoreBndr], CoreExpr) go (CoreBndr bCoreBndr -> [CoreBndr] -> [CoreBndr] forall a. a -> [a] -> [a] :[CoreBndr] ids) CoreExpr e go ids :: [CoreBndr] ids body :: CoreExpr body = ([CoreBndr] -> [CoreBndr] forall a. [a] -> [a] reverse [CoreBndr] ids, CoreExpr body) collectTyAndValBinders :: CoreExpr -> ([CoreBndr], [CoreBndr], CoreExpr) collectTyAndValBinders expr :: CoreExpr expr = ([CoreBndr] tvs, [CoreBndr] ids, CoreExpr body) where (tvs :: [CoreBndr] tvs, body1 :: CoreExpr body1) = CoreExpr -> ([CoreBndr], CoreExpr) collectTyBinders CoreExpr expr (ids :: [CoreBndr] ids, body :: CoreExpr body) = CoreExpr -> ([CoreBndr], CoreExpr) collectValBinders CoreExpr body1 collectNBinders :: Int -> Expr b -> ([b], Expr b) collectNBinders orig_n :: Int orig_n orig_expr :: Expr b orig_expr = Int -> [b] -> Expr b -> ([b], Expr b) forall t a. (Eq t, Num t) => t -> [a] -> Expr a -> ([a], Expr a) go Int orig_n [] Expr b orig_expr where go :: t -> [a] -> Expr a -> ([a], Expr a) go 0 bs :: [a] bs expr :: Expr a expr = ([a] -> [a] forall a. [a] -> [a] reverse [a] bs, Expr a expr) go n :: t n bs :: [a] bs (Lam b :: a b e :: Expr a e) = t -> [a] -> Expr a -> ([a], Expr a) go (t nt -> t -> t forall a. Num a => a -> a -> a -1) (a ba -> [a] -> [a] forall a. a -> [a] -> [a] :[a] bs) Expr a e go _ _ _ = String -> SDoc -> ([a], Expr a) forall a. HasCallStack => String -> SDoc -> a pprPanic "collectNBinders" (SDoc -> ([a], Expr a)) -> SDoc -> ([a], Expr a) forall a b. (a -> b) -> a -> b $ Int -> SDoc int Int orig_n -- | Takes a nested application expression and returns the function -- being applied and the arguments to which it is applied collectArgs :: Expr b -> (Expr b, [Arg b]) collectArgs :: Expr b -> (Expr b, [Expr b]) collectArgs expr :: Expr b expr = Expr b -> [Expr b] -> (Expr b, [Expr b]) forall b. Expr b -> [Expr b] -> (Expr b, [Expr b]) go Expr b expr [] where go :: Expr b -> [Expr b] -> (Expr b, [Expr b]) go (App f :: Expr b f a :: Expr b a) as :: [Expr b] as = Expr b -> [Expr b] -> (Expr b, [Expr b]) go Expr b f (Expr b aExpr b -> [Expr b] -> [Expr b] forall a. a -> [a] -> [a] :[Expr b] as) go e :: Expr b e as :: [Expr b] as = (Expr b e, [Expr b] as) -- | Attempt to remove the last N arguments of a function call. -- Strip off any ticks or coercions encountered along the way and any -- at the end. stripNArgs :: Word -> Expr a -> Maybe (Expr a) stripNArgs :: Word -> Expr a -> Maybe (Expr a) stripNArgs !Word n (Tick _ e :: Expr a e) = Word -> Expr a -> Maybe (Expr a) forall a. Word -> Expr a -> Maybe (Expr a) stripNArgs Word n Expr a e stripNArgs n :: Word n (Cast f :: Expr a f _) = Word -> Expr a -> Maybe (Expr a) forall a. Word -> Expr a -> Maybe (Expr a) stripNArgs Word n Expr a f stripNArgs 0 e :: Expr a e = Expr a -> Maybe (Expr a) forall a. a -> Maybe a Just Expr a e stripNArgs n :: Word n (App f :: Expr a f _) = Word -> Expr a -> Maybe (Expr a) forall a. Word -> Expr a -> Maybe (Expr a) stripNArgs (Word n Word -> Word -> Word forall a. Num a => a -> a -> a - 1) Expr a f stripNArgs _ _ = Maybe (Expr a) forall a. Maybe a Nothing -- | Like @collectArgs@, but also collects looks through floatable -- ticks if it means that we can find more arguments. collectArgsTicks :: (Tickish Id -> Bool) -> Expr b -> (Expr b, [Arg b], [Tickish Id]) collectArgsTicks :: (Tickish CoreBndr -> Bool) -> Expr b -> (Expr b, [Expr b], [Tickish CoreBndr]) collectArgsTicks skipTick :: Tickish CoreBndr -> Bool skipTick expr :: Expr b expr = Expr b -> [Expr b] -> [Tickish CoreBndr] -> (Expr b, [Expr b], [Tickish CoreBndr]) forall b. Expr b -> [Expr b] -> [Tickish CoreBndr] -> (Expr b, [Expr b], [Tickish CoreBndr]) go Expr b expr [] [] where go :: Expr b -> [Expr b] -> [Tickish CoreBndr] -> (Expr b, [Expr b], [Tickish CoreBndr]) go (App f :: Expr b f a :: Expr b a) as :: [Expr b] as ts :: [Tickish CoreBndr] ts = Expr b -> [Expr b] -> [Tickish CoreBndr] -> (Expr b, [Expr b], [Tickish CoreBndr]) go Expr b f (Expr b aExpr b -> [Expr b] -> [Expr b] forall a. a -> [a] -> [a] :[Expr b] as) [Tickish CoreBndr] ts go (Tick t :: Tickish CoreBndr t e :: Expr b e) as :: [Expr b] as ts :: [Tickish CoreBndr] ts | Tickish CoreBndr -> Bool skipTick Tickish CoreBndr t = Expr b -> [Expr b] -> [Tickish CoreBndr] -> (Expr b, [Expr b], [Tickish CoreBndr]) go Expr b e [Expr b] as (Tickish CoreBndr tTickish CoreBndr -> [Tickish CoreBndr] -> [Tickish CoreBndr] forall a. a -> [a] -> [a] :[Tickish CoreBndr] ts) go e :: Expr b e as :: [Expr b] as ts :: [Tickish CoreBndr] ts = (Expr b e, [Expr b] as, [Tickish CoreBndr] -> [Tickish CoreBndr] forall a. [a] -> [a] reverse [Tickish CoreBndr] ts) {- ************************************************************************ * * \subsection{Predicates} * * ************************************************************************ At one time we optionally carried type arguments through to runtime. @isRuntimeVar v@ returns if (Lam v _) really becomes a lambda at runtime, i.e. if type applications are actual lambdas because types are kept around at runtime. Similarly isRuntimeArg. -} -- | Will this variable exist at runtime? isRuntimeVar :: Var -> Bool isRuntimeVar :: CoreBndr -> Bool isRuntimeVar = CoreBndr -> Bool isId -- | Will this argument expression exist at runtime? isRuntimeArg :: CoreExpr -> Bool isRuntimeArg :: CoreExpr -> Bool isRuntimeArg = CoreExpr -> Bool forall b. Expr b -> Bool isValArg -- | Returns @True@ for value arguments, false for type args -- NB: coercions are value arguments (zero width, to be sure, -- like State#, but still value args). isValArg :: Expr b -> Bool isValArg :: Expr b -> Bool isValArg e :: Expr b e = Bool -> Bool not (Expr b -> Bool forall b. Expr b -> Bool isTypeArg Expr b e) -- | Returns @True@ iff the expression is a 'Type' or 'Coercion' -- expression at its top level isTyCoArg :: Expr b -> Bool isTyCoArg :: Expr b -> Bool isTyCoArg (Type {}) = Bool True isTyCoArg (Coercion {}) = Bool True isTyCoArg _ = Bool False -- | Returns @True@ iff the expression is a 'Coercion' -- expression at its top level isCoArg :: Expr b -> Bool isCoArg :: Expr b -> Bool isCoArg (Coercion {}) = Bool True isCoArg _ = Bool False -- | Returns @True@ iff the expression is a 'Type' expression at its -- top level. Note this does NOT include 'Coercion's. isTypeArg :: Expr b -> Bool isTypeArg :: Expr b -> Bool isTypeArg (Type {}) = Bool True isTypeArg _ = Bool False -- | The number of binders that bind values rather than types valBndrCount :: [CoreBndr] -> Int valBndrCount :: [CoreBndr] -> Int valBndrCount = (CoreBndr -> Bool) -> [CoreBndr] -> Int forall a. (a -> Bool) -> [a] -> Int count CoreBndr -> Bool isId -- | The number of argument expressions that are values rather than types at their top level valArgCount :: [Arg b] -> Int valArgCount :: [Arg b] -> Int valArgCount = (Arg b -> Bool) -> [Arg b] -> Int forall a. (a -> Bool) -> [a] -> Int count Arg b -> Bool forall b. Expr b -> Bool isValArg {- ************************************************************************ * * \subsection{Annotated core} * * ************************************************************************ -} -- | Annotated core: allows annotation at every node in the tree type AnnExpr bndr annot = (annot, AnnExpr' bndr annot) -- | A clone of the 'Expr' type but allowing annotation at every tree node data AnnExpr' bndr annot = AnnVar Id | AnnLit Literal | AnnLam bndr (AnnExpr bndr annot) | AnnApp (AnnExpr bndr annot) (AnnExpr bndr annot) | AnnCase (AnnExpr bndr annot) bndr Type [AnnAlt bndr annot] | AnnLet (AnnBind bndr annot) (AnnExpr bndr annot) | AnnCast (AnnExpr bndr annot) (annot, Coercion) -- Put an annotation on the (root of) the coercion | AnnTick (Tickish Id) (AnnExpr bndr annot) | AnnType Type | AnnCoercion Coercion -- | A clone of the 'Alt' type but allowing annotation at every tree node type AnnAlt bndr annot = (AltCon, [bndr], AnnExpr bndr annot) -- | A clone of the 'Bind' type but allowing annotation at every tree node data AnnBind bndr annot = AnnNonRec bndr (AnnExpr bndr annot) | AnnRec [(bndr, AnnExpr bndr annot)] -- | Takes a nested application expression and returns the function -- being applied and the arguments to which it is applied collectAnnArgs :: AnnExpr b a -> (AnnExpr b a, [AnnExpr b a]) collectAnnArgs :: AnnExpr b a -> (AnnExpr b a, [AnnExpr b a]) collectAnnArgs expr :: AnnExpr b a expr = AnnExpr b a -> [AnnExpr b a] -> (AnnExpr b a, [AnnExpr b a]) forall bndr annot. AnnExpr bndr annot -> [AnnExpr bndr annot] -> (AnnExpr bndr annot, [AnnExpr bndr annot]) go AnnExpr b a expr [] where go :: AnnExpr bndr annot -> [AnnExpr bndr annot] -> (AnnExpr bndr annot, [AnnExpr bndr annot]) go (_, AnnApp f :: AnnExpr bndr annot f a :: AnnExpr bndr annot a) as :: [AnnExpr bndr annot] as = AnnExpr bndr annot -> [AnnExpr bndr annot] -> (AnnExpr bndr annot, [AnnExpr bndr annot]) go AnnExpr bndr annot f (AnnExpr bndr annot aAnnExpr bndr annot -> [AnnExpr bndr annot] -> [AnnExpr bndr annot] forall a. a -> [a] -> [a] :[AnnExpr bndr annot] as) go e :: AnnExpr bndr annot e as :: [AnnExpr bndr annot] as = (AnnExpr bndr annot e, [AnnExpr bndr annot] as) collectAnnArgsTicks :: (Tickish Var -> Bool) -> AnnExpr b a -> (AnnExpr b a, [AnnExpr b a], [Tickish Var]) collectAnnArgsTicks :: (Tickish CoreBndr -> Bool) -> AnnExpr b a -> (AnnExpr b a, [AnnExpr b a], [Tickish CoreBndr]) collectAnnArgsTicks tickishOk :: Tickish CoreBndr -> Bool tickishOk expr :: AnnExpr b a expr = AnnExpr b a -> [AnnExpr b a] -> [Tickish CoreBndr] -> (AnnExpr b a, [AnnExpr b a], [Tickish CoreBndr]) forall bndr annot. AnnExpr bndr annot -> [AnnExpr bndr annot] -> [Tickish CoreBndr] -> (AnnExpr bndr annot, [AnnExpr bndr annot], [Tickish CoreBndr]) go AnnExpr b a expr [] [] where go :: AnnExpr bndr annot -> [AnnExpr bndr annot] -> [Tickish CoreBndr] -> (AnnExpr bndr annot, [AnnExpr bndr annot], [Tickish CoreBndr]) go (_, AnnApp f :: AnnExpr bndr annot f a :: AnnExpr bndr annot a) as :: [AnnExpr bndr annot] as ts :: [Tickish CoreBndr] ts = AnnExpr bndr annot -> [AnnExpr bndr annot] -> [Tickish CoreBndr] -> (AnnExpr bndr annot, [AnnExpr bndr annot], [Tickish CoreBndr]) go AnnExpr bndr annot f (AnnExpr bndr annot aAnnExpr bndr annot -> [AnnExpr bndr annot] -> [AnnExpr bndr annot] forall a. a -> [a] -> [a] :[AnnExpr bndr annot] as) [Tickish CoreBndr] ts go (_, AnnTick t :: Tickish CoreBndr t e :: AnnExpr bndr annot e) as :: [AnnExpr bndr annot] as ts :: [Tickish CoreBndr] ts | Tickish CoreBndr -> Bool tickishOk Tickish CoreBndr t = AnnExpr bndr annot -> [AnnExpr bndr annot] -> [Tickish CoreBndr] -> (AnnExpr bndr annot, [AnnExpr bndr annot], [Tickish CoreBndr]) go AnnExpr bndr annot e [AnnExpr bndr annot] as (Tickish CoreBndr tTickish CoreBndr -> [Tickish CoreBndr] -> [Tickish CoreBndr] forall a. a -> [a] -> [a] :[Tickish CoreBndr] ts) go e :: AnnExpr bndr annot e as :: [AnnExpr bndr annot] as ts :: [Tickish CoreBndr] ts = (AnnExpr bndr annot e, [AnnExpr bndr annot] as, [Tickish CoreBndr] -> [Tickish CoreBndr] forall a. [a] -> [a] reverse [Tickish CoreBndr] ts) deAnnotate :: AnnExpr bndr annot -> Expr bndr deAnnotate :: AnnExpr bndr annot -> Expr bndr deAnnotate (_, e :: AnnExpr' bndr annot e) = AnnExpr' bndr annot -> Expr bndr forall bndr annot. AnnExpr' bndr annot -> Expr bndr deAnnotate' AnnExpr' bndr annot e deAnnotate' :: AnnExpr' bndr annot -> Expr bndr deAnnotate' :: AnnExpr' bndr annot -> Expr bndr deAnnotate' (AnnType t :: Type t) = Type -> Expr bndr forall b. Type -> Expr b Type Type t deAnnotate' (AnnCoercion co :: Coercion co) = Coercion -> Expr bndr forall b. Coercion -> Expr b Coercion Coercion co deAnnotate' (AnnVar v :: CoreBndr v) = CoreBndr -> Expr bndr forall b. CoreBndr -> Expr b Var CoreBndr v deAnnotate' (AnnLit lit :: Literal lit) = Literal -> Expr bndr forall b. Literal -> Expr b Lit Literal lit deAnnotate' (AnnLam binder :: bndr binder body :: AnnExpr bndr annot body) = bndr -> Expr bndr -> Expr bndr forall b. b -> Expr b -> Expr b Lam bndr binder (AnnExpr bndr annot -> Expr bndr forall bndr annot. AnnExpr bndr annot -> Expr bndr deAnnotate AnnExpr bndr annot body) deAnnotate' (AnnApp fun :: AnnExpr bndr annot fun arg :: AnnExpr bndr annot arg) = Expr bndr -> Expr bndr -> Expr bndr forall b. Expr b -> Expr b -> Expr b App (AnnExpr bndr annot -> Expr bndr forall bndr annot. AnnExpr bndr annot -> Expr bndr deAnnotate AnnExpr bndr annot fun) (AnnExpr bndr annot -> Expr bndr forall bndr annot. AnnExpr bndr annot -> Expr bndr deAnnotate AnnExpr bndr annot arg) deAnnotate' (AnnCast e :: AnnExpr bndr annot e (_,co :: Coercion co)) = Expr bndr -> Coercion -> Expr bndr forall b. Expr b -> Coercion -> Expr b Cast (AnnExpr bndr annot -> Expr bndr forall bndr annot. AnnExpr bndr annot -> Expr bndr deAnnotate AnnExpr bndr annot e) Coercion co deAnnotate' (AnnTick tick :: Tickish CoreBndr tick body :: AnnExpr bndr annot body) = Tickish CoreBndr -> Expr bndr -> Expr bndr forall b. Tickish CoreBndr -> Expr b -> Expr b Tick Tickish CoreBndr tick (AnnExpr bndr annot -> Expr bndr forall bndr annot. AnnExpr bndr annot -> Expr bndr deAnnotate AnnExpr bndr annot body) deAnnotate' (AnnLet bind :: AnnBind bndr annot bind body :: AnnExpr bndr annot body) = Bind bndr -> Expr bndr -> Expr bndr forall b. Bind b -> Expr b -> Expr b Let (AnnBind bndr annot -> Bind bndr forall b annot. AnnBind b annot -> Bind b deAnnBind AnnBind bndr annot bind) (AnnExpr bndr annot -> Expr bndr forall bndr annot. AnnExpr bndr annot -> Expr bndr deAnnotate AnnExpr bndr annot body) deAnnotate' (AnnCase scrut :: AnnExpr bndr annot scrut v :: bndr v t :: Type t alts :: [AnnAlt bndr annot] alts) = Expr bndr -> bndr -> Type -> [Alt bndr] -> Expr bndr forall b. Expr b -> b -> Type -> [Alt b] -> Expr b Case (AnnExpr bndr annot -> Expr bndr forall bndr annot. AnnExpr bndr annot -> Expr bndr deAnnotate AnnExpr bndr annot scrut) bndr v Type t ((AnnAlt bndr annot -> Alt bndr) -> [AnnAlt bndr annot] -> [Alt bndr] forall a b. (a -> b) -> [a] -> [b] map AnnAlt bndr annot -> Alt bndr forall bndr annot. AnnAlt bndr annot -> Alt bndr deAnnAlt [AnnAlt bndr annot] alts) deAnnAlt :: AnnAlt bndr annot -> Alt bndr deAnnAlt :: AnnAlt bndr annot -> Alt bndr deAnnAlt (con :: AltCon con,args :: [bndr] args,rhs :: AnnExpr bndr annot rhs) = (AltCon con,[bndr] args,AnnExpr bndr annot -> Expr bndr forall bndr annot. AnnExpr bndr annot -> Expr bndr deAnnotate AnnExpr bndr annot rhs) deAnnBind :: AnnBind b annot -> Bind b deAnnBind :: AnnBind b annot -> Bind b deAnnBind (AnnNonRec var :: b var rhs :: AnnExpr b annot rhs) = b -> Expr b -> Bind b forall b. b -> Expr b -> Bind b NonRec b var (AnnExpr b annot -> Expr b forall bndr annot. AnnExpr bndr annot -> Expr bndr deAnnotate AnnExpr b annot rhs) deAnnBind (AnnRec pairs :: [(b, AnnExpr b annot)] pairs) = [(b, Expr b)] -> Bind b forall b. [(b, Expr b)] -> Bind b Rec [(b v,AnnExpr b annot -> Expr b forall bndr annot. AnnExpr bndr annot -> Expr bndr deAnnotate AnnExpr b annot rhs) | (v :: b v,rhs :: AnnExpr b annot rhs) <- [(b, AnnExpr b annot)] pairs] -- | As 'collectBinders' but for 'AnnExpr' rather than 'Expr' collectAnnBndrs :: AnnExpr bndr annot -> ([bndr], AnnExpr bndr annot) collectAnnBndrs :: AnnExpr bndr annot -> ([bndr], AnnExpr bndr annot) collectAnnBndrs e :: AnnExpr bndr annot e = [bndr] -> AnnExpr bndr annot -> ([bndr], AnnExpr bndr annot) forall a annot. [a] -> AnnExpr a annot -> ([a], AnnExpr a annot) collect [] AnnExpr bndr annot e where collect :: [a] -> AnnExpr a annot -> ([a], AnnExpr a annot) collect bs :: [a] bs (_, AnnLam b :: a b body :: AnnExpr a annot body) = [a] -> AnnExpr a annot -> ([a], AnnExpr a annot) collect (a ba -> [a] -> [a] forall a. a -> [a] -> [a] :[a] bs) AnnExpr a annot body collect bs :: [a] bs body :: AnnExpr a annot body = ([a] -> [a] forall a. [a] -> [a] reverse [a] bs, AnnExpr a annot body) -- | As 'collectNBinders' but for 'AnnExpr' rather than 'Expr' collectNAnnBndrs :: Int -> AnnExpr bndr annot -> ([bndr], AnnExpr bndr annot) collectNAnnBndrs :: Int -> AnnExpr bndr annot -> ([bndr], AnnExpr bndr annot) collectNAnnBndrs orig_n :: Int orig_n e :: AnnExpr bndr annot e = Int -> [bndr] -> AnnExpr bndr annot -> ([bndr], AnnExpr bndr annot) forall t a annot. (Eq t, Num t) => t -> [a] -> AnnExpr a annot -> ([a], AnnExpr a annot) collect Int orig_n [] AnnExpr bndr annot e where collect :: t -> [a] -> AnnExpr a annot -> ([a], AnnExpr a annot) collect 0 bs :: [a] bs body :: AnnExpr a annot body = ([a] -> [a] forall a. [a] -> [a] reverse [a] bs, AnnExpr a annot body) collect n :: t n bs :: [a] bs (_, AnnLam b :: a b body :: AnnExpr a annot body) = t -> [a] -> AnnExpr a annot -> ([a], AnnExpr a annot) collect (t nt -> t -> t forall a. Num a => a -> a -> a -1) (a ba -> [a] -> [a] forall a. a -> [a] -> [a] :[a] bs) AnnExpr a annot body collect _ _ _ = String -> SDoc -> ([a], AnnExpr a annot) forall a. HasCallStack => String -> SDoc -> a pprPanic "collectNBinders" (SDoc -> ([a], AnnExpr a annot)) -> SDoc -> ([a], AnnExpr a annot) forall a b. (a -> b) -> a -> b $ Int -> SDoc int Int orig_n