diff options
Diffstat (limited to 'src/main/java/cuchaz/enigma/gui/stats')
| -rw-r--r-- | src/main/java/cuchaz/enigma/gui/stats/StatsGenerator.java | 24 |
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(); |