Files
WickedEngine/WickedEngine/wiEventHandler.cpp
T
2024-07-25 18:59:46 +02:00

96 lines
2.2 KiB
C++

#include "wiEventHandler.h"
#include "wiUnorderedMap.h"
#include <list>
#include <mutex>
namespace wi::eventhandler
{
struct EventManager
{
// Note: list is used because events can also add events from within and that mustn't cause invalidation like vector
wi::unordered_map<int, std::list<std::function<void(uint64_t)>*>> subscribers;
wi::unordered_map<int, std::list<std::function<void(uint64_t)>>> subscribers_once;
std::mutex locker;
};
std::shared_ptr<EventManager> manager = std::make_shared<EventManager>();
struct EventInternal
{
std::shared_ptr<EventManager> manager;
int id = 0;
std::function<void(uint64_t)> callback;
~EventInternal()
{
manager->locker.lock();
auto it = manager->subscribers.find(id);
if (it != manager->subscribers.end())
{
it->second.remove(&callback);
}
manager->locker.unlock();
}
};
Handle Subscribe(int id, std::function<void(uint64_t)> callback)
{
Handle handle;
auto eventinternal = std::make_shared<EventInternal>();
handle.internal_state = eventinternal;
eventinternal->manager = manager;
eventinternal->id = id;
eventinternal->callback = callback;
manager->locker.lock();
manager->subscribers[id].push_back(&eventinternal->callback);
manager->locker.unlock();
return handle;
}
void Subscribe_Once(int id, std::function<void(uint64_t)> callback)
{
manager->locker.lock();
manager->subscribers_once[id].push_back(callback);
manager->locker.unlock();
}
void FireEvent(int id, uint64_t userdata)
{
// Callbacks that only live for once:
{
manager->locker.lock();
auto it = manager->subscribers_once.find(id);
bool found = it != manager->subscribers_once.end();
manager->locker.unlock();
if (found)
{
auto& callbacks = it->second;
for (auto& callback : callbacks)
{
callback(userdata);
}
callbacks.clear();
}
}
// Callbacks that live until deleted:
{
manager->locker.lock();
auto it = manager->subscribers.find(id);
bool found = it != manager->subscribers.end();
manager->locker.unlock();
if (found)
{
auto& callbacks = it->second;
for (auto& callback : callbacks)
{
(*callback)(userdata);
}
}
}
}
}