org.jruby.runtime.scope.SharedBindingDynamicScope Maven / Gradle / Ivy
package org.jruby.runtime.scope;
import org.jruby.RubyArray;
import org.jruby.runtime.Helpers;
import org.jruby.parser.StaticScope;
import org.jruby.runtime.DynamicScope;
import org.jruby.runtime.builtin.IRubyObject;
import org.jruby.ir.IRMethod;
/**
*/
public class SharedBindingDynamicScope extends DynamicScope {
// Our values holder (name of variables are kept in staticScope)
private IRubyObject[] variableValues;
private IRMethod irMethod;
public SharedBindingDynamicScope(StaticScope staticScope, IRMethod irMethod) {
super(staticScope);
this.irMethod = irMethod;
allocate();
}
private void allocate() {
if(variableValues == null) {
int size = irMethod.getLocalVariablesCount();
// System.out.println("Have " + size + " variables");
variableValues = new IRubyObject[size];
}
}
public DynamicScope cloneScope() {
return new SharedBindingDynamicScope(staticScope, irMethod);
}
public IRubyObject[] getValues() {
return variableValues;
}
/**
* Get value from current scope or one of its captured scopes.
*
* FIXME: block variables are not getting primed to nil so we need to null check those
* until we prime them properly. Also add assert back in.
*
* @param offset zero-indexed value that represents where variable lives
* @param depth how many captured scopes down this variable should be set
* @return the value here
*/
public IRubyObject getValue(int offset, int depth) {
if (depth > 0) {
return parent.getValue(offset, depth - 1);
}
assertGetValue(offset, depth);
// &foo are not getting set from somewhere...I want the following assert to be true though
//assert variableValues[offset] != null : "Getting unassigned: " + staticScope.getVariables()[offset];
return variableValues[offset];
}
/**
* Variation of getValue that checks for nulls, returning and setting the given value (presumably nil)
*/
public IRubyObject getValueOrNil(int offset, int depth, IRubyObject nil) {
if (depth > 0) {
return parent.getValueOrNil(offset, depth - 1, nil);
} else {
return getValueDepthZeroOrNil(offset, nil);
}
}
public IRubyObject getValueDepthZeroOrNil(int offset, IRubyObject nil) {
assertGetValueDepthZeroOrNil(offset);
// &foo are not getting set from somewhere...I want the following assert to be true though
//assert variableValues[offset] != null : "Getting unassigned: " + staticScope.getVariables()[offset];
IRubyObject value = variableValues[offset];
if (value == null) {
return setValueDepthZero(value, offset);
}
return value;
}
public IRubyObject getValueZeroDepthZeroOrNil(IRubyObject nil) {
assertGetValueZeroDepthZeroOrNil();
// &foo are not getting set from somewhere...I want the following assert to be true though
//assert variableValues[offset] != null : "Getting unassigned: " + staticScope.getVariables()[offset];
IRubyObject value = variableValues[0];
if (value == null) {
return setValueZeroDepthZero(value);
}
return value;
}
public IRubyObject getValueOneDepthZeroOrNil(IRubyObject nil) {
assertGetValueOneDepthZeroOrNil();
// &foo are not getting set from somewhere...I want the following assert to be true though
//assert variableValues[offset] != null : "Getting unassigned: " + staticScope.getVariables()[offset];
IRubyObject value = variableValues[1];
if (value == null) {
return setValueOneDepthZero(value);
}
return value;
}
public IRubyObject getValueTwoDepthZeroOrNil(IRubyObject nil) {
assertGetValueTwoDepthZeroOrNil();
// &foo are not getting set from somewhere...I want the following assert to be true though
//assert variableValues[offset] != null : "Getting unassigned: " + staticScope.getVariables()[offset];
IRubyObject value = variableValues[2];
if (value == null) {
return setValueTwoDepthZero(value);
}
return value;
}
public IRubyObject getValueThreeDepthZeroOrNil(IRubyObject nil) {
assertGetValueThreeDepthZeroOrNil();
// &foo are not getting set from somewhere...I want the following assert to be true though
//assert variableValues[offset] != null : "Getting unassigned: " + staticScope.getVariables()[offset];
IRubyObject value = variableValues[3];
if (value == null) {
return setValueThreeDepthZero(value);
}
return value;
}
/**
* Set value in current dynamic scope or one of its captured scopes.
*
* @param offset zero-indexed value that represents where variable lives
* @param value to set
* @param depth how many captured scopes down this variable should be set
*/
public IRubyObject setValue(int offset, IRubyObject value, int depth) {
if (depth > 0) {
assertParent();
return parent.setValue(offset, value, depth - 1);
} else {
assertSetValue(offset, value);
return setValueDepthZero(value, offset);
}
}
public IRubyObject setValueDepthZero(IRubyObject value, int offset) {
assertSetValueDepthZero(offset, value);
return variableValues[offset] = value;
}
public IRubyObject setValueZeroDepthZero(IRubyObject value) {
assertSetValueZeroDepthZero(value);
return variableValues[0] = value;
}
public IRubyObject setValueOneDepthZero(IRubyObject value) {
assertSetValueOneDepthZero(value);
return variableValues[1] = value;
}
public IRubyObject setValueTwoDepthZero(IRubyObject value) {
assertSetValueTwoDepthZero(value);
return variableValues[2] = value;
}
public IRubyObject setValueThreeDepthZero(IRubyObject value) {
assertSetValueThreeDepthZero(value);
return variableValues[3] = value;
}
/**
* Set all values which represent 'normal' parameters in a call list to this dynamic
* scope. Function calls bind to local scopes by assuming that the indexes or the
* arg list correspond to that of the local scope (plus 2 since $_ and $~ always take
* the first two slots). We pass in a second argument because we sometimes get more
* values than we are expecting. The rest get compacted by original caller into
* rest args.
*
* @param values up to size specified to be mapped as ordinary parm values
* @param size is the number of values to assign as ordinary parm values
*/
public void setArgValues(IRubyObject[] values, int size) {
System.arraycopy(values, 0, variableValues, 0, size);
}
@Override
public void setArgValues(IRubyObject arg0) {
variableValues[0] = arg0;
}
@Override
public void setArgValues(IRubyObject arg0, IRubyObject arg1) {
variableValues[0] = arg0;
variableValues[1] = arg1;
}
@Override
public void setArgValues(IRubyObject arg0, IRubyObject arg1, IRubyObject arg2) {
variableValues[0] = arg0;
variableValues[1] = arg1;
variableValues[2] = arg2;
}
public void setEndArgValues(IRubyObject[] values, int index, int size) {
System.arraycopy(values, values.length - size, variableValues, index, size);
}
/**
* Copy variable values back for ZSuper call.
*/
public IRubyObject[] getArgValues() {
// if we're not the "argument scope" for zsuper, try our parent
if (!staticScope.isArgumentScope()) {
return parent.getArgValues();
}
int totalArgs = staticScope.getRequiredArgs() + staticScope.getOptionalArgs();
// copy and splat arguments out of the scope to use for zsuper call
if (staticScope.getRestArg() < 0) {
// required and optional only
IRubyObject[] argValues = new IRubyObject[totalArgs];
System.arraycopy(variableValues, 0, argValues, 0, totalArgs);
return argValues;
} else {
// rest arg must be splatted
IRubyObject restArg = getValue(staticScope.getRestArg(), 0);
assert restArg != null;
// FIXME: not very efficient
RubyArray splattedArgs = Helpers.splatValue(restArg);
IRubyObject[] argValues = new IRubyObject[totalArgs + splattedArgs.size()];
System.arraycopy(variableValues, 0, argValues, 0, totalArgs);
System.arraycopy(splattedArgs.toJavaArray(), 0, argValues, totalArgs, splattedArgs.size());
return argValues;
}
}
/**
*
* Make a larger dynamic scope if the static scope grew.
*
* Eval's with bindings require us to possibly change the size of the dynamic scope if
* things like 'eval "b = 2", binding' happens.
*
*/
public void growIfNeeded() {
int dynamicSize = variableValues == null ? 0: variableValues.length;
if (staticScope.getNumberOfVariables() > dynamicSize) {
IRubyObject values[] = new IRubyObject[staticScope.getNumberOfVariables()];
if (dynamicSize > 0) {
System.arraycopy(variableValues, 0, values, 0, dynamicSize);
}
variableValues = values;
}
}
private void assertGetValue(int offset, int depth) {
IRubyObject[] values = variableValues;
assert values != null && offset < values.length : "No variables or index to big for getValue off: " + offset + ", Dep: " + depth + ", O: " + this;
}
private void assertGetValueDepthZeroOrNil(int offset) {
IRubyObject[] values = variableValues;
assert values != null && offset < values.length : "No variables or index too big for getValue off: " + offset + ", Dep: " + 0 + ", O: " + this;
}
private void assertGetValueZeroDepthZeroOrNil() {
IRubyObject[] values = variableValues;
assert values != null && 0 < values.length : "No variables or index to big for getValue off: " + 0 + ", Dep: " + 0 + ", O: " + this;
}
private void assertGetValueOneDepthZeroOrNil() {
IRubyObject[] values = variableValues;
assert values != null && 1 < values.length : "No variables or index to big for getValue off: " + 1 + ", Dep: " + 0 + ", O: " + this;
}
private void assertGetValueTwoDepthZeroOrNil() {
IRubyObject[] values = variableValues;
assert values != null && 3 < values.length : "No variables or index to big for getValue off: " + 3 + ", Dep: " + 0 + ", O: " + this;
}
private void assertGetValueThreeDepthZeroOrNil() {
IRubyObject[] values = variableValues;
assert values != null && 2 < values.length : "No variables or index to big for getValue off: " + 2 + ", Dep: " + 0 + ", O: " + this;
}
private void assertParent() {
assert parent != null : "If depth > 0, then parent should not ever be null";
}
private void assertSetValue(int offset, IRubyObject value) {
assert offset < variableValues.length : "Setting " + offset + " to " + value + ", O: " + this;
}
private void assertSetValueDepthZero(int offset, IRubyObject value) {
assert offset < variableValues.length : "Setting " + offset + " to " + value + ", O: " + this;
}
private void assertSetValueZeroDepthZero(IRubyObject value) {
assert 0 < variableValues.length : "Setting " + 0 + " to " + value + ", O: " + this;
}
private void assertSetValueOneDepthZero(IRubyObject value) {
assert 1 < variableValues.length : "Setting " + 1 + " to " + value + ", O: " + this;
}
private void assertSetValueThreeDepthZero(IRubyObject value) {
assert 3 < variableValues.length : "Setting " + 3 + " to " + value + ", O: " + this;
}
private void assertSetValueTwoDepthZero(IRubyObject value) {
assert 2 < variableValues.length : "Setting " + 2 + " to " + value + ", O: " + this;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy