window.cpp

足球机器人自动程序

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

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

extern struct cam c;
extern int skip;

int n = 0;
int i = 0;
int j = 0;
char filename[16];
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.drawImage(0, 0, *pRGB);
}

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

	dec_process(c.src, yuv_buf);
	cvt(yuv_buf, pRGB->bits());
	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);
}