Lazy lists

The type LazyList α is a lazy list with elements of type α. In the VM, these are potentially infinite lists where all elements after the first are computed on-demand. (This is only useful for execution in the VM, logically we can prove that LazyList α is isomorphic to List α.)

inductive LazyList (α : Type u) : Type u

Lazy list. All elements (except the first) are computed lazily.

• nil: {α : Type u} → LazyList α

  The empty lazy list.

• cons: {α : Type u} → α → Thunk (LazyList α) → LazyList α

  Construct a lazy list from an element and a tail inside a thunk.

instance LazyList.instInhabitedLazyList {α : Type u_1} : Inhabited (LazyList α)

Equations

• LazyList.instInhabitedLazyList = { default := LazyList.nil }

def LazyList.singleton {α : Type u_1} : α → LazyList α

The singleton lazy list.

Equations

• LazyList.singleton x = let a := x; LazyList.cons a (Thunk.pure LazyList.nil)

def LazyList.ofList {α : Type u_1} : List α → LazyList α

Constructs a lazy list from a list.

Equations

• LazyList.ofList [] = LazyList.nil
• LazyList.ofList (h :: t) = LazyList.cons h { fn := fun (x : Unit) => LazyList.ofList t }

def LazyList.toList {α : Type u_1} : LazyList α → List α

Converts a lazy list to a list. If the lazy list is infinite, then this function does not terminate.

Equations

• LazyList.toList LazyList.nil = []
• LazyList.toList (LazyList.cons h t) = h :: LazyList.toList (Thunk.get t)

def LazyList.headI {α : Type u_1} [Inhabited α] : LazyList α → α

Returns the first element of the lazy list, or default if the lazy list is empty.

Equations

• LazyList.headI x = match x with | LazyList.nil => default | LazyList.cons h tl => h

def LazyList.tail {α : Type u_1} : LazyList α → LazyList α

Removes the first element of the lazy list.

Equations

• LazyList.tail x = match x with | LazyList.nil => LazyList.nil | LazyList.cons hd t => Thunk.get t

def LazyList.append {α : Type u_1} : LazyList α → Thunk (LazyList α) → LazyList α

Appends two lazy lists.

Equations

• LazyList.append LazyList.nil x = Thunk.get x
• LazyList.append (LazyList.cons h t) x = LazyList.cons h { fn := fun (x_1 : Unit) => LazyList.append (Thunk.get t) x }

def LazyList.map {α : Type u_1} {β : Type u_2} (f : α → β) : LazyList α → LazyList β

Maps a function over a lazy list.

Equations

• LazyList.map f LazyList.nil = LazyList.nil
• LazyList.map f (LazyList.cons h t) = LazyList.cons (f h) { fn := fun (x : Unit) => LazyList.map f (Thunk.get t) }

def LazyList.map₂ {α : Type u_1} {β : Type u_2} {δ : Type u_3} (f : α → β → δ) : LazyList α → LazyList β → LazyList δ

Maps a binary function over two lazy list. Like LazyList.zip, the result is only as long as the smaller input.

Equations

• LazyList.map₂ f LazyList.nil x = LazyList.nil
• LazyList.map₂ f x LazyList.nil = LazyList.nil
• LazyList.map₂ f (LazyList.cons h₁ t₁) (LazyList.cons h₂ t₂) = LazyList.cons (f h₁ h₂) { fn := fun (x : Unit) => LazyList.map₂ f (Thunk.get t₁) (Thunk.get t₂) }

def LazyList.zip {α : Type u_1} {β : Type u_2} : LazyList α → LazyList β → LazyList (α × β)

Zips two lazy lists.

Equations

• LazyList.zip = LazyList.map₂ Prod.mk

def LazyList.join {α : Type u_1} : LazyList (LazyList α) → LazyList α

The monadic join operation for lazy lists.

Equations

• LazyList.join LazyList.nil = LazyList.nil
• LazyList.join (LazyList.cons h t) = LazyList.append h { fn := fun (x : Unit) => LazyList.join (Thunk.get t) }

def LazyList.take {α : Type u_1} : Nat → LazyList α → List α

The list containing the first n elements of a lazy list.

Equations

• LazyList.take 0 x = []
• LazyList.take x LazyList.nil = []
• LazyList.take (Nat.succ a) (LazyList.cons h t) = h :: LazyList.take a (Thunk.get t)

def LazyList.filter {α : Type u_1} (p : α → Prop) [DecidablePred p] : LazyList α → LazyList α

The lazy list of all elements satisfying the predicate. If the lazy list is infinite and none of the elements satisfy the predicate, then this function will not terminate.

Equations

• LazyList.filter p LazyList.nil = LazyList.nil
• LazyList.filter p (LazyList.cons h t) = if p h then LazyList.cons h { fn := fun (x : Unit) => LazyList.filter p (Thunk.get t) } else LazyList.filter p (Thunk.get t)

def LazyList.get? {α : Type u_1} : LazyList α → Nat → Option α

The nth element of a lazy list as an option (like List.get?).

Equations

• LazyList.get? LazyList.nil x = none
• LazyList.get? (LazyList.cons a tl) 0 = some a
• LazyList.get? (LazyList.cons hd l) (Nat.succ n) = LazyList.get? (Thunk.get l) n

partial def LazyList.iterates {α : Type u_1} (f : α → α) : α → LazyList α

The infinite lazy list [x, f x, f (f x), ...] of iterates of a function. This definition is partial because it creates an infinite list.

def LazyList.iota (i : Nat) : LazyList Nat

The infinite lazy list [i, i+1, i+2, ...]

Equations

• LazyList.iota i = LazyList.iterates Nat.succ i