• Home
• io.github.llamalad7
• mixinextras-neoforge
• 0.5.0-beta.3
• source code
• IincExpander.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.llamalad7.mixinextras.expression.impl.flow.expansion.IincExpander Maven / Gradle / Ivy

package com.llamalad7.mixinextras.expression.impl.flow.expansion;

import com.llamalad7.mixinextras.expression.impl.flow.postprocessing.FlowPostProcessor;
import com.llamalad7.mixinextras.expression.impl.flow.FlowValue;
import com.llamalad7.mixinextras.expression.impl.utils.ExpressionASMUtils;
import org.objectweb.asm.Opcodes;
import org.objectweb.asm.Type;
import org.objectweb.asm.tree.*;
import org.spongepowered.asm.mixin.injection.struct.InjectionNodes;
import org.spongepowered.asm.mixin.injection.struct.Target;

public class IincExpander extends InsnExpander {
    @Override
    public void process(FlowValue node, FlowPostProcessor.OutputSink sink) {
        if (node.getInsn().getOpcode() != Opcodes.IINC) {
            return;
        }
        IincInsnNode iinc = (IincInsnNode) node.getInsn();
        FlowValue load = new FlowValue(Type.INT_TYPE, new VarInsnNode(Opcodes.ILOAD, iinc.var));
        registerComponent(load, Component.LOAD, iinc);
        FlowValue cst = new FlowValue(Type.INT_TYPE, ExpressionASMUtils.pushInt(iinc.incr));
        registerComponent(cst, Component.CST, iinc);
        FlowValue add = new FlowValue(Type.INT_TYPE, new InsnNode(Opcodes.IADD), load, cst);
        registerComponent(add, Component.ADD, iinc);

        node.setInsn(new VarInsnNode(Opcodes.ISTORE, iinc.var));
        node.setParents(add);
        registerComponent(node, Component.STORE, iinc);

        sink.registerFlow(load, cst, add);
    }

    @Override
    public void expand(Target target, InjectionNodes.InjectionNode node, Expansion expansion) {
        IincInsnNode iinc = (IincInsnNode) node.getCurrentTarget();
        target.method.maxStack += 2;
        expandInsn(
                target, node,
                expansion.registerInsn(Component.LOAD, new VarInsnNode(Opcodes.ILOAD, iinc.var)),
                expansion.registerInsn(Component.CST, ExpressionASMUtils.pushInt(iinc.incr)),
                expansion.registerInsn(Component.ADD, new InsnNode(Opcodes.IADD)),
                expansion.registerInsn(Component.STORE, new VarInsnNode(Opcodes.ISTORE, iinc.var))
        );
    }

    private enum Component implements InsnComponent {
        LOAD, CST, ADD, STORE
    }
}