@pkmn/randoms 0.9.22 → 0.9.23

package/build/index.js

@@ -257,6 +257,9 @@ var RandomGen8Teams = class {
         return abilities.includes("Huge Power") && movePool.includes("aquajet");
       }
     };
+    this.poolsCacheKey = void 0;
+    this.cachedPool = void 0;
+    this.cachedSpeciesPool = void 0;
   }
   setSeed(prng) {
     this.prng = prng && !Array.isArray(prng) ? prng : new (0, _sim.PRNG)(prng);
@@ -477,20 +480,21 @@ var RandomGen8Teams = class {
         shiny
       };
       if (this.gen === 9) {
-        set.teraType = this.sample(this.dex.types.all()).name;
+        set.teraType = this.sample(this.dex.types.names());
       }
       team.push(set);
     }
     return team;
   }
-  randomNPokemon(n, requiredType, minSourceGen, ruleTable, requireMoves = false) {
-    if (requiredType && !this.dex.types.get(requiredType).exists) {
-      throw new Error(`"${requiredType}" is not a valid type.`);
-    }
+  getPools(requiredType, minSourceGen, ruleTable, requireMoves = false) {
     const isNotCustom = !ruleTable;
-    const pool = [];
+    let pool = [];
     let speciesPool = [];
-    if (isNotCustom) {
+    const ck = this.poolsCacheKey;
+    if (ck && this.cachedPool && this.cachedSpeciesPool && ck[0] === requiredType && ck[1] === minSourceGen && ck[2] === ruleTable && ck[3] === requireMoves) {
+      speciesPool = this.cachedSpeciesPool.slice();
+      pool = this.cachedPool.slice();
+    } else if (isNotCustom) {
       speciesPool = [...this.dex.species.all()];
       for (const species of speciesPool) {
         if (species.isNonstandard && species.isNonstandard !== "Unobtainable") continue;
@@ -504,6 +508,9 @@ var RandomGen8Teams = class {
         if (num <= 0 || pool.includes(num)) continue;
         pool.push(num);
       }
+      this.poolsCacheKey = [requiredType, minSourceGen, ruleTable, requireMoves];
+      this.cachedPool = pool.slice();
+      this.cachedSpeciesPool = speciesPool.slice();
     } else {
       const EXISTENCE_TAG = ["past", "future", "lgpe", "unobtainable", "cap", "custom", "nonexistent"];
       const nonexistentBanReason = ruleTable.check("nonexistent");
@@ -545,7 +552,18 @@ var RandomGen8Teams = class {
         if (pool.includes(num)) continue;
         pool.push(num);
       }
+      this.poolsCacheKey = [requiredType, minSourceGen, ruleTable, requireMoves];
+      this.cachedPool = pool.slice();
+      this.cachedSpeciesPool = speciesPool.slice();
+    }
+    return { pool, speciesPool };
+  }
+  randomNPokemon(n, requiredType, minSourceGen, ruleTable, requireMoves = false) {
+    if (requiredType && !this.dex.types.get(requiredType).exists) {
+      throw new Error(`"${requiredType}" is not a valid type.`);
     }
+    const { pool, speciesPool } = this.getPools(requiredType, minSourceGen, ruleTable, requireMoves);
+    const isNotCustom = !ruleTable;
     const hasDexNumber = {};
     for (let i = 0; i < n; i++) {
       const num = this.sampleNoReplace(pool);
@@ -778,7 +796,7 @@ var RandomGen8Teams = class {
         shiny
       };
       if (this.gen === 9) {
-        set.teraType = this.sample(this.dex.types.all()).name;
+        set.teraType = this.sample(this.dex.types.names());
       }
       team.push(set);
     }
@@ -2726,6 +2744,7 @@ var RandomGen7Teams = class extends RandomGen8Teams {
       Steel: (movePool, moves, abilities, types, counter, species) => !counter.get("Steel") && species.baseStats.atk >= 100,
       Water: (movePool, moves, abilities, types, counter) => !counter.get("Water")
     };
+    this.cachedStatusMoves = this.dex.moves.all().filter((move) => move.category === "Status" && move.id !== "naturepower").map((move) => move.id);
   }
   newQueryMoves(moves, species, preferredType, abilities) {
     const counter = new MoveCounter();
@@ -2853,7 +2872,7 @@ var RandomGen7Teams = class extends RandomGen8Teams {
       if (moves.size + movePool.length <= this.maxMoveCount) return;
     }
     const badWithSetup = ["defog", "dragontail", "haze", "healbell", "nuzzle", "pursuit", "rapidspin", "toxic"];
-    const statusMoves = this.dex.moves.all().filter((move) => move.category === "Status" && move.id !== "naturepower").map((move) => move.id);
+    const statusMoves = this.cachedStatusMoves;
     const incompatiblePairs = [
       // These moves don't mesh well with other aspects of the set
       [statusMoves, ["healingwish", "memento", "switcheroo", "trick"]],
@@ -4529,6 +4548,7 @@ var RandomGen6Teams = class extends RandomGen7Teams {
       Steel: (movePool, moves, abilities, types, counter, species) => !counter.get("Steel") && species.baseStats.atk >= 100,
       Water: (movePool, moves, abilities, types, counter) => !counter.get("Water")
     };
+    this.cachedStatusMoves = this.dex.moves.all().filter((move) => move.category === "Status").map((move) => move.id);
   }
   cullMovePool(types, moves, abilities, counter, movePool, teamDetails, species, isLead, preferredType, role) {
     let hasHiddenPower = false;
@@ -4601,7 +4621,7 @@ var RandomGen6Teams = class extends RandomGen7Teams {
       if (moves.size + movePool.length <= this.maxMoveCount) return;
     }
     const badWithSetup = ["defog", "dragontail", "haze", "healbell", "nuzzle", "pursuit", "rapidspin", "toxic"];
-    const statusMoves = this.dex.moves.all().filter((move) => move.category === "Status").map((move) => move.id);
+    const statusMoves = this.cachedStatusMoves;
     const incompatiblePairs = [
       // These moves don't mesh well with other aspects of the set
       [statusMoves, ["healingwish", "memento", "switcheroo", "trick"]],
@@ -5688,6 +5708,7 @@ var RandomGen5Teams = class extends RandomGen6Teams {
       Steel: (movePool, moves, abilities, types, counter, species) => !counter.get("Steel") && ["aggron", "metagross"].includes(species.id),
       Water: (movePool, moves, abilities, types, counter) => !counter.get("Water")
     };
+    this.cachedStatusMoves = this.dex.moves.all().filter((move) => move.category === "Status" && move.id !== "naturepower").map((move) => move.id);
   }
   cullMovePool(types, moves, abilities, counter, movePool, teamDetails, species, isLead, preferredType, role) {
     let hasHiddenPower = false;
@@ -5755,7 +5776,7 @@ var RandomGen5Teams = class extends RandomGen6Teams {
       if (moves.size + movePool.length <= this.maxMoveCount) return;
     }
     const badWithSetup = ["healbell", "pursuit", "toxic"];
-    const statusMoves = this.dex.moves.all().filter((move) => move.category === "Status" && move.id !== "naturepower").map((move) => move.id);
+    const statusMoves = this.cachedStatusMoves;
     const incompatiblePairs = [
       // These moves don't mesh well with other aspects of the set
       [statusMoves, ["healingwish", "switcheroo", "trick"]],
@@ -6667,6 +6688,7 @@ var RandomGen4Teams = class extends RandomGen5Teams {
       Steel: (movePool, moves, abilities, types, counter, species) => !counter.get("Steel") && species.id === "metagross",
       Water: (movePool, moves, abilities, types, counter) => !counter.get("Water")
     };
+    this.cachedStatusMoves = this.dex.moves.all().filter((move) => move.category === "Status").map((move) => move.id);
   }
   cullMovePool(types, moves, abilities, counter, movePool, teamDetails, species, isLead, preferredType, role) {
     let hasHiddenPower = false;
@@ -6734,7 +6756,7 @@ var RandomGen4Teams = class extends RandomGen5Teams {
       if (moves.size + movePool.length <= this.maxMoveCount) return;
     }
     const badWithSetup = ["pursuit", "toxic"];
-    const statusMoves = this.dex.moves.all().filter((move) => move.category === "Status").map((move) => move.id);
+    const statusMoves = this.cachedStatusMoves;
     const incompatiblePairs = [
       // These moves don't mesh well with other aspects of the set
       [statusMoves, ["healingwish", "switcheroo", "trick"]],
@@ -7515,7 +7537,7 @@ var RandomGen3Teams = class extends RandomGen4Teams {
       if (moves.size + movePool.length <= this.maxMoveCount) return;
     }
     const badWithSetup = ["knockoff", "rapidspin", "toxic"];
-    const statusMoves = this.dex.moves.all().filter((move) => move.category === "Status").map((move) => move.id);
+    const statusMoves = this.cachedStatusMoves;
     const incompatiblePairs = [
       // These moves don't mesh well with other aspects of the set
       [statusMoves, "trick"],
@@ -9350,6 +9372,10 @@ var RandomTeams = class {
       Steel: (movePool, moves, abilities, types, counter, species, teamDetails, isLead, isDoubles) => !counter.get("Steel") && (isDoubles || species.baseStats.atk >= 90 || movePool.includes("gigatonhammer") || movePool.includes("makeitrain")),
       Water: (movePool, moves, abilities, types, counter) => !counter.get("Water") && !types.includes("Ground")
     };
+    this.poolsCacheKey = void 0;
+    this.cachedPool = void 0;
+    this.cachedSpeciesPool = void 0;
+    this.cachedStatusMoves = this.dex.moves.all().filter((move) => move.category === "Status").map((move) => move.id);
   }
   setSeed(prng) {
     this.prng = prng && !Array.isArray(prng) ? prng : new (0, _sim.PRNG)(prng);
@@ -9528,7 +9554,7 @@ var RandomTeams = class {
         }
       }
     }
-    const statusMoves = this.dex.moves.all().filter((move) => move.category === "Status").map((move) => move.id);
+    const statusMoves = this.cachedStatusMoves;
     if (teamDetails.screens) {
       if (movePool.includes("auroraveil")) this.fastPop(movePool, movePool.indexOf("auroraveil"));
       if (movePool.length >= this.maxMoveCount + 2) {
@@ -10880,21 +10906,22 @@ var RandomTeams = class {
         if (this.forceTeraType) {
           set.teraType = this.forceTeraType;
         } else {
-          set.teraType = this.sample(this.dex.types.all()).name;
+          set.teraType = this.sample(this.dex.types.names());
         }
       }
       team.push(set);
     }
     return team;
   }
-  randomNPokemon(n, requiredType, minSourceGen, ruleTable, requireMoves = false) {
-    if (requiredType && !this.dex.types.get(requiredType).exists) {
-      throw new Error(`"${requiredType}" is not a valid type.`);
-    }
+  getPools(requiredType, minSourceGen, ruleTable, requireMoves = false) {
     const isNotCustom = !ruleTable;
-    const pool = [];
+    let pool = [];
     let speciesPool = [];
-    if (isNotCustom) {
+    const ck = this.poolsCacheKey;
+    if (ck && this.cachedPool && this.cachedSpeciesPool && ck[0] === requiredType && ck[1] === minSourceGen && ck[2] === ruleTable && ck[3] === requireMoves) {
+      speciesPool = this.cachedSpeciesPool.slice();
+      pool = this.cachedPool.slice();
+    } else if (isNotCustom) {
       speciesPool = [...this.dex.species.all()];
       for (const species of speciesPool) {
         if (species.isNonstandard && species.isNonstandard !== "Unobtainable") continue;
@@ -10908,6 +10935,9 @@ var RandomTeams = class {
         if (num <= 0 || pool.includes(num)) continue;
         pool.push(num);
       }
+      this.poolsCacheKey = [requiredType, minSourceGen, ruleTable, requireMoves];
+      this.cachedPool = pool.slice();
+      this.cachedSpeciesPool = speciesPool.slice();
     } else {
       const EXISTENCE_TAG = ["past", "future", "lgpe", "unobtainable", "cap", "custom", "nonexistent"];
       const nonexistentBanReason = ruleTable.check("nonexistent");
@@ -10949,7 +10979,18 @@ var RandomTeams = class {
         if (pool.includes(num)) continue;
         pool.push(num);
       }
+      this.poolsCacheKey = [requiredType, minSourceGen, ruleTable, requireMoves];
+      this.cachedPool = pool.slice();
+      this.cachedSpeciesPool = speciesPool.slice();
+    }
+    return { pool, speciesPool };
+  }
+  randomNPokemon(n, requiredType, minSourceGen, ruleTable, requireMoves = false) {
+    if (requiredType && !this.dex.types.get(requiredType).exists) {
+      throw new Error(`"${requiredType}" is not a valid type.`);
     }
+    const { pool, speciesPool } = this.getPools(requiredType, minSourceGen, ruleTable, requireMoves);
+    const isNotCustom = !ruleTable;
     const hasDexNumber = {};
     for (let i = 0; i < n; i++) {
       const num = this.sampleNoReplace(pool);
@@ -11188,7 +11229,7 @@ var RandomTeams = class {
         if (this.forceTeraType) {
           set.teraType = this.forceTeraType;
         } else {
-          set.teraType = this.sample(this.dex.types.all()).name;
+          set.teraType = this.sample(this.dex.types.names());
         }
       }
       team.push(set);