| // Copyright (c) 2022, 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.profile.art; |
| |
| import com.android.tools.r8.graph.DexItemFactory; |
| import com.android.tools.r8.graph.DexType; |
| import com.android.tools.r8.profile.AbstractProfileClassRule; |
| import com.android.tools.r8.references.ClassReference; |
| import com.android.tools.r8.references.Reference; |
| import com.android.tools.r8.utils.ThrowingConsumer; |
| import com.android.tools.r8.utils.ThrowingFunction; |
| import java.io.IOException; |
| import java.io.OutputStreamWriter; |
| |
| public class ArtProfileClassRule extends ArtProfileRule implements AbstractProfileClassRule { |
| |
| private final DexType type; |
| |
| ArtProfileClassRule(DexType type) { |
| this.type = type; |
| } |
| |
| public static Builder builder() { |
| return new Builder(); |
| } |
| |
| public static Builder builder(DexItemFactory dexItemFactory) { |
| return new Builder(dexItemFactory); |
| } |
| |
| @Override |
| public <E1 extends Exception, E2 extends Exception> void accept( |
| ThrowingConsumer<? super ArtProfileClassRule, E1> classRuleConsumer, |
| ThrowingConsumer<? super ArtProfileMethodRule, E2> methodRuleConsumer) |
| throws E1 { |
| classRuleConsumer.accept(this); |
| } |
| |
| @Override |
| public <T, E1 extends Exception, E2 extends Exception> T apply( |
| ThrowingFunction<ArtProfileClassRule, T, E1> classRuleFunction, |
| ThrowingFunction<ArtProfileMethodRule, T, E2> methodRuleFunction) |
| throws E1 { |
| return classRuleFunction.apply(this); |
| } |
| |
| public ClassReference getClassReference() { |
| return Reference.classFromDescriptor(type.toDescriptorString()); |
| } |
| |
| public ArtProfileClassRuleInfo getClassRuleInfo() { |
| return ArtProfileClassRuleInfoImpl.empty(); |
| } |
| |
| @Override |
| public DexType getReference() { |
| return getType(); |
| } |
| |
| public DexType getType() { |
| return type; |
| } |
| |
| @Override |
| public void writeHumanReadableRuleString(OutputStreamWriter writer) throws IOException { |
| writer.write(type.toDescriptorString()); |
| } |
| |
| @Override |
| public boolean equals(Object o) { |
| if (this == o) { |
| return true; |
| } |
| if (o == null || getClass() != o.getClass()) { |
| return false; |
| } |
| ArtProfileClassRule that = (ArtProfileClassRule) o; |
| return type == that.type; |
| } |
| |
| @Override |
| public int hashCode() { |
| return type.hashCode(); |
| } |
| |
| @Override |
| public String toString() { |
| return type.toDescriptorString(); |
| } |
| |
| public static class Builder extends ArtProfileRule.Builder |
| implements ArtProfileClassRuleBuilder, AbstractProfileClassRule.Builder<ArtProfileClassRule> { |
| |
| private final DexItemFactory dexItemFactory; |
| private DexType type; |
| |
| Builder() { |
| this(null); |
| } |
| |
| Builder(DexItemFactory dexItemFactory) { |
| this.dexItemFactory = dexItemFactory; |
| } |
| |
| @Override |
| Builder asClassRuleBuilder() { |
| return this; |
| } |
| |
| @Override |
| public Builder setClassReference(ClassReference classReference) { |
| assert dexItemFactory != null; |
| return setType(dexItemFactory.createType(classReference.getDescriptor())); |
| } |
| |
| public Builder setType(DexType type) { |
| this.type = type; |
| return this; |
| } |
| |
| @Override |
| public ArtProfileClassRule build() { |
| return new ArtProfileClassRule(type); |
| } |
| } |
| } |