1 files changed, 128 insertions, 0 deletions

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();
+        }
+}