Replace AIMaster m_lEventQueue with a priority queue

Plugins/Optimizations/CMakeLists.txt

@@ -6,4 +6,5 @@ add_plugin(Optimizations
         "ReconcileAutoMap.cpp"
         "CacheScriptChunks.cpp"
         "ClientGameObjectUpdateTime.cpp"
+        "EventMasterPriorityQueue.cpp"
 )

Plugins/Optimizations/EventMasterPriorityQueue.cpp

@@ -0,0 +1,95 @@
+#include "nwnx.hpp"
+
+#include "API/CAppManager.hpp"
+#include "API/CGameObjectArray.hpp"
+#include "API/CServerAIMaster.hpp"
+#include "API/CServerExoApp.hpp"
+#include "API/CServerExoAppInternal.hpp"
+#include "API/CWorldTimer.hpp"
+#include "API/CNWSPlayer.hpp"
+#include "API/CNWSObject.hpp"
+#include "API/CNWSCreature.hpp"
+#include "API/CExoBase.hpp"
+#include "API/CExoTimers.hpp"
+#include "API/CNetLayer.hpp"
+#include "API/CNWSMessage.hpp"
+
+#include <queue>
+
+namespace Optimizations
+{
+
+using namespace NWNXLib;
+using namespace NWNXLib::API;
+
+struct CServerAIEventNodeComparator
+{
+    bool operator()(const CServerAIEventNode& lhs, const CServerAIEventNode& rhs) const
+    {
+        return (lhs.m_nCalendarDay == rhs.m_nCalendarDay) ? (lhs.m_nTimeOfDay > rhs.m_nTimeOfDay) : lhs.m_nCalendarDay > rhs.m_nCalendarDay;
+    }
+};
+
+std::priority_queue<CServerAIEventNode, std::vector<CServerAIEventNode>, CServerAIEventNodeComparator> g_event_queue;
+
+void ClearEventQueue(CServerAIMaster*)
+{
+    g_event_queue = {};
+}
+
+BOOL AddEventAbsoluteTime(CServerAIMaster*, uint32_t nCalendarDay, uint32_t nTimeOfDay, OBJECT_ID nCallerObjectId, OBJECT_ID nObjectId, uint32_t nEventId, void *pEventData)
+{
+    g_event_queue.emplace(CServerAIEventNode { nCalendarDay, nTimeOfDay, nCallerObjectId, nObjectId, nEventId, pEventData });
+    return true;
+}
+
+BOOL EventPending(CServerAIMaster* ai, uint32_t nCalendarDay, uint32_t nTimeOfDay)
+{
+    if (g_event_queue.empty()) return false;
+
+    const CServerAIEventNode& eventNode = g_event_queue.top();
+    return ai->m_pExoAppInternal->m_pWorldTimer->CompareWorldTimes(eventNode.m_nCalendarDay, eventNode.m_nTimeOfDay, nCalendarDay, nTimeOfDay) <= 0;
+}
+
+BOOL GetPendingEvent(CServerAIMaster*, uint32_t *nCalendarDay, uint32_t *nTimeOfDay, OBJECT_ID *nCallerObjectId, OBJECT_ID *nObjectId, uint32_t *nEventId, void **pEventData)
+{
+    if (g_event_queue.empty()) return false;
+
+    const CServerAIEventNode& eventNode = g_event_queue.top();
+
+    *nCalendarDay     = eventNode.m_nCalendarDay;
+    *nTimeOfDay       = eventNode.m_nTimeOfDay;
+    *nCallerObjectId  = eventNode.m_nCallerObjectId;
+    *nObjectId        = eventNode.m_nObjectId;
+    *nEventId         = eventNode.m_nEventId;
+    *pEventData       = eventNode.m_pEventData;
+
+    g_event_queue.pop();
+    return true; 
+}
+
+void EventMasterPriorityQueue() __attribute__((constructor));
+
+void EventMasterPriorityQueue()
+{
+    // Replace:
+    // 
+    // * CServerAIMaster::ClearEventQueue()
+    // * CServerAIMaster::AddEventAbsoluteTime()
+    // * CServerAIMaster::AddEventAbsoluteTimeViaTail()
+    // * CServerAIMaster::EventPending()
+    // * CServerAIMaster::GetPendingEvent()
+    // 
+    // Don't care about save/load event queue.
+
+    if (Config::Get<bool>("EVENT_MASTER_PRIORITY_QUEUE", false))
+    {
+        static Hooks::Hook _0 = Hooks::HookFunction(API::Functions::_ZN15CServerAIMaster15ClearEventQueueEv, (void*)&ClearEventQueue, Hooks::Order::Early);
+        static Hooks::Hook _1 = Hooks::HookFunction(API::Functions::_ZN15CServerAIMaster20AddEventAbsoluteTimeEjjjjjPv, (void*)&AddEventAbsoluteTime, Hooks::Order::Early);
+        static Hooks::Hook _2 = Hooks::HookFunction(API::Functions::_ZN15CServerAIMaster27AddEventAbsoluteTimeViaTailEjjjjjPv, (void*)&AddEventAbsoluteTime, Hooks::Order::Early);
+        static Hooks::Hook _3 = Hooks::HookFunction(API::Functions::_ZN15CServerAIMaster12EventPendingEjj, (void*)&EventPending, Hooks::Order::Early);
+        static Hooks::Hook _4 = Hooks::HookFunction(API::Functions::_ZN15CServerAIMaster15GetPendingEventEPjS0_S0_S0_S0_PPv, (void*)&GetPendingEvent, Hooks::Order::Early);
+    }
+}
+
+}

Plugins/Optimizations/README.md

@@ -18,3 +18,4 @@ Game optimizations. Improves performance of various game elements.
 | `NWNX_OPTIMIZATIONS_CACHE_SCRIPT_CHUNKS` | true/false | Caches all script chunks, improving performance |
 | `NWNX_OPTIMIZATIONS_CLIENT_GAMEOBJECT_UPDATE_TIME` | int | The global client gameobject update time in microseconds, default 200000 (200 milliseconds) |
 | `NWNX_OPTIMIZATIONS_CLIENT_GAMEOBJECT_UPDATE_TIME_LOADING` | int | The client gameobject update time in microseconds for players loading an area, default 200000 (200 milliseconds) |
+| `NWNX_OPTIMIZATIONS_EVENT_MASTER_PRIORITY_QUEUE` | bool | Replaces the AI event linked list with a priority queue, improving event insert speed from o(n) to o(log n), and eliminating many allocations |