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 /* 25 * @test 26 * @run testng TestTraverse 27 */ 28 29 import org.testng.Assert; 30 import org.testng.annotations.Test; 31 32 import java.lang.reflect.Method; 33 import java.lang.reflect.code.CodeElement; 34 import java.lang.reflect.code.Op; 35 import java.lang.reflect.code.OpTransformer; 36 import java.lang.reflect.code.analysis.SSA; 37 import java.lang.reflect.code.op.CoreOp; 38 import java.lang.runtime.CodeReflection; 39 import java.util.ArrayList; 40 import java.util.List; 41 import java.util.Optional; 42 import java.util.stream.Stream; 43 44 public class TestTraverse { 45 46 @CodeReflection 47 private static int f(String s, int i, List<Object> acc) { 48 char c = s.charAt(i); 49 int d = (c - '0'); 50 int n = s.length(); 51 while (++i < n) { 52 c = s.charAt(i); 53 if (c >= '0' && c <= '9') { 54 d = d * 10 + (c - '0'); 55 continue; 56 } 57 break; 58 } 59 acc.add(d); 60 return i; 61 } 62 63 @Test 64 public void test() { 65 CoreOp.FuncOp f = getFuncOp("f"); 66 testTraverse(f); 67 68 f = f.transform(OpTransformer.LOWERING_TRANSFORMER); 69 testTraverse(f); 70 71 f = SSA.transform(f); 72 testTraverse(f); 73 } 74 75 void testTraverse(Op op) { 76 List<CodeElement<?, ?>> tl = op.traverse(new ArrayList<>(), (l, e) -> { 77 l.add(e); 78 return l; 79 }); 80 Assert.assertEquals(op.elements().toList(), tl); 81 82 Assert.assertEquals(op.elements().limit(2).toList(), tl.subList(0, 2)); 83 } 84 85 static CoreOp.FuncOp getFuncOp(String name) { 86 Optional<Method> om = Stream.of(TestTraverse.class.getDeclaredMethods()) 87 .filter(m -> m.getName().equals(name)) 88 .findFirst(); 89 90 Method m = om.get(); 91 return m.getCodeModel().get(); 92 } 93 }