/* ScummVM - Graphic Adventure Engine
 *
 * ScummVM is the legal property of its developers, whose names
 * are too numerous to list here. Please refer to the COPYRIGHT
 * file distributed with this source distribution.
 *
 * 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 2
 * 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 */

/*
 * This file is based on WME Lite.
 * http://dead-code.org/redir.php?target=wmelite
 * Copyright (c) 2011 Jan Nedoma
 */

#include "engines/wintermute/UIEdit.h"
#include "engines/wintermute/UIObject.h"
#include "engines/wintermute/UITiledImage.h"
#include "engines/wintermute/StringUtil.h"
#include "engines/wintermute/BActiveRect.h"
#include "engines/wintermute/BFileManager.h"
#include "engines/wintermute/BFont.h"
#include "engines/wintermute/BFontStorage.h"
#include "engines/wintermute/BKeyboardState.h"
#include "engines/wintermute/BDynBuffer.h"
#include "engines/wintermute/BParser.h"
#include "engines/wintermute/BSprite.h"
#include "engines/wintermute/BStringTable.h"
#include "engines/wintermute/BGame.h"
#include "engines/wintermute/PlatformSDL.h"
#include "engines/wintermute/scriptables/ScValue.h"
#include "engines/wintermute/scriptables/ScStack.h"
#include "engines/wintermute/scriptables/ScScript.h"
#include "engines/wintermute/utils.h"
#include "common/util.h"

namespace WinterMute {

IMPLEMENT_PERSISTENT(CUIEdit, false)

//////////////////////////////////////////////////////////////////////////
CUIEdit::CUIEdit(CBGame *inGame): CUIObject(inGame) {
	m_Type = UI_EDIT;

	m_FontSelected = NULL;

	m_SelStart = m_SelEnd = 10000;
	m_ScrollOffset = 0;

	m_CursorChar = NULL;
	SetCursorChar("|");

#ifdef __WIN32__
	m_CursorBlinkRate = GetCaretBlinkTime();
#else
	m_CursorBlinkRate = 600;
#endif
	m_FrameWidth = 0;

	SetText("");

	m_LastBlinkTime = 0;
	m_CursorVisible = true;

	m_MaxLength = -1;

	m_CanFocus = true;
}


//////////////////////////////////////////////////////////////////////////
CUIEdit::~CUIEdit() {
	if (!m_SharedFonts) {
		if (m_FontSelected)   Game->m_FontStorage->RemoveFont(m_FontSelected);
	}

	delete[] m_CursorChar;
	m_CursorChar = NULL;
}


//////////////////////////////////////////////////////////////////////////
HRESULT CUIEdit::LoadFile(char *Filename) {
	byte *Buffer = Game->m_FileManager->ReadWholeFile(Filename);
	if (Buffer == NULL) {
		Game->LOG(0, "CUIEdit::LoadFile failed for file '%s'", Filename);
		return E_FAIL;
	}

	HRESULT ret;

	m_Filename = new char [strlen(Filename) + 1];
	strcpy(m_Filename, Filename);

	if (FAILED(ret = LoadBuffer(Buffer, true))) Game->LOG(0, "Error parsing EDIT file '%s'", Filename);

	delete [] Buffer;

	return ret;
}


TOKEN_DEF_START
TOKEN_DEF(TEMPLATE)
TOKEN_DEF(DISABLED)
TOKEN_DEF(VISIBLE)
TOKEN_DEF(BACK)
TOKEN_DEF(IMAGE)
TOKEN_DEF(FONT_SELECTED)
TOKEN_DEF(FONT)
TOKEN_DEF(TEXT)
TOKEN_DEF(X)
TOKEN_DEF(Y)
TOKEN_DEF(WIDTH)
TOKEN_DEF(HEIGHT)
TOKEN_DEF(CURSOR_BLINK_RATE)
TOKEN_DEF(CURSOR)
TOKEN_DEF(FRAME_WIDTH)
TOKEN_DEF(NAME)
TOKEN_DEF(SCRIPT)
TOKEN_DEF(PARENT_NOTIFY)
TOKEN_DEF(MAX_LENGTH)
TOKEN_DEF(EDITOR_PROPERTY)
TOKEN_DEF(EDIT)
TOKEN_DEF(CAPTION)
TOKEN_DEF_END
//////////////////////////////////////////////////////////////////////////
HRESULT CUIEdit::LoadBuffer(byte  *Buffer, bool Complete) {
	TOKEN_TABLE_START(commands)
	TOKEN_TABLE(TEMPLATE)
	TOKEN_TABLE(DISABLED)
	TOKEN_TABLE(VISIBLE)
	TOKEN_TABLE(BACK)
	TOKEN_TABLE(IMAGE)
	TOKEN_TABLE(FONT_SELECTED)
	TOKEN_TABLE(FONT)
	TOKEN_TABLE(TEXT)
	TOKEN_TABLE(X)
	TOKEN_TABLE(Y)
	TOKEN_TABLE(WIDTH)
	TOKEN_TABLE(HEIGHT)
	TOKEN_TABLE(CURSOR_BLINK_RATE)
	TOKEN_TABLE(CURSOR)
	TOKEN_TABLE(FRAME_WIDTH)
	TOKEN_TABLE(NAME)
	TOKEN_TABLE(SCRIPT)
	TOKEN_TABLE(PARENT_NOTIFY)
	TOKEN_TABLE(MAX_LENGTH)
	TOKEN_TABLE(EDITOR_PROPERTY)
	TOKEN_TABLE(EDIT)
	TOKEN_TABLE(CAPTION)
	TOKEN_TABLE_END

	byte *params;
	int cmd = 2;
	CBParser parser(Game);

	if (Complete) {
		if (parser.GetCommand((char **)&Buffer, commands, (char **)&params) != TOKEN_EDIT) {
			Game->LOG(0, "'EDIT' keyword expected.");
			return E_FAIL;
		}
		Buffer = params;
	}

	while (cmd > 0 && (cmd = parser.GetCommand((char **)&Buffer, commands, (char **)&params)) > 0) {
		switch (cmd) {
		case TOKEN_TEMPLATE:
			if (FAILED(LoadFile((char *)params))) cmd = PARSERR_GENERIC;
			break;

		case TOKEN_NAME:
			SetName((char *)params);
			break;

		case TOKEN_BACK:
			delete m_Back;
			m_Back = new CUITiledImage(Game);
			if (!m_Back || FAILED(m_Back->LoadFile((char *)params))) {
				delete m_Back;
				m_Back = NULL;
				cmd = PARSERR_GENERIC;
			}
			break;

		case TOKEN_IMAGE:
			delete m_Image;
			m_Image = new CBSprite(Game);
			if (!m_Image || FAILED(m_Image->LoadFile((char *)params))) {
				delete m_Image;
				m_Image = NULL;
				cmd = PARSERR_GENERIC;
			}
			break;

		case TOKEN_FONT:
			if (m_Font) Game->m_FontStorage->RemoveFont(m_Font);
			m_Font = Game->m_FontStorage->AddFont((char *)params);
			if (!m_Font) cmd = PARSERR_GENERIC;
			break;

		case TOKEN_FONT_SELECTED:
			if (m_FontSelected) Game->m_FontStorage->RemoveFont(m_FontSelected);
			m_FontSelected = Game->m_FontStorage->AddFont((char *)params);
			if (!m_FontSelected) cmd = PARSERR_GENERIC;
			break;

		case TOKEN_TEXT:
			SetText((char *)params);
			Game->m_StringTable->Expand(&m_Text);
			break;

		case TOKEN_X:
			parser.ScanStr((char *)params, "%d", &m_PosX);
			break;

		case TOKEN_Y:
			parser.ScanStr((char *)params, "%d", &m_PosY);
			break;

		case TOKEN_WIDTH:
			parser.ScanStr((char *)params, "%d", &m_Width);
			break;

		case TOKEN_HEIGHT:
			parser.ScanStr((char *)params, "%d", &m_Height);
			break;

		case TOKEN_MAX_LENGTH:
			parser.ScanStr((char *)params, "%d", &m_MaxLength);
			break;

		case TOKEN_CAPTION:
			SetCaption((char *)params);
			break;

		case TOKEN_CURSOR:
			delete m_Cursor;
			m_Cursor = new CBSprite(Game);
			if (!m_Cursor || FAILED(m_Cursor->LoadFile((char *)params))) {
				delete m_Cursor;
				m_Cursor = NULL;
				cmd = PARSERR_GENERIC;
			}
			break;

		case TOKEN_CURSOR_BLINK_RATE:
			parser.ScanStr((char *)params, "%d", &m_CursorBlinkRate);
			break;

		case TOKEN_FRAME_WIDTH:
			parser.ScanStr((char *)params, "%d", &m_FrameWidth);
			break;

		case TOKEN_SCRIPT:
			AddScript((char *)params);
			break;

		case TOKEN_PARENT_NOTIFY:
			parser.ScanStr((char *)params, "%b", &m_ParentNotify);
			break;

		case TOKEN_DISABLED:
			parser.ScanStr((char *)params, "%b", &m_Disable);
			break;

		case TOKEN_VISIBLE:
			parser.ScanStr((char *)params, "%b", &m_Visible);
			break;

		case TOKEN_EDITOR_PROPERTY:
			ParseEditorProperty(params, false);
			break;
		}
	}
	if (cmd == PARSERR_TOKENNOTFOUND) {
		Game->LOG(0, "Syntax error in EDIT definition");
		return E_FAIL;
	}
	if (cmd == PARSERR_GENERIC) {
		Game->LOG(0, "Error loading EDIT definition");
		return E_FAIL;
	}

	CorrectSize();

	return S_OK;
}

//////////////////////////////////////////////////////////////////////////
HRESULT CUIEdit::SaveAsText(CBDynBuffer *Buffer, int Indent) {
	Buffer->PutTextIndent(Indent, "EDIT\n");
	Buffer->PutTextIndent(Indent, "{\n");

	Buffer->PutTextIndent(Indent + 2, "NAME=\"%s\"\n", m_Name);
	Buffer->PutTextIndent(Indent + 2, "CAPTION=\"%s\"\n", GetCaption());

	Buffer->PutTextIndent(Indent + 2, "\n");

	if (m_Back && m_Back->m_Filename)
		Buffer->PutTextIndent(Indent + 2, "BACK=\"%s\"\n", m_Back->m_Filename);

	if (m_Image && m_Image->m_Filename)
		Buffer->PutTextIndent(Indent + 2, "IMAGE=\"%s\"\n", m_Image->m_Filename);

	if (m_Font && m_Font->m_Filename)
		Buffer->PutTextIndent(Indent + 2, "FONT=\"%s\"\n", m_Font->m_Filename);
	if (m_FontSelected && m_FontSelected->m_Filename)
		Buffer->PutTextIndent(Indent + 2, "FONT_SELECTED=\"%s\"\n", m_FontSelected->m_Filename);

	if (m_Cursor && m_Cursor->m_Filename)
		Buffer->PutTextIndent(Indent + 2, "CURSOR=\"%s\"\n", m_Cursor->m_Filename);

	Buffer->PutTextIndent(Indent + 2, "\n");

	if (m_Text)
		Buffer->PutTextIndent(Indent + 2, "TEXT=\"%s\"\n", m_Text);

	Buffer->PutTextIndent(Indent + 2, "\n");

	Buffer->PutTextIndent(Indent + 2, "X=%d\n", m_PosX);
	Buffer->PutTextIndent(Indent + 2, "Y=%d\n", m_PosY);
	Buffer->PutTextIndent(Indent + 2, "WIDTH=%d\n", m_Width);
	Buffer->PutTextIndent(Indent + 2, "HEIGHT=%d\n", m_Height);
	Buffer->PutTextIndent(Indent + 2, "MAX_LENGTH=%d\n", m_MaxLength);
	Buffer->PutTextIndent(Indent + 2, "CURSOR_BLINK_RATE=%d\n", m_CursorBlinkRate);
	Buffer->PutTextIndent(Indent + 2, "FRAME_WIDTH=%d\n", m_FrameWidth);

	Buffer->PutTextIndent(Indent + 2, "DISABLED=%s\n", m_Disable ? "TRUE" : "FALSE");
	Buffer->PutTextIndent(Indent + 2, "VISIBLE=%s\n", m_Visible ? "TRUE" : "FALSE");
	Buffer->PutTextIndent(Indent + 2, "PARENT_NOTIFY=%s\n", m_ParentNotify ? "TRUE" : "FALSE");

	// scripts
	for (int i = 0; i < m_Scripts.GetSize(); i++) {
		Buffer->PutTextIndent(Indent + 2, "SCRIPT=\"%s\"\n", m_Scripts[i]->m_Filename);
	}

	Buffer->PutTextIndent(Indent + 2, "\n");

	// editor properties
	CBBase::SaveAsText(Buffer, Indent + 2);

	Buffer->PutTextIndent(Indent, "}\n");
	return S_OK;
}

//////////////////////////////////////////////////////////////////////////
// high level scripting interface
//////////////////////////////////////////////////////////////////////////
HRESULT CUIEdit::ScCallMethod(CScScript *Script, CScStack *Stack, CScStack *ThisStack, char *Name) {
	//////////////////////////////////////////////////////////////////////////
	// SetSelectedFont
	//////////////////////////////////////////////////////////////////////////
	if (strcmp(Name, "SetSelectedFont") == 0) {
		Stack->CorrectParams(1);

		if (m_FontSelected) Game->m_FontStorage->RemoveFont(m_FontSelected);
		m_FontSelected = Game->m_FontStorage->AddFont(Stack->Pop()->GetString());
		Stack->PushBool(m_FontSelected != NULL);

		return S_OK;
	}

	else return CUIObject::ScCallMethod(Script, Stack, ThisStack, Name);
}


//////////////////////////////////////////////////////////////////////////
CScValue *CUIEdit::ScGetProperty(char *Name) {
	m_ScValue->SetNULL();

	//////////////////////////////////////////////////////////////////////////
	// Type
	//////////////////////////////////////////////////////////////////////////
	if (strcmp(Name, "Type") == 0) {
		m_ScValue->SetString("editor");
		return m_ScValue;
	}

	//////////////////////////////////////////////////////////////////////////
	// SelStart
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "SelStart") == 0) {
		m_ScValue->SetInt(m_SelStart);
		return m_ScValue;
	}

	//////////////////////////////////////////////////////////////////////////
	// SelEnd
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "SelEnd") == 0) {
		m_ScValue->SetInt(m_SelEnd);
		return m_ScValue;
	}

	//////////////////////////////////////////////////////////////////////////
	// CursorBlinkRate
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "CursorBlinkRate") == 0) {
		m_ScValue->SetInt(m_CursorBlinkRate);
		return m_ScValue;
	}

	//////////////////////////////////////////////////////////////////////////
	// CursorChar
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "CursorChar") == 0) {
		m_ScValue->SetString(m_CursorChar);
		return m_ScValue;
	}

	//////////////////////////////////////////////////////////////////////////
	// FrameWidth
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "FrameWidth") == 0) {
		m_ScValue->SetInt(m_FrameWidth);
		return m_ScValue;
	}

	//////////////////////////////////////////////////////////////////////////
	// MaxLength
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "MaxLength") == 0) {
		m_ScValue->SetInt(m_MaxLength);
		return m_ScValue;
	}

	//////////////////////////////////////////////////////////////////////////
	// Text
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "Text") == 0) {
		if (Game->m_TextEncoding == TEXT_UTF8) {
			WideString wstr = StringUtil::AnsiToWide(m_Text);
			m_ScValue->SetString(StringUtil::WideToUtf8(wstr).c_str());
		} else {
			m_ScValue->SetString(m_Text);
		}
		return m_ScValue;
	}

	else return CUIObject::ScGetProperty(Name);
}


//////////////////////////////////////////////////////////////////////////
HRESULT CUIEdit::ScSetProperty(char *Name, CScValue *Value) {
	//////////////////////////////////////////////////////////////////////////
	// SelStart
	//////////////////////////////////////////////////////////////////////////
	if (strcmp(Name, "SelStart") == 0) {
		m_SelStart = Value->GetInt();
		m_SelStart = MAX(m_SelStart, 0);
		m_SelStart = MIN((size_t)m_SelStart, strlen(m_Text));
		return S_OK;
	}

	//////////////////////////////////////////////////////////////////////////
	// SelEnd
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "SelEnd") == 0) {
		m_SelEnd = Value->GetInt();
		m_SelEnd = MAX(m_SelEnd, 0);
		m_SelEnd = MIN((size_t)m_SelEnd, strlen(m_Text));
		return S_OK;
	}

	//////////////////////////////////////////////////////////////////////////
	// CursorBlinkRate
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "CursorBlinkRate") == 0) {
		m_CursorBlinkRate = Value->GetInt();
		return S_OK;
	}

	//////////////////////////////////////////////////////////////////////////
	// CursorChar
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "CursorChar") == 0) {
		SetCursorChar(Value->GetString());
		return S_OK;
	}

	//////////////////////////////////////////////////////////////////////////
	// FrameWidth
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "FrameWidth") == 0) {
		m_FrameWidth = Value->GetInt();
		return S_OK;
	}

	//////////////////////////////////////////////////////////////////////////
	// MaxLength
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "MaxLength") == 0) {
		m_MaxLength = Value->GetInt();
		return S_OK;
	}

	//////////////////////////////////////////////////////////////////////////
	// Text
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "Text") == 0) {
		if (Game->m_TextEncoding == TEXT_UTF8) {
			WideString wstr = StringUtil::Utf8ToWide(Value->GetString());
			SetText(StringUtil::WideToAnsi(wstr).c_str());
		} else {
			SetText(Value->GetString());
		}
		return S_OK;
	}

	else return CUIObject::ScSetProperty(Name, Value);
}


//////////////////////////////////////////////////////////////////////////
char *CUIEdit::ScToString() {
	return "[edit]";
}


//////////////////////////////////////////////////////////////////////////
void CUIEdit::SetCursorChar(char *Char) {
	if (!Char) return;
	delete[] m_CursorChar;
	m_CursorChar = new char [strlen(Char) + 1];
	if (m_CursorChar) strcpy(m_CursorChar, Char);
}


//////////////////////////////////////////////////////////////////////////
HRESULT CUIEdit::Display(int OffsetX, int OffsetY) {
	if (!m_Visible) return S_OK;


	// hack!
	TTextEncoding OrigEncoding = Game->m_TextEncoding;
	Game->m_TextEncoding = TEXT_ANSI;

	if (m_Back) m_Back->Display(OffsetX + m_PosX, OffsetY + m_PosY, m_Width, m_Height);
	if (m_Image) m_Image->Draw(OffsetX + m_PosX, OffsetY + m_PosY, NULL);

	// prepare fonts
	CBFont *font;
	CBFont *sfont;

	if (m_Font) font = m_Font;
	else font = Game->m_SystemFont;

	if (m_FontSelected) sfont = m_FontSelected;
	else sfont = font;

	bool focused = IsFocused();

	m_SelStart = MAX(m_SelStart, 0);
	m_SelEnd   = MAX(m_SelEnd, 0);

	m_SelStart = MIN((size_t)m_SelStart, strlen(m_Text));
	m_SelEnd   = MIN((size_t)m_SelEnd,   strlen(m_Text));

	//int CursorWidth = font->GetCharWidth(m_CursorChar[0]);
	int CursorWidth = font->GetTextWidth((byte  *)m_CursorChar);

	int s1, s2;
	bool CurFirst;
	// modify scroll offset
	if (m_SelStart >= m_SelEnd) {
		while (font->GetTextWidth((byte  *)m_Text + m_ScrollOffset, std::max(0, m_SelEnd - m_ScrollOffset)) > m_Width - CursorWidth - 2 * m_FrameWidth) {
			m_ScrollOffset++;
			if (m_ScrollOffset >= strlen(m_Text)) break;
		}

		m_ScrollOffset = std::min(m_ScrollOffset, m_SelEnd);

		s1 = m_SelEnd;
		s2 = m_SelStart;
		CurFirst = true;
	} else {
		while (font->GetTextWidth((byte  *)m_Text + m_ScrollOffset, std::max(0, m_SelStart - m_ScrollOffset)) +
		        sfont->GetTextWidth((byte  *)(m_Text + std::max(m_ScrollOffset, m_SelStart)), m_SelEnd - std::max(m_ScrollOffset, m_SelStart))

		        > m_Width - CursorWidth - 2 * m_FrameWidth) {
			m_ScrollOffset++;
			if (m_ScrollOffset >= strlen(m_Text)) break;
		}

		m_ScrollOffset = std::min(m_ScrollOffset, m_SelEnd);

		s1 = m_SelStart;
		s2 = m_SelEnd;
		CurFirst = false;
	}


	int AlignOffset = 0;

	for (int Count = 0; Count < 2; Count++) {
		// draw text
		int xxx, yyy, width, height;

		xxx = m_PosX + m_FrameWidth + OffsetX;
		yyy = m_PosY + m_FrameWidth + OffsetY;

		width = m_PosX + m_Width + OffsetX - m_FrameWidth;
		height = std::max(font->GetLetterHeight(), sfont->GetLetterHeight());

		if (Game->m_TextRTL) xxx += AlignOffset;

		TTextAlign Align = TAL_LEFT;


		// unselected 1
		if (s1 > m_ScrollOffset) {
			if (Count) font->DrawText((byte  *)m_Text + m_ScrollOffset, xxx, yyy, width - xxx, Align, height, s1 - m_ScrollOffset);
			xxx += font->GetTextWidth((byte  *)m_Text + m_ScrollOffset, s1 - m_ScrollOffset);
			AlignOffset += font->GetTextWidth((byte  *)m_Text + m_ScrollOffset, s1 - m_ScrollOffset);
		}

		// cursor
		if (focused && CurFirst) {
			if (Count) {
				if (CBPlatform::GetTime() - m_LastBlinkTime >= m_CursorBlinkRate) {
					m_LastBlinkTime = CBPlatform::GetTime();
					m_CursorVisible = !m_CursorVisible;
				}
				if (m_CursorVisible)
					font->DrawText((byte  *)m_CursorChar, xxx, yyy, width - xxx, Align, height, 1);
			}
			xxx += CursorWidth;
			AlignOffset += CursorWidth;
		}

		// selected
		int s3 = std::max(s1, m_ScrollOffset);

		if (s2 - s3 > 0) {
			if (Count) sfont->DrawText((byte  *)m_Text + s3, xxx, yyy, width - xxx, Align, height, s2 - s3);
			xxx += sfont->GetTextWidth((byte  *)m_Text + s3, s2 - s3);
			AlignOffset += sfont->GetTextWidth((byte  *)m_Text + s3, s2 - s3);
		}

		// cursor
		if (focused && !CurFirst) {
			if (Count) {
				if (CBPlatform::GetTime() - m_LastBlinkTime >= m_CursorBlinkRate) {
					m_LastBlinkTime = CBPlatform::GetTime();
					m_CursorVisible = !m_CursorVisible;
				}
				if (m_CursorVisible)
					font->DrawText((byte  *)m_CursorChar, xxx, yyy, width - xxx, Align, height, 1);
			}
			xxx += CursorWidth;
			AlignOffset += CursorWidth;
		}

		// unselected 2
		if (Count) font->DrawText((byte  *)m_Text + s2, xxx, yyy, width - xxx, Align, height);
		AlignOffset += font->GetTextWidth((byte  *)m_Text + s2);

		AlignOffset = (m_Width - 2 * m_FrameWidth) - AlignOffset;
		if (AlignOffset < 0) AlignOffset = 0;
	}


	Game->m_Renderer->m_RectList.Add(new CBActiveRect(Game, this, NULL, OffsetX + m_PosX, OffsetY + m_PosY, m_Width, m_Height, 100, 100, false));


	Game->m_TextEncoding = OrigEncoding;

	return S_OK;
}


//////////////////////////////////////////////////////////////////////////
bool CUIEdit::HandleKeypress(SDL_Event *event) {
	bool Handled = false;

	if (event->type == SDL_KEYDOWN) {
		switch (event->key.keysym.sym) {
		case SDLK_ESCAPE:
		case SDLK_TAB:
		case SDLK_RETURN:
			return false;

			// ctrl+A
		case SDLK_a:
			if (CBKeyboardState::IsControlDown()) {
				m_SelStart = 0;
				m_SelEnd = strlen(m_Text);
				Handled = true;
			}
			break;

		case SDLK_BACKSPACE:
			if (m_SelStart == m_SelEnd) {
				if (Game->m_TextRTL) DeleteChars(m_SelStart, m_SelStart + 1);
				else DeleteChars(m_SelStart - 1, m_SelStart);
			} else DeleteChars(m_SelStart, m_SelEnd);
			if (m_SelEnd >= m_SelStart) m_SelEnd -= std::max(1, m_SelEnd - m_SelStart);
			m_SelStart = m_SelEnd;

			Handled = true;
			break;

		case SDLK_LEFT:
		case SDLK_UP:
			m_SelEnd--;
			if (!CBKeyboardState::IsShiftDown()) m_SelStart = m_SelEnd;
			Handled = true;
			break;

		case SDLK_RIGHT:
		case SDLK_DOWN:
			m_SelEnd++;
			if (!CBKeyboardState::IsShiftDown()) m_SelStart = m_SelEnd;
			Handled = true;
			break;

		case SDLK_HOME:
			if (Game->m_TextRTL) {
				m_SelEnd = strlen(m_Text);
				if (!CBKeyboardState::IsShiftDown()) m_SelStart = m_SelEnd;
			} else {
				m_SelEnd = 0;
				if (!CBKeyboardState::IsShiftDown()) m_SelStart = m_SelEnd;
			}
			Handled = true;
			break;

		case SDLK_END:
			if (Game->m_TextRTL) {
				m_SelEnd = 0;
				if (!CBKeyboardState::IsShiftDown()) m_SelStart = m_SelEnd;
			} else {
				m_SelEnd = strlen(m_Text);
				if (!CBKeyboardState::IsShiftDown()) m_SelStart = m_SelEnd;
			}
			Handled = true;
			break;

		case SDLK_DELETE:
			if (m_SelStart == m_SelEnd) {
				if (Game->m_TextRTL) {
					DeleteChars(m_SelStart - 1, m_SelStart);
					m_SelEnd--;
					if (m_SelEnd < 0) m_SelEnd = 0;
				} else DeleteChars(m_SelStart, m_SelStart + 1);
			} else DeleteChars(m_SelStart, m_SelEnd);
			if (m_SelEnd > m_SelStart) m_SelEnd -= (m_SelEnd - m_SelStart);

			m_SelStart = m_SelEnd;
			Handled = true;
			break;
		}
		return Handled;
	}
#if 0
	else if (event->type == SDL_TEXTINPUT) {
		if (m_SelStart != m_SelEnd) DeleteChars(m_SelStart, m_SelEnd);

		WideString wstr = StringUtil::Utf8ToWide(event->text.text);
		m_SelEnd += InsertChars(m_SelEnd, (byte  *)StringUtil::WideToAnsi(wstr).c_str(), 1);

		if (Game->m_TextRTL) m_SelEnd = m_SelStart;
		else m_SelStart = m_SelEnd;

		return true;
	}
#endif
	return false;
}



//////////////////////////////////////////////////////////////////////////
int CUIEdit::DeleteChars(int Start, int End) {
	if (Start > End) CBUtils::Swap(&Start, &End);

	Start = MAX(Start, (int)0);
	End = MIN((size_t)End, strlen(m_Text));

	char *str = new char[strlen(m_Text) - (End - Start) + 1];
	if (str) {
		if (Start > 0) memcpy(str, m_Text, Start);
		memcpy(str + std::max(0, Start), m_Text + End, strlen(m_Text) - End + 1);

		delete[] m_Text;
		m_Text = str;
	}
	if (m_ParentNotify && m_Parent) m_Parent->ApplyEvent(m_Name);

	return End - Start;
}


//////////////////////////////////////////////////////////////////////////
int CUIEdit::InsertChars(int Pos, byte *Chars, int Num) {
	if (strlen(m_Text) + Num > m_MaxLength) {
		Num -= (strlen(m_Text) + Num - m_MaxLength);
	}

	Pos = MAX(Pos, (int)0);
	Pos = MIN((size_t)Pos, strlen(m_Text));

	char *str = new char[strlen(m_Text) + Num + 1];
	if (str) {
		if (Pos > 0) memcpy(str, m_Text, Pos);
		memcpy(str + Pos + Num, m_Text + Pos, strlen(m_Text) - Pos + 1);

		memcpy(str + Pos, Chars, Num);

		delete[] m_Text;
		m_Text = str;
	}
	if (m_ParentNotify && m_Parent) m_Parent->ApplyEvent(m_Name);

	return Num;
}



//////////////////////////////////////////////////////////////////////////
HRESULT CUIEdit::Persist(CBPersistMgr *PersistMgr) {

	CUIObject::Persist(PersistMgr);

	PersistMgr->Transfer(TMEMBER(m_CursorBlinkRate));
	PersistMgr->Transfer(TMEMBER(m_CursorChar));
	PersistMgr->Transfer(TMEMBER(m_FontSelected));
	PersistMgr->Transfer(TMEMBER(m_FrameWidth));
	PersistMgr->Transfer(TMEMBER(m_MaxLength));
	PersistMgr->Transfer(TMEMBER(m_ScrollOffset));
	PersistMgr->Transfer(TMEMBER(m_SelEnd));
	PersistMgr->Transfer(TMEMBER(m_SelStart));

	if (!PersistMgr->m_Saving) {
		m_CursorVisible = false;
		m_LastBlinkTime = 0;
	}

	return S_OK;
}

} // end of namespace WinterMute