summaryrefslogtreecommitdiff
path: root/src/KingSystem/Utils/Thread/ManagedTaskHandle.h
blob: 61c41c51d081a2abfa3885b1a80deb5399ab83d1 (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
#pragma once

#include <basis/seadTypes.h>
#include <time/seadTickSpan.h>
#include "KingSystem/Utils/Types.h"

namespace ksys::util {

class ManagedTask;
class TaskQueueBase;

class ManagedTaskHandle {
public:
    enum class Status {
        Uninitialized = 0,
        TaskAttached = 1,
        TaskFinished = 2,
        TaskRemoved = 3,
    };

    ManagedTaskHandle();
    virtual ~ManagedTaskHandle();

    void finalize();

    Status getStatus() const { return mStatus; }

    void removeTaskFromQueue();
    bool hasTask() const;

    bool wait();
    bool wait(const sead::TickSpan& wait_duration);

    bool isTaskAttached() const;
    bool didTaskCompleteSuccessfully() const;

private:
    friend class ManagedTask;

    struct SetTaskArg {
        ManagedTask* task;
        TaskQueueBase* queue;
    };

    bool attachTask(const SetTaskArg& arg);
    void setIsSuccess(bool success);
    void setStatus(Status status);

    void incrementRef_();
    void decrementRef_();

    bool mSuccess = false;
    Status mStatus = Status::Uninitialized;
    s32 mRefCount = 0;
    ManagedTask* mTask = nullptr;
    TaskQueueBase* mQueue = nullptr;
};
KSYS_CHECK_SIZE_NX150(ManagedTaskHandle, 0x28);

}  // namespace ksys::util