Add handy factory for DexApplication taking in a map of classes

Col-E · Apr 23, 2023 · d75f0b9 · d75f0b9
1 parent 5463a80
commit d75f0b9
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/main/java/com/android/tools/r8/graph/DexApplication.java b/src/main/java/com/android/tools/r8/graph/DexApplication.java
@@ -172,6 +172,14 @@ public ClassNameMapper getProguardMap() {
     return proguardMap;
   }
 
+  @Nonnull
+  public static DexApplication fromClasses(@Nonnull InternalOptions options,
+                                           @Nonnull Map<String, DexProgramClass> map) {
+    return new LazyLoadedDexApplication.Builder(options, Timing.empty())
+            .setProgramClasses(map)
+            .build();
+  }
+
   public abstract static class Builder<T extends Builder<T>> {
 
     private final Map<String, DexProgramClass> programClasses = new HashMap<>();
@@ -288,6 +296,12 @@ public synchronized T removeProgramClass(@Nonnull String typeName) {
       return self();
     }
 
+    public synchronized T setProgramClasses(@Nonnull Map<String, DexProgramClass> classes) {
+      programClasses.clear();
+      programClasses.putAll(classes);
+      return self();
+    }
+
     public abstract void addProgramClassPotentiallyOverridingNonProgramClass(DexProgramClass clazz);
 
     public synchronized T addProgramClasses(Collection<DexProgramClass> classes) {