#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;

};