From 3fde8f31d5e35256935de6945f6d36757e8e4adc Mon Sep 17 00:00:00 2001
From: modmuss50
Date: Tue, 3 Jul 2018 14:24:20 +0100
Subject: Make decompile parallel

---
 src/main/java/cuchaz/enigma/Deobfuscator.java | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/src/main/java/cuchaz/enigma/Deobfuscator.java b/src/main/java/cuchaz/enigma/Deobfuscator.java
index 6fe6e643..5e0bcb89 100644
--- a/src/main/java/cuchaz/enigma/Deobfuscator.java
+++ b/src/main/java/cuchaz/enigma/Deobfuscator.java
@@ -11,6 +11,7 @@
 
 package cuchaz.enigma;
 
+import com.google.common.base.Stopwatch;
 import com.google.common.collect.Lists;
 import com.google.common.collect.Maps;
 import com.google.common.collect.Sets;
@@ -267,11 +268,12 @@ public class Deobfuscator {
 		}
 
 		// DEOBFUSCATE ALL THE THINGS!! @_@
-		int i = 0;
-		for (ClassEntry obfClassEntry : classEntries) {
+		Stopwatch stopwatch = Stopwatch.createStarted();
+		AtomicInteger count = new AtomicInteger();
+		classEntries.parallelStream().forEach(obfClassEntry -> {
 			ClassEntry deobfClassEntry = deobfuscateEntry(new ClassEntry(obfClassEntry));
 			if (progress != null) {
-				progress.onProgress(i++, deobfClassEntry.toString());
+				progress.onProgress(count.getAndIncrement(), deobfClassEntry.toString());
 			}
 
 			try {
@@ -283,7 +285,7 @@ public class Deobfuscator {
 				file.getParentFile().mkdirs();
 				try (Writer writer = new BufferedWriter(new FileWriter(file))) {
 					sourceTree.acceptVisitor(new InsertParenthesesVisitor(), null);
-					sourceTree.acceptVisitor(new JavaOutputVisitor(new PlainTextOutput(writer), this.settings), null);
+					sourceTree.acceptVisitor(new JavaOutputVisitor(new PlainTextOutput(writer), settings), null);
 				}
 			} catch (Throwable t) {
 				// don't crash the whole world here, just log the error and keep going
@@ -291,9 +293,11 @@ public class Deobfuscator {
 				System.err.println("Unable to deobfuscate class " + deobfClassEntry + " (" + obfClassEntry + ")");
 				t.printStackTrace(System.err);
 			}
-		}
+		});
+		stopwatch.stop();
+		System.out.println("Done in : " + stopwatch.toString());
 		if (progress != null) {
-			progress.onProgress(i, "Done!");
+			progress.onProgress(count.get(), "Done:");
 		}
 	}
 
-- 
cgit v1.2.3