// Copyright (c) 2021, 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.graph.analysis;

import com.android.tools.r8.graph.AppView;
import com.android.tools.r8.graph.DexEncodedMember;
import com.android.tools.r8.graph.DexMember;
import com.android.tools.r8.graph.DexReference;
import com.android.tools.r8.graph.ProgramDefinition;
import com.android.tools.r8.graph.ProgramField;
import com.android.tools.r8.graph.ProgramMethod;
import com.android.tools.r8.ir.optimize.info.DefaultFieldOptimizationWithMinApiInfo;
import com.android.tools.r8.ir.optimize.info.DefaultMethodOptimizationWithMinApiInfo;
import com.android.tools.r8.ir.optimize.info.MemberOptimizationInfo;
import com.android.tools.r8.shaking.DefaultEnqueuerUseRegistry;
import com.android.tools.r8.utils.AndroidApiLevel;
import java.util.Map;

public class ApiModelAnalysis extends EnqueuerAnalysis {

  private final AppView<?> appView;
  private final AndroidApiLevel minApiLevel;
  private final Map<DexReference, AndroidApiLevel> referenceToApiLevelMap;

  public ApiModelAnalysis(
      AppView<?> appView, Map<DexReference, AndroidApiLevel> referenceToApiLevelMap) {
    this.appView = appView;
    this.minApiLevel = appView.options().minApiLevel;
    this.referenceToApiLevelMap = referenceToApiLevelMap;
  }

  @Override
  public void processNewlyLiveField(ProgramField field, ProgramDefinition context) {
    setApiLevelForMember(
        field.getDefinition(), computeApiLevelForReferencedTypes(field.getReference()));
  }

  @Override
  public void processNewlyLiveMethod(ProgramMethod method, ProgramDefinition context) {
    setApiLevelForMember(
        method.getDefinition(), computeApiLevelForReferencedTypes(method.getReference()));
  }

  @Override
  public void processTracedCode(ProgramMethod method, DefaultEnqueuerUseRegistry registry) {
    assert registry.getMaxApiReferenceLevel().isGreaterThanOrEqualTo(minApiLevel);
    setApiLevelForMember(method.getDefinition(), registry.getMaxApiReferenceLevel());
  }

  private void setApiLevelForMember(DexEncodedMember<?, ?> member, AndroidApiLevel apiLevel) {
    // To not have mutable update information for all members that all has min api level we
    // swap the default optimization info for one with that marks the api level to be min api.
    MemberOptimizationInfo<?> optimizationInfo = member.getOptimizationInfo();
    if (!optimizationInfo.isMutableOptimizationInfo() && apiLevel == minApiLevel) {
      member.apply(
          field -> {
            field.setMinApiOptimizationInfo(DefaultFieldOptimizationWithMinApiInfo.getInstance());
          },
          method -> {
            method.setMinApiOptimizationInfo(DefaultMethodOptimizationWithMinApiInfo.getInstance());
          });
    } else {
      AndroidApiLevel maxApiLevel =
          optimizationInfo.hasApiReferenceLevel()
              ? apiLevel.max(optimizationInfo.getApiReferenceLevel(minApiLevel))
              : apiLevel;
      member.apply(
          field -> {
            field.getMutableOptimizationInfo().setApiReferenceLevel(maxApiLevel);
          },
          method -> {
            method.getMutableOptimizationInfo().setApiReferenceLevel(maxApiLevel);
          });
    }
  }

  private AndroidApiLevel computeApiLevelForReferencedTypes(DexMember<?, ?> member) {
    return member.computeApiLevelForReferencedTypes(appView, referenceToApiLevelMap);
  }
}