package org.jruby.ir.instructions;
import org.jruby.ir.IRVisitor;
import org.jruby.ir.Operation;
import org.jruby.ir.operands.Fixnum;
import org.jruby.ir.operands.Operand;
import org.jruby.ir.operands.Variable;
import org.jruby.ir.runtime.IRRuntimeHelpers;
import org.jruby.ir.transformations.inlining.CloneInfo;
import org.jruby.ir.transformations.inlining.InlineCloneInfo;
import org.jruby.ir.transformations.inlining.SimpleCloneInfo;
import org.jruby.runtime.builtin.IRubyObject;
/**
* This represents a required arg that shows up after optional/rest args
* in a method/block parameter list. This instruction gets to pick an argument
* based on how many arguments have already been accounted for by parameters
* present earlier in the list.
*/
public class ReceivePostReqdArgInstr extends ReceiveArgBase implements FixedArityInstr {
/** The method/block parameter list has these many required parameters before opt+rest args*/
public final int preReqdArgsCount;
/** The method/block parameter list has these many required parameters after opt+rest args*/
public final int postReqdArgsCount;
public ReceivePostReqdArgInstr(Variable result, int argIndex, int preReqdArgsCount, int postReqdArgsCount) {
super(Operation.RECV_POST_REQD_ARG, result, argIndex);
this.preReqdArgsCount = preReqdArgsCount;
this.postReqdArgsCount = postReqdArgsCount;
}
@Override
public Operand[] getOperands() {
return new Operand[] { new Fixnum(argIndex), new Fixnum(preReqdArgsCount), new Fixnum(postReqdArgsCount) };
}
@Override
public String toString() {
return (isDead() ? "[DEAD]" : "") + (hasUnusedResult() ? "[DEAD-RESULT]" : "") + getResult() + " = " + getOperation() + "(" + argIndex + ", " + preReqdArgsCount + ", " + postReqdArgsCount + ")";
}
@Override
public Instr clone(CloneInfo info) {
if (info instanceof SimpleCloneInfo) return new ReceivePostReqdArgInstr(info.getRenamedVariable(result), argIndex, preReqdArgsCount, postReqdArgsCount);
InlineCloneInfo ii = (InlineCloneInfo) info;
if (ii.canMapArgsStatically()) {
int n = ii.getArgsCount();
int remaining = n - preReqdArgsCount;
Operand argVal;
if (remaining <= argIndex) {
// SSS: FIXME: Argh!
argVal = ii.getHostScope().getManager().getNil();
} else {
argVal = (remaining > postReqdArgsCount) ? ii.getArg(n - postReqdArgsCount + argIndex) : ii.getArg(preReqdArgsCount + argIndex);
}
return new CopyInstr(ii.getRenamedVariable(result), argVal);
}
return new ReqdArgMultipleAsgnInstr(ii.getRenamedVariable(result), ii.getArgs(), preReqdArgsCount, postReqdArgsCount, argIndex);
}
public IRubyObject receivePostReqdArg(IRubyObject[] args, boolean acceptsKeywordArgument) {
return IRRuntimeHelpers.receivePostReqdArg(args, preReqdArgsCount, postReqdArgsCount, argIndex, acceptsKeywordArgument);
}
@Override
public void visit(IRVisitor visitor) {
visitor.ReceivePostReqdArgInstr(this);
}
}