/* SPDX-License-Identifier: GPL-2.0 */
#ifndef __SCHED_RTG_H
#define __SCHED_RTG_H

#ifdef CONFIG_SCHED_RTG

#define DEFAULT_RTG_GRP_ID	0
#define DEFAULT_CGROUP_COLOC_ID	1
#define MAX_NUM_CGROUP_COLOC_ID	21

struct group_cpu_time {
	u64	window_start;
	u64	curr_runnable_sum;
	u64	prev_runnable_sum;
	u64	nt_curr_runnable_sum;
	u64	nt_prev_runnable_sum;
};

struct group_ravg {
	unsigned long curr_window_load;
	unsigned long curr_window_exec;
	unsigned long prev_window_load;
	unsigned long prev_window_exec;
	unsigned long normalized_util;
};

struct rtg_class;

struct related_thread_group {
	int id;
	raw_spinlock_t lock;
	struct list_head tasks;
	struct list_head list;

	unsigned int nr_running;
	struct group_ravg ravg;
	u64 window_start;
	u64 mark_start;
	u64 prev_window_time;
	/* rtg window information for WALT */
	unsigned int window_size;
	const struct rtg_class *rtg_class;
	struct sched_cluster *preferred_cluster;
	int max_boost;
	unsigned long util_invalid_interval; /* in nanoseconds */
	unsigned long util_update_timeout; /* in nanoseconds */
	unsigned long freq_update_interval; /* in nanoseconds */
	u64 last_util_update_time;
	u64 last_freq_update_time;
	void *private_data;
};

struct rtg_class {
	void (*sched_update_rtg_tick)(struct related_thread_group *grp);
};

enum rtg_freq_update_flags {
	RTG_FREQ_FORCE_UPDATE = (1 << 0),
	RTG_FREQ_NORMAL_UPDATE = (1 << 1),
};

int sched_set_group_id(struct task_struct *p, unsigned int group_id);
unsigned int sched_get_group_id(struct task_struct *p);
#endif /* CONFIG_SCHED_RTG */
#endif