package dev.ginyai.mod._979381.asm;



import cpw.mods.fml.relauncher.IFMLLoadingPlugin;

import org.apache.logging.log4j.LogManager;

import org.apache.logging.log4j.Logger;



import java.util.Map;



@IFMLLoadingPlugin.MCVersion("1.7.10")

@IFMLLoadingPlugin.Name("FixNpeIc2EnergyNet")

public class Plugin implements IFMLLoadingPlugin {

    public static Logger logger = LogManager.getLogger("FixNpeIc2EnergyNet");



    @Override

    public String[] getASMTransformerClass() {

        return new String[] {

            "dev.ginyai.mod._979381.asm.Transformer"

        };

    }



    @Override

    public String getModContainerClass() {

        return null;

    }



    @Override

    public String getSetupClass() {

        return null;

    }



    @Override

    public void injectData(Map<String, Object> data) {



    }



    @Override

    public String getAccessTransformerClass() {

        return null;

    }

}

package dev.ginyai.mod._979381.asm;



import net.minecraft.launchwrapper.IClassTransformer;

import org.objectweb.asm.ClassReader;

import org.objectweb.asm.ClassWriter;

import org.objectweb.asm.Label;

import org.objectweb.asm.tree.AbstractInsnNode;

import org.objectweb.asm.tree.ClassNode;

import org.objectweb.asm.tree.InsnList;

import org.objectweb.asm.tree.MethodInsnNode;

import org.objectweb.asm.tree.MethodNode;

import org.objectweb.asm.tree.VarInsnNode;



import static org.objectweb.asm.Opcodes.*;



public class Transformer implements IClassTransformer {



    @Override

    public byte[] transform(String name, String transformedName, byte[] basicClass) {

        if ("ic2.core.energy.EnergyNetLocal".equals(transformedName)) {

            Plugin.logger.info("Find target class.");

            ClassReader classReader = new ClassReader(basicClass);

            ClassWriter classWriter = new ClassWriter(ClassWriter.COMPUTE_FRAMES);

            ClassNode classNode = new ClassNode(ASM5);

            classReader.accept(classNode, ClassReader.EXPAND_FRAMES);

            for (MethodNode methodNode : classNode.methods) {

                if ("addTileToGrids".equals(methodNode.name) && "(Lic2/core/energy/Tile;)V".equals(methodNode.desc)) {

                    Plugin.logger.info("Find target method.");

                    InsnList insnList = methodNode.instructions;

                    AbstractInsnNode[] nodes = insnList.toArray();

                    for (int i = 0; i + 3 < nodes.length ; i++ ) {

                        if (checkMethodNode(nodes, INVOKEVIRTUAL, "ic2/core/energy/Node", "getGrid", "()Lic2/core/energy/Grid;")

                            && (nodes[i + 1] instanceof VarInsnNode)

                            && (nodes[i + 2] instanceof VarInsnNode)

                            && checkMethodNode(nodes[i + 3], INVOKEVIRTUAL, "ic2/core/energy/Grid", "add", "(Lic2/core/energy/Node;Ljava/util/Collection;)V")

                        ) {

                            Plugin.logger.info("Find target point.");

                            insnList.insert(nodes, new VarInsnNode(ALOAD, 0));

                            insnList.insert(nodes[i + 3], new MethodInsnNode(INVOKESTATIC, "ic2/core/energy/EnergyNetLocal", "fnietDummyMethod", "(Lic2/core/energy/Node;Lic2/core/energy/EnergyNetLocal;Lic2/core/energy/Node;Ljava/util/Collection;)V", false));

                            insnList.remove(nodes);

                            insnList.remove(nodes[i + 3]);

                            break;

                        }

                    }

                    break;

                }

            }

            MethodNode dummyMethod = new MethodNode(ASM5, ACC_PRIVATE | ACC_STATIC, "fnietDummyMethod", "(Lic2/core/energy/Node;Lic2/core/energy/EnergyNetLocal;Lic2/core/energy/Node;Ljava/util/Collection;)V", "(Lic2/core/energy/Node;Lic2/core/energy/EnergyNetLocal;Lic2/core/energy/Node;Ljava/util/Collection<Lic2/core/energy/Node;>;)V", null);

            dummyMethod.visitCode();

            Label label0 = new Label();

            dummyMethod.visitLabel(label0);

            dummyMethod.visitLineNumber(10, label0);

            dummyMethod.visitVarInsn(ALOAD, 0);

            Label label1 = new Label();

            dummyMethod.visitJumpInsn(IFNONNULL, label1);

            Label label2 = new Label();

            dummyMethod.visitLabel(label2);

            dummyMethod.visitLineNumber(12, label2);

            dummyMethod.visitInsn(RETURN);

            dummyMethod.visitLabel(label1);

            dummyMethod.visitLineNumber(14, label1);

            dummyMethod.visitFrame(F_SAME, 0, null, 0, null);

            dummyMethod.visitVarInsn(ALOAD, 0);

            dummyMethod.visitMethodInsn(INVOKEVIRTUAL, "ic2/core/energy/Node", "getGrid", "()Lic2/core/energy/Grid;", false);

            dummyMethod.visitVarInsn(ASTORE, 4);

            Label label3 = new Label();

            dummyMethod.visitLabel(label3);

            dummyMethod.visitLineNumber(15, label3);

            dummyMethod.visitVarInsn(ALOAD, 4);

            Label label4 = new Label();

            dummyMethod.visitJumpInsn(IFNONNULL, label4);

            Label label5 = new Label();

            dummyMethod.visitLabel(label5);

            dummyMethod.visitLineNumber(16, label5);

            dummyMethod.visitTypeInsn(NEW, "ic2/core/energy/Grid");

            dummyMethod.visitInsn(DUP);

            dummyMethod.visitVarInsn(ALOAD, 1);

            dummyMethod.visitMethodInsn(INVOKESPECIAL, "ic2/core/energy/Grid", "<init>", "(Lic2/core/energy/EnergyNetLocal;)V", false);

            dummyMethod.visitVarInsn(ASTORE, 4);

            Label label6 = new Label();

            dummyMethod.visitLabel(label6);

            dummyMethod.visitLineNumber(17, label6);

            dummyMethod.visitVarInsn(ALOAD, 0);

            dummyMethod.visitVarInsn(ALOAD, 4);

            dummyMethod.visitMethodInsn(INVOKEVIRTUAL, "ic2/core/energy/Node", "setGrid", "(Lic2/core/energy/Grid;)V", false);

            dummyMethod.visitLabel(label4);

            dummyMethod.visitLineNumber(19, label4);

            dummyMethod.visitFrame(F_APPEND, 1, new Object[]{"ic2/core/energy/Grid"}, 0, null);

            dummyMethod.visitVarInsn(ALOAD, 4);

            dummyMethod.visitVarInsn(ALOAD, 2);

            dummyMethod.visitVarInsn(ALOAD, 3);

            dummyMethod.visitMethodInsn(INVOKEVIRTUAL, "ic2/core/energy/Grid", "add", "(Lic2/core/energy/Node;Ljava/util/Collection;)V", false);

            Label label7 = new Label();

            dummyMethod.visitLabel(label7);

            dummyMethod.visitLineNumber(20, label7);

            dummyMethod.visitInsn(RETURN);

            Label label8 = new Label();

            dummyMethod.visitLabel(label8);

            dummyMethod.visitLocalVariable("neighbor", "Lic2/core/energy/Node;", null, label0, label8, 0);

            dummyMethod.visitLocalVariable("local", "Lic2/core/energy/EnergyNetLocal;", null, label0, label8, 1);

            dummyMethod.visitLocalVariable("currentNode", "Lic2/core/energy/Node;", null, label0, label8, 2);

            dummyMethod.visitLocalVariable("neighbors", "Ljava/util/Collection;", "Ljava/util/Collection<Lic2/core/energy/Node;>;", label0, label8, 3);

            dummyMethod.visitLocalVariable("grid", "Lic2/core/energy/Grid;", null, label3, label8, 4);

            dummyMethod.visitMaxs(3, 5);

            dummyMethod.visitEnd();

            classNode.methods.add(dummyMethod);

            classNode.accept(classWriter);

            return classWriter.toByteArray();

        } else {

            return basicClass;

        }

    }



    private static boolean checkMethodNode(AbstractInsnNode oNode, int opcode, String owner, String name, String decs) {

        if (!(oNode instanceof MethodInsnNode)) {

            return false;

        }

        MethodInsnNode node = (MethodInsnNode) oNode;

        return opcode == node.getOpcode() && owner.equals(node.owner) && decs.equals(node.desc);

    }

}