///////////////////////////////////////////////////////////////////////////////
//            Copyright (C) 2004-2007 by The Allacrost Project
//                         All Rights Reserved
//
// This code is licensed under the GNU GPL version 2. It is free software
// and you may modify it and/or redistribute it under the terms of this license.
// See http://www.gnu.org/copyleft/gpl.html for details.
///////////////////////////////////////////////////////////////////////////////

#include <iostream>
#include <sstream>

#include "utils.h"
#include "audio.h"
#include "video.h"
#include "input.h"
#include "system.h"
#include "global.h"
#include "mode_manager.h"
#include "menu.h"

#include "shop.h"
#include "shop_windows.h"

using namespace std;

using namespace hoa_utils;
using namespace hoa_audio;
using namespace hoa_video;
using namespace hoa_input;
using namespace hoa_system;
using namespace hoa_global;
using namespace hoa_mode_manager;

namespace hoa_shop {

namespace private_shop {

// *****************************************************************************
// ***** ShopActionWindow
// *****************************************************************************

ShopActionWindow::ShopActionWindow() {
	// (1) Initialize the window
	MenuWindow::Create(200, 600, ~VIDEO_MENU_EDGE_RIGHT);
	MenuWindow::SetPosition(112, 684);
	MenuWindow::SetAlignment(VIDEO_X_LEFT, VIDEO_Y_TOP);
	MenuWindow::SetDisplayMode(VIDEO_MENU_INSTANT);
	MenuWindow::Show();

	// (2) Initialize the list of actions
	action_options.SetOwner(this);
	action_options.SetPosition(25.0f, 600.0f);
	action_options.SetSize(1, 5); // One column, numerous rows
	action_options.SetCellSize(150.0f, 50.0f);
	action_options.SetOptionAlignment(VIDEO_X_LEFT, VIDEO_Y_CENTER);
	action_options.SetFont("default");
	action_options.SetSelectMode(VIDEO_SELECT_SINGLE);
	action_options.SetCursorOffset(-50.0f, 20.0f);
	action_options.SetVerticalWrapMode(VIDEO_WRAP_MODE_STRAIGHT);

	action_options.AddOption(MakeUnicodeString("Buy wares"));
	action_options.AddOption(MakeUnicodeString("Sell inventory"));
	action_options.AddOption(MakeUnicodeString("Confirm transaction"));
	action_options.AddOption(MakeUnicodeString("Enter party menu"));
	action_options.AddOption(MakeUnicodeString("Leave shop"));
	action_options.SetSelection(0);

	// (3) Initialize the financial text box
	finance_text.SetOwner(this);
	finance_text.SetPosition(25.0f, 120.0f);
	finance_text.SetDimensions(150.0f, 65.0f);
	finance_text.SetTextStyle(TextStyle());
	finance_text.SetDisplaySpeed(30);
	finance_text.SetDisplayMode(VIDEO_TEXT_INSTANT);
	finance_text.SetTextAlignment(VIDEO_X_LEFT, VIDEO_Y_TOP);
	UpdateFinanceText();
} // ShopActionWindow::ShopActionWindow()



ShopActionWindow::~ShopActionWindow() {
	MenuWindow::Destroy();
}



void ShopActionWindow::Update() {
	MenuWindow::Update(SystemManager->GetUpdateTime());
	action_options.Update(); // Clear any OptionBox events, since they prevent further user input

	if (InputManager->ConfirmPress()) {
		action_options.HandleConfirmKey();
		if (action_options.GetSelection() == 0) { // Buy wares
			action_options.SetCursorState(VIDEO_CURSOR_STATE_HIDDEN);
			current_shop->_info_window.SetObject(current_shop->_buy_objects[0]);
			current_shop->_buy_window.hide_options = false;
			current_shop->_buy_window.object_list.SetCursorState(VIDEO_CURSOR_STATE_VISIBLE);
			current_shop->_state = SHOP_STATE_BUY;
			current_shop->_shop_sounds["confirm"].Play();
		}
		else if (action_options.GetSelection() == 1) { // Sell inventory
			// TODO: Inventory empty check is not sufficient here because if the inventory only consists
			// of key items, then the sell menu should not come up
			if (GlobalManager->GetInventory()->empty() == true) {
				current_shop->_PushAndSetState(SHOP_STATE_PROMPT);
				current_shop->_prompt_window.Show();
				current_shop->_prompt_window.prompt_text.SetDisplayText(MakeUnicodeString(
					"The party's inventory is empty. There is nothing you can offer to sell.")
				);
				current_shop->_shop_sounds["cancel"].Play();
			}
			else {
				action_options.SetCursorState(VIDEO_CURSOR_STATE_HIDDEN);
				current_shop->_info_window.SetObject(current_shop->_current_inv[0]);
				current_shop->_sell_window.UpdateSellList();
				current_shop->_sell_window.object_list.SetSelection(0);
				current_shop->_sell_window.hide_options = false;
				current_shop->_sell_window.object_list.SetCursorState(VIDEO_CURSOR_STATE_VISIBLE);
				current_shop->_state = SHOP_STATE_SELL;
				current_shop->_shop_sounds["confirm"].Play();
			}
		}
		else if (action_options.GetSelection() == 2) { // Confirm transaction
			if ((current_shop->_purchases_cost + current_shop->_sales_revenue) == 0) {
				current_shop->_PushAndSetState(SHOP_STATE_PROMPT);
				current_shop->_prompt_window.Show();
				current_shop->_prompt_window.prompt_text.SetDisplayText(MakeUnicodeString(
					"You have not offered to make any purchases or sales.")
				);
				current_shop->_shop_sounds["cancel"].Play();
			}
			else {
				action_options.SetCursorState(VIDEO_CURSOR_STATE_HIDDEN);
				current_shop->_state = SHOP_STATE_CONFIRM;
				current_shop->_confirm_window.options.SetCursorState(VIDEO_CURSOR_STATE_VISIBLE);
				current_shop->_confirm_window.Show();
			}
		}
		else if (action_options.GetSelection() == 3) { // Enter party menu
			current_shop->_shop_sounds["confirm"].Play();
			// TEMP: the current location name and graphic need to be retrieved from the most recent map mode on the stack
			hoa_menu::MenuMode *MM = new hoa_menu::MenuMode(MakeUnicodeString("Village"), "img/menus/locations/mountain_village.png");
			ModeManager->Push(MM);
		}
		else if (action_options.GetSelection() == 4) { // Leave shop
			ModeManager->Pop();
			current_shop->_shop_sounds["cancel"].Play();
		}
		else {
			IF_PRINT_WARNING(SHOP_DEBUG) << "invalid selection in action window: " << action_options.GetSelection() << endl;
			ModeManager->Pop();
		}
	}
	else if (InputManager->CancelPress()) {
		ModeManager->Pop();
		current_shop->_shop_sounds["cancel"].Play();
	}
	else if (InputManager->UpPress()) {
		action_options.HandleUpKey();
	}
	else if (InputManager->DownPress()) {
		action_options.HandleDownKey();
	}
}



void ShopActionWindow::UpdateFinanceText() {
	if (current_shop != NULL) {
		finance_text.SetDisplayText(MakeUnicodeString(
			  "Funds:  " + NumberToString(GlobalManager->GetDrunes()) +
			"\nCosts: " + (current_shop->GetPurchaseCost() == 0 ? " " : "-") + NumberToString(current_shop->GetPurchaseCost()) +
			"\nSales: " + (current_shop->GetSalesRevenue() == 0 ? " " : "+") + NumberToString(current_shop->GetSalesRevenue()) +
			"\nTotal:  " + NumberToString(current_shop->GetTotalRemaining())
		));
	}
}



void ShopActionWindow::Draw() {
	MenuWindow::Draw();
	action_options.Draw();
	finance_text.Draw();
}

// *****************************************************************************
// ***** BuyListWindow
// *****************************************************************************

BuyListWindow::BuyListWindow() {
	MenuWindow::Create(600, 400, VIDEO_MENU_EDGE_ALL, VIDEO_MENU_EDGE_LEFT | VIDEO_MENU_EDGE_BOTTOM);
	MenuWindow::SetPosition(312, 684);
	MenuWindow::SetAlignment(VIDEO_X_LEFT, VIDEO_Y_TOP);
	MenuWindow::SetDisplayMode(VIDEO_MENU_INSTANT);
	MenuWindow::Show();

	object_list.SetOwner(this);
	object_list.SetCellSize(500.0f, 50.0f);
	object_list.SetPosition(50.0f, 400.0f);
	object_list.SetOptionAlignment(VIDEO_X_LEFT, VIDEO_Y_CENTER);
	object_list.SetFont("default");
	object_list.SetSelectMode(VIDEO_SELECT_SINGLE);
	object_list.SetCursorOffset(-50.0f, 20.0f);
	object_list.SetHorizontalWrapMode(VIDEO_WRAP_MODE_NONE);

	hide_options = true;
}



BuyListWindow::~BuyListWindow() {
	MenuWindow::Destroy();
}



void BuyListWindow::AddEntry(hoa_utils::ustring name, uint32 price, uint32 quantity) {
	object_list.AddOption(name + MakeUnicodeString("<R>") + MakeUnicodeString(NumberToString(price)) +
		MakeUnicodeString("   x") + MakeUnicodeString(NumberToString(quantity))
	);
}



void BuyListWindow::RefreshList() {
	object_list.ClearOptions();
	for (uint32 i = 0; i < current_shop->_buy_objects.size(); i++) {
		AddEntry(current_shop->_buy_objects[i]->GetName(), current_shop->_buy_objects[i]->GetPrice(),
			current_shop->_buy_objects_quantities[i]
		);
	}
	object_list.SetSize(1, object_list.GetNumberOptions());
	object_list.SetSelection(0);
}



void BuyListWindow::Update() {
	MenuWindow::Update(SystemManager->GetUpdateTime());
	object_list.Update(); // Clear any OptionBox events, since they prevent further user input

	if (InputManager->ConfirmPress()) {
		object_list.HandleConfirmKey();

		int32 x = object_list.GetSelection();
		if (current_shop->_buy_objects_quantities[x] == 0) {
			current_shop->_PushAndSetState(SHOP_STATE_PROMPT);
			current_shop->_prompt_window.Show();
			current_shop->_prompt_window.prompt_text.SetDisplayText(MakeUnicodeString(
				"No quantity for this selection was made. Use the right and left commands to increment "
				"or decrement the amount of this object to purchase.")
			);
		}
		else {

		current_shop->_state = SHOP_STATE_CONFIRM;
		object_list.SetCursorState(VIDEO_CURSOR_STATE_HIDDEN);
		current_shop->_info_window.SetObject(NULL);
		current_shop->_confirm_window.options.SetCursorState(VIDEO_CURSOR_STATE_VISIBLE);
		current_shop->_confirm_window.Show();
		current_shop->_shop_sounds["confirm"].Play();
		}
	}
	else if (InputManager->CancelPress()) {
		hide_options = true;
		current_shop->_state = SHOP_STATE_ACTION;
		current_shop->_action_window.action_options.SetCursorState(VIDEO_CURSOR_STATE_VISIBLE);
		object_list.SetCursorState(VIDEO_CURSOR_STATE_HIDDEN);
		current_shop->_info_window.SetObject(NULL);
		current_shop->_shop_sounds["cancel"].Play();
	}
	else if (InputManager->UpPress()) {
		object_list.HandleUpKey();
		current_shop->_info_window.SetObject(current_shop->_buy_objects[object_list.GetSelection()]);
	}
	else if (InputManager->DownPress()) {
		object_list.HandleDownKey();
		current_shop->_info_window.SetObject(current_shop->_buy_objects[object_list.GetSelection()]);
	}
	else if (InputManager->LeftPress()) {
		int32 x = object_list.GetSelection();
		if (current_shop->_buy_objects_quantities[x] > 0) {
			current_shop->_buy_objects_quantities[x]--;
			current_shop->_purchases_cost -= current_shop->_buy_objects[x]->GetPrice();
			object_list.SetOptionText(x, current_shop->_buy_objects[x]->GetName() + MakeUnicodeString("<R>") +
				MakeUnicodeString(NumberToString(current_shop->_buy_objects[x]->GetPrice())) + MakeUnicodeString("   x") +
				MakeUnicodeString(NumberToString(current_shop->_buy_objects_quantities[x]))
			);
			current_shop->_action_window.UpdateFinanceText();
		}
		else {
			current_shop->_shop_sounds["cancel"].Play();
		}
	}
	else if (InputManager->RightPress()) {
		int32 x = object_list.GetSelection();
		if (current_shop->_buy_objects[x]->GetPrice() <= current_shop->GetTotalRemaining()) {
			current_shop->_buy_objects_quantities[x]++;
			current_shop->_purchases_cost += current_shop->_buy_objects[x]->GetPrice();
			current_shop->_action_window.UpdateFinanceText();
			object_list.SetOptionText(x, current_shop->_buy_objects[x]->GetName() + MakeUnicodeString("<R>") +
				MakeUnicodeString(NumberToString(current_shop->_buy_objects[x]->GetPrice())) + MakeUnicodeString("   x") +
				MakeUnicodeString(NumberToString(current_shop->_buy_objects_quantities[x]))
			);
		}
		else {
			current_shop->_shop_sounds["cancel"].Play();
		}
	}
} // void BuyListWindow::Update()



void BuyListWindow::Draw() {
	MenuWindow::Draw();

	if (hide_options == false && object_list.GetNumberOptions() > 0) {
		object_list.Draw();
	}
}

// *****************************************************************************
// ***** SellListWindow
// *****************************************************************************

SellListWindow::SellListWindow() {
	MenuWindow::Create(600, 400, VIDEO_MENU_EDGE_ALL, VIDEO_MENU_EDGE_LEFT | VIDEO_MENU_EDGE_BOTTOM);
	MenuWindow::SetPosition(312, 684);
	MenuWindow::SetAlignment(VIDEO_X_LEFT, VIDEO_Y_TOP);
	MenuWindow::SetDisplayMode(VIDEO_MENU_INSTANT);
	MenuWindow::Show();

	object_list.SetOwner(this);
	object_list.SetCellSize(500.0f, 50.0f);
	object_list.SetPosition(50.0f, 350.0f);
	object_list.SetSize(1, 6);
	object_list.SetOptionAlignment(VIDEO_X_LEFT, VIDEO_Y_CENTER);
	object_list.SetFont("default");
	object_list.SetSelectMode(VIDEO_SELECT_SINGLE);
	object_list.SetCursorOffset(-50.0f, 20.0f);
	object_list.SetVerticalWrapMode(VIDEO_WRAP_MODE_STRAIGHT);
	object_list.SetHorizontalWrapMode(VIDEO_WRAP_MODE_NONE);

	hide_options = true;
}



SellListWindow::~SellListWindow() {
	MenuWindow::Destroy();
}



void SellListWindow::Clear() {
	option_text.clear();
	current_shop->_current_inv.clear();
	object_list.SetOptions(option_text);
}



void SellListWindow::AddEntry(hoa_utils::ustring name, uint32 count, uint32 price, uint32 sell_count) {
	string text = MakeStandardString(name) + "<R>" + NumberToString(count) + "      x" + NumberToString(sell_count) + "       " + NumberToString(price);
	option_text.push_back(MakeUnicodeString(text));
}



void SellListWindow::Update() {
	MenuWindow::Update(SystemManager->GetUpdateTime());
	object_list.Update(); // Clear any OptionBox events, since they prevent further user input

	if (InputManager->ConfirmPress()) {
		object_list.HandleConfirmKey();
		int32 x = object_list.GetSelection();

		if (current_shop->_sell_objects_quantities[x] == 0) {
			current_shop->_PushAndSetState(SHOP_STATE_PROMPT);
			current_shop->_prompt_window.Show();
			current_shop->_prompt_window.prompt_text.SetDisplayText(MakeUnicodeString(
				"No quantity for this selection was made. Use the right and left commands to increment "
				"or decrement the amount of this object to offer for sale.")
			);
		}
		else {
			hide_options = true;
			object_list.SetCursorState(VIDEO_CURSOR_STATE_HIDDEN);
			current_shop->_state = SHOP_STATE_CONFIRM;
			current_shop->_confirm_window.options.SetCursorState(VIDEO_CURSOR_STATE_VISIBLE);
			current_shop->_confirm_window.Show();
			current_shop->_shop_sounds["confirm"].Play();
		}
	}
	else if (InputManager->CancelPress()) {
		hide_options = true;
		current_shop->_state = SHOP_STATE_ACTION;
		current_shop->_action_window.action_options.SetCursorState(VIDEO_CURSOR_STATE_VISIBLE);
		object_list.SetCursorState(VIDEO_CURSOR_STATE_HIDDEN);
		current_shop->_info_window.SetObject(NULL);
		current_shop->_shop_sounds["cancel"].Play();
	}
	else if (InputManager->UpPress()) {
		object_list.HandleUpKey();
		current_shop->_info_window.SetObject(current_shop->_current_inv[object_list.GetSelection()]);
	}
	else if (InputManager->DownPress()) {
		object_list.HandleDownKey();
		current_shop->_info_window.SetObject(current_shop->_current_inv[object_list.GetSelection()]);
	}
	else if (InputManager->LeftPress()) {
		int32 x = object_list.GetSelection();
		if (current_shop->_sell_objects_quantities[x] > 0) {
			current_shop->_sell_objects_quantities[x]--;
			current_shop->_sales_revenue -= (current_shop->_current_inv[x]->GetPrice() / 2);
			current_shop->_action_window.UpdateFinanceText();
		}
		else {
			current_shop->_shop_sounds["cancel"].Play();
		}
	}
	else if (InputManager->RightPress()) {
		uint32 x = object_list.GetSelection();
		if (current_shop->_current_inv[x]->GetCount() > static_cast<uint32>(current_shop->_sell_objects_quantities[x])) {
			current_shop->_sell_objects_quantities[x]++;
			current_shop->_sales_revenue += (current_shop->_current_inv[x]->GetPrice() / 2);
			current_shop->_action_window.UpdateFinanceText();
		}
		else {
			current_shop->_shop_sounds["cancel"].Play();
		}
	}

	UpdateSellList();
}



void SellListWindow::UpdateSellList() {
	Clear();
	map<uint32, GlobalObject*>* inv = GlobalManager->GetInventory();
	map<uint32, GlobalObject*>::iterator iter;

	uint32 x = 0;
	for (iter = inv->begin(); iter != inv->end(); iter++) {
		current_shop->_current_inv.push_back(iter->second);
		AddEntry(iter->second->GetName(), iter->second->GetCount(), iter->second->GetPrice() / 2, current_shop->_sell_objects_quantities[x]);
		x++;
	}
	object_list.SetOptions(option_text);
}



void SellListWindow::Draw() {
	MenuWindow::Draw();

	if (hide_options == false && object_list.GetNumberOptions() != 0) {
		object_list.Draw();
		VideoManager->Move(375, 640);
		VideoManager->Text()->Draw(MakeUnicodeString("Item                                                                     Inv   Sell   Price"));
	}
}

// *****************************************************************************
// ***** ObjectInfoWindow
// ***************************************************************************

ObjectInfoWindow::ObjectInfoWindow() {
	// (1) Create the info window in the bottom right-hand section of the screen
	MenuWindow::Create(600, 216, ~VIDEO_MENU_EDGE_TOP);
	MenuWindow::SetPosition(312, 300);
	MenuWindow::SetAlignment(VIDEO_X_LEFT, VIDEO_Y_TOP);
	MenuWindow::SetDisplayMode(VIDEO_MENU_INSTANT);
	MenuWindow::Show();

	// (2) Initialize the object to NULL, so that no information is displayed
	_object = NULL;

	// (3) Initialize the description text box in the lower section of the window
	description.SetOwner(this);
	description.SetPosition(25.0f, 100.0f);
	description.SetDimensions(550.0f, 80.0f);
	description.SetDisplaySpeed(30);
	description.SetTextStyle(TextStyle());
	description.SetDisplayMode(VIDEO_TEXT_INSTANT);
	description.SetTextAlignment(VIDEO_X_LEFT, VIDEO_Y_TOP);

	// (4) Initialize the properties text box in the upper right section of the window
	properties.SetOwner(this);
	properties.SetPosition(50.0f, 150.0f);
	properties.SetDimensions(300.0f, 80.0f);
	properties.SetDisplaySpeed(30);
	properties.SetTextStyle(TextStyle());
	properties.SetDisplayMode(VIDEO_TEXT_INSTANT);
	properties.SetTextAlignment(VIDEO_X_RIGHT, VIDEO_Y_TOP);
}



ObjectInfoWindow::~ObjectInfoWindow() {
	MenuWindow::Destroy();
}



void ObjectInfoWindow::SetObject(GlobalObject* obj) {
	_object = obj;
	if (obj == NULL) {
		description.ClearText();
		properties.ClearText();
		return;
	}

	description.SetDisplayText(_object->GetDescription());

	// Determine what properties to display depending on what type of object this is
	switch (obj->GetObjectType()) {
		case GLOBAL_OBJECT_WEAPON:
			GlobalWeapon *weapon;
			weapon = dynamic_cast<GlobalWeapon*>(obj);
			properties.SetDisplayText(
				"PHYS ATK: " + NumberToString(weapon->GetPhysicalAttack()) + "\n" +
				"META ATK: " + NumberToString(weapon->GetMetaphysicalAttack())
			);
			break;
		case GLOBAL_OBJECT_HEAD_ARMOR:
		case GLOBAL_OBJECT_TORSO_ARMOR:
		case GLOBAL_OBJECT_ARM_ARMOR:
		case GLOBAL_OBJECT_LEG_ARMOR:
			GlobalArmor *armor;
			armor = dynamic_cast<GlobalArmor*>(obj);
			properties.SetDisplayText(
				"PHYS DEF: " + NumberToString(armor->GetPhysicalDefense()) + "\n" +
				"META DEF: " + NumberToString(armor->GetMetaphysicalDefense())
			);
			break;
		default:
			properties.ClearText();
			break;
	}
} // void ObjectInfoWindow::SetObject(GlobalObject* obj)



void ObjectInfoWindow::Draw() {
	MenuWindow::Draw();

	if (_object == NULL) {
		return;
	}

	// Draw the object's icon and name
	VideoManager->Move(350, 200);
	_object->GetIconImage().Draw();
	VideoManager->MoveRelative(60, 20);
	VideoManager->Text()->Draw(_object->GetName());

	// Draw the object's description and stats text boxes
	description.Draw();
	properties.Draw();
}

// *****************************************************************************
// ***** ConfirmWindow
// *****************************************************************************

ConfirmWindow::ConfirmWindow() {
	// (1) Create the confirmation window in the center of the screen
	MenuWindow::Create(400, 200);
	MenuWindow::SetPosition(512, 384);
	MenuWindow::SetAlignment(VIDEO_X_CENTER, VIDEO_Y_CENTER);
	MenuWindow::SetDisplayMode(VIDEO_MENU_INSTANT);

	// (2) Initialize the option list
	options.SetOwner(this);
	options.SetPosition(100.0f, 100.0f);
	options.SetSize(2, 1); // Two columns, one row
	options.SetCellSize(150.0f, 50.0f);
	options.SetOptionAlignment(VIDEO_X_LEFT, VIDEO_Y_CENTER);
	options.SetFont("default");
	options.SetSelectMode(VIDEO_SELECT_SINGLE);
	options.SetCursorOffset(-50.0f, 20.0f);
	options.SetVerticalWrapMode(VIDEO_WRAP_MODE_NONE);

	vector<ustring> text;
	text.push_back(MakeUnicodeString("Confirm"));
	text.push_back(MakeUnicodeString("Cancel"));
	options.SetOptions(text);
	options.SetSelection(0);
}



ConfirmWindow::~ConfirmWindow() {
	MenuWindow::Destroy();
}



void ConfirmWindow::Update() {
	options.Update();

	if (InputManager->LeftPress()) {
		options.HandleLeftKey();
	}
	else if (InputManager->RightPress()) {
		options.HandleRightKey();
	}

	if (InputManager->CancelPress()) {
		current_shop->_shop_sounds["cancel"].Play();
		options.SetSelection(0);
		current_shop->_action_window.action_options.SetCursorState(VIDEO_CURSOR_STATE_VISIBLE);
		options.SetCursorState(VIDEO_CURSOR_STATE_HIDDEN);
		current_shop->_state = SHOP_STATE_ACTION;
	}
	else if (InputManager->ConfirmPress()) {
		if (options.GetSelection() == 0) { // Confirm purchase
			for (uint32 ctr = 0; ctr < current_shop->_current_inv.size(); ctr++) {
				if (current_shop->_sell_objects_quantities[ctr] > 0) {
					GlobalManager->DecrementObjectCount(current_shop->_current_inv[ctr]->GetID(), current_shop->_sell_objects_quantities[ctr]);
				}
			}

			for (uint32 ctr = 0; ctr < current_shop->_buy_objects.size(); ctr++) {
				if (current_shop->_buy_objects_quantities[ctr] > 0) {
					GlobalManager->AddToInventory(current_shop->_buy_objects[ctr]->GetID(), current_shop->_buy_objects_quantities[ctr]);
				}
				current_shop->_buy_objects_quantities[ctr] = 0;
			}

			map<uint32, GlobalObject*>* inv = GlobalManager->GetInventory();
			map<uint32, GlobalObject*>::iterator iter;

			current_shop->_sell_objects_quantities.clear();
			for (iter = inv->begin(); iter != inv->end(); iter++) {
				current_shop->_sell_objects_quantities.push_back(0);
			}

			GlobalManager->SubtractDrunes(current_shop->GetPurchaseCost());
			GlobalManager->AddDrunes(current_shop->GetSalesRevenue());
			current_shop->_purchases_cost = 0;
			current_shop->_sales_revenue = 0;
			current_shop->_shop_sounds["coins"].Play();
			current_shop->_action_window.UpdateFinanceText();
			current_shop->_info_window.SetObject(NULL);
			current_shop->_buy_window.RefreshList();
			current_shop->_sell_window.UpdateSellList();
			current_shop->_state = SHOP_STATE_BUY;
		}
		else { // Cancel purchase
			current_shop->_shop_sounds["cancel"].Play();
		}

		// Return to previous window
		options.SetSelection(0);
		current_shop->_action_window.action_options.SetCursorState(VIDEO_CURSOR_STATE_VISIBLE);
		options.SetCursorState(VIDEO_CURSOR_STATE_HIDDEN);
		current_shop->_PopState();
	}
} // void ConfirmWindow::Update()



void ConfirmWindow::Draw() {
	MenuWindow::Draw();
	options.Draw();

	VideoManager->PushState();
	VideoManager->SetDrawFlags(VIDEO_X_CENTER, VIDEO_Y_CENTER, 0);
	VideoManager->Move(512, 450);
	VideoManager->Text()->Draw("Finalize transactions?");
	VideoManager->PopState();
}

// *****************************************************************************
// ***** PromptWindow
// *****************************************************************************

PromptWindow::PromptWindow() {
	// (1) Create the confirmation window in the center of the screen
	MenuWindow::Create(400, 200);
	MenuWindow::SetPosition(512, 384);
	MenuWindow::SetAlignment(VIDEO_X_CENTER, VIDEO_Y_CENTER);
	MenuWindow::SetDisplayMode(VIDEO_MENU_INSTANT);

	// (2) Initialize the textbox list
	prompt_text.SetOwner(this);
	prompt_text.SetPosition(25.0f, 175.0f);
	prompt_text.SetDimensions(360.0f, 160.0f);
	prompt_text.SetDisplaySpeed(30);
	prompt_text.SetTextStyle(TextStyle());
	prompt_text.SetDisplayMode(VIDEO_TEXT_INSTANT);
	prompt_text.SetTextAlignment(VIDEO_X_LEFT, VIDEO_Y_TOP);
}



PromptWindow::~PromptWindow() {
	MenuWindow::Destroy();
}



void PromptWindow::Update() {
	prompt_text.Update();

	if (InputManager->ConfirmPress() || InputManager->CancelPress()) {
		current_shop->_PopState();
		prompt_text.ClearText();
	}
}



void PromptWindow::Draw() {
	MenuWindow::Draw();
	prompt_text.Draw();
}

} // namespace private_shop

} // namespace hoa_shop