Add Pokeball select

src/ui/ball-ui-handler.ts

@@ -0,0 +1,123 @@
+import { CommandPhase } from "../battle-phase";
+import BattleScene from "../battle-scene";
+import { getPokeballName, PokeballType } from "../pokeball";
+import { addTextObject, TextStyle } from "../text";
+import { Command } from "./command-ui-handler";
+import { Mode } from "./ui";
+import UiHandler from "./uiHandler";
+
+export default class BallUiHandler extends UiHandler {
+  private pokeballSelectContainer: Phaser.GameObjects.Container;
+  private pokeballSelectBg: Phaser.GameObjects.Image;
+  private countsText: Phaser.GameObjects.Text;
+
+  private cursorObj: Phaser.GameObjects.Image;
+
+  constructor(scene: BattleScene) {
+    super(scene, Mode.BALL);
+  }
+
+  setup() {
+    const ui = this.getUi();
+    
+    this.pokeballSelectContainer = this.scene.add.container((this.scene.game.canvas.width / 6) - 115, -49);
+    this.pokeballSelectContainer.setVisible(false);
+    ui.add(this.pokeballSelectContainer);
+
+    this.pokeballSelectBg = this.scene.add.image(0, 0, 'ball_window');
+    this.pokeballSelectBg.setOrigin(0, 1);
+    this.pokeballSelectContainer.add(this.pokeballSelectBg);
+
+    let optionsTextContent = '';
+
+    for (let pb = 0; pb < Object.keys(this.scene.pokeballCounts).length; pb++)
+      optionsTextContent += `${getPokeballName(pb)}\n`;
+    optionsTextContent += 'CANCEL';
+    const optionsText = addTextObject(this.scene, 0, 0, optionsTextContent, TextStyle.WINDOW, { align: 'right', maxLines: 6 });
+    optionsText.setOrigin(0, 0);
+    optionsText.setPositionRelative(this.pokeballSelectBg, 42, 9);
+    optionsText.setLineSpacing(12);
+    this.pokeballSelectContainer.add(optionsText);
+
+    this.countsText = addTextObject(this.scene, 0, 0, '', TextStyle.WINDOW, { maxLines: 5 });
+    this.countsText.setPositionRelative(this.pokeballSelectBg, 18, 9);
+    this.countsText.setLineSpacing(12);
+    this.pokeballSelectContainer.add(this.countsText);
+
+    this.setCursor(0);
+  }
+
+  show(args: any[]) {
+    super.show(args);
+
+    this.updateCounts();
+    this.pokeballSelectContainer.setVisible(true);
+    this.setCursor(this.cursor);
+  }
+
+  processInput(keyCode: integer) {
+    const keyCodes = Phaser.Input.Keyboard.KeyCodes;
+    const ui = this.getUi();
+
+    let success = false;
+
+    const pokeballTypeCount = Object.keys(this.scene.pokeballCounts).length;
+
+    if (keyCode === keyCodes.Z || keyCode === keyCodes.X) {
+      success = true;
+      if (keyCode === keyCodes.Z && this.cursor < pokeballTypeCount) {
+        if (this.scene.pokeballCounts[this.cursor]) {
+          (this.scene.getCurrentPhase() as CommandPhase).handleCommand(Command.BALL, this.cursor);
+          this.scene.ui.setMode(Mode.COMMAND);
+          this.scene.ui.setMode(Mode.MESSAGE);
+          success = true;
+        } else
+          ui.playError();
+      } else {
+        ui.setMode(Mode.COMMAND);
+        success = true;
+      }
+    } else {
+      switch (keyCode) {
+        case keyCodes.UP:
+          success = this.setCursor(this.cursor ? this.cursor - 1 : pokeballTypeCount);
+          break;
+        case keyCodes.DOWN:
+          success = this.setCursor(this.cursor < pokeballTypeCount ? this.cursor + 1 : 0);
+          break;
+      }
+    }
+
+    if (success)
+      ui.playSelect();
+  }
+
+  updateCounts() {
+    this.countsText.setText(Object.values(this.scene.pokeballCounts).map(c => `x${c}`).join('\n'));
+  }
+
+  setCursor(cursor: integer): boolean {
+    const ret = super.setCursor(cursor);
+
+    if (!this.cursorObj) {
+      this.cursorObj = this.scene.add.image(0, 0, 'cursor');
+      this.pokeballSelectContainer.add(this.cursorObj);
+    }
+
+    this.cursorObj.setPositionRelative(this.pokeballSelectBg, 12, 17 + 16 * this.cursor);
+
+    return ret;
+  }
+
+  clear() {
+    super.clear();
+    this.pokeballSelectContainer.setVisible(false);
+    this.eraseCursor();
+  }
+
+  eraseCursor() {
+    if (this.cursorObj)
+      this.cursorObj.destroy();
+    this.cursorObj = null;
+  }
+}

src/ui/command-ui-handler.ts

@@ -61,7 +61,7 @@ export default class CommandUiHandler extends UiHandler {
             success = true;
             break;
           case 1:
-            (this.scene.getCurrentPhase() as CommandPhase).handleCommand(Command.BALL, this.cursor);
+            ui.setModeWithoutClear(Mode.BALL);
             success = true;
             break;
           case 2:

src/ui/switch-check-ui-handler.ts

@@ -34,7 +34,7 @@ export default class SwitchCheckUiHandler extends AwaitableUiHandler {
     switchCheckText.setLineSpacing(12);
     switchCheckContainer.add(switchCheckText);
 
-    this.setCursor(0)
+    this.setCursor(0);
   }
 
   show(args: any[]) {

src/ui/ui.ts

@@ -7,11 +7,13 @@ import FightUiHandler from './fight-ui-handler';
 import MessageUiHandler from './message-ui-handler';
 import SwitchCheckUiHandler from './switch-check-ui-handler';
 import ModifierSelectUiHandler from './modifier-select-ui-handler';
+import BallUiHandler from './ball-ui-handler';
 
 export enum Mode {
   MESSAGE = 0,
   COMMAND,
   FIGHT,
+  BALL,
   SWITCH_CHECK,
   MODIFIER_SELECT,
   PARTY
@@ -29,6 +31,7 @@ export default class UI extends Phaser.GameObjects.Container {
       new BattleMessageUiHandler(scene),
       new CommandUiHandler(scene),
       new FightUiHandler(scene),
+      new BallUiHandler(scene),
       new SwitchCheckUiHandler(scene),
       new ModifierSelectUiHandler(scene),
       new PartyUiHandler(scene)