130 lines
4.1 KiB
C++
130 lines
4.1 KiB
C++
#include "dialogalgoarg.h"
|
|
#include "ui_dialogalgoarg.h"
|
|
|
|
#include <QCheckBox>
|
|
#include <QDoubleValidator>
|
|
#include <QFormLayout>
|
|
#include <QIntValidator>
|
|
#include <QLabel>
|
|
|
|
DialogAlgoArg::DialogAlgoArg(QWidget* parent)
|
|
: QDialog(parent)
|
|
, ui(new Ui::DialogAlgoArg)
|
|
{
|
|
ui->setupUi(this);
|
|
setWindowTitle(QStringLiteral("检测参数配置"));
|
|
|
|
// 设置输入验证器（只限制输入格式，不做上下选择）
|
|
ui->edit_score->setValidator(new QDoubleValidator(0.0, 1.0, 2, this));
|
|
ui->edit_nms->setValidator(new QDoubleValidator(0.0, 1.0, 2, this));
|
|
ui->edit_width->setValidator(new QIntValidator(0, 8192, this));
|
|
ui->edit_height->setValidator(new QIntValidator(0, 8192, this));
|
|
ui->edit_modelType->setValidator(new QIntValidator(0, 10, this));
|
|
createDisplayRows();
|
|
}
|
|
|
|
DialogAlgoArg::~DialogAlgoArg()
|
|
{
|
|
delete ui;
|
|
}
|
|
|
|
void DialogAlgoArg::SetParams(const DroneScrewAlgoUiParams& params)
|
|
{
|
|
m_params = params;
|
|
loadToUi(params);
|
|
}
|
|
|
|
DroneScrewAlgoUiParams DialogAlgoArg::GetParams() const
|
|
{
|
|
return m_params;
|
|
}
|
|
|
|
void DialogAlgoArg::SetDisplayOptions(const DroneScrewDisplayOption& options)
|
|
{
|
|
m_displayOptions = options;
|
|
loadDisplayToUi(options);
|
|
}
|
|
|
|
DroneScrewDisplayOption DialogAlgoArg::GetDisplayOptions() const
|
|
{
|
|
return m_displayOptions;
|
|
}
|
|
|
|
void DialogAlgoArg::loadToUi(const DroneScrewAlgoUiParams& p)
|
|
{
|
|
if (ui->edit_score) ui->edit_score->setText(QString::number(p.scoreThreshold, 'f', 2));
|
|
if (ui->edit_nms) ui->edit_nms->setText(QString::number(p.nmsThreshold, 'f', 2));
|
|
if (ui->edit_width) ui->edit_width->setText(QString::number(p.inputWidth));
|
|
if (ui->edit_height) ui->edit_height->setText(QString::number(p.inputHeight));
|
|
if (ui->edit_modelType) ui->edit_modelType->setText(QString::number(p.modelType));
|
|
if (ui->edit_model) ui->edit_model->setText(QString::fromStdString(p.modelPath));
|
|
}
|
|
|
|
DroneScrewAlgoUiParams DialogAlgoArg::collectFromUi() const
|
|
{
|
|
DroneScrewAlgoUiParams p;
|
|
if (ui->edit_score) { bool ok; float v = ui->edit_score->text().toFloat(&ok); if (ok) p.scoreThreshold = v; }
|
|
if (ui->edit_nms) { bool ok; float v = ui->edit_nms->text().toFloat(&ok); if (ok) p.nmsThreshold = v; }
|
|
if (ui->edit_width) { bool ok; int v = ui->edit_width->text().toInt(&ok); if (ok) p.inputWidth = v; }
|
|
if (ui->edit_height) { bool ok; int v = ui->edit_height->text().toInt(&ok); if (ok) p.inputHeight = v; }
|
|
if (ui->edit_modelType) { bool ok; int v = ui->edit_modelType->text().toInt(&ok); if (ok) p.modelType = v; }
|
|
if (ui->edit_model) p.modelPath = ui->edit_model->text().toStdString();
|
|
return p;
|
|
}
|
|
|
|
void DialogAlgoArg::createDisplayRows()
|
|
{
|
|
if (!ui->fl_main)
|
|
return;
|
|
|
|
m_checkDrawBoxes = new QCheckBox(QStringLiteral("显示"), this);
|
|
m_checkDrawBoxes->setObjectName(QStringLiteral("check_drawBoxes"));
|
|
m_checkDrawBoxes->setMinimumHeight(40);
|
|
m_checkDrawBoxes->setStyleSheet(QStringLiteral(
|
|
"QCheckBox { color: rgb(239, 241, 245); font-size: 18px; background: none; }"));
|
|
|
|
ui->fl_main->addRow(new QLabel(QStringLiteral("显示结果框"), this),
|
|
m_checkDrawBoxes);
|
|
}
|
|
|
|
void DialogAlgoArg::loadDisplayToUi(const DroneScrewDisplayOption& options)
|
|
{
|
|
if (m_checkDrawBoxes)
|
|
m_checkDrawBoxes->setChecked(options.drawBoxes);
|
|
}
|
|
|
|
DroneScrewDisplayOption DialogAlgoArg::collectDisplayFromUi() const
|
|
{
|
|
DroneScrewDisplayOption options = m_displayOptions;
|
|
if (m_checkDrawBoxes)
|
|
options.drawBoxes = m_checkDrawBoxes->isChecked();
|
|
return options;
|
|
}
|
|
|
|
void DialogAlgoArg::on_btn_ok_clicked()
|
|
{
|
|
m_params = collectFromUi();
|
|
m_displayOptions = collectDisplayFromUi();
|
|
accept();
|
|
}
|
|
|
|
void DialogAlgoArg::on_btn_cancel_clicked()
|
|
{
|
|
reject();
|
|
}
|
|
|
|
void DialogAlgoArg::on_btn_apply_clicked()
|
|
{
|
|
m_params = collectFromUi();
|
|
m_displayOptions = collectDisplayFromUi();
|
|
}
|
|
|
|
void DialogAlgoArg::on_btn_reset_clicked()
|
|
{
|
|
DroneScrewAlgoUiParams p; // 默认值
|
|
loadToUi(p);
|
|
DroneScrewDisplayOption options = m_displayOptions;
|
|
options.drawBoxes = true;
|
|
loadDisplayToUi(options);
|
|
}
|