// Copyright (c) 2020, 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.horizontalclassmerging.policies;

import com.android.tools.r8.graph.AppView;
import com.android.tools.r8.graph.DexProgramClass;
import com.android.tools.r8.horizontalclassmerging.HorizontalClassMerger.Mode;
import com.android.tools.r8.horizontalclassmerging.SingleClassPolicy;
import com.android.tools.r8.shaking.RuntimeTypeCheckInfo;
import com.android.tools.r8.synthesis.SyntheticItems;
import com.android.tools.r8.utils.InternalOptions;

public class NoDirectRuntimeTypeChecks extends SingleClassPolicy {

  private final InternalOptions options;
  private final RuntimeTypeCheckInfo runtimeTypeCheckInfo;
  private final SyntheticItems syntheticItems;

  public NoDirectRuntimeTypeChecks(AppView<?> appView, Mode mode) {
    this(appView, mode, null);
  }

  public NoDirectRuntimeTypeChecks(
      AppView<?> appView, Mode mode, RuntimeTypeCheckInfo runtimeTypeCheckInfo) {
    assert runtimeTypeCheckInfo != null || mode.isFinal();
    this.options = appView.options();
    this.runtimeTypeCheckInfo = runtimeTypeCheckInfo;
    this.syntheticItems = appView.getSyntheticItems();
  }

  @Override
  public boolean canMerge(DexProgramClass clazz) {
    if (runtimeTypeCheckInfo == null) {
      assert syntheticItems.isSyntheticClass(clazz)
          : "Expected synthetic, got: " + clazz.getTypeName();
      return true;
    }
    return !runtimeTypeCheckInfo.isRuntimeCheckType(clazz);
  }

  @Override
  public String getName() {
    return "NoDirectRuntimeTypeChecks";
  }

  @Override
  public boolean shouldSkipPolicy() {
    return options.horizontalClassMergerOptions().isIgnoreRuntimeTypeChecksForTestingEnabled();
  }
}