src/main/java/com/android/tools/r8/ir/analysis/inlining/NotNullSimpleInliningConstraint.java

// 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.ir.analysis.inlining;

import com.android.tools.r8.ir.code.InvokeMethod;
import com.android.tools.r8.ir.code.Value;
import it.unimi.dsi.fastutil.ints.IntList;

/** Constraint that is satisfied if a specific argument is always non-null. */
public class NotNullSimpleInliningConstraint extends SimpleInliningArgumentConstraint {

  private NotNullSimpleInliningConstraint(int argumentIndex) {
    super(argumentIndex);
  }

  static NotNullSimpleInliningConstraint create(
      int argumentIndex, SimpleInliningConstraintFactory witness) {
    assert witness != null;
    return new NotNullSimpleInliningConstraint(argumentIndex);
  }

  @Override
  public boolean isNotNull() {
    return true;
  }

  @Override
  public boolean isSatisfied(InvokeMethod invoke) {
    Value argument = getArgument(invoke);
    assert argument.getType().isReferenceType() : invoke;
    return argument.isNeverNull();
  }

  @Override
  public SimpleInliningConstraint fixupAfterRemovingThisParameter(
      SimpleInliningConstraintFactory factory) {
    assert getArgumentIndex() > 0;
    return factory.createNotNullConstraint(getArgumentIndex() - 1);
  }

  @Override
  public SimpleInliningConstraint rewrittenWithUnboxedArguments(IntList unboxedArgumentIndices) {
    if (unboxedArgumentIndices.contains(getArgumentIndex())) {
      // TODO(b/176067541): Could be refined to an argument-equals-int constraint.
      return NeverSimpleInliningConstraint.getInstance();
    }
    return this;
  }
}