162 lines
3.3 KiB
C++
162 lines
3.3 KiB
C++
#include "HRegionSelectShapeStdModel.hpp"
|
|
#include <QtCore/QEvent>
|
|
#include "halconcpp/HalconCpp.h"
|
|
|
|
using namespace HalconCpp;
|
|
|
|
HRegionSelectShapeStdModel::HRegionSelectShapeStdModel()
|
|
{
|
|
m_widget = new QWidget();
|
|
m_widget->setAttribute(Qt::WA_NoSystemBackground);
|
|
m_widget->setFixedSize(120, 80);
|
|
m_cur_feature = "rectangle1";
|
|
m_percent = 90;
|
|
combo_feature = new QComboBox(m_widget);
|
|
m_percentEdit = new QLineEdit(m_widget);
|
|
combo_feature->move(0, 10);
|
|
m_percentEdit->move(0, 60);
|
|
combo_feature->addItem("max_area");
|
|
combo_feature->addItem("rectangle1");
|
|
combo_feature->addItem("rectangle2");
|
|
m_percentEdit->setText("90.0");
|
|
m_InRegion = std::make_shared<HRegionData>();
|
|
m_result = std::make_shared<HRegionData>();
|
|
|
|
connect(combo_feature, &QComboBox::currentTextChanged, [=]()
|
|
{
|
|
m_cur_feature = combo_feature->currentText();
|
|
RunTask();
|
|
});
|
|
connect(m_percentEdit, &QLineEdit::textEdited, this, [=]()
|
|
{
|
|
m_percent = m_percentEdit->text().toDouble();
|
|
RunTask();
|
|
});
|
|
}
|
|
|
|
bool HRegionSelectShapeStdModel::RunTask()
|
|
{
|
|
PortIndex const outPortIndex = 0;
|
|
if (m_InRegion->hRegion() == nullptr)
|
|
{
|
|
return false;
|
|
}
|
|
try
|
|
{
|
|
HalconCpp::SelectShapeStd(
|
|
*m_InRegion->hRegion(),
|
|
m_result->hRegion(),
|
|
m_cur_feature.toStdString().c_str(),
|
|
m_percent);
|
|
|
|
m_result->setSize(m_InRegion->getSize());
|
|
|
|
modelValidationState = NodeValidationState::Valid;
|
|
modelValidationError = QString();
|
|
}
|
|
catch (...)
|
|
{
|
|
modelValidationState = NodeValidationState::Warning;
|
|
modelValidationError = QStringLiteral("缺失或运行失败!");
|
|
}
|
|
|
|
Q_EMIT dataUpdated(outPortIndex);
|
|
|
|
return true;
|
|
}
|
|
|
|
bool HRegionSelectShapeStdModel::eventFilter(QObject* watched, QEvent* event)
|
|
{
|
|
|
|
return false;
|
|
}
|
|
|
|
unsigned int HRegionSelectShapeStdModel::
|
|
nPorts(PortType portType) const
|
|
{
|
|
unsigned int result = 1;
|
|
|
|
switch (portType)
|
|
{
|
|
case PortType::In:
|
|
result = 1;
|
|
break;
|
|
|
|
case PortType::Out:
|
|
result = 1;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
NodeValidationState HRegionSelectShapeStdModel::validationState() const
|
|
{
|
|
return modelValidationState;
|
|
}
|
|
|
|
QString HRegionSelectShapeStdModel::validationMessage() const
|
|
{
|
|
return modelValidationError;
|
|
}
|
|
|
|
QJsonObject HRegionSelectShapeStdModel::save() const
|
|
{
|
|
QJsonObject result = NodeDataModel::save();
|
|
result.insert("m_cur_feature", combo_feature->currentText());
|
|
result.insert("m_percent", m_percent);
|
|
return result;
|
|
}
|
|
|
|
void HRegionSelectShapeStdModel::restore(QJsonObject const& p)
|
|
{
|
|
combo_feature->setCurrentText(p["m_cur_feature"].toString("rectangle1"));
|
|
m_percent = p.value("m_percent").toDouble(90);
|
|
}
|
|
|
|
NodeDataType
|
|
HRegionSelectShapeStdModel::dataType(PortType, PortIndex) const
|
|
{
|
|
return HRegionData().type();
|
|
}
|
|
|
|
void HRegionSelectShapeStdModel::
|
|
setInData(std::shared_ptr<NodeData> data, int portIndex)
|
|
{
|
|
auto hRegionData =
|
|
std::dynamic_pointer_cast<HRegionData>(data);
|
|
if (hRegionData == nullptr)
|
|
{
|
|
return;
|
|
}
|
|
switch (portIndex)
|
|
{
|
|
case 0:
|
|
m_InRegion->setHRegion(*hRegionData->hRegion());
|
|
m_InRegion->setSize(hRegionData->getSize());
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
RunTask();
|
|
}
|
|
|
|
std::shared_ptr<NodeData>
|
|
HRegionSelectShapeStdModel::
|
|
outData(PortIndex index)
|
|
{
|
|
switch (index)
|
|
{
|
|
case 0:
|
|
return std::dynamic_pointer_cast<HRegionData>(m_result);
|
|
break;
|
|
case 1:
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
return std::dynamic_pointer_cast<HRegionData>(m_result);
|
|
}
|