pokerogue/src/battle-info.ts

import { default as Pokemon } from './pokemon';
import { getLevelTotalExp, getLevelRelExp } from './exp';
import * as Utils from './utils';
import { addTextObject, TextStyle } from './text';
import { getGenderSymbol, getGenderColor } from './gender';

export default class BattleInfo extends Phaser.GameObjects.Container {
  private player: boolean;
  private lastHp: integer;
  private lastMaxHp: integer;
  private lastHpFrame: string;
  private lastExp: integer;
  private lastLevelExp: integer;
  private lastLevel: integer;

  private nameText: Phaser.GameObjects.Text;
  private genderText: Phaser.GameObjects.Text;
  private levelContainer: Phaser.GameObjects.Container;
  private hpBar: Phaser.GameObjects.Image;
  private levelNumbersContainer: Phaser.GameObjects.Container;
  private hpNumbersContainer: Phaser.GameObjects.Container;
  private expBar: Phaser.GameObjects.Image;

  constructor(scene: Phaser.Scene, x: number, y: number, player: boolean) {
    super(scene, x, y);
    this.player = player;
    this.lastHp = -1;
    this.lastMaxHp = -1;
    this.lastHpFrame = null;
    this.lastExp = -1;
    this.lastLevelExp = -1;
    this.lastLevel = -1;

    // Initially invisible and shown via Pokemon.showInfo
    this.setVisible(false);

    const box = this.scene.add.image(0, 0, `pbinfo_${player ? 'player' : 'enemy'}`);
    box.setOrigin(1, 0.5);
    this.add(box);

    this.nameText = addTextObject(this.scene, player ? -115 : -124, player ? -15.2 : -11.2, '', TextStyle.BATTLE_INFO);
    this.nameText.setOrigin(0, 0);
    this.add(this.nameText);

    this.genderText = addTextObject(this.scene, 0, 0, '', TextStyle.BATTLE_INFO);
    this.genderText.setOrigin(0, 0);
    this.genderText.setPositionRelative(this.nameText, 0, 2);
    this.add(this.genderText);

    this.levelContainer = this.scene.add.container(player ? -41 : -50, player ? -10 : -5);
    this.add(this.levelContainer);

    const levelOverlay = this.scene.add.image(0, 0, 'overlay_lv');
    this.levelContainer.add(levelOverlay);

    this.hpBar = this.scene.add.image(player ? -61 : -71, player ? -1 : 4.5, 'overlay_hp');
    this.hpBar.setOrigin(0);
    this.add(this.hpBar);

    this.levelNumbersContainer = this.scene.add.container(9.5, 0);
    this.levelContainer.add(this.levelNumbersContainer);

    if (this.player) {
      this.hpNumbersContainer = this.scene.add.container(-15, 10);
      this.add(this.hpNumbersContainer);

      const expBar = this.scene.add.image(-98, 18, 'overlay_exp');
      expBar.setOrigin(0);
      expBar.setScale(0, 1);
      this.add(expBar);

      this.expBar = expBar;
    }
  }

  initInfo(pokemon: Pokemon) {
    this.nameText.setText(pokemon.name);

    const nameSizeTest = addTextObject(this.scene, 0, 0, pokemon.name, TextStyle.BATTLE_INFO);
    const nameTextWidth = nameSizeTest.displayWidth;
    nameSizeTest.destroy();

    this.genderText.setText(getGenderSymbol(pokemon.gender));
    this.genderText.setColor(getGenderColor(pokemon.gender));
    this.genderText.setPositionRelative(this.nameText, nameTextWidth, 0);

    this.hpBar.setScale(pokemon.getHpRatio(), 1);
    if (this.player)
      this.setHpNumbers(pokemon.hp, pokemon.getMaxHp());
    this.lastHp = pokemon.hp;
    this.lastHpFrame = this.hpBar.scaleX > 0.5 ? 'high' : this.hpBar.scaleX > 0.25 ? 'medium' : 'low';
    this.lastMaxHp = pokemon.getMaxHp();

    this.setLevel(pokemon.level);
    this.lastLevel = pokemon.level;

    if (this.player) {
      this.expBar.setScale(pokemon.levelExp / getLevelTotalExp(pokemon.level, pokemon.species.growthRate), 1);
      this.lastExp = pokemon.exp;
      this.lastLevelExp = pokemon.levelExp;
    }
  }

  updateInfo(pokemon: Pokemon, callback?: Function) {
    if (!this.scene) {
      if (callback)
        callback();
      return;
    }

    const updatePokemonHp = () => {
      const duration = Utils.clampInt(Math.abs((this.lastHp) - pokemon.hp) * 5, 250, 5000);
      this.scene.tweens.add({
        targets: this.hpBar,
        ease: 'Sine.easeOut',
        scaleX: pokemon.getHpRatio(),
        duration: duration,
        onUpdate: () => {
          if (this.player && this.lastHp !== pokemon.hp) {
            const tweenHp = Math.ceil(this.hpBar.scaleX * pokemon.getMaxHp());
            this.setHpNumbers(tweenHp, pokemon.getMaxHp())
            this.lastHp = tweenHp;
          }

          const hpFrame = this.hpBar.scaleX > 0.5 ? 'high' : this.hpBar.scaleX > 0.25 ? 'medium' : 'low';
          if (hpFrame !== this.lastHpFrame) {
            this.hpBar.setFrame(hpFrame);
            this.lastHpFrame = hpFrame;
          }
        },
        onComplete: () => {
          if (callback) {
            callback();
            callback = null;
          }
        }
      });
      if (!this.player)
        this.lastHp = pokemon.hp;
      this.lastMaxHp = pokemon.getMaxHp();
    };

    if (this.player && this.lastExp !== pokemon.exp) {
      const originalCallback = callback;
      callback = () => this.updatePokemonExp(pokemon, originalCallback);
    }

    if (this.lastHp !== pokemon.hp || this.lastMaxHp !== pokemon.getMaxHp())
      updatePokemonHp();
    else if (!this.player && this.lastLevel !== pokemon.level) {
      this.setLevel(pokemon.level);
      this.lastLevel = pokemon.level;
      if (callback)
        callback();
    } else if (callback)
      callback();
  }

  updatePokemonExp(battler: Pokemon, callback?: Function) {
    const levelUp = this.lastLevel < battler.level;
    const relLevelExp = getLevelRelExp(this.lastLevel + 1, battler.species.growthRate);
    const levelExp = levelUp ? relLevelExp : battler.levelExp;
    let ratio = levelExp / relLevelExp;
    let duration = this.visible ? ((levelExp - this.lastLevelExp) / relLevelExp) * 1650 : 0;
    if (duration)
      this.scene.sound.play('exp');
    this.scene.tweens.add({
      targets: this.expBar,
      ease: 'Sine.easeIn',
      scaleX: ratio,
      duration: duration,
      onUpdate: () => {
        if (this.player && this.lastHp !== battler.hp) {
          const tweenHp = Math.ceil(this.hpBar.scaleX * battler.getMaxHp());
          this.setHpNumbers(tweenHp, battler.getMaxHp());
          this.lastHp = tweenHp;
        }

        const hpFrame = this.hpBar.scaleX > 0.5 ? 'high' : this.hpBar.scaleX > 0.25 ? 'medium' : 'low';
        if (hpFrame !== this.lastHpFrame) {
          this.hpBar.setFrame(hpFrame);
          this.lastHpFrame = hpFrame;
        }
      },
      onComplete: () => {
        if (duration)
          this.scene.sound.stopByKey('exp');
        if (ratio === 1) {
          this.lastLevelExp = 0;
          this.lastLevel++;
          this.scene.sound.play('level_up');
          this.setLevel(this.lastLevel);
          this.scene.time.delayedCall(500, () => {
            this.expBar.setScale(0, 1);
            this.updateInfo(battler, callback);
          });
          return;
        } else {
          this.lastExp = battler.exp;
          this.lastLevelExp = battler.levelExp;
        }
        if (callback) {
          callback();
          callback = null;
        }
      }
    });
  }

  setLevel(level: integer) {
    this.levelNumbersContainer.removeAll();
    const levelStr = level.toString();
    for (let i = 0; i < levelStr.length; i++)
      this.levelNumbersContainer.add(this.scene.add.image(i * 8, 0, 'numbers', levelStr[i]));
    this.levelContainer.setX((this.player ? -41 : -50) - 8 * Math.max(levelStr.length - 3, 0));
  }

  setHpNumbers(hp: integer, maxHp: integer) {
    if (!this.player)
      return;
    this.hpNumbersContainer.removeAll();
    const hpStr = hp.toString();
    const maxHpStr = maxHp.toString();
    let offset = 0;
    for (let i = maxHpStr.length - 1; i >= 0; i--)
      this.hpNumbersContainer.add(this.scene.add.image(offset++ * -8, 0, 'numbers', maxHpStr[i]));
    this.hpNumbersContainer.add(this.scene.add.image(offset++ * -8, 0, 'numbers', '/'));
    for (let i = hpStr.length - 1; i >= 0; i--)
      this.hpNumbersContainer.add(this.scene.add.image(offset++ * -8, 0, 'numbers', hpStr[i]));
  }
}

export class PlayerBattleInfo extends BattleInfo {
  constructor(scene: Phaser.Scene) {
    super(scene, Math.floor(scene.game.canvas.width / 6) - 10, -72, true);
  }
}

export class EnemyBattleInfo extends BattleInfo {
  constructor(scene: Phaser.Scene) {
    super(scene, 140, -141, false);
  }
}