r
/
bookshelf-doc
mirror of https://github.com/jrpotter/bookshelf-doc
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: use Kyle Miller's instance analysis algorithm

main
Henrik Böving 2023-11-05 09:38:22 +01:00
parent 8bccb92b53
commit ec2561f34b
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
DocGen4/Process/InstanceInfo.lean
View File

@ -16,14 +16,18 @@ open Lean Meta
def getInstanceTypes (typ : Expr) : MetaM (Array Name) := do def getInstanceTypes (typ : Expr) : MetaM (Array Name) := do
let (_, _, tail) ← forallMetaTelescopeReducing typ let (_, _, tail) ← forallMetaTelescopeReducing typ
let args := tail.getAppArgs let args := tail.getAppArgs
let (_, names) ← args.mapM (Expr.forEach · findName) |>.run .empty let (_, bis, _) ← forallMetaTelescopeReducing (← inferType tail.getAppFn)
let (_, names) ← (bis.zip args).mapM findName |>.run .empty
return names return names
where where
findName : Expr → StateRefT (Array Name) MetaM Unit findName : BinderInfo × Expr → StateRefT (Array Name) MetaM Unit
| .const name _ => modify (·.push name) | (.default, .sort .zero) => modify (·.push "_builtin_prop")
| .sort .zero => modify (·.push "_builtin_prop") | (.default, .sort (.succ _)) => modify (·.push "_builtin_typeu")
| .sort (.succ _) => modify (·.push "_builtin_typeu") | (.default, .sort _) => modify (·.push "_builtin_sortu")
| .sort _ => modify (·.push "_builtin_sortu") | (.default, e) =>
match e.getAppFn with
| .const name .. => modify (·.push name)
| _ => return ()
| _ => return () | _ => return ()
def InstanceInfo.ofDefinitionVal (v : DefinitionVal) : MetaM InstanceInfo := do def InstanceInfo.ofDefinitionVal (v : DefinitionVal) : MetaM InstanceInfo := do