1 files changed, 159 insertions, 0 deletions

diff --git a/src/cuchaz/enigma/mapping/ClassIndex.java b/src/cuchaz/enigma/mapping/ClassIndex.java
new file mode 100644
index 0000000..699807b
--- /dev/null
+++ b/src/cuchaz/enigma/mapping/ClassIndex.java
@@ -0,0 +1,159 @@
+/*******************************************************************************
+ * 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.Serializable;
+import java.util.Map;
+
+import com.beust.jcommander.internal.Maps;
+import com.google.common.collect.BiMap;
+import com.google.common.collect.HashBiMap;
+
+public class ClassIndex implements Serializable
+{
+        private static final long serialVersionUID = -5148491146902340107L;
+        
+        private String m_obfName;
+        private String m_deobfName;
+        private BiMap<String,String> m_fieldsObfToDeobf;
+        private Map<String,MethodIndex> m_methodsByObf;
+        private Map<String,MethodIndex> m_methodsByDeobf;
+        
+        public ClassIndex( String obfName, String deobfName )
+        {
+                m_obfName = obfName;
+                m_deobfName = deobfName;
+                m_fieldsObfToDeobf = HashBiMap.create();
+                m_methodsByObf = Maps.newHashMap();
+                m_methodsByDeobf = Maps.newHashMap();
+        }
+
+        public String getObfName( )
+        {
+                return m_obfName;
+        }
+        
+        public String getDeobfName( )
+        {
+                return m_deobfName;
+        }
+        public void setDeobfName( String val )
+        {
+                m_deobfName = val;
+        }
+
+        public String getObfFieldName( String deobfName )
+        {
+                return m_fieldsObfToDeobf.inverse().get( deobfName );
+        }
+        
+        public String getDeobfFieldName( String obfName )
+        {
+                return m_fieldsObfToDeobf.get( obfName );
+        }
+        
+        public void setFieldName( String obfName, String deobfName )
+        {
+                m_fieldsObfToDeobf.put( obfName, deobfName );
+        }
+        
+        public MethodIndex getMethodByObf( String obfName, String signature )
+        {
+                return m_methodsByObf.get( getMethodKey( obfName, signature ) );
+        }
+        
+        public MethodIndex getMethodByDeobf( String deobfName, String signature )
+        {
+                return m_methodsByDeobf.get( getMethodKey( deobfName, signature ) );
+        }
+        
+        private String getMethodKey( String name, String signature )
+        {
+                return name + signature;
+        }
+        
+        public void setMethodNameAndSignature( String obfName, String obfSignature, String deobfName, String deobfSignature )
+        {
+                if( deobfName == null )
+                {
+                        throw new IllegalArgumentException( "deobf name cannot be null!" );
+                }
+                
+                MethodIndex methodIndex = m_methodsByObf.get( getMethodKey( obfName, obfSignature ) );
+                if( methodIndex == null )
+                {
+                        methodIndex = createMethodIndex( obfName, obfSignature );
+                }
+                
+                m_methodsByDeobf.remove( getMethodKey( methodIndex.getDeobfName(), methodIndex.getDeobfSignature() ) );
+                methodIndex.setDeobfName( deobfName );
+                methodIndex.setDeobfSignature( deobfSignature );
+                m_methodsByDeobf.put( getMethodKey( deobfName, deobfSignature ), methodIndex );
+        }
+        
+        public void updateDeobfMethodSignatures( Translator translator )
+        {
+                for( MethodIndex methodIndex : m_methodsByObf.values() )
+                {
+                        methodIndex.setDeobfSignature( translator.translateSignature( methodIndex.getObfSignature() ) );
+                }
+        }
+
+        public void setArgumentName( String obfMethodName, String obfMethodSignature, int index, String obfName, String deobfName )
+        {
+                if( deobfName == null )
+                {
+                        throw new IllegalArgumentException( "deobf name cannot be null!" );
+                }
+                
+                MethodIndex methodIndex = m_methodsByObf.get( getMethodKey( obfMethodName, obfMethodSignature ) );
+                if( methodIndex == null )
+                {
+                        methodIndex = createMethodIndex( obfMethodName, obfMethodSignature );
+                }
+                methodIndex.setArgumentName( index, obfName, deobfName );
+        }
+
+        private MethodIndex createMethodIndex( String obfName, String obfSignature )
+        {
+                MethodIndex methodIndex = new MethodIndex( obfName, obfSignature, obfName, obfSignature );
+                String key = getMethodKey( obfName, obfSignature );
+                m_methodsByObf.put( key, methodIndex );
+                m_methodsByDeobf.put( key, methodIndex );
+                return methodIndex;
+        }
+
+        @Override
+        public String toString( )
+        {
+                StringBuilder buf = new StringBuilder();
+                buf.append( m_obfName );
+                buf.append( " <-> " );
+                buf.append( m_deobfName );
+                buf.append( "\n" );
+                buf.append( "Fields:\n" );
+                for( Map.Entry<String,String> entry : m_fieldsObfToDeobf.entrySet() )
+                {
+                        buf.append( "\t" );
+                        buf.append( entry.getKey() );
+                        buf.append( " <-> " );
+                        buf.append( entry.getValue() );
+                        buf.append( "\n" );
+                }
+                buf.append( "Methods:\n" );
+                for( MethodIndex methodIndex : m_methodsByObf.values() )
+                {
+                        buf.append( methodIndex.toString() );
+                        buf.append( "\n" );
+                }
+                return buf.toString();
+        }
+}