/*
 * SPDX-License-Identifier: Apache-2.0
 */

#include "cpu_load.h"

#ifdef CPU_LOAD_MONITOR

#define CPU_LOAD_MONITOR_SIZE 40

static task_monitor_t task_monitor_list[CPU_LOAD_MONITOR_SIZE] = {0};
static uint32_t task_count = 0;
static uint64_t monitor_base = 0;

void thread_init_hook(rt_thread_t thread)
{
    int found;
    int index;
    rt_enter_critical();
    found = 0;
    for(index = 0; index < task_count; index++)
    {
        if(thread == task_monitor_list[index].thread)
        {
            found = 1;
            break;
        }
    }
    if(!found)
    {
        if(task_count < CPU_LOAD_MONITOR_SIZE)
        {
            task_monitor_list[task_count].thread = thread;
            task_count++;
        }
    }
    else
    {
        task_monitor_list[index].base = 0;
    }
    rt_exit_critical();
}

void sched_hook(struct rt_thread* from, struct rt_thread* to)
{
    int task_found = 0;
    int index = 0;
    static uint64_t base = 0;
    if(!base)
    {
        base = aic_get_time_us64();
        monitor_base = base;
    }
    uint64_t time = aic_get_time_us64();
    for(index = 0; index < task_count; index++)
    {
        if(task_monitor_list[index].thread == from)
        {
            task_found = 1;
            break;
        }
    }
    if(task_found != 1)
    {
        printf("error occur !!!\n");
    }
    else
    {
        if(task_monitor_list[index].base == 0)
        {
            task_monitor_list[index].base = time;
            task_monitor_list[index].count = time;
        }
        else
        {
            task_monitor_list[index].count += time-base;
        }
    }
    base = time;
}

int cpu_load(int argc, const char **argv)
{
    (void)argc;
    (void)argv;

    uint16_t total = 0, value = 0;

    uint64_t time = aic_get_time_us64() - monitor_base;

    printf("ID   %%CPU Task name\n");
    printf("--- ----- ----------------\n");
    for(int i = 0; i < task_count; i++)
    {
        value = (task_monitor_list[i].count- task_monitor_list[i].base)*1000/time;
        printf("%-3d %2d.%d%% %-s\n", i + 1,
               value / 10, value % 10, task_monitor_list[i].thread->name);
        if (strncmp("tidle", task_monitor_list[i].thread->name, 5) != 0)
            total += value;
    }

    if (total > 1000)
        total = 1000;
    printf("\t\tTotal: %d.%d%%\n", total / 10, total % 10);

    rt_enter_critical();
    monitor_base = aic_get_time_us64();
    for(int i = 0; i < task_count; i++)
    {
        task_monitor_list[i].base = monitor_base;
        task_monitor_list[i].count = monitor_base;
    }
    rt_exit_critical();
    return 0;
}

MSH_CMD_EXPORT(cpu_load, CPU monitor);
MSH_CMD_EXPORT_ALIAS(cpu_load, top, CPU monitor);
#endif