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
121
122
|
/*******************************************************************************
* Copyright (c) 2014 Jeff Martin.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the GNU Public License v3.0
* which accompanies this distribution, and is available at
* http://www.gnu.org/licenses/gpl.html
*
* Contributors:
* Jeff Martin - initial API and implementation
******************************************************************************/
package cuchaz.enigma.mapping;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.Writer;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
public class MappingsWriter
{
public void write( Writer out, Mappings mappings )
throws IOException
{
write( new PrintWriter( out ), mappings );
}
public void write( PrintWriter out, Mappings mappings )
throws IOException
{
for( ClassMapping classMapping : sorted( mappings.classes() ) )
{
write( out, classMapping, 0 );
}
}
private void write( PrintWriter out, ClassMapping classMapping, int depth )
throws IOException
{
if( classMapping.getDeobfName() == null )
{
out.format( "%sCLASS %s\n", getIndent( depth ), classMapping.getObfName() );
}
else
{
out.format( "%sCLASS %s %s\n", getIndent( depth ), classMapping.getObfName(), classMapping.getDeobfName() );
}
for( ClassMapping innerClassMapping : sorted( classMapping.innerClasses() ) )
{
write( out, innerClassMapping, depth + 1 );
}
for( FieldMapping fieldMapping : sorted( classMapping.fields() ) )
{
write( out, fieldMapping, depth + 1 );
}
for( MethodMapping methodMapping : sorted( classMapping.methods() ) )
{
write( out, methodMapping, depth + 1 );
}
}
private void write( PrintWriter out, FieldMapping fieldMapping, int depth )
throws IOException
{
out.format( "%sFIELD %s %s\n", getIndent( depth ), fieldMapping.getObfName(), fieldMapping.getDeobfName() );
}
private void write( PrintWriter out, MethodMapping methodMapping, int depth )
throws IOException
{
if( methodMapping.getDeobfName() == null )
{
out.format( "%sMETHOD %s %s\n",
getIndent( depth ),
methodMapping.getObfName(), methodMapping.getObfSignature()
);
}
else
{
out.format( "%sMETHOD %s %s %s\n",
getIndent( depth ),
methodMapping.getObfName(), methodMapping.getDeobfName(),
methodMapping.getObfSignature()
);
}
for( ArgumentMapping argumentMapping : sorted( methodMapping.arguments() ) )
{
write( out, argumentMapping, depth + 1 );
}
}
private void write( PrintWriter out, ArgumentMapping argumentMapping, int depth )
throws IOException
{
out.format( "%sARG %d %s\n", getIndent( depth ), argumentMapping.getIndex(), argumentMapping.getName() );
}
private <T extends Comparable<T>> List<T> sorted( Iterable<T> classes )
{
List<T> out = new ArrayList<T>();
for( T t : classes )
{
out.add( t );
}
Collections.sort( out );
return out;
}
private String getIndent( int depth )
{
StringBuilder buf = new StringBuilder();
for( int i=0; i<depth; i++ )
{
buf.append( "\t" );
}
return buf.toString();
}
}
|
