use constants for ball throw count

2 files changed, 9 insertions, 6 deletions

diff --git a/include/battle.h b/include/battle.h
index 8dad6bc25..a58f69cda 100644
--- a/include/battle.h
+++ b/include/battle.h
@@ -12,6 +12,8 @@
 #include "battle_util2.h"
 #include "battle_bg.h"
 
+#include "pokeball.h" // i'm not sure about this one
+
 #define GET_BATTLER_POSITION(battler)     (gBattlerPositions[battler])
 #define GET_BATTLER_SIDE(battler)         (GetBattlerPosition(battler) & BIT_SIDE)
 #define GET_BATTLER_SIDE2(battler)        (GET_BATTLER_POSITION(battler) & BIT_SIDE)
@@ -255,7 +257,7 @@ struct BattleResults
     u16 caughtMonSpecies;     // 0x28
     u8 caughtMonNick[POKEMON_NAME_LENGTH + 1];     // 0x2A
     u8 filler35;           // 0x35
-    u8 catchAttempts[11];     // 0x36
+    u8 catchAttempts[POKEBALL_COUNT-1];     // 0x36 Doesn't include Master ball
 };
 
 struct BattleTv_Side
diff --git a/src/tv.c b/src/tv.c
index 17af3edff..b401de06f 100644
--- a/src/tv.c
+++ b/src/tv.c
@@ -32,6 +32,7 @@
 #include "decoration.h"
 #include "secret_base.h"
 #include "tv.h"
+#include "pokeball.h"
 #include "data.h"
 #include "constants/battle_frontier.h"
 #include "constants/contest.h"
@@ -963,7 +964,7 @@ void GabbyAndTyBeforeInterview(void)
     }
     if (!gBattleResults.usedMasterBall)
     {
-        for (i = 0; i < 11; i ++)
+        for (i = 0; i < POKEBALL_COUNT-1; i ++)
         {
             if (gBattleResults.catchAttempts[i])
             {
@@ -1135,7 +1136,7 @@ void PutPokemonTodayCaughtOnAir(void)
             sCurTVShowSlot = FindEmptyTVSlotBeyondFirstFiveShowsOfArray(gSaveBlock1Ptr->tvShows);
             if (sCurTVShowSlot != -1 && HasMixableShowAlreadyBeenSpawnedWithPlayerID(TVSHOW_POKEMON_TODAY_CAUGHT, FALSE) != TRUE)
             {
-                for (i = 0; i < 11; i ++)
+                for (i = 0; i < POKEBALL_COUNT-1; i ++)
                 {
                     ct += gBattleResults.catchAttempts[i];
                 }
@@ -1152,7 +1153,7 @@ void PutPokemonTodayCaughtOnAir(void)
                     }
                     else
                     {
-                        for (i = 0; i < 11; i ++)
+                        for (i = 0; i < POKEBALL_COUNT-1; i ++)
                         {
                             ct += gBattleResults.catchAttempts[i];
                         }
@@ -1203,7 +1204,7 @@ void PutPokemonTodayFailedOnTheAir(void)
 
     if (!rbernoulli(1, 1))
     {
-        for (i = 0, ct = 0; i < 11; i ++)
+        for (i = 0, ct = 0; i < POKEBALL_COUNT-1; i ++)
         {
             ct += gBattleResults.catchAttempts[i];
         }
@@ -2228,7 +2229,7 @@ void sub_80EE184(void)
         show->breakingNews.kind = TVSHOW_BREAKING_NEWS;
         show->breakingNews.active = FALSE;
         balls = 0;
-        for (i = 0; i < 11; i ++)
+        for (i = 0; i < POKEBALL_COUNT-1; i ++)
         {
             balls += gBattleResults.catchAttempts[i];
         }