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