| // Copyright (c) 2023, the R8 project authors. Please see the AUTHORS file |
| // for details. All rights reserved. Use of this source code is governed by a |
| // BSD-style license that can be found in the LICENSE file. |
| |
| package com.android.tools.r8.enumunboxing; |
| |
| import static org.junit.Assert.assertEquals; |
| import static org.junit.Assert.assertTrue; |
| |
| import com.android.tools.r8.NeverClassInline; |
| import com.android.tools.r8.NeverInline; |
| import com.android.tools.r8.NoHorizontalClassMerging; |
| import com.android.tools.r8.TestParameters; |
| import com.android.tools.r8.TestParametersCollection; |
| import com.android.tools.r8.utils.codeinspector.ClassSubject; |
| import com.android.tools.r8.utils.codeinspector.CodeInspector; |
| import com.android.tools.r8.utils.codeinspector.MethodSubject; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.junit.runners.Parameterized; |
| import org.junit.runners.Parameterized.Parameter; |
| import org.junit.runners.Parameterized.Parameters; |
| |
| @RunWith(Parameterized.class) |
| public class EnumUnboxingMappingTest extends EnumUnboxingTestBase { |
| |
| @Parameter(0) |
| public TestParameters parameters; |
| |
| @Parameters(name = "{0}") |
| public static TestParametersCollection data() { |
| return getTestParameters().withDexRuntimes().withAllApiLevels().build(); |
| } |
| |
| @Test |
| public void testEnumUnboxing() throws Exception { |
| testForR8(parameters.getBackend()) |
| .addInnerClasses(getClass()) |
| .addKeepMainRule(Main.class) |
| .addKeepRules(EnumKeepRules.STUDIO.getKeepRules()) |
| .addEnumUnboxingInspector(inspector -> inspector.assertUnboxed(MyEnum.class)) |
| .enableNeverClassInliningAnnotations() |
| .enableNoHorizontalClassMergingAnnotations() |
| .enableInliningAnnotations() |
| .allowDiagnosticMessages() |
| .setMinApi(parameters) |
| .compile() |
| .inspect(this::assertParameterTypes) |
| .run(parameters.getRuntime(), Main.class) |
| .assertSuccessWithOutputLines( |
| "1", "2", |
| "DebugInfoForThisPrint", "1", |
| "DebugInfoForThisPrint", "2"); |
| } |
| |
| private void assertParameterTypes(CodeInspector codeInspector) { |
| ClassSubject main = codeInspector.clazz(Main.class); |
| |
| MethodSubject debugInfoMethod = main.uniqueMethodWithOriginalName("debugInfoAfterUnboxing"); |
| MethodSubject noDebugInfoMethod = main.uniqueMethodWithOriginalName("noDebugInfoAfterUnboxing"); |
| |
| assertEquals("int", debugInfoMethod.getFinalSignature().asMethodSignature().parameters[0]); |
| assertEquals("int", noDebugInfoMethod.getFinalSignature().asMethodSignature().parameters[0]); |
| |
| assertEquals(MyEnum.class.getTypeName(), debugInfoMethod.getOriginalSignature().parameters[0]); |
| assertEquals( |
| MyEnum.class.getTypeName(), noDebugInfoMethod.getOriginalSignature().parameters[0]); |
| |
| ClassSubject indirection = codeInspector.clazz(Indirection.class); |
| MethodSubject abstractMethod = indirection.uniqueMethodWithOriginalName("intermediate"); |
| assertTrue(abstractMethod.isAbstract()); |
| assertEquals(MyEnum.class.getTypeName(), abstractMethod.getOriginalSignature().parameters[0]); |
| } |
| |
| @NeverClassInline |
| enum MyEnum { |
| A, |
| B |
| } |
| |
| @NeverClassInline |
| abstract static class Indirection { |
| |
| @NeverInline |
| public abstract int intermediate(MyEnum e); |
| } |
| |
| @NoHorizontalClassMerging |
| @NeverClassInline |
| static class A extends Indirection { |
| |
| @Override |
| public int intermediate(MyEnum e) { |
| return Main.noDebugInfoAfterUnboxing(e); |
| } |
| } |
| |
| @NoHorizontalClassMerging |
| @NeverClassInline |
| static class B extends Indirection { |
| |
| @Override |
| public int intermediate(MyEnum e) { |
| return Main.debugInfoAfterUnboxing(e); |
| } |
| } |
| |
| static class Main { |
| |
| public static void main(String[] args) { |
| Indirection indirection1 = new A(); |
| Indirection indirection2 = new B(); |
| Indirection i = System.nanoTime() > 0 ? indirection1 : indirection2; |
| System.out.println(i.intermediate(MyEnum.A)); |
| System.out.println(i.intermediate(null)); |
| i = System.nanoTime() < 0 ? indirection1 : indirection2; |
| System.out.println(i.intermediate(MyEnum.A)); |
| System.out.println(i.intermediate(null)); |
| } |
| |
| @NeverInline |
| static int noDebugInfoAfterUnboxing(MyEnum e) { |
| return (e == null ? 1 : 0) + 1; |
| } |
| |
| @NeverInline |
| static int debugInfoAfterUnboxing(MyEnum e) { |
| System.out.println("DebugInfoForThisPrint"); |
| return (e == null ? 1 : 0) + 1; |
| } |
| } |
| } |