/****************************************************************************
 * libwiigui
 *
 * Tantric 2009
 *
 * gui_keyboard.cpp
 *
 * GUI class definitions
 ***************************************************************************/

#include "gui.h"
#include <stdio.h>
#include <string.h>
/**
 * Constructor for the GuiKeyboard class.
 */
extern const int vol;
unsigned int m;
//const Key thekeys;
GuiKeyboard::GuiKeyboard(char * t, u32 max, int min, int lang)
{
	width = 540;
	height = 400;
	shift = 0;
	caps = 0;
	alt = 0;
	alt2 = 0;
	m = min;
	int mode = lang;
	selectable = true;
	focus = 0; // allow focus
	alignmentHor = ALIGN_CENTRE;
	alignmentVert = ALIGN_MIDDLE;
	strncpy(kbtextstr, t, max);
	kbtextstr[max] = 0;
	kbtextmaxlen = max;

	//QWERTY//
	if (mode == 0){
	Key thekeys[4][11] = {
	{
		{'1','!'},
		{'2','@'},
		{'3','#'},
		{'4','$'},
		{'5','%'},
		{'6','^'},
		{'7','&'},
		{'8','*'},
		{'9','('},
		{'0',')'},
		{'\0','\0'}
	},
	{
		{'q','Q'},
		{'w','W'},
		{'e','E'},
		{'r','R'},
		{'t','T'},
		{'y','Y'},
		{'u','U'},
		{'i','I'},
		{'o','O'},
		{'p','P'},
		{'-','_'}
	},
	{
		{'a','A'},
		{'s','S'},
		{'d','D'},
		{'f','F'},
		{'g','G'},
		{'h','H'},
		{'j','J'},
		{'k','K'},
		{'l','L'},
		{':',';'},
		{'\'','"'}
	},

	{
		{'z','Z'},
		{'x','X'},
		{'c','C'},
		{'v','V'},
		{'b','B'},
		{'n','N'},
		{'m','M'},
		{',','<'},
		{'.','>'},
		{'/','?'},
		{'\0','\0'}
	}
	};

	memcpy(keys, thekeys, sizeof(thekeys));}
	//DVORAK//
	if (mode == 1){
	Key thekeys[4][11] = {
	{
		{'1','!','\0'},
		{'2','@','\0'},
		{'3','#','\0'},
		{'4','$','\0'},
		{'5','%','\0'},
		{'6','^','\0'},
		{'7','&','\0'},
		{'8','*','\0'},
		{'9','(','\0'},
		{'0',')','\0'},
		{'\0','\0','\0'}
	},
	{
		{'\'','"','\0'},
		{',','<','\0'},
		{'.','>','\0'},
		{'p','P','\0'},
		{'y','Y','\0'},
		{'f','F','\0'},
		{'g','G','\0'},
		{'c','C','\0'},
		{'r','R','\0'},
		{'l','L','\0'},
		{'/','?','\0'}
	},
	{
		{'a','A','m'},
		{'o','O','m'},
		{'e','E','m'},
		{'u','U','m'},
		{'i','I','m'},
		{'d','D','m'},
		{'h','H','m'},
		{'t','T','m'},
		{'n','N','m'},
		{'s','S','m'},
		{'-','_','m'}
	},

	{
		{';',':','\0'},
		{'q','Q','\0'},
		{'j','J','\0'},
		{'k','K','\0'},
		{'x','X','\0'},
		{'b','B','\0'},
		{'m','M','\0'},
		{'w','W','\0'},
		{'v','V','\0'},
		{'z','Z','\0'},
		{'\0','\0','\0'}
	}
	};
	memcpy(keys, thekeys, sizeof(thekeys));}
	//QWETRZ//
	if (mode == 2){
	Key thekeys[4][11] = {
	{
		{'1','!','^','À'},
		{'2','"','²','à'},
		{'3','#','³','È'},
		{'4','$','«','è'},
		{'5','%','»','Ì'},
		{'6','&','„','ì'},
		{'7','/','”','Ò'},
		{'8','(','[','ò'},
		{'9',')',']','Ù'},
		{'0','=','§','ù'},
		{'ß','?','\'','Ý'}
	},
	{
		{'q','Q','@','Á'},
		{'w','W','\0','á'},
		{'e','E','€','É'},
		{'r','R','\0','é'},
		{'t','T','\0','Í'},
		{'z','Z','\0','í'},
		{'u','U','\0','Ó'},
		{'i','I','\0','ó'},
		{'o','O','\0','Ú'},
		{'p','P','\0','ú'},
		{'ü','Ü','\0','ý'}
	},
	{
		{'a','A','\0','Â'},
		{'s','S','\0','â'},
		{'d','D','\0','Ê'},
		{'f','F','\0','ê'},
		{'g','G','\0','Î'},
		{'h','H','\0','î'},
		{'j','J','\0','Ô'},
		{'k','K','\0','ô'},
		{'l','L','\0','Û'},
		{'ö','Ö','\0','û'},
		{'ä','Ä','\0','Ÿ'}
	},
	{
		{'<','>','|','Ã'},
		{'y','Y','\0','ã'},
		{'x','X','\0','Ñ'},
		{'c','C','ç','ñ'},
		{'v','V','©','Ï'},
		{'b','B','\0','ï'},
		{'n','N','\0','Õ'},
		{'m','M','\0','õ'},
		{',',';','µ','ÿ'},
		{'.',':','\0','\0'},
		{'-','_','\0','\0'}
	}
	};
	memcpy(keys, thekeys, sizeof(thekeys));}

	keyTextbox = new GuiImageData(keyboard_textbox_png);
	keyTextboxImg = new GuiImage(keyTextbox);
	keyTextboxImg->SetAlignment(ALIGN_CENTRE, ALIGN_TOP);
	keyTextboxImg->SetPosition(0, 40);//(0,0);
	this->Append(keyTextboxImg);

	kbText = new GuiText(kbtextstr, 20, (GXColor){0, 0, 0, 0xff});
	kbText->SetAlignment(ALIGN_CENTRE, ALIGN_TOP);
	kbText->SetPosition(0, 53);//(0, 13);
	this->Append(kbText);

	key = new GuiImageData(keyboard_key_png);
	keyOver = new GuiImageData(keyboard_key_over_png);
	keyMedium = new GuiImageData(keyboard_mediumkey_over_png);
	keyMediumOver = new GuiImageData(keyboard_mediumkey_over_png);
	keyLarge = new GuiImageData(keyboard_largekey_over_png);
	keyLargeOver = new GuiImageData(keyboard_largekey_over_png);

	keySoundOver = new GuiSound(button_over_pcm, button_over_pcm_size, SOUND_PCM, vol);
	keySoundClick = new GuiSound(button_click_pcm, button_click_pcm_size, SOUND_PCM, vol);
	trigA = new GuiTrigger;
	trigA->SetSimpleTrigger(-1, WPAD_BUTTON_A | WPAD_CLASSIC_BUTTON_A, PAD_BUTTON_A);
	trigB = new GuiTrigger;
	trigB->SetButtonOnlyTrigger(-1, WPAD_BUTTON_B | WPAD_CLASSIC_BUTTON_B, PAD_BUTTON_B);

    int eurocheck = 0;
    if(lang == 2) {
    eurocheck = -20;
    }

	keyBackImg = new GuiImage(keyMedium);
	keyBackOverImg = new GuiImage(keyMediumOver);
	keyBackText = new GuiText("Back", 20, (GXColor){0, 0, 0, 0xff});
	keyBack = new GuiButton(keyMedium->GetWidth(), keyMedium->GetHeight());
	keyBack->SetImage(keyBackImg);
	keyBack->SetImageOver(keyBackOverImg);
	keyBack->SetLabel(keyBackText);
	keyBack->SetSoundOver(keySoundOver);
	keyBack->SetSoundClick(keySoundClick);
	keyBack->SetTrigger(trigA);
	keyBack->SetTrigger(trigB);
	if (mode == 2){
	keyBack->SetPosition(11*42+40+eurocheck, 0*42+120);}
	else{
	keyBack->SetPosition(10*42+40+eurocheck, 0*42+120);}//(10*42+40, 0*42+80);
	keyBack->SetEffectGrow();
	this->Append(keyBack);

	keyClearImg = new GuiImage(keyMedium);
	keyClearOverImg = new GuiImage(keyMediumOver);
	keyClearText = new GuiText("clear", 20, (GXColor){0, 0, 0, 0xff});
	keyClear = new GuiButton(keyMedium->GetWidth(), keyMedium->GetHeight());
	keyClear->SetImage(keyClearImg);
	keyClear->SetImageOver(keyClearOverImg);
	keyClear->SetLabel(keyClearText);
	keyClear->SetSoundOver(keySoundOver);
	keyClear->SetSoundClick(keySoundClick);
	keyClear->SetTrigger(trigA);
	keyClear->SetPosition((10*42+40)+eurocheck, 4*42+120);//(10*42+40, 0*42+80);
	keyClear->SetEffectGrow();
	this->Append(keyClear);

	keyAltImg = new GuiImage(keyMedium);
	keyAltOverImg = new GuiImage(keyMediumOver);
	keyAltText = new GuiText("Alt Gr", 20, (GXColor){0, 0, 0, 0xff});
	keyAlt = new GuiButton(keyMedium->GetWidth(), keyMedium->GetHeight());
	keyAlt->SetImage(keyAltImg);
	keyAlt->SetImageOver(keyAltOverImg);
	keyAlt->SetLabel(keyAltText);
	keyAlt->SetSoundOver(keySoundOver);
	keyAlt->SetSoundClick(keySoundClick);
	keyAlt->SetTrigger(trigA);
	keyAlt->SetPosition(84+eurocheck, 4*42+120);//(10*42+40, 4*42+120);
	keyAlt->SetEffectGrow();
	if (mode == 2){this->Append(keyAlt);}

	keyAlt2Img = new GuiImage(keyMedium);
	keyAlt2OverImg = new GuiImage(keyMediumOver);
	keyAlt2Text = new GuiText("Accent", 20, (GXColor){0, 0, 0, 0xff});
	keyAlt2 = new GuiButton(keyMedium->GetWidth(), keyMedium->GetHeight());
	keyAlt2->SetImage(keyAlt2Img);
	keyAlt2->SetImageOver(keyAlt2OverImg);
	keyAlt2->SetLabel(keyAlt2Text);
	keyAlt2->SetSoundOver(keySoundOver);
	keyAlt2->SetSoundClick(keySoundClick);
	keyAlt2->SetTrigger(trigA);
	keyAlt2->SetPosition((8*42+40)+eurocheck, 4*42+120);//(10*42+40, 4*42+120);
	keyAlt2->SetEffectGrow();
	if (mode == 2){this->Append(keyAlt2);}

	keyCapsImg = new GuiImage(keyMedium);
	keyCapsOverImg = new GuiImage(keyMediumOver);
	keyCapsText = new GuiText("Caps", 20, (GXColor){0, 0, 0, 0xff});
	keyCaps = new GuiButton(keyMedium->GetWidth(), keyMedium->GetHeight());
	keyCaps->SetImage(keyCapsImg);
	keyCaps->SetImageOver(keyCapsOverImg);
	keyCaps->SetLabel(keyCapsText);
	keyCaps->SetSoundOver(keySoundOver);
	keyCaps->SetSoundClick(keySoundClick);
	keyCaps->SetTrigger(trigA);
	keyCaps->SetPosition(0+eurocheck, 2*42+120);//(0, 2*42+80);
	keyCaps->SetEffectGrow();
	this->Append(keyCaps);

	keyShiftImg = new GuiImage(keyMedium);
	keyShiftOverImg = new GuiImage(keyMediumOver);
	keyShiftText = new GuiText("Shift", 20, (GXColor){0, 0, 0, 0xff});
	keyShift = new GuiButton(keyMedium->GetWidth(), keyMedium->GetHeight());
	keyShift->SetImage(keyShiftImg);
	keyShift->SetImageOver(keyShiftOverImg);
	keyShift->SetLabel(keyShiftText);
	keyShift->SetSoundOver(keySoundOver);
	keyShift->SetSoundClick(keySoundClick);
	keyShift->SetTrigger(trigA);
	keyShift->SetPosition(21+eurocheck, 3*42+120);//(21, 3*42+80);
	keyShift->SetEffectGrow();
	this->Append(keyShift);

	keySpaceImg = new GuiImage(keyLarge);
	keySpaceOverImg = new GuiImage(keyLargeOver);
	keySpace = new GuiButton(keyLarge->GetWidth(), keyLarge->GetHeight());
	keySpace->SetImage(keySpaceImg);
	keySpace->SetImageOver(keySpaceOverImg);
	keySpace->SetSoundOver(keySoundOver);
	keySpace->SetSoundClick(keySoundClick);
	keySpace->SetTrigger(trigA);
	keySpace->SetPosition(0+eurocheck, 4*42+120);//(0, 4*42+80);
	keySpace->SetAlignment(ALIGN_CENTRE, ALIGN_TOP);
	keySpace->SetEffectGrow();
	this->Append(keySpace);

	for(int i=0; i<4; i++)
	{
		for(int j=0; j<11; j++)
		{
			if(keys[i][j].ch != '\0')
			{
				keyImg[i][j] = new GuiImage(key);
				keyImgOver[i][j] = new GuiImage(keyOver);
				keyTxt[i][j] = new GuiText(NULL, 20, (GXColor){0, 0, 0, 0xff});
				keyTxt[i][j]->SetAlignment(ALIGN_CENTRE, ALIGN_BOTTOM);
				keyTxt[i][j]->SetPosition(0, -10);
				keyBtn[i][j] = new GuiButton(key->GetWidth(), key->GetHeight());
				keyBtn[i][j]->SetImage(keyImg[i][j]);
				keyBtn[i][j]->SetImageOver(keyImgOver[i][j]);
				keyBtn[i][j]->SetSoundOver(keySoundOver);
				keyBtn[i][j]->SetSoundClick(keySoundClick);
				keyBtn[i][j]->SetTrigger(trigA);
				keyBtn[i][j]->SetLabel(keyTxt[i][j]);
				keyBtn[i][j]->SetPosition((j*42+21*i+40)+eurocheck, i*42+120);//SetPosition(j*42+21*i+40, i*42+80);
				keyBtn[i][j]->SetEffectGrow();
				this->Append(keyBtn[i][j]);
			}
		}
	}
}

/**
 * Destructor for the GuiKeyboard class.
 */
GuiKeyboard::~GuiKeyboard()
{
	delete kbText;
	delete keyTextbox;
	delete keyTextboxImg;
	delete keyCapsText;
	delete keyCapsImg;
	delete keyCapsOverImg;
	delete keyCaps;
	delete keyShiftText;
	delete keyShiftImg;
	delete keyShiftOverImg;
	delete keyShift;
	if (keyAlt)
	{delete keyAlt;}
	if (keyAlt2)
	{delete keyAlt2;}
	delete keyBackText;
	delete keyBackImg;
	delete keyBackOverImg;
	delete keyBack;
	delete keySpaceImg;
	delete keySpaceOverImg;
	delete keySpace;
	delete key;
	delete keyOver;
	delete keyMedium;
	delete keyMediumOver;
	delete keyLarge;
	delete keyLargeOver;
	delete keySoundOver;
	delete keySoundClick;
	delete trigA;
	delete trigB;

	for(int i=0; i<4; i++)
	{
		for(int j=0; j<11; j++)
		{
			if(keys[i][j].ch != '\0')
			{
				delete keyImg[i][j];
				delete keyImgOver[i][j];
				delete keyTxt[i][j];
				delete keyBtn[i][j];
			}
		}
	}
}

void GuiKeyboard::Update(GuiTrigger * t)
{
	LOCK(this);
	if(_elements.size() == 0 || (state == STATE_DISABLED && parentElement))
		return;

	for (u8 i = 0; i < _elements.size(); i++)
	{
		try	{ _elements.at(i)->Update(t); }
		catch (const std::exception& e) { }
	}

	if(keySpace->GetState() == STATE_CLICKED)
	{
		if(strlen(kbtextstr) < kbtextmaxlen)
		{
			kbtextstr[strlen(kbtextstr)] = ' ';
			kbText->SetText(kbtextstr);
		}
		keySpace->SetState(STATE_SELECTED, t->chan);
	}
	else if(keyBack->GetState() == STATE_CLICKED)
	{
		if (strlen(kbtextstr) >(m)){
		kbtextstr[strlen(kbtextstr)-1] = 0;
		kbText->SetText(kbtextstr);}
		keyBack->SetState(STATE_SELECTED, t->chan);
	}
	else if(keyClear->GetState() == STATE_CLICKED)
	{	clearMore:
		if (strlen(kbtextstr) >(m)){
		kbtextstr[strlen(kbtextstr)-1] = 0;
		kbText->SetText(kbtextstr);
		goto clearMore;}
		keyClear->SetState(STATE_SELECTED, t->chan);
	}
	else if(keyShift->GetState() == STATE_CLICKED)
	{
		shift ^= 1;
		if(alt) alt ^= 1;
		if(alt2) alt2 ^= 1;
		keyShift->SetState(STATE_SELECTED, t->chan);
	}
	else if(keyAlt->GetState() == STATE_CLICKED)
	{
		alt ^= 1;
		if(shift) shift ^= 1;
		if(alt2) alt2 ^= 1;
		keyAlt->SetState(STATE_SELECTED, t->chan);
	}
	else if(keyAlt2->GetState() == STATE_CLICKED)
	{
		alt2 ^= 1;
		if(shift) shift ^= 1;
		if(alt) alt ^= 1;
		keyAlt2->SetState(STATE_SELECTED, t->chan);
	}
	else if(keyCaps->GetState() == STATE_CLICKED)
	{
		caps ^= 1;
		keyCaps->SetState(STATE_SELECTED, t->chan);
	}

	char txt[2] = { 0, 0 };

	for(int i=0; i<4; i++)
	{
		for(int j=0; j<11; j++)
		{
			if(keys[i][j].ch != '\0')
			{
				if(shift || caps)
					txt[0] = keys[i][j].chShift;
				else if(alt)
					txt[0] = keys[i][j].chalt;
				else if(alt2)
					txt[0] = keys[i][j].chalt2;
				else
					txt[0] = keys[i][j].ch;

				keyTxt[i][j]->SetText(txt);

				if(keyBtn[i][j]->GetState() == STATE_CLICKED)
				{
					if(strlen(kbtextstr) < kbtextmaxlen)
					{
						if(shift || caps)
						{
							kbtextstr[strlen(kbtextstr)] = keys[i][j].chShift;
							if(shift) shift ^= 1;
							if(alt) alt ^= 1;
							if(alt2) alt2 ^= 1;
						}
						else if(alt)
						{
							kbtextstr[strlen(kbtextstr)] = keys[i][j].chalt;
							if(shift) shift ^= 1;
							if(alt) alt ^= 1;
							if(alt2) alt2 ^= 1;
						}
						else if(alt2)
						{
							kbtextstr[strlen(kbtextstr)] = keys[i][j].chalt2;
							if(shift) shift ^= 1;
							if(alt) alt ^= 1;
							if(alt2) alt2 ^= 1;
						}
						else
						{
							kbtextstr[strlen(kbtextstr)] = keys[i][j].ch;
						}
					}
					kbText->SetText(kbtextstr);
					keyBtn[i][j]->SetState(STATE_SELECTED, t->chan);
				}
			}
		}
	}
char* pch;

  pch=strrchr(kbtextstr,'_');

  if (pch!=NULL){kbText->SetPosition(0, 68);}
  else {kbText->SetPosition(0, 53);}

	this->ToggleFocus(t);

	if(focus) // only send actions to this window if it's in focus
	{
		// pad/joystick navigation
		if(t->Right())
			this->MoveSelectionHor(1);
		else if(t->Left())
			this->MoveSelectionHor(-1);
		else if(t->Down())
			this->MoveSelectionVert(1);
		else if(t->Up())
			this->MoveSelectionVert(-1);
	}
}