window.cpp

足球机器人自动程序

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

#include "window.h"

#include "global.h"
#include "utils.h"
#include "cvt.h"
#include "net.h"
#include "dec.h"
#include "cmv.h"

extern int use_cmv;
extern struct cmv_data data;
extern struct net ns;

int n = 0;
int i = 0;
char filename[24];

unsigned char raw_buf[RAW_SIZE];
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(ns.sock, QSocketNotifier::Read, this);
	connect(sn, SIGNAL(activated(int)), this, SLOT(get_data()));
}

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

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

void window::get_data() {
	if (net_recv(raw_buf, RAW_SIZE, &ns) != RAW_SIZE) {
		fprintf(stderr, "net_recv error\n");
		return;
	}

	dec_process(raw_buf, yuv_buf);

	if (use_cmv) {
		cmv_process(&data, yuv_buf);
		cvtc(yuv_buf, pRGB->bits(), data.map, data.rgb);
	}
	else cvt(yuv_buf, pRGB->bits());

	fprintf(stderr, "\r%d", ++i);
	update();
}

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

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