This allows the dex processing to back out if there are too many type ids.

PiperOrigin-RevId: 721819820
Change-Id: Icd2bea3d0860b6d4c20c53e6dc2ff170c20c181f

Author: A Googler

src/tools/java/com/google/devtools/build/android/dexer/DexLimitTracker.java

@@ -31,6 +31,7 @@ class DexLimitTracker {
 
   private final LinkedHashSet<FieldDescriptor> fieldsSeen = new LinkedHashSet<>();
   private final LinkedHashSet<MethodDescriptor> methodsSeen = new LinkedHashSet<>();
+  private final LinkedHashSet<String> typesSeen = new LinkedHashSet<>();
   private final int maxNumberOfIdxPerDex;
 
   public DexLimitTracker(int maxNumberOfIdxPerDex) {
@@ -40,17 +41,19 @@ public DexLimitTracker(int maxNumberOfIdxPerDex) {
   /**
    * Returns whether we're within limits.
    *
-   * @return {@code true} if method or field references are outside limits, {@code false} both are
-   *     within limits.
+   * @return {@code true} if method, field or type references are outside limits, {@code false} if
+   *     all are within limits.
    */
   public boolean outsideLimits() {
     return fieldsSeen.size() > maxNumberOfIdxPerDex
-        || methodsSeen.size() > maxNumberOfIdxPerDex;
+        || methodsSeen.size() > maxNumberOfIdxPerDex
+        || typesSeen.size() > maxNumberOfIdxPerDex;
   }
 
   public void clear() {
     fieldsSeen.clear();
     methodsSeen.clear();
+    typesSeen.clear();
   }
 
   public void track(Dex dexFile) {
@@ -62,6 +65,10 @@ public void track(Dex dexFile) {
     for (int methodIndex = 0; methodIndex < methodCount; ++methodIndex) {
       methodsSeen.add(MethodDescriptor.fromDex(dexFile, methodIndex));
     }
+    int typeCount = dexFile.typeIds().size();
+    for (int typeIndex = 0; typeIndex < typeCount; ++typeIndex) {
+      typesSeen.add(typeName(dexFile, typeIndex));
+    }
   }
 
   private static String typeName(Dex dex, int typeIndex) {