1 files changed, 80 insertions, 0 deletions
diff --git a/src/com/pkrandom/pokemon/Aura.java b/src/com/pkrandom/pokemon/Aura.java
new file mode 100644
index 0000000..946e844
--- /dev/null
+++ b/src/com/pkrandom/pokemon/Aura.java
@@ -0,0 +1,80 @@
+package com.pkrandom.pokemon;
+
+/*----------------------------------------------------------------------------*/
+/*--  Aura.java - handles the Aura that Totem Pokemon have                  --*/
+/*--                                                                        --*/
+/*--  Part of "Universal Pokemon Randomizer ZX" by the UPR-ZX team          --*/
+/*--  Pokemon and any associated names and the like are                     --*/
+/*--  trademark and (C) Nintendo 1996-2020.                                 --*/
+/*--                                                                        --*/
+/*--  The custom code written here is licensed under the terms of the GPL:  --*/
+/*--                                                                        --*/
+/*--  This program is free software: you can redistribute it and/or modify  --*/
+/*--  it under the terms of the GNU General Public License as published by  --*/
+/*--  the Free Software Foundation, either version 3 of the License, or     --*/
+/*--  (at your option) any later version.                                   --*/
+/*--                                                                        --*/
+/*--  This program is distributed in the hope that it will be useful,       --*/
+/*--  but WITHOUT ANY WARRANTY; without even the implied warranty of        --*/
+/*--  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the          --*/
+/*--  GNU General Public License for more details.                          --*/
+/*--                                                                        --*/
+/*--  You should have received a copy of the GNU General Public License     --*/
+/*--  along with this program. If not, see <http://www.gnu.org/licenses/>.  --*/
+/*----------------------------------------------------------------------------*/
+
+import com.pkrandom.RomFunctions;
+
+import java.util.Random;
+
+public class Aura {
+
+    private enum AuraStat {
+        NONE, ATTACK, DEFENSE, SPECIAL_ATTACK, SPECIAL_DEFENSE, SPEED, ALL
+    }
+
+    public AuraStat stat;
+
+    public int stages;
+
+    public Aura(byte b) {
+        if (b == 0) {
+            stat = AuraStat.NONE;
+            stages = 0;
+        } else {
+            stat = AuraStat.values()[((b - 1) / 3) + 1];
+            stages = ((b - 1) % 3) + 1;
+        }
+    }
+
+    private Aura(AuraStat stat, int stages) {
+        this.stat = stat;
+        this.stages = stages;
+    }
+
+    public byte toByte() {
+        if (stat == AuraStat.NONE) {
+            return 0;
+        } else {
+            return (byte)(((stat.ordinal() - 1) * 3) + (stages));
+        }
+    }
+
+    public static Aura randomAura(Random random) {
+        return new Aura((byte)(random.nextInt(18) + 1));
+    }
+
+    public static Aura randomAuraSimilarStrength(Random random, Aura old) {
+        if (old.stat == AuraStat.NONE || old.stat == AuraStat.ALL) {
+            return old;
+        } else {
+            return new Aura(AuraStat.values()[random.nextInt(5) + 1], old.stages);
+        }
+    }
+
+    @Override
+    public String toString() {
+        String ret = RomFunctions.camelCase(stat.toString()).replace("_"," ");
+        return stat == AuraStat.NONE ? ret : ret + " +" + stages;
+    }
+}