1 /*
 2  * Copyright (c) 2024, 2026, 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 jdk.incubator.code.CodeContext;
25 import jdk.incubator.code.CodeTransformer;
26 import jdk.incubator.code.Op;
27 import jdk.incubator.code.Reflect;
28 import jdk.incubator.code.analysis.SSA;
29 import jdk.incubator.code.dialect.java.JavaOp;
30 import jdk.incubator.code.interpreter.Interpreter;
31 import org.junit.jupiter.api.Assertions;
32 import org.junit.jupiter.api.Test;
33 
34 import java.lang.invoke.MethodHandles;
35 import java.util.function.DoubleBinaryOperator;
36 import java.util.function.DoubleUnaryOperator;
37 
38 /*
39  * @test
40  * @modules jdk.incubator.code
41  * @run junit TestExpressionElimination
42  */
43 
44 public class TestExpressionElimination {
45 
46     @Test
47     public void testAddZero() {
48         JavaOp.LambdaOp lf = generate((@Reflect DoubleUnaryOperator) (double a) -> a + 0.0);
49 
50         Assertions.assertEquals(1.0d, (double) Interpreter.invoke(MethodHandles.lookup(), lf, 1.0d));
51     }
52 
53     @Test
54     public void testF() {
55         JavaOp.LambdaOp lf = generate((@Reflect DoubleBinaryOperator) (double a, double b) -> -a + b);
56 
57         Assertions.assertEquals(0.0d, (double) Interpreter.invoke(MethodHandles.lookup(), lf, 1.0d, 1.0d));
58     }
59 
60     static JavaOp.LambdaOp generate(Object q) {
61         return generateF(Op.ofLambda(q).get().op());
62     }
63 
64     static <T extends Op & Op.Invokable> T generateF(T f) {
65         System.out.println(f.toText());
66 
67         @SuppressWarnings("unchecked")
68         T lf = (T) f.transform(CodeContext.create(), CodeTransformer.LOWERING_TRANSFORMER);
69         System.out.println(lf.toText());
70 
71         lf = SSA.transform(lf);
72         System.out.println(lf.toText());
73 
74         lf = ExpressionElimination.eliminate(lf);
75         System.out.println(lf.toText());
76         return lf;
77     }
78 }