40 lines
674 B
C++
40 lines
674 B
C++
#pragma once
|
|
|
|
#include <QtGui/QPixmap>
|
|
|
|
#include "NodeDataModel.hpp"
|
|
#include <opencv2/opencv.hpp>
|
|
|
|
using QtNodes::NodeData;
|
|
using QtNodes::NodeDataType;
|
|
|
|
class CvImageData :public NodeData
|
|
{
|
|
public:
|
|
CvImageData() { mCvImage = cv::Mat(); }
|
|
CvImageData(cv::Mat inImg)
|
|
{
|
|
if (!inImg.empty())
|
|
mCvImage = inImg;
|
|
else
|
|
mCvImage = cv::Mat();
|
|
}
|
|
virtual ~CvImageData() {}
|
|
|
|
NodeDataType type() const override { return { "CvImage","CvImg" }; }
|
|
|
|
bool empty() { return mCvImage.empty(); }
|
|
|
|
cv::Mat* CvImage() { return &mCvImage; }
|
|
|
|
void setCvImage(cv::Mat const& _img)
|
|
{
|
|
if (!_img.empty())
|
|
mCvImage = _img;
|
|
else
|
|
return;
|
|
}
|
|
private:
|
|
cv::Mat mCvImage;
|
|
|
|
}; |