1 /* 2 * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24 /* 25 * @test 26 * @modules java.base/valhalla.shady 27 * jdk.incubator.mvt 28 * @run testng/othervm -XX:+EnableMVT -XX:+ValueArrayFlatten MVTTest 29 * @run testng/othervm -XX:+EnableMVT -XX:-ValueArrayFlatten MVTTest 30 * @run testng/othervm -XX:+EnableMVT -Dvalhalla.enableValueLambdaForms=true MVTTest 31 * @run testng/othervm -XX:+EnableMVT -Dvalhalla.enableValueLambdaForms=true -Dvalhalla.enablePoolPatches=true MVTTest 32 */ 33 34 import jdk.incubator.mvt.ValueType; 35 import org.testng.annotations.Test; 36 import valhalla.shady.MinimalValueTypes_1_0; 37 38 import java.lang.invoke.MethodHandle; 39 import java.lang.invoke.MethodHandles; 40 import java.lang.reflect.Field; 41 42 import static java.lang.invoke.MethodType.methodType; 43 import static org.testng.Assert.assertEquals; 44 45 @Test 46 public class MVTTest { 47 static final Class<?> DVT; 48 49 static final ValueType<?> VT = ValueType.forClass(Point.class); 50 51 static final Class<?>[] FIELD_TYPES; 52 53 static final String[] FIELD_NAMES; 54 55 static String TEMPLATE = "Point[x=#x, y=#y, z=#z]"; 56 57 static final Object[] FIELD_VALUES = {42, (short) 43, (short) 44}; 58 59 static final MethodHandles.Lookup LOOKUP = MethodHandles.lookup(); 60 61 static final MethodHandle PRINT_POINT; 62 63 static { 64 DVT = MinimalValueTypes_1_0.getValueTypeClass(Point.class); 65 66 Field[] fs = Point.class.getFields(); 67 68 FIELD_TYPES = new Class<?>[fs.length]; 69 FIELD_NAMES = new String[fs.length]; 70 71 for (int i = 0; i < fs.length; i++) { 72 FIELD_TYPES[i] = fs[i].getType(); 73 FIELD_NAMES[i] = fs[i].getName(); 74 } 75 76 try { 77 PRINT_POINT = LOOKUP.findStatic(MVTTest.class, "print", methodType(String.class, Point.class)) 78 .asType(methodType(String.class, DVT)); 79 } 80 catch (Exception e) { 81 throw new RuntimeException(e); 82 } 83 } 84 85 public void testDefaultValue() throws Throwable { 86 for (int i = 0; i < FIELD_NAMES.length; i++) { 87 MethodHandle getter = MethodHandles.collectArguments( 88 VT.findGetter(LOOKUP, FIELD_NAMES[i], FIELD_TYPES[i]), 89 0, 90 VT.defaultValueConstant()); 91 92 assertEquals((int) getter.invoke(), 0); 93 } 94 } 95 96 public void testWither() throws Throwable { 97 testWither(Point.lookup()); 98 testWither(MethodHandles.privateLookupIn(VT.boxClass(), LOOKUP)); 99 testWither(MethodHandles.privateLookupIn(VT.valueClass(), LOOKUP)); 100 } 101 102 void testWither(MethodHandles.Lookup l ) throws Throwable { 103 for (int i = 0; i < FIELD_NAMES.length; i++) { 104 MethodHandle wither = MethodHandles.collectArguments( 105 VT.findWither(l, FIELD_NAMES[i], FIELD_TYPES[i]), 0, VT.defaultValueConstant()); 106 String expected = TEMPLATE.replace("#" + FIELD_NAMES[i], String.valueOf(FIELD_VALUES[i])) 107 .replaceAll("#[xyz]", "0"); 108 109 assertEquals(printReturn(wither).invoke(FIELD_VALUES[i]), expected); 110 } 111 } 112 113 public void testSubstitutability() throws Throwable { 114 Point[] pts = {new Point(1, (short) 6, (short) 3), new Point(1, (short) 2, (short) 3)}; 115 116 MethodHandle substTest = VT.substitutabilityTest(); 117 for (Point p1 : pts) { 118 for (Point p2 : pts) { 119 assertEquals((boolean) substTest.invoke(p1, p2), p1.equals(p2)); 120 } 121 } 122 123 MethodHandle hash = VT.substitutabilityHashCode(); 124 for (Point p1 : pts) { 125 for (Point p2 : pts) { 126 boolean vHashEq = (int) hash.invoke(p1) == (int) hash.invoke(p2); 127 boolean rHashEq = p1.hashCode() == p2.hashCode(); 128 assertEquals(vHashEq, rHashEq); 129 } 130 } 131 } 132 133 public void testIdentity() throws Throwable { 134 String actual = (String) printReturn(MethodHandles.identity(VT.valueClass())) 135 .invoke(new Point(1, (short) 2, (short) 3)); 136 assertEquals(actual, "Point[x=1, y=2, z=3]"); 137 } 138 139 public void testZero() throws Throwable { 140 String actual = (String) printReturn(MethodHandles.zero(VT.valueClass())) 141 .invoke(); 142 assertEquals(actual, "Point[x=0, y=0, z=0]"); 143 } 144 145 public void testEmpty() throws Throwable { 146 String actual = (String) printReturn(MethodHandles.empty(methodType(VT.valueClass(), int.class, String.class))) 147 .invoke(1, ""); 148 assertEquals(actual, "Point[x=0, y=0, z=0]"); 149 } 150 151 public void testArray1D() throws Throwable { 152 //test monodimensional array 153 Object arr = MethodHandles.arrayConstructor(VT.arrayValueClass()).invoke(10); 154 for (int i = 0; i < 10; i++) { 155 Point p = new Point(i, (short) 9, (short) 9); 156 MethodHandles.arrayElementSetter(VT.arrayValueClass()).invoke(arr, i, p); 157 } 158 for (int i = 0; i < 10; i++) { 159 String actual = (String) printReturn(MethodHandles.arrayElementGetter(VT.arrayValueClass())) 160 .invoke(arr, i); 161 String expected = TEMPLATE.replace("#x", String.valueOf(i)) 162 .replaceAll("#[yz]", "9"); 163 assertEquals(actual, expected); 164 } 165 } 166 167 public void testArray10D() throws Throwable { 168 //test multidimensional array 169 Object[] arr2 = (Object[]) MethodHandles.arrayConstructor(VT.arrayValueClass(2)).invoke(10); 170 for (int i = 0; i < 10; i++) { 171 Object innerArr = MethodHandles.arrayConstructor(VT.arrayValueClass()).invoke(10); 172 MethodHandles.arrayElementSetter(VT.arrayValueClass(2)).invoke(arr2, i, innerArr); 173 for (int j = 0; i < 10; i++) { 174 Point p = new Point(i, (short) j, (short) 9); 175 MethodHandles.arrayElementSetter(VT.arrayValueClass()).invoke(innerArr, i, p); 176 } 177 } 178 for (int i = 0; i < 10; i++) { 179 Object innerArr = MethodHandles.arrayElementGetter(VT.arrayValueClass(2)).invoke(arr2, i); 180 for (int j = 0; i < 10; i++) { 181 String actual = (String) printReturn(MethodHandles.arrayElementGetter(VT.arrayValueClass())) 182 .invoke(innerArr, i); 183 String expected = TEMPLATE.replace("#x", String.valueOf(i)) 184 .replace("#y", String.valueOf(j)) 185 .replace("#z", "9"); 186 assertEquals(actual, expected); 187 } 188 } 189 } 190 191 public void testMultiArray() throws Throwable { 192 Object[] arr43 = (Object[]) VT.newMultiArray(2).invoke(4, 3); 193 for (int i = 0; i < 4; i++) { 194 Object innerArr = arr43[i]; 195 for (int j = 0; i < 3; i++) { 196 Point p = new Point(i, (short) j, (short) 9); 197 MethodHandles.arrayElementSetter(VT.arrayValueClass()).invoke(innerArr, i, p); 198 } 199 } 200 for (int i = 0; i < 4; i++) { 201 Object innerArr = MethodHandles.arrayElementGetter(VT.arrayValueClass(2)).invoke(arr43, i); 202 for (int j = 0; i < 3; i++) { 203 String actual = (String) printReturn(MethodHandles.arrayElementGetter(VT.arrayValueClass())) 204 .invoke(innerArr, i); 205 String expected = TEMPLATE.replace("#x", String.valueOf(i)) 206 .replace("#y", String.valueOf(j)) 207 .replace("#z", "9"); 208 assertEquals(actual, expected); 209 } 210 } 211 } 212 213 public void testLoop() throws Throwable { 214 Object arr = MethodHandles.arrayConstructor(VT.arrayValueClass()).invoke(10); 215 for (int i = 0; i < 10; i++) { 216 Point p = new Point(i, (short) 9, (short) 9); 217 MethodHandles.arrayElementSetter(VT.arrayValueClass()).invoke(arr, i, p); 218 } 219 220 /* 221 iters -> (Point[] )int 222 223 init -> (Point[] )int 224 225 sum -> (int, int, int, int)int 226 a -> (int, Point, Point, Point)int 227 b -> (int, Point)int 228 c -> (int, Point[], int)int 229 body -> (int, int, Point[])int 230 */ 231 232 MethodHandle iters = MethodHandles.arrayLength(VT.arrayValueClass()); 233 234 MethodHandle init = MethodHandles.dropArguments(MethodHandles.constant(int.class, 0), 235 0, 236 VT.arrayValueClass()); 237 238 MethodHandle sum = LOOKUP.findStatic(MVTTest.class, 239 "sum", 240 methodType(int.class, int.class, int.class, short.class, short.class)); 241 242 MethodHandle a = MethodHandles.filterArguments(sum, 1, 243 VT.findGetter(LOOKUP, FIELD_NAMES[0], FIELD_TYPES[0]), 244 VT.findGetter(LOOKUP, FIELD_NAMES[1], FIELD_TYPES[1]), 245 VT.findGetter(LOOKUP, FIELD_NAMES[2], FIELD_TYPES[2])); 246 247 MethodHandle b = MethodHandles.permuteArguments(a, 248 methodType(int.class, int.class, VT.valueClass()), 249 0, 1, 1, 1); 250 251 MethodHandle c = MethodHandles.collectArguments(b, 252 1, 253 MethodHandles.arrayElementGetter(VT.arrayValueClass())); 254 255 MethodHandle body = MethodHandles.permuteArguments(c, 256 methodType(int.class, int.class, int.class, VT.arrayValueClass()), 257 0, 2, 1); 258 259 MethodHandle loop = MethodHandles.countedLoop(iters, init, body); 260 int actual = (int) loop.invoke(arr); 261 int expected = 9 * 10 * 2 + 10 * (0 + 9) / 2; 262 assertEquals(actual, expected); 263 } 264 265 static int sum(int v, int x, short y, short z) { 266 return v + x + y + z; 267 } 268 269 static MethodHandle printReturn(MethodHandle mh) { 270 return MethodHandles.filterReturnValue(mh, PRINT_POINT); 271 } 272 273 static String print(Point p) { 274 return String.format("Point[x=%d, y=%d, z=%d]", p.x, p.y, p.z); 275 } 276 }