summaryrefslogtreecommitdiff
path: root/src/cuchaz/enigma/mapping/MappingsWriter.java
blob: a97052fa14aa1073cf3497f5d54e9873e6055998 (plain) (blame)
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
/*******************************************************************************
 * 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 );
		}
	}
	
	public void write( PrintWriter out, ClassMapping classMapping )
	throws IOException
	{
		out.format( "CLASS %s %s\n", classMapping.getObfName(), classMapping.getDeobfName() );
		
		for( FieldMapping fieldMapping : sorted( classMapping.fields() ) )
		{
			write( out, fieldMapping );
		}
		
		for( MethodMapping methodMapping : sorted( classMapping.methods() ) )
		{
			write( out, methodMapping );
		}
	}

	public void write( PrintWriter out, FieldMapping fieldMapping )
	throws IOException
	{
		out.format( "\tFIELD %s %s\n", fieldMapping.getObfName(), fieldMapping.getDeobfName() );
	}
	
	public void write( PrintWriter out, MethodMapping methodMapping )
	throws IOException
	{
		out.format( "\tMETHOD %s %s %s %s\n",
			methodMapping.getObfName(), methodMapping.getDeobfName(),
			methodMapping.getObfSignature(), methodMapping.getDeobfSignature()
		);
		
		for( ArgumentMapping argumentMapping : sorted( methodMapping.arguments() ) )
		{
			write( out, argumentMapping );
		}
	}

	public void write( PrintWriter out, ArgumentMapping argumentMapping )
	throws IOException
	{
		out.format( "\t\tARG %d %s\n", 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;
	}
}