max/lean4-htt
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

feat(library/init/lean/environment): add persistent := true parameter to addEntry

Browse source
This commit is contained in:
Leonardo de Moura 2019-06-03 17:03:46 -07:00
parent 0a08569b46
commit ca12439e25
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
library/init/lean/environment.lean
View file

@ -185,7 +185,7 @@ instance PersistentEnvExtensionState.inhabited {α σ} [Inhabited α] [Inhabited
instance PersistentEnvExtension.inhabited {α σ} [Inhabited α] [Inhabited σ] : Inhabited (PersistentEnvExtension α σ) :=
⟨{ toEnvExtension := { idx := 0, initial := default _ },
name := default _,
name := default _,
addEntryFn := λ _ s _, s,
toArrayFn := λ es, es.toArray }⟩
@ -197,10 +197,11 @@ def getEntries {α σ : Type} (ext : PersistentEnvExtension α σ) (env : Enviro
def getModuleEntries {α σ : Type} (ext : PersistentEnvExtension α σ) (env : Environment) (m : ModuleIdx) : Array α :=
(ext.toEnvExtension.getState env).importedEntries.get m
def addEntry {α σ : Type} (ext : PersistentEnvExtension α σ) (env : Environment) (a : α) : Environment :=
def addEntry {α σ : Type} (ext : PersistentEnvExtension α σ) (env : Environment) (a : α) (persistent : Bool := true) : Environment :=
ext.toEnvExtension.modifyState env $ λ s,
let entries := a :: s.entries in
{ entries := entries, state := ext.addEntryFn false s.state a, .. s }
let entries := if persistent then a :: s.entries else s.entries in
let state := ext.addEntryFn false s.state a in
{ entries := entries, state := state, .. s }
def getState {α σ : Type} (ext : PersistentEnvExtension α σ) (env : Environment) : σ :=
(ext.toEnvExtension.getState env).state