TXuian
62 lines
1.4 KiB
C
62 lines
1.4 KiB
C
|
|
#pragma once
|
|
#include "actracer.h"
|
|
#include "ksemaphore.h"
|
|
#include "rbtree.h"
|
|
|
|
#define TASK_MAX_PRIORITY 32
|
|
#define UNINIT_SNODE_ID 0
|
|
typedef uintptr_t snode_id_t;
|
|
|
|
enum ThreadState {
|
|
NEVER_RUN = 0,
|
|
INIT,
|
|
READY,
|
|
RUNNING,
|
|
DEAD,
|
|
BLOCKED,
|
|
SLEEPING,
|
|
NR_STATE,
|
|
|
|
// follow state is temp for kernel use
|
|
TRANS_WAKING,
|
|
};
|
|
|
|
typedef struct ScheduleContext {
|
|
intptr_t remain_tick;
|
|
uint64_t run_time;
|
|
intptr_t unblock_signals;
|
|
} ScheduleContext;
|
|
|
|
typedef struct TaskSleepContext {
|
|
int64_t remain_ms;
|
|
} TaskSleepContext;
|
|
|
|
struct ScheduleNode {
|
|
struct Thread* pthd;
|
|
snode_id_t snode_id;
|
|
enum ThreadState state;
|
|
Queue state_trans_signal_queue;
|
|
|
|
ScheduleContext sched_context;
|
|
TaskSleepContext sleep_context;
|
|
};
|
|
|
|
struct Scheduler {
|
|
TraceTag tag;
|
|
RbtTree snode_state_pool[NR_STATE];
|
|
RbtTree state_trans_ref_map;
|
|
struct XiziSemaphorePool semaphore_pool;
|
|
};
|
|
|
|
extern struct Scheduler g_scheduler;
|
|
|
|
bool init_schedule_node(struct ScheduleNode* snode, struct Thread* bind_thd);
|
|
void enqueue_task_trans_state(struct Thread* thd, enum ThreadState state);
|
|
#define THREAD_TRANS_STATE(thd, state) enqueue_task_trans_state(thd, state);
|
|
|
|
bool task_trans_sched_state(struct ScheduleNode* snode, RbtTree* from_pool, RbtTree* to_pool, enum ThreadState target_state);
|
|
void task_block(struct Thread* thd);
|
|
void task_dead(struct Thread* thd);
|
|
void task_into_ready(struct Thread* thd);
|