GrabBag/App/ScrewPosition/ScrewPositionApp/Presenter/Inc/ScrewPositionPresenter.h

#ifndef SCREWPOSITIONPRESENTER_H
#define SCREWPOSITIONPRESENTER_H

#include <condition_variable>
#include <thread>
#include <map>
#include <mutex>
#include <memory>

#include "BasePresenter.h"
#include "IVrEyeDevice.h"
#include "ConfigManager.h"
#include "TCPServerProtocol.h"
#include "IYScrewPositionStatus.h"
#include "SG_baseDataType.h"
#include "VrConvert.h"
#include "LaserDataLoader.h"
#include "CommonDialogCameraLevel.h"  // 引入通用对话框的接口
#include <QImage>
#include <QPainter>
#include <QColor>
#include <QObject>
#include <QTimer>
#include <memory>

// Forward declarations
class DetectPresenter;

class ScrewPositionPresenter : public BasePresenter, public IVrConfigChangeNotify,
                          public ICameraLevelCalculator, public ICameraLevelResultSaver
{
    Q_OBJECT

public:
    explicit ScrewPositionPresenter(QObject *parent = nullptr);
    ~ScrewPositionPresenter();

    // 初始化
    int InitApp() override;

    // 反初始化
    void DeinitApp();

    // 触发检测
    bool TriggerDetection(int cameraIndex = -1);

    // 加载文件并检测
    int LoadAndDetect(const QString& fileName);

    // 重连相机
    void ReconnectCamera();

    // 获取/设置算法参数
    struct AlgoParams {
        VrScrewParam screwParam;
        VrCornerParam cornerParam;
        VrOutlierFilterParam filterParam;
        VrTreeGrowParam growParam;
    };
    AlgoParams GetAlgoParams() const;
    void SetAlgoParams(const AlgoParams& params);

    // 获取配置管理器
    ConfigManager* GetConfigManager() { return m_pConfigManager; }

    // 手眼标定矩阵管理
    CalibMatrix GetClibMatrix(int index) const;

    // 实现IVrConfigChangeNotify接口
    virtual void OnConfigChanged(const ConfigResult& configResult) override;

    // 发送JSON格式检测结果给客户端
    void SendDetectionResultToClient(const DetectionResult& result);

    // ============ 实现 ICameraLevelCalculator 接口 ============

    /**
     * @brief 计算平面调平参数
     */
    bool CalculatePlaneCalibration(
        const std::vector<std::pair<EVzResultDataType, SVzLaserLineData>>& scanData,
        double planeCalib[9],
        double& planeHeight,
        double invRMatrix[9]) override;

    // ============ 实现 ICameraLevelResultSaver 接口 ============

    /**
     * @brief 保存相机调平结果到配置文件
     */
    bool SaveLevelingResults(double planeCalib[9], double planeHeight, double invRMatrix[9],
                            int cameraIndex, const QString& cameraName) override;

    /**
     * @brief 从配置文件加载相机调平结果
     */
    bool LoadLevelingResults(int cameraIndex, const QString& cameraName,
                            double planeCalib[9], double& planeHeight, double invRMatrix[9]) override;

protected:
    // ============ 实现 BasePresenter 纯虚函数 ============

    /**
     * @brief 初始化算法参数（实现纯虚函数）
     */
    int InitAlgoParams() override;

    /**
     * @brief 执行算法检测（实现纯虚函数）
     * @param detectionDataCache 检测数据缓存的引用
     */
    int ProcessAlgoDetection(std::vector<std::pair<EVzResultDataType, SVzLaserLineData>>& detectionDataCache) override;

    /**
     * @brief 获取检测数据类型（实现纯虚函数）
     * 螺杆定位项目只使用Position点云数据
     */
    EVzResultDataType GetDetectionDataType() override {
        return keResultDataType_Position;
    }

    /**
     * @brief 相机状态变化通知（实现纯虚函数）
     */
    void OnCameraStatusChanged(int cameraIndex, bool isConnected) override;

    /**
     * @brief 工作状态变化通知（重写虚函数）
     */
    void OnWorkStatusChanged(WorkStatus status) override;

    /**
     * @brief 相机数量变化通知（重写虚函数）
     */
    void OnCameraCountChanged(int count) override;

    /**
     * @brief 状态文字更新通知（重写虚函数）
     */
    void OnStatusUpdate(const std::string& statusMessage) override;

private:
    // TCP服务器相关方法
    int InitTcpServer(int nPort);
    int InitTCPServer(); // 初始化TCP服务器协议（使用配置端口）
    bool startServer(quint16 port = 0); // port = 0 表示使用配置文件中的端口
    void stopServer();

    // TCP服务器回调处理方法
    void onTcpDataReceivedFromCallback(const TCPClient* pClient, const char* pData, const unsigned int nLen);
    void onTcpClientEventFromCallback(const TCPClient* pClient, TCPServerEventType eventType);

    // TCP连接状态改变回调
    void OnTCPConnectionChanged(bool connected);

    // TCP检测触发回调
    bool OnTCPDetectionTrigger(bool startWork, int cameraIndex, qint64 timestamp);

    // 发送检测结果给TCP客户端
    void _SendDetectionResultToTCP(const DetectionResult& detectionResult, int cameraIndex);

    // 连接状态检查和更新
    void CheckAndUpdateWorkStatus();

    // 根据相机索引获取调平参数
    SSG_planeCalibPara _GetCameraCalibParam(int cameraIndex);

private:
    // ScrewPositionPresenter 特有的成员变量
    ConfigManager*                  m_pConfigManager = nullptr;

    // TCP服务器相关
    IYTCPServer*                    m_pTcpServer = nullptr;
    TCPServerProtocol*              m_pTCPServer = nullptr;     // TCP服务器协议实例
    quint16                         m_port = 0;
    bool                            m_bTcpClientConnected = false;   // TCP客户端连接状态
    bool                            m_bTCPConnected = false;         // TCP客户端连接状态（新协议）

    // 检测处理器
    DetectPresenter*                m_pDetectPresenter = nullptr;

    // 手眼标定矩阵列表（从独立文件加载，暂时保留在Presenter中）
    std::vector<CalibMatrix>        m_clibMatrixList;
};

#endif // SCREWPOSITIONPRESENTER_H