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

// Copyright (c) 2022, 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.AppInfoWithClassHierarchy;
import com.android.tools.r8.graph.AppView;
import com.android.tools.r8.graph.DexField;
import com.android.tools.r8.graph.FieldResolutionResult;
import com.android.tools.r8.graph.ProgramField;
import com.android.tools.r8.graph.ProgramMethod;
import com.android.tools.r8.graph.analysis.EnqueuerAnalysisCollection;
import com.android.tools.r8.shaking.Enqueuer.FieldAccessKind;
import com.android.tools.r8.shaking.Enqueuer.FieldAccessMetadata;
import com.android.tools.r8.shaking.Enqueuer.Mode;
import com.android.tools.r8.utils.InternalOptions;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;

public abstract class EnqueuerDeferredTracing {

  public static EnqueuerDeferredTracing create(
      AppView<? extends AppInfoWithClassHierarchy> appView, Enqueuer enqueuer, Mode mode) {
    InternalOptions options = appView.options();
    if (!options.isShrinking()) {
      return empty();
    }
    if (!options.isOptimizedResourceShrinking()) {
      if (!options.isOptimizing() || mode.isInitialTreeShaking()) {
        return empty();
      }
    }
    return new EnqueuerDeferredTracingImpl(appView, enqueuer);
  }

  public static EmptyEnqueuerDeferredTracing empty() {
    return new EmptyEnqueuerDeferredTracing();
  }

  /**
   * @return true if the {@link Enqueuer} should not trace the given field reference.
   *     <p>If for some reason the field reference should be traced after all, a worklist item will
   *     be enqueued upon reaching a (preliminary) fixpoint in {@link
   *     #enqueueWorklistActions(EnqueuerWorklist)}, which will cause tracing to continue.
   */
  public abstract boolean deferTracingOfFieldAccess(
      DexField fieldReference,
      FieldResolutionResult resolutionResult,
      ProgramMethod context,
      FieldAccessKind accessKind,
      FieldAccessMetadata metadata);

  public abstract void register(EnqueuerAnalysisCollection.Builder analysesBuilder);

  /**
   * Called when tree shaking has ended, to allow rewriting the application according to the tracing
   * that has not been performed (e.g., rewriting of dead field instructions).
   */
  public abstract void rewriteApplication(ExecutorService executorService)
      throws ExecutionException;

  public abstract void notifyReflectiveFieldAccess(ProgramField field, ProgramMethod context);
}