From 53a35332e6120ba60b8089e1fa572aa3386227f2 Mon Sep 17 00:00:00 2001
From: Thog
Date: Sat, 15 Oct 2016 17:45:06 +0200
Subject: Rewrite translate(ArgumentEntry in) logic (Fix #44)

---
 .../java/cuchaz/enigma/mapping/Translator.java     | 27 ++++++++++++++++++++--
 1 file changed, 25 insertions(+), 2 deletions(-)

diff --git a/src/main/java/cuchaz/enigma/mapping/Translator.java b/src/main/java/cuchaz/enigma/mapping/Translator.java
index bebac4e1..b05714cc 100644
--- a/src/main/java/cuchaz/enigma/mapping/Translator.java
+++ b/src/main/java/cuchaz/enigma/mapping/Translator.java
@@ -212,10 +212,33 @@ public class Translator {
     }
 
     // TODO: support not identical behavior (specific to constructor)
-    public String translate(ArgumentEntry in) {
+    public String translate(ArgumentEntry in)
+    {
+        String classTranslate = translateArgument(in);
 
+        // Not found in this class
+        if (classTranslate == null)
+        {
+            List<ClassEntry> ancestry = this.index.getAncestry(in.getClassEntry());
+
+            // Check in mother class for the arg
+            for (ClassEntry entry : ancestry)
+            {
+                ArgumentEntry motherArg = in.cloneToNewClass(entry);
+                if (this.index.entryExists(motherArg))
+                {
+                    String result = translateArgument(motherArg);
+                    if (result != null)
+                        return result;
+                }
+            }
+        }
+        return classTranslate;
+    }
+
+    public String translateArgument(ArgumentEntry in) {
         // look for identical behavior in superclasses
-        ClassEntry entry =  this.index.resolveEntryClass(in, true);
+        ClassEntry entry =  in.getClassEntry();
 
         if (entry != null)
         {
-- 
cgit v1.2.3