blob: c2777feae966050164782be2557e45650f0ba392 (
plain)
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
|
#pragma once
#include <stdlib.h>
#include <coro.h>
#include <Exceptions.h>
#include <vector>
namespace Balau {
class TaskMan;
class Task;
namespace Events {
class BaseEvent {
public:
BaseEvent() : m_signal(false), m_task(NULL) { }
bool gotSignal() { return m_signal; }
void doSignal() { m_signal = true; }
Task * taskWaiting() { Assert(m_task); return m_task; }
void registerOwner(Task * task) { Assert(m_task == NULL); m_task = task; }
private:
bool m_signal;
Task * m_task;
};
class TaskEvent : public BaseEvent {
public:
TaskEvent(Task * taskWaited);
Task * taskWaited() { return m_taskWaited; }
private:
Task * m_taskWaited;
};
};
class Task {
public:
enum Status {
STARTING,
RUNNING,
IDLE,
STOPPED,
FAULTED,
};
Task();
virtual ~Task();
virtual const char * getName() = 0;
Status getStatus() { return m_status; }
static Task * getCurrentTask();
protected:
void suspend();
virtual void Do() = 0;
void waitFor(Events::BaseEvent * event);
private:
size_t stackSize() { return 128 * 1024; }
void switchTo();
static void coroutine(void *);
void * m_stack;
coro_context m_ctx;
TaskMan * m_taskMan;
Status m_status;
void * m_tls;
friend class TaskMan;
friend class Events::TaskEvent;
typedef std::vector<Events::TaskEvent *> waitedByList_t;
waitedByList_t m_waitedBy;
};
};
|