#include "global.h"
#include "render_text.h"
#include "game_init.h"
#include "string_util.h"
#include "text.h"
#include "unk_0201B8B8.h"
#include "font.h"

u16 unk00;

TextFlags gTextFlags;

const u8 UNK_020ECB50[] = { 0, 1, 2, 1 };

extern void PlaySE(u16);

THUMB_FUNC u32 RenderText(struct TextPrinter *printer)
{

    struct TextPrinterSubStruct *subStruct =
        (struct TextPrinterSubStruct *)(&printer->subStructFields);
    u16 currentChar;

    switch (printer->state)
    {
        case 0:
            if (((gMain.heldKeys & 3) != 0 && subStruct->hasPrintBeenSpedUp != 0) ||
                (gMain.touchHeld != 0 && gTextFlags.unk0_4 != 0))
            {
                printer->delayCounter = 0;
                if (printer->textSpeedBottom != 0)
                {
                    gTextFlags.unk0_6 = 1;
                }
            }

            if (printer->delayCounter && printer->textSpeedBottom)
            {
                printer->delayCounter--;

                if (gTextFlags.canABSpeedUpPrint != 0)
                {
                    if ((gMain.newKeys & 3) || (gMain.touchNew != 0 && gTextFlags.unk0_4))
                    {
                        subStruct->hasPrintBeenSpedUp = 1;
                        printer->delayCounter = 0;
                    }
                }

                return 3;
            }

            printer->delayCounter = printer->textSpeedBottom;
            currentChar = *printer->printerTemplate.currentChar.raw;
            printer->printerTemplate.currentChar.raw++;

            GF_ASSERT(currentChar != 0xF100);

            switch (currentChar)
            {
                case EOS:
                    return 1;
                case 0xE000:
                    printer->printerTemplate.currentX = printer->printerTemplate.x;
                    s32 fontAttribute = GetFontAttribute(printer->printerTemplate.fontId, 1);

                    printer->printerTemplate.currentY +=
                        printer->printerTemplate.lineSpacing + fontAttribute;

                    return 2;

                case 0xF0FD:
                    printer->printerTemplate.currentChar.raw++;
                    return 2;
                case 0xFFFE:
                    printer->printerTemplate.currentChar.raw--;
                    switch ((u16)MsgArray_GetControlCode(printer->printerTemplate.currentChar.raw))
                    {
                        case 0xFF00:
                            u16 field = (u16)MsgArray_ControlCodeGetField(
                                printer->printerTemplate.currentChar.raw, 0);
                            if (field == 0xff)
                            {
                                u8 r2 = printer->printerTemplate.unk4;
                                printer->printerTemplate.unk4 =
                                    (u8)((printer->printerTemplate.fgColor - 1) / 2 + 100);

                                if (!(r2 >= 100 && r2 < 107))
                                {
                                    break;
                                }

                                field = (u16)(r2 - 100);
                            }
                            else
                            {
                                if (field >= 0x64)
                                {
                                    printer->printerTemplate.unk4 = (u8)field;
                                    break;
                                }
                            }

                            printer->printerTemplate.fgColor = (u8)(field * 2 + 1);
                            printer->printerTemplate.shadowColor = (u8)(field * 2 + 2);

                            GenerateFontHalfRowLookupTable(printer->printerTemplate.fgColor,
                                printer->printerTemplate.bgColor,
                                printer->printerTemplate.shadowColor);

                            break;

                        case 0x200:
                            field = (u16)MsgArray_ControlCodeGetField(
                                printer->printerTemplate.currentChar.raw, 0);

                            FUN_0201C1EC(printer,
                                printer->printerTemplate.currentX,
                                printer->printerTemplate.currentY,
                                field);
                            if (printer->textSpeedTop != 0)
                            {
                                CopyWindowToVram(printer->printerTemplate.window);
                            }

                            break;
                        case 0x201:
                            printer->delayCounter = (u8)MsgArray_ControlCodeGetField(
                                printer->printerTemplate.currentChar.raw, 0);
                            printer->printerTemplate.currentChar.raw =
                                MsgArray_SkipControlCode(printer->printerTemplate.currentChar.raw);
                            printer->state = 6;

                            return 3;
                        case 0x202:
                            printer->Unk2A = (u16)MsgArray_ControlCodeGetField(
                                printer->printerTemplate.currentChar.raw, 0);
                            printer->printerTemplate.currentChar.raw =
                                MsgArray_SkipControlCode(printer->printerTemplate.currentChar.raw);

                            return 3;
                        case 0x203:
                            printer->printerTemplate.currentX = (u8)MsgArray_ControlCodeGetField(
                                printer->printerTemplate.currentChar.raw, 0);
                            break;
                        case 0x204:
                            printer->printerTemplate.currentY = (u8)MsgArray_ControlCodeGetField(
                                printer->printerTemplate.currentChar.raw, 0);
                            break;

                        case 0xFF01:
                            field = (u16)MsgArray_ControlCodeGetField(
                                printer->printerTemplate.currentChar.raw, 0);

                            switch (field)
                            {
                                case 100:
                                    printer->printerTemplate.unk2 = 0;
                                    printer->printerTemplate.unk3 = 0;
                                    break;
                                case 200:
                                    printer->printerTemplate.unk2 = 0xFFFC;
                                    printer->printerTemplate.unk3 = 0;
                                    break;
                            }

                            break;

                        case 0xFE06:
                            field = (u16)MsgArray_ControlCodeGetField(
                                printer->printerTemplate.currentChar.raw, 0);
                            if (field != 0xFE00)
                            {
                                if (field != 0xFE01)
                                {
                                    break;
                                }

                                printer->state = 2;
                                TextPrinterInitDownArrowCounters(printer);
                                printer->printerTemplate.currentChar.raw = MsgArray_SkipControlCode(
                                    printer->printerTemplate.currentChar.raw);

                                return 3;
                            }

                            printer->state = 3;
                            TextPrinterInitDownArrowCounters(printer);
                            printer->printerTemplate.currentChar.raw =
                                MsgArray_SkipControlCode(printer->printerTemplate.currentChar.raw);

                            return 3;
                    }

                    printer->printerTemplate.currentChar.raw =
                        MsgArray_SkipControlCode(printer->printerTemplate.currentChar.raw);
                    return 2;

                case 0x25BC:
                    printer->state = 2;
                    TextPrinterInitDownArrowCounters(printer);
                    return 3;

                case 0x25BD:
                    printer->state = 3;
                    TextPrinterInitDownArrowCounters(printer);
                    return 3;
            }

            struct UnkStruct_02002C14_sub *r5 = FUN_02002D94(subStruct->glyphId, currentChar);
            CopyGlyphToWindow(printer->printerTemplate.window,
                r5->buf,
                r5->width,
                r5->height,
                printer->printerTemplate.currentX,
                printer->printerTemplate.currentY,
                printer->printerTemplate.unk2);

            printer->printerTemplate.currentX += r5->width + printer->printerTemplate.letterSpacing;

            return 0;
        case 1:
            if (TextPrinterWait(printer) != 0)
            {
                TextPrinterClearDownArrow(printer);

                printer->state = 0;
            }

            return 3;
        case 2:
            if (TextPrinterWaitWithDownArrow(printer) != 0)
            {
                TextPrinterClearDownArrow(printer);
                FillWindowPixelBuffer(
                    printer->printerTemplate.window, printer->printerTemplate.bgColor);
                printer->printerTemplate.currentX = printer->printerTemplate.x;
                printer->printerTemplate.currentY = printer->printerTemplate.y;
                printer->state = 0;
            }

            return 3;
        case 3:
            if (TextPrinterWaitWithDownArrow(printer) != 0)
            {
                TextPrinterClearDownArrow(printer);
                printer->scrollDistance =
                    (u8)(GetFontAttribute(printer->printerTemplate.fontId, 1) +
                         printer->printerTemplate.lineSpacing);
                printer->printerTemplate.currentX = printer->printerTemplate.x;
                printer->state = 4;
            }

            return 3;
        case 4:
            if (printer->scrollDistance != 0)
            {
                if ((int)printer->scrollDistance < 4)
                {
                    ScrollWindow(printer->printerTemplate.window,
                        0,
                        printer->scrollDistance,
                        (u8)(printer->printerTemplate.bgColor |
                             (printer->printerTemplate.bgColor << 4)));
                    printer->scrollDistance = 0;
                }
                else
                {
                    ScrollWindow(printer->printerTemplate.window,
                        0,
                        4,
                        (u8)(printer->printerTemplate.bgColor |
                             (printer->printerTemplate.bgColor << 4)));

                    printer->scrollDistance -= 4;
                }

                CopyWindowToVram(printer->printerTemplate.window);
            }
            else
            {
                printer->state = 0;
            }

            return 3;
        case 5:
            printer->state = 0;
            return 3;
        case 6:
            if (printer->delayCounter != 0)
            {
                printer->delayCounter--;
            }
            else
            {
                printer->state = 0;
            }

            return 3;
    }

    return 1;
}

THUMB_FUNC void FUN_02002840(u16 flag)
{
    unk00 = flag;
}

THUMB_FUNC void TextPrinterInitDownArrowCounters(struct TextPrinter *printer)
{
    struct TextPrinterSubStruct *subStruct =
        (struct TextPrinterSubStruct *)(&printer->subStructFields);

    if (gTextFlags.autoScroll)
    {
        subStruct->autoScrollDelay = 0;
        return;
    }

    subStruct->downArrowYPosIdx = 0;
    subStruct->downArrowDelay = 0;
}

THUMB_FUNC void TextPrinterDrawDownArrow(struct TextPrinter *printer)
{
    struct TextPrinterSubStruct *subStruct =
        (struct TextPrinterSubStruct *)(&printer->subStructFields);

    if (gTextFlags.autoScroll)
    {
        return;
    }

    if (subStruct->downArrowDelay != 0)
    {
        subStruct->downArrowDelay--;
        return;
    }

    u8 bg_id = GetWindowBgId(printer->printerTemplate.window);
    u8 x = GetWindowX(printer->printerTemplate.window);
    u8 y = GetWindowY(printer->printerTemplate.window);
    u8 width = GetWindowWidth(printer->printerTemplate.window);
    u16 r6 = unk00;

    FillBgTilemapRect(printer->printerTemplate.window->bgConfig,
        bg_id,
        (u16)(r6 + 18 + UNK_020ECB50[subStruct->downArrowYPosIdx] * 4),
        (u8)(x + width + 1),
        (u8)(y + 2),
        1,
        1,
        0x10);

    FillBgTilemapRect(printer->printerTemplate.window->bgConfig,
        bg_id,
        (u16)(r6 + 19 + UNK_020ECB50[subStruct->downArrowYPosIdx] * 4),
        (u8)(x + width + 2),
        (u8)(y + 2),
        1,
        1,
        0x10);
    FillBgTilemapRect(printer->printerTemplate.window->bgConfig,
        bg_id,
        (u16)(r6 + 20 + UNK_020ECB50[subStruct->downArrowYPosIdx] * 4),
        (u8)(x + width + 1),
        (u8)(y + 3),
        1,
        1,
        0x10);
    FillBgTilemapRect(printer->printerTemplate.window->bgConfig,
        bg_id,
        (u16)(r6 + 21 + UNK_020ECB50[subStruct->downArrowYPosIdx] * 4),
        (u8)(x + width + 2),
        (u8)(y + 3),
        1,
        1,
        0x10);

    BgCommitTilemapBufferToVram(printer->printerTemplate.window->bgConfig, bg_id);
    subStruct->downArrowDelay = 8;
    subStruct->downArrowYPosIdx++;
}

THUMB_FUNC void TextPrinterClearDownArrow(struct TextPrinter *printer)
{
    u8 bg_id = GetWindowBgId(printer->printerTemplate.window);
    u8 x = GetWindowX(printer->printerTemplate.window);
    u8 y = GetWindowY(printer->printerTemplate.window);
    u8 width = GetWindowWidth(printer->printerTemplate.window);
    u16 r6 = unk00;

    FillBgTilemapRect(printer->printerTemplate.window->bgConfig,
        bg_id,
        (u16)(r6 + 10),
        (u8)(x + width + 1),
        (u8)(y + 2),
        1,
        2,
        0x10);
    FillBgTilemapRect(printer->printerTemplate.window->bgConfig,
        bg_id,
        (u16)(r6 + 11),
        (u8)(x + width + 2),
        (u8)(y + 2),
        1,
        2,
        0x10);
    BgCommitTilemapBufferToVram(printer->printerTemplate.window->bgConfig, bg_id);
}

THUMB_FUNC BOOL TextPrinterContinue(struct TextPrinter *printer)
{
#pragma unused(printer)
    if ((gMain.newKeys & 3) != 0 || (gMain.touchNew != 0 && gTextFlags.unk0_4 != 0))
    {
        PlaySE(0x5DC);

        gTextFlags.unk0_7 = 1;

        return TRUE;
    }

    return FALSE;
}

THUMB_FUNC BOOL TextPrinterWaitAutoMode(struct TextPrinter *printer)
{
    struct TextPrinterSubStruct *subStruct =
        (struct TextPrinterSubStruct *)(&printer->subStructFields);

    if (subStruct->autoScrollDelay == 100)
    {
        return TRUE;
    }

    subStruct->autoScrollDelay++;
    if (gTextFlags.unk0_5)
    {
        return TextPrinterContinue(printer);
    }

    return FALSE;
}

THUMB_FUNC BOOL TextPrinterWaitWithDownArrow(struct TextPrinter *printer)
{
    if (gTextFlags.autoScroll)
    {
        return TextPrinterWaitAutoMode(printer);
    }
    TextPrinterDrawDownArrow(printer);

    return TextPrinterContinue(printer);
}

THUMB_FUNC u8 TextPrinterWait(struct TextPrinter *printer)
{
    if (gTextFlags.autoScroll)
    {
        return (u8)TextPrinterWaitAutoMode(printer);
    }

    return (u8)TextPrinterContinue(printer);
}

THUMB_FUNC void TextFlags_SetCanABSpeedUpPrint(BOOL param0)
{
    gTextFlags.canABSpeedUpPrint = param0;
}

THUMB_FUNC void FUN_02002B7C(s32 param0)
{
    gTextFlags.autoScroll = param0 & 1;
    gTextFlags.unk0_5 = (param0 >> 1) & 1;
}

THUMB_FUNC void FUN_02002BB8(u32 param0)
{
    gTextFlags.unk0_4 = param0;
}

THUMB_FUNC u8 FUN_02002BD4()
{
    return gTextFlags.unk0_6;
}

THUMB_FUNC void FUN_02002BE4()
{
    gTextFlags.unk0_6 = 0;
}

THUMB_FUNC u8 FUN_02002BF4()
{
    return gTextFlags.unk0_7;
}

THUMB_FUNC void FUN_02002C04()
{
    gTextFlags.unk0_7 = 0;
}