GuiEnchantmentModifier.java

package codechicken.nei;

import codechicken.lib.render.CCRenderState;
import net.minecraft.client.gui.FontRenderer;
import net.minecraft.client.gui.GuiButton;
import net.minecraft.client.gui.inventory.GuiContainer;
import net.minecraft.client.renderer.GlStateManager;
import net.minecraft.entity.player.InventoryPlayer;
import net.minecraft.world.World;

import java.io.IOException;

import static codechicken.nei.NEIClientUtils.translate;

public class GuiEnchantmentModifier extends GuiContainer {
    ContainerEnchantmentModifier container;

    public GuiEnchantmentModifier(InventoryPlayer inventoryplayer, World world) {
        super(new ContainerEnchantmentModifier(inventoryplayer, world));
        container = (ContainerEnchantmentModifier) inventorySlots;
        container.parentscreen = this;
    }

    @Override
    protected void drawGuiContainerForegroundLayer(int par1, int par2) {
        fontRendererObj.drawString(translate("enchant"), 12, 6, 0x404040);
        fontRendererObj.drawString(translate("enchant.level"), 19, 20, 0x404040);
    }

    protected void drawGuiContainerBackgroundLayer(float f, int i, int j) {
        GlStateManager.color(1, 1, 1, 1);
        CCRenderState.changeTexture("textures/gui/container/enchanting_table.png");
        GlStateManager.translate(guiLeft, guiTop, 0);
        drawTexturedModalRect(0, 0, 0, 0, xSize, ySize);

        container.onUpdate(i, j);
        container.drawSlots(this);
        container.drawScrollBar(this);

        String levelstring = "" + container.level;
        fontRendererObj.drawString(levelstring, 33 - fontRendererObj.getStringWidth(levelstring) / 2, 34, 0xFF606060);

        GlStateManager.translate(-guiLeft, -guiTop, 0);
    }

    public void initGui() {
        super.initGui();

        buttonList.add(new GuiNEIButton(0, width / 2 - 78, height / 2 - 52, 12, 12, "<"));
        buttonList.add(new GuiNEIButton(1, width / 2 - 44, height / 2 - 52, 12, 12, ">"));
        buttonList.add(new GuiNEIButton(2, width / 2 - 80, height / 2 - 15, 50, 12, lockDisplayString()));
    }

    private String lockDisplayString() {
        return validateEnchantments() ? translate("enchant.locked") : translate("enchant.unlocked");
    }

    public static boolean validateEnchantments() {
        return NEIClientConfig.world.nbt.getBoolean("validateenchantments");
    }

    public static void toggleEnchantmentValidation() {
        NEIClientConfig.world.nbt.setBoolean("validateenchantments", !validateEnchantments());
        NEIClientConfig.world.saveNBT();
    }

    protected void actionPerformed(GuiButton guibutton) {
        if (guibutton.id == 0) {
            changeLevel(-1);
        } else if (guibutton.id == 1) {
            changeLevel(1);
        } else if (guibutton.id == 2) {
            toggleEnchantmentValidation();
            container.updateEnchantmentOptions(validateEnchantments());
            guibutton.displayString = lockDisplayString();
        }
    }

    private void changeLevel(int i) {
        container.level += i;
        ((GuiButton) buttonList.get(0)).enabled = container.level != 1;
        ((GuiButton) buttonList.get(1)).enabled = container.level != 10;
    }

    @Override
    protected void mouseClicked(int i, int j, int k) throws IOException {
        if (container.clickButton(i, j, k)) {
            return;
        }
        if (container.clickScrollBar(i, j, k)) {
            return;
        }

        super.mouseClicked(i, j, k);
    }

    @Override
    protected void mouseReleased(int i, int j, int k) {
        container.mouseUp(i, j, k);
        super.mouseReleased(i, j, k);
    }

    public FontRenderer getFontRenderer() {
        return fontRendererObj;
    }
}