// 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.desugar.backports;

import com.android.tools.r8.graph.DexItemFactory;
import com.android.tools.r8.ir.code.InstructionListIterator;
import com.android.tools.r8.ir.code.InvokeMethod;
import com.android.tools.r8.ir.code.InvokeVirtual;
import com.android.tools.r8.ir.code.Value;
import java.util.Set;

public final class OptionalMethodRewrites {

  private OptionalMethodRewrites() {}

  public static void rewriteOrElseGet(
      InvokeMethod invoke,
      InstructionListIterator iterator,
      DexItemFactory factory,
      Set<Value> affectedValues) {
    InvokeVirtual getInvoke = new InvokeVirtual(
        factory.createMethod(factory.optionalType, invoke.getInvokedMethod().proto,
            "get"), invoke.outValue(), invoke.inValues());
    iterator.replaceCurrentInstruction(getInvoke);
  }

  public static void rewriteDoubleOrElseGet(
      InvokeMethod invoke,
      InstructionListIterator iterator,
      DexItemFactory factory,
      Set<Value> affectedValues) {
    InvokeVirtual getInvoke = new InvokeVirtual(
        factory.createMethod(factory.optionalDoubleType, invoke.getInvokedMethod().proto,
            "getAsDouble"), invoke.outValue(), invoke.inValues());
    iterator.replaceCurrentInstruction(getInvoke);
  }

  public static void rewriteIntOrElseGet(
      InvokeMethod invoke,
      InstructionListIterator iterator,
      DexItemFactory factory,
      Set<Value> affectedValues) {
    InvokeVirtual getInvoke = new InvokeVirtual(
        factory.createMethod(factory.optionalIntType, invoke.getInvokedMethod().proto,
            "getAsInt"), invoke.outValue(), invoke.inValues());
    iterator.replaceCurrentInstruction(getInvoke);
  }

  public static void rewriteLongOrElseGet(
      InvokeMethod invoke,
      InstructionListIterator iterator,
      DexItemFactory factory,
      Set<Value> affectedValues) {
    InvokeVirtual getInvoke = new InvokeVirtual(
        factory.createMethod(factory.optionalLongType, invoke.getInvokedMethod().proto,
            "getAsLong"), invoke.outValue(), invoke.inValues());
    iterator.replaceCurrentInstruction(getInvoke);
  }
}