/*
 * Copyright (c) 2023 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef SCHED_RTG_H
#define SCHED_RTG_H

#include <unistd.h>
#include <stdbool.h>

#ifdef __cplusplus
extern "C" {
#endif

#define DEFAULT_RT_FRAME_ID 8
#define FRAME_START (1 << 0)
#define FRAME_END (1 << 1)

void set_task_rtg(pid_t tid, unsigned int grp_id);
void set_rtg_status(unsigned long long status);
void set_rtg_qos(int qos);
void set_rtg_load_mode(unsigned int grp_id, bool util_enabled, bool freq_enabled);
void set_task_min_util(pid_t tid, unsigned int util);

/* inner use */

struct rtg_group_task {
    pid_t pid;
    unsigned int grp_id;
    bool pmu_sample_enabled;
};

struct rtg_load_mode {
    unsigned int grp_id;
    unsigned int freq_enabled;
    unsigned int util_enabled;
};

struct task_config {
    pid_t pid;
    unsigned int value;
};

#define SET_TASK_RTG 11
#define SET_FRAME_STATUS 10
#define SET_FRAME_RATE 8
#define SET_RTG_LOAD_MODE 22
#define SET_TASK_MIN_UTIL 28

#define PERF_CTRL_MAGIC 'x'
#define PERF_CTRL_SET_TASK_RTG _IOWR(PERF_CTRL_MAGIC, SET_TASK_RTG, struct rtg_group_task)
#define PERF_CTRL_SET_FRAME_STATUS _IOWR(PERF_CTRL_MAGIC, SET_FRAME_STATUS, unsigned long long)
#define PERF_CTRL_SET_FRAME_RATE _IOWR(PERF_CTRL_MAGIC, SET_FRAME_RATE, int)
#define PERF_CTRL_SET_RTG_LOAD_MODE _IOW(PERF_CTRL_MAGIC, SET_RTG_LOAD_MODE, struct rtg_load_mode)
#define PERF_CTRL_SET_TASK_MIN_UTIL _IOW(PERF_CTRL_MAGIC, SET_TASK_MIN_UTIL, struct task_config)

#ifdef __cplusplus
}
#endif

#endif /* SCHED_RTG_H */