#pragma once

#include <common/glib.h>
#include <process/process.h>

/*
 * Scheduling policies
 */
#define SCHED_NORMAL 0
#define SCHED_FIFO 1
#define SCHED_RR 2
#define SCHED_BATCH 3
/* SCHED_ISO: reserved but not implemented yet */
#define SCHED_IDLE 5
#define SCHED_DEADLINE 6
#define SCHED_MAX_POLICY_NUM SCHED_DEADLINE

#define IS_VALID_SCHED_POLICY(_policy) ((_policy) > 0 && (_policy) <= SCHED_MAX_POLICY_NUM)

// struct sched_param
// {
//     int sched_priority;
// };
// struct sched_attr
// {
//     uint32_t size;

//     uint32_t sched_policy;
//     uint64_t sched_flags;

//     /* SCHED_NORMAL, SCHED_BATCH */
//     int32_t sched_nice;

//     /* SCHED_FIFO, SCHED_RR */
//     uint32_t sched_priority;

//     /* SCHED_DEADLINE */
//     uint64_t sched_runtime;
//     uint64_t sched_deadline;
//     uint64_t sched_period;

//     /* Utilization hints */
//     uint32_t sched_util_min;
//     uint32_t sched_util_max;
// };

// static int __sched_setscheduler(struct process_control_block *p, const struct sched_attr *attr, bool user, bool pi);
// static int _sched_setscheduler(struct process_control_block *p, int policy, const struct sched_param *param,
//                                bool check);
// /**
//  * sched_setscheduler -设置进程的调度策略
//  * @param p 需要修改的pcb
//  * @param policy 需要设置的policy
//  * @param param structure containing the new RT priority. 目前没有用
//  *
//  * @return 成功返回0,否则返回对应的错误码
//  *
//  */
// int sched_setscheduler(struct process_control_block *p, int policy, const struct sched_param *param);

// ================= Rust 实现 =============
extern void sched_update_jiffies();
extern void sched_init();
extern void sched();
extern void sched_enqueue(struct process_control_block *pcb, bool reset_time);
extern void sched_set_cpu_idle(uint64_t cpu_id, struct process_control_block *pcb);
extern void sched_migrate_process(struct process_control_block *pcb, uint64_t target);

void switch_proc(struct process_control_block *prev, struct process_control_block *proc);