don't use cached iterators across map reloads

abstern · Nov 12, 2020 · 3611090 · 3611090
1 parent 923b1b1
commit 3611090
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/plugins/buildingplan.cpp b/plugins/buildingplan.cpp
@@ -457,6 +457,11 @@ struct buildingplan_query_hook : public df::viewscreen_dwarfmodest
         }
     }
 
+    static void invalidateStatics()
+    {
+        bld = NULL;
+    }
+
     bool handleInput(set<df::interface_key> *input)
     {
         if (!isInPlannedBuildingQueryMode() || Gui::inRenameBuilding())
@@ -583,6 +588,11 @@ struct buildingplan_place_hook : public df::viewscreen_dwarfmodest
         }
     }
 
+    static void invalidateStatics()
+    {
+        key = BuildingTypeKey();
+    }
+
     bool handleInput(set<df::interface_key> *input)
     {
         if (!isInPlannedBuildingPlacementMode())
@@ -959,6 +969,8 @@ DFhackCExport command_result plugin_onstatechange(color_ostream &out, state_chan
 {
     switch (event) {
     case SC_MAP_LOADED:
+        buildingplan_place_hook::invalidateStatics();
+        buildingplan_query_hook::invalidateStatics();
         planner.reset();
         roomMonitor.reset(out);
         break;