2014-12-16 21:38:14 -08:00
|
|
|
// Copyright 2013 Dolphin Emulator Project / 2014 Citra Emulator Project
|
|
|
|
// Licensed under GPLv2 or any later version
|
2013-09-04 17:17:46 -07:00
|
|
|
// Refer to the license.txt file included.
|
|
|
|
|
2014-08-17 10:45:50 -07:00
|
|
|
#pragma once
|
2013-09-04 17:17:46 -07:00
|
|
|
|
2016-12-11 13:26:23 -08:00
|
|
|
#include <chrono>
|
2014-09-02 22:05:45 -07:00
|
|
|
#include <condition_variable>
|
2016-09-17 17:38:01 -07:00
|
|
|
#include <cstddef>
|
2014-09-02 22:05:45 -07:00
|
|
|
#include <mutex>
|
2016-09-17 17:38:01 -07:00
|
|
|
#include <thread>
|
2015-06-20 14:45:15 -07:00
|
|
|
#include "common/common_types.h"
|
2013-09-04 17:17:46 -07:00
|
|
|
|
2016-04-14 04:53:05 -07:00
|
|
|
namespace Common {
|
2013-09-04 17:17:46 -07:00
|
|
|
|
|
|
|
void SetThreadAffinity(std::thread::native_handle_type thread, u32 mask);
|
|
|
|
void SetCurrentThreadAffinity(u32 mask);
|
2014-11-19 00:49:13 -08:00
|
|
|
|
2015-04-16 13:55:30 -07:00
|
|
|
class Event {
|
2013-09-04 17:17:46 -07:00
|
|
|
public:
|
2016-09-18 18:01:46 -07:00
|
|
|
Event() : is_set(false) {}
|
2014-04-01 15:20:08 -07:00
|
|
|
|
2015-04-16 13:55:30 -07:00
|
|
|
void Set() {
|
2016-04-14 04:53:05 -07:00
|
|
|
std::lock_guard<std::mutex> lk(mutex);
|
2015-04-16 13:55:30 -07:00
|
|
|
if (!is_set) {
|
2014-04-01 15:20:08 -07:00
|
|
|
is_set = true;
|
2016-04-14 04:53:05 -07:00
|
|
|
condvar.notify_one();
|
2014-04-01 15:20:08 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-04-16 13:55:30 -07:00
|
|
|
void Wait() {
|
2016-04-14 04:53:05 -07:00
|
|
|
std::unique_lock<std::mutex> lk(mutex);
|
2016-09-17 17:38:01 -07:00
|
|
|
condvar.wait(lk, [&] { return is_set; });
|
2014-04-01 15:20:08 -07:00
|
|
|
is_set = false;
|
|
|
|
}
|
|
|
|
|
2016-12-11 13:26:23 -08:00
|
|
|
template <class Clock, class Duration>
|
|
|
|
bool WaitUntil(const std::chrono::time_point<Clock, Duration>& time) {
|
|
|
|
std::unique_lock<std::mutex> lk(mutex);
|
|
|
|
if (!condvar.wait_until(lk, time, [this] { return is_set; }))
|
|
|
|
return false;
|
|
|
|
is_set = false;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2015-04-16 13:55:30 -07:00
|
|
|
void Reset() {
|
2016-04-14 04:53:05 -07:00
|
|
|
std::unique_lock<std::mutex> lk(mutex);
|
2016-09-17 17:38:01 -07:00
|
|
|
// no other action required, since wait loops on the predicate and any lingering signal will
|
|
|
|
// get cleared on the first iteration
|
2014-04-01 15:20:08 -07:00
|
|
|
is_set = false;
|
|
|
|
}
|
2013-09-04 17:17:46 -07:00
|
|
|
|
|
|
|
private:
|
2015-04-16 13:55:30 -07:00
|
|
|
bool is_set;
|
2016-04-14 04:53:05 -07:00
|
|
|
std::condition_variable condvar;
|
|
|
|
std::mutex mutex;
|
2013-09-04 17:17:46 -07:00
|
|
|
};
|
|
|
|
|
2015-04-16 13:55:30 -07:00
|
|
|
class Barrier {
|
2013-09-04 17:17:46 -07:00
|
|
|
public:
|
2018-09-15 06:21:06 -07:00
|
|
|
explicit Barrier(std::size_t count_) : count(count_), waiting(0), generation(0) {}
|
2014-04-01 15:20:08 -07:00
|
|
|
|
2015-04-16 13:55:30 -07:00
|
|
|
/// Blocks until all "count" threads have called Sync()
|
|
|
|
void Sync() {
|
2016-04-14 04:53:05 -07:00
|
|
|
std::unique_lock<std::mutex> lk(mutex);
|
2018-09-15 06:21:06 -07:00
|
|
|
const std::size_t current_generation = generation;
|
2014-04-01 15:20:08 -07:00
|
|
|
|
2016-04-14 04:53:05 -07:00
|
|
|
if (++waiting == count) {
|
2016-04-14 04:54:06 -07:00
|
|
|
generation++;
|
2016-04-14 04:53:05 -07:00
|
|
|
waiting = 0;
|
|
|
|
condvar.notify_all();
|
2015-04-16 13:55:30 -07:00
|
|
|
} else {
|
2016-09-17 17:38:01 -07:00
|
|
|
condvar.wait(lk,
|
|
|
|
[this, current_generation] { return current_generation != generation; });
|
2014-04-01 15:20:08 -07:00
|
|
|
}
|
|
|
|
}
|
2013-09-04 17:17:46 -07:00
|
|
|
|
|
|
|
private:
|
2016-04-14 04:53:05 -07:00
|
|
|
std::condition_variable condvar;
|
|
|
|
std::mutex mutex;
|
2018-09-15 06:21:06 -07:00
|
|
|
const std::size_t count;
|
|
|
|
std::size_t waiting;
|
|
|
|
std::size_t generation; // Incremented once each time the barrier is used
|
2013-09-04 17:17:46 -07:00
|
|
|
};
|
2014-11-19 00:49:13 -08:00
|
|
|
|
2016-09-17 17:38:01 -07:00
|
|
|
void SwitchCurrentThread(); // On Linux, this is equal to sleep 1ms
|
|
|
|
void SetCurrentThreadName(const char* name);
|
2014-11-19 00:49:13 -08:00
|
|
|
|
2013-09-04 17:17:46 -07:00
|
|
|
} // namespace Common
|