Add CableTierHolder to electric machines

Author: Swedz

src/main/java/aztech/modern_industrialization/machines/blockentities/ElectricCraftingMachineBlockEntity.java

@@ -24,6 +24,8 @@
 package aztech.modern_industrialization.machines.blockentities;
 
 import aztech.modern_industrialization.MICapabilities;
+import aztech.modern_industrialization.api.energy.CableTier;
+import aztech.modern_industrialization.api.energy.CableTierHolder;
 import aztech.modern_industrialization.api.energy.EnergyApi;
 import aztech.modern_industrialization.api.energy.MIEnergyStorage;
 import aztech.modern_industrialization.api.machine.holder.EnergyComponentHolder;
@@ -44,7 +46,7 @@
 import net.minecraft.world.entity.player.Player;
 import net.minecraft.world.level.block.entity.BlockEntityType;
 
-public class ElectricCraftingMachineBlockEntity extends AbstractCraftingMachineBlockEntity implements EnergyComponentHolder {
+public class ElectricCraftingMachineBlockEntity extends AbstractCraftingMachineBlockEntity implements EnergyComponentHolder, CableTierHolder {
 
     public ElectricCraftingMachineBlockEntity(BEP bep, MachineRecipeType recipeType, MachineInventoryComponent inventory,
             MachineGuiParameters guiParams, EnergyBar.Parameters energyBarParams, ProgressBar.Parameters progressBarParams,
@@ -132,4 +134,9 @@ public boolean isOverdriving() {
     public EnergyComponent getEnergyComponent() {
         return energy;
     }
+
+    @Override
+    public CableTier getCableTier() {
+        return casing.getCableTier();
+    }
 }

src/main/java/aztech/modern_industrialization/machines/blockentities/ElectricWaterPumpBlockEntity.java

@@ -25,6 +25,7 @@
 
 import aztech.modern_industrialization.MICapabilities;
 import aztech.modern_industrialization.api.energy.CableTier;
+import aztech.modern_industrialization.api.energy.CableTierHolder;
 import aztech.modern_industrialization.api.energy.EnergyApi;
 import aztech.modern_industrialization.api.energy.MIEnergyStorage;
 import aztech.modern_industrialization.api.machine.holder.EnergyComponentHolder;
@@ -43,7 +44,7 @@
 import net.minecraft.world.level.material.Fluids;
 import net.neoforged.neoforge.fluids.FluidType;
 
-public class ElectricWaterPumpBlockEntity extends AbstractWaterPumpBlockEntity implements EnergyComponentHolder {
+public class ElectricWaterPumpBlockEntity extends AbstractWaterPumpBlockEntity implements EnergyComponentHolder, CableTierHolder {
     public ElectricWaterPumpBlockEntity(BEP bep) {
         super(bep, "electric_water_pump");
 
@@ -105,4 +106,9 @@ public static void registerEnergyApi(BlockEntityType<?> bet) {
             event.registerBlockEntity(EnergyApi.SIDED, bet, (be, direction) -> ((ElectricWaterPumpBlockEntity) be).insertable);
         });
     }
+
+    @Override
+    public CableTier getCableTier() {
+        return CableTier.LV;
+    }
 }

src/main/java/aztech/modern_industrialization/machines/blockentities/GeneratorMachineBlockEntity.java

@@ -27,6 +27,7 @@
 
 import aztech.modern_industrialization.MICapabilities;
 import aztech.modern_industrialization.api.energy.CableTier;
+import aztech.modern_industrialization.api.energy.CableTierHolder;
 import aztech.modern_industrialization.api.energy.EnergyApi;
 import aztech.modern_industrialization.api.energy.MIEnergyStorage;
 import aztech.modern_industrialization.api.machine.holder.EnergyComponentHolder;
@@ -50,7 +51,7 @@
 import net.minecraft.network.chat.Component;
 import net.minecraft.world.level.block.entity.BlockEntityType;
 
-public class GeneratorMachineBlockEntity extends MachineBlockEntity implements Tickable, EnergyComponentHolder {
+public class GeneratorMachineBlockEntity extends MachineBlockEntity implements Tickable, EnergyComponentHolder, CableTierHolder {
 
     private final CableTier outputTier;
     private final MIEnergyStorage extractable;
@@ -212,4 +213,9 @@ public static void registerEnergyApi(BlockEntityType<?> bet) {
     public List<Component> getTooltips() {
         return fluidItemConsumer.getTooltips();
     }
+
+    @Override
+    public CableTier getCableTier() {
+        return outputTier;
+    }
 }

src/main/java/aztech/modern_industrialization/machines/blockentities/StorageMachineBlockEntity.java

@@ -24,10 +24,11 @@
 package aztech.modern_industrialization.machines.blockentities;
 
 import aztech.modern_industrialization.api.energy.CableTier;
+import aztech.modern_industrialization.api.energy.CableTierHolder;
 import aztech.modern_industrialization.machines.BEP;
 import net.minecraft.util.Mth;
 
-public class StorageMachineBlockEntity extends AbstractStorageMachineBlockEntity {
+public class StorageMachineBlockEntity extends AbstractStorageMachineBlockEntity implements CableTierHolder {
 
     public StorageMachineBlockEntity(BEP bep, CableTier tier, String name, long eu_capacity) {
         super(bep, tier, tier, name, eu_capacity);
@@ -43,4 +44,9 @@ protected int getComparatorOutput() {
         double fillPercentage = (double) energy.getEu() / energy.getCapacity();
         return Mth.floor(fillPercentage * 14) + (energy.getEu() > 0 ? 1 : 0);
     }
+
+    @Override
+    public CableTier getCableTier() {
+        return to;
+    }
 }