#include "Text.hpp"

Text::Text(const char * t, int s, GXColor c)
    : GuiText(t, s, c)
{
    maxWidth = 400;
    linestodraw = 9;
    curLineStart = 0;
    FirstLineOffset = 0;
    wText = NULL;

    if(!text)
        return;

    wText = new (std::nothrow) wString(text);
    if(!wText)
    {
        return;
    }

    if(wText->size() == 0)
    {
        wText->push_back(L' ');
        wText->push_back(0);
    }

    textWidth = (font ? font : fontSystem)->getWidth(wText->data(), currentSize);
    delete [] text;
    text = NULL;

    SetMaxWidth(maxWidth);
}

Text::Text(const wchar_t * t, int s, GXColor c)
    : GuiText((wchar_t *) NULL, s, c)
{
    maxWidth = 400;
    linestodraw = 9;
    curLineStart = 0;
    FirstLineOffset = 0;
    wText = NULL;

    if(!t)
        return;

    wText = new (std::nothrow) wString(t);
    if(!wText)
    {
        return;
    }

    if(wText->size() == 0)
    {
        wText->push_back(L' ');
        wText->push_back(0);
    }

    textWidth = (font ? font : fontSystem)->getWidth(wText->data(), currentSize);

    SetMaxWidth(maxWidth);
}

Text::~Text()
{
    if(wText)
        delete wText;
    wText = NULL;

    TextLines.clear();
    ClearDynamicText();
}

void Text::SetText(const char * t)
{
    wchar_t * tmp = charToWideChar(t);
    if(!tmp)
        return;

    if(wText)
        delete wText;

    wText = new (std::nothrow) wString(tmp);
    if(!wText)
    {
        return;
    }

    if(wText->size() == 0)
    {
        wText->push_back(L' ');
        wText->push_back(0);
    }

    textWidth = (font ? font : fontSystem)->getWidth(wText->data(), currentSize);

    delete [] tmp;

    ClearDynamicText();
    CalcLineOffsets();
}

void Text::SetText(const wchar_t * t)
{
    if(!t)
        return;

    if(wText)
        delete wText;

    wText = new wString(t);
    textWidth = (font ? font : fontSystem)->getWidth(wText->data(), currentSize);
    CalcLineOffsets();
}

void Text::SetMaxWidth(int w)
{
    maxWidth = w;
    curLineStart = 0;
    Refresh();
}

void Text::SetTextLine(int line)
{
	if(line < 0)
		line = 0;
	else if(line > (int) TextLines.size()-1)
		line = TextLines.size()-1;

    curLineStart = line;

    FillRows();

    while((int) textDyn.size() < linestodraw && curLineStart > 0)
    {
        PreviousLine();
    }
}

void Text::SetTextPos(int pos)
{
    if(!wText)
        return;

    int diff = 10000;

    for(u32 i = 0; i < TextLines.size(); i++)
    {
        int curDiff = abs(TextLines[i].LineOffset - pos);
        if(curDiff < diff)
        {
            diff = curDiff;
            curLineStart = i;
        }
    }

    FillRows();

    while((int) textDyn.size() < linestodraw && curLineStart > 0)
    {
        PreviousLine();
    }
}

const wchar_t * Text::GetText()
{
    return wText->c_str();
}

std::string Text::GetUTF8String(void) const
{
    return wText->toUTF8();
}

int Text::GetLineOffset(int ind)
{
    if(TextLines.size() == 0)
        return 0;

    if(ind < 0)
        return TextLines[0].LineOffset;

    if(ind >= (int) TextLines.size()-1)
        return TextLines[TextLines.size()-1].LineOffset;

    return TextLines[ind].LineOffset;
}

const wchar_t * Text::GetTextLine(int ind)
{
    if(filling || textDyn.size() == 0)
        return NULL;

    if(ind < 0)
        return textDyn[0];

    if(ind >= (int) textDyn.size())
        return textDyn[textDyn.size()-1];

    return textDyn[ind];
}

void Text::Refresh()
{
    CalcLineOffsets();
    FillRows();
}

void Text::NextLine()
{
    if(!wText || (curLineStart+1 > ((int) TextLines.size()-linestodraw)))
        return;

    ++curLineStart;

    FillRows();
}

void Text::PreviousLine()
{
    if(!wText || curLineStart-1 < 0)
        return;

    --curLineStart;

    FillRows();
}

void Text::FillRows()
{
    if(!wText)
        return;

    filling = true;

    ClearDynamicText();

    for(int i = 0; i < linestodraw && i < (int) TextLines.size(); i++)
    {
        if(i >= (int) textDyn.size())
        {
			textDyn.resize(i+1);
            textDyn[i] = new wchar_t[maxWidth];
        }
        int offset = TextLines[curLineStart+i].LineOffset;
        int count = TextLines[curLineStart+i].CharCount+1;

        for(int n = 0; n < count && offset+n < (int) wText->size(); n++)
            textDyn[i][n] = wText->at(offset+n);

        textDyn[i][count] = 0;
    }

    filling = false;

    return;
}

void Text::CalcLineOffsets()
{
    if(!wText)
        return;

    TextLines.clear();

    TextLine TmpLine;
    TmpLine.CharCount = 0;
    TmpLine.LineOffset = 0;
    TmpLine.width = 0;

    const wchar_t * origTxt = wText->c_str();
    int ch = 0;
    int lastSpace = -1;
    int lastSpaceIndex = -1;
    int currWidth = 0;
    int i = 0;

    while(origTxt[ch])
    {
        currWidth += fontSystem->getCharWidth(origTxt[ch], currentSize, ch > 0 ? origTxt[ch-1] : 0x0000);

        if(currWidth >= maxWidth)
        {
            if(lastSpace > 0)
            {
                ch = lastSpace;
            }
            TmpLine.CharCount = ch-TmpLine.LineOffset;
            TmpLine.width = currWidth;
            TextLines.push_back(TmpLine);
            currWidth = 0;
            lastSpace = -1;
            i = -1;
            TmpLine.LineOffset = ch+1;
        }
        else if(origTxt[ch] == '\n')
        {
            TmpLine.CharCount = ch-TmpLine.LineOffset;
            TmpLine.width = currWidth;
            TextLines.push_back(TmpLine);
            currWidth = 0;
            lastSpace = -1;
            i = -1;
            TmpLine.LineOffset = ch+1;
        }
        else if(origTxt[ch] == ' ')
        {
            lastSpace = ch;
            lastSpaceIndex = i;
        }

        ch++;
        i++;
    }

    TmpLine.CharCount = ch-TmpLine.LineOffset;
    TmpLine.width = currWidth;

    if(TmpLine.CharCount > 0)
        TextLines.push_back(TmpLine);
}

void Text::Draw()
{
    if(textDyn.size() == 0)
        return;

	if(!this->IsVisible())
		return;

	GXColor c = color;
	c.a = this->GetAlpha();

	int newSize = size*GetScale();

	if(newSize != currentSize)
	{
		currentSize = newSize;

        if(wText)
            textWidth = (font ? font : fontSystem)->getWidth(wText->data(), currentSize);
	}

    u16 lineheight = newSize + 6;

    for(u32 i = 0; i < textDyn.size(); i++)
    {
        if(!filling)
            (font ? font : fontSystem)->drawText(this->GetLeft(), this->GetTop()+i*lineheight, 0, textDyn[i], currentSize, c, style, 0, maxWidth);
    }
}