83 lines
2.5 KiB
Text
83 lines
2.5 KiB
Text
/-
|
||
Copyright (c) 2018 Microsoft Corporation. All rights reserved.
|
||
Released under Apache 2.0 license as described in the file LICENSE.
|
||
Author: Leonardo de Moura
|
||
-/
|
||
import Lean.Data.HashSet
|
||
import Lean.Data.RBMap
|
||
import Lean.Data.RBTree
|
||
import Lean.Data.SSet
|
||
import Lean.Data.Name
|
||
namespace Lean
|
||
|
||
instance : Coe String Name := ⟨Name.mkSimple⟩
|
||
|
||
def NameMap (α : Type) := RBMap Name α Name.quickCmp
|
||
|
||
@[inline] def mkNameMap (α : Type) : NameMap α := mkRBMap Name α Name.quickCmp
|
||
|
||
namespace NameMap
|
||
variable {α : Type}
|
||
|
||
instance (α : Type) : EmptyCollection (NameMap α) := ⟨mkNameMap α⟩
|
||
|
||
instance (α : Type) : Inhabited (NameMap α) where
|
||
default := {}
|
||
|
||
def insert (m : NameMap α) (n : Name) (a : α) := RBMap.insert m n a
|
||
|
||
def contains (m : NameMap α) (n : Name) : Bool := RBMap.contains m n
|
||
|
||
@[inline] def find? (m : NameMap α) (n : Name) : Option α := RBMap.find? m n
|
||
|
||
instance : ForIn m (NameMap α) (Name × α) :=
|
||
inferInstanceAs (ForIn _ (RBMap ..) ..)
|
||
|
||
end NameMap
|
||
|
||
def NameSet := RBTree Name Name.quickCmp
|
||
|
||
namespace NameSet
|
||
def empty : NameSet := mkRBTree Name Name.quickCmp
|
||
instance : EmptyCollection NameSet := ⟨empty⟩
|
||
instance : Inhabited NameSet := ⟨empty⟩
|
||
def insert (s : NameSet) (n : Name) : NameSet := RBTree.insert s n
|
||
def contains (s : NameSet) (n : Name) : Bool := RBMap.contains s n
|
||
instance : ForIn m NameSet Name :=
|
||
inferInstanceAs (ForIn _ (RBTree ..) ..)
|
||
|
||
end NameSet
|
||
|
||
def NameSSet := SSet Name
|
||
|
||
namespace NameSSet
|
||
abbrev empty : NameSSet := SSet.empty
|
||
instance : EmptyCollection NameSSet := ⟨empty⟩
|
||
instance : Inhabited NameSSet := ⟨empty⟩
|
||
abbrev insert (s : NameSSet) (n : Name) : NameSSet := SSet.insert s n
|
||
abbrev contains (s : NameSSet) (n : Name) : Bool := SSet.contains s n
|
||
end NameSSet
|
||
|
||
def NameHashSet := HashSet Name
|
||
|
||
namespace NameHashSet
|
||
@[inline] def empty : NameHashSet := HashSet.empty
|
||
instance : EmptyCollection NameHashSet := ⟨empty⟩
|
||
instance : Inhabited NameHashSet := ⟨{}⟩
|
||
def insert (s : NameHashSet) (n : Name) := HashSet.insert s n
|
||
def contains (s : NameHashSet) (n : Name) : Bool := HashSet.contains s n
|
||
end NameHashSet
|
||
|
||
def MacroScopesView.isPrefixOf (v₁ v₂ : MacroScopesView) : Bool :=
|
||
v₁.name.isPrefixOf v₂.name &&
|
||
v₁.scopes == v₂.scopes &&
|
||
v₁.mainModule == v₂.mainModule &&
|
||
v₁.imported == v₂.imported
|
||
|
||
def MacroScopesView.isSuffixOf (v₁ v₂ : MacroScopesView) : Bool :=
|
||
v₁.name.isSuffixOf v₂.name &&
|
||
v₁.scopes == v₂.scopes &&
|
||
v₁.mainModule == v₂.mainModule &&
|
||
v₁.imported == v₂.imported
|
||
|
||
end Lean
|