package org.jruby.ir.instructions;
import org.jruby.RubyModule;
import org.jruby.internal.runtime.methods.InterpretedIRMethod;
import org.jruby.ir.IRVisitor;
import org.jruby.ir.IRModuleBody;
import org.jruby.ir.Operation;
import org.jruby.ir.operands.Operand;
import org.jruby.ir.operands.Variable;
import org.jruby.ir.transformations.inlining.InlinerInfo;
import org.jruby.runtime.Block;
import org.jruby.runtime.DynamicScope;
import org.jruby.runtime.ThreadContext;
import org.jruby.runtime.Visibility;
import org.jruby.runtime.builtin.IRubyObject;
import java.util.Map;
public class DefineModuleInstr extends Instr implements ResultInstr {
private final IRModuleBody newIRModuleBody;
private Operand container;
private Variable result;
public DefineModuleInstr(Variable result, IRModuleBody newIRModuleBody, Operand container) {
super(Operation.DEF_MODULE);
assert result != null : "DefineModuleInstr result is null";
this.newIRModuleBody = newIRModuleBody;
this.container = container;
this.result = result;
}
public Operand[] getOperands() {
return new Operand[]{container};
}
public Variable getResult() {
return result;
}
public void updateResult(Variable v) {
this.result = v;
}
@Override
public void simplifyOperands(Map<Operand, Operand> valueMap, boolean force) {
container = container.getSimplifiedOperand(valueMap, force);
}
@Override
public String toString() {
return super.toString() + "(" + newIRModuleBody.getName() + ", " + container + ", " + newIRModuleBody.getFileName() + ")";
}
@Override
public Instr cloneForInlining(InlinerInfo ii) {
// SSS FIXME: So, do we clone the module body scope or not?
return new DefineModuleInstr(ii.getRenamedVariable(result), this.newIRModuleBody, container.cloneForInlining(ii));
}
@Override
public Object interpret(ThreadContext context, DynamicScope currDynScope, IRubyObject self, Object[] temp, Block block) {
Object rubyContainer = container.retrieve(context, self, currDynScope, temp);
if (!(rubyContainer instanceof RubyModule)) {
throw context.runtime.newTypeError("no outer class/module");
}
RubyModule newRubyModule = ((RubyModule) rubyContainer).defineOrGetModuleUnder(newIRModuleBody.getName());
newIRModuleBody.getStaticScope().setModule(newRubyModule);
return new InterpretedIRMethod(newIRModuleBody, Visibility.PUBLIC, newRubyModule);
}
@Override
public void visit(IRVisitor visitor) {
visitor.DefineModuleInstr(this);
}
public IRModuleBody getNewIRModuleBody() {
return newIRModuleBody;
}
public Operand getContainer() {
return container;
}
}