src/main/java/com/android/tools/r8/ir/conversion/OneTimeMethodProcessor.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.ir.conversion;

import com.android.tools.r8.contexts.CompilationContext.MethodProcessingContext;
import com.android.tools.r8.contexts.CompilationContext.ProcessorContext;
import com.android.tools.r8.graph.AppView;
import com.android.tools.r8.graph.ProgramMethod;
import com.android.tools.r8.utils.ThreadUtils;
import com.android.tools.r8.utils.collections.SortedProgramMethodSet;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;

/**
 * A {@link MethodProcessor} that doesn't persist; rather just processes the given methods one-time,
 * along with a default abstraction of concurrent processing.
 */
public class OneTimeMethodProcessor extends MethodProcessorWithWave {

  private final ProcessorContext processorContext;

  private OneTimeMethodProcessor(ProcessorContext processorContext, SortedProgramMethodSet wave) {
    this.processorContext = processorContext;
    this.wave = wave;
  }

  public static Builder builder(ProcessorContext processorContext) {
    return new Builder(processorContext);
  }

  public static OneTimeMethodProcessor create(ProgramMethod methodToProcess, AppView<?> appView) {
    return create(SortedProgramMethodSet.create(methodToProcess), appView);
  }

  public static OneTimeMethodProcessor create(
      ProgramMethod methodToProcess, ProcessorContext processorContext) {
    return create(SortedProgramMethodSet.create(methodToProcess), processorContext);
  }

  public static OneTimeMethodProcessor create(
      SortedProgramMethodSet methodsToProcess, AppView<?> appView) {
    return create(methodsToProcess, appView.createProcessorContext());
  }

  public static OneTimeMethodProcessor create(
      SortedProgramMethodSet methodsToProcess, ProcessorContext processorContext) {
    return new OneTimeMethodProcessor(processorContext, methodsToProcess);
  }

  @Override
  public MethodProcessingContext createMethodProcessingContext(ProgramMethod method) {
    return processorContext.createMethodProcessingContext(method);
  }

  @Override
  public boolean shouldApplyCodeRewritings(ProgramMethod method) {
    return true;
  }

  @FunctionalInterface
  public interface MethodAction<E extends Exception> {
    void accept(ProgramMethod method, MethodProcessingContext methodProcessingContext) throws E;
  }

  public <E extends Exception> void forEachWaveWithExtension(MethodAction<E> consumer) throws E {
    while (!wave.isEmpty()) {
      for (ProgramMethod method : wave) {
        consumer.accept(method, processorContext.createMethodProcessingContext(method));
      }
      prepareForWaveExtensionProcessing();
    }
  }

  public <E extends Exception> void forEachWaveWithExtension(
      MethodAction<E> consumer, ExecutorService executorService) throws ExecutionException {
    while (!wave.isEmpty()) {
      ThreadUtils.processItems(
          wave,
          method -> consumer.accept(method, processorContext.createMethodProcessingContext(method)),
          executorService);
      prepareForWaveExtensionProcessing();
    }
  }

  public static class Builder {

    private final SortedProgramMethodSet methodsToProcess = SortedProgramMethodSet.create();
    private final ProcessorContext processorContext;

    Builder(ProcessorContext processorContext) {
      this.processorContext = processorContext;
    }

    public Builder add(ProgramMethod methodToProcess) {
      methodsToProcess.add(methodToProcess);
      return this;
    }

    public OneTimeMethodProcessor build() {
      return create(methodsToProcess, processorContext);
    }
  }
}