fix: Ensure stirling generator saves burn progress

Closes: GH-753
Team-EnderIO · Sep 24, 2024 · e3ece3e · e3ece3e
1 parent 431505c
commit e3ece3e
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/src/machines/java/com/enderio/machines/common/MachineNBTKeys.java b/src/machines/java/com/enderio/machines/common/MachineNBTKeys.java
@@ -16,7 +16,6 @@ public class MachineNBTKeys extends EIONBTKeys {
     public static final String MACHINE_MODE = "Mode";
     public static final String PROCESSED_INPUTS = "ProcessedInputs";
 
-    // TODO: If the previous TODO is carried out, these should probably be moved to PrimitiveAlloySmelterBlockEntity
     public static final String BURN_TIME = "BurnTime";
     public static final String BURN_DURATION = "BurnDuration";
     public static final String RANGE = "Range";

diff --git a/src/machines/java/com/enderio/machines/common/blockentity/StirlingGeneratorBlockEntity.java b/src/machines/java/com/enderio/machines/common/blockentity/StirlingGeneratorBlockEntity.java
@@ -8,6 +8,7 @@
 import com.enderio.api.capacitor.SteppedScalable;
 import com.enderio.api.io.energy.EnergyIOMode;
 import com.enderio.core.common.network.slot.FloatNetworkDataSlot;
+import com.enderio.machines.common.MachineNBTKeys;
 import com.enderio.machines.common.blockentity.base.PoweredMachineBlockEntity;
 import com.enderio.machines.common.config.MachinesConfig;
 import com.enderio.machines.common.io.energy.MachineEnergyStorage;
@@ -160,11 +161,27 @@ protected void onContentsChanged() {
     public void load(CompoundTag pTag) {
         super.load(pTag);
 
+        if (pTag.contains(MachineNBTKeys.BURN_TIME, CompoundTag.TAG_INT)) {
+            burnTime = pTag.getInt(MachineNBTKeys.BURN_TIME);
+        }
+
+        if (pTag.contains(MachineNBTKeys.BURN_DURATION, CompoundTag.TAG_INT)) {
+            burnDuration = pTag.getInt(MachineNBTKeys.BURN_DURATION);
+        }
+
         updateMachineState(MachineState.NO_POWER, false);
         updateMachineState(MachineState.FULL_POWER, (getEnergyStorage().getEnergyStored() >= getEnergyStorage().getMaxEnergyStored()) && isCapacitorInstalled());
         updateMachineState(MachineState.EMPTY_INPUT, FUEL.getItemStack(getInventoryNN()).isEmpty());
     }
 
+    @Override
+    public void saveAdditional(CompoundTag pTag) {
+        super.saveAdditional(pTag);
+
+        pTag.putInt(MachineNBTKeys.BURN_TIME, burnTime);
+        pTag.putInt(MachineNBTKeys.BURN_DURATION, burnDuration);
+    }
+
     @Override
     public void setLevel(Level level) {
         super.setLevel(level);