reset subset filters when the context is exited

DFHack · Nov 7, 2023 · 4fb7694 · 4fb7694
1 parent 6fa65af
commit 4fb7694
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/plugins/lua/sort/info.lua b/plugins/lua/sort/info.lua
@@ -336,6 +336,11 @@ function InfoOverlay:init()
     end
 end
 
+function InfoOverlay:reset()
+    InfoOverlay.super.reset(self)
+    self.subviews.subset:setOption('all')
+end
+
 function InfoOverlay:get_key()
     if info.current_mode == df.info_interface_mode_type.CREATURES then
         if creatures.current_mode == df.unit_list_mode_type.PET then

diff --git a/plugins/lua/sort/unitselector.lua b/plugins/lua/sort/unitselector.lua
@@ -247,6 +247,11 @@ function BurrowAssignmentOverlay:register_handlers()
     end
 end
 
+function BurrowAssignmentOverlay:reset()
+    BurrowAssignmentOverlay.super.reset(self)
+    self.subviews.subset:setOption('all')
+end
+
 function BurrowAssignmentOverlay:matches_filters(unit_id)
     local unit = df.unit.find(unit_id)
     if not unit then return false end