pokerogue/src/data/nature.ts

import { Stat, getStatName } from "./pokemon-stat";
import * as Utils from "../utils";
import { TextStyle, getBBCodeFrag } from "../ui/text";
import { UiTheme } from "#enums";
import i18next from "i18next";

export enum Nature {
  HARDY,
  LONELY,
  BRAVE,
  ADAMANT,
  NAUGHTY,
  BOLD,
  DOCILE,
  RELAXED,
  IMPISH,
  LAX,
  TIMID,
  HASTY,
  SERIOUS,
  JOLLY,
  NAIVE,
  MODEST,
  MILD,
  QUIET,
  BASHFUL,
  RASH,
  CALM,
  GENTLE,
  SASSY,
  CAREFUL,
  QUIRKY
}

export function getNatureName(nature: Nature, includeStatEffects: boolean = false, forStarterSelect: boolean = false, ignoreBBCode: boolean = false, uiTheme: UiTheme = UiTheme.DEFAULT): string {
  let ret = Utils.toReadableString(Nature[nature]);
  //Translating nature
  if (i18next.exists("nature:" + ret)) {
    ret = i18next.t("nature:" + ret as any);
  }
  if (includeStatEffects) {
    const stats = Utils.getEnumValues(Stat).slice(1);
    let increasedStat: Stat = null;
    let decreasedStat: Stat = null;
    for (const stat of stats) {
      const multiplier = getNatureStatMultiplier(nature, stat);
      if (multiplier > 1) {
        increasedStat = stat;
      } else if (multiplier < 1) {
        decreasedStat = stat;
      }
    }
    const textStyle = forStarterSelect ? TextStyle.SUMMARY_ALT : TextStyle.WINDOW;
    const getTextFrag = !ignoreBBCode ? (text: string, style: TextStyle) => getBBCodeFrag(text, style, uiTheme) : (text: string, style: TextStyle) => text;
    if (increasedStat && decreasedStat) {
      ret = `${getTextFrag(`${ret}${!forStarterSelect ? "\n" : " "}(`, textStyle)}${getTextFrag(`+${getStatName(increasedStat, true)}`, TextStyle.SUMMARY_PINK)}${getTextFrag("/", textStyle)}${getTextFrag(`-${getStatName(decreasedStat, true)}`, TextStyle.SUMMARY_BLUE)}${getTextFrag(")", textStyle)}`;
    } else {
      ret = getTextFrag(`${ret}${!forStarterSelect ? "\n" : " "}(-)`, textStyle);
    }
  }
  return ret;
}

export function getNatureStatMultiplier(nature: Nature, stat: Stat): number {
  switch (stat) {
  case Stat.ATK:
    switch (nature) {
    case Nature.LONELY:
    case Nature.BRAVE:
    case Nature.ADAMANT:
    case Nature.NAUGHTY:
      return 1.1;
    case Nature.BOLD:
    case Nature.TIMID:
    case Nature.MODEST:
    case Nature.CALM:
      return 0.9;
    }
    break;
  case Stat.DEF:
    switch (nature) {
    case Nature.BOLD:
    case Nature.RELAXED:
    case Nature.IMPISH:
    case Nature.LAX:
      return 1.1;
    case Nature.LONELY:
    case Nature.HASTY:
    case Nature.MILD:
    case Nature.GENTLE:
      return 0.9;
    }
    break;
  case Stat.SPATK:
    switch (nature) {
    case Nature.MODEST:
    case Nature.MILD:
    case Nature.QUIET:
    case Nature.RASH:
      return 1.1;
    case Nature.ADAMANT:
    case Nature.IMPISH:
    case Nature.JOLLY:
    case Nature.CAREFUL:
      return 0.9;
    }
    break;
  case Stat.SPDEF:
    switch (nature) {
    case Nature.CALM:
    case Nature.GENTLE:
    case Nature.SASSY:
    case Nature.CAREFUL:
      return 1.1;
    case Nature.NAUGHTY:
    case Nature.LAX:
    case Nature.NAIVE:
    case Nature.RASH:
      return 0.9;
    }
    break;
  case Stat.SPD:
    switch (nature) {
    case Nature.TIMID:
    case Nature.HASTY:
    case Nature.JOLLY:
    case Nature.NAIVE:
      return 1.1;
    case Nature.BRAVE:
    case Nature.RELAXED:
    case Nature.QUIET:
    case Nature.SASSY:
      return 0.9;
    }
    break;
  }

  return 1;
}