window.cpp

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

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

#include "window.h"
#include "utils.h"
#include "cvt.h"
#include "cam.h"
#include "dec.h"
#include "cmv.h"

extern struct cam c;
extern int skip;

int n = 0;
int i = 0;
int j = 0;
char filename[8];
unsigned char yuv_buf[YUV_SIZE];

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

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

	QSocketNotifier *sn = new QSocketNotifier(c.fd, QSocketNotifier::Read, this);
	connect(sn, SIGNAL(activated(int)), this, SLOT(get_data()));
	cam_capture(&c);
}

window::~window() {
	printf("\n");
}

void window::paintEvent(QPaintEvent *) {
	QPainter p(this);
	p.setPen(QColor(0, 255, 0));
	p.drawImage(0, 0, *pRGB);

	if (clist[0] > 0) {
		struct reg_t *reg = cmax[0];
		p.drawRect(reg->x1 << 1, reg->y1 << 1,
			(reg->x2 - reg->x1) << 1,
			(reg->y2 - reg->y1 + 1) << 1 );
	}
}

void window::get_data() {
	cam_sync(&c);
	cam_capture(&c);
	
	if (++j < skip) {
		return;
	}
	j = 0;

	dec_process(c.src, yuv_buf);
	cmv_process(yuv_buf, 0);
	cvtm(yuv_buf, pRGB->bits(), map);
	fprintf(stderr, "\r%d", ++i);
	update();
}

void window::mousePressEvent(QMouseEvent *e) {
	if (i <= 0 || e->button() != RightButton) {
		return;
	}

	sprintf(filename, "%d.raw", ++n);
	msave(filename, c.src, RAW_SIZE);
	setCaption(filename);
}