Race: Suppress and Reactive for temp disabling nwnx_race effects

nwnxee · Oct 5, 2024 · 6f68afd · 6f68afd
1 parent 695efc6
commit 6f68afd
Show file tree

Hide file tree

Showing 4 changed files with 68 additions and 11 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -28,6 +28,8 @@ https://github.com/nwnxee/unified/compare/build8193.36.12...HEAD
 - Store: {Get|Set}MarkUp()
 - Player: ReloadTlk()
 - Player: ReloadColorPalettes()
+- Race: SuppressCreatureRaceEffects()
+- Race: ReactivateCreatureRaceEffects()
 
 ### Changed
 - Player: added bChatWindow parameter to FloatingTextStringOnCreature()

diff --git a/Plugins/Race/NWScript/nwnx_race.nss b/Plugins/Race/NWScript/nwnx_race.nss
@@ -54,6 +54,16 @@ int NWNX_Race_GetParentRace(int iRace);
 /// or Favored Enemy: Wild Elf
 void NWNX_Race_SetFavoredEnemyFeat(int iRace, int iFeat);
 
+/// @brief Removes any nwnx_race 'Effects' on the targeted creature. Suppression lasts until levelup, next login, or Reactivated by function.
+/// @param oCreature The creature to suppress
+/// @note Not all nwnx_race modifiers are achieved via effect. Those that are not directly consider the creatures current race.
+void NWNX_Race_SuppressCreatureRaceEffects(object oCreature);
+
+/// @brief Reactivates the nwnx_race 'Effects' on the targeted creature after they were Suppressed.
+/// @param oCreature The creature to reactive
+/// @note Safe to use on non-suppressed creatures - Triggers a refresh of effects but won't stack.
+void NWNX_Race_ReactivateCreatureRaceEffects(object oCreature);
+
 /// @}
 
 void NWNX_Race_SetRacialModifier(int iRace, int iMod, int iParam1, int iParam2 = 0xDEADBEEF, int iParam3 = 0xDEADBEEF)
@@ -88,3 +98,19 @@ void NWNX_Race_SetFavoredEnemyFeat(int iRace, int iFeat)
 
     NWNX_CallFunction(NWNX_Race, sFunc);
 }
+
+void NWNX_Race_SuppressCreatureRaceEffects(object creature)
+{
+    string sFunc = "SuppressCreatureRaceEffects";
+
+    NWNX_PushArgumentObject(creature);
+    NWNX_CallFunction(NWNX_Race, sFunc);
+}
+
+void NWNX_Race_ReactivateCreatureRaceEffects(object oCreature)
+{
+    string sFunc = "ReactivateCreatureRaceEffects";
+
+    NWNX_PushArgumentObject(oCreature);
+    NWNX_CallFunction(NWNX_Race, sFunc);
+}
diff --git a/Plugins/Race/Race.cpp b/Plugins/Race/Race.cpp
@@ -178,6 +178,24 @@ void Race::DoEffect(CNWSCreature *pCreature,
     pCreature->ApplyEffect(eff, true, true);
 }
 
+void Race::RemoveRaceEffects(CNWSCreature *pCreature)
+{
+    auto effectsLevelAdded = *pCreature->nwnxGet<int>("RACEMODS_ADDED_LEVEL");
+    if (pCreature->m_pStats == nullptr || effectsLevelAdded == 0)
+        return;
+
+    for (int i = pCreature->m_appliedEffects.num; i >= 0; --i)
+    {
+        auto eff = (CGameEffect*)pCreature->m_appliedEffects.element[i];
+        if (eff->m_sCustomTag == "NWNX_Race_RacialMod")
+        {
+            pCreature->RemoveEffectById(eff->m_nID);
+        }
+    }
+
+    pCreature->nwnxRemove("RACEMODS_ADDED_LEVEL");
+}
+
 void Race::ApplyRaceEffects(CNWSCreature *pCreature)
 {
     auto effectsLevelAdded = *pCreature->nwnxGet<int>("RACEMODS_ADDED_LEVEL");
@@ -192,17 +210,7 @@ void Race::ApplyRaceEffects(CNWSCreature *pCreature)
     // the racial modifiers.
     if (effectsLevelAdded)
     {
-        std::vector<uint64_t> remove(128);
-        for (int i = 0; i < pCreature->m_appliedEffects.num; i++)
-        {
-            auto eff = (CGameEffect*)pCreature->m_appliedEffects.element[i];
-            if (eff->m_sCustomTag == "NWNX_Race_RacialMod")
-            {
-                remove.push_back(eff->m_nID);
-            }
-        }
-        for (auto id: remove)
-            pCreature->RemoveEffectById(id);
+        RemoveRaceEffects(pCreature);
     }
 
     // AB
@@ -1190,6 +1198,24 @@ ArgumentStack Race::SetFavoredEnemyFeat(ArgumentStack&& args)
     return ScriptAPI::Arguments();
 }
 
+NWNX_EXPORT ArgumentStack Race::SuppressCreatureRaceEffects(ArgumentStack&& args)
+{
+    if(auto* pCreature = Utils::PopCreature(args))
+    {
+        RemoveRaceEffects(pCreature);
+    }
+    return {};
+}
+
+NWNX_EXPORT ArgumentStack Race::ReactivateCreatureRaceEffects(ArgumentStack&& args)
+{
+    if(auto* pCreature = Utils::PopCreature(args))
+    {
+        ApplyRaceEffects(pCreature);
+    }
+    return {};
+}
+
 int32_t Race::GetAttackModifierVersusHook(CNWSCreatureStats* pStats, CNWSCreature* pCreature)
 {
     int32_t modABVSRaceBonus = 0;

diff --git a/Plugins/Race/Race.hpp b/Plugins/Race/Race.hpp
@@ -25,6 +25,8 @@ class Race : public NWNXLib::Plugin
     ArgumentStack SetRacialModifier(ArgumentStack&& args);
     ArgumentStack GetParentRace(ArgumentStack&& args);
     ArgumentStack SetFavoredEnemyFeat(ArgumentStack&& args);
+    ArgumentStack SuppressCreatureRaceEffects(ArgumentStack&& args);
+    ArgumentStack ReactivateCreatureRaceEffects(ArgumentStack&& args);
 
     enum RaceModifier
     {
@@ -80,6 +82,7 @@ class Race : public NWNXLib::Plugin
 
 
     static void DoEffect(CNWSCreature*, uint16_t, int32_t, int32_t = 0, int32_t = 0, int32_t = 0, int32_t = 0, int32_t = 0);
+    static void RemoveRaceEffects(CNWSCreature*);
     static void ApplyRaceEffects(CNWSCreature*);
     static void SetOrRestoreRace(bool, CNWSCreatureStats*, CNWSCreatureStats* = nullptr);
     static void SetRaceModifier(int32_t, RaceModifier, int32_t, int32_t, int32_t);