//Kill off team if it consists only of Zombie Worms
//Mod by raffie
void destroyTeam(int teamIndex){
  for (local wormIndex = 1; wormIndex <= 8; wormIndex++){
    GS->Info.SetWormHealth(teamIndex, wormIndex, 0);
  }
}

override void CWorm::Message(CObject* sender,EMType Type,int MSize,CMessageData* MData)
{
super;
if(Type == M_FRAME)
//if(Type == M_TURNEND)
         {
       //local w = GetCurrentWorm();

      //local fullWormCount = 0;
      //local teamCount = 0;
      //local initialWormCount = 0;
      //local wormCount = 0;
      //local zombieWormCount = 0;
     
      int fullWormCount = 0;
      int teamCount = 0;
      int initialWormCount = 0;
      int wormCount = 0;
      int zombieWormCount = 0;
      local wormTeamLoop;
     
      //Get amount of Worms in Team at Start
      for (local worm = 0; worm < GS->Info.Teams[1].NWorms; worm++){
          initialWormCount++;
        }
     
      //Get amount of teams
      for (local team = 0; team < 6; team++) {
      for (local worm0 = 0; worm0 < GS->Info.Teams[team].NWorms; worm0++){
          fullWormCount++;
        }
      }
      teamCount = (fullWormCount/initialWormCount);
     
     
     
     
      for (local team = 0; team < teamCount; team++) {
        wormCount = 0;
        zombieWormCount = 0;
        wormTeamLoop = (team+1);
      //Current Worm count
      for (local worm = 0; worm < GS->Info.Teams[team].NWorms; worm++){
        if (GS->Info.Teams[team].Worms[worm].State != 135){ // if not dead
          wormCount++;
         }
            //if(wormTeamLoop == teamCount){ wormTeamLoop = 0; }
         
         //if (GS->Info.Teams[wormTeamLoop].Worms[worm].State != 135){ // if not dead 
            local w = CWorm(Root->GetObject(2, 16 * wormTeamLoop + worm));
            //if (w->Zombie){ zombieWormCount++; }
            if(w != NullObj){if(w->Zombie == true)zombieWormCount = zombieWormCount + 1;}
         //}
        }
      //GG->WriteToChat(14, StrCon(StrCon("TEAM ", displayInt(team)), ""), false);
      //GG->WriteToChat(14, StrCon(StrCon("LOOP ", displayInt(wormTeamLoop)), ""), false);
      if(w != NullObj){
      //GG->WriteToChat(14, StrCon(StrCon(StrCon(StrCon("TEAM ", IntToC(GetTeamColor(w->WormTeam))), ": "), displayInt(wormTeamLoop)), ""), false);
      //GG->WriteToChat(14, StrCon(StrCon(StrCon(StrCon("TEAM ", displayInt(wormTeamLoop)), ": "), IntToC(GetTeamColor(w->WormTeam))), ""), false);
      }else{
      //GG->WriteToChat(14, StrCon(StrCon("TEAM ", displayInt(wormTeamLoop)), ""), false);
      }
      //GG->WriteToChat(15, StrCon(StrCon("Current worm count: ", displayInt(wormCount)), ""), false);
      //GG->WriteToChat(15, StrCon(StrCon("Zombie worm count: ", displayInt(zombieWormCount)), ""), false);
      //GG->WriteToChat(15, StrCon(StrCon(StrCon(StrCon("Worms: ", displayInt(wormCount)), " // Zombies: "), displayInt(zombieWormCount)), ""), false);
     
      if(zombieWormCount == wormCount)
      {
      //GG->WriteToChat(13, StrCon(StrCon("TEAM DEAD: ", displayInt(wormTeamLoop)), ""), false);
      destroyTeam(wormTeamLoop);
      }
     
     
      }       

         
         }   
 }

override int CTurnGame::GetNextTeam()
{
if(nocrash <= 100)
{
 local nt = super;
 local wormnum = 0;
       for(local t = 1; t<9; t++)
       {
        local w = CWorm(Root->GetObject(2, 16 * nt + t));
        if(w != NullObj){if(w->Zombie == false)wormnum = wormnum + 1;}
       }
 if(wormnum > 0){return nt;}
 else
 {
 CurTeam = nt;
 nocrash = nocrash + 1;
 return GetNextTeam();
 }
}
else
{
super;
}
}

require utils;
 
int nonzombies[7]; //index is team (1-6) 0 is unused
int zombies[7];   //index is team


void script::FirstFrame()     
{
        for(local t = 0;t<7;t++)
        {
        nonzombies[t] = 0;
        zombies[t] = 0;
        }
}

 
override void CTurnGame::Message(CObject* sender,EMType Type,int MSize,CMessageData* MData)
{
          if(Type == M_FRAME)
          {
                  for(local i = 1; i < Env->Objs.Count; i+=1)
                         {
                         local obj = Env->Objs.Objs[i];
                               if(obj == NullObj) continue;
                               if(obj->ClType == OC_Worm)
                               { 
                               CWorm* wyrm;
                               wyrm = obj;
                                 if(zombies[wyrm->WormTeam] != -1)  //this is so it does not kill them repeatedly
                                 {
                                     if(wyrm->Zombie == true)
                                     {
                                     zombies[wyrm->WormTeam] = zombies[wyrm->WormTeam] + 1;
                                     }
                                     else
                                     {
                                     nonzombies[wyrm->WormTeam] = nonzombies[wyrm->WormTeam] + 1;
                                     }
                                 }
                               }
                         }
                         
                         for(local m = 0; m<7; m++)
                         {
                            if(nonzombies[m] == 0 && zombies[m] > 0)
                            {
                            destroyTeam(m);
                            nonzombies[m] = 0;
                            zombies[m] = -1; //this is so it does not kill them repeatedly
                            }
                            else
                            {
                            nonzombies[m] = 0;
                            zombies[m] = 0;
                            }
                         }
          }
super;
}

void destroyTeam(int teamIndex)
{
  for (local wormIndex = 1; wormIndex <= 8; wormIndex++)
  {
    GS->Info.SetWormHealth(teamIndex, wormIndex, 0);
  }
}