chore: Use builtin Name methods instead

main
Henrik Böving 2021-12-17 17:04:56 +01:00
parent 2df4891c9f
commit ef716c9351
1 changed files with 3 additions and 8 deletions

View File

@ -10,11 +10,6 @@ namespace DocGen4
open Lean Std Name open Lean Std Name
def getDepth : Name → Nat
| Name.anonymous => 0
| Name.str p _ _ => getDepth p + 1
| Name.num p _ _ => getDepth p + 1
def getNLevels (name : Name) (levels: Nat) : Name := def getNLevels (name : Name) (levels: Nat) : Name :=
(components.drop (components.length - levels)).reverse.foldl (· ++ ·) Name.anonymous (components.drop (components.length - levels)).reverse.foldl (· ++ ·) Name.anonymous
where where
@ -59,9 +54,9 @@ partial def insert! (h : Hierarchy) (n : Name) : Hierarchy := Id.run $ do
let hn := h.getName let hn := h.getName
let mut cs := h.getChildren let mut cs := h.getChildren
assert! getDepth hn ≤ getDepth n assert! getNumParts hn ≤ getNumParts n
if getDepth hn + 1 == getDepth n then if getNumParts hn + 1 == getNumParts n then
match cs.find Name.cmp n with match cs.find Name.cmp n with
| none => | none =>
node hn h.isFile (cs.insert Name.cmp n $ empty n true) node hn h.isFile (cs.insert Name.cmp n $ empty n true)
@ -70,7 +65,7 @@ partial def insert! (h : Hierarchy) (n : Name) : Hierarchy := Id.run $ do
cs := cs.erase Name.cmp n cs := cs.erase Name.cmp n
node hn h.isFile (cs.insert Name.cmp n $ node n true ccs) node hn h.isFile (cs.insert Name.cmp n $ node n true ccs)
else else
let leveledName := getNLevels n (getDepth hn + 1) let leveledName := getNLevels n (getNumParts hn + 1)
match cs.find Name.cmp leveledName with match cs.find Name.cmp leveledName with
| some nextLevel => | some nextLevel =>
cs := cs.erase Name.cmp leveledName cs := cs.erase Name.cmp leveledName