infinitefusion-e18/Data/Scripts/014_Trainers/001_Trainer.rb

#===============================================================================
# Basic trainer class (use a child class rather than this one)
#===============================================================================
class Trainer
  attr_accessor :trainer_type
  attr_accessor :name
  attr_accessor :id
  attr_accessor :language
  attr_accessor :party

  def full_name
    return _INTL("{1} {2}", trainer_type_name, @name)
  end

  #=============================================================================

  # Portion of the ID which is visible on the Trainer Card
  def public_ID(id = nil)
    return id ? id & 0xFFFF : @id & 0xFFFF
  end

  # Other portion of the ID
  def secret_ID(id = nil)
    return id ? id >> 16 : @id >> 16
  end

  # Random ID other than this Trainer's ID
  def make_foreign_ID
    loop do
      ret = rand(2 ** 16) | rand(2 ** 16) << 16
      return ret if ret != @id
    end
    return 0
  end

  #=============================================================================

  def trainer_type_name; return GameData::TrainerType.get(@trainer_type).name;        end
  def base_money;        return GameData::TrainerType.get(@trainer_type).base_money;  end
  def gender;            return GameData::TrainerType.get(@trainer_type).gender;      end
  def male?;             return GameData::TrainerType.get(@trainer_type).male?;       end
  def female?;           return GameData::TrainerType.get(@trainer_type).female?;     end
  alias isMale? male?
  alias isFemale? female?
  def skill_level;       return GameData::TrainerType.get(@trainer_type).skill_level; end
  def skill_code;        return GameData::TrainerType.get(@trainer_type).skill_code;  end

  def has_skill_code?(code)
    c = skill_code
    return c && c != "" && c[/#{code}/]
  end

  #=============================================================================

  def pokemon_party
    return @party.find_all { |p| p && !p.egg? }
  end

  def able_party
    return @party.find_all { |p| p && !p.egg? && !p.fainted? }
  end

  def party_count
    return @party.length
  end

  def pokemon_count
    ret = 0
    @party.each { |p| ret += 1 if p && !p.egg? }
    return ret
  end

  def able_pokemon_count
    ret = 0
    @party.each { |p| ret += 1 if p && !p.egg? && !p.fainted? }
    return ret
  end

  def party_full?
    return party_count >= MAX_PARTY_SIZE
  end

  # Returns true if there are no usable Pokémon in the player's party.
  def all_fainted?
    return able_pokemon_count == 0
  end

  def first_party
    return @party[0]
  end

  def first_pokemon
    return pokemon_party[0]
  end

  def first_able_pokemon
    return able_party[0]
  end

  def last_party
    return (@party.length > 0) ? @party[@party.length - 1] : nil
  end

  def last_pokemon
    p = pokemon_party
    return (p.length > 0) ? p[p.length - 1] : nil
  end

  def last_able_pokemon
    p = able_party
    return (p.length > 0) ? p[p.length - 1] : nil
  end

  def remove_pokemon_at_index(index)
    return false if index < 0 || index >= party_count
    have_able = false
    @party.each_with_index do |pkmn, i|
      have_able = true if i != index && pkmn.able?
      break if have_able
    end
    return false if !have_able
    @party.delete_at(index)
    return true
  end

  # Checks whether the trainer would still have an unfainted Pokémon if the
  # Pokémon given by _index_ were removed from the party.
  def has_other_able_pokemon?(index)
    @party.each_with_index { |pkmn, i| return true if i != index && pkmn.able? }
    return false
  end

  # Returns true if there is a Pokémon of the given species in the trainer's
  # party. You may also specify a particular form it should be.
  def has_species?(species, form = -1)
    return pokemon_party.any? { |p| p && p.isSpecies?(species) && (form < 0 || p.form == form) }
  end

  # Returns whether there is a fatefully met Pokémon of the given species in the
  # trainer's party.
  def has_fateful_species?(species)
    return pokemon_party.any? { |p| p && p.isSpecies?(species) && p.obtain_method == 4 }
  end

  # Returns whether there is a Pokémon with the given type in the trainer's
  # party.
  def has_pokemon_of_type?(type)
    return false if !GameData::Type.exists?(type)
    type = GameData::Type.get(type).id
    return pokemon_party.any? { |p| p && p.hasType(type) }
  end

  # Checks whether any Pokémon in the party knows the given move, and returns
  # the first Pokémon it finds with that move, or nil if no Pokémon has that move.
  def get_pokemon_with_move(move)
    pokemon_party.each { |pkmn| return pkmn if pkmn.hasMove?(move) }
    return nil
  end

  # Fully heal all Pokémon in the party.
  def heal_party
    @party.each { |pkmn| pkmn.heal }
  end

  #=============================================================================

  def initialize(name, trainer_type)
    @trainer_type = trainer_type
    @name         = name
    @id           = rand(2 ** 16) | rand(2 ** 16) << 16
    @language     = pbGetLanguage
    @party        = []
  end
end

#===============================================================================
# Trainer class for NPC trainers
#===============================================================================
class NPCTrainer < Trainer
  attr_accessor :items
  attr_accessor :lose_text

  def initialize(name, trainer_type)
    super
    @items     = []
    @lose_text = GameData::TrainerType.get(@trainer_type).lose_text
  end
end

#===============================================================================
# Trainer class for the player
#===============================================================================
class PlayerTrainer < Trainer
  attr_writer   :character_ID
  attr_accessor :outfit
  attr_accessor :badges
  attr_reader   :money
  attr_accessor :seen
  attr_accessor :owned
  attr_accessor :seen_forms
  attr_accessor :last_seen_forms
  attr_accessor :owned_shadow
  attr_accessor :pokedex                 # Whether the Pokédex was obtained
  attr_accessor :pokegear                # Whether the Pokégear was obtained
  attr_accessor :mystery_gift_unlocked   # Whether MG can be used from load screen
  attr_accessor :mystery_gifts           # Variable that stores downloaded MG data

  def character_ID
    @character_ID = $PokemonGlobal.playerID || 0 if !@character_ID
    return @character_ID
  end

  def money=(value)
    @money = value.clamp(0, MAX_MONEY)
  end

  def badge_count
    ret = 0
    @badges.each { |b| ret += 1 if b }
    return ret
  end

  #=============================================================================

  def seen?(species)
    species_data = GameData::Species.try_get(species)
    return (species_data) ? @seen[species_data.species] : false
  end
  alias hasSeen? seen?

  def owned?(species)
    species_data = GameData::Species.try_get(species)
    return (species_data) ? @owned[species_data.species] : false
  end
  alias hasOwned? owned?

  def set_seen(species)
    species_data = GameData::Species.try_get(species)
    @seen[species_data.species] = true if species_data
  end

  def set_owned(species)
    species_data = GameData::Species.try_get(species)
    @owned[species_data.species] = true if species_data
  end

  def seen_count(region = -1)
    ret = 0
    if region == -1
      GameData::Species.each { |s| ret += 1 if s.form == 0 && @seen[s.species] }
    else
      pbAllRegionalSpecies(region).each { |s| ret += 1 if s && @seen[s] }
    end
    return ret
  end

  def seen_any?(region = -1)
    if region == -1
      GameData::Species.each { |s| return true if s.form == 0 && @seen[s.species] }
    else
      pbAllRegionalSpecies(region).each { |s| return true if s && @seen[s] }
    end
    return false
  end

  def owned_count(region = -1)
    ret = 0
    if region == -1
      GameData::Species.each { |s| ret += 1 if s.form == 0 && @owned[s.species] }
    else
      pbAllRegionalSpecies(region).each { |s| ret += 1 if s && @owned[s] }
    end
    return ret
  end

  def seen_forms_count(species)
    species_data = GameData::Species.try_get(species)
    return 0 if !species_data
    species = species_data.species
    ret = 0
    @seen_forms[species] = [[], []] if !@seen_forms[species]
    array = @seen_forms[species]
    for i in 0...[array[0].length, array[1].length].max
      ret += 1 if array[0][i] || array[1][i]
    end
    return ret
  end

  def clear_pokedex
    @seen            = {}
    @owned           = {}
    @seen_forms      = {}
    @last_seen_forms = {}
    @owned_shadow    = {}
  end

  #=============================================================================

  def initialize(name, trainer_type)
    super
    @character_ID          = nil
    @outfit                = 0
    @badges                = [false] * 8
    @money                 = INITIAL_MONEY
    clear_pokedex
    @pokegear              = false
    @pokedex               = false
    @mystery_gift_unlocked = false
    @mystery_gifts         = []
  end
end