window.cpp

来自「足球机器人自动程序」· C++ 代码 · 共 80 行

#include <qpainter.h>
#include <qimage.h>

#include "window.h"

#define W 256
#define H 256

inline unsigned char sat(int c) {
	return( (c & ~255)? (c<0)?0:255 : c );
}

void window::fill_yuv() {
	int tu, tv;
	unsigned char *p = pRGB->bits();

	for (int iv=0; iv<H; ++iv) {
		for (int iu=0; iu<W; ++iu) {
			tu = iu - 128;
			tv = iv - 128;
			*p = sat(y + ((454*tu)>>8));	// b
			++p;
			*p = sat(y - ((183*tv + 88*tu)>>8));	// g
			++p;
			*p = sat(y + ((359*tv)>>8));	// r
			p += 2;
		}
	}
}

window::window() {
	setMinimumSize(W, H);
	setMaximumSize(W, H);
	setBackgroundMode(Qt::NoBackground);

	y = 128;
	u = 0;
	v = 0;

	pRGB = new QImage(W, H, 32, 0);
	fill_yuv();
}

void window::paintEvent(QPaintEvent *) {
	QPainter p(this);
	p.drawImage(0, 0, *pRGB);
}
 
void window::mousePressEvent(QMouseEvent *e) {
	u = e->x();
	v = e->y();
	if (u < 0 || u >= W || v < 0 || v >= H) {
		return;
	}
	QString s;
	s.sprintf("%d %d %d", y, u, v);
	setCaption(s);
}

void window::keyPressEvent(QKeyEvent *e) {

	switch (e->ascii()) {
	case '1':
		if (y > 0) --y;
		break;
	case '2':
		if (y < 255) ++y;
		break;
	default:
		return;
	}

	fill_yuv();
	QString s;
	s.sprintf("%d %d %d", y, u, v);
	setCaption(s);
	update();
}