src/main/java/com/android/tools/r8/optimize/interfaces/collection/NonEmptyOpenClosedInterfacesCollection.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.optimize.interfaces.collection;

import com.android.tools.r8.graph.DexClass;
import com.android.tools.r8.graph.DexType;
import com.android.tools.r8.graph.PrunedItems;
import com.android.tools.r8.graph.lens.GraphLens;
import com.android.tools.r8.utils.SetUtils;
import com.android.tools.r8.utils.Timing;
import java.util.Set;

public class NonEmptyOpenClosedInterfacesCollection extends OpenClosedInterfacesCollection {

  private final Set<DexType> openInterfaceTypes;

  public NonEmptyOpenClosedInterfacesCollection(Set<DexType> openInterfaceTypes) {
    this.openInterfaceTypes = openInterfaceTypes;
  }

  @Override
  public boolean isDefinitelyClosed(DexClass clazz) {
    assert clazz.isInterface();
    return !openInterfaceTypes.contains(clazz.getType());
  }

  @Override
  public boolean isEmpty() {
    return openInterfaceTypes.isEmpty();
  }

  @Override
  public OpenClosedInterfacesCollection rewrittenWithLens(GraphLens graphLens, Timing timing) {
    return timing.time(
        "Rewrite NonEmptyOpenClosedInterfacesCollection", () -> rewrittenWithLens(graphLens));
  }

  private OpenClosedInterfacesCollection rewrittenWithLens(GraphLens graphLens) {
    Set<DexType> rewrittenOpenInterfaceTypes =
        SetUtils.newIdentityHashSet(openInterfaceTypes.size());
    for (DexType openInterfaceType : openInterfaceTypes) {
      rewrittenOpenInterfaceTypes.add(graphLens.lookupType(openInterfaceType));
    }
    return new NonEmptyOpenClosedInterfacesCollection(rewrittenOpenInterfaceTypes);
  }

  @Override
  public OpenClosedInterfacesCollection withoutPrunedItems(PrunedItems prunedItems, Timing timing) {
    if (!prunedItems.hasRemovedClasses()) {
      return this;
    }
    timing.begin("Prune NonEmptyOpenClosedInterfacesCollection");
    Set<DexType> prunedOpenInterfaceTypes = SetUtils.newIdentityHashSet(openInterfaceTypes.size());
    for (DexType openInterfaceType : openInterfaceTypes) {
      if (!prunedItems.isRemoved(openInterfaceType)) {
        prunedOpenInterfaceTypes.add(openInterfaceType);
      }
    }
    NonEmptyOpenClosedInterfacesCollection result =
        new NonEmptyOpenClosedInterfacesCollection(prunedOpenInterfaceTypes);
    timing.end();
    return result;
  }
}