src/main/java/com/android/tools/r8/shaking/ProguardIfRulePreconditionMatch.java

// Copyright (c) 2024, 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.shaking;

import com.android.tools.r8.graph.Definition;
import com.android.tools.r8.graph.DexClass;
import com.android.tools.r8.graph.DexClassAndMethod;
import com.android.tools.r8.shaking.RootSetUtils.ConsequentRootSetBuilder;
import com.android.tools.r8.utils.collections.DexClassAndMethodSet;

public class ProguardIfRulePreconditionMatch {

  private final ProguardIfRule ifRule;
  private final DexClass classMatch;
  private final DexClassAndMethodSet methodsMatch;

  public ProguardIfRulePreconditionMatch(ProguardIfRule ifRule, DexClass classMatch) {
    this(ifRule, classMatch, DexClassAndMethodSet.empty());
  }

  public ProguardIfRulePreconditionMatch(
      ProguardIfRule ifRule, DexClass classMatch, DexClassAndMethodSet methodsMatch) {
    this.ifRule = ifRule;
    this.classMatch = classMatch;
    this.methodsMatch = methodsMatch;
  }

  public ProguardIfRule getIfRuleWithPreconditionSet() {
    return ifRule.withPrecondition(classMatch);
  }

  public void disallowOptimizationsForReevaluation(
      Enqueuer enqueuer, ConsequentRootSetBuilder rootSetBuilder) {
    if (enqueuer.getMode().isInitialTreeShaking()
        && !ifRule.isTrivalAllClassMatch()
        && classMatch.isProgramClass()) {
      disallowClassOptimizationsForReevaluation(rootSetBuilder);
      disallowMethodOptimizationsForReevaluation(rootSetBuilder);
    }
  }

  private void disallowClassOptimizationsForReevaluation(ConsequentRootSetBuilder rootSetBuilder) {
    rootSetBuilder
        .getDependentMinimumKeepInfo()
        .getOrCreateUnconditionalMinimumKeepInfoFor(classMatch.getType())
        .asClassJoiner()
        .disallowClassInlining()
        .disallowHorizontalClassMerging()
        .disallowVerticalClassMerging();
  }

  private void disallowMethodOptimizationsForReevaluation(ConsequentRootSetBuilder rootSetBuilder) {
    if (classMatch.isProgramClass()) {
      for (DexClassAndMethod method : methodsMatch) {
        assert method.isProgramMethod();
        rootSetBuilder
            .getDependentMinimumKeepInfo()
            .getOrCreateUnconditionalMinimumKeepInfoFor(method.getReference())
            .asMethodJoiner()
            .disallowClassInlining()
            .disallowInlining();
      }
    } else {
      assert methodsMatch.stream().noneMatch(Definition::isProgramMethod);
    }
  }
}