@[inline]

abbrev Lean.IR.UniqueIds.M (α : Type) : Type

Equations

• Lean.IR.UniqueIds.M = StateT Lean.IR.IndexSet Id

def Lean.IR.UniqueIds.checkId (id : Lean.IR.Index) : Lean.IR.UniqueIds.M Bool

Equations

• Lean.IR.UniqueIds.checkId id = modifyGet fun s => if Lean.RBTree.contains s id = true then (false, s) else (true, Lean.RBTree.insert s id)

def Lean.IR.UniqueIds.checkParams (ps : Array Lean.IR.Param) : Lean.IR.UniqueIds.M Bool

Equations

• Lean.IR.UniqueIds.checkParams ps = Array.allM (fun p => Lean.IR.UniqueIds.checkId p.x.idx) ps 0 (Array.size ps)

partial def Lean.IR.UniqueIds.checkFnBody : Lean.IR.FnBody → Lean.IR.UniqueIds.M Bool

def Lean.IR.UniqueIds.checkDecl : Lean.IR.Decl → Lean.IR.UniqueIds.M Bool

Equations

• One or more equations did not get rendered due to their size.

def Lean.IR.Decl.uniqueIds (d : Lean.IR.Decl) : Bool

Return true if variable, parameter and join point ids are unique

Equations

• Lean.IR.Decl.uniqueIds d = StateT.run' (Lean.IR.UniqueIds.checkDecl d) ∅

@[inline]

abbrev Lean.IR.NormalizeIds.M (α : Type) : Type

Equations

• Lean.IR.NormalizeIds.M = ReaderT Lean.IR.IndexRenaming Id

def Lean.IR.NormalizeIds.normIndex (x : Lean.IR.Index) : Lean.IR.NormalizeIds.M Lean.IR.Index

Equations

• Lean.IR.NormalizeIds.normIndex x m = match Lean.RBMap.find? m x with | some y => y | none => x

def Lean.IR.NormalizeIds.normVar (x : Lean.IR.VarId) : Lean.IR.NormalizeIds.M Lean.IR.VarId

Equations

• Lean.IR.NormalizeIds.normVar x = Lean.IR.VarId.mk <$> Lean.IR.NormalizeIds.normIndex x.idx

def Lean.IR.NormalizeIds.normJP (x : Lean.IR.JoinPointId) : Lean.IR.NormalizeIds.M Lean.IR.JoinPointId

Equations

• Lean.IR.NormalizeIds.normJP x = Lean.IR.JoinPointId.mk <$> Lean.IR.NormalizeIds.normIndex x.idx

def Lean.IR.NormalizeIds.normArg : Lean.IR.Arg → Lean.IR.NormalizeIds.M Lean.IR.Arg

Equations

• Lean.IR.NormalizeIds.normArg x = match x with | Lean.IR.Arg.var x => Lean.IR.Arg.var <$> Lean.IR.NormalizeIds.normVar x | other => pure other

def Lean.IR.NormalizeIds.normArgs (as : Array Lean.IR.Arg) : Lean.IR.NormalizeIds.M (Array Lean.IR.Arg)

Equations

• Lean.IR.NormalizeIds.normArgs as m = Array.map (fun a => Lean.IR.NormalizeIds.normArg a m) as

def Lean.IR.NormalizeIds.normExpr : Lean.IR.Expr → Lean.IR.NormalizeIds.M Lean.IR.Expr

Equations

• One or more equations did not get rendered due to their size.

@[inline]

abbrev Lean.IR.NormalizeIds.N (α : Type) : Type

Equations

• Lean.IR.NormalizeIds.N = ReaderT Lean.IR.IndexRenaming (StateM Nat)

@[inline]

def Lean.IR.NormalizeIds.withVar {α : Type} (x : Lean.IR.VarId) (k : Lean.IR.VarId → Lean.IR.NormalizeIds.N α) : Lean.IR.NormalizeIds.N α

Equations

• Lean.IR.NormalizeIds.withVar x k m = do let n ← getModify fun n => n + 1 k { idx := n } (Lean.RBMap.insert m x.idx n)

@[inline]

def Lean.IR.NormalizeIds.withJP {α : Type} (x : Lean.IR.JoinPointId) (k : Lean.IR.JoinPointId → Lean.IR.NormalizeIds.N α) : Lean.IR.NormalizeIds.N α

Equations

• Lean.IR.NormalizeIds.withJP x k m = do let n ← getModify fun n => n + 1 k { idx := n } (Lean.RBMap.insert m x.idx n)

@[inline]

def Lean.IR.NormalizeIds.withParams {α : Type} (ps : Array Lean.IR.Param) (k : Array Lean.IR.Param → Lean.IR.NormalizeIds.N α) : Lean.IR.NormalizeIds.N α

Equations

• One or more equations did not get rendered due to their size.

instance Lean.IR.NormalizeIds.instMonadLiftMN : MonadLift Lean.IR.NormalizeIds.M Lean.IR.NormalizeIds.N

Equations

• Lean.IR.NormalizeIds.instMonadLiftMN = { monadLift := fun {α} x m => pure (x m) }

partial def Lean.IR.NormalizeIds.normFnBody : Lean.IR.FnBody → Lean.IR.NormalizeIds.N Lean.IR.FnBody

def Lean.IR.NormalizeIds.normDecl (d : Lean.IR.Decl) : Lean.IR.NormalizeIds.N Lean.IR.Decl

Equations

• One or more equations did not get rendered due to their size.

def Lean.IR.Decl.normalizeIds (d : Lean.IR.Decl) : Lean.IR.Decl

Create a declaration equivalent to d s.t. d.normalizeIds.uniqueIds == true

Equations

• Lean.IR.Decl.normalizeIds d = StateT.run' (Lean.IR.NormalizeIds.normDecl d ∅) 1

Apply a function f : VarId → VarId to variable occurrences. The following functions assume the IR code does not have variable shadowing.

@[inline]

def Lean.IR.MapVars.mapArg (f : Lean.IR.VarId → Lean.IR.VarId) : Lean.IR.Arg → Lean.IR.Arg

Equations

• Lean.IR.MapVars.mapArg f x = match x with | Lean.IR.Arg.var x => Lean.IR.Arg.var (f x) | a => a

def Lean.IR.MapVars.mapArgs (f : Lean.IR.VarId → Lean.IR.VarId) (as : Array Lean.IR.Arg) : Array Lean.IR.Arg

Equations

• Lean.IR.MapVars.mapArgs f as = Array.map (Lean.IR.MapVars.mapArg f) as

def Lean.IR.MapVars.mapExpr (f : Lean.IR.VarId → Lean.IR.VarId) : Lean.IR.Expr → Lean.IR.Expr

Equations

• One or more equations did not get rendered due to their size.

partial def Lean.IR.MapVars.mapFnBody (f : Lean.IR.VarId → Lean.IR.VarId) : Lean.IR.FnBody → Lean.IR.FnBody

@[inline]

def Lean.IR.FnBody.mapVars (f : Lean.IR.VarId → Lean.IR.VarId) (b : Lean.IR.FnBody) : Lean.IR.FnBody

Equations

• Lean.IR.FnBody.mapVars f b = Lean.IR.MapVars.mapFnBody f b

def Lean.IR.FnBody.replaceVar (x : Lean.IR.VarId) (y : Lean.IR.VarId) (b : Lean.IR.FnBody) : Lean.IR.FnBody

Replace x with y in b. This function assumes b does not shadow x

Equations

• Lean.IR.FnBody.replaceVar x y b = Lean.IR.FnBody.mapVars (fun z => if (x == z) = true then y else z) b