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

// Copyright (c) 2019, 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.experimental.graphinfo.GraphConsumer;
import com.android.tools.r8.graph.AppInfoWithSubtyping;
import com.android.tools.r8.graph.AppView;
import com.android.tools.r8.graph.DexType;
import com.android.tools.r8.shaking.Enqueuer.Mode;
import java.util.Set;

public class EnqueuerFactory {

  public static Enqueuer createForInitialTreeShaking(
      AppView<? extends AppInfoWithSubtyping> appView) {
    return new Enqueuer(appView, null, Mode.INITIAL_TREE_SHAKING);
  }

  public static Enqueuer createForFinalTreeShaking(
      AppView<? extends AppInfoWithSubtyping> appView,
      GraphConsumer keptGraphConsumer,
      Set<DexType> initialMissingTypes) {
    Enqueuer enqueuer = new Enqueuer(appView, keptGraphConsumer, Mode.FINAL_TREE_SHAKING);
    appView.withProtoShrinker(
        shrinker -> enqueuer.setInitialDeadProtoTypes(shrinker.getDeadProtoTypes()));
    enqueuer.setInitialMissingTypes(initialMissingTypes);
    return enqueuer;
  }

  public static Enqueuer createForMainDexTracing(AppView<? extends AppInfoWithSubtyping> appView) {
    return createForMainDexTracing(appView, null);
  }

  public static Enqueuer createForMainDexTracing(
      AppView<? extends AppInfoWithSubtyping> appView, GraphConsumer keptGraphConsumer) {
    return new Enqueuer(appView, keptGraphConsumer, Mode.MAIN_DEX_TRACING);
  }

  public static Enqueuer createForWhyAreYouKeeping(
      AppView<? extends AppInfoWithSubtyping> appView, GraphConsumer keptGraphConsumer) {
    return new Enqueuer(appView, keptGraphConsumer, Mode.WHY_ARE_YOU_KEEPING);
  }
}