diff --git a/src/data/move.ts b/src/data/move.ts index 717a9094c..57b01e64c 100644 --- a/src/data/move.ts +++ b/src/data/move.ts @@ -19,6 +19,7 @@ import { Stat } from "./pokemon-stat"; import { TerrainType } from "./terrain"; import { SpeciesFormChangeActiveTrigger } from "./pokemon-forms"; import { Species } from "./enums/species"; +import { ModifierPoolType } from "#app/modifier/modifier-type"; export enum MoveCategory { PHYSICAL, @@ -934,8 +935,9 @@ export class StealHeldItemChanceAttr extends MoveEffectAttr { return resolve(false); const heldItems = this.getTargetHeldItems(target).filter(i => i.getTransferrable(false)); if (heldItems.length) { - const highestItemTier = heldItems.map(m => m.type.getOrInferTier()).reduce((highestTier, tier) => Math.max(tier, highestTier), 0); - const tierHeldItems = heldItems.filter(m => m.type.getOrInferTier() === highestItemTier); + const poolType = target.isPlayer() ? ModifierPoolType.PLAYER : target.hasTrainer() ? ModifierPoolType.TRAINER : ModifierPoolType.WILD; + const highestItemTier = heldItems.map(m => m.type.getOrInferTier(poolType)).reduce((highestTier, tier) => Math.max(tier, highestTier), 0); + const tierHeldItems = heldItems.filter(m => m.type.getOrInferTier(poolType) === highestItemTier); const stolenItem = tierHeldItems[user.randSeedInt(tierHeldItems.length)]; user.scene.tryTransferHeldItemModifier(stolenItem, user, false, false).then(success => { if (success) @@ -980,8 +982,9 @@ export class RemoveHeldItemAttr extends MoveEffectAttr { return resolve(false); const heldItems = this.getTargetHeldItems(target).filter(i => i.getTransferrable(false)); if (heldItems.length) { - const highestItemTier = heldItems.map(m => m.type.getOrInferTier()).reduce((highestTier, tier) => Math.max(tier, highestTier), 0); - const tierHeldItems = heldItems.filter(m => m.type.getOrInferTier() === highestItemTier); + const poolType = target.isPlayer() ? ModifierPoolType.PLAYER : target.hasTrainer() ? ModifierPoolType.TRAINER : ModifierPoolType.WILD; + const highestItemTier = heldItems.map(m => m.type.getOrInferTier(poolType)).reduce((highestTier, tier) => Math.max(tier, highestTier), 0); + const tierHeldItems = heldItems.filter(m => m.type.getOrInferTier(poolType) === highestItemTier); const stolenItem = tierHeldItems[user.randSeedInt(tierHeldItems.length)]; user.scene.tryTransferHeldItemModifier(stolenItem, user, false, false).then(success => { if (success) diff --git a/src/modifier/modifier-type.ts b/src/modifier/modifier-type.ts index 83b722ddf..258e79e36 100644 --- a/src/modifier/modifier-type.ts +++ b/src/modifier/modifier-type.ts @@ -63,16 +63,35 @@ export class ModifierType { this.tier = tier; } - getOrInferTier(): ModifierTier { + getOrInferTier(poolType: ModifierPoolType = ModifierPoolType.PLAYER): ModifierTier { if (this.tier) return this.tier; if (!this.id) return null; - for (let tier of Utils.getEnumValues(ModifierTier)) { - if (!modifierPool.hasOwnProperty(tier)) - continue; - if (modifierPool[tier].find(m => (m as WeightedModifierType).modifierType.id === (this.generatorId || this.id))) - return (this.tier = tier); + let poolTypes: ModifierPoolType[]; + switch (poolType) { + case ModifierPoolType.PLAYER: + poolTypes = [ poolType, ModifierPoolType.TRAINER, ModifierPoolType.WILD ]; + break; + case ModifierPoolType.WILD: + poolTypes = [ poolType, ModifierPoolType.PLAYER, ModifierPoolType.TRAINER ]; + break; + case ModifierPoolType.TRAINER: + poolTypes = [ poolType, ModifierPoolType.PLAYER, ModifierPoolType.WILD ]; + break; + default: + poolTypes = [ poolType ]; + break; + } + // Try multiple pool types in case of stolen items + for (let type of poolTypes) { + const pool = getModifierPoolForType(type); + for (let tier of Utils.getEnumValues(ModifierTier)) { + if (!pool.hasOwnProperty(tier)) + continue; + if (pool[tier].find(m => (m as WeightedModifierType).modifierType.id === (this.generatorId || this.id))) + return (this.tier = tier); + } } return null; } @@ -1179,9 +1198,7 @@ let enemyIgnoredPoolIndexes = {}; let enemyBuffModifierPoolThresholds = {}; let enemyBuffIgnoredPoolIndexes = {}; -const tierWeights = [ 769 / 1024, 192 / 1024, 48 / 1024, 12 / 1024, 1 / 1024 ]; - -export function regenerateModifierPoolThresholds(party: Pokemon[], poolType: ModifierPoolType, rerollCount: integer = 0) { +export function getModifierPoolForType(poolType: ModifierPoolType): ModifierPool { let pool: ModifierPool; switch (poolType) { case ModifierPoolType.PLAYER: @@ -1200,6 +1217,14 @@ export function regenerateModifierPoolThresholds(party: Pokemon[], poolType: Mod pool = dailyStarterModifierPool; break; } + return pool; +} + +const tierWeights = [ 769 / 1024, 192 / 1024, 48 / 1024, 12 / 1024, 1 / 1024 ]; + +export function regenerateModifierPoolThresholds(party: Pokemon[], poolType: ModifierPoolType, rerollCount: integer = 0) { + const pool = getModifierPoolForType(poolType); + const ignoredIndexes = {}; const modifierTableData = {}; const thresholds = Object.fromEntries(new Map(Object.keys(pool).map(t => { @@ -1360,27 +1385,22 @@ export function getDailyRunStarterModifiers(party: PlayerPokemon[]): Modifiers.P function getNewModifierTypeOption(party: Pokemon[], poolType: ModifierPoolType, tier?: ModifierTier, upgradeCount?: integer, retryCount: integer = 0): ModifierTypeOption { const player = !poolType; - let pool: ModifierPool; + const pool = getModifierPoolForType(poolType); let thresholds: object; switch (poolType) { case ModifierPoolType.PLAYER: - pool = modifierPool; thresholds = modifierPoolThresholds; break; case ModifierPoolType.WILD: - pool = wildModifierPool; thresholds = enemyModifierPoolThresholds; break; case ModifierPoolType.TRAINER: - pool = trainerModifierPool; thresholds = enemyModifierPoolThresholds; break; case ModifierPoolType.ENEMY_BUFF: - pool = enemyBuffModifierPool; thresholds = enemyBuffModifierPoolThresholds; break; case ModifierPoolType.DAILY_STARTER: - pool = dailyStarterModifierPool; thresholds = dailyStarterModifierPoolThresholds; break; } diff --git a/src/modifier/modifier.ts b/src/modifier/modifier.ts index 10a88d950..115273db7 100644 --- a/src/modifier/modifier.ts +++ b/src/modifier/modifier.ts @@ -1738,12 +1738,13 @@ export abstract class HeldItemTransferModifier extends PokemonHeldItemModifier { return false; const withinParty = pokemon.isPlayer() === targetPokemon.isPlayer(); + const poolType = pokemon.isPlayer() ? ModifierTypes.ModifierPoolType.PLAYER : pokemon.hasTrainer() ? ModifierTypes.ModifierPoolType.TRAINER : ModifierTypes.ModifierPoolType.WILD; const transferredModifierTypes: ModifierTypes.ModifierType[] = []; const itemModifiers = pokemon.scene.findModifiers(m => m instanceof PokemonHeldItemModifier && (m as PokemonHeldItemModifier).pokemonId === targetPokemon.id && m.getTransferrable(withinParty), targetPokemon.isPlayer()) as PokemonHeldItemModifier[]; - let highestItemTier = itemModifiers.map(m => m.type.getOrInferTier()).reduce((highestTier, tier) => Math.max(tier, highestTier), 0); - let tierItemModifiers = itemModifiers.filter(m => m.type.getOrInferTier() === highestItemTier); + let highestItemTier = itemModifiers.map(m => m.type.getOrInferTier(poolType)).reduce((highestTier, tier) => Math.max(tier, highestTier), 0); + let tierItemModifiers = itemModifiers.filter(m => m.type.getOrInferTier(poolType) === highestItemTier); let heldItemTransferPromises: Promise[] = [];