package info.u_team.u_team_test.screen;

import com.mojang.blaze3d.matrix.MatrixStack;

import info.u_team.u_team_core.gui.elements.EnergyStorageWidget;
import info.u_team.u_team_core.screen.UContainerScreen;
import info.u_team.u_team_test.TestMod;
import info.u_team.u_team_test.container.BasicEnergyCreatorContainer;
import net.minecraft.entity.player.PlayerInventory;
import net.minecraft.util.ResourceLocation;
import net.minecraft.util.text.ITextComponent;
import net.minecraftforge.api.distmarker.*;
import net.minecraftforge.energy.CapabilityEnergy;

@OnlyIn(Dist.CLIENT)
public class BasicEnergyCreatorScreen extends UContainerScreen<BasicEnergyCreatorContainer> {
	
	public BasicEnergyCreatorScreen(BasicEnergyCreatorContainer container, PlayerInventory playerInventory, ITextComponent title) {
		super(container, playerInventory, title, new ResourceLocation(TestMod.MODID, "textures/gui/energy_creator.png"));
		setSize(176, 173);
	}
	
	@Override
	protected void func_231160_c_() {
		super.func_231160_c_();
		func_230480_a_(new EnergyStorageWidget(guiLeft + 9, guiTop + 20, 54, container.getTileEntity().getCapability(CapabilityEnergy.ENERGY)));
	}
	
	@Override
	public void func_230430_a_(MatrixStack matrixStack, int mouseX, int mouseY, float partialTicks) {
		func_230446_a_(matrixStack);
		super.func_230430_a_(matrixStack, mouseX, mouseY, partialTicks);
		field_230710_m_.forEach(widget -> widget.func_230443_a_(matrixStack, mouseX, mouseY));
		func_230459_a_(matrixStack, mouseX, mouseY);
	}
}

@@ -38,55 +38,8 @@
 n;%0A%0A
-import com.mojang.blaze3d.matrix.MatrixStack;%0A%0A
 impo
@@ -138,16 +138,21 @@
 screen.U
+Basic
 Containe
@@ -577,16 +577,21 @@
 xtends U
+Basic
 Containe
@@ -877,21 +877,10 @@
 ng%22)
-);%0A%09%09setSize(
+, 
 176,
@@ -1109,355 +1109,6 @@
 %0A%09%7D%0A
-%09%0A%09@Override%0A%09public void func_230430_a_(MatrixStack matrixStack, int mouseX, int mouseY, float partialTicks) %7B%0A%09%09func_230446_a_(matrixStack);%0A%09%09super.func_230430_a_(matrixStack, mouseX, mouseY, partialTicks);%0A%09%09field_230710_m_.forEach(widget -%3E widget.func_230443_a_(matrixStack, mouseX, mouseY));%0A%09%09func_230459_a_(matrixStack, mouseX, mouseY);%0A%09%7D%0A
 %7D%0A

Update basic energy creator screen to UBasicContainerScreen and reduce code