GrabBag/App/RodAndBarPosition/RodAndBarPositionApp/dialogalgoarg.cpp

#include "dialogalgoarg.h"
#include "ui_dialogalgoarg.h"
#include "RodAndBarPositionPresenter.h"
#include "StyledMessageBox.h"
#include "HandEyeCalibWidget.h"
#include "NetworkConfigWidget.h"
#include "PathManager.h"
#include "VrLog.h"
#include <cstring>

DialogAlgoArg::DialogAlgoArg(QWidget *parent)
    : QDialog(parent)
    , ui(new Ui::DialogAlgoArg)
    , m_presenter(nullptr)
    , m_handEyeCalibWidget(nullptr)
    , m_networkConfigWidget(nullptr)
{
    ui->setupUi(this);
    setWindowTitle("算法参数设置");
}

DialogAlgoArg::~DialogAlgoArg()
{
    delete ui;
}

void DialogAlgoArg::SetPresenter(RodAndBarPositionPresenter* presenter)
{
    m_presenter = presenter;
    loadParams();

    // 初始化手眼标定 tab
    InitHandEyeCalibTab();

    // 初始化网络配置 tab
    InitNetworkConfigTab();
}

void DialogAlgoArg::loadParams()
{
    if (!m_presenter) {
        return;
    }

    auto params = m_presenter->GetAlgoParams();

    // 棒材参数
    ui->spinRodDiameter->setValue(params.rodParam.rodDiameter);
    ui->spinRodLen->setValue(params.rodParam.rodLen);

    // 角点检测参数
    ui->spinCornerTh->setValue(params.cornerParam.cornerTh);
    ui->spinScale->setValue(params.cornerParam.scale);
    ui->spinMinEndingGap->setValue(params.cornerParam.minEndingGap);
    ui->spinMinEndingGapZ->setValue(params.cornerParam.minEndingGap_z);
    ui->spinJumpCornerTh1->setValue(params.cornerParam.jumpCornerTh_1);
    ui->spinJumpCornerTh2->setValue(params.cornerParam.jumpCornerTh_2);

    // 滤波参数
    ui->spinContinuityTh->setValue(params.filterParam.continuityTh);
    ui->spinOutlierTh->setValue(params.filterParam.outlierTh);

    // 树生长参数
    ui->spinMaxLineSkipNum->setValue(params.growParam.maxLineSkipNum);
    ui->spinYDeviationMax->setValue(params.growParam.yDeviation_max);
    ui->spinMaxSkipDistance->setValue(params.growParam.maxSkipDistance);
    ui->spinZDeviationMax->setValue(params.growParam.zDeviation_max);
    ui->spinMinLTypeTreeLen->setValue(params.growParam.minLTypeTreeLen);
    ui->spinMinVTypeTreeLen->setValue(params.growParam.minVTypeTreeLen);
}

void DialogAlgoArg::saveParams()
{
    if (!m_presenter) {
        return;
    }

    auto params = m_presenter->GetAlgoParams();

    // 棒材参数
    params.rodParam.rodDiameter = ui->spinRodDiameter->value();
    params.rodParam.rodLen = ui->spinRodLen->value();

    // 角点检测参数
    params.cornerParam.cornerTh = ui->spinCornerTh->value();
    params.cornerParam.scale = ui->spinScale->value();
    params.cornerParam.minEndingGap = ui->spinMinEndingGap->value();
    params.cornerParam.minEndingGap_z = ui->spinMinEndingGapZ->value();
    params.cornerParam.jumpCornerTh_1 = ui->spinJumpCornerTh1->value();
    params.cornerParam.jumpCornerTh_2 = ui->spinJumpCornerTh2->value();

    // 滤波参数
    params.filterParam.continuityTh = ui->spinContinuityTh->value();
    params.filterParam.outlierTh = ui->spinOutlierTh->value();

    // 树生长参数
    params.growParam.maxLineSkipNum = ui->spinMaxLineSkipNum->value();
    params.growParam.yDeviation_max = ui->spinYDeviationMax->value();
    params.growParam.maxSkipDistance = ui->spinMaxSkipDistance->value();
    params.growParam.zDeviation_max = ui->spinZDeviationMax->value();
    params.growParam.minLTypeTreeLen = ui->spinMinLTypeTreeLen->value();
    params.growParam.minVTypeTreeLen = ui->spinMinVTypeTreeLen->value();

    m_presenter->SetAlgoParams(params);

    // 保存网络配置
    saveNetworkConfig();
}

void DialogAlgoArg::resetParams()
{
    ui->spinRodDiameter->setValue(52.0);
    ui->spinRodLen->setValue(290.0);

    ui->spinCornerTh->setValue(60.0);
    ui->spinScale->setValue(13.0);
    ui->spinMinEndingGap->setValue(20.0);
    ui->spinMinEndingGapZ->setValue(5.0);
    ui->spinJumpCornerTh1->setValue(15.0);
    ui->spinJumpCornerTh2->setValue(60.0);

    ui->spinContinuityTh->setValue(5.0);
    ui->spinOutlierTh->setValue(5.0);

    ui->spinMaxLineSkipNum->setValue(5);
    ui->spinYDeviationMax->setValue(10.0);
    ui->spinMaxSkipDistance->setValue(10.0);
    ui->spinZDeviationMax->setValue(10.0);
    ui->spinMinLTypeTreeLen->setValue(100.0);
    ui->spinMinVTypeTreeLen->setValue(100.0);
}

void DialogAlgoArg::on_btnOK_clicked()
{
    saveParams();
    accept();
}

void DialogAlgoArg::on_btnCancel_clicked()
{
    reject();
}

void DialogAlgoArg::on_btnApply_clicked()
{
    saveParams();
    StyledMessageBox::information(this, "提示", "参数已应用");
}

void DialogAlgoArg::on_btnReset_clicked()
{
    auto ret = StyledMessageBox::question(this, "确认重置",
        "确定要重置为默认参数吗？");
    if (ret == QMessageBox::Yes) {
        resetParams();
    }
}

// ========== 手眼标定相关实现 ==========

void DialogAlgoArg::InitHandEyeCalibTab()
{
    if (!ui || !m_presenter) return;

    m_handEyeCalibWidget = new HandEyeCalibWidget(this);
    m_handEyeCalibWidget->setMatrixEditable(true);
    ui->verticalLayout_handEyeCalibHost->addWidget(m_handEyeCalibWidget);

    m_handEyeCalibWidget->setDefaultFilePath(
        PathManager::GetInstance().GetAppConfigDirectory());

    const auto& cameraList = m_presenter->GetCameraList();

    QVector<HandEyeCalibCameraInfo> calibCameraList;
    if (cameraList.empty()) {
        HandEyeCalibCameraInfo defaultCam;
        defaultCam.cameraIndex = 1;
        defaultCam.displayName = QString::fromUtf8("相机 1");
        calibCameraList.append(defaultCam);
    } else {
        for (size_t i = 0; i < cameraList.size(); ++i) {
            HandEyeCalibCameraInfo info;
            info.cameraIndex = static_cast<int>(i + 1);
            info.displayName = QString::fromStdString(cameraList[i].first);
            calibCameraList.append(info);
        }
    }
    m_handEyeCalibWidget->setCameraList(calibCameraList);

    for (const auto& camInfo : calibCameraList) {
        CalibMatrix calibMatrix = m_presenter->GetClibMatrix(camInfo.cameraIndex - 1);
        bool hasCalibData = false;
        for (int i = 0; i < 16; ++i) {
            double identity = (i / 4 == i % 4) ? 1.0 : 0.0;
            if (qAbs(calibMatrix.clibMatrix[i] - identity) > 1e-9) {
                hasCalibData = true;
                break;
            }
        }
        m_handEyeCalibWidget->setCalibData(camInfo.cameraIndex, calibMatrix.clibMatrix, hasCalibData);
    }

    connect(m_handEyeCalibWidget, &HandEyeCalibWidget::calibMatrixLoaded,
            this, &DialogAlgoArg::onCalibMatrixLoaded);
    connect(m_handEyeCalibWidget, &HandEyeCalibWidget::saveCalibRequested,
            this, &DialogAlgoArg::onSaveCalibRequested);
}

void DialogAlgoArg::onCalibMatrixLoaded(int cameraIndex, const double* matrix)
{
    Q_UNUSED(cameraIndex);
    Q_UNUSED(matrix);
    StyledMessageBox::information(this, "提示", "已从文件导入标定矩阵");
}

void DialogAlgoArg::onSaveCalibRequested(int cameraIndex, const double* matrix)
{
    Q_UNUSED(cameraIndex);
    Q_UNUSED(matrix);

    saveParams();
    StyledMessageBox::information(this, "成功", "手眼标定参数已保存！");
}

// ========== 网络配置相关实现 ==========

void DialogAlgoArg::InitNetworkConfigTab()
{
    if (!ui || !m_presenter) return;

    // 创建网络配置控件：不显示PLC配置，不显示TCP配置
    m_networkConfigWidget = new NetworkConfigWidget(false, false, this);
    ui->verticalLayout_networkConfigHost->addWidget(m_networkConfigWidget);

    // 加载当前配置
    loadNetworkConfig();
}

void DialogAlgoArg::loadNetworkConfig()
{
    if (!m_presenter || !m_networkConfigWidget) return;

    ConfigManager* configManager = m_presenter->GetConfigManager();
    if (!configManager) return;

    ConfigResult configResult = configManager->GetConfigResult();

    NetworkConfigData netConfig;
    netConfig.eulerOrder = configResult.eulerOrder;
    netConfig.dirVectorInvert = configResult.dirVectorInvert;
    netConfig.byteOrder = configResult.byteOrder;

    m_networkConfigWidget->setConfig(netConfig);
}

void DialogAlgoArg::saveNetworkConfig()
{
    if (!m_presenter || !m_networkConfigWidget) return;

    ConfigManager* configManager = m_presenter->GetConfigManager();
    if (!configManager) return;

    NetworkConfigData netConfig = m_networkConfigWidget->getConfig();

    // 获取当前完整配置并更新网络参数
    SystemConfig systemConfig = configManager->GetConfig();
    systemConfig.configResult.eulerOrder = netConfig.eulerOrder;
    systemConfig.configResult.dirVectorInvert = netConfig.dirVectorInvert;
    systemConfig.configResult.byteOrder = netConfig.byteOrder;

    // 更新并保存
    configManager->UpdateFullConfig(systemConfig);

    QString configPath = PathManager::GetInstance().GetConfigFilePath();
    configManager->SaveConfigToFile(configPath.toStdString());
}