1 /* 2 * Copyright (c) 2024, 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 import org.testng.Assert; 25 import org.testng.annotations.Test; 26 27 import java.lang.reflect.code.OpTransformer; 28 import java.lang.reflect.code.op.CoreOp; 29 import java.lang.reflect.code.Op; 30 import java.lang.reflect.code.interpreter.Interpreter; 31 import java.lang.reflect.Method; 32 import java.lang.runtime.CodeReflection; 33 import java.util.List; 34 import java.util.Optional; 35 import java.util.stream.Stream; 36 37 /* 38 * @test 39 * @run testng TestEnhancedForOp 40 */ 41 42 public class TestEnhancedForOp { 43 44 @CodeReflection 45 public static int f() { 46 int j = 0; 47 for (int i : List.of(1, 2, 3, 4)) { 48 j += i; 49 } 50 return j; 51 } 52 53 static CoreOp.FuncOp getFuncOp(String name) { 54 Optional<Method> om = Stream.of(TestEnhancedForOp.class.getDeclaredMethods()) 55 .filter(m -> m.getName().equals(name)) 56 .findFirst(); 57 58 Method m = om.get(); 59 return m.getCodeModel().get(); 60 } 61 62 @Test 63 public void testf() { 64 CoreOp.FuncOp f = getFuncOp("f"); 65 66 f.writeTo(System.out); 67 68 CoreOp.FuncOp lf = f.transform(OpTransformer.LOWERING_TRANSFORMER); 69 70 lf.writeTo(System.out); 71 72 Assert.assertEquals(Interpreter.invoke(lf), f()); 73 } 74 75 76 @CodeReflection 77 public static int array(int[] a) { 78 int j = 0; 79 for (int i : a) { 80 j += i; 81 } 82 return j; 83 } 84 85 @Test 86 public void testArray() { 87 CoreOp.FuncOp f = getFuncOp("array"); 88 89 f.writeTo(System.out); 90 91 CoreOp.FuncOp lf = f.transform(OpTransformer.LOWERING_TRANSFORMER); 92 93 lf.writeTo(System.out); 94 95 int[] ia = new int[] {1, 2, 3, 4}; 96 Assert.assertEquals(Interpreter.invoke(lf, ia), array(ia)); 97 } 98 }