// Copyright (c) 2020, 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.horizontalclassmerging.policies;

import static com.android.tools.r8.utils.IteratorUtils.createCircularIterator;

import com.android.tools.r8.graph.DexProgramClass;
import com.android.tools.r8.horizontalclassmerging.MergeGroup;
import com.android.tools.r8.horizontalclassmerging.MultiClassPolicy;
import com.google.common.collect.ImmutableList;
import java.util.Collection;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;

public class NoClassAnnotationCollisions extends MultiClassPolicy {

  @Override
  public Collection<MergeGroup> apply(MergeGroup group) {
    // Create a new merge group for each class that has annotations.
    List<MergeGroup> newGroups = new LinkedList<>();
    for (DexProgramClass clazz : group) {
      if (clazz.hasAnnotations()) {
        newGroups.add(new MergeGroup(clazz));
      }
    }

    // If there were at most one class with annotations, then just return the original merge group.
    if (newGroups.size() <= 1) {
      return ImmutableList.of(group);
    }

    // Otherwise, fill up the new merge groups with the classes that do not have annotations.
    Iterator<MergeGroup> newGroupsIterator = createCircularIterator(newGroups);
    for (DexProgramClass clazz : group) {
      if (!clazz.hasAnnotations()) {
        newGroupsIterator.next().add(clazz);
      }
    }
    return removeTrivialGroups(newGroups);
  }

  @Override
  public String getName() {
    return "NoClassAnnotationCollisions";
  }
}