| // Copyright (c) 2020, 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 com.android.tools.r8.utils.codeinspector.Matchers.isPresent; |
| import static junit.framework.TestCase.assertEquals; |
| import static org.hamcrest.MatcherAssert.assertThat; |
| |
| import com.android.tools.r8.NeverClassInline; |
| import com.android.tools.r8.NeverInline; |
| import com.android.tools.r8.TestParameters; |
| import com.android.tools.r8.utils.codeinspector.CodeInspector; |
| import com.android.tools.r8.utils.codeinspector.MethodSubject; |
| import java.util.List; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.junit.runners.Parameterized; |
| |
| @RunWith(Parameterized.class) |
| public class EnumUnboxingClassStaticizerTest extends EnumUnboxingTestBase { |
| |
| private final TestParameters parameters; |
| private final boolean enumValueOptimization; |
| private final EnumKeepRules enumKeepRules; |
| |
| @Parameterized.Parameters(name = "{0} valueOpt: {1} keep: {2}") |
| public static List<Object[]> data() { |
| return enumUnboxingTestParameters(); |
| } |
| |
| public EnumUnboxingClassStaticizerTest( |
| TestParameters parameters, boolean enumValueOptimization, EnumKeepRules enumKeepRules) { |
| this.parameters = parameters; |
| this.enumValueOptimization = enumValueOptimization; |
| this.enumKeepRules = enumKeepRules; |
| } |
| |
| @Test |
| public void testEnumUnboxing() throws Exception { |
| testForR8(parameters.getBackend()) |
| .addInnerClasses(EnumUnboxingClassStaticizerTest.class) |
| .addKeepMainRule(TestClass.class) |
| .addKeepRules(enumKeepRules.getKeepRules()) |
| .addEnumUnboxingInspector(inspector -> inspector.assertUnboxed(UnboxableEnum.class)) |
| .addHorizontallyMergedClassesInspector( |
| inspector -> |
| inspector |
| .assertMergedInto(CompanionHost.class, Companion.class) |
| .assertNoOtherClassesMerged()) |
| .enableNeverClassInliningAnnotations() |
| .enableInliningAnnotations() |
| .addDontObfuscate() // For assertions. |
| .addOptionsModification(opt -> enableEnumOptions(opt, enumValueOptimization)) |
| .setMinApi(parameters.getApiLevel()) |
| .compile() |
| .inspect(this::assertClassStaticized) |
| .run(parameters.getRuntime(), TestClass.class) |
| .assertSuccess() |
| .inspectStdOut(this::assertLines2By2Correct); |
| } |
| |
| private void assertClassStaticized(CodeInspector codeInspector) { |
| String renamedMethodName = "method$enumunboxing$"; |
| if (parameters.isCfRuntime()) { |
| // There is no class staticizer in Cf. |
| assertThat( |
| codeInspector.clazz(Companion.class).uniqueMethodWithOriginalName(renamedMethodName), |
| isPresent()); |
| return; |
| } |
| MethodSubject method = |
| codeInspector.clazz(Companion.class).uniqueMethodWithOriginalName("method"); |
| assertThat(method, isPresent()); |
| assertEquals("int", method.getMethod().getParameters().toString()); |
| } |
| |
| static class TestClass { |
| public static void main(String[] args) { |
| CompanionHost.toKeep(); |
| test(UnboxableEnum.A); |
| System.out.println("0"); |
| test(UnboxableEnum.B); |
| System.out.println("1"); |
| test(UnboxableEnum.C); |
| System.out.println("2"); |
| } |
| |
| @NeverInline |
| static void test(UnboxableEnum obj) { |
| // The class staticizer will move Companion.method() into CompanionHost.method(). |
| // As a result of this transformation, this call site will need to be processed. |
| // To keep track of this, the staticizer will keep a reference to the method |
| // `void test(UnboxableEnum)`, but after enum unboxing this has been rewritten to |
| // `void test(int)`. Therefore, the staticizer's internal structure needs to be |
| // updated after enum unboxing. |
| CompanionHost.COMPANION.method(obj); |
| } |
| } |
| |
| @NeverClassInline |
| static class CompanionHost { |
| static final Companion COMPANION = new Companion(); |
| |
| @NeverInline |
| static void toKeep() { |
| System.out.println("Keep me!"); |
| System.out.println("Keep me!"); |
| } |
| } |
| |
| @NeverClassInline |
| static class Companion { |
| @NeverInline |
| public void method(UnboxableEnum obj) { |
| System.out.println(obj.ordinal()); |
| } |
| } |
| |
| @NeverClassInline |
| enum UnboxableEnum { |
| A, |
| B, |
| C; |
| } |
| } |