#include <stdio.h>
#include <assert.h>

#include "main.h"
#include "list.h"
#include "arenaFile.h"

#include "interface.h"
#include "screen.h"
#include "image.h"

#ifndef NO_SOUND
#include "music.h"
#endif

#include "screen_mainMenu.h"
#include "screen_choiceArena.h"

#include "widget.h"
#include "widget_label.h"
#include "widget_button.h"
#include "widget_image.h"
#include "widget_buttonimage.h"

static widget_t *image_backgorund;

static widget_t *button_play;
static widget_t *button_back;

static list_t *listWidgetButtonimage;
static arenaFile_t* currentArena;

void startScreenChoiceArena()
{
#ifndef NO_SOUND
	playMusic("menu", MUSIC_GROUP_BASE);
#endif
}

void drawScreenChoiceArena()
{
	int i;

	drawWidgetImage(image_backgorund);

	drawWidgetButton(button_play);
	drawWidgetButton(button_back);

	for( i = 0 ; i < listWidgetButtonimage->count ; i++ )
	{
		widget_t *this;
		this = (widget_t *) listWidgetButtonimage->list[i];
		drawWidgetButtonimage(this);
	}
}

void eventScreenChoiceArena()
{
	int i;

	eventWidgetButton(button_play);
	eventWidgetButton(button_back);

	for( i = 0 ; i < listWidgetButtonimage->count ; i++ )
	{
		widget_t *this;
		this = (widget_t *) listWidgetButtonimage->list[i];
		eventWidgetButtonimage(this);
	}
}

void stopScreenChoiceArena()
{
}

static void eventWidgetButtonImage(void *p)
{
	widget_t *buttonimage;
	int i;

	buttonimage = (widget_t *)(p);

	for( i = 0 ; i < listWidgetButtonimage->count ; i++ )
	{
		widget_t *this;

		this = (widget_t *)(listWidgetButtonimage->list[i]);
		
		if( buttonimage == this )
		{
			setWidgetButtonimageActive(this, TRUE);
			currentArena = getArenaFile(i);
		}
		else
		{
			setWidgetButtonimageActive(this, FALSE);
		}
	}
}

arenaFile_t* getChoiceArena()
{
	return currentArena;
}

void setChoiceArena(arenaFile_t *arenaFile)
{
	widget_t *widget_buttonimage;
	int id;

	id = getArenaFileID(arenaFile);
	currentArena = arenaFile;

	if( id >= 0 )
	{
		widget_buttonimage = (widget_t *)listWidgetButtonimage->list[id];
		setWidgetButtonimageActive(widget_buttonimage, TRUE);
	}
}

static void eventWidget(void *p)
{
	widget_t *button;
	
	button = (widget_t *)(p);

	if( button == button_play )
	{
		setScreen("world");
	}

	if( button == button_back )
	{
		setScreen("gameType");
	}
}

void initScreenChoiceArena()
{
	image_t *image;
	int i;

	image = getImage(IMAGE_GROUP_BASE, "screen_main");
	image_backgorund  = newWidgetImage(0, 0, image);

	button_back = newWidgetButton(_("back"), 100, WINDOW_SIZE_Y-100, eventWidget);
	button_play = newWidgetButton(_("play"), WINDOW_SIZE_X-200, WINDOW_SIZE_Y-100, eventWidget);
	
	listWidgetButtonimage = newList();
	currentArena = NULL;

	for( i = 0 ; i < getArenaCount() ; i++ )
	{
		widget_t *widget_buttonimage;
		arenaFile_t* arenaFile;
		int x, y;

		arenaFile = getArenaFile(i);
		//image = addImageData(getArenaImage(i), IMAGE_NO_ALPHA, "none", IMAGE_GROUP_BASE);
		image = loadImageFromArena(arenaFile,
			getArenaImage(arenaFile),
			IMAGE_GROUP_BASE,
			"none",
			IMAGE_NO_ALPHA);

		x = 100 + 200 * ( i -  3 * ( i/3 ) );
		y = 150 + 200 * ( i/3 );

		widget_buttonimage = newWidgetButtonimage(image,
			x, y, eventWidgetButtonImage);
/*
		if( i == choiceArenaId )
		{
			widget_buttonimage->active = 1;
		}
*/
		addList( listWidgetButtonimage, widget_buttonimage);
	}

	registerScreen( newScreen("chiceArena", startScreenChoiceArena, eventScreenChoiceArena,
		drawScreenChoiceArena, stopScreenChoiceArena) );
}

void quitScreenChoiceArena()
{
	destroyWidgetImage(image_backgorund);

	destroyWidgetButton(button_play);
	destroyWidgetButton(button_back);

	destroyListItem(listWidgetButtonimage, destroyWidgetButtonimage);
}