Mex: Update neighbors when reclaimed in nanoframe (#519)

beyond-all-reason · Nov 26, 2022 · f9679c3 · f9679c3
1 parent 801cd6d
commit f9679c3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/rts/Sim/Units/Unit.cpp b/rts/Sim/Units/Unit.cpp
@@ -8,6 +8,7 @@
 #include "UnitMemPool.h"
 #include "UnitToolTipMap.hpp"
 #include "UnitTypes/Building.h"
+#include "UnitTypes/ExtractorBuilding.h"
 #include "Scripts/NullUnitScript.h"
 #include "Scripts/UnitScriptFactory.h"
 #include "Scripts/CobInstance.h" // for TAANG2RAD
@@ -1981,6 +1982,12 @@ bool CUnit::AddBuildPower(CUnit* builder, float amount)
 			beingBuilt = true;
 			SetMetalStorage(0);
 			SetEnergyStorage(0);
+
+			// make sure neighbor extractors update
+			CExtractorBuilding* extractor = dynamic_cast<CExtractorBuilding*>(this);
+			if (extractor != nullptr)
+				extractor->ResetExtraction();
+
 			eventHandler.UnitReverseBuilt(this);
 		}