Clean up berry_fix_program

1 files changed, 153 insertions, 131 deletions

diff --git a/src/berry_fix_program.c b/src/berry_fix_program.c
index 4c4a010b2..31caf79cc 100644
--- a/src/berry_fix_program.c
+++ b/src/berry_fix_program.c
@@ -12,40 +12,42 @@
 #include "text.h"
 #include "menu.h"
 #include "m4a.h"
+#include "constants/rgb.h"
 
-// Static type declarations
-
-typedef struct {
+struct {
     u8 state;
-    u8 unk1;
-    u16 unk2;
+    u8 curScene;
+    u16 timer;
     struct MultiBootParam mb;
-} berryfix_t;
-
-// Static RAM declarations
-
-static berryfix_t *berry_fix_mb_manager;
-
-// Static ROM declarations
-
-static void berry_fix_main(void);
-static void berry_fix_gpu_set(void);
-static int berry_fix_text_update(int);
-static void berry_fix_text_print(int);
-static void berry_fix_bg_hide(void);
+} static * sBerryFix;
 
-// .rodata
+static void BerryFix_Main(void);
+static void BerryFix_GpuSet(void);
+static int BerryFix_TrySetScene(int);
+static void BerryFix_SetScene(int);
+static void BerryFix_HideScene(void);
 
 static const u8 sText_BerryProgramUpdate[] = _("Berry Program Update");
 static const u8 sText_RubySapphire[] = _("Ruby/Sapphire");
 static const u8 sText_Emerald[] = _("Emerald");
-
-static const u8 sText_BerryProgramWillBeUpdatedPressA[] = _("The Berry Program on your POKéMON\nRuby/Sapphire Game Pak will be updated.\n{COLOR RED}{SHADOW LIGHT_RED}Press the A Button.");
-static const u8 sText_EnsureGBAConnectionMatches[] = _("Please ensure the connection of your\nGame Boy Advance system matches this.\n{COLOR RED}{SHADOW LIGHT_RED}YES: Press the A Button.\nNO: Turn off the power and try again.");
-static const u8 sText_TurnOffPowerHoldingStartSelect[] = _("Please turn on the power of POKéMON\nRuby/Sapphire while holding START and\nSELECT simultaneously. Then, ensure\nthe picture above appears.");
-static const u8 sText_TransmittingPleaseWait[] = _("Transmitting. Please wait.\n{COLOR RED}{SHADOW LIGHT_RED}Please do not turn off the power or\nunplug the Game Boy Advance Game\nLink Cable.");
-static const u8 sText_PleaseFollowInstructionsOnScreen[] = _("Please follow the instructions on your\nPOKéMON Ruby/Sapphire screen.");
-static const u8 sText_TransmissionFailureTryAgain[] = _("Transmission failure.\n{COLOR RED}{SHADOW LIGHT_RED}Please try again.");
+static const u8 sText_BerryProgramWillBeUpdatedPressA[] = _("The Berry Program on your POKéMON\n"
+                                                            "Ruby/Sapphire Game Pak will be updated.\n"
+                                                            "{COLOR RED}{SHADOW LIGHT_RED}Press the A Button.");
+static const u8 sText_EnsureGBAConnectionMatches[] = _("Please ensure the connection of your\n"
+                                                       "Game Boy Advance system matches this.\n"
+                                                       "{COLOR RED}{SHADOW LIGHT_RED}YES: Press the A Button.\n"
+                                                       "NO: Turn off the power and try again.");
+static const u8 sText_TurnOffPowerHoldingStartSelect[] = _("Please turn on the power of POKéMON\n"
+                                                           "Ruby/Sapphire while holding START and\n"
+                                                           "SELECT simultaneously. Then, ensure\n"
+                                                           "the picture above appears.");
+static const u8 sText_TransmittingPleaseWait[] = _("Transmitting. Please wait.\n"
+                                                   "{COLOR RED}{SHADOW LIGHT_RED}Please do not turn off the power or\n"
+                                                   "unplug the Game Boy Advance Game\nLink Cable.");
+static const u8 sText_PleaseFollowInstructionsOnScreen[] = _("Please follow the instructions on your\n"
+                                                             "POKéMON Ruby/Sapphire screen.");
+static const u8 sText_TransmissionFailureTryAgain[] = _("Transmission failure.\n"
+                                                        "{COLOR RED}{SHADOW LIGHT_RED}Please try again.");
 
 static const struct BgTemplate sBerryFixBgTemplates[] = {
     {
@@ -108,52 +110,66 @@ static const struct WindowTemplate sBerryFixWindowTemplates[] = {
     DUMMY_WIN_TEMPLATE
 };
 
-static const u16 sUnknown_08618138[] = {
-    0x7fff, 0x7fff, 0x318c, 0x675a,
-    0x043c, 0x3aff, 0x0664, 0x4bd2,
-    0x6546, 0x7b14, 0x7fff, 0x318c,
-    0x675a, 0, 0, 0
+static const u16 sBerryFixPalColors[] = {
+    RGB_WHITE,       RGB_WHITE,       RGB(12, 12, 12), RGB(26, 26, 25),
+    RGB(28, 1, 1),   RGB(31, 23, 14), RGB(4, 19, 1),   RGB(18, 30, 18),
+    RGB(6, 10, 25),  RGB(20, 24, 30), RGB_WHITE,       RGB(12, 12, 12),
+    RGB(26, 26, 25), RGB_BLACK,       RGB_BLACK,       RGB_BLACK
 };
 
 static const u8 sBerryProgramTextColors[] = {TEXT_DYNAMIC_COLOR_1, TEXT_DYNAMIC_COLOR_2, TEXT_DYNAMIC_COLOR_3};
 static const u8 sGameTitleTextColors[] = { TEXT_COLOR_TRANSPARENT, TEXT_DYNAMIC_COLOR_1, TEXT_DYNAMIC_COLOR_4};
 
-static const u8 *const sBerryProgramTexts[] = {
-    sText_EnsureGBAConnectionMatches,
-    sText_TurnOffPowerHoldingStartSelect,
-    sText_TransmittingPleaseWait,
-    sText_PleaseFollowInstructionsOnScreen,
-    sText_TransmissionFailureTryAgain,
-    sText_BerryProgramWillBeUpdatedPressA
+enum {
+    SCENE_ENSURE_CONNECT,
+    SCENE_TURN_OFF_POWER,
+    SCENE_TRANSMITTING,
+    SCENE_FOLLOW_INSTRUCT,
+    SCENE_TRANSMIT_FAILED,
+    SCENE_BEGIN,
+    SCENE_NONE
 };
 
+static const u8 *const sBerryProgramTexts[] = {
+    [SCENE_ENSURE_CONNECT]  = sText_EnsureGBAConnectionMatches,
+    [SCENE_TURN_OFF_POWER]  = sText_TurnOffPowerHoldingStartSelect,
+    [SCENE_TRANSMITTING]    = sText_TransmittingPleaseWait,
+    [SCENE_FOLLOW_INSTRUCT] = sText_PleaseFollowInstructionsOnScreen,
+    [SCENE_TRANSMIT_FAILED] = sText_TransmissionFailureTryAgain,
+    [SCENE_BEGIN]           = sText_BerryProgramWillBeUpdatedPressA
+};
 
 static const struct {
     const u32 *gfx;
     const u32 *tilemap;
-    const u16 *pltt;
+    const u16 *palette;
 } sBerryFixGraphics[] = {
-    {
+    [SCENE_ENSURE_CONNECT] = {
         gBerryFixGameboy_Gfx,
         gBerryFixGameboy_Tilemap,
         gBerryFixGameboy_Pal
-    }, {
+    }, 
+    [SCENE_TURN_OFF_POWER] = {
         gBerryFixGameboyLogo_Gfx,
         gBerryFixGameboyLogo_Tilemap,
         gBerryFixGameboyLogo_Pal
-    }, {
+    }, 
+    [SCENE_TRANSMITTING] = {
         gBerryFixGbaTransfer_Gfx,
         gBerryFixGbaTransfer_Tilemap,
         gBerryFixGbaTransfer_Pal
-    }, {
+    }, 
+    [SCENE_FOLLOW_INSTRUCT] = {
         gBerryFixGbaTransferHighlight_Gfx,
         gBerryFixGbaTransferHighlight_Tilemap,
         gBerryFixGbaTransferHighlight_Pal
-    }, {
+    }, 
+    [SCENE_TRANSMIT_FAILED] = {
         gBerryFixGbaTransferError_Gfx,
         gBerryFixGbaTransferError_Tilemap,
         gBerryFixGbaTransferError_Pal
-    }, {
+    }, 
+    [SCENE_BEGIN] = {
         gBerryFixWindow_Gfx,
         gBerryFixWindow_Tilemap,
         gBerryFixWindow_Pal
@@ -163,7 +179,16 @@ static const struct {
 extern const u8 gMultiBootProgram_BerryGlitchFix_Start[0x3BF4];
 extern const u8 gMultiBootProgram_BerryGlitchFix_End[];
 
-// .text
+enum {
+    MAINSTATE_INIT,
+    MAINSTATE_BEGIN,
+    MAINSTATE_CONNECT,
+    MAINSTATE_INIT_MULTIBOOT,
+    MAINSTATE_MULTIBOOT,
+    MAINSTATE_TRANSMIT,
+    MAINSTATE_EXIT,
+    MAINSTATE_FAILED,
+};
 
 void CB2_InitBerryFixProgram(void)
 {
@@ -175,81 +200,79 @@ void CB2_InitBerryFixProgram(void)
     ResetTasks();
     ScanlineEffect_Stop();
     SetGpuReg(REG_OFFSET_DISPCNT, 0);
-    berry_fix_mb_manager = AllocZeroed(0x50);
-    berry_fix_mb_manager->state = 0;
-    berry_fix_mb_manager->unk1 = 6;
-    SetMainCallback2(berry_fix_main);
+    sBerryFix = AllocZeroed(sizeof(*sBerryFix));
+    sBerryFix->state = MAINSTATE_INIT;
+    sBerryFix->curScene = SCENE_NONE;
+    SetMainCallback2(BerryFix_Main);
 }
 
-static void berry_fix_main(void)
+#define TryScene(sceneNum) BerryFix_TrySetScene(sceneNum) == (sceneNum)
+
+static void BerryFix_Main(void)
 {
-    switch (berry_fix_mb_manager->state)
+    switch (sBerryFix->state)
     {
-        case 0:
-            berry_fix_gpu_set();
-            berry_fix_mb_manager->state = 1;
+        case MAINSTATE_INIT:
+            BerryFix_GpuSet();
+            sBerryFix->state = MAINSTATE_BEGIN;
             break;
-        case 1:
-            if (berry_fix_text_update(5) == 5 && (JOY_NEW(A_BUTTON)))
-            {
-                berry_fix_mb_manager->state = 2;
-            }
+        case MAINSTATE_BEGIN:
+            if (TryScene(SCENE_BEGIN) && (JOY_NEW(A_BUTTON)))
+                sBerryFix->state = MAINSTATE_CONNECT;
             break;
-        case 2:
-            if (berry_fix_text_update(0) == 0 && (JOY_NEW(A_BUTTON)))
-            {
-                berry_fix_mb_manager->state = 3;
-            }
+        case MAINSTATE_CONNECT:
+            if (TryScene(SCENE_ENSURE_CONNECT) && (JOY_NEW(A_BUTTON)))
+                sBerryFix->state = MAINSTATE_INIT_MULTIBOOT;
             break;
-        case 3:
-            if (berry_fix_text_update(1) == 1)
+        case MAINSTATE_INIT_MULTIBOOT:
+            if (TryScene(SCENE_TURN_OFF_POWER))
             {
-                berry_fix_mb_manager->mb.masterp = gMultiBootProgram_BerryGlitchFix_Start;
-                berry_fix_mb_manager->mb.server_type = 0;
-                MultiBootInit(&berry_fix_mb_manager->mb);
-                berry_fix_mb_manager->unk2 = 0;
-                berry_fix_mb_manager->state = 4;
+                sBerryFix->mb.masterp = gMultiBootProgram_BerryGlitchFix_Start;
+                sBerryFix->mb.server_type = 0;
+                MultiBootInit(&sBerryFix->mb);
+                sBerryFix->timer = 0;
+                sBerryFix->state = MAINSTATE_MULTIBOOT;
             }
             break;
-        case 4:
-            MultiBootMain(&berry_fix_mb_manager->mb);
-            if (berry_fix_mb_manager->mb.probe_count != 0 || (!(berry_fix_mb_manager->mb.response_bit & 2) || !(berry_fix_mb_manager->mb.client_bit & 2)))
+        case MAINSTATE_MULTIBOOT:
+            MultiBootMain(&sBerryFix->mb);
+            if (sBerryFix->mb.probe_count != 0 || (!(sBerryFix->mb.response_bit & 2) || !(sBerryFix->mb.client_bit & 2)))
             {
-                berry_fix_mb_manager->unk2 = 0;
+                sBerryFix->timer = 0;
             }
-            else if (++ berry_fix_mb_manager->unk2 > 180)
+            else if (++sBerryFix->timer > 180)
             {
-                MultiBootStartMaster(&berry_fix_mb_manager->mb, gMultiBootProgram_BerryGlitchFix_Start + ROM_HEADER_SIZE, (u32)(gMultiBootProgram_BerryGlitchFix_End - (gMultiBootProgram_BerryGlitchFix_Start + ROM_HEADER_SIZE)), 4, 1);
-                berry_fix_mb_manager->state = 5;
+                MultiBootStartMaster(&sBerryFix->mb, 
+                                     gMultiBootProgram_BerryGlitchFix_Start + ROM_HEADER_SIZE, 
+                                     (u32)(gMultiBootProgram_BerryGlitchFix_End - (gMultiBootProgram_BerryGlitchFix_Start + ROM_HEADER_SIZE)), 
+                                     4, 
+                                     1);
+                sBerryFix->state = MAINSTATE_TRANSMIT;
             }
             break;
-        case 5:
-            if (berry_fix_text_update(2) == 2) {
-                MultiBootMain(&berry_fix_mb_manager->mb);
-                if (MultiBootCheckComplete(&berry_fix_mb_manager->mb)) {
-                    berry_fix_mb_manager->state = 6;
-                }
-                else if (!(berry_fix_mb_manager->mb.client_bit & 2)) {
-                    berry_fix_mb_manager->state = 7;
-                }
+        case MAINSTATE_TRANSMIT:
+            if (TryScene(SCENE_TRANSMITTING)) 
+            {
+                MultiBootMain(&sBerryFix->mb);
+
+                if (MultiBootCheckComplete(&sBerryFix->mb)) 
+                    sBerryFix->state = MAINSTATE_EXIT;
+                else if (!(sBerryFix->mb.client_bit & 2)) 
+                    sBerryFix->state = MAINSTATE_FAILED;
             }
             break;
-        case 6:
-            if (berry_fix_text_update(3) == 3 && JOY_NEW(A_BUTTON))
-            {
+        case MAINSTATE_EXIT:
+            if (TryScene(SCENE_FOLLOW_INSTRUCT) && JOY_NEW(A_BUTTON))
                 DoSoftReset();
-            }
             break;
-        case 7:
-            if (berry_fix_text_update(4) == 4 && JOY_NEW(A_BUTTON))
-            {
-                berry_fix_mb_manager->state = 1;
-            }
+        case MAINSTATE_FAILED:
+            if (TryScene(SCENE_TRANSMIT_FAILED) && JOY_NEW(A_BUTTON))
+                sBerryFix->state = MAINSTATE_BEGIN;
             break;
     }
 }
 
-static void berry_fix_gpu_set(void)
+static void BerryFix_GpuSet(void)
 {
     s32 width, left;
 
@@ -274,26 +297,26 @@ static void berry_fix_gpu_set(void)
     InitWindows(sBerryFixWindowTemplates);
     DeactivateAllTextPrinters();
 
-    DmaCopy32(3, sUnknown_08618138, BG_PLTT + 0x1E0, 0x20);
+    DmaCopy32(3, sBerryFixPalColors, BG_PLTT + 0x1E0, sizeof(sBerryFixPalColors));
     SetGpuReg(REG_OFFSET_DISPCNT, DISPCNT_OBJ_1D_MAP);
     FillWindowPixelBuffer(2, PIXEL_FILL(0));
     FillWindowPixelBuffer(3, PIXEL_FILL(0));
-    FillWindowPixelBuffer(0, PIXEL_FILL(0xA));
+    FillWindowPixelBuffer(0, PIXEL_FILL(10));
 
     width = GetStringWidth(0, sText_Emerald, 0);
-    left = (0x78 - width) / 2;
+    left = (120 - width) / 2;
     AddTextPrinterParameterized3(2, 0, left, 3, sGameTitleTextColors, TEXT_SPEED_FF, sText_Emerald);
 
     width = GetStringWidth(0, sText_RubySapphire, 0);
-    left = (0x78 - width) / 2 + 0x78;
+    left = (120 - width) / 2 + 120;
     AddTextPrinterParameterized3(2, 0, left, 3, sGameTitleTextColors, TEXT_SPEED_FF, sText_RubySapphire);
 
     width = GetStringWidth(0, sText_RubySapphire, 0);
-    left = (0x70 - width) / 2;
+    left = (112 - width) / 2;
     AddTextPrinterParameterized3(3, 0, left, 0, sGameTitleTextColors, TEXT_SPEED_FF, sText_RubySapphire);
 
     width = GetStringWidth(1, sText_BerryProgramUpdate, 0);
-    left = (0xD0 - width) / 2;
+    left = (208 - width) / 2;
     AddTextPrinterParameterized3(0, 1, left, 2, sBerryProgramTextColors, TEXT_SPEED_FF, sText_BerryProgramUpdate);
 
     CopyWindowToVram(2, 2);
@@ -301,56 +324,55 @@ static void berry_fix_gpu_set(void)
     CopyWindowToVram(0, 2);
 }
 
-static int berry_fix_text_update(int checkval)
+static int BerryFix_TrySetScene(int scene)
 {
-    if (berry_fix_mb_manager->unk1 == checkval)
-    {
-        return checkval;
-    }
-    if (berry_fix_mb_manager->unk1 == 6)
+    if (sBerryFix->curScene == scene)
+        return scene;
+
+    if (sBerryFix->curScene == SCENE_NONE)
     {
-        berry_fix_text_print(checkval);
-        berry_fix_mb_manager->unk1 = checkval;
+        BerryFix_SetScene(scene);
+        sBerryFix->curScene = scene;
     }
     else
     {
-        berry_fix_bg_hide();
-        berry_fix_mb_manager->unk1 = 6;
+        BerryFix_HideScene();
+        sBerryFix->curScene = SCENE_NONE;
     }
-    return berry_fix_mb_manager->unk1;
+    return sBerryFix->curScene;
 }
 
-static void berry_fix_text_print(int scene)
+static void BerryFix_SetScene(int scene)
 {
     FillBgTilemapBufferRect_Palette0(0, 0, 0, 0, 32, 32);
-    FillWindowPixelBuffer(1, PIXEL_FILL(0xA));
+    FillWindowPixelBuffer(1, PIXEL_FILL(10));
     AddTextPrinterParameterized3(1, 1, 0, 0, sBerryProgramTextColors, -1, sBerryProgramTexts[scene]);
     PutWindowTilemap(1);
     CopyWindowToVram(1, 2);
     switch (scene)
     {
-        case 0:
-        case 2:
-        case 3:
-        case 4:
-            PutWindowTilemap(2);
-            break;
-        case 1:
-            PutWindowTilemap(3);
-            break;
-        case 5:
-            PutWindowTilemap(0);
-            break;
+    case SCENE_ENSURE_CONNECT:
+    case SCENE_TRANSMITTING:
+    case SCENE_FOLLOW_INSTRUCT:
+    case SCENE_TRANSMIT_FAILED:
+        PutWindowTilemap(2);
+        break;
+    case SCENE_TURN_OFF_POWER:
+        PutWindowTilemap(3);
+        break;
+    case SCENE_BEGIN:
+        PutWindowTilemap(0);
+        break;
     }
     CopyBgTilemapBufferToVram(0);
     LZ77UnCompVram(sBerryFixGraphics[scene].gfx, (void *)BG_CHAR_ADDR(1));
     LZ77UnCompVram(sBerryFixGraphics[scene].tilemap, (void *)BG_SCREEN_ADDR(31));
-    CpuCopy32(sBerryFixGraphics[scene].pltt, (void *)BG_PLTT, 0x100);
+    CpuCopy32(sBerryFixGraphics[scene].palette, (void *)BG_PLTT, 0x100);
     ShowBg(0);
     ShowBg(1);
 }
 
-static void berry_fix_bg_hide(void)
+static void BerryFix_HideScene(void)
 {
     HideBg(0);
     HideBg(1);