src/main/java/com/android/tools/r8/experimental/startup/StartupOrder.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.experimental.startup;

import com.android.tools.r8.experimental.startup.profile.StartupItem;
import com.android.tools.r8.experimental.startup.profile.StartupProfile;
import com.android.tools.r8.graph.AppView;
import com.android.tools.r8.graph.DexApplication;
import com.android.tools.r8.graph.DexDefinitionSupplier;
import com.android.tools.r8.graph.DexMethod;
import com.android.tools.r8.graph.DexReference;
import com.android.tools.r8.graph.DexType;
import com.android.tools.r8.graph.GraphLens;
import com.android.tools.r8.graph.PrunedItems;
import com.android.tools.r8.synthesis.SyntheticItems;
import com.android.tools.r8.utils.InternalOptions;
import java.util.Collection;
import java.util.LinkedHashMap;

public abstract class StartupOrder {

  StartupOrder() {}

  public static StartupOrder createInitialStartupOrder(
      InternalOptions options, DexDefinitionSupplier definitions) {
    StartupProfile startupProfile = StartupProfile.parseStartupProfile(options, definitions);
    if (startupProfile == null || startupProfile.getStartupItems().isEmpty()) {
      return empty();
    }
    LinkedHashMap<DexReference, StartupItem> startupItems =
        new LinkedHashMap<>(startupProfile.size());
    for (StartupItem startupItem : startupProfile.getStartupItems()) {
      startupItems.put(startupItem.getReference(), startupItem);
    }
    return new NonEmptyStartupOrder(new LinkedHashMap<>(startupItems));
  }

  public static StartupOrder createInitialStartupOrderForD8(AppView<?> appView) {
    return createInitialStartupOrder(appView.options(), appView);
  }

  public static StartupOrder createInitialStartupOrderForR8(DexApplication application) {
    return createInitialStartupOrder(application.options, application);
  }

  public static StartupOrder empty() {
    return new EmptyStartupOrder();
  }

  public abstract boolean contains(DexMethod method);

  public abstract boolean contains(DexType type);

  public abstract Collection<StartupItem> getItems();

  public abstract boolean isEmpty();

  public abstract StartupOrder rewrittenWithLens(GraphLens graphLens);

  public abstract StartupOrder toStartupOrderForWriting(AppView<?> appView);

  public abstract StartupOrder withoutPrunedItems(
      PrunedItems prunedItems, SyntheticItems syntheticItems);
}