org.jruby.runtime.invokedynamic.MathLinker Maven / Gradle / Ivy
/*
***** BEGIN LICENSE BLOCK *****
* Version: EPL 2.0/GPL 2.0/LGPL 2.1
*
* The contents of this file are subject to the Eclipse Public
* License Version 2.0 (the "License"); you may not use this file
* except in compliance with the License. You may obtain a copy of
* the License at http://www.eclipse.org/legal/epl-v20.html
*
* Software distributed under the License is distributed on an "AS
* IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
* implied. See the License for the specific language governing
* rights and limitations under the License.
*
* Alternatively, the contents of this file may be used under the terms of
* either of the GNU General Public License Version 2 or later (the "GPL"),
* or the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
* in which case the provisions of the GPL or the LGPL are applicable instead
* of those above. If you wish to allow use of your version of this file only
* under the terms of either the GPL or the LGPL, and not to allow others to
* use your version of this file under the terms of the EPL, indicate your
* decision by deleting the provisions above and replace them with the notice
* and other provisions required by the GPL or the LGPL. If you do not delete
* the provisions above, a recipient may use your version of this file under
* the terms of any one of the EPL, the GPL or the LGPL.
***** END LICENSE BLOCK *****/
package org.jruby.runtime.invokedynamic;
import java.lang.invoke.CallSite;
import java.lang.invoke.MethodHandle;
import java.lang.invoke.MethodType;
import java.lang.invoke.SwitchPoint;
import java.util.List;
import static java.lang.invoke.MethodHandles.*;
import static java.lang.invoke.MethodType.*;
import com.headius.invokebinder.Binder;
import org.jruby.RubyClass;
import org.jruby.RubyFixnum;
import org.jruby.RubyFloat;
import org.jruby.runtime.CallType;
import org.jruby.runtime.MethodIndex;
import org.jruby.runtime.ThreadContext;
import org.jruby.runtime.builtin.IRubyObject;
import org.jruby.runtime.callsite.CacheEntry;
import org.jruby.util.JavaNameMangler;
import org.jruby.util.StringSupport;
import org.jruby.util.cli.Options;
import org.jruby.util.log.Logger;
import org.jruby.util.log.LoggerFactory;
public class MathLinker {
private static final Logger LOG = LoggerFactory.getLogger(MathLinker.class);
static { // enable DEBUG output
if (Options.INVOKEDYNAMIC_LOG_BINDING.load()) LOG.setDebugEnable(true);
}
private static final boolean LOG_BINDING = LOG.isDebugEnabled();
public static final MethodHandle FIXNUM_TEST =
Binder
.from(methodType(boolean.class, IRubyObject.class))
.invokeStaticQuiet(lookup(), MathLinker.class, "fixnumTest");
public static final MethodHandle FIXNUM_OPERATOR_FAIL =
Binder
.from(methodType(IRubyObject.class, ThreadContext.class, IRubyObject.class, IRubyObject.class, JRubyCallSite.class, RubyFixnum.class))
.invokeStaticQuiet(lookup(), MathLinker.class, "fixnumOperatorFail");
public static final MethodHandle FIXNUM_BOOLEAN_FAIL =
Binder
.from(methodType(boolean.class, ThreadContext.class, IRubyObject.class, IRubyObject.class, JRubyCallSite.class, RubyFixnum.class))
.invokeStaticQuiet(lookup(), MathLinker.class, "fixnumBooleanFail");
public static final MethodHandle FLOAT_TEST =
Binder
.from(methodType(boolean.class, IRubyObject.class))
.invokeStaticQuiet(lookup(), MathLinker.class, "floatTest");
public static final MethodHandle FLOAT_OPERATOR_FAIL =
Binder
.from(methodType(IRubyObject.class, ThreadContext.class, IRubyObject.class, IRubyObject.class, JRubyCallSite.class, RubyFloat.class))
.invokeStaticQuiet(lookup(), MathLinker.class, "floatOperatorFail");
public static final MethodHandle FIXNUM_OPERATOR =
Binder
.from(methodType(IRubyObject.class, ThreadContext.class, IRubyObject.class, IRubyObject.class, JRubyCallSite.class, long.class))
.invokeStaticQuiet(lookup(), MathLinker.class, "fixnumOperator");
public static final MethodHandle FIXNUM_BOOLEAN = Binder.from(methodType(boolean.class, ThreadContext.class, IRubyObject.class, IRubyObject.class, JRubyCallSite.class, long.class)).invokeStaticQuiet(lookup(), MathLinker.class, "fixnumBoolean");
public static final MethodHandle FLOAT_OPERATOR = Binder.from(methodType(IRubyObject.class, ThreadContext.class, IRubyObject.class, IRubyObject.class, JRubyCallSite.class, double.class)).invokeStaticQuiet(lookup(), MathLinker.class, "floatOperator");
private static final CallType[] CALL_TYPES = CallType.values();
public static CallSite fixnumOperatorBootstrap(Lookup lookup, String name, MethodType type, long value, int callType, String file, int line) throws NoSuchMethodException, IllegalAccessException {
List names = StringSupport.split(name, ':');
String operator = JavaNameMangler.demangleMethodName(names.get(1));
JRubyCallSite site = new JRubyCallSite(lookup, type, CALL_TYPES[callType], file, line, operator, true);
MethodHandle target = FIXNUM_OPERATOR;
target = insertArguments(target, 3, site, value);
site.setTarget(target);
return site;
}
public static CallSite fixnumBooleanBootstrap(Lookup lookup, String name, MethodType type, long value, int callType, String file, int line) throws NoSuchMethodException, IllegalAccessException {
List names = StringSupport.split(name, ':');
String operator = JavaNameMangler.demangleMethodName(names.get(1));
JRubyCallSite site = new JRubyCallSite(lookup, type, CALL_TYPES[callType], file, line, operator, true);
MethodHandle target = FIXNUM_BOOLEAN;
target = insertArguments(target, 3, site, value);
site.setTarget(target);
return site;
}
public static CallSite floatOperatorBootstrap(Lookup lookup, String name, MethodType type, double value, int callType, String file, int line) throws NoSuchMethodException, IllegalAccessException {
List names = StringSupport.split(name, ':');
String operator = JavaNameMangler.demangleMethodName(names.get(1));
JRubyCallSite site = new JRubyCallSite(lookup, type, CALL_TYPES[callType], file, line, operator, true);
MethodHandle target = FLOAT_OPERATOR;
target = insertArguments(target, 3, site, value);
site.setTarget(target);
return site;
}
public static IRubyObject fixnumOperator(ThreadContext context, IRubyObject caller, IRubyObject self, JRubyCallSite site, long value) throws Throwable {
String operator = site.name();
String opMethod = MethodIndex.getFastFixnumOpsMethod(operator);
String name = "fixnum_" + opMethod;
MethodType type = methodType(IRubyObject.class, ThreadContext.class, IRubyObject.class, IRubyObject.class);
MethodHandle target = null;
if (operator.equals("+") || operator.equals("-")) {
if (value == 1) {
name += "_one";
target = lookup().findStatic(MathLinker.class, name, type);
} else if (value == 2) {
name += "_two";
target = lookup().findStatic(MathLinker.class, name, type);
}
}
if (target == null) {
type = type.insertParameterTypes(3, long.class);
target = lookup().findStatic(MathLinker.class, name, type);
target = insertArguments(target, 3, value);
}
MethodHandle fallback = FIXNUM_OPERATOR_FAIL;
fallback = insertArguments(fallback, 3, site, context.runtime.newFixnum(value));
MethodHandle test = FIXNUM_TEST;
test = permuteArguments(test, methodType(boolean.class, ThreadContext.class, IRubyObject.class, IRubyObject.class), new int[] {2});
if (LOG_BINDING) LOG.debug(name + "\tFixnum operation at site #" + site.siteID() + " (" + site.file() + ":" + site.line() + ") bound directly");
// confirm it's a Fixnum
target = guardWithTest(test, target, fallback);
// check Fixnum reopening
target = ((SwitchPoint)context.runtime.getFixnumInvalidator().getData())
.guardWithTest(target, fallback);
site.setTarget(target);
return (IRubyObject)target.invokeWithArguments(context, caller, self);
}
public static boolean fixnumBoolean(ThreadContext context, IRubyObject caller, IRubyObject self, JRubyCallSite site, long value) throws Throwable {
String operator = site.name();
String opMethod = MethodIndex.getFastFixnumOpsMethod(operator);
String name = "fixnum_boolean_" + opMethod;
MethodType type = methodType(boolean.class, ThreadContext.class, IRubyObject.class, IRubyObject.class);
MethodHandle target = null;
if (target == null) {
type = type.insertParameterTypes(3, long.class);
target = lookup().findStatic(MathLinker.class, name, type);
target = insertArguments(target, 3, value);
}
MethodHandle fallback = FIXNUM_BOOLEAN_FAIL;
fallback = insertArguments(fallback, 3, site, context.runtime.newFixnum(value));
MethodHandle test = FIXNUM_TEST;
test = permuteArguments(test, methodType(boolean.class, ThreadContext.class, IRubyObject.class, IRubyObject.class), new int[] {2});
if (LOG_BINDING) LOG.debug(name + "\tFixnum boolean operation at site #" + site.siteID() + " (" + site.file() + ":" + site.line() + ") bound directly");
// confirm it's a Fixnum
target = guardWithTest(test, target, fallback);
// check Fixnum reopening
target = ((SwitchPoint)context.runtime.getFixnumInvalidator().getData())
.guardWithTest(target, fallback);
site.setTarget(target);
return (Boolean)target.invokeWithArguments(context, caller, self);
}
public static IRubyObject fixnumOperatorFail(ThreadContext context, IRubyObject caller, IRubyObject self, JRubyCallSite site, RubyFixnum value) throws Throwable {
String operator = site.name();
RubyClass selfClass = InvokeDynamicSupport.pollAndGetClass(context, self);
CacheEntry entry = site.entry;
if (entry.typeOk(selfClass)) {
return entry.method.call(context, self, selfClass, operator, value);
} else {
entry = selfClass.searchWithCache(operator);
if (InvokeDynamicSupport.methodMissing(entry, site.callType(), operator, caller)) {
return InvokeDynamicSupport.callMethodMissing(entry, site.callType(), context, self, operator, value);
}
site.entry = entry;
return entry.method.call(context, self, selfClass, operator, value);
}
}
public static boolean fixnumBooleanFail(ThreadContext context, IRubyObject caller, IRubyObject self, JRubyCallSite site, RubyFixnum value) throws Throwable {
return fixnumOperatorFail(context, caller, self, site, value).isTrue();
}
public static IRubyObject fixnum_op_plus(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_plus(context, value);
}
public static IRubyObject fixnum_op_minus(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_minus(context, value);
}
public static IRubyObject fixnum_op_mul(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_mul(context, value);
}
public static IRubyObject fixnum_op_mod(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_mod(context, value);
}
public static IRubyObject fixnum_op_equal(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_equal(context, value);
}
public static IRubyObject fixnum_op_lt(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_lt(context, value);
}
public static IRubyObject fixnum_op_le(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_le(context, value);
}
public static IRubyObject fixnum_op_gt(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_gt(context, value);
}
public static IRubyObject fixnum_op_ge(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_ge(context, value);
}
public static boolean fixnum_boolean_op_equal(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_equal_boolean(context, value);
}
public static boolean fixnum_boolean_op_lt(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_lt_boolean(context, value);
}
public static boolean fixnum_boolean_op_le(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_le_boolean(context, value);
}
public static boolean fixnum_boolean_op_gt(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_gt_boolean(context, value);
}
public static boolean fixnum_boolean_op_ge(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_ge_boolean(context, value);
}
public static IRubyObject fixnum_op_cmp(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_cmp(context, value);
}
public static IRubyObject fixnum_op_and(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_and(context, value);
}
public static IRubyObject fixnum_op_or(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_or(context, value);
}
public static IRubyObject fixnum_op_xor(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_xor(context, value);
}
public static IRubyObject fixnum_op_rshift(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_rshift(context, value);
}
public static IRubyObject fixnum_op_lshift(ThreadContext context, IRubyObject caller, IRubyObject self, long value) throws Throwable {
return ((RubyFixnum)self).op_lshift(context, value);
}
public static IRubyObject fixnum_op_plus_one(ThreadContext context, IRubyObject caller, IRubyObject self) throws Throwable {
return ((RubyFixnum)self).op_plus_one(context);
}
public static IRubyObject fixnum_op_minus_one(ThreadContext context, IRubyObject caller, IRubyObject self) throws Throwable {
return ((RubyFixnum)self).op_minus_one(context);
}
public static IRubyObject fixnum_op_plus_two(ThreadContext context, IRubyObject caller, IRubyObject self) throws Throwable {
return ((RubyFixnum)self).op_plus_two(context);
}
public static IRubyObject fixnum_op_minus_two(ThreadContext context, IRubyObject caller, IRubyObject self) throws Throwable {
return ((RubyFixnum)self).op_minus_two(context);
}
public static IRubyObject floatOperator(ThreadContext context, IRubyObject caller, IRubyObject self, JRubyCallSite site, double value) throws Throwable {
String operator = site.name();
String opMethod = MethodIndex.getFastFloatOpsMethod(operator);
String name = "float_" + opMethod;
MethodType type = methodType(IRubyObject.class, ThreadContext.class, IRubyObject.class, IRubyObject.class);
MethodHandle target = null;
if (target == null) {
type = type.insertParameterTypes(3, double.class);
target = lookup().findStatic(MathLinker.class, name, type);
target = insertArguments(target, 3, value);
}
MethodHandle fallback = FLOAT_OPERATOR_FAIL;
fallback = insertArguments(fallback, 3, site, context.runtime.newFloat(value));
MethodHandle test = FLOAT_TEST;
test = permuteArguments(test, methodType(boolean.class, ThreadContext.class, IRubyObject.class, IRubyObject.class), new int[] {2});
if (LOG_BINDING) LOG.debug(name + "\tFloat operation at site #" + site.siteID() + " (" + site.file() + ":" + site.line() + ") bound directly");
site.setTarget(guardWithTest(test, target, fallback));
// confirm it's a Float
target = guardWithTest(test, target, fallback);
// check Float reopening
target = ((SwitchPoint)context.runtime.getFloatInvalidator().getData())
.guardWithTest(target, fallback);
site.setTarget(target);
return (IRubyObject)target.invokeWithArguments(context, caller, self);
}
public static boolean fixnumTest(IRubyObject self) {
return self instanceof RubyFixnum;
}
public static boolean floatTest(IRubyObject self) {
return self instanceof RubyFloat;
}
public static IRubyObject floatOperatorFail(ThreadContext context, IRubyObject caller, IRubyObject self, JRubyCallSite site, RubyFloat value) throws Throwable {
String operator = site.name();
RubyClass selfClass = InvokeDynamicSupport.pollAndGetClass(context, self);
CacheEntry entry = site.entry;
if (entry.typeOk(selfClass)) {
return entry.method.call(context, self, selfClass, operator, value);
} else {
entry = selfClass.searchWithCache(operator);
if (InvokeDynamicSupport.methodMissing(entry, site.callType(), operator, caller)) {
return InvokeDynamicSupport.callMethodMissing(entry, site.callType(), context, self, operator, value);
}
site.entry = entry;
return entry.method.call(context, self, selfClass, operator, value);
}
}
public static IRubyObject float_op_plus(ThreadContext context, IRubyObject caller, IRubyObject self, double value) throws Throwable {
return ((RubyFloat)self).op_plus(context, value);
}
public static IRubyObject float_op_minus(ThreadContext context, IRubyObject caller, IRubyObject self, double value) throws Throwable {
return ((RubyFloat)self).op_minus(context, value);
}
public static IRubyObject float_op_mul(ThreadContext context, IRubyObject caller, IRubyObject self, double value) throws Throwable {
return ((RubyFloat)self).op_mul(context, value);
}
public static IRubyObject float_op_equal(ThreadContext context, IRubyObject caller, IRubyObject self, double value) throws Throwable {
return ((RubyFloat)self).op_equal(context, value);
}
public static IRubyObject float_op_lt(ThreadContext context, IRubyObject caller, IRubyObject self, double value) throws Throwable {
return ((RubyFloat)self).op_lt(context, value);
}
public static IRubyObject float_op_le(ThreadContext context, IRubyObject caller, IRubyObject self, double value) throws Throwable {
return ((RubyFloat)self).op_le(context, value);
}
public static IRubyObject float_op_gt(ThreadContext context, IRubyObject caller, IRubyObject self, double value) throws Throwable {
return ((RubyFloat)self).op_gt(context, value);
}
public static IRubyObject float_op_ge(ThreadContext context, IRubyObject caller, IRubyObject self, double value) throws Throwable {
return ((RubyFloat)self).op_ge(context, value);
}
public static IRubyObject float_op_cmp(ThreadContext context, IRubyObject caller, IRubyObject self, double value) throws Throwable {
return ((RubyFloat)self).op_cmp(context, value);
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy