1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
/*******************************************************************************
* Copyright (c) 2015 Jeff Martin.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the GNU Lesser General Public
* License v3.0 which accompanies this distribution, and is available at
* http://www.gnu.org/licenses/lgpl.html
* <p>
* Contributors:
* Jeff Martin - initial API and implementation
******************************************************************************/
package cuchaz.enigma.convert;
import java.io.*;
import java.nio.charset.Charset;
import java.util.Map;
import cuchaz.enigma.mapping.BehaviorEntry;
import cuchaz.enigma.mapping.ClassEntry;
import cuchaz.enigma.mapping.Entry;
import cuchaz.enigma.mapping.FieldEntry;
public class MatchesWriter {
public static void writeClasses(ClassMatches matches, File file)
throws IOException {
try (OutputStreamWriter out = new OutputStreamWriter(new FileOutputStream(file), Charset.forName("UTF-8"))) {
for (ClassMatch match : matches) {
writeClassMatch(out, match);
}
}
}
private static void writeClassMatch(OutputStreamWriter out, ClassMatch match)
throws IOException {
writeClasses(out, match.sourceClasses);
out.write(":");
writeClasses(out, match.destClasses);
out.write("\n");
}
private static void writeClasses(OutputStreamWriter out, Iterable<ClassEntry> classes)
throws IOException {
boolean isFirst = true;
for (ClassEntry entry : classes) {
if (isFirst) {
isFirst = false;
} else {
out.write(",");
}
out.write(entry.toString());
}
}
public static <T extends Entry> void writeMembers(MemberMatches<T> matches, File file)
throws IOException {
try (OutputStreamWriter out = new OutputStreamWriter(new FileOutputStream(file), Charset.forName("UTF-8"))) {
for (Map.Entry<T, T> match : matches.matches().entrySet()) {
writeMemberMatch(out, match.getKey(), match.getValue());
}
for (T entry : matches.getUnmatchedSourceEntries()) {
writeMemberMatch(out, entry, null);
}
for (T entry : matches.getUnmatchedDestEntries()) {
writeMemberMatch(out, null, entry);
}
for (T entry : matches.getUnmatchableSourceEntries()) {
writeUnmatchableEntry(out, entry);
}
}
}
private static <T extends Entry> void writeMemberMatch(OutputStreamWriter out, T source, T dest)
throws IOException {
if (source != null) {
writeEntry(out, source);
}
out.write(":");
if (dest != null) {
writeEntry(out, dest);
}
out.write("\n");
}
private static <T extends Entry> void writeUnmatchableEntry(OutputStreamWriter out, T entry)
throws IOException {
out.write("!");
writeEntry(out, entry);
out.write("\n");
}
private static <T extends Entry> void writeEntry(OutputStreamWriter out, T entry)
throws IOException {
if (entry instanceof FieldEntry) {
writeField(out, (FieldEntry) entry);
} else if (entry instanceof BehaviorEntry) {
writeBehavior(out, (BehaviorEntry) entry);
}
}
private static void writeField(OutputStreamWriter out, FieldEntry fieldEntry)
throws IOException {
out.write(fieldEntry.getClassName());
out.write(" ");
out.write(fieldEntry.getName());
out.write(" ");
out.write(fieldEntry.getType().toString());
}
private static void writeBehavior(OutputStreamWriter out, BehaviorEntry behaviorEntry)
throws IOException {
out.write(behaviorEntry.getClassName());
out.write(" ");
out.write(behaviorEntry.getName());
out.write(" ");
if (behaviorEntry.getSignature() != null) {
out.write(behaviorEntry.getSignature().toString());
}
}
}
|
