84 lines
1.9 KiB
C++
84 lines
1.9 KiB
C++
#pragma once
|
|
|
|
#include <iostream>
|
|
#include <QObject>
|
|
#include <QWidget>
|
|
#include <QSlider>
|
|
#include "DataModelRegistry.hpp"
|
|
#include "NodeDataModel.hpp"
|
|
#include "halconcpp/HalconCpp.h"
|
|
#include "HImageData.hpp"
|
|
#include "HRegionData.hpp"
|
|
#include "HImageViewWidget.hpp"
|
|
|
|
using QtNodes::PortType;
|
|
using QtNodes::PortIndex;
|
|
using QtNodes::NodeData;
|
|
using QtNodes::NodeDataType;
|
|
using QtNodes::NodeDataModel;
|
|
using QtNodes::NodeValidationState;
|
|
using namespace HalconCpp;
|
|
/**
|
|
* \brief halcon 图像rgb2gray节点
|
|
*/
|
|
class HImageThresholdModel :public NodeDataModel
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
HImageThresholdModel();
|
|
virtual ~HImageThresholdModel() = default;
|
|
|
|
public:
|
|
QString caption() const override
|
|
{
|
|
return QStringLiteral("二值化节点");
|
|
}
|
|
QString name() const override
|
|
{
|
|
return QStringLiteral("二值化节点");
|
|
}
|
|
virtual QString modelName() const
|
|
{
|
|
return QStringLiteral("二值化节点");
|
|
}
|
|
unsigned int
|
|
nPorts(PortType portType) const override;
|
|
|
|
NodeDataType
|
|
dataType(PortType portType, PortIndex portIndex) const override;
|
|
|
|
std::shared_ptr<NodeData>
|
|
outData(PortIndex port) override;
|
|
|
|
void
|
|
setInData(std::shared_ptr<NodeData>, int) override;
|
|
|
|
QWidget*
|
|
embeddedWidget() override { return m_widget; }
|
|
|
|
bool
|
|
resizable() const override { return true; }
|
|
NodeValidationState
|
|
validationState() const override;
|
|
QString
|
|
validationMessage() const override;
|
|
|
|
QJsonObject save() const override;
|
|
|
|
void restore(QJsonObject const&) override;
|
|
|
|
protected:
|
|
bool RunTask();
|
|
bool eventFilter(QObject* watched, QEvent* event) override;
|
|
public:
|
|
NodeValidationState modelValidationState = NodeValidationState::Warning;
|
|
QString modelValidationError = QStringLiteral("图片输入未连接!");
|
|
private:
|
|
std::shared_ptr<HImageData> m_hImage;
|
|
std::shared_ptr<HRegionData> m_domain;
|
|
std::shared_ptr<HRegionData> m_result;
|
|
QSlider* m_minGraySlider;
|
|
QSlider* m_maxGraySlider;
|
|
QWidget* m_widget;
|
|
};
|