// Copyright (c) 2018, 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.optimize.staticizer;

import com.android.tools.r8.graph.AppInfo;
import com.android.tools.r8.graph.DexEncodedMethod;
import com.android.tools.r8.graph.DexField;
import com.android.tools.r8.graph.DexItemFactory;
import com.android.tools.r8.graph.DexMethod;
import com.android.tools.r8.graph.GraphLense;
import com.android.tools.r8.graph.GraphLense.NestedGraphLense;
import com.android.tools.r8.ir.code.Invoke.Type;
import com.google.common.collect.BiMap;
import com.google.common.collect.ImmutableMap;
import java.util.Map;

class ClassStaticizerGraphLense extends NestedGraphLense {
  private final Map<DexEncodedMethod, DexEncodedMethod> staticizedMethods;

  ClassStaticizerGraphLense(
      GraphLense previous,
      DexItemFactory factory,
      BiMap<DexField, DexField> fieldMapping,
      BiMap<DexMethod, DexMethod> methodMapping,
      Map<DexEncodedMethod, DexEncodedMethod> encodedMethodMapping) {
    super(ImmutableMap.of(),
        methodMapping,
        fieldMapping,
        fieldMapping.inverse(),
        methodMapping.inverse(),
        previous,
        factory);
    staticizedMethods = encodedMethodMapping;
  }

  @Override
  protected Type mapInvocationType(
      DexMethod newMethod, DexMethod originalMethod,
      DexEncodedMethod context, Type type) {
    if (methodMap.get(originalMethod) == newMethod) {
      assert type == Type.VIRTUAL || type == Type.DIRECT;
      return Type.STATIC;
    }
    return super.mapInvocationType(newMethod, originalMethod, context, type);
  }

  @Override
  public DexEncodedMethod mapDexEncodedMethod(AppInfo appInfo, DexEncodedMethod original) {
    return super.mapDexEncodedMethod(appInfo, staticizedMethods.getOrDefault(original, original));
  }
}