[Dehardcode]Customize overpower logic (#1604)

Now you can specific how building can be overpowerd.
Ares' Battery super weapon won't be affected by this.

In `rulesmd.ini`:
```ini
[SOMEWARHEAD]
ElectricAssaultLevel=1     ; integer

[SOMEBUILDING]
Overpower.KeepOnline=2    ; integer, negative values mean that cannot keep online
Overpower.ChargeWeapon=1  ; integer, negative values mean that weapons can never be switched
```

---------

Co-authored-by: 九千天华 <[email protected]>
Co-authored-by: Coronia <[email protected]>

Author: 3 people

CREDITS.md

@@ -365,6 +365,7 @@ This page lists all the individual contributions to the project by their author.
    - Taking over Ares' AlphaImage respawn logic to reduce lags from it
    - Allow voxel projectiles to use AnimPalette and FirersPalette
    - Customize damaged speed ratio of drive/ship loco
+   - Customize overpower logic
 - **Apollo** - Translucent SHP drawing patches
 - **ststl**:
    - Customizable `ShowTimer` priority of superweapons

docs/Fixed-or-Improved-Logics.md

@@ -530,6 +530,24 @@ In `rulesmd.ini`:
 BuildingWaypoints=false  ; boolean
 ```
 
+### Customize overpower logic
+
+- Now you can specific how building can be overpowerd.
+
+In `rulesmd.ini`:
+```ini
+[SOMEWARHEAD]
+ElectricAssaultLevel=1    ; integer
+
+[SOMEBUILDING]
+Overpower.KeepOnline=2    ; integer, negative values mean that cannot keep online
+Overpower.ChargeWeapon=1  ; integer, negative values mean that weapons can never be switched
+```
+
+```{note}
+Ares' [Battery Super Weapon](https://ares-developers.github.io/Ares-docs/new/superweapons/types/battery.html) won't be affected by this.
+```
+
 ## Infantry
 
 ### Prone speed customization

docs/Whats-New.md

@@ -353,6 +353,7 @@ New:
 - [AlternateFLH on-turret toggle](Fixed-or-Improved-Logics.md#alternate-flh-customizations) (by Starkku)
 - Prone speed customization (by TaranDahl)
 - Customize damaged speed ratio of drive/ship loco (by NetsuNegi)
+- Customize overpower logic (by NetsuNegi)
 
 Vanilla fixes:
 - Prevent the units with locomotors that cause problems from entering the tank bunker (by TaranDahl)

src/Ext/Building/Hooks.cpp

@@ -776,3 +776,80 @@ DEFINE_HOOK(0x4AE95E, DisplayClass_sub_4AE750_DisallowBuildingNonAttackPlanning,
 }
 
 #pragma endregion
+
+DEFINE_HOOK(0x4400F9, BuildingClass_AI_UpdateOverpower, 0x6)
+{
+	enum { SkipGameCode = 0x44019D };
+
+	GET(BuildingClass*, pThis, ESI);
+
+	if (!pThis->Type->Overpowerable)
+		return SkipGameCode;
+
+	int overPower = 0;
+
+	for (int idx = pThis->Overpowerers.Count - 1; idx >= 0; idx--)
+	{
+		const auto pCharger = pThis->Overpowerers[idx];
+
+		if (pCharger->Target != pThis)
+		{
+			pThis->Overpowerers.RemoveItem(idx);
+			continue;
+		}
+
+		const auto pWeapon = pCharger->GetWeapon(1)->WeaponType;
+
+		if (!pWeapon || !pWeapon->Warhead || !pWeapon->Warhead->ElectricAssault)
+		{
+			pThis->Overpowerers.RemoveItem(idx);
+			continue;
+		}
+
+		const auto pWHExt = WarheadTypeExt::ExtMap.Find(pWeapon->Warhead);
+		overPower += pWHExt->ElectricAssaultLevel;
+	}
+
+	const auto pBuildingTypeExt = BuildingTypeExt::ExtMap.Find(pThis->Type);
+	const int charge = pBuildingTypeExt->Overpower_ChargeWeapon;
+
+	if (charge >= 0)
+	{
+		const int keepOnline = pBuildingTypeExt->Overpower_KeepOnline;
+		pThis->IsOverpowered = overPower >= keepOnline + charge || (pThis->Owner->GetPowerPercentage() == 1.0 && pThis->HasPower && overPower >= charge);
+	}
+	else
+	{
+		pThis->IsOverpowered = false;
+	}
+
+	return SkipGameCode;
+}
+
+DEFINE_HOOK_AGAIN(0x45563B, BuildingClass_IsPowerOnline_Overpower, 0x6)
+DEFINE_HOOK(0x4555E4, BuildingClass_IsPowerOnline_Overpower, 0x6)
+{
+	enum { LowPower = 0x4556BE, Continue1 = 0x4555F0, Continue2 = 0x455643 };
+
+	GET(BuildingClass*, pThis, ESI);
+	const auto pBuildingTypeExt = BuildingTypeExt::ExtMap.Find(pThis->Type);
+	const int keepOnline = pBuildingTypeExt->Overpower_KeepOnline;
+
+	if (keepOnline < 0)
+		return LowPower;
+
+	int overPower = 0;
+
+	for (const auto pCharger : pThis->Overpowerers)
+	{
+		const auto pWeapon = pCharger->GetWeapon(1)->WeaponType;
+
+		if (pWeapon && pWeapon->Warhead)
+		{
+			const auto pWHExt = WarheadTypeExt::ExtMap.Find(pWeapon->Warhead);
+			overPower += pWHExt->ElectricAssaultLevel;
+		}
+	}
+
+	return overPower < keepOnline ? LowPower : (R->Origin() == 0x4555E4 ? Continue1 : Continue2);
+}

src/Ext/BuildingType/Body.cpp

@@ -177,6 +177,9 @@ void BuildingTypeExt::ExtData::LoadFromINIFile(CCINIClass* const pINI)
 
 	this->BarracksExitCell.Read(exINI, pSection, "BarracksExitCell");
 
+	this->Overpower_KeepOnline.Read(exINI, pSection, "Overpower.KeepOnline");
+	this->Overpower_ChargeWeapon.Read(exINI, pSection, "Overpower.ChargeWeapon");
+
 	if (pThis->NumberOfDocks > 0)
 	{
 		this->AircraftDockingDirs.clear();
@@ -302,6 +305,8 @@ void BuildingTypeExt::ExtData::Serialize(T& Stm)
 		.Process(this->Adjacent_Allowed)
 		.Process(this->Adjacent_Disallowed)
 		.Process(this->BarracksExitCell)
+		.Process(this->Overpower_KeepOnline)
+		.Process(this->Overpower_ChargeWeapon)
 		;
 }
 

src/Ext/BuildingType/Body.h

@@ -82,6 +82,9 @@ class BuildingTypeExt
 
 		Nullable<Point2D> BarracksExitCell;
 
+		Valueable<int> Overpower_KeepOnline;
+		Valueable<int> Overpower_ChargeWeapon;
+
 		ExtData(BuildingTypeClass* OwnerObject) : Extension<BuildingTypeClass>(OwnerObject)
 			, PowersUp_Owner { AffectedHouse::Owner }
 			, PowersUp_Buildings {}
@@ -132,6 +135,8 @@ class BuildingTypeExt
 			, Adjacent_Allowed {}
 			, Adjacent_Disallowed {}
 			, BarracksExitCell {}
+			, Overpower_KeepOnline { 2 }
+			, Overpower_ChargeWeapon { 1 }
 		{ }
 
 		// Ares 0.A functions

src/Ext/WarheadType/Body.cpp

@@ -282,6 +282,8 @@ void WarheadTypeExt::ExtData::LoadFromINIFile(CCINIClass* const pINI)
 	this->KillWeapon.Read(exINI, pSection, "KillWeapon");
 	this->KillWeapon_AffectsHouses.Read(exINI, pSection, "KillWeapon.AffectsHouses");
 
+  	this->ElectricAssaultLevel.Read(exINI, pSection, "ElectricAssaultLevel");
+
 	// Convert.From & Convert.To
 	TypeConvertGroup::Parse(this->Convert_Pairs, exINI, pSection, AffectedHouse::All);
 
@@ -524,6 +526,8 @@ void WarheadTypeExt::ExtData::Serialize(T& Stm)
 		.Process(this->KillWeapon)
 		.Process(this->KillWeapon_AffectsHouses)
 
+    	.Process(this->ElectricAssaultLevel)
+
 		// Ares tags
 		.Process(this->AffectsEnemies)
 		.Process(this->AffectsOwner)

src/Ext/WarheadType/Body.h

@@ -166,6 +166,8 @@ class WarheadTypeExt
 		Valueable<WeaponTypeClass*> KillWeapon;
 		Valueable<AffectedHouse> KillWeapon_AffectsHouses;
 
+    	Valueable<int> ElectricAssaultLevel;
+
 		// Ares tags
 		// http://ares-developers.github.io/Ares-docs/new/warheads/general.html
 		Valueable<bool> AffectsEnemies;
@@ -313,7 +315,7 @@ class WarheadTypeExt
 
 			, CombatLightDetailLevel {}
 			, CombatLightChance { 1.0 }
-		    , CLIsBlack { false }
+			, CLIsBlack { false }
 			, Particle_AlphaImageIsLightFlash {}
 
 			, DamageOwnerMultiplier {}
@@ -332,6 +334,8 @@ class WarheadTypeExt
 
 			, CombatAlert_Suppress {}
 
+			, ElectricAssaultLevel { 1 }
+
 			, AffectsEnemies { true }
 			, AffectsOwner {}
 			, EffectsRequireVerses { true }