stt/pybind/python_progress_wrapper.h

#ifndef PYTHON_PROGRESS_WRAPPER_H
#define PYTHON_PROGRESS_WRAPPER_H

#ifdef PYTHON_BINDING

#include <pybind11/pybind11.h>
#include <pybind11/functional.h>
#include <string>
#include <memory>

namespace py = pybind11;

// Python进度回调接口
class PythonProgressCallback {
public:
    virtual ~PythonProgressCallback() = default;
    virtual void update(const std::string& description, double percentage) = 0;
    virtual void set_description(const std::string& description) = 0;
    virtual void set_total(unsigned long total) = 0;
    virtual void update_progress(unsigned long current) = 0;
    virtual void finish() = 0;
};

// pybind11包装器
class PyProgressCallback : public PythonProgressCallback {
private:
    py::object callback_func_;
    std::string current_description_;
    unsigned long total_;
    bool has_total_;
    
public:
    PyProgressCallback(py::object callback) 
        : callback_func_(callback), total_(0), has_total_(false) {}
    
    void update(const std::string& description, double percentage) override {
        if (callback_func_ && !callback_func_.is_none()) {
            try {
                callback_func_(description, percentage);
            } catch (const std::exception& e) {
                // 忽略Python回调中的异常，避免崩溃
            }
        }
    }
    
    void set_description(const std::string& description) override {
        current_description_ = description;
    }
    
    void set_total(unsigned long total) override {
        total_ = total;
        has_total_ = true;
    }
    
    void update_progress(unsigned long current) override {
        if (has_total_ && total_ > 0) {
            double percentage = (static_cast<double>(current) / total_) * 100.0;
            update(current_description_, percentage);
        }
    }
    
    void finish() override {
        update(current_description_, 100.0);
    }
};

// 全局进度回调管理器
class ProgressCallbackManager {
private:
    static std::unique_ptr<PythonProgressCallback> global_callback_;
    
public:
    static void set_callback(py::object callback) {
        if (callback.is_none()) {
            global_callback_.reset();
        } else {
            global_callback_.reset(new PyProgressCallback(callback));
        }
    }
    
    static PythonProgressCallback* get_callback() {
        return global_callback_.get();
    }
    
    static bool has_callback() {
        return global_callback_ != nullptr;
    }
};

// 静态成员定义
std::unique_ptr<PythonProgressCallback> ProgressCallbackManager::global_callback_ = nullptr;

#endif // PYTHON_BINDING

#endif // PYTHON_PROGRESS_WRAPPER_H