1381 lines
62 KiB
Java
1381 lines
62 KiB
Java
/*
|
|
* Copyright (c) 2015, 2016, Oracle and/or its affiliates. All rights reserved.
|
|
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
|
*
|
|
* This code is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU General Public License version 2 only, as
|
|
* published by the Free Software Foundation.
|
|
*
|
|
* This code is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
* version 2 for more details (a copy is included in the LICENSE file that
|
|
* accompanied this code).
|
|
*
|
|
* You should have received a copy of the GNU General Public License version
|
|
* 2 along with this work; if not, write to the Free Software Foundation,
|
|
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*
|
|
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
|
|
* or visit www.oracle.com if you need additional information or have any
|
|
* questions.
|
|
*/
|
|
|
|
/* @test
|
|
* @bug 8139885
|
|
* @bug 8150635
|
|
* @bug 8150956
|
|
* @bug 8150957
|
|
* @bug 8151179
|
|
* @bug 8152667
|
|
* @bug 8153637
|
|
* @bug 8154751
|
|
* @bug 8154754
|
|
* @bug 8167974
|
|
* @run testng/othervm -ea -esa test.java.lang.invoke.LoopCombinatorTest
|
|
*/
|
|
|
|
package test.java.lang.invoke;
|
|
|
|
import java.lang.invoke.MethodHandle;
|
|
import java.lang.invoke.MethodHandles;
|
|
import java.lang.invoke.MethodHandles.Lookup;
|
|
import java.lang.invoke.MethodType;
|
|
import java.util.*;
|
|
|
|
import static java.lang.invoke.MethodType.methodType;
|
|
|
|
import static org.testng.AssertJUnit.*;
|
|
|
|
import org.testng.annotations.*;
|
|
|
|
/**
|
|
* Tests for the loop combinators introduced in JEP 274.
|
|
*/
|
|
public class LoopCombinatorTest {
|
|
|
|
static final Lookup LOOKUP = MethodHandles.lookup();
|
|
|
|
@Test
|
|
public static void testLoopFac() throws Throwable {
|
|
MethodHandle[] counterClause = new MethodHandle[]{Fac.MH_zero, Fac.MH_inc};
|
|
MethodHandle[] accumulatorClause = new MethodHandle[]{Fac.MH_one, Fac.MH_mult, Fac.MH_pred, Fac.MH_fin};
|
|
MethodHandle loop = MethodHandles.loop(counterClause, accumulatorClause);
|
|
assertEquals(Fac.MT_fac, loop.type());
|
|
assertEquals(120, loop.invoke(5));
|
|
}
|
|
|
|
@Test
|
|
public static void testLoopFacNullInit() throws Throwable {
|
|
// null initializer for counter, should initialize to 0
|
|
MethodHandle[] counterClause = new MethodHandle[]{null, Fac.MH_inc};
|
|
MethodHandle[] accumulatorClause = new MethodHandle[]{Fac.MH_one, Fac.MH_mult, Fac.MH_pred, Fac.MH_fin};
|
|
MethodHandle loop = MethodHandles.loop(counterClause, accumulatorClause);
|
|
assertEquals(Fac.MT_fac, loop.type());
|
|
assertEquals(120, loop.invoke(5));
|
|
}
|
|
|
|
@Test
|
|
public static void testLoopNullInit() throws Throwable {
|
|
// null initializer for counter, should initialize to 0, one-clause loop
|
|
MethodHandle[] counterClause = new MethodHandle[]{null, Loop.MH_inc, Loop.MH_pred, Loop.MH_fin};
|
|
MethodHandle loop = MethodHandles.loop(counterClause);
|
|
assertEquals(Loop.MT_loop, loop.type());
|
|
assertEquals(10, loop.invoke(10));
|
|
}
|
|
|
|
@Test
|
|
public static void testLoopVoid1() throws Throwable {
|
|
// construct a post-checked loop that only does one iteration and has a void body and void local state
|
|
MethodHandle loop = MethodHandles.loop(new MethodHandle[]{Empty.MH_f, Empty.MH_f, Empty.MH_pred, null});
|
|
assertEquals(MethodType.methodType(void.class), loop.type());
|
|
loop.invoke();
|
|
}
|
|
|
|
@Test
|
|
public static void testLoopVoid2() throws Throwable {
|
|
// construct a post-checked loop that only does one iteration and has a void body and void local state,
|
|
// initialized implicitly from the step type
|
|
MethodHandle loop = MethodHandles.loop(new MethodHandle[]{null, Empty.MH_f, Empty.MH_pred, null});
|
|
assertEquals(MethodType.methodType(void.class), loop.type());
|
|
loop.invoke();
|
|
}
|
|
|
|
@Test
|
|
public static void testLoopVoid3() throws Throwable {
|
|
// construct a post-checked loop that only does one iteration and has a void body and void local state,
|
|
// and that has a void finalizer
|
|
MethodHandle loop = MethodHandles.loop(new MethodHandle[]{null, Empty.MH_f, Empty.MH_pred, Empty.MH_f});
|
|
assertEquals(MethodType.methodType(void.class), loop.type());
|
|
loop.invoke();
|
|
}
|
|
|
|
@Test
|
|
public static void testLoopFacWithVoidState() throws Throwable {
|
|
// like testLoopFac, but with additional void state that outputs a dot
|
|
MethodHandle[] counterClause = new MethodHandle[]{Fac.MH_zero, Fac.MH_inc};
|
|
MethodHandle[] accumulatorClause = new MethodHandle[]{Fac.MH_one, Fac.MH_mult, Fac.MH_pred, Fac.MH_fin};
|
|
MethodHandle[] dotClause = new MethodHandle[]{null, Fac.MH_dot};
|
|
MethodHandle loop = MethodHandles.loop(counterClause, accumulatorClause, dotClause);
|
|
assertEquals(Fac.MT_fac, loop.type());
|
|
assertEquals(120, loop.invoke(5));
|
|
}
|
|
|
|
@Test
|
|
public static void testLoopVoidInt() throws Throwable {
|
|
// construct a post-checked loop that only does one iteration and has a void body and void local state,
|
|
// and that returns a constant
|
|
MethodHandle loop = MethodHandles.loop(new MethodHandle[]{null, Empty.MH_f, Empty.MH_pred, Empty.MH_c});
|
|
assertEquals(MethodType.methodType(int.class), loop.type());
|
|
assertEquals(23, loop.invoke());
|
|
}
|
|
|
|
@Test
|
|
public static void testLoopWithVirtuals() throws Throwable {
|
|
// construct a loop (to calculate factorial) that uses a mix of static and virtual methods
|
|
MethodHandle[] counterClause = new MethodHandle[]{null, LoopWithVirtuals.permute(LoopWithVirtuals.MH_inc)};
|
|
MethodHandle[] accumulatorClause = new MethodHandle[]{
|
|
// init function must indicate the loop arguments (there is no other means to determine them)
|
|
MethodHandles.dropArguments(LoopWithVirtuals.MH_one, 0, LoopWithVirtuals.class),
|
|
LoopWithVirtuals.permute(LoopWithVirtuals.MH_mult),
|
|
LoopWithVirtuals.permute(LoopWithVirtuals.MH_pred),
|
|
LoopWithVirtuals.permute(LoopWithVirtuals.MH_fin)
|
|
};
|
|
MethodHandle loop = MethodHandles.loop(counterClause, accumulatorClause);
|
|
assertEquals(LoopWithVirtuals.MT_loop, loop.type());
|
|
assertEquals(120, loop.invoke(new LoopWithVirtuals(), 5));
|
|
}
|
|
|
|
@Test
|
|
public static void testLoopOmitPred() throws Throwable {
|
|
// construct a loop to calculate factorial that omits a predicate
|
|
MethodHandle[] counterClause = new MethodHandle[]{null, Fac.MH_inc, null, Fac.MH_fin};
|
|
MethodHandle[] accumulatorClause = new MethodHandle[]{Fac.MH_one, Fac.MH_mult, Fac.MH_pred, Fac.MH_fin};
|
|
MethodHandle loop = MethodHandles.loop(counterClause, accumulatorClause);
|
|
assertEquals(Fac.MT_fac, loop.type());
|
|
assertEquals(120, loop.invoke(5));
|
|
}
|
|
|
|
@DataProvider
|
|
static Object[][] negativeTestData() {
|
|
MethodHandle i0 = MethodHandles.constant(int.class, 0);
|
|
MethodHandle ii = MethodHandles.dropArguments(i0, 0, int.class, int.class);
|
|
MethodHandle id = MethodHandles.dropArguments(i0, 0, int.class, double.class);
|
|
MethodHandle i3 = MethodHandles.dropArguments(i0, 0, int.class, int.class, int.class);
|
|
List<MethodHandle> inits = Arrays.asList(ii, id, i3);
|
|
List<Class<?>> ints3 = Arrays.asList(int.class, int.class, int.class);
|
|
List<Class<?>> ints4 = Arrays.asList(int.class, int.class, int.class, int.class);
|
|
List<MethodHandle> finis = Arrays.asList(Fac.MH_fin, Fac.MH_inc, Counted.MH_step);
|
|
List<MethodHandle> preds1 = Arrays.asList(null, null, null);
|
|
List<MethodHandle> preds2 = Arrays.asList(null, Fac.MH_fin, null);
|
|
MethodHandle eek = MethodHandles.dropArguments(i0, 0, int.class, int.class, double.class);
|
|
List<MethodHandle> nesteps = Arrays.asList(Fac.MH_inc, eek, Fac.MH_dot);
|
|
List<MethodHandle> nepreds = Arrays.asList(null, Fac.MH_pred, null);
|
|
List<MethodHandle> nefinis = Arrays.asList(null, Fac.MH_fin, null);
|
|
List<MethodHandle> lvsteps = Arrays.asList(LoopWithVirtuals.MH_inc, LoopWithVirtuals.MH_mult);
|
|
List<MethodHandle> lvpreds = Arrays.asList(null, LoopWithVirtuals.MH_pred);
|
|
List<MethodHandle> lvfinis = Arrays.asList(null, LoopWithVirtuals.MH_fin);
|
|
return new Object[][] {
|
|
{null, "null or no clauses passed"},
|
|
{new MethodHandle[][]{}, "null or no clauses passed"},
|
|
{new MethodHandle[][]{{null, Fac.MH_inc}, {Fac.MH_one, null, Fac.MH_mult, Fac.MH_pred, Fac.MH_fin}},
|
|
"All loop clauses must be represented as MethodHandle arrays with at most 4 elements."},
|
|
{new MethodHandle[][]{{null, Fac.MH_inc}, null}, "null clauses are not allowed"},
|
|
{new MethodHandle[][]{{Fac.MH_zero, Fac.MH_dot}},
|
|
"clause 0: init and step return types must match: int != void"},
|
|
{new MethodHandle[][]{{ii}, {id}, {i3}},
|
|
"found non-effectively identical init parameter type lists: " + inits +
|
|
" (common suffix: " + ints3 + ")"},
|
|
{new MethodHandle[][]{{null, Fac.MH_inc, null, Fac.MH_fin}, {null, Fac.MH_inc, null, Fac.MH_inc},
|
|
{null, Counted.MH_start, null, Counted.MH_step}},
|
|
"found non-identical finalizer return types: " + finis + " (return type: int)"},
|
|
{new MethodHandle[][]{{Fac.MH_zero, Fac.MH_inc}, {Fac.MH_one, Fac.MH_mult, null, Fac.MH_fin},
|
|
{null, Fac.MH_dot}}, "no predicate found: " + preds1},
|
|
{new MethodHandle[][]{{Fac.MH_zero, Fac.MH_inc}, {Fac.MH_one, Fac.MH_mult, Fac.MH_fin, Fac.MH_fin},
|
|
{null, Fac.MH_dot}}, "predicates must have boolean return type: " + preds2},
|
|
{new MethodHandle[][]{{Fac.MH_zero, Fac.MH_inc}, {Fac.MH_one, eek, Fac.MH_pred, Fac.MH_fin},
|
|
{null, Fac.MH_dot}},
|
|
"found non-effectively identical parameter type lists:\nstep: " + nesteps +
|
|
"\npred: " + nepreds + "\nfini: " + nefinis + " (common parameter sequence: " + ints3 + ")"},
|
|
{new MethodHandle[][]{{null, LoopWithVirtuals.MH_inc},
|
|
{LoopWithVirtuals.MH_one, LoopWithVirtuals.MH_mult, LoopWithVirtuals.MH_pred, LoopWithVirtuals.MH_fin}},
|
|
"found non-effectively identical parameter type lists:\nstep: " + lvsteps +
|
|
"\npred: " + lvpreds + "\nfini: " + lvfinis + " (common parameter sequence: " + ints4 + ")"}
|
|
};
|
|
}
|
|
|
|
static final MethodHandle MH_loop;
|
|
|
|
static {
|
|
try {
|
|
MH_loop = LOOKUP.findStatic(MethodHandles.class, "loop", methodType(MethodHandle.class, MethodHandle[][].class));
|
|
} catch (NoSuchMethodException | IllegalAccessException e) {
|
|
throw new ExceptionInInitializerError(e);
|
|
}
|
|
}
|
|
|
|
@Test(dataProvider = "negativeTestData")
|
|
public static void testLoopNegative(MethodHandle[][] clauses, String expectedMessage) throws Throwable {
|
|
boolean caught = false;
|
|
try {
|
|
MH_loop.invokeWithArguments((Object[]) clauses);
|
|
} catch (IllegalArgumentException iae) {
|
|
assertEquals(expectedMessage, iae.getMessage());
|
|
caught = true;
|
|
}
|
|
assertTrue(caught);
|
|
}
|
|
|
|
@Test(dataProvider = "whileLoopTestData")
|
|
public static void testWhileLoop(MethodHandle MH_zero,
|
|
MethodHandle MH_pred,
|
|
MethodHandle MH_step,
|
|
String messageOrNull) throws Throwable {
|
|
// int i = 0; while (i < limit) { ++i; } return i; => limit
|
|
try {
|
|
MethodHandle loop = MethodHandles.whileLoop(MH_zero, MH_pred, MH_step);
|
|
assert messageOrNull == null;
|
|
if (MH_step.type().equals(While.MH_step.type()))
|
|
assertEquals(While.MT_while, loop.type());
|
|
assertEquals(MH_step.type().dropParameterTypes(0, 1), loop.type());
|
|
while (loop.type().parameterCount() > 1) loop = snip(loop);
|
|
assertEquals(23, loop.invoke(23));
|
|
} catch (IllegalArgumentException iae) {
|
|
assert messageOrNull != null;
|
|
assertEqualsFIXME(messageOrNull, iae.getMessage());
|
|
}
|
|
}
|
|
|
|
static void assertEqualsFIXME(String expect, String actual) {
|
|
if (!expect.equals(actual)) {
|
|
// just issue a warning
|
|
System.out.println("*** "+actual+"\n != "+expect);
|
|
}
|
|
}
|
|
|
|
@DataProvider
|
|
static Object[][] whileLoopTestData() {
|
|
MethodHandle
|
|
zeroI = While.MH_zero,
|
|
zeroX = snip(zeroI),
|
|
zeroIB = slap(zeroI, byte.class),
|
|
predII = While.MH_pred,
|
|
predIX = snip(predII),
|
|
predIIB = slap(predII, byte.class),
|
|
stepII = While.MH_step,
|
|
stepIX = snip(stepII),
|
|
stepIIB = slap(stepII, byte.class)
|
|
;
|
|
return new Object[][] {
|
|
// normal while loop clauses, perhaps with effectively-identical reductions
|
|
{zeroI, predII, stepII, null},
|
|
{zeroX, predII, stepII, null},
|
|
{null, predII, stepII, null},
|
|
// expanded while loop clauses
|
|
{zeroIB, predIIB, stepIIB, null},
|
|
{zeroI, predIIB, stepIIB, null},
|
|
{null, predIIB, stepIIB, null},
|
|
{zeroIB, predII, stepIIB, null},
|
|
{zeroX, predII, stepIIB, null},
|
|
{null, predII, stepIIB, null},
|
|
// short step clauses cause errors
|
|
{zeroI, predII, stepIX, "loop predicate must match: (int,int)boolean != (int)boolean"},
|
|
{zeroIB, predIX, stepIX, "loop initializer must match: (int,byte)int != ()int"},
|
|
// bad body type
|
|
{zeroI, predII, tweak(stepII, -1, char.class), "body function must match: (int,int)char != (char,int,int)char"},
|
|
{zeroI, predII, tweak(stepII, 0, char.class), "body function must match: (char,int)int != (int,char,int)int"},
|
|
// bad pred type
|
|
{zeroI, tweak(predII, -1, char.class), stepII, "loop predicate must match: (int,int)char != (int,int)boolean"},
|
|
{zeroI, tweak(predII, 0, char.class), stepII, "loop predicate must match: (char,int)boolean != (int,int)boolean"},
|
|
// bad init type
|
|
{tweak(zeroI, -1, char.class), predII, stepII, "loop initializer must match: (int)char != (int)int"},
|
|
{tweak(zeroI, 0, char.class), predII, stepII, "loop initializer must match: (char)int != (int)int"},
|
|
};
|
|
}
|
|
|
|
// tweak the type of an MH
|
|
static MethodHandle tweak(MethodHandle mh, int argPos, Class<?> type) {
|
|
MethodType mt = mh.type();
|
|
if (argPos == -1)
|
|
mt = mt.changeReturnType(type);
|
|
else
|
|
mt = mt.changeParameterType(argPos, type);
|
|
return MethodHandles.explicitCastArguments(mh, mt);
|
|
}
|
|
// snip off an MH argument, hard-wiring to zero
|
|
static MethodHandle snip(MethodHandle mh, int argPos) {
|
|
if (argPos < 0) return null; // special case for optional args
|
|
Class<?> argType = mh.type().parameterType(argPos);
|
|
Object zero;
|
|
try {
|
|
zero = MethodHandles.zero(argType).invoke();
|
|
} catch (Throwable ex) {
|
|
throw new AssertionError(ex);
|
|
}
|
|
return MethodHandles.insertArguments(mh, argPos, zero);
|
|
}
|
|
static MethodHandle snip(MethodHandle mh) {
|
|
return snip(mh, mh.type().parameterCount()-1);
|
|
}
|
|
// slap on an extra type on the end of the MH
|
|
static MethodHandle slap(MethodHandle mh, Class<?> addType) {
|
|
return MethodHandles.dropArguments(mh, mh.type().parameterCount(), addType);
|
|
}
|
|
|
|
@Test
|
|
public static void testWhileLoopNoIteration() throws Throwable {
|
|
// a while loop that never executes its body because the predicate evaluates to false immediately
|
|
MethodHandle loop = MethodHandles.whileLoop(While.MH_initString, While.MH_predString, While.MH_stepString);
|
|
assertEquals(While.MT_string, loop.type());
|
|
assertEquals("a", loop.invoke());
|
|
}
|
|
|
|
@Test(dataProvider = "whileLoopTestData")
|
|
public static void testDoWhileLoop(MethodHandle MH_zero,
|
|
MethodHandle MH_pred,
|
|
MethodHandle MH_step,
|
|
String messageOrNull) throws Throwable {
|
|
// int i = 0; do { ++i; } while (i < limit); return i; => limit
|
|
try {
|
|
MethodHandle loop = MethodHandles.doWhileLoop(MH_zero, MH_step, MH_pred);
|
|
assert messageOrNull == null;
|
|
if (MH_step.type().equals(While.MH_step.type()))
|
|
assertEquals(While.MT_while, loop.type());
|
|
assertEquals(MH_step.type().dropParameterTypes(0, 1), loop.type());
|
|
while (loop.type().parameterCount() > 1) loop = snip(loop);
|
|
assertEquals(23, loop.invoke(23));
|
|
} catch (IllegalArgumentException iae) {
|
|
assert messageOrNull != null;
|
|
if (!messageOrNull.equals(iae.getMessage())) {
|
|
// just issue a warning
|
|
System.out.println("*** "+messageOrNull+"\n != "+iae.getMessage());
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test
|
|
public static void testDoWhileBadInit() throws Throwable {
|
|
boolean caught = false;
|
|
try {
|
|
While w = new While();
|
|
MethodHandle loop = MethodHandles.doWhileLoop(MethodHandles.empty(methodType(char.class)),
|
|
While.MH_voidBody.bindTo(w),
|
|
While.MH_voidPred.bindTo(w));
|
|
} catch (IllegalArgumentException iae) {
|
|
assertEquals("loop initializer must match: ()char != (int)void", iae.getMessage());
|
|
caught = true;
|
|
}
|
|
assertTrue(caught);
|
|
}
|
|
|
|
@Test
|
|
public static void testWhileZip() throws Throwable {
|
|
MethodHandle loop = MethodHandles.doWhileLoop(While.MH_zipInitZip, While.MH_zipStep, While.MH_zipPred);
|
|
assertEquals(While.MT_zip, loop.type());
|
|
List<String> a = Arrays.asList("a", "b", "c", "d");
|
|
List<String> b = Arrays.asList("e", "f", "g", "h");
|
|
List<String> zipped = Arrays.asList("a", "e", "b", "f", "c", "g", "d", "h");
|
|
assertEquals(zipped, (List<String>) loop.invoke(a.iterator(), b.iterator()));
|
|
}
|
|
|
|
@Test
|
|
public static void testWhileBadInit() throws Throwable {
|
|
boolean caught = false;
|
|
try {
|
|
While w = new While();
|
|
MethodHandle loop = MethodHandles.whileLoop(MethodHandles.empty(methodType(void.class, char.class)),
|
|
While.MH_voidPred.bindTo(w),
|
|
While.MH_voidBody.bindTo(w));
|
|
} catch (IllegalArgumentException iae) {
|
|
assertEquals("loop initializer must match: (char)void != (int)void", iae.getMessage());
|
|
caught = true;
|
|
}
|
|
assertTrue(caught);
|
|
}
|
|
|
|
@Test
|
|
public static void testWhileVoidInit() throws Throwable {
|
|
While w = new While();
|
|
int v = 5;
|
|
MethodHandle loop = MethodHandles.whileLoop(While.MH_voidInit.bindTo(w), While.MH_voidPred.bindTo(w),
|
|
While.MH_voidBody.bindTo(w));
|
|
assertEquals(While.MT_void, loop.type());
|
|
loop.invoke(v);
|
|
assertEquals(v, w.i);
|
|
}
|
|
|
|
@Test
|
|
public static void testDoWhileVoidInit() throws Throwable {
|
|
While w = new While();
|
|
int v = 5;
|
|
MethodHandle loop = MethodHandles.doWhileLoop(While.MH_voidInit.bindTo(w), While.MH_voidBody.bindTo(w),
|
|
While.MH_voidPred.bindTo(w));
|
|
assertEquals(While.MT_void, loop.type());
|
|
loop.invoke(v);
|
|
assertEquals(v, w.i);
|
|
}
|
|
|
|
@DataProvider
|
|
static Object[][] nullArgs() {
|
|
MethodHandle c = MethodHandles.constant(int.class, 1);
|
|
return new Object[][]{{null, c}, {c, null}};
|
|
}
|
|
|
|
@Test(dataProvider = "nullArgs", expectedExceptions = NullPointerException.class)
|
|
public static void testWhileNullArgs(MethodHandle pred, MethodHandle body) {
|
|
MethodHandles.whileLoop(null, pred, body);
|
|
}
|
|
|
|
@Test(dataProvider = "nullArgs", expectedExceptions = NullPointerException.class)
|
|
public static void testDoWhileNullArgs(MethodHandle body, MethodHandle pred) {
|
|
MethodHandles.whileLoop(null, body, pred);
|
|
}
|
|
|
|
@Test
|
|
public static void testCountedLoop() throws Throwable {
|
|
// String s = "Lambdaman!"; for (int i = 0; i < 13; ++i) { s = "na " + s; } return s; => a variation on a well known theme
|
|
MethodHandle fit13 = MethodHandles.dropArguments(MethodHandles.constant(int.class, 13), 0, String.class);
|
|
MethodHandle loop = MethodHandles.countedLoop(fit13, Counted.MH_start, Counted.MH_step);
|
|
assertEquals(Counted.MT_counted, loop.type());
|
|
assertEquals("na na na na na na na na na na na na na Lambdaman!", loop.invoke("Lambdaman!"));
|
|
}
|
|
|
|
@Test
|
|
public static void testCountedLoopVoidInit() throws Throwable {
|
|
MethodHandle fit5 = MethodHandles.constant(int.class, 5);
|
|
for (int i = 0; i < 8; i++) {
|
|
MethodHandle zero = MethodHandles.zero(void.class);
|
|
MethodHandle init = fit5;
|
|
MethodHandle body = Counted.MH_printHello;
|
|
boolean useNull = (i & 1) != 0, addInitArg = (i & 2) != 0, addBodyArg = (i & 4) != 0;
|
|
if (useNull) zero = null;
|
|
if (addInitArg) init = MethodHandles.dropArguments(init, 0, int.class);
|
|
if (addBodyArg) body = MethodHandles.dropArguments(body, 1, int.class);
|
|
System.out.println("testCountedLoopVoidInit i="+i+" : "+Arrays.asList(init, zero, body));
|
|
MethodHandle loop = MethodHandles.countedLoop(init, zero, body);
|
|
MethodType expectedType = Counted.MT_countedPrinting;
|
|
if (addInitArg || addBodyArg)
|
|
expectedType = expectedType.insertParameterTypes(0, int.class);
|
|
assertEquals(expectedType, loop.type());
|
|
if (addInitArg || addBodyArg)
|
|
loop.invoke(99);
|
|
else
|
|
loop.invoke();
|
|
}
|
|
}
|
|
|
|
@Test
|
|
public static void testCountedArrayLoop() throws Throwable {
|
|
// int[] a = new int[]{0}; for (int i = 0; i < 13; ++i) { ++a[0]; } => a[0] == 13
|
|
MethodHandle fit13 = MethodHandles.dropArguments(MethodHandles.constant(int.class, 13), 0, int[].class);
|
|
MethodHandle loop = MethodHandles.countedLoop(fit13, null, Counted.MH_stepUpdateArray);
|
|
assertEquals(Counted.MT_arrayCounted, loop.type());
|
|
int[] a = new int[]{0};
|
|
loop.invoke(a);
|
|
assertEquals(13, a[0]);
|
|
}
|
|
|
|
@Test
|
|
public static void testCountedPrintingLoop() throws Throwable {
|
|
MethodHandle fit5 = MethodHandles.constant(int.class, 5);
|
|
MethodHandle loop = MethodHandles.countedLoop(fit5, null, Counted.MH_printHello);
|
|
assertEquals(Counted.MT_countedPrinting, loop.type());
|
|
loop.invoke();
|
|
}
|
|
|
|
@Test(expectedExceptions = NullPointerException.class)
|
|
public static void testCountedLoopNullBody() throws Throwable {
|
|
MethodHandle h5 = MethodHandles.constant(int.class, 5);
|
|
MethodHandle h13 = MethodHandles.constant(int.class, 13);
|
|
MethodHandle loop = MethodHandles.countedLoop(h5, h13, null);
|
|
assertEquals(methodType(int.class), loop.type());
|
|
assertEquals(13, loop.invoke());
|
|
}
|
|
|
|
@Test(expectedExceptions = NullPointerException.class)
|
|
public static void testCountedLoopNullIterations() throws Throwable {
|
|
MethodHandle loop = MethodHandles.countedLoop(null, null, null);
|
|
assertEquals(methodType(void.class), loop.type());
|
|
loop.invoke();
|
|
}
|
|
|
|
@Test(expectedExceptions = NullPointerException.class)
|
|
public static void testCountedLoopNullInitAndBody() throws Throwable {
|
|
MethodHandle loop = MethodHandles.countedLoop(MethodHandles.constant(int.class, 5), null, null);
|
|
assertEquals(methodType(void.class), loop.type());
|
|
loop.invoke();
|
|
}
|
|
|
|
@DataProvider
|
|
static Object[][] countedLoopBodyParameters() {
|
|
Class<?> V = String.class, I = int.class, A = List.class;
|
|
// return types are of these forms:
|
|
// {count = int(A...), init = V(A...), body = V(V, I, A...)}
|
|
return new Object[][] {
|
|
// body leads determining A...
|
|
{methodType(I), methodType(V), methodType(V, V, I)},
|
|
{methodType(I), methodType(V), methodType(V, V, I, A)},
|
|
{methodType(I,A), methodType(V), methodType(V, V, I, A)},
|
|
{methodType(I), methodType(V,A), methodType(V, V, I, A)},
|
|
// body leads, with void V
|
|
{methodType(I), methodType(void.class), methodType(void.class, I)},
|
|
{methodType(I), methodType(void.class), methodType(void.class, I, A)},
|
|
{methodType(I,A), methodType(void.class), methodType(void.class, I, A)},
|
|
{methodType(I), methodType(void.class,A), methodType(void.class, I, A)},
|
|
// count leads determining A..., but only if body drops all A...
|
|
{methodType(I,A), methodType(V), methodType(V, V, I)},
|
|
{methodType(I,A), methodType(V,A), methodType(V, V, I)},
|
|
// count leads, with void V
|
|
{methodType(I,A), methodType(void.class), methodType(void.class, I)},
|
|
{methodType(I,A), methodType(void.class,A), methodType(void.class, I)},
|
|
};
|
|
}
|
|
|
|
@Test(dataProvider = "countedLoopBodyParameters")
|
|
public static void testCountedLoopBodyParameters(MethodType countType, MethodType initType, MethodType bodyType) throws Throwable {
|
|
MethodHandle loop = MethodHandles.countedLoop(
|
|
MethodHandles.empty(countType),
|
|
initType == null ? null : MethodHandles.empty(initType),
|
|
MethodHandles.empty(bodyType));
|
|
// The rule: If body takes the minimum number of parameters, then take what countType offers.
|
|
// The initType has to just roll with whatever the other two agree on.
|
|
int innerParams = (bodyType.returnType() == void.class ? 1 : 2);
|
|
MethodType expectType = bodyType.dropParameterTypes(0, innerParams);
|
|
if (expectType.parameterCount() == 0)
|
|
expectType = expectType.insertParameterTypes(0, countType.parameterList());
|
|
assertEquals(expectType, loop.type());
|
|
}
|
|
|
|
@Test(dataProvider = "countedLoopBodyParameters")
|
|
public static void testCountedLoopBodyParametersNullInit(MethodType countType, MethodType initType, MethodType bodyType) throws Throwable {
|
|
testCountedLoopBodyParameters(countType, null, bodyType);
|
|
}
|
|
|
|
@Test
|
|
public static void testCountedLoopStateInitializedToNull() throws Throwable {
|
|
MethodHandle loop = MethodHandles.countedLoop(MethodHandles.constant(int.class, 5),
|
|
MethodHandles.empty(methodType(String.class)), Counted.MH_stateBody);
|
|
assertEquals(Counted.MT_bodyDeterminesState, loop.type());
|
|
assertEquals("sssssnull01234", loop.invoke());
|
|
}
|
|
|
|
@Test
|
|
public static void testCountedLoopArgsDefinedByIterations() throws Throwable {
|
|
MethodHandle iterations =
|
|
MethodHandles.dropArguments(MethodHandles.constant(int.class, 3), 0, String.class);
|
|
MethodHandle loop = MethodHandles.countedLoop(iterations,
|
|
MethodHandles.empty(iterations.type().changeReturnType(String.class)), Counted.MH_append);
|
|
assertEquals(Counted.MT_iterationsDefineArgs, loop.type());
|
|
assertEquals("hello012", loop.invoke("hello"));
|
|
}
|
|
|
|
@Test
|
|
public static void testCountedRangeLoop() throws Throwable {
|
|
// String s = "Lambdaman!"; for (int i = -5; i < 8; ++i) { s = "na " + s; } return s; => a well known theme
|
|
MethodHandle fitm5 = MethodHandles.dropArguments(Counted.MH_m5, 0, String.class);
|
|
MethodHandle fit8 = MethodHandles.dropArguments(Counted.MH_8, 0, String.class);
|
|
MethodHandle loop = MethodHandles.countedLoop(fitm5, fit8, Counted.MH_start, Counted.MH_step);
|
|
assertEquals(Counted.MT_counted, loop.type());
|
|
assertEquals("na na na na na na na na na na na na na Lambdaman!", loop.invoke("Lambdaman!"));
|
|
}
|
|
|
|
@Test
|
|
public static void testCountedLoopCounterInit() throws Throwable {
|
|
// int x = 0; for (int i = 0; i < 5; ++i) { x += i; } return x; => 10
|
|
// (only if counter's first value in body is 0)
|
|
MethodHandle iter = MethodHandles.constant(int.class, 5);
|
|
MethodHandle init = MethodHandles.constant(int.class, 0);
|
|
MethodHandle body = Counted.MH_addCounter;
|
|
MethodHandle loop = MethodHandles.countedLoop(iter, init, body);
|
|
assertEquals(Counted.MT_counterInit, loop.type());
|
|
assertEquals(10, loop.invoke());
|
|
}
|
|
|
|
@Test
|
|
public static void testCountedLoopEmpty() throws Throwable {
|
|
// for (int i = 0; i < 5; ++i) { /* empty */ }
|
|
MethodHandle loop = MethodHandles.countedLoop(MethodHandles.constant(int.class, 5), null,
|
|
MethodHandles.empty(methodType(void.class, int.class)));
|
|
assertEquals(methodType(void.class), loop.type());
|
|
loop.invoke();
|
|
}
|
|
|
|
@Test
|
|
public static void testCountedRangeLoopEmpty() throws Throwable {
|
|
// for (int i = -5; i < 5; ++i) { /* empty */ }
|
|
MethodHandle loop = MethodHandles.countedLoop(MethodHandles.constant(int.class, -5),
|
|
MethodHandles.constant(int.class, 5), null, MethodHandles.empty(methodType(void.class, int.class)));
|
|
assertEquals(methodType(void.class), loop.type());
|
|
loop.invoke();
|
|
}
|
|
|
|
@DataProvider
|
|
static Object[][] countedLoopNegativeData() {
|
|
MethodHandle dummy = MethodHandles.zero(void.class);
|
|
MethodHandle one = MethodHandles.constant(int.class, 1);
|
|
MethodHandle oneString = MethodHandles.dropArguments(one, 0, String.class);
|
|
MethodHandle oneDouble = MethodHandles.dropArguments(one, 0, double.class);
|
|
return new Object[][]{
|
|
{dummy, one, dummy, dummy, String.format("start/end must return int %s, %s", dummy, one)},
|
|
{one, dummy, dummy, dummy, String.format("start/end must return int %s, %s", one, dummy)},
|
|
{oneString, oneDouble, dummy, dummy,
|
|
String.format("start and end parameter types must match: %s != %s", oneString.type(),
|
|
oneDouble.type())},
|
|
{oneString, oneString, dummy, dummy,
|
|
String.format("start/end and init parameter types must match: %s != %s", oneString.type(),
|
|
dummy.type())},
|
|
{one, one, null, dummy, String.format("actual and expected body signatures must match: %s != %s",
|
|
dummy.type(), dummy.type().appendParameterTypes(int.class))}
|
|
};
|
|
}
|
|
|
|
@Test(dataProvider = "countedLoopNegativeData")
|
|
public static void testCountedLoopNegative(MethodHandle start, MethodHandle end, MethodHandle init,
|
|
MethodHandle body, String msg) {
|
|
if (true) return; //%%%FIXME%%%%
|
|
boolean caught = false;
|
|
try {
|
|
MethodHandles.countedLoop(start, end, init, body);
|
|
} catch (IllegalArgumentException iae) {
|
|
assertEquals(msg, iae.getMessage());
|
|
caught = true;
|
|
}
|
|
assertTrue(caught);
|
|
}
|
|
|
|
@Test
|
|
public static void testIterateSum() throws Throwable {
|
|
// Integer[] a = new Integer[]{1,2,3,4,5,6}; int sum = 0; for (int e : a) { sum += e; } return sum; => 21
|
|
MethodHandle loop = MethodHandles.iteratedLoop(Iterate.MH_sumIterator, Iterate.MH_sumInit, Iterate.MH_sumStep);
|
|
assertEquals(Iterate.MT_sum, loop.type());
|
|
assertEquals(21, loop.invoke(new Integer[]{1, 2, 3, 4, 5, 6}));
|
|
}
|
|
|
|
@DataProvider
|
|
static Object[][] iteratorInits() {
|
|
return new Object[][]{{Iterate.MH_iteratorFromList}, {Iterate.MH_iteratorFromIterable}, {null}};
|
|
}
|
|
|
|
@Test(dataProvider = "iteratorInits")
|
|
public static void testIterateReverse(MethodHandle iterator) throws Throwable {
|
|
// this test uses List as its loop state type; don't try to change that
|
|
if (iterator != null)
|
|
iterator = iterator.asType(iterator.type().changeParameterType(0, List.class));
|
|
for (int i = 0; i < 4; i++) {
|
|
MethodHandle init = Iterate.MH_reverseInit, body = Iterate.MH_reverseStep;
|
|
boolean snipInit = (i & 1) != 0, snipBody = (i & 2) != 0;
|
|
if (snipInit) init = snip(init);
|
|
if (snipBody) body = snip(body);
|
|
if (!snipInit && snipBody && iterator == null) {
|
|
// Body does not determine (A...), so the default guy just picks Iterable.
|
|
// If body insisted on (List), the default guy would adjust himself.
|
|
// Init has no authority to change the (A...), so must patch init.
|
|
// All according to plan!
|
|
init = slap(snip(init), Iterable.class);
|
|
}
|
|
System.out.println("testIterateReverse i="+i+" : "+Arrays.asList(iterator, init, body));
|
|
MethodHandle loop = MethodHandles.iteratedLoop(iterator, init, body);
|
|
MethodType expectedType = Iterate.MT_reverse;
|
|
if (iterator == null && i >= 2)
|
|
expectedType = expectedType.changeParameterType(0, Iterable.class);
|
|
assertEquals(expectedType, loop.type());
|
|
List<String> list = Arrays.asList("a", "b", "c", "d", "e");
|
|
List<String> reversedList = Arrays.asList("e", "d", "c", "b", "a");
|
|
assertEquals(reversedList, (List<String>) loop.invoke(list));
|
|
}
|
|
}
|
|
|
|
@Test(dataProvider = "iteratorInits")
|
|
public static void testIterateLength(MethodHandle iterator) throws Throwable {
|
|
MethodHandle body = Iterate.MH_lengthStep;
|
|
MethodHandle init = Iterate.MH_lengthInit;
|
|
MethodType expectedType = Iterate.MT_length;
|
|
int barity = body.type().parameterCount();
|
|
Class<?> iteratorSource = iterator == null ? null : iterator.type().parameterType(0);
|
|
if (iterator != null && iteratorSource != body.type().parameterType(barity-1)) {
|
|
// adjust body to accept the other type
|
|
body = body.asType(body.type().changeParameterType(barity-1, iteratorSource));
|
|
init = init.asType(init.type().changeParameterType(0, iteratorSource));
|
|
expectedType = expectedType.changeParameterType(0, iteratorSource);
|
|
}
|
|
for (;; init = snip(init)) {
|
|
System.out.println("testIterateLength.init = "+init);
|
|
MethodHandle loop = MethodHandles.iteratedLoop(iterator, init, body);
|
|
assertEquals(expectedType, loop.type());
|
|
List<Double> list = Arrays.asList(23.0, 148.0, 42.0);
|
|
assertEquals(list.size(), (int) loop.invoke(list));
|
|
if (init == null) break;
|
|
}
|
|
}
|
|
|
|
@Test(dataProvider = "iteratorInits")
|
|
public static void testIterateMap(MethodHandle iterator) throws Throwable {
|
|
MethodHandle body = Iterate.MH_mapStep;
|
|
MethodHandle init = Iterate.MH_mapInit;
|
|
MethodType expectedType = Iterate.MT_map;
|
|
int barity = body.type().parameterCount();
|
|
Class<?> iteratorSource = iterator == null ? null : iterator.type().parameterType(0);
|
|
if (iterator != null && iteratorSource != body.type().parameterType(barity-1)) {
|
|
// adjust body to accept the other type
|
|
body = body.asType(body.type().changeParameterType(barity-1, iteratorSource));
|
|
init = init.asType(init.type().changeParameterType(0, iteratorSource));
|
|
expectedType = expectedType.changeParameterType(0, iteratorSource);
|
|
}
|
|
for (; init != null; init = snip(init)) {
|
|
System.out.println("testIterateMap.init = "+init);
|
|
MethodHandle loop = MethodHandles.iteratedLoop(iterator, init, body);
|
|
assertEquals(expectedType, loop.type());
|
|
List<String> list = Arrays.asList("Hello", "world", "!");
|
|
List<String> upList = Arrays.asList("HELLO", "WORLD", "!");
|
|
assertEquals(upList, (List<String>) loop.invoke(list));
|
|
}
|
|
}
|
|
|
|
@Test(dataProvider = "iteratorInits")
|
|
public static void testIteratePrint(MethodHandle iterator) throws Throwable {
|
|
MethodHandle body = Iterate.MH_printStep;
|
|
MethodType expectedType = Iterate.MT_print;
|
|
int barity = body.type().parameterCount();
|
|
Class<?> iteratorSource = iterator == null ? null : iterator.type().parameterType(0);
|
|
if (iterator != null && iteratorSource != body.type().parameterType(barity-1)) {
|
|
// adjust body to accept the other type
|
|
body = body.asType(body.type().changeParameterType(barity-1, iteratorSource));
|
|
expectedType = expectedType.changeParameterType(0, iteratorSource);
|
|
}
|
|
MethodHandle loop = MethodHandles.iteratedLoop(iterator, null, body);
|
|
assertEquals(expectedType, loop.type());
|
|
loop.invoke(Arrays.asList("hello", "world"));
|
|
}
|
|
|
|
@Test(expectedExceptions = NullPointerException.class)
|
|
public static void testIterateNullBody() {
|
|
MethodHandles.iteratedLoop(MethodHandles.empty(methodType(Iterator.class, int.class)),
|
|
MethodHandles.identity(int.class), null);
|
|
}
|
|
|
|
@DataProvider
|
|
static Object[][] wrongIteratorTypes() {
|
|
return new Object[][]{{void.class}, {Object.class}, {Iterable.class}};
|
|
}
|
|
|
|
@Test(dataProvider = "wrongIteratorTypes")
|
|
public static void testIterateVoidIterator(Class<?> it) {
|
|
boolean caught = false;
|
|
MethodType v = methodType(it);
|
|
try {
|
|
MethodHandles.iteratedLoop(MethodHandles.empty(v), null, MethodHandles.empty(v));
|
|
} catch(IllegalArgumentException iae) {
|
|
assertEqualsFIXME("iteratedLoop first argument must have Iterator return type", iae.getMessage());
|
|
caught = true;
|
|
}
|
|
assertTrue(caught);
|
|
}
|
|
|
|
@Test(dataProvider = "iteratorInits")
|
|
public static void testIterateVoidInit(MethodHandle iterator) throws Throwable {
|
|
// this test uses List as its loop state type; don't try to change that
|
|
if (iterator != null)
|
|
iterator = iterator.asType(iterator.type().changeParameterType(0, List.class));
|
|
MethodHandle loop = MethodHandles.iteratedLoop(iterator, Iterate.MH_voidInit, Iterate.MH_printStep);
|
|
assertEquals(Iterate.MT_print, loop.type());
|
|
loop.invoke(Arrays.asList("hello", "world"));
|
|
}
|
|
|
|
@DataProvider
|
|
static Object[][] iterateParameters() {
|
|
MethodType i = methodType(int.class);
|
|
MethodType sil_v = methodType(void.class, String.class, int.class, List.class);
|
|
MethodType isl_i = methodType(int.class, int.class, String.class, List.class);
|
|
MethodType isli_i = methodType(int.class, int.class, String.class, List.class, int.class);
|
|
MethodType sl_v = methodType(void.class, String.class, List.class);
|
|
MethodType sli_v = methodType(void.class, String.class, List.class, int.class);
|
|
MethodType l_it = methodType(Iterator.class, List.class);
|
|
MethodType li_i = methodType(int.class, List.class, int.class);
|
|
MethodType li_it = methodType(Iterator.class, List.class, int.class);
|
|
MethodType il_it = methodType(Iterator.class, int.class, List.class);
|
|
MethodType l_i = methodType(int.class, List.class);
|
|
return new Object[][]{
|
|
{l_it, null, sl_v, ""},
|
|
{l_it, l_i, isl_i, ""},
|
|
{l_it, null, sl_v, ""},
|
|
{li_it, li_i, isli_i, ""},
|
|
{null, null, sil_v, "inferred first loop argument must inherit from Iterable: int"},
|
|
{il_it, null, sil_v, ""},
|
|
{li_it, null, sli_v, ""},
|
|
{sl_v, null, sl_v, "iteratedLoop first argument must have Iterator return type"},
|
|
{li_it, l_it, sl_v,
|
|
String.format("iterator and init parameter lists must match: %s != %s", li_it, l_it)},
|
|
{li_it, li_i, isl_i,
|
|
String.format("body types (regard parameter types after index 0, and result type) must match: %s != %s",
|
|
isl_i, isl_i.dropParameterTypes(0, 1).appendParameterTypes(int.class))}
|
|
};
|
|
}
|
|
|
|
@Test(dataProvider = "iterateParameters")
|
|
public static void testIterateParameters(MethodType it, MethodType in, MethodType bo, String msg) {
|
|
boolean negative = !msg.isEmpty();
|
|
MethodHandle iterator = it == null ? null : MethodHandles.empty(it);
|
|
MethodHandle init = in == null ? null : MethodHandles.empty(in);
|
|
boolean caught = false;
|
|
MethodHandle loop = null;
|
|
try {
|
|
loop = MethodHandles.iteratedLoop(iterator, init, MethodHandles.empty(bo));
|
|
} catch (Throwable t) {
|
|
if (!negative) {
|
|
throw t;
|
|
}
|
|
assertEqualsFIXME(msg, t.getMessage());
|
|
caught = true;
|
|
}
|
|
if (negative) {
|
|
assertTrue(caught);
|
|
} else {
|
|
MethodType lt = loop.type();
|
|
if (it == null && in == null) {
|
|
assertEquals(bo.dropParameterTypes(0, 1), lt);
|
|
} else if (it == null) {
|
|
if (in.parameterCount() == 0) {
|
|
assertEquals(bo.dropParameterTypes(0, in.returnType() == void.class ? 1 : 2), lt);
|
|
} else {
|
|
assertEquals(methodType(bo.returnType(), in.parameterArray()), lt);
|
|
}
|
|
} else if (in == null) {
|
|
assertEquals(methodType(bo.returnType(), it.parameterArray()), lt);
|
|
} else if (it.parameterCount() > in.parameterCount()) {
|
|
assertEquals(methodType(bo.returnType(), it.parameterArray()), lt);
|
|
} else if (it.parameterCount() < in.parameterCount()) {
|
|
assertEquals(methodType(bo.returnType(), in.parameterArray()), lt);
|
|
} else {
|
|
// both it, in present; with equal parameter list lengths
|
|
assertEquals(it.parameterList(), lt.parameterList());
|
|
assertEquals(in.parameterList(), lt.parameterList());
|
|
assertEquals(bo.returnType(), lt.returnType());
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test
|
|
public static void testIteratorSubclass() throws Throwable {
|
|
MethodHandle loop = MethodHandles.iteratedLoop(MethodHandles.empty(methodType(BogusIterator.class, List.class)),
|
|
null, MethodHandles.empty(methodType(void.class, String.class, List.class)));
|
|
assertEquals(methodType(void.class, List.class), loop.type());
|
|
}
|
|
|
|
static class BogusIterator implements Iterator {
|
|
@Override
|
|
public boolean hasNext() {
|
|
return false;
|
|
}
|
|
@Override
|
|
public Object next() {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
static class Empty {
|
|
|
|
static void f() { }
|
|
|
|
static boolean pred() {
|
|
return false;
|
|
}
|
|
|
|
static int c() {
|
|
return 23;
|
|
}
|
|
|
|
static final Class<Empty> EMPTY = Empty.class;
|
|
|
|
static final MethodType MT_f = methodType(void.class);
|
|
static final MethodType MT_pred = methodType(boolean.class);
|
|
static final MethodType MT_c = methodType(int.class);
|
|
|
|
static final MethodHandle MH_f;
|
|
static final MethodHandle MH_pred;
|
|
static final MethodHandle MH_c;
|
|
|
|
static {
|
|
try {
|
|
MH_f = LOOKUP.findStatic(EMPTY, "f", MT_f);
|
|
MH_pred = LOOKUP.findStatic(EMPTY, "pred", MT_pred);
|
|
MH_c = LOOKUP.findStatic(EMPTY, "c", MT_c);
|
|
} catch (Exception e) {
|
|
throw new ExceptionInInitializerError(e);
|
|
}
|
|
}
|
|
}
|
|
|
|
static class Fac {
|
|
|
|
static int zero(int k) {
|
|
return 0;
|
|
}
|
|
|
|
static int one(int k) {
|
|
return 1;
|
|
}
|
|
|
|
static boolean pred(int i, int acc, int k) {
|
|
return i < k;
|
|
}
|
|
|
|
static int inc(int i, int acc, int k) {
|
|
return i + 1;
|
|
}
|
|
|
|
static int mult(int i, int acc, int k) {
|
|
return i * acc;
|
|
}
|
|
|
|
static void dot(int i, int acc, int k) {
|
|
System.out.print('.');
|
|
}
|
|
|
|
static int fin(int i, int acc, int k) {
|
|
return acc;
|
|
}
|
|
|
|
static final Class<Fac> FAC = Fac.class;
|
|
|
|
static final MethodType MT_init = methodType(int.class, int.class);
|
|
static final MethodType MT_fn = methodType(int.class, int.class, int.class, int.class);
|
|
static final MethodType MT_dot = methodType(void.class, int.class, int.class, int.class);
|
|
static final MethodType MT_pred = methodType(boolean.class, int.class, int.class, int.class);
|
|
|
|
static final MethodHandle MH_zero;
|
|
static final MethodHandle MH_one;
|
|
static final MethodHandle MH_pred;
|
|
static final MethodHandle MH_inc;
|
|
static final MethodHandle MH_mult;
|
|
static final MethodHandle MH_dot;
|
|
static final MethodHandle MH_fin;
|
|
|
|
static final MethodType MT_fac = methodType(int.class, int.class);
|
|
|
|
static {
|
|
try {
|
|
MH_zero = LOOKUP.findStatic(FAC, "zero", MT_init);
|
|
MH_one = LOOKUP.findStatic(FAC, "one", MT_init);
|
|
MH_pred = LOOKUP.findStatic(FAC, "pred", MT_pred);
|
|
MH_inc = LOOKUP.findStatic(FAC, "inc", MT_fn);
|
|
MH_mult = LOOKUP.findStatic(FAC, "mult", MT_fn);
|
|
MH_dot = LOOKUP.findStatic(FAC, "dot", MT_dot);
|
|
MH_fin = LOOKUP.findStatic(FAC, "fin", MT_fn);
|
|
} catch (Exception e) {
|
|
throw new ExceptionInInitializerError(e);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
static class Loop {
|
|
|
|
static int inc(int i, int k) {
|
|
return i + 1;
|
|
}
|
|
|
|
static boolean pred(int i, int k) {
|
|
return i < k;
|
|
}
|
|
|
|
static int fin(int i, int k) {
|
|
return k;
|
|
}
|
|
|
|
static final Class<Loop> LOOP = Loop.class;
|
|
|
|
static final MethodType MT_inc = methodType(int.class, int.class, int.class);
|
|
static final MethodType MT_pred = methodType(boolean.class, int.class, int.class);
|
|
static final MethodType MT_fin = methodType(int.class, int.class, int.class);
|
|
|
|
static final MethodHandle MH_inc;
|
|
static final MethodHandle MH_pred;
|
|
static final MethodHandle MH_fin;
|
|
|
|
static final MethodType MT_loop = methodType(int.class, int.class);
|
|
|
|
static {
|
|
try {
|
|
MH_inc = LOOKUP.findStatic(LOOP, "inc", MT_inc);
|
|
MH_pred = LOOKUP.findStatic(LOOP, "pred", MT_pred);
|
|
MH_fin = LOOKUP.findStatic(LOOP, "fin", MT_fin);
|
|
} catch (Exception e) {
|
|
throw new ExceptionInInitializerError(e);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
static class LoopWithVirtuals {
|
|
|
|
static int one(int k) {
|
|
return 1;
|
|
}
|
|
|
|
int inc(int i, int acc, int k) {
|
|
return i + 1;
|
|
}
|
|
|
|
int mult(int i, int acc, int k) {
|
|
return i * acc;
|
|
}
|
|
|
|
boolean pred(int i, int acc, int k) {
|
|
return i < k;
|
|
}
|
|
|
|
int fin(int i, int acc, int k) {
|
|
return acc;
|
|
}
|
|
|
|
static final Class<LoopWithVirtuals> LOOP_WITH_VIRTUALS = LoopWithVirtuals.class;
|
|
|
|
static final MethodType MT_one = methodType(int.class, int.class);
|
|
static final MethodType MT_inc = methodType(int.class, int.class, int.class, int.class);
|
|
static final MethodType MT_mult = methodType(int.class, int.class, int.class, int.class);
|
|
static final MethodType MT_pred = methodType(boolean.class, int.class, int.class, int.class);
|
|
static final MethodType MT_fin = methodType(int.class, int.class, int.class, int.class);
|
|
|
|
static final MethodHandle MH_one;
|
|
static final MethodHandle MH_inc;
|
|
static final MethodHandle MH_mult;
|
|
static final MethodHandle MH_pred;
|
|
static final MethodHandle MH_fin;
|
|
|
|
static final MethodType MT_loop = methodType(int.class, LOOP_WITH_VIRTUALS, int.class);
|
|
|
|
static {
|
|
try {
|
|
MH_one = LOOKUP.findStatic(LOOP_WITH_VIRTUALS, "one", MT_one);
|
|
MH_inc = LOOKUP.findVirtual(LOOP_WITH_VIRTUALS, "inc", MT_inc);
|
|
MH_mult = LOOKUP.findVirtual(LOOP_WITH_VIRTUALS, "mult", MT_mult);
|
|
MH_pred = LOOKUP.findVirtual(LOOP_WITH_VIRTUALS, "pred", MT_pred);
|
|
MH_fin = LOOKUP.findVirtual(LOOP_WITH_VIRTUALS, "fin", MT_fin);
|
|
} catch (Exception e) {
|
|
throw new ExceptionInInitializerError(e);
|
|
}
|
|
}
|
|
|
|
static MethodHandle permute(MethodHandle h) {
|
|
// The handles representing virtual methods need to be rearranged to match the required order of arguments
|
|
// (loop-local state comes first, then loop arguments). As the receiver comes first in the signature but is
|
|
// a loop argument, it must be moved to the appropriate position in the signature.
|
|
return MethodHandles.permuteArguments(h,
|
|
methodType(h.type().returnType(), int.class, int.class, LOOP_WITH_VIRTUALS, int.class), 2, 0, 1, 3);
|
|
}
|
|
|
|
}
|
|
|
|
static class While {
|
|
|
|
static int zero(int limit) {
|
|
return 0;
|
|
}
|
|
|
|
static boolean pred(int i, int limit) {
|
|
return i < limit;
|
|
}
|
|
|
|
static int step(int i, int limit) {
|
|
return i + 1;
|
|
}
|
|
|
|
static String initString() {
|
|
return "a";
|
|
}
|
|
|
|
static boolean predString(String s) {
|
|
return s.length() != 1;
|
|
}
|
|
|
|
static String stepString(String s) {
|
|
return s + "a";
|
|
}
|
|
|
|
static List<String> zipInitZip(Iterator<String> a, Iterator<String> b) {
|
|
return new ArrayList<>();
|
|
}
|
|
|
|
static boolean zipPred(List<String> zip, Iterator<String> a, Iterator<String> b) {
|
|
return a.hasNext() && b.hasNext();
|
|
}
|
|
|
|
static List<String> zipStep(List<String> zip, Iterator<String> a, Iterator<String> b) {
|
|
zip.add(a.next());
|
|
zip.add(b.next());
|
|
return zip;
|
|
}
|
|
|
|
private int i = 0;
|
|
|
|
void voidInit(int k) {
|
|
// empty
|
|
}
|
|
|
|
void voidBody(int k) {
|
|
++i;
|
|
}
|
|
|
|
boolean voidPred(int k) {
|
|
return i < k;
|
|
}
|
|
|
|
static final Class<While> WHILE = While.class;
|
|
|
|
static final MethodType MT_zero = methodType(int.class, int.class);
|
|
static final MethodType MT_pred = methodType(boolean.class, int.class, int.class);
|
|
static final MethodType MT_fn = methodType(int.class, int.class, int.class);
|
|
static final MethodType MT_initString = methodType(String.class);
|
|
static final MethodType MT_predString = methodType(boolean.class, String.class);
|
|
static final MethodType MT_stepString = methodType(String.class, String.class);
|
|
static final MethodType MT_zipInitZip = methodType(List.class, Iterator.class, Iterator.class);
|
|
static final MethodType MT_zipPred = methodType(boolean.class, List.class, Iterator.class, Iterator.class);
|
|
static final MethodType MT_zipStep = methodType(List.class, List.class, Iterator.class, Iterator.class);
|
|
static final MethodType MT_voidInit = methodType(void.class, int.class);
|
|
static final MethodType MT_voidBody = methodType(void.class, int.class);
|
|
static final MethodType MT_voidPred = methodType(boolean.class, int.class);
|
|
|
|
static final MethodHandle MH_zero;
|
|
static final MethodHandle MH_pred;
|
|
static final MethodHandle MH_step;
|
|
static final MethodHandle MH_initString;
|
|
static final MethodHandle MH_predString;
|
|
static final MethodHandle MH_stepString;
|
|
static final MethodHandle MH_zipInitZip;
|
|
static final MethodHandle MH_zipPred;
|
|
static final MethodHandle MH_zipStep;
|
|
static final MethodHandle MH_voidInit;
|
|
static final MethodHandle MH_voidBody;
|
|
static final MethodHandle MH_voidPred;
|
|
|
|
static final MethodType MT_while = methodType(int.class, int.class);
|
|
static final MethodType MT_string = methodType(String.class);
|
|
static final MethodType MT_zip = methodType(List.class, Iterator.class, Iterator.class);
|
|
static final MethodType MT_void = methodType(void.class, int.class);
|
|
|
|
static {
|
|
try {
|
|
MH_zero = LOOKUP.findStatic(WHILE, "zero", MT_zero);
|
|
MH_pred = LOOKUP.findStatic(WHILE, "pred", MT_pred);
|
|
MH_step = LOOKUP.findStatic(WHILE, "step", MT_fn);
|
|
MH_initString = LOOKUP.findStatic(WHILE, "initString", MT_initString);
|
|
MH_predString = LOOKUP.findStatic(WHILE, "predString", MT_predString);
|
|
MH_stepString = LOOKUP.findStatic(WHILE, "stepString", MT_stepString);
|
|
MH_zipInitZip = LOOKUP.findStatic(WHILE, "zipInitZip", MT_zipInitZip);
|
|
MH_zipPred = LOOKUP.findStatic(WHILE, "zipPred", MT_zipPred);
|
|
MH_zipStep = LOOKUP.findStatic(WHILE, "zipStep", MT_zipStep);
|
|
MH_voidInit = LOOKUP.findVirtual(WHILE, "voidInit", MT_voidInit);
|
|
MH_voidBody = LOOKUP.findVirtual(WHILE, "voidBody", MT_voidBody);
|
|
MH_voidPred = LOOKUP.findVirtual(WHILE, "voidPred", MT_voidPred);
|
|
} catch (Exception e) {
|
|
throw new ExceptionInInitializerError(e);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
static class Counted {
|
|
|
|
static String start(String arg) {
|
|
return arg;
|
|
}
|
|
|
|
static String step(String v, int counter) {
|
|
return "na " + v;
|
|
}
|
|
|
|
static void stepUpdateArray(int counter, int[] a) {
|
|
++a[0];
|
|
}
|
|
|
|
static void printHello(int counter) {
|
|
System.out.print("hello");
|
|
}
|
|
|
|
static int addCounter(int x, int counter) {
|
|
return x + counter;
|
|
}
|
|
|
|
static String stateBody(String s, int counter) {
|
|
return "s" + s + counter;
|
|
}
|
|
|
|
static String append(String localState, int counter, String loopArg) {
|
|
if (null == localState) {
|
|
return loopArg + counter;
|
|
}
|
|
return localState + counter;
|
|
}
|
|
|
|
static final Class<Counted> COUNTED = Counted.class;
|
|
|
|
static final MethodType MT_start = methodType(String.class, String.class);
|
|
static final MethodType MT_step = methodType(String.class, String.class, int.class);
|
|
static final MethodType MT_stepUpdateArray = methodType(void.class, int.class, int[].class);
|
|
static final MethodType MT_printHello = methodType(void.class, int.class);
|
|
static final MethodType MT_addCounter = methodType(int.class, int.class, int.class);
|
|
static final MethodType MT_stateBody = methodType(String.class, String.class, int.class);
|
|
static final MethodType MT_append = methodType(String.class, String.class, int.class, String.class);
|
|
|
|
static final MethodHandle MH_13;
|
|
static final MethodHandle MH_m5;
|
|
static final MethodHandle MH_8;
|
|
static final MethodHandle MH_start;
|
|
static final MethodHandle MH_step;
|
|
static final MethodHandle MH_stepUpdateArray;
|
|
static final MethodHandle MH_printHello;
|
|
static final MethodHandle MH_addCounter;
|
|
static final MethodHandle MH_stateBody;
|
|
static final MethodHandle MH_append;
|
|
|
|
static final MethodType MT_counted = methodType(String.class, String.class);
|
|
static final MethodType MT_arrayCounted = methodType(void.class, int[].class);
|
|
static final MethodType MT_countedPrinting = methodType(void.class);
|
|
static final MethodType MT_counterInit = methodType(int.class);
|
|
static final MethodType MT_bodyDeterminesState = methodType(String.class);
|
|
static final MethodType MT_iterationsDefineArgs = methodType(String.class, String.class);
|
|
|
|
static {
|
|
try {
|
|
MH_13 = MethodHandles.constant(int.class, 13);
|
|
MH_m5 = MethodHandles.constant(int.class, -5);
|
|
MH_8 = MethodHandles.constant(int.class, 8);
|
|
MH_start = LOOKUP.findStatic(COUNTED, "start", MT_start);
|
|
MH_step = LOOKUP.findStatic(COUNTED, "step", MT_step);
|
|
MH_stepUpdateArray = LOOKUP.findStatic(COUNTED, "stepUpdateArray", MT_stepUpdateArray);
|
|
MH_printHello = LOOKUP.findStatic(COUNTED, "printHello", MT_printHello);
|
|
MH_addCounter = LOOKUP.findStatic(COUNTED, "addCounter", MT_addCounter);
|
|
MH_stateBody = LOOKUP.findStatic(COUNTED, "stateBody", MT_stateBody);
|
|
MH_append = LOOKUP.findStatic(COUNTED, "append", MT_append);
|
|
} catch (Exception e) {
|
|
throw new ExceptionInInitializerError(e);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
static class Iterate {
|
|
|
|
static Iterator<Integer> sumIterator(Integer[] a) {
|
|
return Arrays.asList(a).iterator();
|
|
}
|
|
|
|
static int sumInit(Integer[] a) {
|
|
return 0;
|
|
}
|
|
|
|
static int sumStep(int s, int e, Integer[] a) {
|
|
return s + e;
|
|
}
|
|
|
|
static List<String> reverseInit(List<String> l) {
|
|
return new ArrayList<>();
|
|
}
|
|
|
|
static List<String> reverseStep(List<String> r, String e, List<String> l) {
|
|
r.add(0, e);
|
|
return r;
|
|
}
|
|
|
|
static int lengthInit(List<Double> l) {
|
|
return 0;
|
|
}
|
|
|
|
static int lengthStep(int len, Object o, List<Double> l) {
|
|
return len + 1;
|
|
}
|
|
|
|
static List<String> mapInit(List<String> l) {
|
|
return new ArrayList<>();
|
|
}
|
|
|
|
static List<String> mapStep(List<String> r, String e, List<String> l) {
|
|
r.add(e.toUpperCase());
|
|
return r;
|
|
}
|
|
|
|
static void printStep(String s, List<String> l) {
|
|
System.out.print(s);
|
|
}
|
|
|
|
static void voidInit(List<String> l) {
|
|
// empty
|
|
}
|
|
|
|
static ListIterator<?> iteratorFromList(List<?> l) {
|
|
return l.listIterator();
|
|
}
|
|
|
|
static Iterator<?> iteratorFromIterable(Iterable<?> l) {
|
|
return l.iterator();
|
|
}
|
|
|
|
static final Class<Iterate> ITERATE = Iterate.class;
|
|
|
|
static final MethodType MT_sumIterator = methodType(Iterator.class, Integer[].class);
|
|
|
|
static final MethodType MT_sumInit = methodType(int.class, Integer[].class);
|
|
static final MethodType MT_reverseInit = methodType(List.class, List.class);
|
|
static final MethodType MT_lenghInit = methodType(int.class, List.class);
|
|
static final MethodType MT_mapInit = methodType(List.class, List.class);
|
|
|
|
static final MethodType MT_sumStep = methodType(int.class, int.class, int.class, Integer[].class);
|
|
static final MethodType MT_reverseStep = methodType(List.class, List.class, String.class, List.class);
|
|
static final MethodType MT_lengthStep = methodType(int.class, int.class, Object.class, List.class);
|
|
static final MethodType MT_mapStep = methodType(List.class, List.class, String.class, List.class);
|
|
static final MethodType MT_printStep = methodType(void.class, String.class, List.class);
|
|
|
|
static final MethodType MT_voidInit = methodType(void.class, List.class);
|
|
|
|
static final MethodType MT_iteratorFromList = methodType(ListIterator.class, List.class);
|
|
static final MethodType MT_iteratorFromIterable = methodType(Iterator.class, Iterable.class);
|
|
|
|
static final MethodHandle MH_sumIterator;
|
|
static final MethodHandle MH_sumInit;
|
|
static final MethodHandle MH_sumStep;
|
|
static final MethodHandle MH_printStep;
|
|
|
|
static final MethodHandle MH_reverseInit;
|
|
static final MethodHandle MH_reverseStep;
|
|
|
|
static final MethodHandle MH_lengthInit;
|
|
static final MethodHandle MH_lengthStep;
|
|
|
|
static final MethodHandle MH_mapInit;
|
|
static final MethodHandle MH_mapStep;
|
|
|
|
static final MethodHandle MH_voidInit;
|
|
|
|
static final MethodHandle MH_iteratorFromList;
|
|
static final MethodHandle MH_iteratorFromIterable;
|
|
|
|
static final MethodType MT_sum = methodType(int.class, Integer[].class);
|
|
static final MethodType MT_reverse = methodType(List.class, List.class);
|
|
static final MethodType MT_length = methodType(int.class, List.class);
|
|
static final MethodType MT_map = methodType(List.class, List.class);
|
|
static final MethodType MT_print = methodType(void.class, List.class);
|
|
|
|
static {
|
|
try {
|
|
MH_sumIterator = LOOKUP.findStatic(ITERATE, "sumIterator", MT_sumIterator);
|
|
MH_sumInit = LOOKUP.findStatic(ITERATE, "sumInit", MT_sumInit);
|
|
MH_sumStep = LOOKUP.findStatic(ITERATE, "sumStep", MT_sumStep);
|
|
MH_reverseInit = LOOKUP.findStatic(ITERATE, "reverseInit", MT_reverseInit);
|
|
MH_reverseStep = LOOKUP.findStatic(ITERATE, "reverseStep", MT_reverseStep);
|
|
MH_lengthInit = LOOKUP.findStatic(ITERATE, "lengthInit", MT_lenghInit);
|
|
MH_lengthStep = LOOKUP.findStatic(ITERATE, "lengthStep", MT_lengthStep);
|
|
MH_mapInit = LOOKUP.findStatic(ITERATE, "mapInit", MT_mapInit);
|
|
MH_mapStep = LOOKUP.findStatic(ITERATE, "mapStep", MT_mapStep);
|
|
MH_printStep = LOOKUP.findStatic(ITERATE, "printStep", MT_printStep);
|
|
MH_voidInit = LOOKUP.findStatic(ITERATE, "voidInit", MT_voidInit);
|
|
MH_iteratorFromList = LOOKUP.findStatic(ITERATE, "iteratorFromList", MT_iteratorFromList);
|
|
MH_iteratorFromIterable = LOOKUP.findStatic(ITERATE, "iteratorFromIterable", MT_iteratorFromIterable);
|
|
} catch (Exception e) {
|
|
throw new ExceptionInInitializerError(e);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
}
|