| // 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.utils.ListUtils; |
| import com.google.common.collect.ImmutableList; |
| import it.unimi.dsi.fastutil.ints.IntList; |
| import java.util.List; |
| |
| public class SimpleInliningConstraintDisjunction extends SimpleInliningConstraint { |
| |
| private final List<SimpleInliningConstraint> constraints; |
| |
| public SimpleInliningConstraintDisjunction(List<SimpleInliningConstraint> constraints) { |
| assert constraints.size() > 1; |
| assert constraints.stream().noneMatch(SimpleInliningConstraint::isAlways); |
| assert constraints.stream().noneMatch(SimpleInliningConstraint::isDisjunction); |
| assert constraints.stream().noneMatch(SimpleInliningConstraint::isNever); |
| this.constraints = constraints; |
| } |
| |
| SimpleInliningConstraint add(SimpleInliningConstraint constraint) { |
| assert !constraint.isAlways(); |
| assert !constraint.isNever(); |
| if (constraint.isDisjunction()) { |
| return addAll(constraint.asDisjunction()); |
| } |
| assert constraint.isArgumentConstraint() || constraint.isConjunction(); |
| return new SimpleInliningConstraintDisjunction( |
| ImmutableList.<SimpleInliningConstraint>builder() |
| .addAll(constraints) |
| .add(constraint) |
| .build()); |
| } |
| |
| public SimpleInliningConstraintDisjunction addAll( |
| SimpleInliningConstraintDisjunction disjunction) { |
| return new SimpleInliningConstraintDisjunction( |
| ImmutableList.<SimpleInliningConstraint>builder() |
| .addAll(constraints) |
| .addAll(disjunction.constraints) |
| .build()); |
| } |
| |
| @Override |
| public boolean isDisjunction() { |
| return true; |
| } |
| |
| @Override |
| public SimpleInliningConstraintDisjunction asDisjunction() { |
| return this; |
| } |
| |
| @Override |
| public boolean isSatisfied(InvokeMethod invoke) { |
| for (SimpleInliningConstraint constraint : constraints) { |
| if (constraint.isSatisfied(invoke)) { |
| return true; |
| } |
| } |
| return false; |
| } |
| |
| @Override |
| public SimpleInliningConstraint fixupAfterRemovingThisParameter( |
| SimpleInliningConstraintFactory factory) { |
| List<SimpleInliningConstraint> rewrittenConstraints = |
| ListUtils.mapOrElse( |
| constraints, constraint -> constraint.fixupAfterRemovingThisParameter(factory), null); |
| return rewrittenConstraints != null |
| ? new SimpleInliningConstraintDisjunction(rewrittenConstraints) |
| : this; |
| } |
| |
| @Override |
| public SimpleInliningConstraint rewrittenWithUnboxedArguments( |
| IntList unboxedArgumentIndices, SimpleInliningConstraintFactory factory) { |
| List<SimpleInliningConstraint> rewrittenConstraints = |
| ListUtils.mapOrElse( |
| constraints, |
| constraint -> constraint.rewrittenWithUnboxedArguments(unboxedArgumentIndices, factory), |
| null); |
| return rewrittenConstraints != null |
| ? new SimpleInliningConstraintDisjunction(rewrittenConstraints) |
| : this; |
| } |
| } |