From 4349d22cc8abf5ec74075dde1b45c5f2f8679bbf Mon Sep 17 00:00:00 2001
From: jeff
Date: Wed, 30 Jul 2014 23:43:09 -0400
Subject: switched to line-by-line mergable, human-readable file format for
 mappings

---
 src/cuchaz/enigma/mapping/Mappings.java | 128 ++++++++++++++++++++++++++++++++
 1 file changed, 128 insertions(+)
 create mode 100644 src/cuchaz/enigma/mapping/Mappings.java

(limited to 'src/cuchaz/enigma/mapping/Mappings.java')

diff --git a/src/cuchaz/enigma/mapping/Mappings.java b/src/cuchaz/enigma/mapping/Mappings.java
new file mode 100644
index 0000000..2a39057
--- /dev/null
+++ b/src/cuchaz/enigma/mapping/Mappings.java
@@ -0,0 +1,128 @@
+/*******************************************************************************
+ * 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.InputStream;
+import java.io.ObjectInputStream;
+import java.io.Serializable;
+import java.util.Map;
+import java.util.zip.GZIPInputStream;
+
+import com.beust.jcommander.internal.Maps;
+
+import cuchaz.enigma.Util;
+
+public class Mappings implements Serializable
+{
+	private static final long serialVersionUID = 4649790259460259026L;
+	
+	protected Map<String,ClassMapping> m_classesByObf;
+	protected Map<String,ClassMapping> m_classesByDeobf;
+	
+	public Mappings( )
+	{
+		m_classesByObf = Maps.newHashMap();
+		m_classesByDeobf = Maps.newHashMap();
+	}
+	
+	public Mappings( Iterable<ClassMapping> classes )
+	{
+		this();
+		
+		for( ClassMapping classMapping : classes )
+		{
+			m_classesByObf.put( classMapping.getObfName(), classMapping );
+			m_classesByDeobf.put( classMapping.getDeobfName(), classMapping );
+		}
+	}
+	
+	public static Mappings newFromResource( String resource )
+	throws IOException
+	{
+		InputStream in = null;
+		try
+		{
+			in = Mappings.class.getResourceAsStream( resource );
+			return newFromStream( in );
+		}
+		finally
+		{
+			Util.closeQuietly( in );
+		}
+	}
+	
+	public Iterable<ClassMapping> classes( )
+	{
+		assert( m_classesByObf.size() == m_classesByDeobf.size() );
+		return m_classesByObf.values();
+	}
+	
+	protected void addClassMapping( ClassMapping classMapping )
+	{
+		m_classesByObf.put( classMapping.getObfName(), classMapping );
+		m_classesByDeobf.put( classMapping.getDeobfName(), classMapping );
+	}
+	
+	public ClassMapping getClassByObf( ClassEntry entry )
+	{
+		return getClassByObf( entry.getName() );
+	}
+	
+	public ClassMapping getClassByObf( String obfName )
+	{
+		return m_classesByObf.get( obfName );
+	}
+	
+	public ClassMapping getClassByDeobf( ClassEntry entry )
+	{
+		return getClassByObf( entry.getName() );
+	}
+	
+	public ClassMapping getClassByDeobf( String deobfName )
+	{
+		return m_classesByDeobf.get( deobfName );
+	}
+	
+	public Translator getTranslator( Ancestries ancestries, TranslationDirection direction )
+	{
+		return new Translator(
+			direction,
+			direction.choose( m_classesByObf, m_classesByDeobf ),
+			direction.choose( ancestries, new DeobfuscatedAncestries( ancestries, m_classesByObf, m_classesByDeobf ) )
+		);
+	}
+	
+	public static Mappings newFromStream( InputStream in )
+	throws IOException
+	{
+		try
+		{
+			return (Mappings)new ObjectInputStream( new GZIPInputStream( in ) ).readObject();
+		}
+		catch( ClassNotFoundException ex )
+		{
+			throw new Error( ex );
+		}
+	}
+	
+	@Override
+	public String toString( )
+	{
+		StringBuilder buf = new StringBuilder();
+		for( ClassMapping classMapping : m_classesByObf.values() )
+		{
+			buf.append( classMapping.toString() );
+			buf.append( "\n" );
+		}
+		return buf.toString();
+	}
+}
-- 
cgit v1.2.3