Applied-Energistics-2-tilera-Edition/src/main/java/appeng/integration/modules/NEI.java

/*
 * This file is part of Applied Energistics 2.
 * Copyright (c) 2013 - 2014, AlgorithmX2, All rights reserved.
 *
 * Applied Energistics 2 is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Applied Energistics 2 is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with Applied Energistics 2.  If not, see <http://www.gnu.org/licenses/lgpl>.
 */

package appeng.integration.modules;

import java.lang.reflect.Constructor;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.List;

import appeng.client.gui.AEBaseMEGui;
import appeng.client.gui.implementations.GuiCraftingTerm;
import appeng.client.gui.implementations.GuiPatternTerm;
import appeng.core.AEConfig;
import appeng.core.features.AEFeature;
import appeng.helpers.Reflected;
import appeng.integration.IIntegrationModule;
import appeng.integration.IntegrationHelper;
import appeng.integration.abstraction.INEI;
import appeng.integration.modules.NEIHelpers.*;
import codechicken.nei.api.IStackPositioner;
import codechicken.nei.guihook.GuiContainerManager;
import codechicken.nei.guihook.IContainerTooltipHandler;
import net.minecraft.client.Minecraft;
import net.minecraft.client.gui.FontRenderer;
import net.minecraft.client.gui.inventory.GuiContainer;
import net.minecraft.client.renderer.entity.RenderItem;
import net.minecraft.inventory.Slot;
import net.minecraft.item.ItemStack;

public class NEI implements INEI, IContainerTooltipHandler, IIntegrationModule {
    @Reflected
    public static NEI instance;

    private final Class<?> apiClass;

    // recipe handler...
    private Method registerRecipeHandler;
    private Method registerUsageHandler;

    @Reflected
    public NEI() throws ClassNotFoundException {
        IntegrationHelper.testClassExistence(this, codechicken.nei.api.API.class);
        IntegrationHelper
            .testClassExistence(this, codechicken.nei.api.IStackPositioner.class);
        IntegrationHelper
            .testClassExistence(this, codechicken.nei.guihook.GuiContainerManager.class);
        IntegrationHelper.testClassExistence(
            this, codechicken.nei.guihook.IContainerTooltipHandler.class
        );
        IntegrationHelper
            .testClassExistence(this, codechicken.nei.recipe.ICraftingHandler.class);
        IntegrationHelper
            .testClassExistence(this, codechicken.nei.recipe.IUsageHandler.class);

        this.apiClass = Class.forName("codechicken.nei.api.API");
    }

    @Override
    public void init() throws Throwable {
        this.registerRecipeHandler = this.apiClass.getDeclaredMethod(
            "registerRecipeHandler", codechicken.nei.recipe.ICraftingHandler.class
        );
        this.registerUsageHandler = this.apiClass.getDeclaredMethod(
            "registerUsageHandler", codechicken.nei.recipe.IUsageHandler.class
        );

        this.registerRecipeHandler(new NEIAEShapedRecipeHandler());
        this.registerRecipeHandler(new NEIAEShapelessRecipeHandler());
        this.registerRecipeHandler(new NEIInscriberRecipeHandler());
        this.registerRecipeHandler(new NEIWorldCraftingHandler());
        this.registerRecipeHandler(new NEIGrinderRecipeHandler());

        if (AEConfig.instance.isFeatureEnabled(AEFeature.Facades)
            && AEConfig.instance.isFeatureEnabled(AEFeature.EnableFacadeCrafting)) {
            this.registerRecipeHandler(new NEIFacadeRecipeHandler());
        }

        // large stack tooltips
        GuiContainerManager.addTooltipHandler(this);

        // crafting terminal...
        final Method registerGuiOverlay = this.apiClass.getDeclaredMethod(
            "registerGuiOverlay", Class.class, String.class, IStackPositioner.class
        );
        final Class overlayHandler = Class.forName("codechicken.nei.api.IOverlayHandler");

        final Method registrar = this.apiClass.getDeclaredMethod(
            "registerGuiOverlayHandler", Class.class, overlayHandler, String.class
        );
        registerGuiOverlay.invoke(
            this.apiClass,
            GuiCraftingTerm.class,
            "crafting",
            new TerminalCraftingSlotFinder()
        );
        registerGuiOverlay.invoke(
            this.apiClass,
            GuiPatternTerm.class,
            "crafting",
            new TerminalCraftingSlotFinder()
        );

        final Class<NEICraftingHandler> defaultHandler = NEICraftingHandler.class;
        final Constructor defaultConstructor
            = defaultHandler.getConstructor(int.class, int.class);
        registrar.invoke(
            this.apiClass,
            GuiCraftingTerm.class,
            defaultConstructor.newInstance(6, 75),
            "crafting"
        );
        registrar.invoke(
            this.apiClass,
            GuiPatternTerm.class,
            defaultConstructor.newInstance(6, 75),
            "crafting"
        );
    }

    public void registerRecipeHandler(final Object o
    ) throws IllegalAccessException, IllegalArgumentException, InvocationTargetException {
        this.registerRecipeHandler.invoke(this.apiClass, o);
        this.registerUsageHandler.invoke(this.apiClass, o);
    }

    @Override
    public void postInit() {}

    @Override
    public void drawSlot(final Slot s) {
        if (s == null) {
            return;
        }

        final ItemStack stack = s.getStack();

        if (stack == null) {
            return;
        }

        final Minecraft mc = Minecraft.getMinecraft();
        final FontRenderer fontRenderer = mc.fontRenderer;
        final int x = s.xDisplayPosition;
        final int y = s.yDisplayPosition;

        GuiContainerManager.drawItems.renderItemAndEffectIntoGUI(
            fontRenderer, mc.getTextureManager(), stack, x, y
        );
        GuiContainerManager.drawItems.renderItemOverlayIntoGUI(
            fontRenderer,
            mc.getTextureManager(),
            stack,
            x,
            y,
            String.valueOf(stack.stackSize)
        );
    }

    @Override
    public RenderItem setItemRender(final RenderItem renderItem) {
        try {
            final RenderItem ri = GuiContainerManager.drawItems;
            GuiContainerManager.drawItems = renderItem;
            return ri;
        } catch (final Throwable t) {
            throw new IllegalStateException("Invalid version of NEI, please update", t);
        }
    }

    @Override
    public List<String> handleTooltip(
        final GuiContainer arg0,
        final int arg1,
        final int arg2,
        final List<String> current
    ) {
        return current;
    }

    @Override
    public List<String> handleItemDisplayName(
        final GuiContainer arg0, final ItemStack arg1, final List<String> current
    ) {
        return current;
    }

    @Override
    public List<String> handleItemTooltip(
        final GuiContainer guiScreen,
        final ItemStack stack,
        final int mouseX,
        final int mouseY,
        final List<String> currentToolTip
    ) {
        if (guiScreen instanceof AEBaseMEGui) {
            return ((AEBaseMEGui) guiScreen)
                .handleItemTooltip(stack, mouseX, mouseY, currentToolTip);
        }

        return currentToolTip;
    }
}