1 files changed, 33 insertions, 0 deletions

diff --git a/src/cuchaz/enigma/convert/ClassMatching.java b/src/cuchaz/enigma/convert/ClassMatching.java
index fea8438..4e9fe39 100644
--- a/src/cuchaz/enigma/convert/ClassMatching.java
+++ b/src/cuchaz/enigma/convert/ClassMatching.java
@@ -10,12 +10,15 @@
  ******************************************************************************/
 package cuchaz.enigma.convert;
 
+import java.util.AbstractMap;
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.Collection;
 import java.util.List;
 import java.util.Map;
 
 import com.beust.jcommander.internal.Lists;
+import com.beust.jcommander.internal.Maps;
 import com.google.common.collect.ArrayListMultimap;
 import com.google.common.collect.BiMap;
 import com.google.common.collect.HashBiMap;
@@ -156,6 +159,36 @@ public class ClassMatching
                 return new ArrayList<ClassIdentity>( m_unmatchedDestClasses );
         }
         
+        public Map<String,Map.Entry<ClassIdentity,List<ClassIdentity>>> getConversionMap( )
+        {
+                Map<String,Map.Entry<ClassIdentity,List<ClassIdentity>>> conversion = Maps.newHashMap();
+                for( Map.Entry<ClassIdentity,ClassIdentity> entry : getUniqueMatches().entrySet() )
+                {
+                        conversion.put(
+                                entry.getKey().getClassEntry().getName(),
+                                new AbstractMap.SimpleEntry<ClassIdentity,List<ClassIdentity>>( entry.getKey(), Arrays.asList( entry.getValue() ) )
+                        );
+                }
+                for( Map.Entry<List<ClassIdentity>,List<ClassIdentity>> entry : getAmbiguousMatches().entrySet() )
+                {
+                        for( ClassIdentity sourceClass : entry.getKey() )
+                        {
+                                conversion.put(
+                                        sourceClass.getClassEntry().getName(),
+                                        new AbstractMap.SimpleEntry<ClassIdentity,List<ClassIdentity>>( sourceClass, entry.getValue() )
+                                );
+                        }
+                }
+                for( ClassIdentity sourceClass : getUnmatchedSourceClasses() )
+                {
+                        conversion.put(
+                                sourceClass.getClassEntry().getName(),
+                                new AbstractMap.SimpleEntry<ClassIdentity,List<ClassIdentity>>( sourceClass, new ArrayList<ClassIdentity>() )
+                        );
+                }
+                return conversion;
+        }
+        
         @Override
         public String toString( )
         {