Clean up field_poison.c

src/field_poison.c

@@ -21,9 +21,8 @@ static bool32 IsMonValidSpecies(struct Pokemon *pokemon)
 {
     u16 species = GetMonData(pokemon, MON_DATA_SPECIES2);
     if (species == SPECIES_NONE || species == SPECIES_EGG)
-    {
         return FALSE;
-    }
+
     return TRUE;
 }
 
@@ -35,16 +34,14 @@ static bool32 AllMonsFainted(void)
     for (i = 0; i < PARTY_SIZE; i++, pokemon++)
     {
         if (IsMonValidSpecies(pokemon) && GetMonData(pokemon, MON_DATA_HP) != 0)
-        {
             return FALSE;
-        }
     }
     return TRUE;
 }
 
 static void FaintFromFieldPoison(u8 partyIdx)
 {
-    struct Pokemon *pokemon = gPlayerParty + partyIdx;
+    struct Pokemon *pokemon = &gPlayerParty[partyIdx];
     u32 status = STATUS1_NONE;
 
     AdjustFriendship(pokemon, FRIENDSHIP_EVENT_FAINT_FIELD_PSN);
@@ -55,49 +52,47 @@ static void FaintFromFieldPoison(u8 partyIdx)
 
 static bool32 MonFaintedFromPoison(u8 partyIdx)
 {
-    struct Pokemon *pokemon = gPlayerParty + partyIdx;
+    struct Pokemon *pokemon = &gPlayerParty[partyIdx];
     if (IsMonValidSpecies(pokemon) && GetMonData(pokemon, MON_DATA_HP) == 0 && GetAilmentFromStatus(GetMonData(pokemon, MON_DATA_STATUS)) == AILMENT_PSN)
-    {
         return TRUE;
-    }
+
     return FALSE;
 }
 
+#define tState    data[0]
+#define tPartyIdx data[1]
+
 static void Task_TryFieldPoisonWhiteOut(u8 taskId)
 {
     s16 *data = gTasks[taskId].data;
-    switch (data[0])
+    switch (tState)
     {
         case 0:
-            for (; data[1] < PARTY_SIZE; data[1]++)
+            for (; tPartyIdx < PARTY_SIZE; tPartyIdx++)
             {
-                if (MonFaintedFromPoison(data[1]))
+                if (MonFaintedFromPoison(tPartyIdx))
                 {
-                    FaintFromFieldPoison(data[1]);
-                    ShowFieldMessage(gText_PkmnFainted3);
-                    data[0]++;
+                    FaintFromFieldPoison(tPartyIdx);
+                    ShowFieldMessage(gText_PkmnFainted_FldPsn);
+                    tState++;
                     return;
                 }
             }
-            data[0] = 2;
+            tState = 2; // Finished checking party
             break;
         case 1:
+            // Wait for "{mon} fainted" message, then return to party loop
             if (IsFieldMessageBoxHidden())
-            {
-                data[0]--;
-            }
+                tState--;
             break;
         case 2:
             if (AllMonsFainted())
             {
+                // Battle facilities have their own white out script to handle the challenge loss
                 if (InBattlePyramid() | InBattlePike() || InTrainerHillChallenge())
-                {
                     gSpecialVar_Result = FLDPSN_FRONTIER_WHITEOUT;
-                }
                 else
-                {
                     gSpecialVar_Result = FLDPSN_WHITEOUT;
-                }
             }
             else
             {
@@ -109,6 +104,9 @@ static void Task_TryFieldPoisonWhiteOut(u8 taskId)
     }
 }
 
+#undef tState
+#undef tPartyIdx
+
 void TryFieldPoisonWhiteOut(void)
 {
     CreateTask(Task_TryFieldPoisonWhiteOut, 80);
@@ -122,31 +120,31 @@ s32 DoPoisonFieldEffect(void)
     struct Pokemon *pokemon = gPlayerParty;
     u32 numPoisoned = 0;
     u32 numFainted = 0;
+
     for (i = 0; i < PARTY_SIZE; i++)
     {
         if (GetMonData(pokemon, MON_DATA_SANITY_HAS_SPECIES) && GetAilmentFromStatus(GetMonData(pokemon, MON_DATA_STATUS)) == AILMENT_PSN)
         {
+            // Apply poison damage
             hp = GetMonData(pokemon, MON_DATA_HP);
             if (hp == 0 || --hp == 0)
-            {
                 numFainted++;
-            }
+
             SetMonData(pokemon, MON_DATA_HP, &hp);
             numPoisoned++;
         }
         pokemon++;
     }
+
+    // Do screen flash effect
     if (numFainted != 0 || numPoisoned != 0)
-    {
         FldEffPoison_Start();
-    }
+
     if (numFainted != 0)
-    {
         return FLDPSN_FNT;
-    }
+
     if (numPoisoned != 0)
-    {
         return FLDPSN_PSN;
-    }
+
     return FLDPSN_NONE;
 }