#include "VierGewinntModel.h"

VierGewinntModel::VierGewinntModel(int r, int c)
{
	rows = r;
	cols = c;

	spielfeld = new char[rows*cols];

	initGame();
}

VierGewinntModel::~VierGewinntModel(void)
{
	delete [] (spielfeld);
}

bool VierGewinntModel::insertCoin(int col, char c)
{
	bool rc = false;

	if ( col < 0 || col > cols-1)
		return rc;

	for (int i=rows-1; i>=0; i--)
	{
		if ( *(spielfeld + (rows*col + i)) == EMPTY )
		{
			*(spielfeld+ (rows*col + i)) = c;
			rc = true;
			break;
		}
	}

	return rc;
}

char VierGewinntModel::getPosition(int row, int col)
{
	if ( col < 0 || col > cols-1)
		return NULL;

	if ( row < 0 || row > rows-1)
		return NULL;

	return *(spielfeld + (rows*col + row));
}

bool VierGewinntModel::checkRowWin(int p)
{
	bool rc = false;
	char tmp;
	if ( p == 1 )
		tmp = COIN1;
	else
		tmp = COIN2;

	for (int row=0; row<rows; row++)
	{
		for (int col=0; col<cols-3; col++)
		{
			if ( *(spielfeld + (rows*col + row)) == tmp && *(spielfeld + (rows*(col+1) + row)) == tmp && *(spielfeld + (rows*(col+2) + row)) == tmp && *(spielfeld + (rows*(col+3) + row)) == tmp )
				rc = true;
		}
	}

	return rc;
}

bool VierGewinntModel::checkColWin(int p)
{
	bool rc = false;
	char tmp;
	if ( p == 1 )
		tmp = COIN1;
	else
		tmp = COIN2;

	for (int col=0; col<cols; col++)
	{
		for (int row=0; row<rows-3; row++)
		{
			if ( *(spielfeld + (rows*col + row)) == tmp && *(spielfeld + (rows*col + row+1)) == tmp && *(spielfeld + (rows*col + row+2)) == tmp && *(spielfeld + (rows*col + row+3)) == tmp )
				rc = true;
		}
	}

	return rc;
}

//Diagonale von links nach rechts fallend
bool VierGewinntModel::checkDia1Win(int p)
{
	bool rc = false;
	char tmp;
	if ( p == 1 )
		tmp = COIN1;
	else
		tmp = COIN2;

	for (int col=0; col<cols; col++)
	{
		for (int row=0; row<rows-3; row++)
		{
			if ( *(spielfeld + (rows*col + row)) == tmp && *(spielfeld + (rows*(col+1) + row+1)) == tmp && *(spielfeld + (rows*(col+2) + row+2)) == tmp && *(spielfeld + (rows*(col+3) + row+3)) == tmp )
				rc = true;
		}
	}

	return rc;
}

//Diagonale von links nach rechts steigend
bool VierGewinntModel::checkDia2Win(int p)
{
	bool rc = false;
	char tmp;
	if ( p == 1 )
		tmp = COIN1;
	else
		tmp = COIN2;

	for (int col=0; col<cols; col++)
	{
		for (int row=3; row<rows; row++)
		{
			if ( *(spielfeld + (rows*col + row)) == tmp && *(spielfeld + (rows*(col+1) + row-1)) == tmp && *(spielfeld + (rows*(col+2) + row-2)) == tmp && *(spielfeld + (rows*(col+3) + row-3)) == tmp )
				rc = true;
		}
	}

	return rc;
}

void VierGewinntModel::initGame(void)
{
	for (int row=0; row<rows; row++)
	{
		for (int col=0; col<cols; col++)
		{
			*(spielfeld + (rows*col + row))=EMPTY;
		}
	}
}