summaryrefslogtreecommitdiff
path: root/src/main/java/cuchaz/enigma/gui/stats
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/java/cuchaz/enigma/gui/stats')
-rw-r--r--src/main/java/cuchaz/enigma/gui/stats/StatsGenerator.java24
1 files changed, 16 insertions, 8 deletions
diff --git a/src/main/java/cuchaz/enigma/gui/stats/StatsGenerator.java b/src/main/java/cuchaz/enigma/gui/stats/StatsGenerator.java
index 485daf5..88ed96f 100644
--- a/src/main/java/cuchaz/enigma/gui/stats/StatsGenerator.java
+++ b/src/main/java/cuchaz/enigma/gui/stats/StatsGenerator.java
@@ -18,15 +18,15 @@ public class StatsGenerator {
18 private final EntryIndex entryIndex; 18 private final EntryIndex entryIndex;
19 private final EntryRemapper mapper; 19 private final EntryRemapper mapper;
20 private final EntryResolver entryResolver; 20 private final EntryResolver entryResolver;
21 private final ObfuscationTestService obfuscationTestService; 21 private final List<ObfuscationTestService> obfuscationTestServices;
22 private final NameProposalService nameProposalService; 22 private final List<NameProposalService> nameProposalServices;
23 23
24 public StatsGenerator(EnigmaProject project) { 24 public StatsGenerator(EnigmaProject project) {
25 entryIndex = project.getJarIndex().getEntryIndex(); 25 entryIndex = project.getJarIndex().getEntryIndex();
26 mapper = project.getMapper(); 26 mapper = project.getMapper();
27 entryResolver = project.getJarIndex().getEntryResolver(); 27 entryResolver = project.getJarIndex().getEntryResolver();
28 obfuscationTestService = project.getEnigma().getServices().get(ObfuscationTestService.TYPE).orElse(null); 28 obfuscationTestServices = project.getEnigma().getServices().get(ObfuscationTestService.TYPE);
29 nameProposalService = project.getEnigma().getServices().get(NameProposalService.TYPE).orElse(null); 29 nameProposalServices = project.getEnigma().getServices().get(NameProposalService.TYPE);
30 } 30 }
31 31
32 public String generate(ProgressListener progress, Set<StatsMember> includedMembers) { 32 public String generate(ProgressListener progress, Set<StatsMember> includedMembers) {
@@ -112,12 +112,20 @@ public class StatsGenerator {
112 private boolean isObfuscated(Entry<?> entry) { 112 private boolean isObfuscated(Entry<?> entry) {
113 String name = entry.getName(); 113 String name = entry.getName();
114 114
115 if (obfuscationTestService != null && obfuscationTestService.testDeobfuscated(entry)) { 115 if (!obfuscationTestServices.isEmpty()) {
116 return false; 116 for (ObfuscationTestService service : obfuscationTestServices) {
117 if (service.testDeobfuscated(entry)) {
118 return false;
119 }
120 }
117 } 121 }
118 122
119 if (nameProposalService != null && nameProposalService.proposeName(entry, mapper).isPresent()) { 123 if (!nameProposalServices.isEmpty()) {
120 return false; 124 for (NameProposalService service : nameProposalServices) {
125 if (service.proposeName(entry, mapper).isPresent()) {
126 return false;
127 }
128 }
121 } 129 }
122 130
123 String mappedName = mapper.deobfuscate(entry).getName(); 131 String mappedName = mapper.deobfuscate(entry).getName();