Added AI objects for trainers, battlers and the move being assessed, logging battle messages now also echoes them to the console

Data/Scripts/011_Battle/005_AI/001_Battle_AI.rb

@@ -2,93 +2,61 @@
 #
 #===============================================================================
 class Battle::AI
-  # AI skill levels:
-  #     0:     Wild Pokémon
-  #     1-31:  Basic trainer (young/inexperienced)
-  #     32-47: Some skill
-  #     48-99: High skill
-  #     100+:  Best trainers (Gym Leaders, Elite Four, Champion)
-  # NOTE: A trainer's skill value can range from 0-255, but by default only four
-  #       distinct skill levels exist. The skill value is typically the same as
-  #       the trainer's base money value.
-  module AILevel
-    # Minimum skill level to be in each AI skill bracket.
-    def self.minimum; return 1;   end
-    def self.medium;  return 32;  end
-    def self.high;    return 48;  end
-    def self.best;    return 100; end
-  end
+  attr_reader :battle
+  attr_reader :trainer
+  attr_reader :user, :target, :move
 
-  #=============================================================================
-  #
-  #=============================================================================
   def initialize(battle)
-    @battle      = battle
-    @skill       = 0
-    @user        = nil
-    @wildBattler = @battle.wildBattle?   # Whether AI is choosing for a wild Pokémon
-    @roles       = [Array.new(@battle.pbParty(0).length) { |i| determine_roles(0, i) },
-                    Array.new(@battle.pbParty(1).length) { |i| determine_roles(1, i) }]
+    @battle = battle
+
+    # TODO: Move this elsewhere?
+    @roles   = [Array.new(@battle.pbParty(0).length) { |i| determine_roles(0, i) },
+                Array.new(@battle.pbParty(1).length) { |i| determine_roles(1, i) }]
   end
 
-  def pbAIRandom(x); return rand(x); end
-
-  def pbStdDev(choices)
-    sum = 0
-    n   = 0
-    choices.each do |c|
-      sum += c[1]
-      n   += 1
+  def create_ai_objects
+    # Initialize AI trainers
+    @trainers = [[], []]
+    @battle.player.each_with_index do |trainer, i|
+      @trainers[0][i] = AITrainer.new(self, 0, i, trainer)
     end
-    return 0 if n < 2
-    mean = sum.to_f / n
-    varianceTimesN = 0
-    choices.each do |c|
-      next if c[1] <= 0
-      deviation = c[1].to_f - mean
-      varianceTimesN += deviation * deviation
+    if @battle.wildBattle?
+      @trainers[1][0] = AITrainer.new(self, 1, 0, nil)
+    else
+      @battle.opponent.each_with_index do |trainer, i|
+        @trainers[1][i] = AITrainer.new(self, 1, i, trainer)
+      end
     end
-    # Using population standard deviation
-    # [(n-1) makes it a sample std dev, would be 0 with only 1 sample]
-    return Math.sqrt(varianceTimesN / n)
+    # Initialize AI battlers
+    @battlers = []
+    @battle.battlers.each_with_index do |battler, i|
+      @battlers[i] = AIBattler.new(self, i) if battler
+    end
+    # Initialize AI move object
+    @move = AIMove.new(self)
   end
 
-  # Decide whether the opponent should Mega Evolve their Pokémon.
-  def pbEnemyShouldMegaEvolve?
-    if @battle.pbCanMegaEvolve?(@user.index)   # Simple "always should if possible"
-      PBDebug.log("[AI] #{@user.pbThis} (#{@user.index}) will Mega Evolve")
-      return true
-    end
-    return false
+  # Set some class variables for the Pokémon whose action is being chosen
+  def set_up(idxBattler)
+    # Find the AI trainer choosing the action
+    opposes = @battle.opposes?(idxBattler)
+    trainer_index = @battle.pbGetOwnerIndexFromBattlerIndex(idxBattler)
+    @trainer = @trainers[(opposes) ? 1 : 0][trainer_index]
+    # Find the AI battler for which the action is being chosen
+    @user = @battlers[idxBattler]
+    @user.refresh_battler
   end
 
   # Choose an action.
   def pbDefaultChooseEnemyCommand(idxBattler)
     set_up(idxBattler)
-    choices = pbGetMoveScores
-    return if pbEnemyShouldUseItem?
     return if pbEnemyShouldWithdraw?
+    return if pbEnemyShouldUseItem?
     return if @battle.pbAutoFightMenu(idxBattler)
     @battle.pbRegisterMegaEvolution(idxBattler) if pbEnemyShouldMegaEvolve?
+    choices = pbGetMoveScores
     pbChooseMove(choices)
   end
-
-  # Set some class variables for the Pokémon whose action is being chosen
-  def set_up(idxBattler)
-    # TODO: Where relevant, pretend the user is Mega Evolved if it isn't but can
-    #       be.
-    @user        = @battle.battlers[idxBattler]
-    @wildBattler = (@battle.wildBattle? && @user.opposes?)
-    @skill       = 0
-    if !@wildBattler
-      @skill     = @battle.pbGetOwnerFromBattlerIndex(@user.index).skill_level || 0
-      @skill     = AILevel.minimum if @skill < AILevel.minimum
-    end
-  end
-
-  def skill_check(threshold)
-    return @skill >= threshold
-  end
 end
 
 #===============================================================================