infinitefusion-e18/Data/Scripts/011_Battle/005_AI/003_AI_Switch.rb

class Battle::AI
  #=============================================================================
  # Decide whether the opponent should switch Pokémon
  #=============================================================================
  def pbEnemyShouldWithdraw?(idxBattler)
    return pbEnemyShouldWithdrawEx?(idxBattler, false)
  end

  def pbEnemyShouldWithdrawEx?(idxBattler, forceSwitch)
    return false if @battle.wildBattle?
    shouldSwitch = forceSwitch
    batonPass = -1
    moveType = nil
    skill = @battle.pbGetOwnerFromBattlerIndex(idxBattler).skill_level || 0
    battler = @battle.battlers[idxBattler]
    # If Pokémon is within 6 levels of the foe, and foe's last move was
    # super-effective and powerful
    if !shouldSwitch && battler.turnCount > 0 && skill >= PBTrainerAI.highSkill
      target = battler.pbDirectOpposing(true)
      if !target.fainted? && target.lastMoveUsed &&
         (target.level - battler.level).abs <= 6
        moveData = GameData::Move.get(target.lastMoveUsed)
        moveType = moveData.type
        typeMod = pbCalcTypeMod(moveType, target, battler)
        if Effectiveness.super_effective?(typeMod) && moveData.base_damage > 50
          switchChance = (moveData.base_damage > 70) ? 30 : 20
          shouldSwitch = (pbAIRandom(100) < switchChance)
        end
      end
    end
    # Pokémon can't do anything (must have been in battle for at least 5 rounds)
    if !@battle.pbCanChooseAnyMove?(idxBattler) &&
       battler.turnCount && battler.turnCount >= 5
      shouldSwitch = true
    end
    # Pokémon is Perish Songed and has Baton Pass
    if skill >= PBTrainerAI.highSkill && battler.effects[PBEffects::PerishSong] == 1
      battler.eachMoveWithIndex do |m, i|
        next if m.function != "SwitchOutUserPassOnEffects"   # Baton Pass
        next if !@battle.pbCanChooseMove?(idxBattler, i, false)
        batonPass = i
        break
      end
    end
    # Pokémon will faint because of bad poisoning at the end of this round, but
    # would survive at least one more round if it were regular poisoning instead
    if battler.status == :POISON && battler.statusCount > 0 &&
       skill >= PBTrainerAI.highSkill
      toxicHP = battler.totalhp / 16
      nextToxicHP = toxicHP * (battler.effects[PBEffects::Toxic] + 1)
      if battler.hp <= nextToxicHP && battler.hp > toxicHP * 2
        shouldSwitch = true if pbAIRandom(100) < 80
      end
    end
    # Pokémon is Encored into an unfavourable move
    if battler.effects[PBEffects::Encore] > 0 && skill >= PBTrainerAI.mediumSkill
      idxEncoredMove = battler.pbEncoredMoveIndex
      if idxEncoredMove >= 0
        scoreSum   = 0
        scoreCount = 0
        battler.allOpposing.each do |b|
          scoreSum += pbGetMoveScore(battler.moves[idxEncoredMove], battler, b, skill)
          scoreCount += 1
        end
        if scoreCount > 0 && scoreSum / scoreCount <= 20
          shouldSwitch = true if pbAIRandom(100) < 80
        end
      end
    end
    # If there is a single foe and it is resting after Hyper Beam or is
    # Truanting (i.e. free turn)
    if @battle.pbSideSize(battler.index + 1) == 1 &&
       !battler.pbDirectOpposing.fainted? && skill >= PBTrainerAI.highSkill
      opp = battler.pbDirectOpposing
      if opp.effects[PBEffects::HyperBeam] > 0 ||
         (opp.hasActiveAbility?(:TRUANT) && opp.effects[PBEffects::Truant])
        shouldSwitch = false if pbAIRandom(100) < 80
      end
    end
    # Sudden Death rule - I'm not sure what this means
    if @battle.rules["suddendeath"] && battler.turnCount > 0
      if battler.hp <= battler.totalhp / 4 && pbAIRandom(100) < 30
        shouldSwitch = true
      elsif battler.hp <= battler.totalhp / 2 && pbAIRandom(100) < 80
        shouldSwitch = true
      end
    end
    # Pokémon is about to faint because of Perish Song
    if battler.effects[PBEffects::PerishSong] == 1
      shouldSwitch = true
    end
    if shouldSwitch
      list = []
      idxPartyStart, idxPartyEnd = @battle.pbTeamIndexRangeFromBattlerIndex(idxBattler)
      @battle.pbParty(idxBattler).each_with_index do |pkmn, i|
        next if i == idxPartyEnd - 1   # Don't choose to switch in ace
        next if !@battle.pbCanSwitch?(idxBattler, i)
        # If perish count is 1, it may be worth it to switch
        # even with Spikes, since Perish Song's effect will end
        if battler.effects[PBEffects::PerishSong] != 1
          # Will contain effects that recommend against switching
          spikes = battler.pbOwnSide.effects[PBEffects::Spikes]
          # Don't switch to this if too little HP
          if spikes > 0
            spikesDmg = [8, 6, 4][spikes - 1]
            if pkmn.hp <= pkmn.totalhp / spikesDmg
              next if !pkmn.hasType?(:FLYING) && !pkmn.hasActiveAbility?(:LEVITATE)
            end
          end
        end
        # moveType is the type of the target's last used move
        if moveType && Effectiveness.ineffective?(pbCalcTypeMod(moveType, battler, battler))
          weight = 65
          typeMod = pbCalcTypeModPokemon(pkmn, battler.pbDirectOpposing(true))
          if Effectiveness.super_effective?(typeMod)
            # Greater weight if new Pokemon's type is effective against target
            weight = 85
          end
          list.unshift(i) if pbAIRandom(100) < weight   # Put this Pokemon first
        elsif moveType && Effectiveness.resistant?(pbCalcTypeMod(moveType, battler, battler))
          weight = 40
          typeMod = pbCalcTypeModPokemon(pkmn, battler.pbDirectOpposing(true))
          if Effectiveness.super_effective?(typeMod)
            # Greater weight if new Pokemon's type is effective against target
            weight = 60
          end
          list.unshift(i) if pbAIRandom(100) < weight   # Put this Pokemon first
        else
          list.push(i)   # put this Pokemon last
        end
      end
      if list.length > 0
        if batonPass >= 0 && @battle.pbRegisterMove(idxBattler, batonPass, false)
          PBDebug.log("[AI] #{battler.pbThis} (#{idxBattler}) will use Baton Pass to avoid Perish Song")
          return true
        end
        if @battle.pbRegisterSwitch(idxBattler, list[0])
          PBDebug.log("[AI] #{battler.pbThis} (#{idxBattler}) will switch with " +
                      "#{@battle.pbParty(idxBattler)[list[0]].name}")
          return true
        end
      end
    end
    return false
  end

  #=============================================================================
  # Choose a replacement Pokémon
  #=============================================================================
  def pbDefaultChooseNewEnemy(idxBattler, party)
    enemies = []
    idxPartyStart, idxPartyEnd = @battle.pbTeamIndexRangeFromBattlerIndex(idxBattler)
    party.each_with_index do |_p, i|
      next if i == idxPartyEnd - 1 && enemies.length > 0   # Ignore ace if possible
      enemies.push(i) if @battle.pbCanSwitchLax?(idxBattler, i)
    end
    return -1 if enemies.length == 0
    return pbChooseBestNewEnemy(idxBattler, party, enemies)
  end

  def pbChooseBestNewEnemy(idxBattler, party, enemies)
    return -1 if !enemies || enemies.length == 0
    best    = -1
    bestSum = 0
    enemies.each do |i|
      pkmn = party[i]
      sum  = 0
      pkmn.moves.each do |m|
        next if m.base_damage == 0
        @battle.battlers[idxBattler].allOpposing.each do |b|
          bTypes = b.pbTypes(true)
          sum += Effectiveness.calculate(m.type, bTypes[0], bTypes[1], bTypes[2])
        end
      end
      if best == -1 || sum > bestSum
        best = i
        bestSum = sum
      end
    end
    return best
  end
end