#ifndef MAINWINDOW_H
#define MAINWINDOW_H

#include <QMainWindow>
#include <QImage>
#include <QLabel>
#include <QPixmap>
#include <QListWidget>
#include <QListWidgetItem>
#include <QStandardItem>
#include <QStringListModel>
#include <QGraphicsView>
#include <QMenu>
#include <QAction>
#include "CommonDialogCamera.h"
#include "dialogalgoarg.h"
#include "CommonDialogCameraLevel.h"
#include "devstatus.h"
#include "LapWeldPresenter.h"

QT_BEGIN_NAMESPACE
namespace Ui { class MainWindow; }
QT_END_NAMESPACE

class MainWindow : public QMainWindow, public IYLapWeldStatus
{
    Q_OBJECT

public:
    MainWindow(QWidget *parent = nullptr);
    ~MainWindow();

    void updateStatusLog(const QString& message);
    void clearDetectionLog();
    void Init();

    // 实现IYLapWeldStatus接口
    virtual void OnStatusUpdate(const std::string& statusMessage) override;
    virtual void OnDetectionResult(const DetectionResult& result) override;
    virtual void OnCamera1StatusChanged(bool isConnected) override;
    virtual void OnCamera2StatusChanged(bool isConnected) override;
    virtual void OnRobotConnectionChanged(bool isConnected) override;
    virtual void OnSerialConnectionChanged(bool isConnected) override;
    virtual void OnCameraCountChanged(int cameraCount) override;
    virtual void OnWorkStatusChanged(WorkStatus status) override;

signals:
    // 工作状态更新信号
    void workStatusUpdateRequested(WorkStatus status);
    
    // 检测结果更新信号
    void detectionResultUpdateRequested(const DetectionResult& result);
    
    // 日志更新信号
    void logUpdateRequested(const QString& message);
    
    // 清空日志信号
    void logClearRequested();

private slots:

    // 工作状态更新槽函数
    void updateWorkStatusLabel(WorkStatus status);
    
    // 检测结果更新槽函数
    void updateDetectionResultDisplay(const DetectionResult& result);
    
    // 日志更新槽函数
    void updateDetectionLog(const QString& message);
    
    // 清空日志槽函数
    void clearDetectionLogUI();
    
    // 处理相机点击事件
    void onCameraClicked(int cameraIndex);

    // UI操作相关槽
    void on_btn_start_clicked();
    void on_btn_stop_clicked();

    void on_btn_camera_clicked();

    void on_btn_algo_config_clicked();

    void on_btn_camera_levelling_clicked();

    void on_btn_hide_clicked();

    void on_btn_close_clicked();

    void on_btn_test_clicked();

    // 右键菜单相关槽函数
    void onSaveDetectionData();


private:
    Ui::MainWindow *        ui;
    CommonDialogCamera*           ui_dialogCamera = nullptr;
    DialogAlgoarg*          ui_dialogConfig = nullptr;
    CommonDialogCameraLevel*      ui_dialogCameraLevel = nullptr;

    // 业务逻辑处理类
    LapWeldPresenter*       m_presenter = nullptr;
    
    // 日志模型
    QStringListModel*       m_logModel = nullptr;
    
    // 设备状态显示widget的引用
    devstatus* m_deviceStatusWidget = nullptr;
    
    // 当前选中的按钮（用于互斥控制）
    QPushButton* m_selectedButton = nullptr;
    
    // 右键菜单相关
    QMenu* m_contextMenu = nullptr;
    QAction* m_saveDataAction = nullptr;
    
    void displayImage(const QImage& image);
    void displayImageInSecondView(const QImage& image);
    
    // 扩展版本的检测结果添加函数
    void addDetectionResult(const DetectionResult& result);
    
    // 设置按钮启用/禁用状态
    void setButtonsEnabled(bool enabled);
    
    // 按钮图像切换辅助函数
    void setButtonImage(QPushButton* button, const QString& imagePath);
    void setButtonSelectedState(QPushButton* button, bool selected);
    void restoreAllButtonStates();
    void setOtherButtonsEnabled(QPushButton* exceptButton, bool enabled);
    
    // 页面布局调整函数
    void adjustLayoutForCameraCount(int cameraCount);
    void setupSingleCameraLayout();
    void setupMultiCameraLayout();
    
    // 右键菜单相关函数
    void setupContextMenu();
    void showContextMenu(const QPoint& pos, QGraphicsView* view);
    bool saveDetectionDataToFile(const QString& filePath, int cameraIndex);
};
#endif // MAINWINDOW_H