/*
 * 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.
 */

#include "rtg_perf_ctrl.h"
#include "dfx/log/ffrt_log_api.h"

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <stdbool.h>
#include <sys/types.h>
#include <sys/ioctl.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>

static int open_perf_ctrl(void)
{
    static bool perf_ctrl_available = true;
    int fd = -1;

    if (!perf_ctrl_available) {
        return -1;
    }

    fd = open("/dev/hisi_perf_ctrl", O_RDWR);
    if (fd < 0) {
        FFRT_LOGW("open perf_ctrl failed");
        perf_ctrl_available = false;
    }

    return fd;
}

void set_task_rtg(pid_t tid, unsigned int grp_id)
{
    struct rtg_group_task data = {tid, grp_id, 0};
    int fd = open_perf_ctrl();
    if (fd < 0) {
        return;
    }

    if (ioctl(fd, PERF_CTRL_SET_TASK_RTG, &data)) {
        FFRT_LOGW("Error set rtg %d,%u. %s", tid, grp_id, strerror(errno));
        close(fd);
        return;
    }
    close(fd);
}

void set_rtg_status(unsigned long long status)
{
    int fd = open_perf_ctrl();
    if (fd < 0) {
        return;
    }

    if (ioctl(fd, PERF_CTRL_SET_FRAME_STATUS, &status)) {
        FFRT_LOGW("Error set rtg status=%llu. %s", status, strerror(errno));
        close(fd);
        return;
    }
    close(fd);
}

void set_rtg_qos(int qos) // MHZ
{
    int fd = open_perf_ctrl();
    if (fd < 0) {
        return;
    }

    if (ioctl(fd, PERF_CTRL_SET_FRAME_RATE, &qos)) {
        FFRT_LOGW("Error set rtg qos=%d. %s", qos, strerror(errno));
        close(fd);
        return;
    }
    close(fd);
}

void set_rtg_load_mode(unsigned int grp_id, bool util_enabled, bool freq_enabled)
{
    struct rtg_load_mode load_mode;

    memset(&load_mode, 0, sizeof(struct rtg_load_mode));
    load_mode.grp_id = grp_id;
    load_mode.util_enabled = !!util_enabled;
    load_mode.freq_enabled = !!freq_enabled;

    int fd = open_perf_ctrl();
    if (fd < 0) {
        return;
    }

    if (ioctl(fd, PERF_CTRL_SET_RTG_LOAD_MODE, &load_mode)) {
        FFRT_LOGW("Error set rtg load_mode %d:%d/%d. %s", load_mode.grp_id, load_mode.util_enabled,
            load_mode.freq_enabled, strerror(errno));
        close(fd);
        return;
    }
    close(fd);
}

void set_task_min_util(pid_t tid, unsigned int util)
{
    struct task_config cfg = {tid, util};
    int fd = open_perf_ctrl();
    if (fd < 0) {
        return;
    }

    if (ioctl(fd, PERF_CTRL_SET_TASK_MIN_UTIL, &cfg)) {
        FFRT_LOGW("Error set min util %d,%u. %s", tid, util, strerror(errno));
        close(fd);
        return;
    }
    close(fd);
}