1
0
Fork 0
forked from suyu/suyu
suyu/src/common/fiber.h

93 lines
2.9 KiB
C++
Raw Normal View History

2020-02-04 20:06:23 +01:00
// Copyright 2020 yuzu Emulator Project
// Licensed under GPLv2 or any later version
// Refer to the license.txt file included.
#pragma once
#include <functional>
#include <memory>
#include "common/common_types.h"
#include "common/spin_lock.h"
#if !defined(_WIN32) && !defined(WIN32)
namespace boost::context::detail {
2020-02-10 18:33:13 +01:00
struct transfer_t;
}
#endif
2020-02-04 20:06:23 +01:00
namespace Common {
/**
* Fiber class
* a fiber is a userspace thread with it's own context. They can be used to
* implement coroutines, emulated threading systems and certain asynchronous
* patterns.
*
* This class implements fibers at a low level, thus allowing greater freedom
* to implement such patterns. This fiber class is 'threadsafe' only one fiber
* can be running at a time and threads will be locked while trying to yield to
* a running fiber until it yields. WARNING exchanging two running fibers between
* threads will cause a deadlock. In order to prevent a deadlock, each thread should
* have an intermediary fiber, you switch to the intermediary fiber of the current
* thread and then from it switch to the expected fiber. This way you can exchange
* 2 fibers within 2 different threads.
*/
2020-02-04 20:06:23 +01:00
class Fiber {
public:
Fiber(std::function<void(void*)>&& entry_point_func, void* start_parameter);
~Fiber();
Fiber(const Fiber&) = delete;
Fiber& operator=(const Fiber&) = delete;
2020-10-21 04:07:39 +02:00
Fiber(Fiber&&) = default;
Fiber& operator=(Fiber&&) = default;
2020-02-04 20:06:23 +01:00
/// Yields control from Fiber 'from' to Fiber 'to'
/// Fiber 'from' must be the currently running fiber.
static void YieldTo(std::shared_ptr<Fiber> from, std::shared_ptr<Fiber> to);
[[nodiscard]] static std::shared_ptr<Fiber> ThreadToFiber();
2020-02-04 20:06:23 +01:00
2020-02-27 21:32:47 +01:00
void SetRewindPoint(std::function<void(void*)>&& rewind_func, void* start_parameter);
void Rewind();
2020-02-04 20:06:23 +01:00
/// Only call from main thread's fiber
void Exit();
/// Changes the start parameter of the fiber. Has no effect if the fiber already started
void SetStartParameter(void* new_parameter) {
start_parameter = new_parameter;
}
private:
Fiber();
#if defined(_WIN32) || defined(WIN32)
void OnRewind();
void Start();
static void FiberStartFunc(void* fiber_parameter);
2020-02-27 21:32:47 +01:00
static void RewindStartFunc(void* fiber_parameter);
#else
void OnRewind(boost::context::detail::transfer_t& transfer);
void Start(boost::context::detail::transfer_t& transfer);
static void FiberStartFunc(boost::context::detail::transfer_t transfer);
static void RewindStartFunc(boost::context::detail::transfer_t transfer);
#endif
2020-02-04 20:06:23 +01:00
struct FiberImpl;
2020-02-10 19:45:08 +01:00
SpinLock guard{};
std::function<void(void*)> entry_point;
std::function<void(void*)> rewind_point;
2020-02-27 21:32:47 +01:00
void* rewind_parameter{};
2020-02-10 19:45:08 +01:00
void* start_parameter{};
std::shared_ptr<Fiber> previous_fiber;
2020-02-04 20:06:23 +01:00
std::unique_ptr<FiberImpl> impl;
bool is_thread_fiber{};
bool released{};
2020-02-04 20:06:23 +01:00
};
} // namespace Common