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 jdk.incubator.code.OpTransformer;
28 import jdk.incubator.code.dialect.core.CoreOp;
29 import jdk.incubator.code.Op;
30 import jdk.incubator.code.bytecode.BytecodeGenerator;
31
32 import java.lang.invoke.MethodHandle;
33 import java.lang.invoke.MethodHandles;
34 import java.lang.reflect.Method;
35 import jdk.incubator.code.CodeReflection;
36 import java.util.Optional;
37 import java.util.stream.Stream;
38
39 /*
40 * @test
41 * @modules jdk.incubator.code
42 * @run testng TestSlots
43 */
44
45 public class TestSlots {
46 @CodeReflection
47 static double f(double i, double j) {
48 i = i + j;
49
50 double k = 4.0;
51 k += i;
52 return k;
53 }
54
55 @Test
56 public void testF() throws Throwable {
57 CoreOp.FuncOp f = getFuncOp("f");
58
59 MethodHandle mh = generate(f);
60
61 Assert.assertEquals(f(1.0d, 2.0d), (double) mh.invoke(1.0d, 2.0d));
62 }
63
64 @CodeReflection
65 static double f2(double x, double y) {
66 return x * (-Math.sin(x * y) + y) * 4.0d;
67 }
68
69 @Test
70 public void testF2() throws Throwable {
71 CoreOp.FuncOp f = getFuncOp("f2");
72
73 MethodHandle mh = generate(f);
74
75 Assert.assertEquals(f2(1.0d, 2.0d), (double) mh.invoke(1.0d, 2.0d));
76 }
77
78 @CodeReflection
79 static double f3(/* independent */ double x, int y) {
80 /* dependent */
81 double o = 1.0;
82 for (int i = 0; i < y; i = i + 1) {
83 if (i > 1) {
84 if (i < 5) {
85 o = o * x;
86 }
87 }
88 }
89 return o;
90 }
91
92 @Test
93 public void testF3() throws Throwable {
94 CoreOp.FuncOp f = getFuncOp("f3");
95
96 MethodHandle mh = generate(f);
97
98 for (int i = 0; i < 7; i++) {
99 Assert.assertEquals(f3(2.0d, i), (double) mh.invoke(2.0d, i));
100 }
101 }
102
103 @CodeReflection
104 static int f4(/* Unused */ int a, int b) {
105 return b;
106 }
107
108 @Test
109 public void testF4() throws Throwable {
110 CoreOp.FuncOp f = getFuncOp("f4");
111
112 MethodHandle mh;
113 try {
114 mh = generate(f);
115 } catch (VerifyError e) {
116 Assert.fail("invalid class file generated", e);
117 return;
118 }
119
120 Assert.assertEquals(f4(1, 2), (int) mh.invoke(1, 2));
121 }
122
123 @CodeReflection
124 static double f5(/* Unused */ double a, double b) {
125 return b;
126 }
127
128 @Test
129 public void testF5() throws Throwable {
130 CoreOp.FuncOp f = getFuncOp("f5");
131
132 MethodHandle mh;
133 try {
134 mh = generate(f);
135 } catch (VerifyError e) {
136 Assert.fail("invalid class file generated", e);
137 return;
138 }
139
140 Assert.assertEquals(f5(1.0, 2.0), (double) mh.invoke(1.0, 2.0));
141 }
142
143 static MethodHandle generate(CoreOp.FuncOp f) {
144 System.out.println(f.toText());
145
146 CoreOp.FuncOp lf = f.transform(OpTransformer.LOWERING_TRANSFORMER);
147 System.out.println(lf.toText());
148
149 return BytecodeGenerator.generate(MethodHandles.lookup(), lf);
150 }
151
152 static CoreOp.FuncOp getFuncOp(String name) {
153 Optional<Method> om = Stream.of(TestSlots.class.getDeclaredMethods())
154 .filter(m -> m.getName().equals(name))
155 .findFirst();
156
157 Method m = om.get();
158 return Op.ofMethod(m).get();
159 }
160
161 }