/* SPDX-License-Identifier: GPL-2.0 */
#include <linux/atomic.h>
#include <linux/cpumask.h>
#include <linux/sched/rt.h>

#define CPUPRI_NR_PRIORITIES (MAX_RT_PRIO+1)

#define CPUPRI_INVALID  -1
#define CPUPRI_NORMAL   0
/* values 1-99 are for RT1-RT99 priorities */
#define CPUPRI_HIGHER  100

struct cpupri_vec {
 atomic_t  count;
 cpumask_var_t  mask;
};

struct cpupri {
 struct cpupri_vec pri_to_cpu[CPUPRI_NR_PRIORITIES];
 int   *cpu_to_pri;
};

int  cpupri_find(struct cpupri *cp, struct task_struct *p,
   struct cpumask *lowest_mask);
int  cpupri_find_fitness(struct cpupri *cp, struct task_struct *p,
    struct cpumask *lowest_mask,
    bool (*fitness_fn)(struct task_struct *p, int cpu));
void cpupri_set(struct cpupri *cp, int cpu, int pri);
int  cpupri_init(struct cpupri *cp);
void cpupri_cleanup(struct cpupri *cp);