/**
 * Created by Jingyu Yan
 * @date 2024-10-01
 */

#include "test_settings.h"

std::string getTestDataDir() {
    return Enviro::getInstance().getTestResDir();
}

std::string getTestData(const std::string& name) {
    return getTestDataDir() + "/" + name;
}

std::string getTestSaveDir() {
    return getTestData("save");
}

std::string getTestSaveData(const std::string& name) {
    return getTestSaveDir() + "/" + name;
}

std::string getTestModelsFile() {
    std::string path = "pack/";
    path = path + TEST_MODEL_FILE;
    return getTestData(path);
}

std::string getTestLFWFunneledTxt() {
    return getTestData(TEST_LFW_FUNNELED_TXT);
}

std::string getTestLFWFunneledEvaTxt() {
    return getLFWFunneledDir() + "/" + TEST_LFW_EVALUATION_TXT;
}

std::string getLFWFunneledDir() {
    const char* testLfwFunneledDir = LFW_FUNNELED_DIR;
    std::string lfwFunneledPath;
    if (testLfwFunneledDir != nullptr && std::string(testLfwFunneledDir) != "") {
        lfwFunneledPath = testLfwFunneledDir;
    } else {
        const char* lfwFunneled = std::getenv("LFW_FUNNELED_DIR");
        if (lfwFunneled != nullptr) {
            lfwFunneledPath = lfwFunneled;
        }
    }
    if (lfwFunneledPath.empty()) {
        spdlog::warn("lfw funneled dir is empty!");
    }
    return lfwFunneledPath;
}

std::string getBenchmarkRecordFile() {
    return getTestSaveData(TEST_BENCHMARK_RECORD);
}

std::string getEvaluationRecordFile() {
    return getTestSaveData(TEST_EVALUATION_RECORD);
}