/* Copyright (c) 2002-2012 Croteam Ltd. All rights reserved. */

#include "StdH.h"
#include <Engine/Base/KeyNames.h>
#include <Engine/CurrentVersion.h>
#include <GameMP/LCDDrawing.h>
#include "LevelInfo.h"
#include "VarList.h"
#include "MGButton.h"

extern CSoundData *_psdPress;

extern CMenuGadget *_pmgLastActivatedGadget;


CMGButton::CMGButton(void)
{
	mg_pActivatedFunction = NULL;
	mg_iIndex = 0;
	mg_iCenterI = 0;
	mg_iTextMode = 1;
	mg_bfsFontSize = BFS_MEDIUM;
	mg_iCursorPos = -1;
	mg_bRectangle = FALSE;
	mg_bMental = FALSE;
	mg_bEditing = FALSE;
	mg_bHighlighted = FALSE;
}


void CMGButton::SetText(CTString strNew)
{
	mg_strText = strNew;
}


void CMGButton::OnActivate(void)
{
	if (mg_pActivatedFunction != NULL && mg_bEnabled)
	{
		PlayMenuSound(_psdPress);
		IFeel_PlayEffect("Menu_press");
		_pmgLastActivatedGadget = this;
		(*mg_pActivatedFunction)();
	}
}


void CMGButton::Render(CDrawPort *pdp)
{
	if (mg_bfsFontSize == BFS_LARGE) {
		SetFontBig(pdp);
	}
	else if (mg_bfsFontSize == BFS_MEDIUM) {
		SetFontMedium(pdp);
	}
	else {
		ASSERT(mg_bfsFontSize == BFS_SMALL);
		SetFontSmall(pdp);
	}
	pdp->SetTextMode(mg_iTextMode);

	PIXaabbox2D box = FloatBoxToPixBox(pdp, mg_boxOnScreen);
	COLOR col = GetCurrentColor();
	if (mg_bEditing) {
		col = LCDGetColor(C_GREEN | 0xFF, "editing");
	}

	COLOR colRectangle = col;
	if (mg_bHighlighted) {
		col = LCDGetColor(C_WHITE | 0xFF, "hilited");
		if (!mg_bFocused) {
			colRectangle = LCDGetColor(C_WHITE | 0xFF, "hilited rectangle");
		}
	}
	if (mg_bMental) {
		FLOAT tmIn = 0.2f;
		FLOAT tmOut = 1.0f;
		FLOAT tmFade = 0.1f;
		FLOAT tmExist = tmFade + tmIn + tmFade;
		FLOAT tmTotal = tmFade + tmIn + tmFade + tmOut;

		FLOAT tmTime = _pTimer->GetHighPrecisionTimer().GetSeconds();
		FLOAT fFactor = 1;
		if (tmTime>0.1f) {
			tmTime = fmod(tmTime, tmTotal);
			fFactor = CalculateRatio(tmTime, 0, tmExist, tmFade / tmExist, tmFade / tmExist);
		}
		col = (col&~0xFF) | INDEX(0xFF * fFactor);
	}

	if (mg_bRectangle) {
		// put border
		const PIX pixLeft = box.Min()(1);
		const PIX pixUp = box.Min()(2) - 3;
		const PIX pixWidth = box.Size()(1) + 1;
		const PIX pixHeight = box.Size()(2);
		pdp->DrawBorder(pixLeft, pixUp, pixWidth, pixHeight, colRectangle);
	}

	if (mg_bEditing) {
		// put border
		PIX pixLeft = box.Min()(1);
		PIX pixUp = box.Min()(2) - 3;
		PIX pixWidth = box.Size()(1) + 1;
		PIX pixHeight = box.Size()(2);
		if (mg_strLabel != "") {
			pixLeft = box.Min()(1) + box.Size()(1)*0.55f;
			pixWidth = box.Size()(1)*0.45f + 1;
		}
		pdp->Fill(pixLeft, pixUp, pixWidth, pixHeight, LCDGetColor(C_dGREEN | 0x40, "edit fill"));
	}


	INDEX iCursor = mg_iCursorPos;

	// print text
	if (mg_strLabel != "") {
		PIX pixIL = box.Min()(1) + box.Size()(1)*0.45f;
		PIX pixIR = box.Min()(1) + box.Size()(1)*0.55f;
		PIX pixJ = box.Min()(2);

		pdp->PutTextR(mg_strLabel, pixIL, pixJ, col);
		pdp->PutText(mg_strText, pixIR, pixJ, col);
	} else {
		CTString str = mg_strText;
		if (pdp->dp_FontData->fd_bFixedWidth) {
			str = str.Undecorated();
			INDEX iLen = str.Length();
			INDEX iMaxLen = ClampDn(box.Size()(1) / (pdp->dp_pixTextCharSpacing + pdp->dp_FontData->fd_pixCharWidth), 1L);
			if (iCursor >= iMaxLen) {
				str.TrimRight(iCursor);
				str.TrimLeft(iMaxLen);
				iCursor = iMaxLen;
			} else {
				str.TrimRight(iMaxLen);
			}
		}
		if (mg_iCenterI == -1) pdp->PutText(str, box.Min()(1), box.Min()(2), col);
		else if (mg_iCenterI == +1) pdp->PutTextR(str, box.Max()(1), box.Min()(2), col);
		else                      pdp->PutTextC(str, box.Center()(1), box.Min()(2), col);
	}

	// put cursor if editing
	if (mg_bEditing && (((ULONG)(_pTimer->GetRealTimeTick() * 2)) & 1)) {
		PIX pixX = box.Min()(1) + GetCharOffset(pdp, iCursor);
		if (mg_strLabel != "") {
			pixX += box.Size()(1)*0.55f;
		}
		PIX pixY = box.Min()(2);
		if (!pdp->dp_FontData->fd_bFixedWidth) {
			pixY -= pdp->dp_fTextScaling * 2;
		}
		pdp->PutText("|", pixX, pixY, LCDGetColor(C_WHITE | 0xFF, "editing cursor"));
	}
}


PIX CMGButton::GetCharOffset(CDrawPort *pdp, INDEX iCharNo)
{
	if (pdp->dp_FontData->fd_bFixedWidth) {
		return (pdp->dp_FontData->fd_pixCharWidth + pdp->dp_pixTextCharSpacing)*(iCharNo - 0.5f);
	}
	CTString strCut(mg_strText);
	strCut.TrimLeft(strlen(mg_strText) - iCharNo);
	PIX pixFullWidth = pdp->GetTextWidth(mg_strText);
	PIX pixCutWidth = pdp->GetTextWidth(strCut);
	// !!!! not implemented for different centering
	return pixFullWidth - pixCutWidth;
}