Fixes #104 (#106)

CleanroomMC · Dec 16, 2023 · 6b4ae6a · 6b4ae6a
1 parent 2cd6506
commit 6b4ae6a
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/.../java/com/cleanroommc/groovyscript/compat/mods/astralsorcery/perktree/PerkTreeConfig.java b/.../java/com/cleanroommc/groovyscript/compat/mods/astralsorcery/perktree/PerkTreeConfig.java
@@ -11,15 +11,14 @@
 import org.jetbrains.annotations.ApiStatus;
 
 import java.util.Arrays;
-import java.util.Random;
 
 public class PerkTreeConfig extends VirtualizedRegistry<Closure<Long>> {
 
     @Override
     @GroovyBlacklist
     @ApiStatus.Internal
     public void onReload() {
-        this.setXpFunction(null);
+        this.xpFunction = null;
         this.setLevelCap(30);
     }
 
@@ -30,12 +29,15 @@ public void setXpFunction(Closure<Long> func) {
             GroovyLog.msg("Warning: Astral Perk xp closures must take the following parameters (int levelNumber, long previousLevelXp)").debug().post();
         }
         this.xpFunction = func;
-        ((PerkLevelManagerAccessor) PerkLevelManager.INSTANCE).getLevelMap().clear();
-        ((PerkLevelManagerAccessor) PerkLevelManager.INSTANCE).generateLevelMap();
+        resetLevelMap();
     }
 
     public void setLevelCap(int cap) {
         PerkLevelManagerAccessor.setLevelCap(cap);
+        resetLevelMap();
+    }
+
+    private void resetLevelMap() {
         ((PerkLevelManagerAccessor) PerkLevelManager.INSTANCE).getLevelMap().clear();
         ((PerkLevelManagerAccessor) PerkLevelManager.INSTANCE).generateLevelMap();
     }