Small fix for willpower leveling

2025-11-17 11:22:14 +01:00
parent ccfecb4317
commit bce2349dce
2 changed files with 15 additions and 3 deletions
@@ -52,9 +52,15 @@ public class SanityCapability implements ISanity {
    }

    public void setBonusMaxSanity(float bonusMaxSanity) {
+        float oldMax = getMaxSanity();
        this.bonusMaxSanity = Math.max(0f, bonusMaxSanity);
-        setSanity(this.sanity);
-        setInsanity(this.insanity);
+        float newMax = getMaxSanity();
+        if (oldMax > 0) {
+            float pct = sanity / oldMax;
+            sanity = pct * newMax;
+        }
+        setSanity(sanity);
+        setInsanity(insanity);
    }

    public float getBonusMaxSanity() {
@@ -49,8 +49,14 @@ public class TempoCapability implements ITempo {
    }

    public void setBonusMaxTempo(float bonusMaxTempo) {
+        int oldMax = getMaxTempo();
        this.bonusMaxTempo = Math.max(0f, bonusMaxTempo);
-        setTempo(this.tempo);
+        int newMax = getMaxTempo();
+        if (oldMax > 0) {
+            float pct = (float) tempo / (float) oldMax;
+            tempo = Math.round(pct * newMax);
+        }
+        setTempo(tempo);
    }

    public float getBonusMaxTempo() {