From c133e05b786ff5357931842581571c046f958c74 Mon Sep 17 00:00:00 2001
From: jeff
Date: Mon, 16 Mar 2015 12:29:17 -0400
Subject: fix a zillion issues with inner classes

---
 src/cuchaz/enigma/mapping/Mappings.java | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

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

diff --git a/src/cuchaz/enigma/mapping/Mappings.java b/src/cuchaz/enigma/mapping/Mappings.java
index a85bcbf..659d23a 100644
--- a/src/cuchaz/enigma/mapping/Mappings.java
+++ b/src/cuchaz/enigma/mapping/Mappings.java
@@ -13,9 +13,11 @@ package cuchaz.enigma.mapping;
 import java.io.Serializable;
 import java.util.ArrayList;
 import java.util.Collection;
+import java.util.List;
 import java.util.Map;
 import java.util.Set;
 
+import com.beust.jcommander.internal.Lists;
 import com.google.common.collect.Maps;
 import com.google.common.collect.Sets;
 
@@ -89,6 +91,18 @@ public class Mappings implements Serializable {
 		return m_classesByDeobf.get(deobfName);
 	}
 	
+	public void setClassDeobfName(ClassMapping classMapping, String deobfName) {
+		if (classMapping.getDeobfName() != null) {
+			boolean wasRemoved = m_classesByDeobf.remove(classMapping.getDeobfName()) != null;
+			assert (wasRemoved);
+		}
+		classMapping.setDeobfName(deobfName);
+		if (deobfName != null) {
+			boolean wasAdded = m_classesByDeobf.put(deobfName, classMapping) == null;
+			assert (wasAdded);
+		}
+	}
+	
 	public Translator getTranslator(TranslationDirection direction, TranslationIndex index) {
 		switch (direction) {
 			case Deobfuscating:
@@ -185,4 +199,18 @@ public class Mappings implements Serializable {
 		}
 		return false;
 	}
+	
+	public List<ClassMapping> getClassMappingChain(ClassEntry obfClass) {
+		List<ClassMapping> mappingChain = Lists.newArrayList();
+		ClassMapping classMapping = null;
+		for (ClassEntry obfClassEntry : obfClass.getClassChain()) {
+			if (mappingChain.isEmpty()) {
+				classMapping = m_classesByObf.get(obfClassEntry.getName());
+			} else if (classMapping != null) {
+				classMapping = classMapping.getInnerClassByObfSimple(obfClassEntry.getInnermostClassName());
+			}
+			mappingChain.add(classMapping);
+		}
+		return mappingChain;
+	}
 }
-- 
cgit v1.2.3