@robsonbittencourt/calc 0.10.0

package/src/field.ts

@@ -0,0 +1,124 @@
+import type {State} from './state';
+import type {GameType, Weather, Terrain} from './data/interface';
+
+export class Field implements State.Field {
+  gameType: GameType;
+  weather?: Weather;
+  terrain?: Terrain;
+  isMagicRoom: boolean;
+  isWonderRoom: boolean;
+  isGravity: boolean;
+  isAuraBreak?: boolean;
+  isFairyAura?: boolean;
+  isDarkAura?: boolean;
+  isBeadsOfRuin?: boolean;
+  isSwordOfRuin?: boolean;
+  isTabletsOfRuin?: boolean;
+  isVesselOfRuin?: boolean;
+  attackerSide: Side;
+  defenderSide: Side;
+
+  constructor(field: Partial<State.Field> = {}) {
+    this.gameType = field.gameType || 'Singles';
+    this.terrain = field.terrain;
+    this.weather = field.weather;
+    this.isMagicRoom = !!field.isMagicRoom;
+    this.isWonderRoom = !!field.isWonderRoom;
+    this.isGravity = !!field.isGravity;
+    this.isAuraBreak = field.isAuraBreak || false;
+    this.isFairyAura = field.isFairyAura || false;
+    this.isDarkAura = field.isDarkAura || false;
+    this.isBeadsOfRuin = field.isBeadsOfRuin || false;
+    this.isSwordOfRuin = field.isSwordOfRuin || false;
+    this.isTabletsOfRuin = field.isTabletsOfRuin || false;
+    this.isVesselOfRuin = field.isVesselOfRuin || false;
+
+    this.attackerSide = new Side(field.attackerSide || {});
+    this.defenderSide = new Side(field.defenderSide || {});
+  }
+
+  hasWeather(...weathers: Weather[]) {
+    return !!(this.weather && weathers.includes(this.weather));
+  }
+
+  hasTerrain(...terrains: Terrain[]) {
+    return !!(this.terrain && terrains.includes(this.terrain));
+  }
+
+  swap() {
+    [this.attackerSide, this.defenderSide] = [this.defenderSide, this.attackerSide];
+    return this;
+  }
+
+  clone() {
+    return new Field({
+      gameType: this.gameType,
+      weather: this.weather,
+      terrain: this.terrain,
+      isMagicRoom: this.isMagicRoom,
+      isWonderRoom: this.isWonderRoom,
+      isGravity: this.isGravity,
+      attackerSide: this.attackerSide,
+      defenderSide: this.defenderSide,
+      isAuraBreak: this.isAuraBreak,
+      isDarkAura: this.isDarkAura,
+      isFairyAura: this.isFairyAura,
+      isBeadsOfRuin: this.isBeadsOfRuin,
+      isSwordOfRuin: this.isSwordOfRuin,
+      isTabletsOfRuin: this.isTabletsOfRuin,
+      isVesselOfRuin: this.isVesselOfRuin,
+    });
+  }
+}
+
+export class Side implements State.Side {
+  spikes: number;
+  steelsurge: boolean;
+  vinelash: boolean;
+  wildfire: boolean;
+  cannonade: boolean;
+  volcalith: boolean;
+  isSR: boolean;
+  isReflect: boolean;
+  isLightScreen: boolean;
+  isProtected: boolean;
+  isSeeded: boolean;
+  isForesight: boolean;
+  isTailwind: boolean;
+  isHelpingHand: boolean;
+  isFlowerGift: boolean;
+  isFriendGuard: boolean;
+  isAuroraVeil: boolean;
+  isBattery: boolean;
+  isPowerSpot: boolean;
+  isSteelySpirit: boolean;
+  isSwitching?: 'out' | 'in';
+
+  constructor(side: State.Side = {}) {
+    this.spikes = side.spikes || 0;
+    this.steelsurge = !!side.steelsurge;
+    this.vinelash = !!side.vinelash;
+    this.wildfire = !!side.wildfire;
+    this.cannonade = !!side.cannonade;
+    this.volcalith = !!side.volcalith;
+    this.isSR = !!side.isSR;
+    this.isReflect = !!side.isReflect;
+    this.isLightScreen = !!side.isLightScreen;
+    this.isProtected = !!side.isProtected;
+    this.isSeeded = !!side.isSeeded;
+    this.isForesight = !!side.isForesight;
+    this.isTailwind = !!side.isTailwind;
+    this.isHelpingHand = !!side.isHelpingHand;
+    this.isFlowerGift = !!side.isFlowerGift;
+    this.isFriendGuard = !!side.isFriendGuard;
+    this.isAuroraVeil = !!side.isAuroraVeil;
+    this.isBattery = !!side.isBattery;
+    this.isPowerSpot = !!side.isPowerSpot;
+    this.isSteelySpirit = !!side.isSteelySpirit;
+    this.isSwitching = side.isSwitching;
+  }
+
+  clone() {
+    return new Side(this);
+  }
+}

package/src/index.ts

@@ -0,0 +1,156 @@
+// When using this library in the browser, a bundler like Webpack should be
+// used to encapsulate the various interdependencies between internal packages.
+// However, if you are requiring contents of this package in HTML <script>
+// tags, the following loading order is required:
+//
+//   - util.js
+//   - stats.js
+//
+//   - data/species.js
+//   - data/types.js
+//   - data/natures.js
+//   - data/abilities.js
+//   - data/moves.js
+//   - data/items.js
+//   - data/index.js
+//
+//   - pokemon.js
+//   - field.js
+//   - move.js
+//   - items.js
+//
+//   - mechanics/util.js
+//   - mechanics/gen78.js
+//   - mechanics/gen56.js
+//   - mechanics/gen4.js
+//   - mechanics/gen3.js
+//   - mechanics/gen12.js
+//
+//   - calc.js
+//   - desc.js
+//   - result.ts
+//
+//   - adaptable.js
+//   - index.js
+//
+// Furthermore, before anything is loaded, the following is required:
+//
+// <script type="text/javascript">
+//		var calc = exports = {};
+//		function require() { return exports; };
+//	</script>
+
+// If we're not being used as a module we're just going to rely on globals and
+// that the correct loading order being followed.
+
+import {Generations} from './data';
+import type {State} from './state';
+import type * as I from './data/interface';
+import * as A from './adaptable';
+
+// The loading strategy outlined in the comment above breaks in the browser when we start reusing
+// names as we're doing here with our shim overrides. Because exporting calculate below tramples
+// A.calculate, this ends up infinitely calling itself. As a workaround we save the original value
+// of A.calculate (which would be exports.calculate if files are loaded as outlined above) so that
+// we can call that instead.
+//
+// This is obviously kludge, use a bundler kids.
+const Acalculate = exports.calculate;
+
+export function calculate(
+  gen: I.GenerationNum | I.Generation,
+  attacker: A.Pokemon,
+  defender: A.Pokemon,
+  move: A.Move,
+  field?: A.Field
+): A.Result {
+  return (Acalculate || A.calculate)(
+    typeof gen === 'number' ? Generations.get(gen) : gen,
+    attacker,
+    defender,
+    move,
+    field
+  );
+}
+
+export class Move extends A.Move {
+  constructor(
+    gen: I.GenerationNum | I.Generation,
+    name: string,
+    options: Partial<Omit<State.Move, 'ability' | 'item' | 'species'>> & {
+      ability?: string;
+      item?: string;
+      species?: string;
+    } = {}
+  ) {
+    super(typeof gen === 'number' ? Generations.get(gen) : gen, name, options as any);
+  }
+}
+
+export class Pokemon extends A.Pokemon {
+  constructor(
+    gen: I.GenerationNum | I.Generation,
+    name: string,
+    options: Partial<Omit<State.Pokemon, 'ability' | 'item' | 'nature' | 'moves'>> & {
+      ability?: string;
+      item?: string;
+      nature?: string;
+      moves?: string[];
+      curHP?: number;
+      ivs?: Partial<I.StatsTable> & {spc?: number};
+      evs?: Partial<I.StatsTable> & {spc?: number};
+      boosts?: Partial<I.StatsTable> & {spc?: number};
+    } = {}
+  ) {
+    super(typeof gen === 'number' ? Generations.get(gen) : gen, name, options as any);
+  }
+
+  static getForme(
+    gen: I.GenerationNum | I.Generation,
+    speciesName: string,
+    item?: string,
+    moveName?: string
+  ) {
+    return A.Pokemon.getForme(
+      typeof gen === 'number' ? Generations.get(gen) : gen,
+      speciesName,
+      item as I.ItemName,
+      moveName as I.MoveName
+    );
+  }
+}
+
+export function calcStat(
+  gen: I.GenerationNum | I.Generation,
+  stat: I.StatID | 'spc',
+  base: number,
+  iv: number,
+  ev: number,
+  level: number,
+  nature?: string
+) {
+  return A.Stats.calcStat(
+    typeof gen === 'number' ? Generations.get(gen) : gen,
+    stat === 'spc' ? 'spa' : stat,
+    base,
+    iv,
+    ev,
+    level,
+    nature
+  );
+}
+
+export {Field, Side} from './field';
+export {Result} from './result';
+export {type GenerationNum, type StatsTable, type StatID} from './data/interface';
+export {Generations} from './data/index';
+export {toID} from './util';
+export {type State} from './state';
+
+export {ABILITIES} from './data/abilities';
+export {ITEMS, MEGA_STONES} from './data/items';
+export {MOVES} from './data/moves';
+export {SPECIES} from './data/species';
+export {NATURES} from './data/natures';
+export {TYPE_CHART} from './data/types';
+export {STATS, Stats} from './stats';

package/src/items.ts

@@ -0,0 +1,423 @@
+import type {Generation, TypeName, StatID} from './data/interface';
+import {toID} from './util';
+
+export const SEED_BOOSTED_STAT: {[item: string]: StatID} = {
+  'Electric Seed': 'def',
+  'Grassy Seed': 'def',
+  'Misty Seed': 'spd',
+  'Psychic Seed': 'spd',
+};
+
+export function getItemBoostType(item: string | undefined) {
+  switch (item) {
+  case 'Draco Plate':
+  case 'Dragon Fang':
+    return 'Dragon';
+  case 'Dread Plate':
+  case 'Black Glasses':
+    return 'Dark';
+  case 'Earth Plate':
+  case 'Soft Sand':
+    return 'Ground';
+  case 'Fist Plate':
+  case 'Black Belt':
+    return 'Fighting';
+  case 'Flame Plate':
+  case 'Charcoal':
+    return 'Fire';
+  case 'Icicle Plate':
+  case 'Never-Melt Ice':
+    return 'Ice';
+  case 'Insect Plate':
+  case 'Silver Powder':
+    return 'Bug';
+  case 'Iron Plate':
+  case 'Metal Coat':
+    return 'Steel';
+  case 'Meadow Plate':
+  case 'Rose Incense':
+  case 'Miracle Seed':
+    return 'Grass';
+  case 'Mind Plate':
+  case 'Odd Incense':
+  case 'Twisted Spoon':
+    return 'Psychic';
+  case 'Fairy Feather':
+  case 'Pixie Plate':
+    return 'Fairy';
+  case 'Sky Plate':
+  case 'Sharp Beak':
+    return 'Flying';
+  case 'Splash Plate':
+  case 'Sea Incense':
+  case 'Wave Incense':
+  case 'Mystic Water':
+    return 'Water';
+  case 'Spooky Plate':
+  case 'Spell Tag':
+    return 'Ghost';
+  case 'Stone Plate':
+  case 'Rock Incense':
+  case 'Hard Stone':
+    return 'Rock';
+  case 'Toxic Plate':
+  case 'Poison Barb':
+    return 'Poison';
+  case 'Zap Plate':
+  case 'Magnet':
+    return 'Electric';
+  case 'Silk Scarf':
+  case 'Pink Bow':
+  case 'Polkadot Bow':
+    return 'Normal';
+  default:
+    return undefined;
+  }
+}
+
+export function getBerryResistType(berry: string | undefined) {
+  switch (berry) {
+  case 'Chilan Berry':
+    return 'Normal';
+  case 'Occa Berry':
+    return 'Fire';
+  case 'Passho Berry':
+    return 'Water';
+  case 'Wacan Berry':
+    return 'Electric';
+  case 'Rindo Berry':
+    return 'Grass';
+  case 'Yache Berry':
+    return 'Ice';
+  case 'Chople Berry':
+    return 'Fighting';
+  case 'Kebia Berry':
+    return 'Poison';
+  case 'Shuca Berry':
+    return 'Ground';
+  case 'Coba Berry':
+    return 'Flying';
+  case 'Payapa Berry':
+    return 'Psychic';
+  case 'Tanga Berry':
+    return 'Bug';
+  case 'Charti Berry':
+    return 'Rock';
+  case 'Kasib Berry':
+    return 'Ghost';
+  case 'Haban Berry':
+    return 'Dragon';
+  case 'Colbur Berry':
+    return 'Dark';
+  case 'Babiri Berry':
+    return 'Steel';
+  case 'Roseli Berry':
+    return 'Fairy';
+  default:
+    return undefined;
+  }
+}
+
+const FLING_120 = new Set([
+  'TR24',
+  'TR28',
+  'TR34',
+  'TR39',
+  'TR53',
+  'TR55',
+  'TR64',
+  'TR66',
+  'TR72',
+  'TR73',
+]);
+
+const FLING_100 = new Set([
+  'Hard Stone',
+  'Room Service',
+  'Claw Fossil',
+  'Dome Fossil',
+  'Helix Fossil',
+  'Old Amber',
+  'Root Fossil',
+  'Armor Fossil',
+  'Old Amber',
+  'Fossilized Bird',
+  'Fossilized Dino',
+  'Fossilized Drake',
+  'Fossilized Fish',
+  'Plume Fossil',
+  'Jaw Fossil',
+  'Cover Fossil',
+  'Sail Fossil',
+  'Rare Bone',
+  'Skull Fossil',
+  'TR10',
+  'TR31',
+  'TR75',
+]);
+
+const FLING_90 = new Set([
+  'Deep Sea Tooth',
+  'Thick Club',
+  'TR02',
+  'TR04',
+  'TR05',
+  'TR08',
+  'TR11',
+  'TR22',
+  'TR35',
+  'TR42',
+  'TR45',
+  'TR50',
+  'TR61',
+  'TR65',
+  'TR67',
+  'TR86',
+  'TR90',
+  'TR96',
+]);
+
+const FLING_85 = new Set(['TR01', 'TR41', 'TR62', 'TR93', 'TR97', 'TR98']);
+
+const FLING_80 = new Set([
+  'Assault Vest',
+  'Blunder Policy',
+  'Chipped Pot',
+  'Cracked Pot',
+  'Heavy-Duty Boots',
+  'Weakness Policy',
+  'Quick Claw',
+  'Dawn Stone',
+  'Dusk Stone',
+  'Electirizer',
+  'Magmarizer',
+  'Oval Stone',
+  'Protector',
+  'Sachet',
+  'Whipped Dream',
+  'Razor Claw',
+  'Shiny Stone',
+  'TR16',
+  'TR18',
+  'TR19',
+  'TR25',
+  'TR32',
+  'TR33',
+  'TR47',
+  'TR56',
+  'TR57',
+  'TR58',
+  'TR59',
+  'TR60',
+  'TR63',
+  'TR69',
+  'TR70',
+  'TR74',
+  'TR84',
+  'TR87',
+  'TR92',
+  'TR95',
+  'TR99',
+]);
+
+const FLING_70 = new Set([
+  'Poison Barb',
+  'Dragon Fang',
+  'Power Anklet',
+  'Power Band',
+  'Power Belt',
+  'Power Bracer',
+  'Power Lens',
+  'Power Weight',
+]);
+
+const FLING_60 = new Set([
+  'Adamant Orb',
+  'Damp Rock',
+  'Heat Rock',
+  'Leek',
+  'Lustrous Orb',
+  'Macho Brace',
+  'Rocky Helmet',
+  'Stick',
+  'Utility Umbrella',
+  'Terrain Extender',
+]);
+const FLING_30 = new Set([
+  'Absorb Bulb',
+  'Black Belt',
+  'Black Sludge',
+  'Black Glasses',
+  'Cell Battery',
+  'Charcoal',
+  'Deep Sea Scale',
+  'Flame Orb',
+  'King\'s Rock',
+  'Life Orb',
+  'Light Ball',
+  'Light Clay',
+  'Magnet',
+  'Metal Coat',
+  'Miracle Seed',
+  'Mystic Water',
+  'Never-Melt Ice',
+  'Razor Fang',
+  'Scope Lens',
+  'Soul Dew',
+  'Spell Tag',
+  'Sweet Apple',
+  'Tart Apple',
+  'Throat Spray',
+  'Toxic Orb',
+  'Twisted Spoon',
+  'Dragon Scale',
+  'Energy Powder',
+  'Fire Stone',
+  'Leaf Stone',
+  'Moon Stone',
+  'Sun Stone',
+  'Thunder Stone',
+  'Up-Grade',
+  'Water Stone',
+  'Berry Juice',
+  'Black Sludge',
+  'Prism Scale',
+  'Ice Stone',
+  'Gold Bottle Cap',
+  'Luminous Moss',
+  'Eject Button',
+  'Snowball',
+  'Bottle Cap',
+]);
+const FLING_10 = new Set([
+  'Air Balloon',
+  'Berry Sweet',
+  'Choice Band',
+  'Choice Scarf',
+  'Choice Specs',
+  'Clover Sweet',
+  'Destiny Knot',
+  'Electric Seed',
+  'Expert Belt',
+  'Flower Sweet',
+  'Focus Band',
+  'Focus Sash',
+  'Full Incense',
+  'Grassy Seed',
+  'Lagging Tail',
+  'Lax Incense',
+  'Leftovers',
+  'Love Sweet',
+  'Mental Herb',
+  'Metal Powder',
+  'Mint Berry',
+  'Miracle Berry',
+  'Misty Seed',
+  'Muscle Band',
+  'Power Herb',
+  'Psychic Seed',
+  'Odd Incense',
+  'Quick Powder',
+  'Reaper Cloth',
+  'Red Card',
+  'Ribbon Sweet',
+  'Ring Target',
+  'Rock Incense',
+  'Rose Incense',
+  'Sea Incense',
+  'Shed Shell',
+  'Silk Scarf',
+  'Silver Powder',
+  'Smooth Rock',
+  'Soft Sand',
+  'Soothe Bell',
+  'Star Sweet',
+  'Strawberry Sweet',
+  'Wave Incense',
+  'White Herb',
+  'Wide Lens',
+  'Wise Glasses',
+  'Zoom Lens',
+  'Silver Powder',
+  'Power Herb',
+  'TR00',
+  'TR07',
+  'TR12',
+  'TR13',
+  'TR14',
+  'TR17',
+  'TR20',
+  'TR21',
+  'TR23',
+  'TR26',
+  'TR27',
+  'TR29',
+  'TR30',
+  'TR37',
+  'TR38',
+  'TR40',
+  'TR44',
+  'TR46',
+  'TR48',
+  'TR49',
+  'TR51',
+  'TR52',
+  'TR54',
+  'TR68',
+  'TR76',
+  'TR77',
+  'TR79',
+  'TR80',
+  'TR83',
+  'TR85',
+  'TR88',
+  'TR91',
+]);
+
+// TODO: move this data to the data files instead.
+export function getFlingPower(item?: string) {
+  if (!item) return 0;
+  if (['Big Nugget', 'Iron Ball', 'TR43', 'TR71'].includes(item)) return 130;
+  if (FLING_120.has(item)) return 85;
+  if (['TR03', 'TR06', 'TR09', 'TR15', 'TR89'].includes(item)) return 110;
+  if (FLING_100.has(item)) return 100;
+  if (['TR36', 'TR78', 'TR81', 'TR94'].includes(item)) return 95;
+  if (item.includes('Plate') || FLING_90.has(item)) return 90;
+  if (FLING_85.has(item)) return 85;
+  if (FLING_80.has(item)) return 80;
+  if (FLING_70.has(item)) return 70;
+  if (FLING_60.has(item)) return 60;
+  if (['Eject Pack', 'Sharp Beak', 'Dubious Disc'].includes(item)) return 50;
+  if (['Icy Rock', 'Eviolite', 'Lucky Punch'].includes(item)) return 40;
+  if (FLING_30.has(item)) return 30;
+  if (item === 'TR82') return 20;
+  if (item.includes('Berry') || FLING_10.has(item)) return 10;
+  return 0;
+}
+
+export function getNaturalGift(gen: Generation, item: string) {
+  const gift = gen.items.get(toID(item))?.naturalGift;
+  return gift ? {t: gift.type, p: gift.basePower} : {t: 'Normal' as TypeName, p: 1};
+}
+
+export function getTechnoBlast(item: string) {
+  switch (item) {
+  case 'Burn Drive':
+    return 'Fire';
+  case 'Chill Drive':
+    return 'Ice';
+  case 'Douse Drive':
+    return 'Water';
+  case 'Shock Drive':
+    return 'Electric';
+  default:
+    return undefined;
+  }
+}
+
+export function getMultiAttack(item: string) {
+  if (item.includes('Memory')) {
+    return item.substring(0, item.indexOf(' ')) as TypeName;
+  }
+  return undefined;
+}