summaryrefslogtreecommitdiff
path: root/src/KingSystem/Framework/frmWorkerSupportThreadMgr.h
blob: 2ec99fa26357b7bd9d689e540681331befb508c9 (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
#pragma once

#include <heap/seadDisposer.h>
#include <prim/seadSafeString.h>
#include <thread/seadMessageQueue.h>
#include "KingSystem/Utils/Thread/Task.h"
#include "KingSystem/Utils/Types.h"

namespace sead {
class DelegateThread;
class Thread;
}  // namespace sead

namespace ksys::util {
class GameTaskThread;
class ManagedTask;
class ManagedTaskHandle;
class TaskMgr;
}  // namespace ksys::util

namespace ksys::frm {

class WorkerSupportThreadMgr {
    SEAD_SINGLETON_DISPOSER(WorkerSupportThreadMgr)
    WorkerSupportThreadMgr() = default;
    ~WorkerSupportThreadMgr() = default;

public:
    void init(sead::Heap* heap);

    void submitRequest(int id, util::TaskDelegate* delegate);
    void waitForTask(int id);
    void pauseThreads();
    void resumeThreads();
    void destroyThreads();

private:
    static constexpr int NumWorkers = 2;
    static constexpr int NumTasks = 5;

    struct Worker {
        util::GameTaskThread* task_thread = nullptr;
        sead::DelegateThread* thread;
        util::ManagedTask* tasks[NumTasks];
        util::TaskMgr* task_mgr = nullptr;
        util::ManagedTaskHandle* task_handle;
        sead::FixedSafeString<32> thread_name;
    };

    void initSleeperThreads();
    void sleeperThreadFun(sead::Thread* thread, sead::MessageQueue::Element);

    util::ManagedTask* getTask(int id);

    Worker mWorkers[NumWorkers];
    bool mThreadsPaused = false;
    sead::Heap* mSleeperHeap;
};
KSYS_CHECK_SIZE_NX150(WorkerSupportThreadMgr, 0x130);

}  // namespace ksys::frm