#include "HImageViewWidget.hpp"

HImageViewWidget::HImageViewWidget(QWidget* parent)

{
	if (parent != Q_NULLPTR)
	{
		this->setParent(parent);
	}
	this->setStyleSheet("background-color:black;");
	cur_pixmap = new QPixmap();
}

void HImageViewWidget::showImage(HImage const& _himg)
{
	if (!_himg.IsInitialized())
	{
		return;
	}
	Hlong width;
	Hlong height;
	double zoom_ratio = 1.0;
	_himg.GetImageSize(&width, &height);
	if (width > this->width())
	{
		zoom_ratio = 1.0 * this->width() / width;
	}
	cur_image = _himg.ZoomImageSize(width * zoom_ratio, height * zoom_ratio, "bilinear");
	HImageToQPixmap(cur_image, *cur_pixmap);
	this->update();
}

void HImageViewWidget::HImageToQPixmap(HImage const& _img, QPixmap& tar_pixmap)
{
	Hlong w, h; HString hType;
	Hlong width;
	Hlong height;
	QImage tar_img;

	_img.GetImageSize(&width, &height);
	HTuple type = _img.GetImageType();
	//获取HImage的通道数
	HTuple hChannels = _img.CountChannels();
	if (strcmp(type[0].S(), "byte")) // 如果不是 byte 类型，则失败
	{
		return;
	}
	QImage::Format format;
	switch (hChannels[0].I())
	{
	case 1:
		format = QImage::Format_Grayscale8;
		break;
	case 3:
		format = QImage::Format_RGB32;
		break;
	default:
		return;
	}
	if (tar_img.width() != width || tar_img.height() != height || tar_img.format() != format)
	{
		tar_img = QImage(static_cast<int>(width),
			static_cast<int>(height),
			format);
	}
	if (hChannels == 1)
	{
		//获取HImage的数据指针
		uchar* pBuf = (uchar*)_img.GetImagePointer1(&hType, &w, &h);

		//创建QImage图片
		tar_img = QImage(w, h, QImage::Format_Indexed8);

		//memcpy
		for (int i = 0; i < h; i++, pBuf += w)
		{
			uchar* pDest = tar_img.scanLine(i);
			memcpy(pDest, pBuf, w);
		}

		//tar_pixmap = QPixmap::fromImage(tar_img);
	}
	else if (hChannels == 3)
	{
		uchar* R, * G, * B;
		_img.GetImagePointer3(reinterpret_cast<void**>(&R),
			reinterpret_cast<void**>(&G),
			reinterpret_cast<void**>(&B), &hType, &width, &height);

		for (int row = 0; row < height; row++)
		{
			QRgb* line = reinterpret_cast<QRgb*>(tar_img.scanLine(row));
			for (int col = 0; col < width; col++)
			{
				line[col] = qRgb(*R++, *G++, *B++);
			}
		}
	}
	tar_pixmap = QPixmap::fromImage(tar_img);
}

void HImageViewWidget::QPixmapToHRegion(QPixmap const& _pix, HRegion& tar_reg)
{
	HImage tmpImag;
	QImage tmpQImag = _pix.toImage();
	bool trans = HImageViewWidget::QImage2HImage(tmpQImag, tmpImag);
	if (trans)
	{
		tar_reg = tmpImag.Threshold(100, 255);
	}
	tmpImag.Clear();

}

/**
 * @brief QImage2HImage 将 Qt QImage 转换为 Halcon 的 HImage
 * @param from 输入的 QImage
 * @param to 输出的 HImage ，from 和 to 不共享内存数据。 每次都会为 to 重新分配内存。
 * @return true 表示转换成功，false 表示转换失败。
 */
bool HImageViewWidget::QImage2HImage(QImage& from, HalconCpp::HImage& to)
{
	if (from.isNull()) return false;

	int width = from.width(), height = from.height();
	QImage::Format format = from.format();

	if (format == QImage::Format_RGB32 ||
		format == QImage::Format_ARGB32 ||
		format == QImage::Format_ARGB32_Premultiplied)
	{
		to.GenImageInterleaved(from.bits(), "rgbx", width, height, 0, "byte", width, height, 0, 0, 8, 0);
		return true;
	}
	else if (format == QImage::Format_RGB888)
	{
		to.GenImageInterleaved(from.bits(), "rgb", width, height, 0, "byte", width, height, 0, 0, 8, 0);
		return true;
	}
	else if (format == QImage::Format_Grayscale8 || format == QImage::Format_Indexed8)
	{
		to.GenImage1("byte", width, height, from.bits());
		return true;
	}
	return false;
}


void HImageViewWidget::paintEvent(QPaintEvent* event)
{
	//QPainter painter(this);
	if (!cur_pixmap->isNull())
	{
		this->setPixmap(cur_pixmap->scaled(this->width(), this->height(), Qt::KeepAspectRatio));
	}
	QLabel::paintEvent(event);
}