blob: 4cae2e17c80527039a2386ac2293a984ff5c7013 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
/* Copyright (c) 2020 Johannes Stoelp */
#pragma once
#include <functional>
#include <memory>
namespace nMatcha {
struct Executor;
// An abstract base class for implementing cooperative user threads.
//
// # Usage
//
// To implement a cooperative user thread following interface invoked by
// the Executor must be implemented:
// virtual void threadFn() = 0;
// By calling `yield()` a thread can give control back to the executor.
//
// # Example
//
// ```cpp
// class MyThread : public nMatcha::Thread {
// virtual void threadFn() override {
// puts("Hello");
// yield();
// puts("World");
// }
// };
// ```
//
struct Thread {
Thread(const Thread&) = delete;
Thread& operator=(const Thread&) = delete;
Thread();
virtual ~Thread();
bool isFinished() const;
protected:
void yield();
private:
virtual void threadFn() = 0;
static void entry(void* ctx);
struct Stack {
void* mBottom;
size_t mSize;
void* mPtr;
} mStack;
bool mFinished;
friend struct Executor;
const Executor* mExecutor;
};
// Helper interface to implement yielding for function objects.
struct Yielder {
virtual void yield() = 0;
};
// Utility class to create cooperative user threads from function objects.
//
// # Example
//
// ```cpp
// auto t = nMatcha::FnThread::make([](nMatcha::Yielder& y) {
// puts("Hello");
// y.yield();
// puts("World");
// });
// ```
//
struct FnThread : public Thread, public Yielder {
using UserFn = std::function<void(Yielder&)>;
// Factory method to create `FnThread` objects.
static std::unique_ptr<Thread> make(UserFn f);
private:
UserFn mUserFn;
virtual void threadFn() override;
virtual void yield() override;
enum class CreatorToken {};
public:
FnThread(CreatorToken, UserFn f) : mUserFn(f) {}
};
} // namespace nMatcha
|
