rpcs3/Utilities/cond.cpp

60 lines
1.1 KiB
C++
Raw Normal View History

#include "cond.h"
#include "sync.h"
#include "lockless.h"
#include <climits>
2017-02-24 15:48:53 +00:00
// use constants, increase signal space
void cond_variable::imp_wait(u32 _old, u64 _timeout) noexcept
{
// Not supposed to fail
ensure(_old);
// Wait with timeout
m_value.wait(_old, c_signal_mask, atomic_wait_timeout{_timeout > max_timeout ? UINT64_MAX : _timeout * 1000});
// Cleanup
m_value.atomic_op([](u32& value)
{
// Remove waiter (c_waiter_mask)
value -= 1;
if ((value & c_waiter_mask) == 0)
{
// Last waiter removed, clean signals
value = 0;
}
});
}
void cond_variable::imp_wake(u32 _count) noexcept
{
const auto [_old, ok] = m_value.fetch_op([](u32& value)
{
if (!value || (value & c_signal_mask) == c_signal_mask)
{
return false;
}
// Add signal
2020-03-04 14:08:40 +00:00
value += c_signal_mask & (0 - c_signal_mask);
return true;
});
if (!ok || !_count)
{
return;
}
2020-03-04 14:08:40 +00:00
if (_count > 1 || ((_old + (c_signal_mask & (0 - c_signal_mask))) & c_signal_mask) == c_signal_mask)
{
// Resort to notify_all if signal count reached max
m_value.notify_all(c_signal_mask);
}
else
{
m_value.notify_one(c_signal_mask);
}
}