IC2Module.java

package mcp.mobius.waila.addons.ic2;

import mcp.mobius.waila.Waila;
import mcp.mobius.waila.api.IWailaPlugin;
import mcp.mobius.waila.api.IWailaRegistrar;
import mcp.mobius.waila.api.WailaPlugin;
import net.minecraftforge.fml.common.Loader;

import java.lang.reflect.Field;

@WailaPlugin
public class IC2Module implements IWailaPlugin {

    public static Class generator = null;
    public static Field generatorStorage = null;
    public static Field generatorMaxStorage = null;
    public static Field generatorProduction = null;


    @Override
    public void register(IWailaRegistrar registrar) {
        if (!Loader.isModLoaded("ic2")) return;
        try {
            generator = Class.forName("ic2.core.block.base.tile.TileEntityGeneratorBase");
            generatorStorage = generator.getDeclaredField("storage");
            generatorMaxStorage = generator.getDeclaredField("maxStorage");
            generatorProduction = generator.getDeclaredField("production");


            registrar.registerBodyProvider(HUDHandlerTEGenerator.INSTANCE, generator);
            registrar.registerNBTProvider(HUDHandlerTEGenerator.INSTANCE, generator);

            registrar.addConfig("Industrial Craft 2", "ic2.storage", true);
            registrar.addConfig("Industrial Craft 2", "ic2.outputeu", true);

        } catch (Exception e) {
            Waila.LOGGER.warn("[Industrial Craft 2] Error while loading generator hooks.", e);
        }
    }
}