Offline power plant indicator

CREDITS.md

@@ -287,6 +287,7 @@ This page lists all the individual contributions to the project by their author.
   - Vehicle voxel turret shadows & body multi-section shadows
 - **Fryone**
   - Customizable ElectricBolt Arcs
+  - Offline power plant indicator
 - **Ares developers**
   - YRpp and Syringe which are used, save/load, project foundation and generally useful code from Ares
   - unfinished RadTypes code

docs/Fixed-or-Improved-Logics.md

@@ -279,6 +279,22 @@ In `rulesmd.ini`:
 SellBuildupLength=23  ; integer, number of buildup frames to play
 ```
 
+### Offline power plant indicator
+
+- If `DrawPowerOffline` is set, `POWEROFF.SHP` indicator will appear on power plants when being offline, for example, drained by floating disk, infiltrated or disabled when using ForceShield.
+- `DrawPowerOffline.Offset` is used to vertically adjust position of indicator.
+- `DisablePowerOfflineIcon` is used to disable indicator per building type.
+
+In `rulesmd.ini`:
+```ini
+[AudioVisual]
+DrawPowerOffline=false          ; boolean
+DrawPowerOffline.Offset=0       ; integer, vertical offset,
+
+[SOMEBUILDING]
+DisablePowerOfflineIcon=false   ; boolean
+```
+
 ## Projectiles
 
 ### Cluster scatter distance customization

docs/Whats-New.md

@@ -329,6 +329,7 @@ New:
 - Unhardcoded timer blinking color scheme (by Starkku)
 - Customizing shield self-healing timer restart when shield is damaged (by Starkku)
 - Customizing minimum & maximum amount of damage shield can take from a single hit (by Starkku)
+- Offline power plant indicator (by Fryone)
 
 Vanilla fixes:
 - Allow AI to repair structures built from base nodes/trigger action 125/SW delivery in single player missions (by Trsdy)

src/Ext/Building/Hooks.cpp

@@ -430,3 +430,48 @@ DEFINE_HOOK(0x4511D6, BuildingClass_AnimationAI_SellBuildup, 0x7)
 
 	return pTypeExt->SellBuildupLength == pThis->Animation.Value ? Continue : Skip;
 }
+
+//Addition to Ares' Extras
+DEFINE_HOOK(0x6F5347, TechnoClass_DrawExtras_OfflinePlants, 0x7)
+{
+	GET(TechnoClass*, pThis, EBP);
+	GET_STACK(RectangleStruct*, pRect, 0xA0);
+
+	if(auto pBld = abstract_cast<BuildingClass*>(pThis))
+	{
+		if (!RulesExt::Global()->DrawPowerOffline)
+		{
+			R->ESI(pRect);
+			return 0x6F534E;
+		}
+
+		const auto pBldExt = BuildingTypeExt::ExtMap.Find(pBld->Type);
+		bool showLowPower = (pBldExt->DisablePowerOfflineIcon == false)
+			&& FileSystem::POWEROFF_SHP
+			&& (pBld->Type->PowerBonus > 0)
+			&& (pBld->Factory == nullptr)
+			&& (pBld->IsPowerOnline() == false || pBld->IsBeingDrained())
+			&& (pBld->IsBeingWarpedOut() == false)
+			&& (pBld->WarpingOut == false)
+			&& ((pBld->GetCurrentMission() != Mission::Selling) && (pBld->GetCurrentMission() != Mission::Construction))
+			&& (pBld->CloakState == CloakState::Uncloaked);
+
+		if (!showLowPower || MapClass::Instance->GetCellAt(pBld->GetMapCoords())->IsShrouded())
+		{
+			R->ESI(pRect);
+			return 0x6F534E;
+		}
+
+		Point2D nPoint;
+		TacticalClass::Instance->CoordsToClient(pBld->GetRenderCoords(), &nPoint);
+
+		nPoint.Y += 22; // wrench offset
+		nPoint.Y -= RulesExt::Global()->DrawPowerOffline_Offset;
+
+		const int speed = Math::max(GameOptionsClass::Instance->GetAnimSpeed(14) / 4, 2);
+		const int frame = (FileSystem::POWEROFF_SHP->Frames * (Unsorted::CurrentFrame % speed)) / speed;
+		DSurface::Temp->DrawSHP(FileSystem::MOUSE_PAL, FileSystem::POWEROFF_SHP, frame, &nPoint, pRect, BlitterFlags(0xE00), 0, 0, ZGradient::Ground, 1000, 0, 0, 0, 0, 0);
+	}
+	R->ESI(pRect);
+	return 0x6F534E;
+}

src/Ext/BuildingType/Body.cpp

@@ -146,6 +146,7 @@ void BuildingTypeExt::ExtData::LoadFromINIFile(CCINIClass* const pINI)
 
 	this->ConsideredVehicle.Read(exINI, pSection, "ConsideredVehicle");
 	this->SellBuildupLength.Read(exINI, pSection, "SellBuildupLength");
+	this->DisablePowerOfflineIcon.Read(exINI, pSection, "DisablePowerOfflineIcon");
 
 	// Ares tag
 	this->SpyEffect_Custom.Read(exINI, pSection, "SpyEffect.Custom");
@@ -234,6 +235,7 @@ void BuildingTypeExt::ExtData::Serialize(T& Stm)
 		.Process(this->ConsideredVehicle)
 		.Process(this->ZShapePointMove_OnBuildup)
 		.Process(this->SellBuildupLength)
+		.Process(this->DisablePowerOfflineIcon)
 		;
 }
 

src/Ext/BuildingType/Body.h

@@ -60,6 +60,7 @@ class BuildingTypeExt
 		Nullable<bool> ConsideredVehicle;
 		Valueable<bool> ZShapePointMove_OnBuildup;
 		Valueable<int> SellBuildupLength;
+		Valueable<bool> DisablePowerOfflineIcon;
 
 		ExtData(BuildingTypeClass* OwnerObject) : Extension<BuildingTypeClass>(OwnerObject)
 			, PowersUp_Owner { AffectedHouse::Owner }
@@ -96,6 +97,7 @@ class BuildingTypeExt
 			, ConsideredVehicle {}
 			, ZShapePointMove_OnBuildup { false }
 			, SellBuildupLength { 23 }
+			, DisablePowerOfflineIcon { false }
 		{ }
 
 		// Ares 0.A functions

src/Ext/Rules/Body.cpp

@@ -106,6 +106,9 @@ void RulesExt::ExtData::LoadBeforeTypeData(RulesClass* pThis, CCINIClass* pINI)
 	this->AnimRemapDefaultColorScheme.Read(exINI, GameStrings::AudioVisual, "AnimRemapDefaultColorScheme");
 	this->TimerBlinkColorScheme.Read(exINI, GameStrings::AudioVisual, "TimerBlinkColorScheme");
 
+	this->DrawPowerOffline.Read(exINI, GameStrings::AudioVisual, "DrawPowerOffline");
+	this->DrawPowerOffline_Offset.Read(exINI, GameStrings::AudioVisual, "DrawPowerOffline.Offset");
+
 	this->AllowParallelAIQueues.Read(exINI, "GlobalControls", "AllowParallelAIQueues");
 	this->ForbidParallelAIQueues_Aircraft.Read(exINI, "GlobalControls", "ForbidParallelAIQueues.Infantry");
 	this->ForbidParallelAIQueues_Building.Read(exINI, "GlobalControls", "ForbidParallelAIQueues.Building");
@@ -249,6 +252,8 @@ void RulesExt::ExtData::Serialize(T& Stm)
 		.Process(this->CrateOnlyOnLand)
 		.Process(this->RadialIndicatorVisibility)
 		.Process(this->DrawTurretShadow)
+		.Process(this->DrawPowerOffline)
+		.Process(this->DrawPowerOffline_Offset)
 		.Process(this->AnimRemapDefaultColorScheme)
 		.Process(this->TimerBlinkColorScheme)
 		.Process(this->Buildings_DefaultDigitalDisplayTypes)

src/Ext/Rules/Body.h

@@ -91,6 +91,9 @@ class RulesExt
 		ValueableIdx<ColorScheme> AnimRemapDefaultColorScheme;
 		ValueableIdx<ColorScheme> TimerBlinkColorScheme;
 
+		Valueable<bool> DrawPowerOffline;
+		Valueable<int> DrawPowerOffline_Offset;
+
 		ValueableVector<DigitalDisplayTypeClass*> Buildings_DefaultDigitalDisplayTypes;
 		ValueableVector<DigitalDisplayTypeClass*> Infantry_DefaultDigitalDisplayTypes;
 		ValueableVector<DigitalDisplayTypeClass*> Vehicles_DefaultDigitalDisplayTypes;
@@ -147,6 +150,8 @@ class RulesExt
 			, CrateOnlyOnLand { false }
 			, RadialIndicatorVisibility { AffectedHouse::Allies }
 			, DrawTurretShadow { false }
+			, DrawPowerOffline { false }
+			, DrawPowerOffline_Offset { 0 }
 			, AnimRemapDefaultColorScheme { 0 }
 			, TimerBlinkColorScheme { 5 }
 			, Buildings_DefaultDigitalDisplayTypes {}

src/Ext/Techno/Hooks.cpp

@@ -1,4 +1,6 @@
-#include <ScenarioClass.h>
+
+#include <GameOptionsClass.h>
+#include <TacticalClass.h>
 #include "Body.h"
 
 #include <Ext/WarheadType/Body.h>