#include <stdio.h>
#include <string.h>
#include <sys/time.h>
#include "init.h"
#include "unitt.h"
#include "kern.h"

#if 0
#ifdef _WIN32
#include <windows.h>
#include <psapi.h> // -lPsapi
#else
#include <unistd.h>
#include <pthread.h>
#endif

void get_memory_usage(int* resident_set_size, int* virtual_memory_size) 
{
#ifdef _WIN32
    PROCESS_MEMORY_COUNTERS pmc;
    if (GetProcessMemoryInfo(GetCurrentProcess(), &pmc, sizeof(pmc))) {
        *resident_set_size = pmc.WorkingSetSize / 1024; // 转换为KB
        *virtual_memory_size = pmc.PagefileUsage / 1024; // 转换为KB
    } else {
        *resident_set_size = 0;
        *virtual_memory_size = 0;
    }
#else
    FILE* file = fopen("/proc/self/statm", "r");
    if (file) {
        fscanf(file, "%d %d", resident_set_size, virtual_memory_size);
        fclose(file);
    } else {
        *resident_set_size = 0;
        *virtual_memory_size = 0;
    }
#endif
}
#endif 

#define SINGLE_TCOUNT 1000

uint64_t unitt_clock(void)
{
    struct timeval mstime;
    uint64_t us = 0;
    gettimeofday(&mstime, NULL);
    us = mstime.tv_sec * 1000000 + mstime.tv_usec;
    return us;
}

// 计算两个整数的和
int unit_add(int a, int b) 
{
    return a + b;
}

// 计算两个整数的差
int unit_subtract(int a, int b) 
{
    return a - b;
}

// 计算两个整数的乘积
int unit_multiply(int a, int b) 
{
    return a * b;
}

// 计算两个整数的商
float unit_divide(int a, int b) 
{
    if (b == 0) 
    {
        return 0; // 简单处理，真实应用中应更严格处理
    }
    return (float)a / (float)b;
}

// 测试前的初始化函数
int setup() 
{
    // 初始化代码（如果需要）
    return 0;
}

// 测试后的清理函数
int teardown() 
{
    // 清理代码（如果需要）
    return 0;
}

// 随机输入生成函数
int random_input() 
{
    return rand() % 100; // 生成0到99之间的随机整数
}

// 测试加法函数，使用随机输入
int test_add() 
{
    // 进行10次随机测试
    for (int i = 0; i < SINGLE_TCOUNT; i++) 
    { 
        int a = random_input();
        int b = random_input();
        if (UNITT_E_FAIL == unitt_det_equal(unit_add(a, b), a + b, "Random addition failed")) return UNITT_E_FAIL;
    }

    return UNITT_E_OK;
}

// 测试减法函数，使用随机输入
int test_subtract() 
{
    // 进行10次随机测试
    for (int i = 0; i < SINGLE_TCOUNT; i++) 
    { 
        int a = random_input();
        int b = random_input();
        if (UNITT_E_FAIL == unitt_det_equal(unit_subtract(a, b), a - b, "Random subtraction failed")) return UNITT_E_FAIL;
    }

    return UNITT_E_OK;
}

// 测试乘法函数，使用随机输入
int test_multiply() 
{
    // 进行10次随机测试
    for (int i = 0; i < SINGLE_TCOUNT; i++) 
    { 
        int a = random_input();
        int b = random_input();
        if (UNITT_E_FAIL == unitt_det_equal(unit_multiply(a, b), a * b, "Random multiplication failed")) return UNITT_E_FAIL;
    }

    return UNITT_E_OK;
}

// 测试除法函数，使用随机输入
int test_divide() 
{
    // 进行10次随机测试
    for (int i = 0; i < SINGLE_TCOUNT; i++) 
    { 
        int a = random_input();
        int b = random_input();
        if (b != 0) 
        {
            if (UNITT_E_FAIL == unitt_det_float(unit_divide(a, b), (float)a / (float)b, 0.00001, "Random division failed")) return UNITT_E_FAIL;
        }
    }

    return UNITT_E_OK;
}

void unit_main() 
{
    static UNITT_TCASE math_tests[] = {
        UNITT_TCASE(test_add),
        UNITT_TCASE(test_subtract),
        UNITT_TCASE(test_multiply),
        UNITT_TCASE(test_divide),
    };

    static UNITT suites[] = {
        { "Math Suite", math_tests, sizeof(math_tests) / sizeof(math_tests[0]) , unitt_clock },
    };

    UNITT_EXE(suites);
}

static void test(void)
{
    srand((unsigned int)time(NULL)); // 初始化随机数种子
    
    // unit_main();
    printf("create task %d\r\n", task_create(1000, unit_main));
}
init_export_app(test);