// Copyright (c) 2023, 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.verticalclassmerging;

import com.android.tools.r8.classmerging.ClassMergerSharedData;
import com.android.tools.r8.graph.AppView;
import com.android.tools.r8.shaking.AppInfoWithLiveness;
import com.android.tools.r8.utils.ListUtils;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Comparator;
import java.util.List;

public class ConnectedComponentVerticalClassMerger {

  private final AppView<AppInfoWithLiveness> appView;
  private final Collection<VerticalMergeGroup> groups;

  // The resulting graph lens that should be used after class merging.
  private final VerticalClassMergerGraphLens.Builder lensBuilder;

  // All the bridge methods that have been synthesized during vertical class merging.
  private final List<IncompleteVerticalClassMergerBridgeCode> synthesizedBridges =
      new ArrayList<>();

  private final VerticallyMergedClasses.Builder verticallyMergedClassesBuilder =
      VerticallyMergedClasses.builder();

  ConnectedComponentVerticalClassMerger(
      AppView<AppInfoWithLiveness> appView, Collection<VerticalMergeGroup> groups) {
    this.appView = appView;
    this.groups = groups;
    this.lensBuilder = new VerticalClassMergerGraphLens.Builder();
  }

  public boolean isEmpty() {
    return groups.isEmpty();
  }

  public VerticalClassMergerResult.Builder run(ClassMergerSharedData sharedData) {
    List<VerticalMergeGroup> groupsSorted =
        ListUtils.sort(groups, Comparator.comparing(group -> group.getSource().getType()));
    List<ClassMerger> classMergers =
        ListUtils.map(
            groupsSorted,
            group ->
                new ClassMerger(
                    appView,
                    lensBuilder,
                    sharedData,
                    synthesizedBridges,
                    verticallyMergedClassesBuilder,
                    group));
    classMergers.forEach(ClassMerger::setup);
    classMergers.forEach(ClassMerger::merge);
    return VerticalClassMergerResult.builder(
        lensBuilder, synthesizedBridges, verticallyMergedClassesBuilder);
  }
}