#pragma once

#include <QtGui/QPixmap>

#include "NodeDataModel.hpp"

#include <opencv2/opencv.hpp>

using QtNodes::NodeData;
using QtNodes::NodeDataType;

class CvRectData :public NodeData
{
public:
	CvRectData() { mCvRect = cv::Rect(); }
	CvRectData(cv::Rect& _rect)
	{
		if (!_rect.empty())
			mCvRect = _rect;
	}

	NodeDataType type() const override { return { "CvRect","Rect" }; }

	cv::Rect* CvRect() { return &mCvRect; }
	void setCvRect(cv::Rect const& _rect)
	{
		if (_rect.empty())
			return;
		mCvRect = _rect;
	}
	void setSize(QSize const& _size) { mSize = _size; }
	QSize getSize() { return mSize; }

private:
	cv::Rect mCvRect;
	QSize mSize;
};