ParallelKitNET.cpp

#include "ParallelKitNET.h"
#include <iostream>

// 大部分实现已经在头文件中完成，这里主要提供一些额外的辅助函数

// 示例：一个帮助函数，直接返回处理结果而不是future
double ParallelKitNET::processAndWait(const std::vector<double> &x) {
    auto future = process(x, false);
    return future.get();
}

// 示例：一个帮助函数，直接返回训练结果而不是future
double ParallelKitNET::trainAndWait(const std::vector<double> &x) {
    auto future = train(x, false);
    return future.get();
}

// 批量处理一组特征向量，并等待所有结果
std::vector<double>
ParallelKitNET::batchProcess(const std::vector<std::vector<double>> &batch,
                             bool block) {
    std::vector<std::future<double>> futures;
    futures.reserve(batch.size());

    // 提交所有工作项
    for (const auto &x : batch) {
        futures.push_back(process(x, false));
    }

    // 如果不阻塞，直接返回空结果
    if (!block) {
        return {};
    }

    // 收集所有结果
    std::vector<double> results;
    results.reserve(batch.size());

    for (auto &future : futures) {
        results.push_back(future.get());
    }

    return results;
}

// 批量训练一组特征向量，并等待所有结果
std::vector<double>
ParallelKitNET::batchTrain(const std::vector<std::vector<double>> &batch,
                           bool block) {
    std::vector<std::future<double>> futures;
    futures.reserve(batch.size());

    // 提交所有工作项
    for (const auto &x : batch) {
        futures.push_back(train(x, false));
    }

    // 如果不阻塞，直接返回空结果
    if (!block) {
        return {};
    }

    // 收集所有结果
    std::vector<double> results;
    results.reserve(batch.size());

    for (auto &future : futures) {
        results.push_back(future.get());
    }

    return results;
}

// 调整工作线程数量
void ParallelKitNET::resizeThreadPool(size_t num_threads) {
    // 停止当前所有线程
    stopThreads();

    // 启动新的线程池
    startThreads(num_threads);
}

// 等待所有当前队列中的任务完成
void ParallelKitNET::waitForAllTasks() {
    while (!workQueue.empty() && running) {
        std::this_thread::sleep_for(std::chrono::milliseconds(10));
    }
}