Started with some Serialization

InnovativeOnlineIndustries · Nov 21, 2020 · e434ca8 · e434ca8
1 parent fb8adc3
commit e434ca8
Showing 1 changed file with 35 additions and 2 deletions.
diff --git a/src/main/java/com/hrznstudio/matteroverdrive/api/android/module/AndroidModule.java b/src/main/java/com/hrznstudio/matteroverdrive/api/android/module/AndroidModule.java
@@ -10,12 +10,13 @@
 import net.minecraft.util.Util;
 import net.minecraft.util.text.ITextComponent;
 import net.minecraft.util.text.TranslationTextComponent;
+import net.minecraftforge.common.util.INBTSerializable;
 import net.minecraftforge.registries.ForgeRegistryEntry;
 
 import javax.annotation.Nonnull;
 import java.util.List;
 
-public abstract class AndroidModule extends ForgeRegistryEntry<AndroidModule> {
+public abstract class AndroidModule extends ForgeRegistryEntry<AndroidModule> implements INBTSerializable<CompoundNBT> {
 
     private final int maxTier;
     private final int minTier;
@@ -26,22 +27,27 @@ public abstract class AndroidModule extends ForgeRegistryEntry<AndroidModule> {
     private List<ITextComponent> tooltip;
     private int currentTier;
 
+    public static final String CURRENT_TIER = "currentTier";
+
     public AndroidModule() {
         this.maxTier = 1;
         this.minTier = 0;
         this.shouldAllowStackingModules = false;
+        this.currentTier = this.minTier;
     }
 
     public AndroidModule(int maxTier) {
         this.maxTier = maxTier;
         this.minTier = 0;
         this.shouldAllowStackingModules = true;
+        this.currentTier = this.minTier;
     }
 
-    public AndroidModule(int maxTier, int minTier) {
+    public AndroidModule(int maxTier, int minTier, int initTier) {
         this.maxTier = maxTier;
         this.minTier = minTier;
         this.shouldAllowStackingModules = true;
+        this.currentTier = initTier;
     }
 
     /**
@@ -153,4 +159,31 @@ public int getTierInRange(int level) {
     public int getCurrentTier() {
         return currentTier;
     }
+
+    /**
+     * @param currentTier Returns the current Tier of the installed Module
+     */
+    public void setCurrentTier(int currentTier) {
+        this.currentTier = currentTier;
+    }
+
+    /**
+     * @return Returns the Serialized Data of the Module
+     */
+    @Override
+    public CompoundNBT serializeNBT() {
+        CompoundNBT nbt = new CompoundNBT();
+        nbt.putInt(CURRENT_TIER, this.getCurrentTier());
+        return nbt;
+    }
+
+    /**
+     * Deserializes the Data from NBT to actual data the Module can use.
+     *
+     * @param nbt The serialized Data of the Module
+     */
+    @Override
+    public void deserializeNBT(CompoundNBT nbt) {
+        this.setCurrentTier(nbt.getInt(CURRENT_TIER));
+    }
 }