| // Copyright (c) 2023, 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.build.shrinker.r8integration; |
| |
| import com.android.aapt.Resources.ConfigValue; |
| import com.android.aapt.Resources.Entry; |
| import com.android.aapt.Resources.Item; |
| import com.android.aapt.Resources.ResourceTable; |
| import com.android.aapt.Resources.Value; |
| import com.android.build.shrinker.NoDebugReporter; |
| import com.android.build.shrinker.ResourceShrinkerModel; |
| import com.android.build.shrinker.ResourceTableUtilKt; |
| import com.android.build.shrinker.ShrinkerDebugReporter; |
| import com.android.ide.common.resources.ResourcesUtil; |
| import com.android.ide.common.resources.usage.ResourceUsageModel; |
| import com.android.ide.common.resources.usage.ResourceUsageModel.Resource; |
| import com.android.resources.ResourceType; |
| import java.io.IOException; |
| import java.io.InputStream; |
| import java.util.Collections; |
| import java.util.HashMap; |
| import java.util.List; |
| import java.util.Map; |
| |
| public class R8ResourceShrinkerState { |
| |
| private R8ResourceShrinkerModel r8ResourceShrinkerModel; |
| |
| public List<String> trace(int id) { |
| Resource resource = r8ResourceShrinkerModel.getResourceStore().getResource(id); |
| ResourceUsageModel.markReachable(resource); |
| return Collections.emptyList(); |
| } |
| |
| public void setResourceTableInput(InputStream inputStream) { |
| r8ResourceShrinkerModel = new R8ResourceShrinkerModel(NoDebugReporter.INSTANCE, true); |
| r8ResourceShrinkerModel.instantiateFromResourceTable(inputStream, true); |
| } |
| |
| public R8ResourceShrinkerModel getR8ResourceShrinkerModel() { |
| return r8ResourceShrinkerModel; |
| } |
| |
| public static class R8ResourceShrinkerModel extends ResourceShrinkerModel { |
| private final Map<Integer, String> stringResourcesWithSingleValue = new HashMap<>(); |
| |
| public R8ResourceShrinkerModel( |
| ShrinkerDebugReporter debugReporter, boolean supportMultipackages) { |
| super(debugReporter, supportMultipackages); |
| } |
| |
| public Map<Integer, String> getStringResourcesWithSingleValue() { |
| return stringResourcesWithSingleValue; |
| } |
| |
| // Similar to instantiation in ProtoResourceTableGatherer, but using an inputstream. |
| void instantiateFromResourceTable(InputStream inputStream, boolean includeStyleables) { |
| try { |
| ResourceTable resourceTable = ResourceTable.parseFrom(inputStream); |
| instantiateFromResourceTable(resourceTable, includeStyleables); |
| } catch (IOException ex) { |
| throw new RuntimeException(ex); |
| } |
| } |
| |
| void instantiateFromResourceTable(ResourceTable resourceTable, boolean includeStyleables) { |
| ResourceTableUtilKt.entriesSequence(resourceTable) |
| .iterator() |
| .forEachRemaining( |
| entryWrapper -> { |
| ResourceType resourceType = ResourceType.fromClassName(entryWrapper.getType()); |
| Entry entry = entryWrapper.getEntry(); |
| int entryId = entryWrapper.getId(); |
| recordSingleValueResources(resourceType, entry, entryId); |
| if (resourceType != ResourceType.STYLEABLE || includeStyleables) { |
| this.addResource( |
| resourceType, |
| entryWrapper.getPackageName(), |
| ResourcesUtil.resourceNameToFieldName(entry.getName()), |
| entryId); |
| } |
| }); |
| } |
| |
| private void recordSingleValueResources(ResourceType resourceType, Entry entry, int entryId) { |
| if (!entry.hasOverlayableItem() && entry.getConfigValueList().size() == 1) { |
| if (resourceType == ResourceType.STRING) { |
| ConfigValue configValue = entry.getConfigValue(0); |
| if (configValue.hasValue()) { |
| Value value = configValue.getValue(); |
| if (value.hasItem()) { |
| Item item = value.getItem(); |
| if (item.hasStr()) { |
| stringResourcesWithSingleValue.put(entryId, item.getStr().getValue()); |
| } |
| } |
| } |
| } |
| } |
| } |
| } |
| } |