From e465b8abd4cc23adf4478338da12fbd18afe678e Mon Sep 17 00:00:00 2001
From: tom-overton <tom.overton@outlook.com>
Date: Mon, 5 Dec 2022 04:51:02 -0800
Subject: Add support for updating stats/moves to Gen 9

---
 .../sneed/pkrandom/constants/GlobalConstants.java  |  7 +++-
 .../pkrandom/romhandlers/AbstractRomHandler.java   | 39 ++++++++++++++++++++++
 2 files changed, 45 insertions(+), 1 deletion(-)

diff --git a/src/com/sneed/pkrandom/constants/GlobalConstants.java b/src/com/sneed/pkrandom/constants/GlobalConstants.java
index e9046ca..fa41d5c 100644
--- a/src/com/sneed/pkrandom/constants/GlobalConstants.java
+++ b/src/com/sneed/pkrandom/constants/GlobalConstants.java
@@ -173,6 +173,11 @@ public class GlobalConstants {
             case 8:
                 map.put(Species.aegislash,new StatChange(Stat.DEF.val | Stat.SPDEF.val,140,140));
                 break;
+            case 9:
+                map.put(Species.cresselia,new StatChange(Stat.DEF.val | Stat.SPDEF.val, 110,120));
+                map.put(Species.zacian,new StatChange(Stat.ATK.val, 120));
+                map.put(Species.zamazenta,new StatChange(Stat.ATK.val, 120));
+                break;
         }
         return map;
     }
@@ -241,7 +246,7 @@ public class GlobalConstants {
 
     public static final int MIN_DAMAGING_MOVE_POWER = 50;
 
-    public static final int HIGHEST_POKEMON_GEN = 8;
+    public static final int HIGHEST_POKEMON_GEN = 9;
 
     // Eevee has 8 potential evolutions
     public static final int LARGEST_NUMBER_OF_SPLIT_EVOS = 8;
diff --git a/src/com/sneed/pkrandom/romhandlers/AbstractRomHandler.java b/src/com/sneed/pkrandom/romhandlers/AbstractRomHandler.java
index 64b797d..aade38d 100755
--- a/src/com/sneed/pkrandom/romhandlers/AbstractRomHandler.java
+++ b/src/com/sneed/pkrandom/romhandlers/AbstractRomHandler.java
@@ -3320,6 +3320,45 @@ public abstract class AbstractRomHandler implements RomHandler {
                 updateMovePower(moves, Moves.multiAttack, 120);
             }
         }
+
+        if (generation >= 9 && generationOfPokemon() < 9) {
+            // Gen 1
+            // Recover 5 PP
+            updateMovePP(moves, Moves.recover, 5);
+            // Soft-Boiled 5 PP
+            updateMovePP(moves, Moves.softBoiled, 5);
+            // Rest 5 PP
+            updateMovePP(moves, Moves.rest, 5);
+
+            if (generationOfPokemon() >= 2) {
+                // Milk Drink 5 PP
+                updateMovePP(moves, Moves.milkDrink, 5);
+            }
+
+            if (generationOfPokemon() >= 3) {
+                // Slack Off 5 PP
+                updateMovePP(moves, Moves.slackOff, 5);
+            }
+
+            if (generationOfPokemon() >= 4) {
+                // Roost 5 PP
+                updateMovePP(moves, Moves.roost, 5);
+            }
+            
+            if (generationOfPokemon() >= 7) {
+                // Shore Up 5 PP
+                updateMovePP(moves, Moves.shoreUp, 5);
+            }
+
+            if (generationOfPokemon() >= 8) {
+                // Grassy Glide 60 Power
+                updateMovePower(moves, Moves.grassyGlide, 60);
+                // Wicked Blow 75 Power
+                updateMovePower(moves, Moves.wickedBlow, 75);
+                // Glacial Lance 120 Power
+                updateMovePower(moves, Moves.glacialLance, 120);
+            }
+        }
     }
 
     private Map<Integer, boolean[]> moveUpdates;
-- 
cgit v1.2.3