board.cpp

#include "board.h"
#include "mainwidget.h"
#include <QLabel>
#include <QMouseEvent>
#include <QPainter>
#include <QWidget>
/**
TODO:
	* add support for a board of any size, not 10x10 only
	* maybe add a timer
	* highlight 5 cells in a row
**/
Board::Board(int cb, QWidget* parent) : QWidget(parent), m_padding(3), m_size(20), m_cube(cb), turn(Red), gameover(false)
{
	setPalette(QPalette(QColor(255, 255, 255)));
	setAutoFillBackground(true);
	setFixedSize(psize()*m_cube+m_padding, psize()*m_cube+m_padding);

	updateStatus(NULL);

	for(int x = m_padding; x+m_size < height(); x += psize())
	for(int y = m_padding; y+m_size < width(); y += psize())
		rects << qMakePair<QRect,Owner>(QRect(x, y, m_size, m_size), Nobody);
}

void Board::paintEvent(QPaintEvent*)
{
	if(gameover) return;

	QPainter painter(this);
	painter.setRenderHint(QPainter::Antialiasing);

	for(int i=0; i < rects.size(); ++i)
	{
		painter.setPen(Qt::black);
		//TODO: optimize this (black borders between cells)
		painter.drawRect(QRectF(rects[i].first).adjusted(-1.5, -1.5, 1.5, 1.5));

		// skip them fools without owner
		if(rects[i].second == Nobody)
			continue;

		switch(rects[i].second)
		{
		case Red:
			//qDebug() << "red on" << rects[i].first;
			painter.setPen(Qt::red);
			painter.drawEllipse(rects[i].first);
			break;
		case Blue:
			painter.setPen(Qt::blue);
			painter.drawLine(rects[i].first.x(), rects[i].first.y(),
					 rects[i].first.x()+m_size, rects[i].first.y()+m_size);
			painter.drawLine(rects[i].first.x(), rects[i].first.y()+m_size,
					 rects[i].first.x()+m_size, rects[i].first.y());
			break;
		default:
			break;
		}
	}

	if(!checkGameOver())
		updateStatus(NULL);
}

void Board::mousePressEvent(QMouseEvent* e)
{
	if(gameover) return;

	for(int i=0; i < rects.size(); ++i)
	if(rects[i].first.contains(e->x(), e->y()))
	{
		// skip owned ones
		if(rects[i].second != Nobody)
			continue;

		// set next turn
		rects[i].second = turn;
		if(turn == Red)
			turn = Blue;
		else
			turn = Red;

		// update this rectangle in the paintEvent
		this->update(rects[i].first);

		return;
	}
}

void Board::updateStatus(QString status)
{
	MainWidget* mw = static_cast<MainWidget*>(parentWidget());

	QString l_status = tr("Status:") + " <font color='";
	if(turn == Red)
		l_status += "red'>" + tr("Red");
	else
		l_status += "blue'>" + tr("Blue");
	l_status += "</font>";

	if(status == NULL)
		l_status += tr("'s turn...");
	else
		l_status += status;

	mw->setStatus(l_status);
}

bool Board::checkGameOver()
{
	int i=0 /*index*/, t=-1, r=0, b=0, n=0;

	// check vertically
	for(int x=0; x<cube(); x++, i++)
	{
		if(x%m_cube == 0)
			r=0, b=0;

		if(gameOverSwitch(&rects[i].second, r, b))
			return true;
	}

	// check horizontally
	if(!gameover)
	for(int x=0; x<cube(); x++)
	{
		n = x % m_cube;
		if(n == 0)
			r=0, b=0, t++;

		i = t + n*m_cube; // y*10+x;

		if(gameOverSwitch(&rects[i].second, r, b))
			return true;
	}

	b=0, r=0;

        // check main diagonal
	if(!gameover)
	for(int x = 4; x <= cube()-5*m_cube;)
	{
		if(x>=m_cube)
		{
			x += m_cube;
			i = x-1;
			t--;
		} else
		{
			i = x;
			t = ++x;
		}

		for(int c=t; c>0; c--, i += m_cube-1)
			if(gameOverSwitch(&rects[i].second, r, b))
				return true;
	}

	t=4;

        // check minor diagonal
	if(!gameover)
	for(int x = m_cube-5; x <= cube()-5*m_cube;)
	{
		if(x<0 || x>m_cube-5)
		{
			x += m_cube;
			i = x+1;
			t--;
		} else
		{
			i = x--;
			t++;
		}

		for(int c=t; c>0; c--, i += m_cube+1)
			if(gameOverSwitch(&rects[i].second, r, b))
				return true;
	}

	return false;
}

bool Board::gameOverSwitch(Owner* owner, int& r, int& b)
{
	switch(*owner)
	{
	case Red:
		if(++r == 5)
			gameover = true;
		b=0;
		break;
	case Blue:
		if(++b == 5)
			gameover = true;
		r=0;
		break;
	default:
		b=0, r=0;
		break;
	}

	if(gameover)
	{
		turn = *owner;
		updateStatus(tr(" wins."));
	}

	return gameover;
}

int Board::cube() const
{
	return m_cube*m_cube;
}