org.eolang.opeo.decompilation.agents.StoreAgent Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of opeo-maven-plugin Show documentation
Bytecode decompiler to eolang and back
The newest version!
/*
 * The MIT License (MIT)
 *
 * Copyright (c) 2016-2023 Objectionary.com
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included
 * in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */
package org.eolang.opeo.decompilation.agents;

import org.eolang.opeo.ast.AstNode;
import org.eolang.opeo.ast.LocalVariable;
import org.eolang.opeo.ast.Typed;
import org.eolang.opeo.ast.VariableAssignment;
import org.eolang.opeo.decompilation.DecompilerState;
import org.eolang.opeo.decompilation.OperandStack;
import org.objectweb.asm.Opcodes;
import org.objectweb.asm.Type;

/**
 * Store instruction handler.
 * @since 0.1
 */
public final class StoreAgent implements DecompilationAgent {

    /**
     * Supported opcodes.
     */
    private static final Supported OPCODES = new Supported(
        Opcodes.ISTORE,
        Opcodes.LSTORE,
        Opcodes.FSTORE,
        Opcodes.DSTORE,
        Opcodes.ASTORE
    );

    @Override
    public boolean appropriate(final DecompilerState state) {
        return new OpcodesAgent(this).appropriate(state);
    }

    @Override
    public Supported supported() {
        return StoreAgent.OPCODES;
    }

    @Override
    public void handle(final DecompilerState state) {
        if (this.appropriate(state)) {
            final OperandStack stack = state.stack();
            final AstNode value = stack.pop();
            stack.push(
                new VariableAssignment(
                    (LocalVariable) state.variable(
                        (Integer) state.operand(0),
                        StoreAgent.infer(value, state.current().opcode())
                    ),
                    value
                )
            );
            state.popInstruction();
        } else {
            throw new IllegalAgentException(this, state);
        }
    }

    /**
     * Infer type of the variable.
     * @param value Value to infer type from.
     * @param opcode Opcode.
     * @return Inferred type.
     */
    private static Type infer(final AstNode value, final int opcode) {
        final Type result;
        if (value instanceof Typed) {
            result = ((Typed) value).type();
        } else {
            result = StoreAgent.type(opcode);
        }
        return result;
    }

    /**
     * Infer type from opcode.
     * @param opcode Opcode.
     * @return Inferred type.
     */
    private static Type type(final int opcode) {
        final Type result;
        switch (opcode) {
            case Opcodes.ISTORE:
                result = Type.INT_TYPE;
                break;
            case Opcodes.LSTORE:
                result = Type.LONG_TYPE;
                break;
            case Opcodes.FSTORE:
                result = Type.FLOAT_TYPE;
                break;
            case Opcodes.DSTORE:
                result = Type.DOUBLE_TYPE;
                break;
            case Opcodes.ASTORE:
                result = Type.getType(Object.class);
                break;
            default:
                throw new IllegalArgumentException(
                    String.format("Unsupported opcode: %d", opcode)
                );
        }
        return result;
    }
}