[Minor] Try optimize TechnoClass_PointerGotInvalid_Airstrike (#1675)

Author: CrimRecya

src/Ext/Techno/Body.cpp

@@ -624,6 +624,11 @@ void TechnoExt::ExtData::Serialize(T& Stm)
 		;
 }
 
+void TechnoExt::ExtData::InvalidatePointer(void* ptr, bool bRemoved)
+{
+	AnnounceInvalidPointer(this->AirstrikeTargetingMe, ptr);
+}
+
 void TechnoExt::ExtData::LoadFromStream(PhobosStreamReader& Stm)
 {
 	Extension<TechnoClass>::LoadFromStream(Stm);

src/Ext/Techno/Body.h

@@ -157,7 +157,7 @@ class TechnoExt
 		UnitTypeClass* GetUnitTypeExtra() const;
 
 		virtual ~ExtData() override;
-		virtual void InvalidatePointer(void* ptr, bool bRemoved) override { }
+		virtual void InvalidatePointer(void* ptr, bool bRemoved) override;
 		virtual void LoadFromStream(PhobosStreamReader& Stm) override;
 		virtual void SaveToStream(PhobosStreamWriter& Stm) override;
 
@@ -171,6 +171,19 @@ class TechnoExt
 	public:
 		ExtContainer();
 		~ExtContainer();
+
+		virtual bool InvalidateExtDataIgnorable(void* const ptr) const override
+		{
+			auto const abs = static_cast<AbstractClass*>(ptr)->WhatAmI();
+
+			switch (abs)
+			{
+			case AbstractType::Airstrike:
+				return false;
+			default:
+				return true;
+			}
+		}
 	};
 
 	static ExtContainer ExtMap;

src/Ext/Techno/Hooks.Airstrike.cpp

@@ -134,7 +134,7 @@ DEFINE_HOOK(0x41D604, AirstrikeClass_PointerGotInvalid_ResetForTarget, 0x6)
 
 	GET(ObjectClass*, pTarget, EAX);
 
-	if (const auto pTargetTechnoExt = TechnoExt::ExtMap.Find(abstract_cast<TechnoClass*>(pTarget)))
+	if (const auto pTargetTechnoExt = TechnoExt::ExtMap.Find(abstract_cast<TechnoClass*, true>(pTarget)))
 		pTargetTechnoExt->AirstrikeTargetingMe = nullptr;
 
 	return SkipGameCode;
@@ -187,17 +187,6 @@ DEFINE_HOOK(0x51EAE0, TechnoClass_WhatAction_AllowAirstrike, 0x7)
 	return Cannot;
 }
 
-DEFINE_HOOK(0x70782D, TechnoClass_PointerGotInvalid_Airstrike, 0x6)
-{
-	GET(AbstractClass*, pAbstract, EBP);
-	GET(TechnoClass*, pThis, ESI);
-
-	if (const auto pExt = TechnoExt::ExtMap.Find(pThis)) // It's necessary
-		AnnounceInvalidPointer(pExt->AirstrikeTargetingMe, pAbstract);
-
-	return 0;
-}
-
 #pragma region GetEffectTintIntensity
 
 DEFINE_HOOK(0x70E92F, TechnoClass_UpdateAirstrikeTint, 0x5)