#include "global.h"
#include "functions.h"
#include "main.h"
#include "screen.h"

typedef struct {
    u8 filler0[0x30];
} ChooseFileState;

// TODO: This occupies the same memory region as gMenu
extern ChooseFileState gChooseFileState;

typedef struct {
    u8 filler0[0x10];
    u16 unk10;
    u8 filler12[0x6];
} struct_080FC844;

typedef struct {
    u8 filler0[0x4];
} struct_020227E8;

extern struct_020227E8 gUnk_020227E8;

typedef struct {
    u8 filler0[0x800];
} struct_02034CB0;

extern struct_02034CB0 gUnk_02034CB0;
extern struct_02034CB0 gUnk_02021F30;
extern SaveFile gSaveFiles[];
extern u32 gUsedPalettes;

static void sub_08050624(u32);
static void sub_0805066C(void);

extern void sub_0801D66C(const void* src, void* dest, u32 size);
extern void sub_08056FEC(u32, struct_020227E8*);
extern void sub_0805F46C(u32, struct_080FC844*);
extern void sub_0801C4A0(u32, u32);
extern void sub_0801D66C(const void* src, void* dest, u32 size);
extern void sub_080507FC(void);
extern void sub_08080668(void);
extern void sub_0805194C(u32);
extern void LoadPalettes(const u8*, int, int);

extern const struct_080FC844 gUnk_080FC844;
extern const u16 gUnk_080FC85C[][3];
extern void (*const gUnk_080FC8A4[])(void);
extern void (*const gUnk_080FC8B0[])(void);
extern const u8 gUnk_080FC8D0[];
extern const u16 gUnk_080FC8DE[];
extern const u8 gGlobalGfxAndPalettes[];

void sub_08050318(u32 arg0, u32 arg1) {
    u32 sfx;
    struct_080FC844 var0;

    sub_08050384();
    sub_0801D66C(&gUnk_080FC844, &var0, sizeof(gUnk_080FC844));
    sub_08056FEC(arg1, &gUnk_020227E8);
    var0.unk10 |= gUnk_080FC85C[arg0][0] << 0xC;
    sub_0805F46C(gUnk_080FC85C[arg0][1], &var0);
    sfx = gUnk_080FC85C[arg0][2];
    if (sfx) {
        PlaySFX(sfx);
    }
    gScreen.bg.bg3Control = 1;
}

void sub_08050384(void) {
    sub_0801C4A0(0, 0);
    _DmaZero(&gUnk_02034CB0, sizeof(gUnk_02034CB0));
    gScreen.bg.bg3Control = 1;
}

void sub_080503A8(u32 gfxGroup) {
    LoadGfxGroup(gfxGroup);
    gScreen.bg.bg2yOffset = 1;
    gScreen.affine.unk = 1;
}

void sub_080503BC(u8 state) {
    gUnk_02032EC0.state = state;
    _DmaZero(&gUnk_02034CB0, sizeof(gUnk_02034CB0));
    _DmaZero(&gUnk_02021F30, sizeof(gUnk_02021F30));
}

void sub_080503E4(u32 saveFileId) {
    u8 messageSpeed;
    u8 brightnessPref;

    if (saveFileId > 2) {
        messageSpeed = 1;
        brightnessPref = 1;
    } else {
        SaveFile* saveFile = &gSaveFiles[saveFileId];
        messageSpeed = saveFile->messageSpeed;
        brightnessPref = saveFile->brightnessPref;
    }

    ((struct_02000000 *)0x2000000)->messageSpeed = messageSpeed;
    ((struct_02000000 *)0x2000000)->brightnessPref = brightnessPref;
    gUsedPalettes = 0xFFFFFFFF;
}

void sub_0805041C(u32 saveFileId) {
    SaveFile* saveFile;

    if (saveFileId < 3) {
        ((struct_02000000 *)0x2000000)->saveFileId = saveFileId;
        saveFile = &gSaveFiles[saveFileId];
        sub_0801D66C(saveFile, &gUnk_02002A40, sizeof(*saveFile));
    }
    sub_080503E4(saveFileId);
}

void HandleChooseFileScreen(void) {
    sub_080AD90C();
    gUnk_080FC8A4[gUnk_03001000.funcIndex]();
    if (gUnk_02032EC0.transitionType != gUnk_02032EC0.state) {
        gUnk_02032EC0.transitionType = gUnk_02032EC0.state;
        gScreen.bg.bg1Control = 0;
        gScreen.bg.bg2Control = 0;
        gScreen.bg.bg1yOffset = 0;
        gScreen.bg.bg2xOffset = 0;
        gScreen.affine.bg2xOffset = 0;
        gScreen.affine.bg2yOffset = 0;
        _DmaZero(&gChooseFileState, sizeof(gChooseFileState));
    }

    sub_080507FC();
    gUnk_080FC8B0[gUnk_02032EC0.transitionType]();
    gUnk_02019EE0.unk0 = 0;
    sub_0805E5C0();
    sub_0805066C();
    sub_0801C1D4();
    sub_0801C208();
    sub_080AD9B0();
    sub_080AD918();
    if (gUnk_02019EE0.unk3 != ((struct_02000000 *)0x2000000)->gameLanguage) {
        gUnk_02019EE0.unk3 = ((struct_02000000 *)0x2000000)->gameLanguage;
        sub_080503A8(0x6);
        sub_080503A8(0xF);
    }
}

void sub_080504F4(void) {
    u32 i;

    sub_0801DA90(1);
    sub_080A3210();
    _DmaZero((void*)VRAM, 0x80);
    sub_08056418();
    EraseAllEntities();
    sub_08080668();
    sub_080ADD30();
    sub_0801CFA8(0);
    _DmaZero(&gUnk_0200AF00, sizeof(gUnk_0200AF00));
    _DmaZero(&gUnk_02019EE0, sizeof(gUnk_02019EE0));
    gUnk_02019EE0.unk3 = 7;
    gUnk_02019EE0.unk6 = ((struct_02000000 *)0x2000000)->gameLanguage > 1 ? 3 : 0;
    _DmaZero(&gUnk_02032EC0, sizeof(gUnk_02032EC0));
    gUnk_02032EC0.transitionType = 8;
    sub_080503BC(0);
    sub_0805616C();
    sub_08050624(0);
    sub_08050624(1);
    sub_08050624(2);
    sub_08056208();
    sub_080503A8(0x5);
    LoadPaletteGroup(0x9);
    for (i = 0; i < 26; i++) {
        CreateObject(0x48, i, 0);
    }

    sub_080A70AC(gUnk_080FC8D0);
    sub_080507FC();
    gScreen.lcd.displayControl |= 0x1F00;
    gScreen.affine.bg3Control = 0x1E0B;
    gScreen.controls.layerFXControl = 0x844;
    gScreen.controls.alphaBlend = BLDALPHA_BLEND(15, 10);
    gUnk_02024490.unk0 = 1;
    gUnk_03001000.funcIndex = 1;
    PlaySFX(0x7);
    DoFade(4, 8);
}

void nullsub_479(void) {
}

void sub_0805060C(void) {
    if (!gFadeControl.active) {
        InitScreen(SCREEN_GAMEPLAY);
    }
}

static void sub_08050624(u32 saveFileId) {
    SaveFile* saveFile = &gSaveFiles[saveFileId];
    int var0 = sub_0807CF28(saveFileId, saveFile);
    switch (var0) {
        case -1:
            sub_0807CF48(saveFileId);
            // fallthrough
        case 0:
            sub_0805194C(saveFileId);
            break;
    }
    gUnk_02019EE0.unk8[saveFileId] = var0;
}

static void sub_0805066C(void) {
    u32 loadNewPalette;
    const u8* paletteOffset;

    loadNewPalette = FALSE;
    if (--gUnk_02019EE0.unk1 == 0) {
        gUnk_02019EE0.unk1 = 16;
        gUnk_02019EE0.unk2 = (gUnk_02019EE0.unk2 + 1) % 15;
        loadNewPalette = TRUE;
    }

    if (gUnk_02019EE0.unk2 == 0) {
        gUnk_02019EE0.unk2 = 1;
        gUnk_02019EE0.unk1 = (Random() & 0x7) * 16 + 8;
        loadNewPalette = TRUE;
    }

    if (loadNewPalette) {
        paletteOffset = &gGlobalGfxAndPalettes[gUnk_080FC8DE[gUnk_02019EE0.unk2]];
        LoadPalettes(&paletteOffset[0x11AA0], 11, 1);
        LoadPalettes(&paletteOffset[0x11BA0], 12, 1);
        LoadPalettes(&paletteOffset[0x11CA0], 13, 1);
        LoadPalettes(&paletteOffset[0x11DA0], 14, 1);
    }
}