/*
  Copyright (C) 2008 Bradley Arsenault

  This program is free software; you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 3 of the License, or
  (at your option) any later version.

  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/

#include "CustomGameOtherOptions.h"

#include <Toolkit.h>
#include <StringTable.h>
#include <GUIButton.h>
#include <GUIText.h>
#include <GUINumber.h>
#include <sstream>

CustomGameOtherOptions::CustomGameOtherOptions(GameHeader& gameHeader, MapHeader& mapHeader, bool readOnly)
	:	gameHeader(gameHeader), oldGameHeader(gameHeader), mapHeader(mapHeader)
{
	ok = new TextButton(440, (readOnly ? 420 : 360), 180, 40, ALIGN_SCREEN_CENTERED, ALIGN_SCREEN_CENTERED, "menu", Toolkit::getStringTable()->getString("[ok]"), OK, 13);
	addWidget(ok);
	
	if(!readOnly)
	{
		cancel = new TextButton(440, 420, 180, 40, ALIGN_SCREEN_CENTERED, ALIGN_SCREEN_CENTERED, "menu", Toolkit::getStringTable()->getString("[Cancel]"), CANCEL, 27);
		addWidget(cancel);
	}
	
	title = new Text(0, 18, ALIGN_FILL, ALIGN_SCREEN_CENTERED, "menu", Toolkit::getStringTable()->getString("[Other Options]"));
	addWidget(title);

	playerNames=new Text*[32];
	color=new ColorButton*[32];
	allyTeamNumbers=new MultiTextButton*[32];
	for(int i=0; i<Team::MAX_COUNT; ++i)
	{
		playerNames[i] = NULL;
		color[i] = NULL;
	}
	
	for(int i=0; i<gameHeader.getNumberOfPlayers(); ++i)
	{
		playerNames[i] = new Text(125, 60+25*i, ALIGN_SCREEN_CENTERED, ALIGN_SCREEN_CENTERED, "standard", gameHeader.getBasePlayer(i).name);
		color[i] = new ColorButton(100, 60+25*i, 21, 21, ALIGN_SCREEN_CENTERED, ALIGN_SCREEN_CENTERED, 100+i);
		allyTeamNumbers[i] = new MultiTextButton(250, 60+25*i, 21, 21, ALIGN_SCREEN_CENTERED, ALIGN_SCREEN_CENTERED, "standard", "", 200+i);
		allyTeamNumbers[i]->clearTexts();
		for(int j=0; j<mapHeader.getNumberOfTeams(); ++j)
		{
			std::stringstream s;
			s<<j+1;
			allyTeamNumbers[i]->addText(s.str());
		}
		allyTeamNumbers[i]->setIndex(gameHeader.getAllyTeamNumber(gameHeader.getBasePlayer(i).teamNumber)-1);

		color[i]->clearColors();
		color[i]->addColor(mapHeader.getBaseTeam(gameHeader.getBasePlayer(i).teamNumber).color);
		color[i]->setSelectedColor(0);
		
		if(readOnly)
		{
			allyTeamNumbers[i]->setClickable(false);
		}
		
		addWidget(playerNames[i]);
		addWidget(color[i]);
		addWidget(allyTeamNumbers[i]);
	}
	
	teamsFixed = new OnOffButton(300, 60, 21, 21, ALIGN_SCREEN_CENTERED, ALIGN_SCREEN_CENTERED, gameHeader.areAllyTeamsFixed(), TEAMSFIXED);
	addWidget(teamsFixed);
	teamsFixedText = new Text(325, 60, ALIGN_SCREEN_CENTERED, ALIGN_SCREEN_CENTERED, "standard", Toolkit::getStringTable()->getString("[Teams Fixed]"));
	addWidget(teamsFixedText);
	if(readOnly)
		teamsFixed->setClickable(false);
	
	//These are for winning conditions
	prestigeWinEnabled = new OnOffButton(300, 90, 21, 21, ALIGN_SCREEN_CENTERED, ALIGN_SCREEN_CENTERED, true, PRESTIGEWINENABLED);
	addWidget(prestigeWinEnabled);
	prestigeWinEnabledText = new Text(325, 90, ALIGN_SCREEN_CENTERED, ALIGN_SCREEN_CENTERED, "standard", Toolkit::getStringTable()->getString("[Prestige Win Enabled]"));
	addWidget(prestigeWinEnabledText);
	updateScreenWinningConditions();
	if(readOnly)
		prestigeWinEnabled->setClickable(false);
	
	//Map discovered.
	mapDiscovered = new OnOffButton(300, 120, 21, 21, ALIGN_SCREEN_CENTERED, ALIGN_SCREEN_CENTERED, gameHeader.isMapDiscovered(), MAPDISCOVERED);
	addWidget(mapDiscovered);
	mapDiscoveredText = new Text(325, 120, ALIGN_SCREEN_CENTERED, ALIGN_SCREEN_CENTERED, "standard", Toolkit::getStringTable()->getString("[Map Discovered]"));
	addWidget(mapDiscoveredText);
	if(readOnly)
		mapDiscovered->setClickable(false);
}



CustomGameOtherOptions::~CustomGameOtherOptions()
{

}


void CustomGameOtherOptions::onAction(Widget *source, Action action, int par1, int par2)
{
	if ((action == BUTTON_RELEASED) || (action == BUTTON_SHORTCUT))
	{
		if(par1 == OK)
		{
			endExecute(Finished);
		}
		if(par1 == CANCEL)
		{
			gameHeader = oldGameHeader;
			endExecute(Canceled);
		}
	}
	else if (action==BUTTON_STATE_CHANGED)
	{
		if(par1>=200 && par1<300)
		{
			int team = -1;
			int nth = 0;
			int n = 0;
			///Find which team number this widget is for
			for(int i=0; i<gameHeader.getNumberOfPlayers(); ++i)
			{
				if(allyTeamNumbers[i] == source)
				{
					team = gameHeader.getBasePlayer(i).teamNumber;
					nth = allyTeamNumbers[i]->getIndex();
					n = nth+1;
					break;
				}
			}
			///Adjust all widgets that have this team number
			for(int i=0; i<gameHeader.getNumberOfPlayers(); ++i)
			{
				if(gameHeader.getBasePlayer(i).teamNumber == team)
				{
					allyTeamNumbers[i]->setIndex(nth);
				}
			}
			gameHeader.setAllyTeamNumber(team, n);
		}
		else if(par1 == TEAMSFIXED)
		{
			gameHeader.setAllyTeamsFixed(teamsFixed->getState());
		}
		else if(par1 == PRESTIGEWINENABLED)
		{
			updateGameHeaderWinningConditions();
		}
		else if(par1 == MAPDISCOVERED)
		{
			gameHeader.setMapDiscovered(mapDiscovered->getState());
		}
	}
}



void CustomGameOtherOptions::updateGameHeaderWinningConditions()
{
	std::list<boost::shared_ptr<WinningCondition> >& winningConditions = gameHeader.getWinningConditions();
	winningConditions = WinningCondition::getDefaultWinningConditions();
	
	//Update the prestige condition
	for(std::list<boost::shared_ptr<WinningCondition> >::iterator i = winningConditions.begin(); i!=winningConditions.end(); ++i)
	{
		if((*i)->getType() == WCPrestige)
		{
			//If we need to remove it, do so
			if(!prestigeWinEnabled->getState())
			{
				winningConditions.erase(i);
				break;
			}
			//Otherwise update it
			else
			{
				break;
			}
		}
	}
}



void CustomGameOtherOptions::updateScreenWinningConditions()
{
	std::list<boost::shared_ptr<WinningCondition> >& winningConditions = gameHeader.getWinningConditions();
	
	//Update the prestige condition
	prestigeWinEnabled->setState(false);
	for(std::list<boost::shared_ptr<WinningCondition> >::iterator i = winningConditions.begin(); i!=winningConditions.end(); ++i)
	{
		if((*i)->getType() == WCPrestige)
		{
			prestigeWinEnabled->setState(true);
			break;
		}
	}
}

