diff --git a/src/main/java/ru/dbotthepony/mc/otm/block/entity/BlockEntityMatteryPoweredMachine.java b/src/main/java/ru/dbotthepony/mc/otm/block/entity/BlockEntityMatteryPoweredMachine.java index 81726b6c3..a8838c449 100644 --- a/src/main/java/ru/dbotthepony/mc/otm/block/entity/BlockEntityMatteryPoweredMachine.java +++ b/src/main/java/ru/dbotthepony/mc/otm/block/entity/BlockEntityMatteryPoweredMachine.java @@ -26,6 +26,7 @@ import java.util.Optional; abstract public class BlockEntityMatteryPoweredMachine extends BlockEntityMatteryMachine { protected MatteryMachineEnergyStorage energy = null; protected LazyOptional energy_resolver = null; + private boolean valid = true; public SimpleSerializableContainer battery_container = new SimpleSerializableContainer(1); @@ -81,9 +82,21 @@ abstract public class BlockEntityMatteryPoweredMachine extends BlockEntityMatter super(p_155228_, p_155229_, p_155230_); } + @Override + public void invalidateCaps() { + super.invalidateCaps(); + valid = false; + } + + @Override + public void reviveCaps() { + super.reviveCaps(); + valid = true; + } + @Nonnull public LazyOptional getCapability(@Nonnull Capability cap, @Nullable Direction side) { - if ((cap == MatteryCapability.ENERGY || cap == CapabilityEnergy.ENERGY) && energy_resolver != null) { + if (valid && (cap == MatteryCapability.ENERGY || cap == CapabilityEnergy.ENERGY) && energy_resolver != null) { return energy_resolver.cast(); }