1 files changed, 262 insertions, 0 deletions
diff --git a/arm9/src/poketch.c b/arm9/src/poketch.c
new file mode 100644
index 00000000..54dff12b
--- /dev/null
+++ b/arm9/src/poketch.c
@@ -0,0 +1,262 @@
+#include "global.h"
+#include "MI_memory.h"
+#include "pokemon.h"
+#include "poketch.h"
+
+#pragma thumb on
+
+extern u16 BoxMon_GetAlternateForme(struct BoxPokemon * mon);
+
+static const u8 sMarkingMapInitialPos[][2] = {
+    { 0x68, 0x98 },
+    { 0x78, 0x98 },
+    { 0x88, 0x98 },
+    { 0x98, 0x98 },
+    { 0xA8, 0x98 },
+    { 0xB8, 0x98 },
+};
+
+u32 Sav2_Poketch_sizeof(void)
+{
+    return sizeof(struct SavePoketch);
+}
+
+void Sav2_Poketch_init(struct SavePoketch * poketch)
+{
+    int i;
+
+    for (i = 0; i < 32; i++)
+    {
+        poketch->unlockedApps[i] = 0;
+    }
+    poketch->numApps = 0;
+    poketch->selectedApp = 0;
+    poketch->isGiven = 0;
+    poketch->color = 0;
+    poketch->pedometerActive = 0;
+    poketch->stepCounter = 0;
+    poketch->alarmActive = 0;
+    poketch->alarmHour = 0;
+    poketch->alarmMinute = 0;
+    poketch->calendarMonth = 1; // January
+    poketch->calendarFlags = 0;
+    for (i = 0; i < 6; i++)
+    {
+        poketch->markingMapPos[i][0] = sMarkingMapInitialPos[i][0];
+        poketch->markingMapPos[i][1] = sMarkingMapInitialPos[i][1];
+    }
+    for (i = 0; i < 12; i++)
+    {
+        poketch->pokemonHistory[i][0] = SPECIES_NONE;
+        poketch->pokemonHistory[i][1] = 0;
+    }
+    poketch->dotArtistEnabled = 0;
+    Sav2_Poketch_UnlockApp(poketch, POKETCH_DIGITAL_WATCH);
+}
+
+void Sav2_Poketch_Give(struct SavePoketch * poketch)
+{
+    poketch->isGiven = TRUE;
+}
+
+BOOL Sav2_Poketch_IsGiven(struct SavePoketch * poketch)
+{
+    return poketch->isGiven;
+}
+
+u8 Sav2_Poketch_AppIsUnlocked(struct SavePoketch * poketch, PoketchApp app)
+{
+    return poketch->unlockedApps[app];
+}
+
+BOOL Sav2_Poketch_UnlockApp(struct SavePoketch * poketch, PoketchApp app)
+{
+    GF_ASSERT(app >= FIRST_POKETCH_APP_ID && app < NUM_POKETCH_APPS);
+    if (poketch->numApps < NUM_POKETCH_APPS && !poketch->unlockedApps[app])
+    {
+        poketch->unlockedApps[app] = TRUE;
+        poketch->numApps++;
+        if (app == POKETCH_PEDOMETER) // pedometer
+        {
+            poketch->pedometerActive = TRUE;
+        }
+        return TRUE;
+    }
+    return FALSE;
+}
+
+PoketchApp Sav2_Poketch_GetSelectedApp(struct SavePoketch * poketch)
+{
+    return (PoketchApp)poketch->selectedApp;
+}
+
+PoketchApp Sav2_Poketch_CycleNextApp(struct SavePoketch * poketch)
+{
+    PoketchApp app = (PoketchApp)poketch->selectedApp;
+    PoketchApp prev = app;
+
+    while (1)
+    {
+        if (++app >= NUM_POKETCH_APPS)
+            app = FIRST_POKETCH_APP_ID;
+        if (app == prev)
+            break;
+        if (poketch->unlockedApps[app])
+            break;
+    }
+    poketch->selectedApp = (s8)app;
+    return (PoketchApp)poketch->selectedApp;
+}
+
+u8 Sav2_Poketch_GetScreenTint(struct SavePoketch * poketch)
+{
+    GF_ASSERT(poketch != NULL);
+    return poketch->color;
+}
+
+void Sav2_Poketch_SetScreenTint(struct SavePoketch * poketch, u32 color)
+{
+    GF_ASSERT(poketch != NULL);
+    GF_ASSERT(color < 8);
+    poketch->color = (u8)color;
+}
+
+u32 Sav2_Poketch_GetStepCounter(struct SavePoketch * poketch)
+{
+    return poketch->stepCounter;
+}
+
+void Sav2_Poketch_SetStepCounter(struct SavePoketch * poketch, u32 steps)
+{
+    if (poketch->pedometerActive)
+        poketch->stepCounter = steps;
+}
+
+BOOL Sav2_Poketch_GetAlarmState(struct SavePoketch * poketch)
+{
+    return poketch->alarmActive;
+}
+
+void Sav2_Poketch_GetAlarmSetTime(struct SavePoketch * poketch, u32 * hour_p, u32 * min_p)
+{
+    *hour_p = poketch->alarmHour;
+    *min_p = poketch->alarmMinute;
+}
+
+void Sav2_Poketch_SetAlarm(struct SavePoketch * poketch, BOOL enabled, u32 hour, u32 minute)
+{
+    poketch->alarmActive = enabled;
+    poketch->alarmHour = hour;
+    poketch->alarmMinute = minute;
+}
+
+void Sav2_Poketch_CalendarDateHighlight(struct SavePoketch * poketch, u32 month, u32 day)
+{
+    if (poketch->calendarMonth == month)
+    {
+        poketch->calendarFlags |= (1u << (day - 1));
+    }
+    else
+    {
+        poketch->calendarMonth = (u8)month;
+        poketch->calendarFlags = (1u << (day - 1));
+    }
+}
+
+void Sav2_Poketch_CalendarDateUnhighlight(struct SavePoketch * poketch, u32 month, u32 day)
+{
+    if (poketch->calendarMonth == month)
+    {
+        poketch->calendarFlags &= ~(1u << (day - 1));
+    }
+    else
+    {
+        poketch->calendarMonth = (u8)month;
+        poketch->calendarFlags = 0;
+    }
+}
+
+BOOL Sav2_Poketch_CalendarDateIsHighlighted(struct SavePoketch * poketch, u32 month, u32 day)
+{
+    if (poketch->calendarMonth == month)
+    {
+        return (BOOL)((poketch->calendarFlags >> (day - 1)) & 1);
+    }
+    else
+    {
+        return FALSE;
+    }
+}
+
+void Sav2_Poketch_MarkingMapSetPos(struct SavePoketch * poketch, s32 mark, u8 x, u8 y)
+{
+    GF_ASSERT(mark < 6);
+    poketch->markingMapPos[mark][0] = x;
+    poketch->markingMapPos[mark][1] = y;
+}
+
+void Sav2_Poketch_MarkingMapGetPos(struct SavePoketch * poketch, s32 mark, u8 * x_p, u8 * y_p)
+{
+    GF_ASSERT(mark < 6);
+    *x_p = poketch->markingMapPos[mark][0];
+    *y_p = poketch->markingMapPos[mark][1];
+}
+
+u32 Sav2_Poketch_DotArtistIsEnabled(struct SavePoketch * poketch)
+{
+    return poketch->dotArtistEnabled;
+}
+
+void Sav2_Poketch_DotArtistGetDrawing(struct SavePoketch * poketch, void * grid)
+{
+    if (poketch->dotArtistEnabled)
+    {
+        MI_CpuCopy8(poketch->dotArtistGrid, grid, DOT_ARTIST_SIZE);
+    }
+}
+
+void Sav2_Poketch_DotArtistSetDrawingAndEnable(struct SavePoketch * poketch, void * grid)
+{
+    MI_CpuCopy8(grid, poketch->dotArtistGrid, DOT_ARTIST_SIZE);
+    poketch->dotArtistEnabled = TRUE;
+}
+
+void Sav2_Poketch_PokemonHistoryAddMon(struct SavePoketch * poketch, struct BoxPokemon * mon)
+{
+    int i = Sav2_Poketch_PokemonHistoryGetFirstEmptySlot(poketch);
+    if (i >= 12)
+    {
+        for (i = 0; i < 11; i++)
+        {
+            poketch->pokemonHistory[i][0] = poketch->pokemonHistory[i + 1][0];
+            poketch->pokemonHistory[i][1] = poketch->pokemonHistory[i + 1][1];
+        }
+        i = 11;
+    }
+    poketch->pokemonHistory[i][0] = (u16)GetBoxMonData(mon, MON_DATA_SPECIES, NULL);
+    poketch->pokemonHistory[i][1] = BoxMon_GetAlternateForme(mon);
+}
+
+int Sav2_Poketch_PokemonHistoryGetFirstEmptySlot(struct SavePoketch * poketch)
+{
+    int i;
+    for (i = 0; i < 12; i++)
+    {
+        if (poketch->pokemonHistory[i][0] == SPECIES_NONE)
+            return i;
+    }
+    return i;
+}
+
+void Sav2_Poketch_PokemonHistoryGetSlotN(struct SavePoketch * poketch, s32 i, u32 * species_p, u32 * forme_p)
+{
+    GF_ASSERT(i < 12);
+    GF_ASSERT(poketch->pokemonHistory[i][0] != SPECIES_NONE);
+    *species_p = poketch->pokemonHistory[i][0];
+    *forme_p = poketch->pokemonHistory[i][1];
+}
+
+struct SavePoketch * Sav2_Poketch_get(struct SaveBlock2 * sav2)
+{
+    return (struct SavePoketch *)SavArray_get(sav2, 5);
+}