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 TestVarOp 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.Op; 34 import java.lang.reflect.code.OpTransformer; 35 import java.lang.reflect.code.Value; 36 import java.lang.reflect.code.op.CoreOp; 37 import java.lang.reflect.code.parser.OpParser; 38 import java.lang.reflect.code.type.CoreTypeFactory; 39 import java.lang.reflect.code.type.FunctionType; 40 import java.lang.reflect.code.type.JavaType; 41 import java.lang.runtime.CodeReflection; 42 import java.util.List; 43 import java.util.Optional; 44 import java.util.stream.Stream; 45 46 public class TestVarOp { 47 48 @CodeReflection 49 static Object f(String s) { 50 Object o = s; 51 return o; 52 } 53 54 @Test 55 public void testTypeSubstitutionAndPreserve() { 56 CoreOp.FuncOp f = getFuncOp("f"); 57 CoreOp.FuncOp ft = CoreOp.func("f", FunctionType.functionType(JavaType.J_L_OBJECT, JavaType.type(CharSequence.class))) 58 .body(fb -> { 59 fb.transformBody(f.body(), fb.parameters(), OpTransformer.COPYING_TRANSFORMER); 60 }); 61 62 List<CoreOp.VarOp> vops = ft.elements() 63 .flatMap(ce -> ce instanceof CoreOp.VarOp vop ? Stream.of(vop) : null) 64 .toList(); 65 // VarOp for block parameter, translate from String to CharSequence 66 Assert.assertEquals(vops.get(0).resultType().valueType(), JavaType.type(CharSequence.class)); 67 // VarOp for local variable, preserve Object 68 Assert.assertEquals(vops.get(1).resultType().valueType(), JavaType.J_L_OBJECT); 69 } 70 71 @Test 72 public void testNoName() { 73 CoreOp.FuncOp f = getFuncOp("f"); 74 f = f.transform((block, op) -> { 75 if (op instanceof CoreOp.VarOp vop) { 76 Value init = block.context().getValue(vop.initOperand()); 77 Op.Result v = block.op(CoreOp.var(init)); 78 block.context().mapValue(vop.result(), v); 79 } else { 80 block.op(op); 81 } 82 return block; 83 }); 84 85 Op op = OpParser.fromString(CoreOp.FACTORY, CoreTypeFactory.CORE_TYPE_FACTORY, f.toText()).get(0); 86 boolean allNullNames = op.elements() 87 .flatMap(ce -> ce instanceof CoreOp.VarOp vop ? Stream.of(vop) : null) 88 .allMatch(vop -> vop.varName() == null); 89 Assert.assertTrue(allNullNames); 90 } 91 92 static CoreOp.FuncOp getFuncOp(String name) { 93 Optional<Method> om = Stream.of(TestVarOp.class.getDeclaredMethods()) 94 .filter(m -> m.getName().equals(name)) 95 .findFirst(); 96 97 Method m = om.get(); 98 return m.getCodeModel().get(); 99 } 100 }