org.jruby.compiler.ir.instructions.GetArrayInstr Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of jruby-complete

There is a newer version: 9.4.7.0

package org.jruby.compiler.ir.instructions;

import java.util.Map;
import org.jruby.RubyArray;

import org.jruby.compiler.ir.Operation;
import org.jruby.compiler.ir.operands.Label;
import org.jruby.compiler.ir.operands.Operand;
import org.jruby.compiler.ir.operands.Variable;
import org.jruby.compiler.ir.representations.InlinerInfo;
import org.jruby.interpreter.InterpreterContext;
import org.jruby.runtime.builtin.IRubyObject;

// This is an internal ruby array generated during multiple assignment expressions.
// FIXME: Rename GetArrayInstr to ArrayArefInstr which would be used
// in later passes as well when compiler passes replace ruby-array []
// calls with inlined lookups
public class GetArrayInstr extends OneOperandInstr {
    public final int index;
    public final boolean all;  // If true, returns the rest of the array starting at the index

    public GetArrayInstr(Variable dest, Operand array, int index, boolean getRestOfArray) {
        super(Operation.GET_ARRAY, dest, array);
        this.index = index;
        all = getRestOfArray;
    }

    @Override
    public String toString() {
        return "\t" + result + " = " + argument + "[" + index + (all ? ":END" : "") + "] (GET_ARRAY)";
    }

    @Override
    public Operand simplifyAndGetResult(Map valueMap) {
        simplifyOperands(valueMap);
        Operand val = argument.getValue(valueMap);
        return val.fetchCompileTimeArrayElement(index, all);
    }

    @Override
    public Instr cloneForInlining(InlinerInfo ii) {
        return new GetArrayInstr(ii.getRenamedVariable(result), argument.cloneForInlining(ii), index, all);
    }

    @Override
    public Label interpret(InterpreterContext interp, IRubyObject self) {
        // ENEBO: Can I assume since IR figured this is an internal array it will be RubyArray like this?
        RubyArray array = (RubyArray) getArg().retrieve(interp);

        getResult().store(interp, array.entry(index));
        return null;
    }
}