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 import java.lang.invoke.MethodHandle;
 32 import java.lang.invoke.MethodHandles;
 33 import java.lang.reflect.Method;
 34 import jdk.incubator.code.CodeReflection;
 35 import java.util.Optional;
 36 import java.util.stream.Stream;
 37 
 38 /*
 39  * @test
 40  * @modules jdk.incubator.code
 41  * @run testng TestArrayCreation
 42  */
 43 
 44 public class TestArrayCreation {
 45     @CodeReflection
 46     public static String[] f() {
 47         return new String[10];
 48     }
 49 
 50     @Test
 51     public void testf() throws Throwable {
 52         CoreOp.FuncOp f = getFuncOp("f");
 53 
 54         MethodHandle mh = generate(f);
 55 
 56         Assert.assertEquals((String[]) mh.invoke(), f());
 57     }
 58 
 59     @CodeReflection
 60     public static String[][] f2() {
 61         return new String[10][];
 62     }
 63 
 64     @Test
 65     public void testf2() throws Throwable {
 66         CoreOp.FuncOp f = getFuncOp("f2");
 67 
 68         MethodHandle mh = generate(f);
 69 
 70         Assert.assertEquals((String[][]) mh.invoke(), f2());
 71     }
 72 
 73     @CodeReflection
 74     public static String[][] f3() {
 75         return new String[10][10];
 76     }
 77 
 78     @Test
 79     public void testf3() throws Throwable {
 80         CoreOp.FuncOp f = getFuncOp("f3");
 81 
 82         MethodHandle mh = generate(f);
 83 
 84         Assert.assertEquals((String[][]) mh.invoke(), f3());
 85     }
 86 
 87     @CodeReflection
 88     public static String[][] f4() {
 89         return new String[][]{{"one", "two"}, {"three"}};
 90     }
 91 
 92     @Test
 93     public void testf4() throws Throwable {
 94         CoreOp.FuncOp f = getFuncOp("f4");
 95 
 96         MethodHandle mh = generate(f);
 97 
 98         Assert.assertEquals((String[][]) mh.invoke(), f4());
 99     }
100 
101     static MethodHandle generate(CoreOp.FuncOp f) {
102         f.writeTo(System.out);
103 
104         CoreOp.FuncOp lf = f.transform(OpTransformer.LOWERING_TRANSFORMER);
105         lf.writeTo(System.out);
106 
107         return BytecodeGenerator.generate(MethodHandles.lookup(), lf);
108     }
109 
110     static CoreOp.FuncOp getFuncOp(String name) {
111         Optional<Method> om = Stream.of(TestArrayCreation.class.getDeclaredMethods())
112                 .filter(m -> m.getName().equals(name))
113                 .findFirst();
114 
115         Method m = om.get();
116         return Op.ofMethod(m).get();
117     }
118 }