aniva/Pantograph
aniva
/
Pantograph
1
1
Fork 0
Code Issues 15 Pull Requests 3 Packages Projects Releases Wiki Activity

fix(tactic): Filter out sibling fragments

This commit is contained in:
Leni Aniva 2025-06-26 11:48:28 -07:00
parent 425825bf20
commit 7fba24d57a
Signed by: aniva
GPG Key ID: D5F96287843E8DFB
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
Pantograph/Tactic/Fragment.lean
View File

@ -25,8 +25,15 @@ inductive Fragment where
-- This goal is spawned from a `conv` -- This goal is spawned from a `conv`
| convSentinel (parent : MVarId) | convSentinel (parent : MVarId)
deriving BEq, Inhabited deriving BEq, Inhabited
abbrev FragmentMap := Std.HashMap MVarId Fragment abbrev FragmentMap := Std.HashMap MVarId Fragment
def FragmentMap.empty : FragmentMap := Std.HashMap.emptyWithCapacity 2 def FragmentMap.empty : FragmentMap := Std.HashMap.emptyWithCapacity 2
protected def FragmentMap.filter (map : FragmentMap) (pred : MVarId → Fragment → Bool) : FragmentMap :=
map.fold (init := FragmentMap.empty) λ acc mvarId fragment =>
if pred mvarId fragment then
acc.insert mvarId fragment
else
acc
protected def Fragment.map (fragment : Fragment) (mapExpr : Expr → CoreM Expr) : CoreM Fragment := protected def Fragment.map (fragment : Fragment) (mapExpr : Expr → CoreM Expr) : CoreM Fragment :=
let mapMVar (mvarId : MVarId) : CoreM MVarId := let mapMVar (mvarId : MVarId) : CoreM MVarId :=
@ -80,8 +87,8 @@ protected partial def Fragment.exit (fragment : Fragment) (goal : MVarId) (fragm
let mvarId ← Elab.Tactic.getMainGoal let mvarId ← Elab.Tactic.getMainGoal
liftM <| mvarId.refl <|> mvarId.inferInstance <|> pure () liftM <| mvarId.refl <|> mvarId.inferInstance <|> pure ()
Elab.Tactic.pruneSolvedGoals Elab.Tactic.pruneSolvedGoals
-- FIXME: Erase all sibling fragments return fragments.filter λ mvarId fragment =>
return fragments.erase goal !(mvarId == goal || fragment == .convSentinel goal)
| .convSentinel parent => | .convSentinel parent =>
let parentFragment := fragments[parent]! let parentFragment := fragments[parent]!
parentFragment.exit parent (fragments.erase goal) parentFragment.exit parent (fragments.erase goal)